TopoclusterJets/src/EventSelection.h

0001 // Tell emacs that this is a C++ source
0002 //  -*- C++ -*-.
0003 #ifndef EVENTSELECTION_H
0004 #define EVENTSELECTION_H
0005
0006 #include <fun4all/SubsysReco.h>
0007 #include <jetbase/Jetv1.h>
0008
0009 #include <memory>
0010 #include <string>
0011 #include <utility>  // std::pair, std::make_pair
0012 #include <vector>
0013
0014 #include <TFile.h>
0015 class PHCompositeNode;
0016 class TTree;
0017
0018 class EventSelection : public SubsysReco
0019 {
0020 public:
0021     // constructor
0022   EventSelection( const std::string &recojetname = "AntiKt_Tower_r04_Sub1",
0023            const std::string &outputfilename = "selecteventtest.root");
0024
0025     ~EventSelection() override; // destructor
0026
0027     void setVzCut(double cut) { m_vtxZ_cut = cut; }
0028
0029     // Standard Fun4All functions
0030     int Init(PHCompositeNode *topNode) override;
0031     int InitRun(PHCompositeNode *topNode) override;
0032     int process_event(PHCompositeNode *topNode) override;
0033     int ResetEvent(PHCompositeNode *topNode) override;
0034     int EndRun(const int runnumber) override;
0035     int End(PHCompositeNode *topNode) override;
0036     int Reset(PHCompositeNode * /*topNode*/) override;
0037
0038     void Print(const std::string &what = "ALL") const override;
0039
0040 private:
0041
0042     // private variables
0043
0044     std::string m_outputfilename;
0045
0046     TFile *outFile;
0047     TTree *m_tree;
0048
0049     double m_vtxZ_cut;
0050     int m_event;
0051     float m_vertex_z;
0052
0053 };
0054
0055 #endif // EVENTSELECTION_H