[u/mrichter/AliRoot.git] / JETAN / AliAnalysisTaskJetCluster.h

#ifndef ALIANALYSISTASKJETCLUSTER_H
#define ALIANALYSISTASKJETCLUSTER_H
 
/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

// **************************************
// task used for comparing different jets D parmaters from fastjet 
// *******************************************

#include  "AliAnalysisTaskSE.h"
#include  "THnSparse.h" // cannot forward declare ThnSparseF
#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 AliAODExtension;
class AliAODJet;
class AliGenPythiaEventHeader;
class AliCFManager;
class AliAODJetEventBackground;
class AliJetFinder;
class AliAODMCParticle;
class TList;
class TChain;
class TH2F;
class TH1F;
class TH3F;
class TProfile;
class TRandom3;
class TRefArray;
class TClonesArray;
class TF1;

class AliAnalysisTaskJetCluster : public AliAnalysisTaskSE
{
 public:
    AliAnalysisTaskJetCluster();
    AliAnalysisTaskJetCluster(const char* name);
    virtual ~AliAnalysisTaskJetCluster();
    // Implementation of interface methods
    virtual void UserCreateOutputObjects();
    virtual void LocalInit();
    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 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;}
    virtual void SetTrackTypeRec(Int_t i){fTrackTypeRec = i;}
    virtual void SetTrackPtCut(Float_t x){fTrackPtCut = x;}
    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 SetBackgroundBranch(const char* c){fBackgroundBranch = c;}
    virtual const char* GetBackgroundBranch(){return fBackgroundBranch.Data();}    
    virtual void SetNSkipLeadingRan(Int_t x){fNSkipLeadingRan = x;}
    virtual void SetNSkipLeadingCone(Int_t x){fNSkipLeadingCone = x;}
    virtual void SetNRandomCones(Int_t x){fNRandomCones = x;}

    virtual void SetJetOutputBranch(const char *c){fNonStdBranch = c;}
    virtual const char* GetJetOutputBranch(){return fNonStdBranch.Data();}
    virtual void SetJetOutputFile(const char *c){fNonStdFile = c;}
    virtual const char* GetJetOutputFile(){return fNonStdFile.Data();}
    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(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;}

    Double_t GetMomentumSmearing(Int_t cat, Double_t pt);
    void FitMomentumResolution();


    // 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;}
    // 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;}
    void SetGhostArea(Double_t f) {fGhostArea = f;}
    void SetActiveAreaRepeats(Int_t f) {fActiveAreaRepeats = f;}
    void SetGhostEtamax(Double_t f) {fGhostEtamax = f;}


    // 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, kTrackAODMCextra, kTrackAODMCextraonly, kTrackAODMCHF};
    enum {kMaxJets = 4};
    enum {kMaxCorrelation =  3};
    enum {kMaxRadius =       5};
    enum {kMaxCent =         4};
    enum {kJet = 1<<0,
	  kJetRan = 1<<1,	  
	  kRC = 1<<2,
	  kRCRan = 1<<3
    };
    

 private:

    AliAnalysisTaskJetCluster(const AliAnalysisTaskJetCluster&);
    AliAnalysisTaskJetCluster& operator=(const AliAnalysisTaskJetCluster&);

    Int_t GetListOfTracks(TList *list,Int_t type);
	Int_t AddDaughters(TList * list, AliAODMCParticle *part, TClonesArray * tca);
	
    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
    Bool_t        fUseAODTrackInput;      // take track from input AOD not from ouptu AOD
    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
    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        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         fNSkipLeadingCone;      // number of leading jets to be for the random cones
    Int_t         fNRandomCones;          // number of generated random cones
    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
    Float_t       fMaxTrackPtInJet;       // maximum track pt within a jet for flagging...
    Float_t       fJetTriggerPtCut;       // minimum jwt pT for AOD to be written
    Float_t       fVtxZCut;               // zvtx cut
    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
    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
    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 fEfficiencyFixed;             // fixed efficiency for all pT and all types of tracks


    // Fast jet
    Double_t fRparam;                  // fastjet distance parameter
    fastjet::JetAlgorithm fAlgorithm; //fastjet::kt_algorithm
    fastjet::Strategy fStrategy;  //= fastjet::Best;
    fastjet::RecombinationScheme fRecombScheme; // = fastjet::BIpt_scheme;
    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

    TRandom3*     fRandom;   //! random number generator
    TProfile*     fh1Xsec;   //! pythia cross section and trials
    TH1F*         fh1Trials; //! trials are added
    TH1F*         fh1PtHard;  //! Pt har of the event...       
    TH1F*         fh1PtHardNoW;  //! Pt har of the event without weigt       
    TH1F*         fh1PtHardTrials;  //! Number of trials 

    TH1F*         fh1NJetsRec; //! number of reconstructed jets
    TH1F*         fh1NConstRec;//! number of constiutens in leading jet
    TH1F*         fh1NConstLeadingRec;//! number of constiutens in leading jet
    TH1F*         fh1PtJetsRecIn;  //! Jet pt for all jets
    TH1F*         fh1PtJetsLeadingRecIn;  //! Jet pt for all jets
    TH1F*         fh1PtJetConstRec;//! pt of constituents
    TH1F*         fh1PtJetConstLeadingRec;// pt of constituents
    TH1F*         fh1PtTracksRecIn;  //! track pt for all tracks
    TH1F*         fh1PtTracksLeadingRecIn;  //! track pt for all tracks

    // Randomized track histos
    TH1F*         fh1NJetsRecRan; //! number of reconstructed jets from randomized
    TH1F*         fh1NConstRecRan;//! number of constiutens in leading jet
    TH1F*         fh1PtJetsLeadingRecInRan;  //! Jet pt for all jets
    TH1F*         fh1NConstLeadingRecRan;//! number of constiutens in leading jet
    TH1F*         fh1PtJetsRecInRan;  //! Jet pt for all jets

    TH1F*         fh1PtTracksGenIn;  //! track pt for all tracks
    TH1F*         fh1Nch;            //! charged particle mult
    TH1F*         fh1BiARandomCones[3]; //! Residual distribtion from reandom cones on real event
    TH1F*         fh1BiARandomConesRan[3]; //! Residual distribtion from reandom cones on random event
    TH1F*         fh1CentralityPhySel;          // ! centrality of anaylsed events 
    TH1F*         fh1Centrality;                // ! centrality of anaylsed events 
    TH1F*         fh1CentralitySelect;          // ! centrality of selected events 
    TH1F*         fh1ZPhySel;          // ! centrality of anaylsed events 
    TH1F*         fh1Z;                // ! centrality of anaylsed events 
    TH1F*         fh1ZSelect;          // ! centrality of selected events 


    TH2F*         fh2NRecJetsPt;            //! Number of found jets above threshold
    TH2F*         fh2NRecTracksPt;          //! Number of found tracks above threshold
    TH2F*         fh2NConstPt;           //! number of constituents vs. pt
    TH2F*         fh2NConstLeadingPt;           //! number of constituents vs. pt
    TH2F*         fh2JetPhiEta;             //! jet phi eta
    TH2F*         fh2LeadingJetPhiEta;      //! leading jet phi eta
    TH2F*         fh2JetEtaPt;              //! leading jet eta
    TH2F*         fh2LeadingJetEtaPt;              //! leading jet eta
    TH2F*         fh2TrackEtaPt;              //! track eta
    TH2F*         fh2LeadingTrackEtaPt;       //! leading track eta
    TH2F*         fh2JetsLeadingPhiEta;     //! jet phi eta
    TH2F*         fh2JetsLeadingPhiPt;      //! jet correlation with leading jet
    TH2F*         fh2TracksLeadingPhiEta;   //! track correlation with leading track
    TH2F*         fh2TracksLeadingPhiPt;    //! track correlation with leading track
    TH2F*         fh2TracksLeadingJetPhiPt; //! track correlation with leading Jet
    TH2F*         fh2JetsLeadingPhiPtW;      //! jet correlation with leading jet
    TH2F*         fh2TracksLeadingPhiPtW;   //! track correlation with leading track
    TH2F*         fh2TracksLeadingJetPhiPtW; //! track correlation with leading Jet
    TH2F*         fh2NRecJetsPtRan;            //! Number of found jets above threshold
    TH2F*         fh2NConstPtRan;           //! number of constituents vs. pt
    TH2F*         fh2NConstLeadingPtRan;           //! number of constituents vs. pt
    TH2F*         fh2PtNch;               //! p_T of cluster vs. multiplicity,
    TH2F*         fh2PtNchRan;            //! p_T of cluster vs. multiplicity,random
    TH2F*         fh2PtNchN;               //! p_T of cluster vs. multiplicity, weigthed with constituents
    TH2F*         fh2PtNchNRan;            //! p_T of cluster vs. multiplicity, weigthed with constituents random
    TH2F*         fh2TracksLeadingJetPhiPtRan; //! track correlation with leading Jet
    TH2F*         fh2TracksLeadingJetPhiPtWRan; //! track correlation with leading Jet


    TH2F*         fh2JetsLeadingPhiPtC[kMaxCent]; //! jet correlation with leading jet    
    TH2F*         fh2JetsLeadingPhiPtWC[kMaxCent];      //! jet 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
   

    ClassDef(AliAnalysisTaskJetCluster, 26) 
};
 
#endif
Commit	Line	Data
dbf053f0	1	#ifndef ALIANALYSISTASKJETCLUSTER_H
	2	#define ALIANALYSISTASKJETCLUSTER_H
	3
	4	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	5	* See cxx source for full Copyright notice */
	6
	7	// **************************************
f840d64c	8	// task used for comparing different jets D parmaters from fastjet
dbf053f0	9	// *******************************************
	10
	11	#include "AliAnalysisTaskSE.h"
	12	#include "THnSparse.h" // cannot forward declare ThnSparseF
13fa2a91	13	#ifndef __CINT__
	14	# include "fastjet/ClusterSequenceArea.hh"
	15	# include "fastjet/AreaDefinition.hh"
	16	# include "fastjet/JetDefinition.hh"
13fa2a91	17	#else
13fa2a91	18	namespace fastjet {
6f3ba560	19	enum JetAlgorithm;
	20	enum Strategy;
	21	enum RecombinationScheme;
	22	enum AreaType;
13fa2a91	23	}
13fa2a91	24	#endif
dbf053f0	25
	26	////////////////
	27	class AliJetHeader;
	28	class AliESDEvent;
	29	class AliAODEvent;
54424110	30	class AliAODExtension;
dbf053f0	31	class AliAODJet;
	32	class AliGenPythiaEventHeader;
	33	class AliCFManager;
d6e66a82	34	class AliAODJetEventBackground;
d6e66a82	35	class AliJetFinder;
d8589aad	36	class AliAODMCParticle;
dbf053f0	37	class TList;
	38	class TChain;
	39	class TH2F;
	40	class TH1F;
	41	class TH3F;
	42	class TProfile;
d7ca0e14	43	class TRandom3;
54424110	44	class TRefArray;
8f85b773	45	class TClonesArray;
006b2a30	46	class TF1;
dbf053f0	47
	48	class AliAnalysisTaskJetCluster : public AliAnalysisTaskSE
	49	{
	50	public:
	51	AliAnalysisTaskJetCluster();
	52	AliAnalysisTaskJetCluster(const char* name);
54424110	53	virtual ~AliAnalysisTaskJetCluster();
dbf053f0	54	// Implementation of interface methods
dbf053f0	55	virtual void UserCreateOutputObjects();
fa7c34ba	56	virtual void LocalInit();
dbf053f0	57	virtual void UserExec(Option_t *option);
	58	virtual void Terminate(Option_t *option);
	59	virtual Bool_t Notify();
	60
d6e66a82	61
dbf053f0	62	virtual void SetAODTrackInput(Bool_t b){fUseAODTrackInput = b;}
dbf053f0	63	virtual void SetAODMCInput(Bool_t b){fUseAODMCInput = b;}
ca6d12f9	64	virtual void SetEventSelection(Bool_t b){fEventSelection = b;}
c18b64ea	65	virtual void SetRequireITSRefit(Int_t i){fRequireITSRefit=i;}
27c67edf	66	virtual void SetSharedClusterCut(Int_t docut){fApplySharedClusterCut=docut;}
dbf053f0	67	virtual void SetRecEtaWindow(Float_t f){fRecEtaWindow = f;}
8a320ab9	68	virtual void SetTrackEtaWindow(Float_t f){fTrackEtaWindow = f;}
dbf053f0	69	virtual void SetTrackTypeGen(Int_t i){fTrackTypeGen = i;}
dbf053f0	70	virtual void SetTrackTypeRec(Int_t i){fTrackTypeRec = i;}
54424110	71	virtual void SetTrackPtCut(Float_t x){fTrackPtCut = x;}
aae7993b	72	virtual void SetCentralityCut(Float_t xLo,Float_t xUp){fCentCutLo = xLo; fCentCutUp = xUp;}
b2c1e955	73	virtual void SetFilterMask(UInt_t i,Int_t iType = 0){fFilterMask = i;
b2c1e955	74	fFilterType = iType;}
2d8b837d	75	virtual void SetFilterMaskBestPt(UInt_t i){fFilterMaskBestPt = i;}
2d8b837d	76
bea24d2d	77	virtual void SetJetTypes(UInt_t i){fJetTypes = i;}
0341d0d8	78	virtual void SetJetTriggerPtCut(Float_t x){fJetTriggerPtCut = x;}
ca6d12f9	79	virtual void SetVtxCuts(Float_t z,Float_t r = 1){fVtxZCut = z; fVtxR2Cut = r *r;}
b5b166c5	80	virtual void SetBackgroundBranch(const char* c){fBackgroundBranch = c;}
b5b166c5	81	virtual const char* GetBackgroundBranch(){return fBackgroundBranch.Data();}
bd80a748	82	virtual void SetNSkipLeadingRan(Int_t x){fNSkipLeadingRan = x;}
af399bc2	83	virtual void SetNSkipLeadingCone(Int_t x){fNSkipLeadingCone = x;}
d7ca0e14	84	virtual void SetNRandomCones(Int_t x){fNRandomCones = x;}
bd80a748	85
54424110	86	virtual void SetJetOutputBranch(const char *c){fNonStdBranch = c;}
b5b166c5	87	virtual const char* GetJetOutputBranch(){return fNonStdBranch.Data();}
54424110	88	virtual void SetJetOutputFile(const char *c){fNonStdFile = c;}
b5b166c5	89	virtual const char* GetJetOutputFile(){return fNonStdFile.Data();}
65c43de8	90	virtual void SetMaxTrackPtInJet(Float_t x){fMaxTrackPtInJet = x;}
54424110	91	virtual void SetJetOutputMinPt(Float_t x){fJetOutputMinPt = x;}
21ead515	92	virtual void SetBackgroundCalc(Bool_t b){fUseBackgroundCalc = b;}
7bedcd31	93	virtual void SetStoreRhoLeadingTrackCorr(Bool_t b) {fStoreRhoLeadingTrackCorr=b;}
54424110	94
006b2a30	95	//Setters for detector level effects
fa7c34ba	96	virtual void SetUseTrResolutionFromOADB(Bool_t b=kTRUE, TString path="$ALICE_ROOT/OADB/PWGJE/Resolution/PtResol_LHCh_Cent0-10_v1.root") {fUseTrPtResolutionFromOADB = b; fPathTrPtResolution=path;}
	97	virtual void SetUseTrEfficiencyFromOADB(Bool_t b=kTRUE, TString path="$ALICE_ROOT/OADB/PWGJE/Efficiency/Efficiency_LHC11a2aj_Cent0_v1.root") {fUseTrEfficiencyFromOADB = b; fPathTrEfficiency=path;}
	98	virtual void LoadTrEfficiencyRootFileFromOADB();
	99	virtual void LoadTrPtResolutionRootFileFromOADB();
	100	virtual void SetChangeEfficiencyFraction(Double_t p) {fChangeEfficiencyFraction = p;}
	101	virtual void SetSmearResolution(Bool_t b){fUseTrPtResolutionSmearing = b;}
190446bf	102	virtual void SetDiceEfficiency(Int_t b){fUseDiceEfficiency = b;}
4bd66a92	103	virtual void SetDiceEfficiencyMinPt(Double_t pt) {fDiceEfficiencyMinPt = pt;}
006b2a30	104	virtual void SetMomentumResolutionHybrid(TProfile p1, TProfile p2, TProfile *p3);
006b2a30	105	virtual void SetEfficiencyHybrid(TH1 h1, TH1 h2, TH1 *h3);
da8f7918	106	virtual void SetFixedEfficiency(Double_t eff) {fEfficiencyFixed = eff;}
9dcbfbd8	107	virtual void SetRequireT0vtx(Bool_t b = true){fRequireTZEROvtx = b;}
9dcbfbd8	108	virtual void SetRequireV0AC(Bool_t b = true){fRequireVZEROAC = b;}
006b2a30	109
	110	Double_t GetMomentumSmearing(Int_t cat, Double_t pt);
	111	void FitMomentumResolution();
65c43de8	112
65c43de8	113
dbf053f0	114	// for Fast Jet
	115	fastjet::JetAlgorithm GetAlgorithm() const {return fAlgorithm;}
	116	fastjet::Strategy GetStrategy() const {return fStrategy;}
	117	fastjet::RecombinationScheme GetRecombScheme() const {return fRecombScheme;}
	118	fastjet::AreaType GetAreaType() const {return fAreaType;}
	119	// Setters
	120	void SetRparam(Double_t f) {fRparam = f;}
ff1ffd3f	121	// Temporary change to integer; problem with dictionary generation?
	122	//void SetAlgorithm(fastjet::JetAlgorithm f) {fAlgorithm = f;}
	123	void SetAlgorithm(Int_t f) {fAlgorithm = (fastjet::JetAlgorithm) f;}
dbf053f0	124	void SetStrategy(fastjet::Strategy f) {fStrategy = f;}
	125	void SetRecombScheme(fastjet::RecombinationScheme f) {fRecombScheme = f;}
	126	void SetAreaType(fastjet::AreaType f) {fAreaType = f;}
d6e66a82	127	void SetGhostArea(Double_t f) {fGhostArea = f;}
	128	void SetActiveAreaRepeats(Int_t f) {fActiveAreaRepeats = f;}
	129	void SetGhostEtamax(Double_t f) {fGhostEtamax = f;}
	130
	131
dbf053f0	132
	133	// Helper
	134	//
d8589aad	135	virtual bool IsBMeson(int pc);
d8589aad	136	virtual bool IsDMeson(int pc);
dbf053f0	137
	138	// we have different cases
	139	// AOD reading -> MC from AOD
	140	// ESD reading -> MC from Kinematics
	141	// this has to match with our selection of input events
d8589aad	142	enum {kTrackUndef = 0, kTrackAOD, kTrackKineAll,kTrackKineCharged, kTrackAODMCAll, kTrackAODMCCharged, kTrackAODMCChargedAcceptance, kTrackAODextra, kTrackAODextraonly, kTrackAODMCextra, kTrackAODMCextraonly, kTrackAODMCHF};
dbf053f0	143	enum {kMaxJets = 4};
	144	enum {kMaxCorrelation = 3};
	145	enum {kMaxRadius = 5};
aae7993b	146	enum {kMaxCent = 4};
bea24d2d	147	enum {kJet = 1<<0,
	148	kJetRan = 1<<1,
	149	kRC = 1<<2,
	150	kRCRan = 1<<3
	151	};
dbf053f0	152
	153
	154	private:
	155
	156	AliAnalysisTaskJetCluster(const AliAnalysisTaskJetCluster&);
	157	AliAnalysisTaskJetCluster& operator=(const AliAnalysisTaskJetCluster&);
	158
	159	Int_t GetListOfTracks(TList *list,Int_t type);
d8589aad	160	Int_t AddDaughters(TList * list, AliAODMCParticle part, TClonesArray tca);
d8589aad	161
54424110	162	AliAODEvent *fAOD; // ! where we take the jets from can be input or output AOD
	163	AliAODExtension *fAODExtension; // ! AOD extension in case we write a non-sdt branch to a separate file and the aod is standard
	164	TRefArray *fRef; // ! trefarray for track references within the jet
dbf053f0	165	Bool_t fUseAODTrackInput; // take track from input AOD not from ouptu AOD
dbf053f0	166	Bool_t fUseAODMCInput; // take MC from input AOD not from ouptu AOD
21ead515	167	Bool_t fUseBackgroundCalc; // switches on background calculations
ca6d12f9	168	Bool_t fEventSelection; // use the event selection of this task, otherwise analyse all
d66558be	169	Bool_t fRequireVZEROAC; // switch to require V0 AC
d66558be	170	Bool_t fRequireTZEROvtx; // switch to require T0 vtx
54424110	171	UInt_t fFilterMask; // filter bit for slecected tracks
b01d3ab7	172	UInt_t fFilterMaskBestPt; // filter bit to mark jets with high quality leading tracks
d7396b04	173
b2c1e955	174	UInt_t fFilterType; // filter type 0 = all, 1 = ITSTPC, 2 = TPC
006b2a30	175	UInt_t fJetTypes; // 1<<0 regular jets, 1<<1 << randomized event 1<<2 random cones 1<<3 random cones randomiuzed event
dbf053f0	176	Int_t fTrackTypeRec; // type of tracks used for FF
dbf053f0	177	Int_t fTrackTypeGen; // type of tracks used for FF
af399bc2	178	Int_t fNSkipLeadingRan; // number of leading tracks to be skipped in the randomized event
af399bc2	179	Int_t fNSkipLeadingCone; // number of leading jets to be for the random cones
d7ca0e14	180	Int_t fNRandomCones; // number of generated random cones
dbf053f0	181	Float_t fAvgTrials; // Average nimber of trials
dbf053f0	182	Float_t fExternalWeight; // external weight
ca6d12f9	183	Float_t fTrackEtaWindow; // eta window used for corraltion plots between rec and gen
c18b64ea	184	Int_t fRequireITSRefit; // to select hybrids with ITS refit only
27c67edf	185	Int_t fApplySharedClusterCut; // flag to apply shared cluster cut (needed for some AODs where this cut was not applied in the filtering)
dbf053f0	186	Float_t fRecEtaWindow; // eta window used for corraltion plots between rec and gen
54424110	187	Float_t fTrackPtCut; // minimum track pt to be accepted
54424110	188	Float_t fJetOutputMinPt; // minimum p_t for jets to be written out
65c43de8	189	Float_t fMaxTrackPtInJet; // maximum track pt within a jet for flagging...
0341d0d8	190	Float_t fJetTriggerPtCut; // minimum jwt pT for AOD to be written
ca6d12f9	191	Float_t fVtxZCut; // zvtx cut
ca6d12f9	192	Float_t fVtxR2Cut; // R vtx cut (squared)
aae7993b	193	Float_t fCentCutUp; // upper limit on centrality
aae7993b	194	Float_t fCentCutLo; // lower limit on centrality
7bedcd31	195
	196	Bool_t fStoreRhoLeadingTrackCorr; //store histos with rho correlation to leading track in event
	197
54424110	198	// output configurartion
54424110	199	TString fNonStdBranch; // the name of the non-std branch name, if empty no branch is filled
0341d0d8	200	TString fBackgroundBranch; // name of the branch used for background subtraction
006b2a30	201	TString fNonStdFile; // The optional name of the output file the non-std branch is written to
	202
	203	//Detector level effects
	204	TProfile *fMomResH1; // Momentum resolution from TrackQA Hybrid Category 1
	205	TProfile *fMomResH2; // Momentum resolution from TrackQA Hybrid Category 2
	206	TProfile *fMomResH3; // Momentum resolution from TrackQA Hybrid Category 3
575dee18	207	TF1 *fMomResH1Fit; //fit
	208	TF1 *fMomResH2Fit; //fit
	209	TF1 *fMomResH3Fit; //fit
006b2a30	210
	211	TH1 *fhEffH1; // Efficiency for Spectra Hybrid Category 1
	212	TH1 *fhEffH2; // Efficiency for Spectra Hybrid Category 2
	213	TH1 *fhEffH3; // Efficiency for Spectra Hybrid Category 3
fa7c34ba	214	Bool_t fUseTrPtResolutionSmearing; // Apply momentum smearing on track level
b0b15bae	215	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
4bd66a92	216	Double_t fDiceEfficiencyMinPt; // Only do efficiency dicing for tracks above this pt
fa7c34ba	217	Bool_t fUseTrPtResolutionFromOADB; // Load track pt resolution root file from OADB path
	218	Bool_t fUseTrEfficiencyFromOADB; // Load tracking efficiency root file from OADB path
	219	TString fPathTrPtResolution; // OADB path to root file
	220	TString fPathTrEfficiency; // OADB path to root file
da8f7918	221	Double_t fChangeEfficiencyFraction; // change efficiency by fraction
da8f7918	222	Double_t fEfficiencyFixed; // fixed efficiency for all pT and all types of tracks
fa7c34ba	223
54424110	224
dbf053f0	225	// Fast jet
f0659f11	226	Double_t fRparam; // fastjet distance parameter
dbf053f0	227	fastjet::JetAlgorithm fAlgorithm; //fastjet::kt_algorithm
	228	fastjet::Strategy fStrategy; //= fastjet::Best;
	229	fastjet::RecombinationScheme fRecombScheme; // = fastjet::BIpt_scheme;
f0659f11	230	fastjet::AreaType fAreaType; // fastjet area type
	231	Double_t fGhostArea; // fasjet ghost area
	232	Int_t fActiveAreaRepeats; // fast jet active area repeats
	233	Double_t fGhostEtamax; // fast jet ghost area
8f85b773	234
006b2a30	235	TClonesArray *fTCAJetsOut; //! TCA of output jets
	236	TClonesArray *fTCAJetsOutRan; //! TCA of output jets in randomized event
	237	TClonesArray *fTCARandomConesOut; //! TCA of output jets in randomized event
8f85b773	238	TClonesArray *fTCARandomConesOutRan; //! TCA of output jets in randomized event
	239	AliAODJetEventBackground *fAODJetBackgroundOut; //! jet background to be written out
	240
d7ca0e14	241	TRandom3* fRandom; //! random number generator
d6e66a82	242	TProfile* fh1Xsec; //! pythia cross section and trials
	243	TH1F* fh1Trials; //! trials are added
	244	TH1F* fh1PtHard; //! Pt har of the event...
	245	TH1F* fh1PtHardNoW; //! Pt har of the event without weigt
	246	TH1F* fh1PtHardTrials; //! Number of trials
	247
	248	TH1F* fh1NJetsRec; //! number of reconstructed jets
	249	TH1F* fh1NConstRec;//! number of constiutens in leading jet
	250	TH1F* fh1NConstLeadingRec;//! number of constiutens in leading jet
	251	TH1F* fh1PtJetsRecIn; //! Jet pt for all jets
	252	TH1F* fh1PtJetsLeadingRecIn; //! Jet pt for all jets
	253	TH1F* fh1PtJetConstRec;//! pt of constituents
dbf053f0	254	TH1F* fh1PtJetConstLeadingRec;// pt of constituents
d6e66a82	255	TH1F* fh1PtTracksRecIn; //! track pt for all tracks
d6e66a82	256	TH1F* fh1PtTracksLeadingRecIn; //! track pt for all tracks
dbf053f0	257
dbf053f0	258	// Randomized track histos
d6e66a82	259	TH1F* fh1NJetsRecRan; //! number of reconstructed jets from randomized
	260	TH1F* fh1NConstRecRan;//! number of constiutens in leading jet
	261	TH1F* fh1PtJetsLeadingRecInRan; //! Jet pt for all jets
	262	TH1F* fh1NConstLeadingRecRan;//! number of constiutens in leading jet
	263	TH1F* fh1PtJetsRecInRan; //! Jet pt for all jets
	264
	265	TH1F* fh1PtTracksGenIn; //! track pt for all tracks
	266	TH1F* fh1Nch; //! charged particle mult
82a35b1a	267	TH1F* fh1BiARandomCones[3]; //! Residual distribtion from reandom cones on real event
82a35b1a	268	TH1F* fh1BiARandomConesRan[3]; //! Residual distribtion from reandom cones on random event
aae7993b	269	TH1F* fh1CentralityPhySel; // ! centrality of anaylsed events
e2ca7519	270	TH1F* fh1Centrality; // ! centrality of anaylsed events
aae7993b	271	TH1F* fh1CentralitySelect; // ! centrality of selected events
e2ca7519	272	TH1F* fh1ZPhySel; // ! centrality of anaylsed events
	273	TH1F* fh1Z; // ! centrality of anaylsed events
	274	TH1F* fh1ZSelect; // ! centrality of selected events
	275
d6e66a82	276
	277	TH2F* fh2NRecJetsPt; //! Number of found jets above threshold
	278	TH2F* fh2NRecTracksPt; //! Number of found tracks above threshold
	279	TH2F* fh2NConstPt; //! number of constituents vs. pt
	280	TH2F* fh2NConstLeadingPt; //! number of constituents vs. pt
	281	TH2F* fh2JetPhiEta; //! jet phi eta
	282	TH2F* fh2LeadingJetPhiEta; //! leading jet phi eta
	283	TH2F* fh2JetEtaPt; //! leading jet eta
	284	TH2F* fh2LeadingJetEtaPt; //! leading jet eta
	285	TH2F* fh2TrackEtaPt; //! track eta
	286	TH2F* fh2LeadingTrackEtaPt; //! leading track eta
	287	TH2F* fh2JetsLeadingPhiEta; //! jet phi eta
	288	TH2F* fh2JetsLeadingPhiPt; //! jet correlation with leading jet
	289	TH2F* fh2TracksLeadingPhiEta; //! track correlation with leading track
	290	TH2F* fh2TracksLeadingPhiPt; //! track correlation with leading track
	291	TH2F* fh2TracksLeadingJetPhiPt; //! track correlation with leading Jet
	292	TH2F* fh2JetsLeadingPhiPtW; //! jet correlation with leading jet
	293	TH2F* fh2TracksLeadingPhiPtW; //! track correlation with leading track
	294	TH2F* fh2TracksLeadingJetPhiPtW; //! track correlation with leading Jet
	295	TH2F* fh2NRecJetsPtRan; //! Number of found jets above threshold
	296	TH2F* fh2NConstPtRan; //! number of constituents vs. pt
	297	TH2F* fh2NConstLeadingPtRan; //! number of constituents vs. pt
	298	TH2F* fh2PtNch; //! p_T of cluster vs. multiplicity,
	299	TH2F* fh2PtNchRan; //! p_T of cluster vs. multiplicity,random
	300	TH2F* fh2PtNchN; //! p_T of cluster vs. multiplicity, weigthed with constituents
	301	TH2F* fh2PtNchNRan; //! p_T of cluster vs. multiplicity, weigthed with constituents random
	302	TH2F* fh2TracksLeadingJetPhiPtRan; //! track correlation with leading Jet
	303	TH2F* fh2TracksLeadingJetPhiPtWRan; //! track correlation with leading Jet
82a35b1a	304
aae7993b	305
	306	TH2F* fh2JetsLeadingPhiPtC[kMaxCent]; //! jet correlation with leading jet
	307	TH2F* fh2JetsLeadingPhiPtWC[kMaxCent]; //! jet correlation with leading jet
	308	TH2F* fh2TracksLeadingJetPhiPtC[kMaxCent]; //! track correlation with leading Jet
	309	TH2F* fh2TracksLeadingJetPhiPtWC[kMaxCent]; //! track correlation with leading Jet
	310
cbfccad4	311	TH3F* fh3CentvsRhoLeadingTrackPt; //! centrality vs background density full event
	312	TH3F* fh3CentvsSigmaLeadingTrackPt; //! centrality vs sigma full event
	313	TH3F* fh3MultvsRhoLeadingTrackPt; //! multiplicity vs background density full event
	314	TH3F* fh3MultvsSigmaLeadingTrackPt; //! multiplicity vs sigma full event
7bedcd31	315
	316	TH3F* fh3CentvsRhoLeadingTrackPtQ1; //! centrality vs background density vs pt leading track near side
	317	TH3F* fh3CentvsRhoLeadingTrackPtQ2; //! centrality vs background density vs pt leading track perpendicular (+0.5*\pi)
	318	TH3F* fh3CentvsRhoLeadingTrackPtQ3; //! centrality vs background density vs pt leading track away side
	319	TH3F* fh3CentvsRhoLeadingTrackPtQ4; //! centrality vs background density vs pt leading track perpendicular (-0.5*\pi)
	320
	321	TH3F* fh3CentvsSigmaLeadingTrackPtQ1; //! centrality vs sigma vs pt leading track near side
	322	TH3F* fh3CentvsSigmaLeadingTrackPtQ2; //! centrality vs sigma vs pt leading track perpendicular (+0.5*\pi)
	323	TH3F* fh3CentvsSigmaLeadingTrackPtQ3; //! centrality vs sigma vs pt leading track away side
	324	TH3F* fh3CentvsSigmaLeadingTrackPtQ4; //! centrality vs sigma vs pt leading track perpendicular (-0.5*\pi)
	325
	326	TH3F* fh3MultvsRhoLeadingTrackPtQ1; //! multiplicity vs background density vs pt leading track near side
	327	TH3F* fh3MultvsRhoLeadingTrackPtQ2; //! multiplicity vs background density vs pt leading track perpendicular (+0.5*\pi)
	328	TH3F* fh3MultvsRhoLeadingTrackPtQ3; //! multiplicity vs background density vs pt leading track away side
	329	TH3F* fh3MultvsRhoLeadingTrackPtQ4; //! multiplicity vs background density vs pt leading track perpendicular (-0.5*\pi)
	330
	331	TH3F* fh3MultvsSigmaLeadingTrackPtQ1; //! multiplicity vs sigma vs pt leading track near side
	332	TH3F* fh3MultvsSigmaLeadingTrackPtQ2; //! multiplicity vs sigma vs pt leading track perpendicular (+0.5*\pi)
	333	TH3F* fh3MultvsSigmaLeadingTrackPtQ3; //! multiplicity vs sigma vs pt leading track away side
	334	TH3F* fh3MultvsSigmaLeadingTrackPtQ4; //! multiplicity vs sigma vs pt leading track perpendicular (-0.5*\pi)
c0fdfe53	335
cbfccad4	336	TH3F* fh3CentvsDeltaRhoLeadingTrackPtQ1; //! centrality vs delta background density vs pt leading track near side
	337	TH3F* fh3CentvsDeltaRhoLeadingTrackPtQ2; //! centrality vs delta background density vs pt leading track perpendicular (+0.5*\pi)
	338	TH3F* fh3CentvsDeltaRhoLeadingTrackPtQ3; //! centrality vs delta background density vs pt leading track away side
	339	TH3F* fh3CentvsDeltaRhoLeadingTrackPtQ4; //! centrality vs delta background density vs pt leading track perpendicular (-0.5*\pi)
	340
006b2a30	341	//Histos for detector level effects from toy model
	342	TH2F *fh2PtGenPtSmeared; //! Control histo smeared momentum
	343	TProfile *fp1Efficiency; //! Control profile efficiency
	344	TProfile *fp1PtResolution; //! Control profile for pT resolution
	345
d7ca0e14	346	TList *fHistList; //!leading tracks to be skipped in the randomized event Output list
dbf053f0	347
dbf053f0	348
27c67edf	349	ClassDef(AliAnalysisTaskJetCluster, 26)
dbf053f0	350	};
	351
	352	#endif