increase class def
[u/mrichter/AliRoot.git] / JETAN / DEV / AliAnalysisTaskJetCluster.h
index 93d7b27..194651f 100644 (file)
@@ -34,6 +34,7 @@ class TProfile;
 class TRandom3;
 class TRefArray;
 class TClonesArray;
 class TRandom3;
 class TRefArray;
 class TClonesArray;
+class TF1;
 
 class AliAnalysisTaskJetCluster : public AliAnalysisTaskSE
 {
 
 class AliAnalysisTaskJetCluster : public AliAnalysisTaskSE
 {
@@ -43,14 +44,12 @@ class AliAnalysisTaskJetCluster : public AliAnalysisTaskSE
     virtual ~AliAnalysisTaskJetCluster();
     // Implementation of interface methods
     virtual void UserCreateOutputObjects();
     virtual ~AliAnalysisTaskJetCluster();
     // Implementation of interface methods
     virtual void UserCreateOutputObjects();
-    virtual void Init();
-    virtual void LocalInit() { Init(); }
+    virtual void LocalInit();
     virtual void UserExec(Option_t *option);
     virtual void Terminate(Option_t *option);
     virtual Bool_t Notify();
 
     
     virtual void UserExec(Option_t *option);
     virtual void Terminate(Option_t *option);
     virtual Bool_t Notify();
 
     
-
     virtual void SetAODTrackInput(Bool_t b){fUseAODTrackInput = b;}
     virtual void SetAODMCInput(Bool_t b){fUseAODMCInput = b;}
     virtual void SetEventSelection(Bool_t b){fEventSelection = b;}
     virtual void SetAODTrackInput(Bool_t b){fUseAODTrackInput = b;}
     virtual void SetAODMCInput(Bool_t b){fUseAODMCInput = b;}
     virtual void SetEventSelection(Bool_t b){fEventSelection = b;}
@@ -62,6 +61,8 @@ class AliAnalysisTaskJetCluster : public AliAnalysisTaskSE
     virtual void SetCentralityCut(Float_t xLo,Float_t xUp){fCentCutLo = xLo; fCentCutUp = xUp;}
     virtual void SetFilterMask(UInt_t i,Int_t iType = 0){fFilterMask = i;
       fFilterType = iType;}
     virtual void SetCentralityCut(Float_t xLo,Float_t xUp){fCentCutLo = xLo; fCentCutUp = xUp;}
     virtual void SetFilterMask(UInt_t i,Int_t iType = 0){fFilterMask = i;
       fFilterType = iType;}
+    virtual void SetFilterMaskBestPt(UInt_t i){fFilterMaskBestPt = i;}
+
     virtual void SetJetTypes(UInt_t i){fJetTypes = i;}
     virtual void SetJetTriggerPtCut(Float_t x){fJetTriggerPtCut = x;}    
     virtual void SetVtxCuts(Float_t z,Float_t r = 1){fVtxZCut = z; fVtxR2Cut = r *r;}    
     virtual void SetJetTypes(UInt_t i){fJetTypes = i;}
     virtual void SetJetTriggerPtCut(Float_t x){fJetTriggerPtCut = x;}    
     virtual void SetVtxCuts(Float_t z,Float_t r = 1){fVtxZCut = z; fVtxR2Cut = r *r;}    
@@ -78,7 +79,22 @@ class AliAnalysisTaskJetCluster : public AliAnalysisTaskSE
     virtual void SetMaxTrackPtInJet(Float_t x){fMaxTrackPtInJet = x;}
     virtual void SetJetOutputMinPt(Float_t x){fJetOutputMinPt = x;}
     virtual void SetBackgroundCalc(Bool_t b){fUseBackgroundCalc = b;} 
     virtual void SetMaxTrackPtInJet(Float_t x){fMaxTrackPtInJet = x;}
     virtual void SetJetOutputMinPt(Float_t x){fJetOutputMinPt = x;}
     virtual void SetBackgroundCalc(Bool_t b){fUseBackgroundCalc = b;} 
+    virtual void SetStoreRhoLeadingTrackCorr(Bool_t b) {fStoreRhoLeadingTrackCorr=b;}
+
+    //Setters for detector level effects
+    virtual void SetUseTrResolutionFromOADB(Bool_t b=kTRUE, TString path="$ALICE_ROOT/OADB/PWGJE/Resolution/PtResol_LHCh_Cent0-10_v1.root") {fUseTrPtResolutionFromOADB = b; fPathTrPtResolution=path;}
+    virtual void SetUseTrEfficiencyFromOADB(Bool_t b=kTRUE, TString path="$ALICE_ROOT/OADB/PWGJE/Efficiency/Efficiency_LHC11a2aj_Cent0_v1.root") {fUseTrEfficiencyFromOADB = b; fPathTrEfficiency=path;}
+    virtual void LoadTrEfficiencyRootFileFromOADB();
+    virtual void LoadTrPtResolutionRootFileFromOADB();
+    virtual void SetChangeEfficiencyFraction(Double_t p) {fChangeEfficiencyFraction = p;}
+    virtual void SetSmearResolution(Bool_t b){fUseTrPtResolutionSmearing = b;} 
+    virtual void SetDiceEfficiency(Bool_t b){fUseDiceEfficiency = b;} 
+    virtual void SetDiceEfficiencyMinPt(Double_t pt) {fDiceEfficiencyMinPt = pt;}
+    virtual void SetMomentumResolutionHybrid(TProfile *p1, TProfile *p2, TProfile *p3);
+    virtual void SetEfficiencyHybrid(TH1 *h1, TH1 *h2, TH1 *h3);
 
 
+    Double_t GetMomentumSmearing(Int_t cat, Double_t pt);
+    void FitMomentumResolution();
 
 
     // for Fast Jet
 
 
     // for Fast Jet
@@ -132,8 +148,10 @@ class AliAnalysisTaskJetCluster : public AliAnalysisTaskSE
     Bool_t        fUseBackgroundCalc;     // switches on background calculations
     Bool_t        fEventSelection;        // use the event selection of this task, otherwise analyse all
     UInt_t        fFilterMask;            // filter bit for slecected tracks
     Bool_t        fUseBackgroundCalc;     // switches on background calculations
     Bool_t        fEventSelection;        // use the event selection of this task, otherwise analyse all
     UInt_t        fFilterMask;            // filter bit for slecected tracks
+    UInt_t        fFilterMaskBestPt;      // filter bit to mark jets with high quality leading tracks
+
     UInt_t        fFilterType;            // filter type 0 = all, 1 = ITSTPC, 2 = TPC
     UInt_t        fFilterType;            // filter type 0 = all, 1 = ITSTPC, 2 = TPC
-    UInt_t        fJetTypes;              // 1<<0 regular jets, 1<<1 << randomized event 1<<2 random cones 1<<3 random cones randomiuzed evetn
+    UInt_t        fJetTypes;              // 1<<0 regular jets, 1<<1 << randomized event 1<<2 random cones 1<<3 random cones randomiuzed event
     Int_t         fTrackTypeRec;          // type of tracks used for FF 
     Int_t         fTrackTypeGen;          // type of tracks used for FF 
     Int_t         fNSkipLeadingRan;       // number of leading tracks to be skipped in the randomized event
     Int_t         fTrackTypeRec;          // type of tracks used for FF 
     Int_t         fTrackTypeGen;          // type of tracks used for FF 
     Int_t         fNSkipLeadingRan;       // number of leading tracks to be skipped in the randomized event
@@ -151,25 +169,48 @@ class AliAnalysisTaskJetCluster : public AliAnalysisTaskSE
     Float_t       fVtxR2Cut;              // R vtx cut (squared) 
     Float_t       fCentCutUp;             // upper limit on centrality
     Float_t       fCentCutLo;             // lower limit on centrality
     Float_t       fVtxR2Cut;              // R vtx cut (squared) 
     Float_t       fCentCutUp;             // upper limit on centrality
     Float_t       fCentCutLo;             // lower limit on centrality
+
+    Bool_t       fStoreRhoLeadingTrackCorr; //store histos with rho correlation to leading track in event
+
     // output configurartion
     TString       fNonStdBranch;      // the name of the non-std branch name, if empty no branch is filled
     TString       fBackgroundBranch;  // name of the branch used for background subtraction
     // output configurartion
     TString       fNonStdBranch;      // the name of the non-std branch name, if empty no branch is filled
     TString       fBackgroundBranch;  // name of the branch used for background subtraction
-    TString       fNonStdFile;        // The optional name of the output file the non-std brnach is written to
-    
+    TString       fNonStdFile;        // The optional name of the output file the non-std branch is written to
+
+    //Detector level effects
+    TProfile *fMomResH1; // Momentum resolution from TrackQA Hybrid Category 1
+    TProfile *fMomResH2; // Momentum resolution from TrackQA Hybrid Category 2
+    TProfile *fMomResH3; // Momentum resolution from TrackQA Hybrid Category 3
+    TF1 *fMomResH1Fit;   //fit
+    TF1 *fMomResH2Fit;   //fit
+    TF1 *fMomResH3Fit;   //fit
+
+    TH1      *fhEffH1;        // Efficiency for Spectra Hybrid Category 1
+    TH1      *fhEffH2;        // Efficiency for Spectra Hybrid Category 2
+    TH1      *fhEffH3;        // Efficiency for Spectra Hybrid Category 3
+    Bool_t    fUseTrPtResolutionSmearing;  // Apply momentum smearing on track level
+    Bool_t    fUseDiceEfficiency;          // Apply efficiency on track level by dicing
+    Double_t  fDiceEfficiencyMinPt;        // Only do efficiency dicing for tracks above this pt
+    Bool_t fUseTrPtResolutionFromOADB;     // Load track pt resolution root file from OADB path
+    Bool_t fUseTrEfficiencyFromOADB;       // Load tracking efficiency root file from OADB path
+    TString fPathTrPtResolution;           // OADB path to root file
+    TString fPathTrEfficiency;             // OADB path to root file
+    Double_t fChangeEfficiencyFraction;    //change efficiency by fraction
+
 
     // Fast jet
 
     // Fast jet
-    Double_t fRparam;
+    Double_t fRparam;                  // fastjet distance parameter
     fastjet::JetAlgorithm fAlgorithm; //fastjet::kt_algorithm
     fastjet::Strategy fStrategy;  //= fastjet::Best;
     fastjet::RecombinationScheme fRecombScheme; // = fastjet::BIpt_scheme;
     fastjet::JetAlgorithm fAlgorithm; //fastjet::kt_algorithm
     fastjet::Strategy fStrategy;  //= fastjet::Best;
     fastjet::RecombinationScheme fRecombScheme; // = fastjet::BIpt_scheme;
-    fastjet::AreaType fAreaType; 
-    Double_t fGhostArea;
-    Int_t fActiveAreaRepeats;
-    Double_t fGhostEtamax;
-
-    TClonesArray  *fTCAJetsOut; //! TCA of output jets
-    TClonesArray  *fTCAJetsOutRan; //! TCA of output jets in randomized event
-    TClonesArray  *fTCARandomConesOut; //! TCA of output jets in randomized event
+    fastjet::AreaType fAreaType;  // fastjet area type
+    Double_t fGhostArea;          // fasjet ghost area
+    Int_t fActiveAreaRepeats;     // fast jet active area repeats
+    Double_t fGhostEtamax;        // fast jet ghost area
+
+    TClonesArray  *fTCAJetsOut;           //! TCA of output jets
+    TClonesArray  *fTCAJetsOutRan;        //! TCA of output jets in randomized event
+    TClonesArray  *fTCARandomConesOut;    //! TCA of output jets in randomized event
     TClonesArray  *fTCARandomConesOutRan; //! TCA of output jets in randomized event
     AliAODJetEventBackground *fAODJetBackgroundOut; //! jet background to be written out
 
     TClonesArray  *fTCARandomConesOutRan; //! TCA of output jets in randomized event
     AliAODJetEventBackground *fAODJetBackgroundOut; //! jet background to be written out
 
@@ -243,10 +284,45 @@ class AliAnalysisTaskJetCluster : public AliAnalysisTaskSE
     TH2F*         fh2TracksLeadingJetPhiPtC[kMaxCent]; //! track correlation with leading Jet
     TH2F*         fh2TracksLeadingJetPhiPtWC[kMaxCent]; //! track correlation with leading Jet
 
     TH2F*         fh2TracksLeadingJetPhiPtC[kMaxCent]; //! track correlation with leading Jet
     TH2F*         fh2TracksLeadingJetPhiPtWC[kMaxCent]; //! track correlation with leading Jet
 
+    TH3F*         fh3CentvsRhoLeadingTrackPt;       //! centrality vs background density full event
+    TH3F*         fh3CentvsSigmaLeadingTrackPt;     //! centrality vs sigma full event
+    TH3F*         fh3MultvsRhoLeadingTrackPt;       //! multiplicity vs background density full event
+    TH3F*         fh3MultvsSigmaLeadingTrackPt;     //! multiplicity vs sigma full event
+
+    TH3F*         fh3CentvsRhoLeadingTrackPtQ1;     //! centrality vs background density vs pt leading track near side
+    TH3F*         fh3CentvsRhoLeadingTrackPtQ2;     //! centrality vs background density vs pt leading track perpendicular (+0.5*\pi)
+    TH3F*         fh3CentvsRhoLeadingTrackPtQ3;     //! centrality vs background density vs pt leading track away side
+    TH3F*         fh3CentvsRhoLeadingTrackPtQ4;     //! centrality vs background density vs pt leading track perpendicular (-0.5*\pi)
+
+    TH3F*         fh3CentvsSigmaLeadingTrackPtQ1;     //! centrality vs sigma vs pt leading track near side
+    TH3F*         fh3CentvsSigmaLeadingTrackPtQ2;     //! centrality vs sigma vs pt leading track perpendicular (+0.5*\pi)
+    TH3F*         fh3CentvsSigmaLeadingTrackPtQ3;     //! centrality vs sigma vs pt leading track away side
+    TH3F*         fh3CentvsSigmaLeadingTrackPtQ4;     //! centrality vs sigma vs pt leading track perpendicular (-0.5*\pi)
+
+    TH3F*         fh3MultvsRhoLeadingTrackPtQ1;    //! multiplicity vs background density vs pt leading track near side
+    TH3F*         fh3MultvsRhoLeadingTrackPtQ2;    //! multiplicity vs background density vs pt leading track perpendicular (+0.5*\pi)
+    TH3F*         fh3MultvsRhoLeadingTrackPtQ3;    //! multiplicity vs background density vs pt leading track away side
+    TH3F*         fh3MultvsRhoLeadingTrackPtQ4;    //! multiplicity vs background density vs pt leading track perpendicular (-0.5*\pi)
+
+    TH3F*         fh3MultvsSigmaLeadingTrackPtQ1;     //! multiplicity vs sigma vs pt leading track near side
+    TH3F*         fh3MultvsSigmaLeadingTrackPtQ2;     //! multiplicity vs sigma vs pt leading track perpendicular (+0.5*\pi)
+    TH3F*         fh3MultvsSigmaLeadingTrackPtQ3;     //! multiplicity vs sigma vs pt leading track away side
+    TH3F*         fh3MultvsSigmaLeadingTrackPtQ4;     //! multiplicity vs sigma vs pt leading track perpendicular (-0.5*\pi)
+
+    TH3F*         fh3CentvsDeltaRhoLeadingTrackPtQ1;     //! centrality vs delta background density vs pt leading track near side
+    TH3F*         fh3CentvsDeltaRhoLeadingTrackPtQ2;     //! centrality vs delta background density vs pt leading track perpendicular (+0.5*\pi)
+    TH3F*         fh3CentvsDeltaRhoLeadingTrackPtQ3;     //! centrality vs delta background density vs pt leading track away side
+    TH3F*         fh3CentvsDeltaRhoLeadingTrackPtQ4;     //! centrality vs delta background density vs pt leading track perpendicular (-0.5*\pi)
+
+    //Histos for detector level effects from toy model
+    TH2F *fh2PtGenPtSmeared;     //! Control histo smeared momentum
+    TProfile *fp1Efficiency;     //! Control profile efficiency
+    TProfile *fp1PtResolution;   //! Control profile for pT resolution
+
     TList *fHistList; //!leading tracks to be skipped in the randomized event Output list
    
 
     TList *fHistList; //!leading tracks to be skipped in the randomized event Output list
    
 
-    ClassDef(AliAnalysisTaskJetCluster, 20
+    ClassDef(AliAnalysisTaskJetCluster, 21
 };
  
 #endif
 };
  
 #endif