packages/intt/InttZVertexFinder.h

0001 // Tell emacs that this is a C++ source
0002 //  -*- C++ -*-.
0003 #ifndef INTT_INTTZVERTEXFINDER_H
0004 #define INTT_INTTZVERTEXFINDER_H
0005
0006 #include <fun4all/SubsysReco.h>
0007
0008 #include <string>
0009
0010 class PHCompositeNode;
0011 class InttVertexMap;
0012
0013 class INTTZvtx;
0014
0015 class InttZVertexFinder : public SubsysReco
0016 {
0017  public:
0018   InttZVertexFinder(const std::string &name = "InttZVertexFinder");
0019
0020   ~InttZVertexFinder() override;
0021
0022   int Init(PHCompositeNode *topNode) override;
0023
0024   int InitRun(PHCompositeNode *topNode) override;
0025
0026   int process_event(PHCompositeNode *topNode) override;
0027
0028   /// Called at the end of all processing.
0029   int End(PHCompositeNode *topNode) override;
0030
0031   void Print(const std::string &what = "ALL") const override;
0032
0033   void SetBeamCenter(const double beamx, const double beamy);
0034   void SetOutDirectory(const std::string &outDirectory);
0035
0036   void EnableQA(const bool enableQA);
0037   void EnableEventDisplay(const bool enableEvtDisp);
0038
0039  private:
0040   int createNodes(PHCompositeNode *topNode);
0041
0042  private:
0043   INTTZvtx *m_inttzvtx{nullptr};
0044   InttVertexMap *m_inttvertexmap{nullptr};
0045 };
0046
0047 #endif  // INTT_INTTZVERTEXFINDER_H