New task for Ds->KKpi analysis (Francesco)
[u/mrichter/AliRoot.git] / PWG3 / vertexingHF / AliAnalysisTaskSEDs.h
CommitLineData
25c94fc8 1#ifndef ALIANALYSISTASKDS_H
2#define ALIANALYSISTASKDS_H
3
4/* Copyright(c) 2007-2009, ALICE Experiment at CERN, All rights reserved. *
5 * See cxx source for full Copyright notice */
6
7
8/* $Id: $ */
9
10///////////////////////////////////////////////////////////////////
11// //
12// Analysis task to produce Ds candidates mass spectra //
13// Origin: F.Prino, Torino, prino@to.infn.it //
14// //
15///////////////////////////////////////////////////////////////////
16
17#include <TROOT.h>
18#include <TSystem.h>
19#include <TH1F.h>
20#include <TH2F.h>
21
22#include "AliAnalysisTaskSE.h"
23#include "AliAnalysisVertexingHF.h"
24
25class AliAnalysisTaskSEDs : public AliAnalysisTaskSE
26{
27 public:
28
29 AliAnalysisTaskSEDs();
30 AliAnalysisTaskSEDs(const char *name);
31 virtual ~AliAnalysisTaskSEDs();
32 void SetReadMC(Bool_t readMC=kTRUE){fReadMC=readMC;}
33 void SetMassRange(Double_t rang=0.2){fMassRange=rang;}
34 void SetPtBins(Int_t n, Double_t* lim);
35
36 // Implementation of interface methods
37 virtual void UserCreateOutputObjects();
38 virtual void Init();
39 virtual void LocalInit() {Init();}
40 virtual void UserExec(Option_t *option);
41 virtual void Terminate(Option_t *option);
42
43 private:
44 Int_t GetHistoIndex(Int_t iPtBin) const { return iPtBin*3;}
45 Int_t GetSignalHistoIndex(Int_t iPtBin) const { return iPtBin*3+1;}
46 Int_t GetBackgroundHistoIndex(Int_t iPtBin) const { return iPtBin*3+2;}
47
48 enum {kMaxPtBins=10};
49
50 AliAnalysisTaskSEDs(const AliAnalysisTaskSEDs &source);
51 AliAnalysisTaskSEDs& operator=(const AliAnalysisTaskSEDs& source);
52
53 TList* fOutput; //! list send on output slot 0
54 TH1F* fHistNEvents; //! hist. for No. of events
55 TH1F* fChanHist[3]; //! hist. with KKpi and piKK candidates (sig,bkg,tot)
56 TH1F* fChanHistCuts[3]; //! hist. with KKpi and piKK candidates analysis cuts
57 TH1F* fMassHist[3*kMaxPtBins]; //! hist. of mass spectra (sig,bkg,tot)
58 TH1F* fMassHistCuts[3*kMaxPtBins];//! hist. of mass spectra (sig,bkg,tot) analysis cuts
59 TH1F* fCosPHist[3*kMaxPtBins]; //! hist. of cos pointing angle (sig,bkg,tot)
60 TH1F* fDLenHist[3*kMaxPtBins]; //! hist. of decay length (sig,bkg,tot)
61 TH2F* fDalitz[3*kMaxPtBins]; //! dalitz plot (sig,bkg,tot)
62 Bool_t fReadMC; // flag for access to MC
63 UChar_t fNPtBins; // number of Pt bins
64 Double_t fPtLimits[kMaxPtBins+1]; // limits for pt bins
65 Double_t fMassRange; // range for mass histogram
66 AliAnalysisVertexingHF *fVHF; // Heavy flavour analysis object
67
68 ClassDef(AliAnalysisTaskSEDs,1); // AliAnalysisTaskSE for Ds mass spectra
69};
70
71#endif
72