JCORRAN code update from DongJo

[u/mrichter/AliRoot.git] / PWGCF / Correlations / JCORRAN / AliJCORRANTask.h

diff --git a/PWGCF/Correlations/JCORRAN/AliJCORRANTask.h b/PWGCF/Correlations/JCORRAN/AliJCORRANTask.h
old mode 100755 (executable)
new mode 100644 (file)
index 0ae9997..abea78a
--- a/PWGCF/Correlations/JCORRAN/AliJCORRANTask.h
+++ b/PWGCF/Correlations/JCORRAN/AliJCORRANTask.h
@@ -32,17 +32,16 @@
 #include "AliESDZDC.h"
 #include "AliJConst.h"
 #include "AliESDpid.h"
-#include "AliEMCALGeometry.h"
 #include "AliPHOSGeoUtils.h"
 #include "AliPIDResponse.h"
 #include "AliPIDCombined.h"
+#include "AliJFilter.h"
+#include "AliJEfficiencyScanner.h"
 
 //==============================================================
 
 using namespace std;
 
-const int kMaxDimBuffer = 300;//max length of a line read to a buffe
-
 class TH1D;
 class TH2D;
 class TNtuple;
@@ -77,88 +76,31 @@ class AliJCORRANTask : public AliAnalysisTaskSE {
 
   // methods to fill from AliAnalysisTaskSE
   virtual void UserCreateOutputObjects(); 
-  virtual void Init();         
+  virtual void Init();   
   virtual void LocalInit() { Init(); }
   virtual void UserExec(Option_t *option);
-  virtual void Terminate(Option_t * opt = "");
-
-  TString GetRunType() const { return fRunType;}
-  void SetRunType( const TString type ){ fRunType = type; }
-  void SetESDtrackCuts(AliESDtrackCuts* esdTrackCuts){ fEsdTrackCuts = esdTrackCuts;}
-  void SetESDFilter( AliAnalysisFilter * filter ){ fESDFilter = filter; }
-  void SetRealOrMC(Bool_t realormc){fIsRealOrMC[0]=realormc;} //flags whether the input 
-  void SetStoreEventPlaneSource(bool dostore ){ fStoreEventPlaneSource = dostore; }
-  bool GetStoreEventPlaneSource(){ return fStoreEventPlaneSource; };
-  void SetStoreTPCTrack(bool dostore ){ fStoreTPCTrack = dostore; }
-  bool GetStoreTPCTrack(){ return fStoreTPCTrack; };
-  //are ESDs from real  exp or MonteCarlo 
+  virtual void Terminate(Option_t* option="");
+
+  AliJFilter *GetFilter() { return fFilter; }
+
+  AliJRunHeader * GetJRunHeader(){ return fAliJRunHeader; }
+  void SetJRunHeader( AliJRunHeader * hdr ){ fAliJRunHeader = hdr ; }
   void SetOutputAODName(const char* aodname){ fAODName=aodname;}
-  //  AliEMCALGeoUtils* GetEMCALGeoUtils (bool doDelete=kFALSE);
+  TString *GetOutputAODName() {return &fAODName;}
+
+  void SetDoStoreJOD(Bool_t doeff){ fDoStoreJOD=doeff; }
+  Bool_t GetDoStoreJOD(){ return fDoStoreJOD; }
 
  private:
-  AliJEventHeader* ReadCommonHeader(AliVEvent *event);
-  // methods to read data from ESD
-  void ReadESDTracks(AliESDEvent* esd);
-  void ReadESDCaloClusters(const AliESDEvent* esd);
-  void ReadESDHeader(AliESDEvent* esd);
-  void ReadESDPID(AliESDtrack* track, AliJTrack* ctrack);
-  // methods to read data from AOD
-  void ReadAODTracks(const AliAODEvent* aod);
-  void ReadAODCaloClusters(const AliAODEvent* aod);
-  void ReadAODHeader(AliAODEvent* aod);
-  void ReadFilter();
-  void ReadMCTracks(AliMCEvent* fMC);
-  Int_t GetSuperModuleNumber(bool isemcal, Int_t absId);
-
-  UInt_t ConvertTriggerMask();//Converts alice trigger mask to JCorran trigger mask
-  //functions used for event selction:
-  bool AcceptAODTrack(AliAODTrack* aodTrack);
-  void SetOADBPath(const char* path) {fOADBPath=path;}
-  const char* GetOADBPath() const { return fOADBPath.Data(); }
-
-  // method to fill jcorran
-  bool SetAliceTriggerDef(AliJRunHeader *runHeader);
-  bool SetAliceFilterMapDef(AliJRunHeader *runHeader); //TODO Check
-  void PrintOut();
   
-  // UTILS
-  void AddListAODBranch(const char* aname, const char* cname, TClonesArray **obj, int nlist);
-
-  // d a t a     m e m b e r s
-  TString fRunType;   // ex) LHC10h
-  TString fInputFormat; // specify the input data format (ESD or AOD)
-  AliESDtrackCuts* fEsdTrackCuts; //track selection cuts
-  AliAnalysisFilter * fESDFilter; //filter set of track selection BS
-  TVectorT<double>  fIsRealOrMC; //flags if the input are real (0) ESDs or MonteCarlo ESDs (1)
+  AliJFilter *fFilter; // filter object
   TString fAODName; //output delta AOD name
-  TString fActiveTriggers[AliJConst::kRangeTriggerTableAlice];//alice table mapping trigger bit to trigger name
-  TString fTriggerTableJCorran[AliJConst::kRangeTriggerTableJCorran];//JCorran trigger table TBit 0 =MinBias
-  bool fStoreEventPlaneSource;
-  bool fStoreTPCTrack;
-  TString fOADBPath;
-
-  // jcorran output objects
-
-  TClonesArray *    fTrackList;   // list of charged track objects
-  TClonesArray *    fMCTrackList; // list of charged track objects
-  TClonesArray *    fPhotonList;  // list of photons objects
-  TClonesArray *    fHeaderList;  // event details
-  TList *          fRunInfoList; // run details
-
-  AliESDpid          *fPIDesd;
-  AliPIDResponse  *fPIDResponse; // PID response object
-  AliPIDCombined  *fPIDCombined;
-
-  AliESDVZERO*        fVZEROData;
-  AliESDTZERO*        fTZEROData;
-  //  AliESDFMD*          fFMDData;
-  AliESDZDC*          fZDCData;
+  TTree      *fJODTree;
+  AliJRunHeader * fAliJRunHeader;
 
-  AliJRunHeader*      fAliRunHeader;//  run details (mg field, trigger mask,etc...)
-  AliEMCALGeometry * fEMCALGeoUtils; // no AliEMCALGeoUtils.h in trunk aliroot (111130)
-  AliPHOSGeoUtils  * fPHOSGeom; //phos geometry matrix 
+  Bool_t fDoStoreJOD;
 
 
-  ClassDef(AliJCORRANTask, 1); 
+  ClassDef(AliJCORRANTask, 2); 
 };
 #endif // AliJCORRANTask_H