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