EMCAL/AliEMCALJetFinder.h

   1 #ifndef ALIEMCALJETFINDER_H
   2 #define ALIEMCALJETFINDER_H
   3
   4 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   5  * See cxx source for full Copyright notice     */
   6
   7 /* $Id$ */
   8 //*-- Author:
   9 //*-- Andreas Morsch (CERN)
  10
  11 #include <TTask.h>
  12 #include "AliEMCALJet.h"
  13
  14 class TClonesArray;
  15 class TH2F;
  16 class TH1F;
  17 class TCanvas;
  18 class TList;
  19 class AliEMCALHadronCorrection;
  20
  21 class AliEMCALJetFinder : public TTask {
  22  public:
  23     AliEMCALJetFinder();
  24     AliEMCALJetFinder(const char* name, const char *title);
  25     virtual ~AliEMCALJetFinder();
  26     virtual void  Init();
  27     virtual void  Find(Int_t ncell, Int_t ncell_tot, Float_t etc[30000],
  28                       Float_t etac[30000], Float_t phic[30000],
  29                       Float_t min_move, Float_t max_move, Int_t mode,
  30                       Float_t prec_bg, Int_t ierror);
  31     virtual void  Find();
  32     virtual void  FindTracksInJetCone();
  33     virtual void  Test();
  34     virtual void  BuildTrackFlagTable();
  35     virtual Int_t SetTrackFlag(Float_t radius, Int_t pdgCode, Double_t charge);
  36     // Geometry
  37     virtual void SetCellSize(Float_t eta, Float_t phi);
  38     // Parameters
  39     virtual void SetDebug(Int_t flag = 0) {fDebug = flag;}
  40     virtual void SetConeRadius(Float_t par);
  41     virtual void SetEtSeed(Float_t par);
  42     virtual void SetMinJetEt(Float_t par);
  43     virtual void SetMinCellEt(Float_t par);
  44     virtual void SetPtCut(Float_t par = 1.);
  45     virtual void SetMomentumSmearing(Bool_t flag = kFALSE) {fSmear = flag;}
  46     virtual void SetEfficiencySim(Bool_t flag = kFALSE)    {fEffic = flag;}
  47     virtual void SetSamplingFraction(Float_t par = 12.9) {fSamplingF = par;}
  48     virtual void SetIncludeK0andN(Bool_t flag = kFALSE) {fK0N = flag;}
  49     // Correction of hadronic energy
  50     virtual void SetHadronCorrector(AliEMCALHadronCorrection* corr)
  51         {fHadronCorrector = corr;}
  52     virtual void SetHadronCorrection(Int_t flag = 1) {fHCorrection = flag;}
  53     // PAI
  54     void SetWriteKey(Bool_t flag = kFALSE) {fWrite = flag;}
  55     void SetMode(Int_t mode = 0) {fMode = mode;}
  56     void SetMinMove(Float_t minMove = 0.05) {fMinMove = minMove;}
  57     void SetMaxMove(Float_t maxMove = 0.15) {fMaxMove = maxMove;}
  58     void SetPrecBg (Float_t precBg = 0.035) {fPrecBg = precBg;}
  59     void SetParametersForBgSubtraction
  60     (Int_t mode=0, Float_t minMove=0.05, Float_t maxMove=0.15, Float_t precBg=0.035);
  61     //    virtual void Print(Option_t* option="") const;    // *MENU*
  62
  63     Bool_t GetWriteKey() {return fWrite;}
  64     AliEMCALJet* GetJetT() {return fJetT[0];}
  65     virtual void DrawHistsForTuning(Int_t mode=0);           // *MENU*
  66     virtual void PrintParameters(Int_t mode=0);              // *MENU*
  67     virtual const Char_t* GetFileNameForParameters(Char_t* dir="RES/");
  68
  69     // Access to Results
  70     virtual Int_t   Njets();
  71     virtual Float_t JetEnergy(Int_t);
  72     virtual Float_t JetPhiL(Int_t);
  73     virtual Float_t JetPhiW(Int_t);
  74     virtual Float_t JetEtaL(Int_t);
  75     virtual Float_t JetEtaW(Int_t);
  76     TH2F*   GetLego()  {return fLego;}
  77     TH2F*   GetLegoB() {return fLegoB;}
  78     TH2F*   GetLegoEMCAL() {return fhLegoEMCAL;}
  79     TH2F*   GethEff() {return fhEff;}
  80     TH1F*   GetCellEt() {return fhCellEt;}
  81     TH1F*   GetCellEMCALEt() {return fhCellEMCALEt;}
  82     TH1F*   GetTrackPt() {return fhTrackPt;}
  83     TH1F*   GetTrackPtBcut() {return fhTrackPtBcut;}
  84     TList*  GetHistsList() {return fHistsList;}
  85     Int_t   GetNChTpc() {return fNChTpc;}
  86     void    DrawLego(Char_t *opt="lego");         // *MENU*
  87     void    DrawLegoEMCAL(Char_t *opt="lego");    // *MENU*
  88     void    DrawLegos();                          // *MENU*
  89     Bool_t  IsThisPartonsOrDiQuark(Int_t pdg);
  90     TString &GetPythiaParticleName(Int_t kf);
  91     // I/O
  92     virtual void SetOutputFileName(char* name) {fOutFileName = name;}
  93     virtual void FillFromHits(Int_t flag = 0);
  94     virtual void FillFromHitFlaggedTracks(Int_t flag = 0);
  95     virtual void FillFromDigits(Int_t flag = 0);
  96     virtual void FillFromTracks(Int_t flag = 0, Int_t ich = 0);
  97     virtual void FillFromParticles();
  98     virtual void FillFromPartons();
  99
 100     virtual void SaveBackgroundEvent();
 101     virtual void InitFromBackground();
 102     virtual void AddJet(const AliEMCALJet& jet);
 103     virtual void WriteJets();
 104     virtual void ResetJets();
 105     virtual TClonesArray* Jets() {return fJets;}
 106  private:
 107     virtual void BookLego();
 108     virtual void DumpLego();
 109     virtual void ResetMap();
 110     virtual Float_t PropagatePhi(Float_t pt, Float_t charge, Bool_t& curls);
 111     virtual void RearrangeParticlesMemory(Int_t npart);
 112  protected:
 113     Bool_t                         fWrite;           // Key for writing
 114     TClonesArray*                  fJets;            //! List of Jets
 115     TH2F*                          fLego;            //! Lego Histo
 116     TH2F*                          fLegoB;           //! Lego Histo Backg
 117     TH2F*                          fhLegoTracks;     //! Lego for Tracks
 118     TH2F*                          fhLegoEMCAL;      //! Lego for EMCAL itself
 119     TH2F*                          fhLegoHadrCorr;   //! Lego for hadron correction
 120     TH2F*                          fhEff;            //! Hist. for controling eff.
 121     TH1F*                          fhCellEt;         //! Et distr. for cells from fLego
 122     TH1F*                          fhCellEMCALEt;    //! Et distr. for cells from fLegoEMCAL
 123     TH1F*                          fhTrackPt;        //! Pt distr. for charge particles
 124     TH1F*                          fhTrackPtBcut;    //! Pt distr. for charge particles + cut due to magnetic field
 125     TH1F*                          fhChPartMultInTpc;//! Ch. part. multiplicity in TPC acceptance
 126     TCanvas*                       fC1;              //! first canvas for drawing
 127     TList*                         fHistsList;       //! List of hists - 4-mar-2002
 128     AliEMCALJet*                   fJetT[10];        //! Jet temporary storage
 129     AliEMCALHadronCorrection*      fHadronCorrector; //! Pointer to hadronic correction
 130     Int_t                          fHCorrection;     //  Hadron correction flag
 131     Int_t                          fDebug;           //! Debug flag
 132     Int_t                          fBackground;      //! Background flag
 133     Float_t                        fConeRadius;      //  Cone radius
 134     Float_t                        fPtCut;           //  Pt cut on charged tracks
 135     Float_t                        fEtSeed;          //  Min. Et for seed
 136     Float_t                        fMinJetEt;        //  Min Et of jet
 137     Float_t                        fMinCellEt;       //  Min Et in one cell
 138     Float_t                        fSamplingF;       //  Sampling Fraction
 139     Bool_t                         fSmear;           //  Flag for momentum smearing
 140     Bool_t                         fEffic;           //  Flag for efficiency simulation
 141     Bool_t                         fK0N;             //  Flag for efficiency simulation
 142     Int_t                          fNjets;           //! Number of Jetsp
 143     Float_t                        fDeta;            //! eta cell size
 144     Float_t                        fDphi;            //! phi cell size
 145     Int_t                          fNcell;           //! number of cells
 146     Int_t                          fNtot;            //! total number of cells
 147     Int_t                          fNbinEta;         //! number of cells in eta
 148     Int_t                          fNbinPhi;         //! number of cells in phi
 149     Float_t                        fEtaMin;          //! minimum eta
 150     Float_t                        fEtaMax;          //! maximum eta
 151     Float_t                        fPhiMin;          //! minimun phi
 152     Float_t                        fPhiMax;          //! maximum phi
 153     Float_t                        fEtCell[30000];   //! Cell Energy
 154     Float_t                        fEtaCell[30000];  //! Cell eta
 155     Float_t                        fPhiCell[30000];  //! Cell phi
 156     Int_t                          fNt;              //! number of tracks
 157     Int_t                          fNChTpc;          //! number of ch.part in TPC
 158
 159     Int_t                          fNtS;             //! number of tracks selected
 160     Int_t*                         fTrackList;       //! List of selected tracks
 161     Float_t*                       fPtT;             //! Pt   of tracks
 162     Float_t*                       fEtaT;            //! Eta  of tracks
 163     Float_t*                       fPhiT;            //! Phi  of tracks
 164
 165     Int_t                          fNtB;             //! number of tracks in Bg
 166     Int_t*                         fTrackListB;      //! List of selected tracks in Bg
 167     Float_t*                       fPtB;             //! Pt   of tracks in Bg
 168     Float_t*                       fEtaB;            //! Eta  of tracks in Bg
 169     Float_t*                       fPhiB;            //! Phi  of tracks in Bg
 170
 171     // parameter for jet_finder_ua1
 172     Int_t                          fMode;            // key for BG subtraction
 173     Float_t                        fMinMove;         // min cone move
 174     Float_t                        fMaxMove;         // max cone move
 175     Float_t                        fPrecBg;          // max value of change for BG (in %)
 176     Int_t                          fError;           // error variables
 177
 178     char*                          fOutFileName;     //! Output file name
 179     TFile*                         fOutFile;         //! Output file
 180     TFile*                         fInFile;          //! Output file
 181     Int_t                          fEvent;           //! Processed event
 182
 183     ClassDef(AliEMCALJetFinder,2)        // JetFinder for EMCAL
 184 }
 185 ;
 186 #endif // ALIEMCALJetFinder_H