QAG4SimulationMvtx.h

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

#ifndef QA_QAG4SIMULATIONMVTX_H
#define QA_QAG4SIMULATIONMVTX_H

#include <trackbase/ClusterErrorPara.h>
#include <trackbase/TrkrDefs.h>

#include <fun4all/SubsysReco.h>

#include <set>
#include <string>

class PHCompositeNode;
class PHG4Hit;
class PHG4HitContainer;
class TrkrClusterContainer;
class TrkrClusterHitAssoc;
class TrkrHitTruthAssoc;
class ActsGeometry;

class QAG4SimulationMvtx : public SubsysReco
{
 public:
  QAG4SimulationMvtx(const std::string& name = "QAG4SimulationMvtx");

  int InitRun(PHCompositeNode* topNode) override;
  int process_event(PHCompositeNode* topNode) override;
 
 private:
  std::string get_histo_prefix() const;

  int load_nodes(PHCompositeNode*);

  void evaluate_clusters();

  // get geant hits associated to a cluster
  using G4HitSet = std::set<PHG4Hit*>;
  G4HitSet find_g4hits(TrkrDefs::cluskey) const;

  bool m_initialized = false;

  TrkrClusterContainer* m_cluster_map = nullptr;

  TrkrClusterHitAssoc* m_cluster_hit_map = nullptr;

  TrkrHitTruthAssoc* m_hit_truth_map = nullptr;

  ActsGeometry* m_tGeometry = nullptr;

  PHG4HitContainer* m_g4hits_mvtx = nullptr;

  /* it is filled at Init stage. It should not change for the full run */
  std::set<int> m_layers;
  ClusterErrorPara _ClusErrPara;
};

#endif