Update (Andrea)
[u/mrichter/AliRoot.git] / PWG3 / vertexingHF / AliAnalysisTaskSEDStarJets.h
1 #ifndef ALIANALYSISTASKSEDSTARJETS_H
2 #define ALIANALYSISTASKSEDSTARJETS_H
3 /**************************************************************************
4  * Copyright(c) 1998-2009, ALICE Experiment at CERN, All rights reserved. *
5  *                                                                        *
6  * Author: The ALICE Off-line Project.                                    *
7  * Contributors are mentioned in the code where appropriate.              *
8  *                                                                        *
9  * Permission to use, copy, modify and distribute this software and its   *
10  * documentation strictly for non-commercial purposes is hereby granted   *
11  * without fee, provided that the above copyright notice appears in all   *
12  * copies and that both the copyright notice and this permission notice   *
13  * appear in the supporting documentation. The authors make no claims     *
14  * about the suitability of this software for any purpose. It is          *
15  * provided "as is" without express or implied warranty.                  *
16  **************************************************************************/
17
18 //-----------------------------------------------------------------------
19 // Author : A. Grelli, UTRECHT
20 //-----------------------------------------------------------------------
21
22
23 #include <TH2F.h>
24 #include "AliAnalysisTaskSE.h"
25 #include "AliAODEvent.h"
26 #include "AliRDHFCutsDStartoKpipi.h"
27
28 class TParticle ;
29 class TClonesArray ;
30 class AliAODMCParticle;
31
32
33 class AliAnalysisTaskSEDStarJets : public AliAnalysisTaskSE 
34 {
35   
36  public:
37   
38   AliAnalysisTaskSEDStarJets();
39   AliAnalysisTaskSEDStarJets(const Char_t* name,AliRDHFCutsDStartoKpipi* cuts);
40   virtual ~AliAnalysisTaskSEDStarJets();
41   
42   virtual void     UserCreateOutputObjects();
43   virtual void     UserExec(Option_t *option);
44   virtual void     Terminate(Option_t *);
45   virtual void     Init();
46   virtual void     LocalInit() {Init();}
47
48   //side band background eval
49   void     SideBandBackground(Double_t finvM, Double_t finvMDStar,  Double_t dStarMomBkg, Double_t fejet, Double_t ejet);
50   
51   // inizializations
52   Bool_t   DefineHistoFroAnalysis();  
53   //MC FF
54   double   FillMCFF(AliAODMCParticle* mcPart, TClonesArray* mcArray, Int_t mcLabel);
55   // correction for UA1 cone algorithm
56   void     SetChargeFractionCorrection(Int_t chargeFrCorr) {fchargeFrCorr =  chargeFrCorr;}
57   Int_t    GetChargeFractionCorrection() const {return fchargeFrCorr;}
58
59   // set MC usage
60   void    SetMC(Bool_t theMCon) {fUseMCInfo = theMCon;}
61   Bool_t  GetMC() const {return fUseMCInfo;}
62   
63  private:
64   
65   AliAnalysisTaskSEDStarJets(const AliAnalysisTaskSEDStarJets &source);
66   AliAnalysisTaskSEDStarJets& operator=(const AliAnalysisTaskSEDStarJets& source); 
67
68   Int_t  fEvents;                //  n. of events
69   Int_t  fchargeFrCorr;          //  Charge fraction correction UA1 algorithm
70   Bool_t fUseMCInfo;             //  Use MC info
71   Bool_t fRequireNormalization;  //  normalization 
72   
73   TList *fOutput;                  //! user output
74   AliRDHFCutsDStartoKpipi *fCuts;  // Cuts 
75
76   // define the histograms 
77
78   TH1F *ftrigger;        //!
79   TH1F *fPtPion;         //!
80   TH1F *fInvMass;        //!
81   TH1F *fRECOPtDStar;    //!
82   TH1F *fRECOPtBkg;      //!
83   TH1F *fDStar;          //!
84   TH1F *fDiff;           //!
85   TH1F *fDiffSideBand;   //!
86   TH1F *fDStarMass;      //!
87   TH1F *fPhi;            //!
88   TH1F *fPhiBkg;         //!
89   TH1F *fTrueDiff;       //!
90   TH1F *fResZ;           //!
91   TH1F *fResZBkg;        //!  
92   TH1F *fEjet;           //!
93   TH1F *fPhijet;         //!
94   TH1F *fEtaJet;         //!
95   TH1F *theMCFF;         //!
96   TH1F *fDphiD0Dstar;    //!
97   TH1F *fPtJet;          //!
98
99   ClassDef(AliAnalysisTaskSEDStarJets,3); // class for charm-jet correlations
100 };
101
102 #endif