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