Fill ntuple only on request; more mass histos (Francesco, Renu)
[u/mrichter/AliRoot.git] / PWG3 / vertexingHF / AliAnalysisTaskSEDplus.h
1 #ifndef ALIANALYSISTASKDPLUS_H
2 #define ALIANALYSISTASKDPLUS_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 AliAnalysisTaskSEDplus
9 // AliAnalysisTaskSE for the comparison of heavy-flavour decay candidates
10 // to MC truth (kinematics stored in the AOD)
11
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 AliAnalysisTaskSEDplus : public AliAnalysisTaskSE
23 {
24  public:
25
26   AliAnalysisTaskSEDplus();
27   AliAnalysisTaskSEDplus(const char *name, Bool_t fillNtuple=kFALSE);
28   virtual ~AliAnalysisTaskSEDplus();
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   AliAnalysisTaskSEDplus(const AliAnalysisTaskSEDplus &source);
42   AliAnalysisTaskSEDplus& operator=(const AliAnalysisTaskSEDplus& source); 
43   TList   *fOutput; //! list send on output slot 0
44   TNtuple *fNtupleDplus; //! output ntuple
45   TNtuple *fNtupleDplusbackg; //! output ntuple
46   Bool_t fFillNtuple;   // flag for filling ntuple
47   AliAnalysisVertexingHF *fVHF;  // Vertexer heavy flavour (used to pass the cuts)
48   
49   ClassDef(AliAnalysisTaskSEDplus,4); // AliAnalysisTaskSE for the MC association of heavy-flavour decay candidates
50 };
51
52 #endif
53