Test/RecoInfoExport/RecoInfoExport.h

0001 #ifndef __RecoInfoExport_H__
0002 #define __RecoInfoExport_H__
0003
0004 #include <fun4all/SubsysReco.h>
0005 #include <string>
0006 #include <vector>
0007 #include <TH1D.h>
0008 #include <TH2D.h>
0009
0010 class PHCompositeNode;
0011
0012 class RecoInfoExport : public SubsysReco
0013 {
0014
0015 public:
0016
0017   explicit
0018   RecoInfoExport(const std::string &name = "RecoInfoExport");
0019
0020   int
0021   Init(PHCompositeNode *topNode);
0022   int
0023   process_event(PHCompositeNode *topNode);
0024   int
0025   End(PHCompositeNode *topNode);
0026
0027   void
0028   set_file_prefix(const std::string &s)
0029   {
0030     _file_prefix = s;
0031   }
0032
0033   double
0034   get_pT_threshold() const
0035   {
0036     return _pT_threshold;
0037   }
0038
0039   void
0040   set_pT_threshold(double tThreshold)
0041   {
0042     _pT_threshold = tThreshold;
0043   }
0044
0045   double
0046   get_tower_threshold() const
0047   {
0048     return _tower_threshold;
0049   }
0050
0051   void
0052   set_tower_threshold(double towerThreshold)
0053   {
0054     _tower_threshold = towerThreshold;
0055   }
0056
0057   double
0058   get_min_track_hit_dist() const
0059   {
0060     return _min_track_hit_dist;
0061   }
0062
0063   void
0064   set_min_track_hit_dist(double minTrackHitDist)
0065   {
0066     _min_track_hit_dist = minTrackHitDist;
0067   }
0068 private:
0069
0070   int _event;
0071   std::string _file_prefix;
0072
0073   std::vector<std::string> _calo_names;
0074
0075   double _tower_threshold;
0076   double _pT_threshold;
0077   double _min_track_hit_dist;
0078 };
0079
0080 #endif // __RecoInfoExport_H__