PHTrackSeeding.cc

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file PHTrackSeeding.cc

#include "PHTrackSeeding.h"

#include <trackbase_historic/TrackSeed.h>
#include <trackbase_historic/TrackSeed_v1.h>
#include <trackbase_historic/TrackSeedContainer.h>
#include <trackbase_historic/TrackSeedContainer_v1.h>

#include <globalvertex/SvtxVertexMap.h>

#include <trackbase/TrkrClusterContainer.h>
#include <trackbase/TrkrClusterHitAssoc.h>
#include <trackbase/TrkrHitSetContainer.h>
#include <trackbase/TrkrClusterIterationMapv1.h>

#include <fun4all/Fun4AllReturnCodes.h>
#include <fun4all/SubsysReco.h>                  // for SubsysReco

#include <phool/PHCompositeNode.h>
#include <phool/PHIODataNode.h>
#include <phool/PHNode.h>                        // for PHNode
#include <phool/PHNodeIterator.h>
#include <phool/PHObject.h>                      // for PHObject
#include <phool/getClass.h>
#include <phool/phool.h>                         // for PHWHERE

#include <iostream>                              // for operator<<, endl

using namespace std;

PHTrackSeeding::PHTrackSeeding(const std::string& name)
  : SubsysReco(name)
  , _iteration_map(nullptr)
  , _n_iteration(0)
{
}

int PHTrackSeeding::InitRun(PHCompositeNode* topNode)
{
  return Setup(topNode);
}

int PHTrackSeeding::process_event(PHCompositeNode* topNode)
{
  if(_n_iteration >0){
    _iteration_map = findNode::getClass<TrkrClusterIterationMapv1>(topNode, "CLUSTER_ITERATION_MAP");
    if (!_iteration_map){
      cerr << PHWHERE << "Cluster Iteration Map missing, aborting." << endl;
      return Fun4AllReturnCodes::ABORTEVENT;
    }
  }
  return Process(topNode);
}

int PHTrackSeeding::End(PHCompositeNode* /*topNode*/)
{
  return End();
}

int PHTrackSeeding::Setup(PHCompositeNode* topNode)
{
  //cout << PHWHERE << "Entering Setup" << endl;
 
  int ret = CreateNodes(topNode);
  if (ret != Fun4AllReturnCodes::EVENT_OK) return ret;

  ret = GetNodes(topNode);
  if (ret != Fun4AllReturnCodes::EVENT_OK) return ret;

  return Fun4AllReturnCodes::EVENT_OK;
}

int PHTrackSeeding::CreateNodes(PHCompositeNode* topNode)
{
  // create nodes...
  PHNodeIterator iter(topNode);

  PHCompositeNode* dstNode = static_cast<PHCompositeNode*>(iter.findFirst(
      "PHCompositeNode", "DST"));
  if (!dstNode)
  {
    cerr << PHWHERE << "DST Node missing, doing nothing." << endl;
    return Fun4AllReturnCodes::ABORTEVENT;
  }
  PHNodeIterator iter_dst(dstNode);

  // Create the SVTX node
  PHCompositeNode* tb_node =
      dynamic_cast<PHCompositeNode*>(iter_dst.findFirst("PHCompositeNode",
                                                        "SVTX"));
  if (!tb_node)
  {
    tb_node = new PHCompositeNode("SVTX");
    dstNode->addNode(tb_node);
    if (Verbosity() > 0)
      cout << PHWHERE << "SVTX node added" << endl;
  }

  
  _track_map = findNode::getClass<TrackSeedContainer>(topNode, _track_map_name);
  if (!_track_map)
    {
      _track_map = new TrackSeedContainer_v1;
      PHIODataNode<PHObject>* tracks_node = 
        new PHIODataNode<PHObject>(_track_map, _track_map_name, "PHObject");
      tb_node->addNode(tracks_node);
      if (Verbosity() > 0){
        cout << PHWHERE << "Svtx/" <<_track_map_name  << " node added" << endl;
      }
    }
  if(Verbosity() > 0)
    _track_map->identify();

  return Fun4AllReturnCodes::EVENT_OK;
}

int PHTrackSeeding::GetNodes(PHCompositeNode* topNode)
{
  //---------------------------------
  // Get Objects off of the Node Tree
  //---------------------------------
  if(_use_truth_clusters)
    _cluster_map = findNode::getClass<TrkrClusterContainer>(topNode, "TRKR_CLUSTER_TRUTH");
  else
    _cluster_map = findNode::getClass<TrkrClusterContainer>(topNode, "TRKR_CLUSTER");

  if (!_cluster_map)
  {
    cerr << PHWHERE << " ERROR: Can't find node TRKR_CLUSTER" << endl;
    return Fun4AllReturnCodes::ABORTEVENT;
  }

  if(do_hit_assoc){
    _cluster_hit_map = findNode::getClass<TrkrClusterHitAssoc>(topNode, "TRKR_CLUSTERHITASSOC");
    if (!_cluster_hit_map)
      {
        cerr << PHWHERE << " ERROR: Can't find node TRKR_CLUSTERHITASSOC" << endl;
      }
  }
  _track_map = findNode::getClass<TrackSeedContainer>(topNode, _track_map_name);
  if (!_track_map)
  {
    cerr << PHWHERE << " ERROR: Can't find " << _track_map_name << endl;
    return Fun4AllReturnCodes::ABORTEVENT;
  }
  /*
  _hitsets = findNode::getClass<TrkrHitSetContainer>(topNode, "TRKR_HITSET");
  if (!_hitsets)
  {
    cerr << PHWHERE << " ERROR: Can't find TrkrHitSetContainer." << endl;
    return Fun4AllReturnCodes::ABORTEVENT;
  }
  */
  return Fun4AllReturnCodes::EVENT_OK;
}