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/TrkrDefs.h>
#include <trackbase_historic/ActsTransformations.h>

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

struct ActsSurfaceMaps;
struct ActsTrackingGeometry;
class PHG4CylinderCellGeomContainer;
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;

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

  int get_embed(PHG4Particle*) const;

  TrackEvaluationContainerv1::ClusterStruct create_cluster( TrkrDefs::cluskey, TrkrCluster* ) 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&, std::set<PHG4Hit*> ) const;

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

  TrackEvaluationContainerv1* m_container = nullptr;

  int m_flags = EvalEvent | EvalClusters | EvalTracks;

  ActsSurfaceMaps *m_surfmaps = nullptr;

  ActsTrackingGeometry *m_tGeometry = nullptr;

  ActsTransformations m_transformer;

  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;

  PHG4CylinderCellGeomContainer* 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;

};

#endif  // G4EVAL_TrackEvaluation_H