]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - JETAN/AliAnalysisTaskJetCluster.h
Merge branch 'master' of https://git.cern.ch/reps/AliRoot
[u/mrichter/AliRoot.git] / JETAN / AliAnalysisTaskJetCluster.h
index 14b6c552597710e87bdae9d08b411fe0df3e2afc..042de7309c020801960244a6f13a7ed053cb17a4 100644 (file)
 
 #include  "AliAnalysisTaskSE.h"
 #include  "THnSparse.h" // cannot forward declare ThnSparseF
-#include "fastjet/ClusterSequenceArea.hh"
-#include "fastjet/AreaDefinition.hh"
-#include "fastjet/JetDefinition.hh"
-#include "fastjet/PseudoJet.hh"
+#ifndef __CINT__
+# include "fastjet/ClusterSequenceArea.hh"
+# include "fastjet/AreaDefinition.hh"
+# include "fastjet/JetDefinition.hh"
+#else
+namespace fastjet {
+  enum JetAlgorithm;
+  enum Strategy;
+  enum RecombinationScheme;
+  enum AreaType;
+}
+#endif
 
 ////////////////
 class AliJetHeader;
 class AliESDEvent;
 class AliAODEvent;
+class AliAODTrack;
 class AliAODExtension;
 class AliAODJet;
 class AliGenPythiaEventHeader;
 class AliCFManager;
 class AliAODJetEventBackground;
 class AliJetFinder;
+class AliAODMCParticle;
 class TList;
 class TChain;
 class TH2F;
@@ -53,6 +63,8 @@ class AliAnalysisTaskJetCluster : public AliAnalysisTaskSE
     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 SetRequireITSRefit(Int_t i){fRequireITSRefit=i;}
+    virtual void SetSharedClusterCut(Int_t docut){fApplySharedClusterCut=docut;}
     virtual void SetRecEtaWindow(Float_t f){fRecEtaWindow = f;}
     virtual void SetTrackEtaWindow(Float_t f){fTrackEtaWindow = f;}
     virtual void SetTrackTypeGen(Int_t i){fTrackTypeGen = i;}
@@ -79,6 +91,7 @@ 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 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;}
@@ -87,23 +100,28 @@ class AliAnalysisTaskJetCluster : public AliAnalysisTaskSE
     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 SetDiceEfficiency(Int_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);
-
+    virtual void SetFixedEfficiency(Double_t eff) {fEfficiencyFixed = eff;}
+    virtual void SetRequireT0vtx(Bool_t b = true){fRequireTZEROvtx = b;}
+    virtual void SetRequireV0AC(Bool_t b = true){fRequireVZEROAC = b;}
+    virtual void SetUseHFcuts(Bool_t b = true){fUseHFcuts = b;}
     Double_t GetMomentumSmearing(Int_t cat, Double_t pt);
     void FitMomentumResolution();
 
 
-    // for Fast Jet
+    // for Fast Jet 
     fastjet::JetAlgorithm        GetAlgorithm()         const {return fAlgorithm;}
     fastjet::Strategy            GetStrategy()          const {return fStrategy;}
     fastjet::RecombinationScheme GetRecombScheme()      const {return fRecombScheme;}
     fastjet::AreaType            GetAreaType()          const {return fAreaType;}
     // Setters
     void SetRparam(Double_t f)                           {fRparam = f;}
-    void SetAlgorithm(fastjet::JetAlgorithm f)           {fAlgorithm = f;}
+    // Temporary change to integer; problem with dictionary generation?
+    //void SetAlgorithm(fastjet::JetAlgorithm f)           {fAlgorithm = f;}
+    void SetAlgorithm(Int_t f)           {fAlgorithm = (fastjet::JetAlgorithm) f;}
     void SetStrategy(fastjet::Strategy f)                {fStrategy = f;}
     void SetRecombScheme(fastjet::RecombinationScheme f) {fRecombScheme = f;}
     void SetAreaType(fastjet::AreaType f)                {fAreaType = f;}
@@ -115,12 +133,14 @@ class AliAnalysisTaskJetCluster : public AliAnalysisTaskSE
 
     // Helper
     //
+       virtual bool    IsBMeson(int pc);
+       virtual bool    IsDMeson(int pc);
 
     // we have different cases
     // AOD reading -> MC from AOD
     // ESD reading -> MC from Kinematics
     // this has to match with our selection of input events
-    enum {kTrackUndef = 0, kTrackAOD, kTrackKineAll,kTrackKineCharged, kTrackAODMCAll, kTrackAODMCCharged, kTrackAODMCChargedAcceptance, kTrackAODextra, kTrackAODextraonly};
+    enum {kTrackUndef = 0, kTrackAOD, kTrackKineAll,kTrackKineCharged, kTrackAODMCAll, kTrackAODMCCharged, kTrackAODMCChargedAcceptance, kTrackAODextra, kTrackAODextraonly, kTrackAODMCextra, kTrackAODMCextraonly, kTrackAODMCHF};
     enum {kMaxJets = 4};
     enum {kMaxCorrelation =  3};
     enum {kMaxRadius =       5};
@@ -138,7 +158,9 @@ class AliAnalysisTaskJetCluster : public AliAnalysisTaskSE
     AliAnalysisTaskJetCluster& operator=(const AliAnalysisTaskJetCluster&);
 
     Int_t GetListOfTracks(TList *list,Int_t type);
-
+       Int_t AddDaughters(TList * list, AliAODMCParticle *part, TClonesArray * tca);
+    Bool_t AvoidDoubleCountingHF(AliAODEvent *aod, AliAODTrack *tr1);
+       
     AliAODEvent     *fAOD;                // ! where we take the jets from can be input or output AOD
     AliAODExtension *fAODExtension;       // ! AOD extension in case we write a non-sdt branch to a separate file and the aod is standard
     TRefArray       *fRef;               // ! trefarray for track references within the jet
@@ -146,6 +168,9 @@ class AliAnalysisTaskJetCluster : public AliAnalysisTaskSE
     Bool_t        fUseAODMCInput;         // take MC from input AOD not from ouptu AOD
     Bool_t        fUseBackgroundCalc;     // switches on background calculations
     Bool_t        fEventSelection;        // use the event selection of this task, otherwise analyse all
+    Bool_t        fRequireVZEROAC;        // switch to require V0 AC
+    Bool_t        fRequireTZEROvtx;       // switch to require T0 vtx
+    Bool_t        fUseHFcuts;       // switch to require T0 vtx
     UInt_t        fFilterMask;            // filter bit for slecected tracks
     UInt_t        fFilterMaskBestPt;      // filter bit to mark jets with high quality leading tracks
 
@@ -159,6 +184,8 @@ class AliAnalysisTaskJetCluster : public AliAnalysisTaskSE
     Float_t       fAvgTrials;             // Average nimber of trials
     Float_t       fExternalWeight;        // external weight
     Float_t       fTrackEtaWindow;        // eta window used for corraltion plots between rec and gen 
+    Int_t         fRequireITSRefit;       // to select hybrids with ITS refit only
+    Int_t         fApplySharedClusterCut; // flag to apply shared cluster cut (needed for some AODs where this cut was not applied in the filtering)
     Float_t       fRecEtaWindow;          // eta window used for corraltion plots between rec and gen 
     Float_t       fTrackPtCut;            // minimum track pt to be accepted
     Float_t       fJetOutputMinPt;        // minimum p_t for jets to be written out
@@ -168,6 +195,9 @@ 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
+
+    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
@@ -177,21 +207,22 @@ class AliAnalysisTaskJetCluster : public AliAnalysisTaskSE
     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
+    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
+    Int_t    fUseDiceEfficiency;           // Flag to apply efficiency on track level by dicing 0: no dicing; 1: dicing wrt to accepted; 2: dicing wrt to generated
     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
+    Double_t fChangeEfficiencyFraction;    // change efficiency by fraction
+    Double_t fEfficiencyFixed;             // fixed efficiency for all pT and all types of tracks
 
 
     // Fast jet
@@ -280,6 +311,36 @@ class AliAnalysisTaskJetCluster : public AliAnalysisTaskSE
     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
@@ -288,7 +349,7 @@ class AliAnalysisTaskJetCluster : public AliAnalysisTaskSE
     TList *fHistList; //!leading tracks to be skipped in the randomized event Output list
    
 
-    ClassDef(AliAnalysisTaskJetCluster, 20
+    ClassDef(AliAnalysisTaskJetCluster, 26
 };
  
 #endif