QAG4SimulationTpc.h

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

#ifndef QA_QAG4SIMULATIONTPC_H
#define QA_QAG4SIMULATIONTPC_H

#include <g4eval/SvtxEvalStack.h>  // for SvtxEvalStack

#include <trackbase/TrkrDefs.h>

#include <fun4all/SubsysReco.h>

#include <map>
#include <memory>
#include <set>
#include <string>

class PHCompositeNode;
class PHG4Hit;
class PHG4HitContainer;
class PHG4TruthInfoContainer;
class TrkrClusterContainer;
class TrkrClusterHitAssoc;
class TrkrHitTruthAssoc;
class SvtxEvalStack;
struct ActsSurfaceMaps;
struct ActsTrackingGeometry;

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

  int InitRun(PHCompositeNode* topNode) override;
  int process_event(PHCompositeNode* topNode) override;

 private:
  std::string get_histo_prefix() const;

  std::unique_ptr<SvtxEvalStack> m_svtxEvalStack;
  PHG4TruthInfoContainer* m_truthContainer;

  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;

  ActsSurfaceMaps* m_surfmaps = nullptr;

  ActsTrackingGeometry* m_tGeometry = nullptr;

  TrkrClusterContainer* m_cluster_map = nullptr;

  TrkrClusterHitAssoc* m_cluster_hit_map = nullptr;

  TrkrHitTruthAssoc* m_hit_truth_map = nullptr;

  PHG4HitContainer* m_g4hits_tpc = nullptr;

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

#endif