PHTruthSiliconAssociation.h

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

// Tell emacs that this is a C++ source
//  -*- C++ -*-.
#ifndef PHTRUTHSILICONASSOCIATION_H
#define PHTRUTHSILICONASSOCIATION_H

#include <fun4all/SubsysReco.h>
#include <trackbase/TrkrDefs.h>

#include <string>
#include <set>
#include <vector>

class PHCompositeNode;
class SvtxTrackMap;
class SvtxTrack;
class SvtxVertexMap;
class TrkrClusterContainer;
class TrkrHitSetContainer;
class TrkrClusterHitAssoc;
class TrkrHitTruthAssoc;
class PHG4TruthInfoContainer;
class PHG4HitContainer;
class PHG4Particle;
class AssocInfoContainer;
class TpcSeedTrackMap;

class PHTruthSiliconAssociation : public SubsysReco
{
 public:

  PHTruthSiliconAssociation(const std::string &name = "PHTruthSiliconAssociation");

  ~PHTruthSiliconAssociation() override;

  int Init(PHCompositeNode *topNode) override;

  int InitRun(PHCompositeNode *topNode) override;

  int process_event(PHCompositeNode *topNode) override;

  int ResetEvent(PHCompositeNode *topNode) override;

  int EndRun(const int runnumber) override;

  int End(PHCompositeNode *topNode) override;

  int Reset(PHCompositeNode * /*topNode*/) override;

  void Print(const std::string &what = "ALL") const override;
  
 private:

  int GetNodes(PHCompositeNode* topNode);
  void copySiliconClustersToCorrectedMap( );

  std::vector<PHG4Particle*> getG4PrimaryParticle(SvtxTrack *track);
  std::set<TrkrDefs::cluskey> getSiliconClustersFromParticle(PHG4Particle* g4particle);
  
  PHG4TruthInfoContainer* _g4truth_container{nullptr};
  PHG4HitContainer *_g4hits_tpc{nullptr};
  PHG4HitContainer *_g4hits_mvtx{nullptr};
  PHG4HitContainer *_g4hits_intt{nullptr};
  
  TrkrClusterContainer *_cluster_map{nullptr};
  TrkrClusterContainer *_corrected_cluster_map{nullptr};
  TrkrHitSetContainer  *_hitsets{nullptr};
  TrkrClusterHitAssoc *_cluster_hit_map{nullptr};
  TrkrHitTruthAssoc *_hit_truth_map{nullptr};
  SvtxTrackMap *_track_map{nullptr};
  AssocInfoContainer *_assoc_container{nullptr};
  SvtxTrack *_tracklet{nullptr};
  SvtxVertexMap * _vertex_map{nullptr};
  TpcSeedTrackMap *_seed_track_map{nullptr};

 std::string _tpcseed_track_map_name = "TpcSeedTrackMap";

};

#endif // PHTRUTHSILICONASSOCIATION_H