GetMedium replaced by CurrentMedium (I.Hrivnacova)
[u/mrichter/AliRoot.git] / TOF / AliTOFPID.h
1 #ifndef ALITOFPID_H
2 #define ALITOFPID_H
3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4  * See cxx source for full Copyright notice                               */
5
6
7 //_________________________________________________________________________
8 //  Task Class for PID in TOF      
9 //                  
10 //-- Authors: B. Zagreev, F. Pierella
11
12
13 #include "TTask.h"
14 #include "TCutG.h"
15 #include "TString.h"
16 #include <TF1.h>
17 #include <TF2.h>
18 #include <TH1.h>
19 #include <TH2.h>
20 #include <TNtuple.h>
21
22 class AliTOFPID: public TTask {
23
24 public:
25   AliTOFPID() ;          // ctor
26   AliTOFPID(char* headerFile, char *cutsFile, const Option_t* opt="pp") ; 
27   virtual ~AliTOFPID() ; // dtor   
28   virtual void  Exec(const Option_t *eventType, const Option_t *outputmode, const Option_t *outputsavemode); // do the main work
29   void          Init(const Option_t* opt);
30   void          SetNEvents(Int_t Nevents) {fNevents = Nevents;}
31   Int_t         GetNEvents() const {return fNevents;}
32   void  SetDbg(Int_t dbgflag)                        {fdbg=dbgflag;}
33   // getter for pointers to cuts
34   TCutG*        GetElectronCut() const {return felectron;}
35   TCutG*        GetPionCut()     const {return fpion;}
36   TCutG*        GetKaonCut()     const {return fkaon;}
37   TCutG*        GetProtonCut()   const {return fproton;}
38
39   TFile *       GetFileCuts()    const {return fcut;}
40   TNtuple*      GetInputNtuple() const {return fNtuple;}
41
42   Bool_t   operator == (const AliTOFPID & tofpid) const ;
43
44 private:
45   TCutG *felectron; // pointer to cut for electron
46   TCutG *fpion;     // pointer to cut for pion
47   TCutG *fkaon;     // pointer to cut for kaon
48   TCutG *fproton;   // pointer to cut for proton
49   TFile *fcut;      // pointer to file containing cuts
50   TNtuple *fNtuple; // pointer to ntuple 
51   TFile *fhfile;
52   TFile *fgen;
53   char* foutfileName; // destination file name for histos
54   Int_t fNevents; // number of events
55   Int_t fdbg;
56   Int_t fTask;
57
58  protected:
59
60   ClassDef(AliTOFPID,0)  // Task class for TOF pid
61
62 };
63
64 #endif // AliTOFPID_H