[u/mrichter/AliRoot.git] / TOF / AliTOFPID.h

#ifndef ALITOFPID_H
#define ALITOFPID_H
/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */


//_________________________________________________________________________
//  Task Class for PID in TOF      
//                  
//-- Authors: B. Zagreev, F. Pierella


#include "TTask.h"
#include "TCutG.h"
#include "TString.h"
#include <TF1.h>
#include <TF2.h>
#include <TH1.h>
#include <TH2.h>
#include <TNtuple.h>

class AliTOFPID: public TTask {

public:
  AliTOFPID() ;          // ctor
  AliTOFPID(char* headerFile, char *cutsFile, const Option_t* opt="pp") ; 
  virtual ~AliTOFPID() ; // dtor   
  virtual void  Exec(const Option_t *eventType, const Option_t *outputmode, const Option_t *outputsavemode); // do the main work
  void          Init(const Option_t* opt);
  void          SetNEvents(Int_t Nevents) {fNevents = Nevents;}
  Int_t         GetNEvents() const {return fNevents;}
  void  SetDbg(Int_t dbgflag)                        {fdbg=dbgflag;}
  // getter for pointers to cuts
  TCutG*        GetElectronCut() const {return felectron;}
  TCutG*        GetPionCut()     const {return fpion;}
  TCutG*        GetKaonCut()     const {return fkaon;}
  TCutG*        GetProtonCut()   const {return fproton;}

  TFile *       GetFileCuts()    const {return fcut;}
  TNtuple*      GetInputNtuple() const {return fNtuple;}

  Bool_t   operator == (const AliTOFPID & tofpid) const ;

private:
  TCutG *felectron; // pointer to cut for electron
  TCutG *fpion;     // pointer to cut for pion
  TCutG *fkaon;     // pointer to cut for kaon
  TCutG *fproton;   // pointer to cut for proton
  TFile *fcut;      // pointer to file containing cuts
  TNtuple *fNtuple; // pointer to ntuple 
  TFile *fhfile;
  TFile *fgen;
  char* foutfileName; // destination file name for histos
  Int_t fNevents; // number of events
  Int_t fdbg;
  Int_t fTask;

 protected:

  ClassDef(AliTOFPID,0)  // Task class for TOF pid

};

#endif // AliTOFPID_H
Commit	Line	Data
190c1f49	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;}
8488c769	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;}
190c1f49	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
5fff655e	60	ClassDef(AliTOFPID,0) // Task class for TOF pid
190c1f49	61
	62	};
	63
	64	#endif // AliTOFPID_H