JETAN/AliAnalysisTaskJetCluster.h

   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 // **************************************
   8 // task used for comapring different jets D parmaters from fastjet
   9 // *******************************************
  10
  11 #include  "AliAnalysisTaskSE.h"
  12 #include  "THnSparse.h" // cannot forward declare ThnSparseF
  13 #include "fastjet/ClusterSequenceArea.hh"
  14 #include "fastjet/AreaDefinition.hh"
  15 #include "fastjet/JetDefinition.hh"
  16 #include "fastjet/PseudoJet.hh"
  17
  18 ////////////////
  19 class AliJetHeader;
  20 class AliESDEvent;
  21 class AliAODEvent;
  22 class AliAODExtension;
  23 class AliAODJet;
  24 class AliGenPythiaEventHeader;
  25 class AliCFManager;
  26 class AliAODJetEventBackground;
  27 class AliJetFinder;
  28 class TList;
  29 class TChain;
  30 class TH2F;
  31 class TH1F;
  32 class TH3F;
  33 class TProfile;
  34 class TRefArray;
  35
  36
  37 class AliAnalysisTaskJetCluster : public AliAnalysisTaskSE
  38 {
  39  public:
  40     AliAnalysisTaskJetCluster();
  41     AliAnalysisTaskJetCluster(const char* name);
  42     virtual ~AliAnalysisTaskJetCluster();
  43     // Implementation of interface methods
  44     virtual void UserCreateOutputObjects();
  45     virtual void Init();
  46     virtual void LocalInit() { Init(); }
  47     virtual void UserExec(Option_t *option);
  48     virtual void Terminate(Option_t *option);
  49     virtual Bool_t Notify();
  50
  51
  52
  53     virtual void SetUseGlobalSelection(Bool_t b){fUseGlobalSelection = b;}
  54     virtual void SetAODTrackInput(Bool_t b){fUseAODTrackInput = b;}
  55     virtual void SetAODMCInput(Bool_t b){fUseAODMCInput = b;}
  56     virtual void SetRecEtaWindow(Float_t f){fRecEtaWindow = f;}
  57     virtual void SetTrackTypeGen(Int_t i){fTrackTypeGen = i;}
  58     virtual void SetTrackTypeRec(Int_t i){fTrackTypeRec = i;}
  59     virtual void SetTrackPtCut(Float_t x){fTrackPtCut = x;}
  60     virtual void SetFilterMask(UInt_t i){fFilterMask = i;}
  61
  62     virtual void SetNSkipLeadingRan(Int_t x){fNSkipLeadingRan = x;}
  63
  64     virtual void SetJetOutputBranch(const char *c){fNonStdBranch = c;}
  65     virtual void SetJetOutputFile(const char *c){fNonStdFile = c;}
  66     virtual void SetJetOutputMinPt(Float_t x){fJetOutputMinPt = x;}
  67
  68     // for Fast Jet
  69     fastjet::JetAlgorithm        GetAlgorithm()         const {return fAlgorithm;}
  70     fastjet::Strategy            GetStrategy()          const {return fStrategy;}
  71     fastjet::RecombinationScheme GetRecombScheme()      const {return fRecombScheme;}
  72     fastjet::AreaType            GetAreaType()          const {return fAreaType;}
  73     // Setters
  74     void SetRparam(Double_t f)                           {fRparam = f;}
  75     void SetAlgorithm(fastjet::JetAlgorithm f)           {fAlgorithm = f;}
  76     void SetStrategy(fastjet::Strategy f)                {fStrategy = f;}
  77     void SetRecombScheme(fastjet::RecombinationScheme f) {fRecombScheme = f;}
  78     void SetAreaType(fastjet::AreaType f)                {fAreaType = f;}
  79     void SetGhostArea(Double_t f) {fGhostArea = f;}
  80     void SetActiveAreaRepeats(Int_t f) {fActiveAreaRepeats = f;}
  81     void SetGhostEtamax(Double_t f) {fGhostEtamax = f;}
  82
  83
  84
  85     // Helper
  86     //
  87
  88     // we have different cases
  89     // AOD reading -> MC from AOD
  90     // ESD reading -> MC from Kinematics
  91     // this has to match with our selection of input events
  92     enum {kTrackUndef = 0, kTrackAOD, kTrackKineAll,kTrackKineCharged, kTrackAODMCAll, kTrackAODMCCharged, kTrackAODMCChargedAcceptance};
  93     enum {kMaxJets = 4};
  94     enum {kMaxCorrelation =  3};
  95     enum {kMaxRadius =       5};
  96
  97
  98  private:
  99
 100     AliAnalysisTaskJetCluster(const AliAnalysisTaskJetCluster&);
 101     AliAnalysisTaskJetCluster& operator=(const AliAnalysisTaskJetCluster&);
 102
 103     Int_t GetListOfTracks(TList *list,Int_t type);
 104
 105     AliAODEvent     *fAOD;                // ! where we take the jets from can be input or output AOD
 106     AliAODExtension *fAODExtension;       // ! AOD extension in case we write a non-sdt branch to a separate file and the aod is standard
 107     TRefArray       *fRef;               // ! trefarray for track references within the jet
 108     Bool_t        fUseAODTrackInput;      // take track from input AOD not from ouptu AOD
 109     Bool_t        fUseAODMCInput;         // take MC from input AOD not from ouptu AOD
 110     Bool_t        fUseGlobalSelection;    // Limit the eta of the generated jets
 111     UInt_t        fFilterMask;            // filter bit for slecected tracks
 112     Int_t         fTrackTypeRec;          // type of tracks used for FF
 113     Int_t         fTrackTypeGen;          // type of tracks used for FF
 114     Int_t         fNSkipLeadingRan;        // number of leading tracks to be skipped in the randomized event
 115     Float_t       fAvgTrials;             // Average nimber of trials
 116     Float_t       fExternalWeight;        // external weight
 117     Float_t       fRecEtaWindow;          // eta window used for corraltion plots between rec and gen
 118     Float_t       fTrackPtCut;            // minimum track pt to be accepted
 119     Float_t       fJetOutputMinPt;        // minimum p_t for jets to be written out
 120
 121     // output configurartion
 122     TString       fNonStdBranch;      // the name of the non-std branch name, if empty no branch is filled
 123     TString       fNonStdFile;        // The optional name of the output file the non-std brnach is written to
 124
 125
 126     // Fast jet
 127     Double_t fRparam;
 128     fastjet::JetAlgorithm fAlgorithm; //fastjet::kt_algorithm
 129     fastjet::Strategy fStrategy;  //= fastjet::Best;
 130     fastjet::RecombinationScheme fRecombScheme; // = fastjet::BIpt_scheme;
 131     fastjet::AreaType fAreaType;
 132     Double_t fGhostArea;
 133     Int_t fActiveAreaRepeats;
 134     Double_t fGhostEtamax;
 135     TProfile*     fh1Xsec;   //! pythia cross section and trials
 136     TH1F*         fh1Trials; //! trials are added
 137     TH1F*         fh1PtHard;  //! Pt har of the event...
 138     TH1F*         fh1PtHardNoW;  //! Pt har of the event without weigt
 139     TH1F*         fh1PtHardTrials;  //! Number of trials
 140
 141     TH1F*         fh1NJetsRec; //! number of reconstructed jets
 142     TH1F*         fh1NConstRec;//! number of constiutens in leading jet
 143     TH1F*         fh1NConstLeadingRec;//! number of constiutens in leading jet
 144     TH1F*         fh1PtJetsRecIn;  //! Jet pt for all jets
 145     TH1F*         fh1PtJetsLeadingRecIn;  //! Jet pt for all jets
 146     TH1F*         fh1PtJetConstRec;//! pt of constituents
 147     TH1F*         fh1PtJetConstLeadingRec;// pt of constituents
 148     TH1F*         fh1PtTracksRecIn;  //! track pt for all tracks
 149     TH1F*         fh1PtTracksLeadingRecIn;  //! track pt for all tracks
 150
 151     // Randomized track histos
 152     TH1F*         fh1NJetsRecRan; //! number of reconstructed jets from randomized
 153     TH1F*         fh1NConstRecRan;//! number of constiutens in leading jet
 154     TH1F*         fh1PtJetsLeadingRecInRan;  //! Jet pt for all jets
 155     TH1F*         fh1NConstLeadingRecRan;//! number of constiutens in leading jet
 156     TH1F*         fh1PtJetsRecInRan;  //! Jet pt for all jets
 157
 158     TH1F*         fh1PtTracksGenIn;  //! track pt for all tracks
 159     TH1F*         fh1Nch;            //! charged particle mult
 160     TH1F*         fh1BiARandomCones[3]; //! Residual distribtion from reandom cones on real event
 161     TH1F*         fh1BiARandomConesRan[3]; //! Residual distribtion from reandom cones on random event
 162
 163
 164     TH2F*         fh2NRecJetsPt;            //! Number of found jets above threshold
 165     TH2F*         fh2NRecTracksPt;          //! Number of found tracks above threshold
 166     TH2F*         fh2NConstPt;           //! number of constituents vs. pt
 167     TH2F*         fh2NConstLeadingPt;           //! number of constituents vs. pt
 168     TH2F*         fh2JetPhiEta;             //! jet phi eta
 169     TH2F*         fh2LeadingJetPhiEta;      //! leading jet phi eta
 170     TH2F*         fh2JetEtaPt;              //! leading jet eta
 171     TH2F*         fh2LeadingJetEtaPt;              //! leading jet eta
 172     TH2F*         fh2TrackEtaPt;              //! track eta
 173     TH2F*         fh2LeadingTrackEtaPt;       //! leading track eta
 174     TH2F*         fh2JetsLeadingPhiEta;     //! jet phi eta
 175     TH2F*         fh2JetsLeadingPhiPt;      //! jet correlation with leading jet
 176     TH2F*         fh2TracksLeadingPhiEta;   //! track correlation with leading track
 177     TH2F*         fh2TracksLeadingPhiPt;    //! track correlation with leading track
 178     TH2F*         fh2TracksLeadingJetPhiPt; //! track correlation with leading Jet
 179     TH2F*         fh2JetsLeadingPhiPtW;      //! jet correlation with leading jet
 180     TH2F*         fh2TracksLeadingPhiPtW;   //! track correlation with leading track
 181     TH2F*         fh2TracksLeadingJetPhiPtW; //! track correlation with leading Jet
 182     TH2F*         fh2NRecJetsPtRan;            //! Number of found jets above threshold
 183     TH2F*         fh2NConstPtRan;           //! number of constituents vs. pt
 184     TH2F*         fh2NConstLeadingPtRan;           //! number of constituents vs. pt
 185     TH2F*         fh2PtNch;               //! p_T of cluster vs. multiplicity,
 186     TH2F*         fh2PtNchRan;            //! p_T of cluster vs. multiplicity,random
 187     TH2F*         fh2PtNchN;               //! p_T of cluster vs. multiplicity, weigthed with constituents
 188     TH2F*         fh2PtNchNRan;            //! p_T of cluster vs. multiplicity, weigthed with constituents random
 189     TH2F*         fh2TracksLeadingJetPhiPtRan; //! track correlation with leading Jet
 190     TH2F*         fh2TracksLeadingJetPhiPtWRan; //! track correlation with leading Jet
 191
 192     TList *fHistList; // Output list
 193
 194
 195     ClassDef(AliAnalysisTaskJetCluster, 6)
 196 };
 197
 198 #endif