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 "DSTCompressor.h"

#include <trackbase/ActsGeometry.h>
#include <trackbase/TrkrDefs.h>
#include <trackbase_historic/ActsTransformations.h>

#include <fun4all/SubsysReco.h>

#include <TRandom.h>

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

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::Vector3 getGlobalPosition(TrkrDefs::cluskey, 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;

  ActsGeometry* 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