TrackEvaluation.h

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

#ifndef G4EVAL_TRACKEVALUATION_H
#define G4EVAL_TRACKEVALUATION_H

#include "TrackEvaluationContainerv1.h"

#include <fun4all/SubsysReco.h>

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

#include <trackbase_historic/SvtxTrackState.h>

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

class ActsGeometry;
class PHG4TpcCylinderGeomContainer;
class PHG4CylinderGeomContainer;
class PHG4Hit;
class PHG4HitContainer;
class PHG4Particle;
class PHG4TruthInfoContainer;
class SvtxTrack;
class SvtxTrackMap;
class TrkrCluster;
class TrkrClusterContainer;
class TrkrClusterHitAssoc;
class TrkrHitSetContainer;
class TrkrHitTruthAssoc;

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

  int Init(PHCompositeNode*) override;

  int InitRun(PHCompositeNode*) override;

  int process_event(PHCompositeNode*) override;

  int End(PHCompositeNode*) override;

  enum Flags
  {
    EvalEvent = 1 << 0,
    EvalClusters = 1 << 1,
    EvalTracks = 1 << 2
  };

  void set_flags(int flags)
  {
    m_flags = flags;
  }

 private:
  int load_nodes(PHCompositeNode*);

  void evaluate_event();

  void evaluate_clusters();

  void evaluate_tracks();

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

  std::pair<int, int> get_max_contributor(SvtxTrack*) const;

  int get_embed(PHG4Particle*) const;

  TrackEvaluationContainerv1::ClusterStruct create_cluster(TrkrDefs::cluskey, TrkrCluster*, SvtxTrack*) const;

  void add_trk_information(TrackEvaluationContainerv1::ClusterStruct&, SvtxTrackState*) const;


  void add_trk_information_micromegas(TrackEvaluationContainerv1::ClusterStruct&, int /* tileid */, SvtxTrackState*) const;

  // add truth information
  void add_truth_information(TrackEvaluationContainerv1::ClusterStruct&, const std::set<PHG4Hit*>&) const;

  // add truth information
  void add_truth_information_micromegas(TrackEvaluationContainerv1::ClusterStruct&, int /* tileid */, const std::set<PHG4Hit*>&) const;

  TrackEvaluationContainerv1* m_container = nullptr;

  int m_flags = EvalEvent | EvalClusters | EvalTracks;

  ActsGeometry* m_tGeometry = nullptr;

  TrkrHitSetContainer* m_hitsetcontainer = nullptr;

  TrkrClusterContainer* m_cluster_map = nullptr;

  TrkrClusterHitAssoc* m_cluster_hit_map = nullptr;

  TrkrHitTruthAssoc* m_hit_truth_map = nullptr;

  SvtxTrackMap* m_track_map = nullptr;


  PHG4HitContainer* m_g4hits_tpc = nullptr;
  PHG4HitContainer* m_g4hits_intt = nullptr;
  PHG4HitContainer* m_g4hits_mvtx = nullptr;
  PHG4HitContainer* m_g4hits_micromegas = nullptr;

  PHG4TruthInfoContainer* m_g4truthinfo = nullptr;

  PHG4TpcCylinderGeomContainer* m_tpc_geom_container = nullptr;

  PHG4CylinderGeomContainer* m_micromegas_geom_container = nullptr;

  // map cluster keys to g4hits
  using G4HitMap = std::map<TrkrDefs::cluskey, G4HitSet>;
  mutable G4HitMap m_g4hit_map;
  ClusterErrorPara _ClusErrPara;
};

#endif  // G4EVAL_TrackEvaluation_H