SQTruthVertexing.h

Go to the documentation of this file.

#ifndef _SQTRUTHVERTEXING_H
#define _SQTRUTHVERTEXING_H
 
#include <fun4all/SubsysReco.h>
#include <TRandom1.h>
 
class TVector3;
class SQTrack;
class SQTrackVector;
class SQDimuonVector;
class SRecEvent;
class SRecTrack;
class SRecDimuon;
 
 
class SQTruthVertexing: public SubsysReco
{
public:
  SQTruthVertexing(const std::string& name = "SQTruthVertexing");
  ~SQTruthVertexing();
 
  int Init(PHCompositeNode* topNode);
  int InitRun(PHCompositeNode* topNode);
  int process_event(PHCompositeNode* topNode);
  int End(PHCompositeNode* topNode);
 
  void set_legacy_rec_container(const bool enable = true) { legacyContainer = enable; }
  void set_vtx_smearing(const double r) { vtxSmearing = r; }
 
private:
  int MakeNodes(PHCompositeNode* topNode);
  int GetNodes(PHCompositeNode* topNode);
 
  bool buildRecDimuon(double z_vtx, SRecTrack* posTrack, SRecTrack* negTrack, SRecDimuon* dimuon);
  double swimTrackToVertex(SRecTrack* track, double z, TVector3* pos = nullptr, TVector3* mom = nullptr);
 
  TRandom1 rndm;
 
  bool legacyContainer;
  double vtxSmearing;
 
  SRecEvent*      recEvent;
  SQTrackVector*  recTrackVec;
  SQTrackVector*  truthTrackVec;
  SQDimuonVector* truthDimuonVec;
 
  SQDimuonVector* recDimuonVec;
};
 
#endif