qaWriter_analysistree2.h 1.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354
  1. #ifndef QATOOLS_WRITERS_ANALYSISTREE2_H
  2. #define QATOOLS_WRITERS_ANALYSISTREE2_H
  3. #include <Rtypes.h>
  4. #include <TString.h>
  5. #include <TFile.h>
  6. #include <TTree.h>
  7. #include <TDatabasePDG.h>
  8. #include <TParticlePDG.h>
  9. #include <qaEvent.h>
  10. #include <qaParticle.h>
  11. #include <Utility.h>
  12. #include <qaWriter_manager.h>
  13. // AnalysisTree headers
  14. #include <AnalysisTree/Configuration.hpp>
  15. #include <AnalysisTree/DataHeader.hpp>
  16. #include <AnalysisTree/EventHeader.hpp>
  17. #include <AnalysisTree/Detector.hpp>
  18. #include <AnalysisTree/Matching.hpp>
  19. class qaWriter_analysistree2 : virtual public qaWriter_manager
  20. {
  21. private:
  22. /* data */
  23. TFile *fFile;
  24. TTree *fTree;
  25. AnalysisTree::DataHeader *fDataHeader;
  26. AnalysisTree::Configuration *fConfig;
  27. AnalysisTree::EventHeader *fEvent;
  28. AnalysisTree::Particles *fParticles;
  29. std::string str_event_branch;
  30. std::string str_particles_branch;
  31. int iB, iPhiRp, icharge;
  32. bool isInit;
  33. public:
  34. qaWriter_analysistree2();
  35. virtual ~qaWriter_analysistree2();
  36. virtual void Init(std::string filename, std::string treename, std::string system, float sqrtSnn);
  37. virtual void WriteEvent(qaEvent* event);
  38. virtual void WriteParticle(qaParticle* particle);
  39. virtual void WriteTree();
  40. virtual void FillTree();
  41. ClassDef(qaWriter_analysistree2,0);
  42. };
  43. #endif