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