StO97DstMaker.h 1.7 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283
  1. //
  2. // The version of the class should be changed every time
  3. // when any changes in the codes are done
  4. // Grigory Nigmatkulov: 2016/12/15
  5. //
  6. // Version 2 Nikita E:
  7. // Added transverse sphericity
  8. //
  9. #ifndef STO97DSTMAKER_H
  10. #define STO97DSTMAKER_H
  11. /*
  12. * General includes
  13. */
  14. #include "StO97Track.h"
  15. #include "StO97Event.h"
  16. #include "StMaker.h"
  17. #include "TTree.h"
  18. #include "TFile.h"
  19. #include "TChain.h"
  20. #include "TSystem.h"
  21. #include "TBranch.h"
  22. #include "TDatime.h"
  23. #include "TLorentzVector.h"
  24. /*
  25. * Includes for transverse sphericity calculations
  26. */
  27. #include <TVectorT.h>
  28. #include <TMatrixTLazy.h>
  29. #include <TMatrixDSymEigen.h>
  30. //_________________
  31. class StO97DstMaker : public StMaker {
  32. private:
  33. const Char_t *mOutFileName;
  34. const Char_t *mInFileName;
  35. TFile *mOutFile;
  36. FILE *mInFile;
  37. TTree *mTree;
  38. Int_t mCompression;
  39. Float_t mSperp; // For primary tracks with pT>0.15 GeV and |eta|<0.5
  40. Float_t mSperp2; // For primary tracks with pT>0.15 GeV and |eta|<1.0
  41. TMatrixTSym<double> *matrix; // transverse sphericity matrix for mSperp
  42. TMatrixTSym<double> *matrix2; // transverse sphericity matrix for mSperp2
  43. Bool_t mStop;
  44. Int_t mNBytes;
  45. Int_t mNEvents;
  46. Int_t mNTracks;
  47. char mBuf[128]; // buffer for general read purposes
  48. StO97Event *mEvent;
  49. Bool_t ReadEvent();
  50. Bool_t ReadTracks();
  51. public:
  52. StO97DstMaker(const Char_t *iFileName,
  53. const Char_t *oFileName);
  54. ~StO97DstMaker();
  55. void Clear(Option_t *option = "") { StMaker::Clear(option); }
  56. Int_t Init();
  57. Int_t Make();
  58. Int_t Finish();
  59. Bool_t IsStop() { return mStop; }
  60. void SetCompressionlevel(Int_t comp);
  61. ClassDef(StO97DstMaker, 2)
  62. };
  63. #endif