Included also 3 and 4 prong decays; added variables to ntuple
[u/mrichter/AliRoot.git] / PWG3 / vertexingHF / AliAnalysisTaskSECompareHF.h
1 #ifndef ALIANALYSISTASKCOMPAREHF_H
2 #define ALIANALYSISTASKCOMPAREHF_H
3
4 /* Copyright(c) 1998-2008, ALICE Experiment at CERN, All rights reserved. *
5  * See cxx source for full Copyright notice                               */
6
7 //*************************************************************************
8 // Class AliAnalysisTaskSECompareHF
9 // AliAnalysisTaskSE for the comparison of heavy-flavour decay candidates
10 // to MC truth (kinematics stored in the AOD)
11 // Author: A.Dainese, andrea.dainese@ln.infn.it
12 //*************************************************************************
13
14 #include <TROOT.h>
15 #include <TSystem.h>
16 #include <TNtuple.h>
17 #include <TH1F.h>
18
19 #include "AliAnalysisTaskSE.h"
20 #include "AliAnalysisVertexingHF.h"
21
22 class AliAnalysisTaskSECompareHF : public AliAnalysisTaskSE
23 {
24  public:
25
26   AliAnalysisTaskSECompareHF();
27   AliAnalysisTaskSECompareHF(const char *name);
28   virtual ~AliAnalysisTaskSECompareHF();
29
30
31   // Implementation of interface methods
32   virtual void UserCreateOutputObjects();
33   virtual void Init();
34   virtual void LocalInit() {Init();}
35   virtual void UserExec(Option_t *option);
36   virtual void Terminate(Option_t *option);
37
38   
39  private:
40
41   AliAnalysisTaskSECompareHF(const AliAnalysisTaskSECompareHF &source);
42   AliAnalysisTaskSECompareHF& operator=(const AliAnalysisTaskSECompareHF& source); 
43   TList   *fOutput; //! list send on output slot 0
44   TNtuple *fNtupleCmp; //! output ntuple
45   TH1F    *fHistMass;    //! output histogram
46   AliAnalysisVertexingHF *fVHF;  // Vertexer heavy flavour (used to pass the cuts)
47   
48   ClassDef(AliAnalysisTaskSECompareHF,3); // AliAnalysisTaskSE for the MC association of heavy-flavour decay candidates
49 };
50
51 #endif
52