DSTEmulator.h

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

#ifndef G4EVAL_DSTEMULATOR_H
#define G4EVAL_DSTEMULATOR_H

#include <fun4all/SubsysReco.h>
#include <trackbase/TrkrDefs.h>
#include <trackbase/ActsSurfaceMaps.h>
#include <trackbase/ActsTrackingGeometry.h>
#include <trackbase_historic/ActsTransformations.h>

#include <map>
#include <set>
#include <string>
#include <vector>
#include <TRandom.h>

#include "DSTCompressor.h"

class PHG4Hit;
class PHG4HitContainer;
class PHG4Particle;
class PHG4TruthInfoContainer;
class SvtxTrack;
class SvtxTrackMap;
class TrackEvaluationContainerv1;
class TrkrCluster;
class TrkrClusterContainer;
class TrkrClusterHitAssoc;
class TrkrHitSetContainer;
class TrkrHitTruthAssoc;

class TFile;
class TNtuple;

class DSTEmulator : public SubsysReco
{
  public:

  DSTEmulator( const std::string& = "DSTEmulator",
               const std::string &filename = "DSTana.root",
               int nBits = 8,
               int sabotage = 0,
               bool compress = true);

  int Init(PHCompositeNode*) override;

  int InitRun(PHCompositeNode*) override;

  int process_event(PHCompositeNode*) override;

  int End(PHCompositeNode*) override;

  private:

  int load_nodes( PHCompositeNode* );

  void evaluate_tracks();
  float compress_dx(float in_val);
  float compress_dy(float in_val);
  // get geant hits associated to a cluster
  using G4HitSet = std::set<PHG4Hit*>;
  G4HitSet find_g4hits( TrkrDefs::cluskey ) const;
  Acts::Vector3D getGlobalPosition(TrkrCluster*) const;
  std::pair<int,int> get_max_contributor( SvtxTrack* ) const;

  int get_embed(PHG4Particle*) const;

  TrackEvaluationContainerv1* m_container = 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;

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

  ActsSurfaceMaps *m_surfMaps = nullptr;
  ActsTrackingGeometry *m_tGeometry = nullptr;
  ActsTransformations m_transform;

  TNtuple *_dst_data;

  // output file
  std::string _filename;
  TFile *_tfile;

  DSTCompressor* m_compressor;

  // Number of bits for the integer representation after compression
  int nBits = 8;
  // replace the decompressed residuals by a large number
  int sabotage = 0;
  // random seed
  TRandom rnd;
  // switch to apply the compressed residuals to cluster positions
  bool apply_compression = true;

};

#endif  // G4EVAL_DSTEMULATOR_H