qaReader_smash_root.h 1.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354
  1. #ifndef QATOOLS_READERS_SMASH_ROOT_H
  2. #define QATOOLS_READERS_SMASH_ROOT_H
  3. #include <Rtypes.h>
  4. #include <TChain.h>
  5. #include <qaEvent.h>
  6. #include <qaParticle.h>
  7. #include <Utility.h>
  8. #include <qaReader_manager.h>
  9. #define MAX_TRACKS 15000
  10. class qaReader_smash_root : virtual public qaReader_manager
  11. {
  12. private:
  13. TChain *fChain;
  14. const char *fChainName = "particles";
  15. bool is_init;
  16. Long64_t fCurrentEvent;
  17. Int_t npart;
  18. Double_t impact_b;
  19. Int_t ev;
  20. Int_t tcounter;
  21. Int_t pdgcode[MAX_TRACKS]; //[npart]
  22. Double_t p0[MAX_TRACKS]; //[npart]
  23. Double_t px[MAX_TRACKS]; //[npart]
  24. Double_t py[MAX_TRACKS]; //[npart]
  25. Double_t pz[MAX_TRACKS]; //[npart]
  26. Double_t t[MAX_TRACKS]; //[npart]
  27. Double_t x[MAX_TRACKS]; //[npart]
  28. Double_t y[MAX_TRACKS]; //[npart]
  29. Double_t z[MAX_TRACKS]; //[npart]
  30. Bool_t ChainCheck();
  31. public:
  32. qaReader_smash_root(/* args */);
  33. virtual ~qaReader_smash_root();
  34. virtual void SetChain(const TString &inputFileName);
  35. virtual Long64_t GetEntries() { return fChain->GetEntries(); }
  36. virtual qaEvent *ReadEvent(Long64_t iev);
  37. virtual qaParticle* ReadParticle(Int_t ipart);
  38. ClassDef(qaReader_smash_root, 0);
  39. };
  40. #endif