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