2c84a264818716e59dcf5cc79a151c2c186e02c1
[u/mrichter/AliRoot.git] / PWG3 / vertexingHF / AliAnalysisTaskSEDStarSpectra.h
1 #ifndef ALIANALYSISTASKSEDSTARSPECTRA_H
2 #define ALIANALYSISTASKSEDSTARSPECTRA_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 /* $Id$ */ 
19
20 #include <TH2F.h>
21 #include "TROOT.h"
22 #include "TSystem.h"
23
24 #include "AliAnalysisTaskSE.h"
25 #include "AliAODEvent.h"
26 #include "AliRDHFCutsDStartoKpipi.h"
27 #include "AliNormalizationCounter.h"
28
29 class AliAnalysisTaskSEDStarSpectra : public AliAnalysisTaskSE 
30 {
31   
32  public:
33   
34   AliAnalysisTaskSEDStarSpectra();
35   AliAnalysisTaskSEDStarSpectra(const Char_t* name,AliRDHFCutsDStartoKpipi* cuts);
36   virtual ~AliAnalysisTaskSEDStarSpectra();
37
38   // Implementation of interface methods  
39   virtual void UserCreateOutputObjects();
40   virtual void Init();
41   virtual void LocalInit() {Init();}
42   virtual void UserExec(Option_t *option);
43   virtual void Terminate(Option_t *option);
44  
45
46  //Background simulation
47   void     SideBandBackground(AliAODRecoCascadeHF *part, AliRDHFCutsDStartoKpipi *cuts, TList *listout);
48   void     WrongSignForDStar(AliAODRecoCascadeHF *part, AliRDHFCutsDStartoKpipi *cuts, TList *listout);
49     // histos
50   void   FillSpectrum(AliAODRecoCascadeHF *part, Int_t isDStar, AliRDHFCutsDStartoKpipi *cuts, TList *listout);
51   void     DefineHistograms();
52   // set analysis type
53   void     SetAnalysisType(Int_t anaType) {fAnalysis = anaType;}
54   void     PrintAnalysisType() {printf("Analysis type: %d\n(0: Heidelberg\t1: Utrecht)",fAnalysis);}
55  // set MC usage
56   void     SetMC(Bool_t theMCon) {fUseMCInfo = theMCon;}
57   Bool_t   GetMC() const {return fUseMCInfo;}
58  // set rare mesons
59   void     SetRareSearch(Bool_t theRareOn) {fDoSearch = theRareOn;}
60   Bool_t   GetRareSearch() const {return fDoSearch;}
61   
62  private:
63   
64   AliAnalysisTaskSEDStarSpectra(const AliAnalysisTaskSEDStarSpectra &source);
65   AliAnalysisTaskSEDStarSpectra& operator=(const AliAnalysisTaskSEDStarSpectra& source); 
66   
67   Int_t  fEvents;                //  n. of events
68   Int_t  fAnalysis;              //  0: HD;     1: UU;
69   Double_t fD0Window;            //  select width on D0Mass
70   Double_t fPeakWindow;          //  select width on DstarMass
71   Bool_t fUseMCInfo;             //  Use MC info
72   Bool_t fDoSearch;              //  Rare mesons
73   TList *fOutput;                //!  User output
74   TList *fOutputAll;             //!  User output2
75   TList *fOutputPID;             //!  User output3
76   Int_t  fNSigma;                //  n sigma for kaon PID
77   AliRDHFCutsDStartoKpipi *fCuts; // Cuts - sent to output slot 3
78   // define the histograms
79   TH1F *fCEvents;             //!
80   TH2F *fTrueDiff2;           //!
81   TH1F *fDeltaMassD1;         //! 
82   AliNormalizationCounter *fCounter;//!Counter for normalization slot 4
83   ClassDef(AliAnalysisTaskSEDStarSpectra,8); // class for D* spectra
84 };
85
86 #endif
87