[u/mrichter/AliRoot.git] / PWG / CaloTrackCorrBase / AliFiducialCut.h

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

//_________________________________________________________________________
// Class for track/cluster acceptance selection
// Selection in Central barrel, EMCAL and PHOS
//  
// Several selection regions possible for the different
// detectors
//
//*-- Author: Gustavo Conesa (INFN-LNF)

// --- ROOT system ---
#include <TObject.h> 
#include <TArrayF.h> 

class TString ;
class TLorentzVector ;


class AliFiducialCut : public TObject {
  
public: 
  AliFiducialCut() ; // ctor
  virtual  ~AliFiducialCut() ;//virtual dtor
  
  void      InitParameters();
    
  Bool_t    CheckFiducialRegion(TLorentzVector lv,
                                const TArrayF* minphi, const TArrayF* maxphi, 
                                const TArrayF* mineta, const TArrayF* maxeta) const ;
  
  Bool_t    IsInFiducialCut    (TLorentzVector lv, TString det) const ;
  
  void      DoCTSFiducialCut  (Bool_t b)     { fCTSFiducialCut   = b    ; }
  void      DoEMCALFiducialCut(Bool_t b)     { fEMCALFiducialCut = b    ; }
  void      DoPHOSFiducialCut (Bool_t b)     { fPHOSFiducialCut  = b    ; }
  
  Bool_t    GetCTSFiducialCutStatus()  const { return fCTSFiducialCut   ; }
  Bool_t    GetEMCALFiducialCut()      const { return fEMCALFiducialCut ; }
  Bool_t    GetPHOSFiducialCutStatus() const { return fPHOSFiducialCut  ; }
  
  void      SetSimpleCTSFiducialCut  (Float_t abseta, Float_t phimin, Float_t phimax) ;
  void      SetSimpleEMCALFiducialCut(Float_t abseta, Float_t phimin, Float_t phimax) ;
  void      SetSimplePHOSFiducialCut (Float_t abseta, Float_t phimin, Float_t phimax) ;
  
  void      Print(const Option_t * opt)const;
  
  void      AddCTSFidCutMaxEtaArray(Int_t size, Float_t* array)  
                                              { fCTSFidCutMaxEta->Set(size,array)   ; } 
  TArrayF * GetCTSFidCutMaxEtaArray() const   { return fCTSFidCutMaxEta             ; }
  
  void      AddCTSFidCutMaxPhiArray(Int_t size, Float_t* array)  
                                              { fCTSFidCutMaxPhi->Set(size,array)   ; }
  TArrayF * GetCTSFidCutMaxPhiArray() const   { return fCTSFidCutMaxPhi             ; }
  
  void      AddCTSFidCutMinEtaArray(Int_t size, Float_t* array)  
                                              { fCTSFidCutMinEta->Set(size,array)   ; } 
  TArrayF * GetCTSFidCutMinEtaArray() const   { return fCTSFidCutMinEta             ; }
  
  void      AddCTSFidCutMinPhiArray(Int_t size, Float_t* array)  
                                              { fCTSFidCutMinPhi->Set(size,array)   ; }
  TArrayF * GetCTSFidCutMinPhiArray() const   { return fCTSFidCutMinPhi             ; }
  
  void      AddEMCALFidCutMaxEtaArray(Int_t size, Float_t* array)  
                                              { fEMCALFidCutMaxEta->Set(size,array) ; } 
  TArrayF * GetEMCALFidCutMaxEtaArray() const { return fEMCALFidCutMaxEta           ; }
  
  void      AddEMCALFidCutMaxPhiArray(Int_t size, Float_t* array)  
                                              { fEMCALFidCutMaxPhi->Set(size,array) ; }
  TArrayF * GetEMCALFidCutMaxPhiArray() const { return fEMCALFidCutMaxPhi           ; }
  
  void      AddEMCALFidCutMinEtaArray(Int_t size, Float_t* array)  
                                              { fEMCALFidCutMinEta->Set(size,array) ; } 
  TArrayF * GetEMCALFidCutMinEtaArray() const { return fEMCALFidCutMinEta           ; }
  
  void      AddEMCALFidCutMinPhiArray(Int_t size, Float_t* array)  
                                              { fEMCALFidCutMinPhi->Set(size,array) ; }
  TArrayF * GetEMCALFidCutMinPhiArray() const { return fEMCALFidCutMinPhi           ; }
  
  void      AddPHOSFidCutMaxEtaArray(Int_t size, Float_t* array)  
                                              { fPHOSFidCutMaxEta->Set(size,array)  ; } 
  TArrayF * GetPHOSFidCutMaxEtaArray() const  { return fPHOSFidCutMaxEta            ; }
  
  void      AddPHOSFidCutMaxPhiArray(Int_t size, Float_t* array)  
                                              { fPHOSFidCutMaxPhi->Set(size,array)  ; }
  TArrayF * GetPHOSFidCutMaxPhiArray() const  { return fPHOSFidCutMaxPhi            ; }
  
  void      AddPHOSFidCutMinEtaArray(Int_t size, Float_t* array)  
                                              { fPHOSFidCutMinEta->Set(size,array)  ; } 
  TArrayF * GetPHOSFidCutMinEtaArray() const  { return fPHOSFidCutMinEta            ; }

  void      AddPHOSFidCutMinPhiArray(Int_t size, Float_t* array)  
                                              { fPHOSFidCutMinPhi->Set(size,array)  ; }
  TArrayF * GetPHOSFidCutMinPhiArray() const  { return fPHOSFidCutMinPhi            ; }
  
private:
  
  //Detector acceptance cuts
  Bool_t    fEMCALFiducialCut ;  // Apply fiducial cuts to EMCAL clusters
  Bool_t    fPHOSFiducialCut  ;  // Apply fiducial cuts to PHOS clusters
  Bool_t    fCTSFiducialCut   ;  // Apply fiducial cuts to  CTS tracks
  
  TArrayF * fCTSFidCutMinEta ;   // Take particles in CTS with eta > fCTSFidCutMinEta
  TArrayF * fCTSFidCutMinPhi ;   // Take particles in CTS with phi > fCTSFidCutMinPhi
  TArrayF * fCTSFidCutMaxEta ;   // Take particles in CTS with eta < fCTSFidCutMaxEta
  TArrayF * fCTSFidCutMaxPhi ;   // Take particles in CTS with phi > fCTSFidCutMaxPhi
  
  TArrayF * fEMCALFidCutMinEta ; // Take particles in EMCAL with eta > fEMCALFidCutMinEta
  TArrayF * fEMCALFidCutMinPhi ; // Take particles in EMCAL with phi > fEMCALFidCutMinPhi
  TArrayF * fEMCALFidCutMaxEta ; // Take particles in EMCAL with eta < fEMCALFidCutMaxEta
  TArrayF * fEMCALFidCutMaxPhi ; // Take particles in EMCAL with phi > fEMCALFidCutMaxPhi
  
  TArrayF * fPHOSFidCutMinEta ;  // Take particles in PHOS with eta > fPHOSFidCutMinEta
  TArrayF * fPHOSFidCutMinPhi ;  // Take particles in PHOS with phi > fPHOSFidCutMinPhi
  TArrayF * fPHOSFidCutMaxEta ;  // Take particles in PHOS with eta < fPHOSFidCutMaxEta
  TArrayF * fPHOSFidCutMaxPhi ;  // Take particles in PHOS with phi > fPHOSFidCutMaxPhi
  
  AliFiducialCut(              const AliFiducialCut & g) ; // cpy ctor
  AliFiducialCut & operator = (const AliFiducialCut & g) ; // cpy assignment
  
  ClassDef(AliFiducialCut,1)
  
} ;


#endif //ALIFIDUCIALCUT_H
Commit	Line	Data
ff45398a	1	#ifndef ALIFIDUCIALCUT_H
ff45398a	2	#define ALIFIDUCIALCUT_H
1c5acb87	3	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
1c5acb87	4	* See cxx source for full Copyright notice */
1c5acb87	5
	6	//_________________________________________________________________________
	7	// Class for track/cluster acceptance selection
	8	// Selection in Central barrel, EMCAL and PHOS
5efec477	9	//
	10	// Several selection regions possible for the different
	11	// detectors
1c5acb87	12	//
	13	//*-- Author: Gustavo Conesa (INFN-LNF)
	14
	15	// --- ROOT system ---
	16	#include <TObject.h>
	17	#include <TArrayF.h>
	18
	19	class TString ;
	20	class TLorentzVector ;
	21
1c5acb87	22
ff45398a	23	class AliFiducialCut : public TObject {
c5693f62	24
c5693f62	25	public:
ff45398a	26	AliFiducialCut() ; // ctor
c5693f62	27	virtual ~AliFiducialCut() ;//virtual dtor
78219bac	28
c5693f62	29	void InitParameters();
c5693f62	30
8a2dbbff	31	Bool_t CheckFiducialRegion(TLorentzVector lv,
c5693f62	32	const TArrayF* minphi, const TArrayF* maxphi,
c5693f62	33	const TArrayF* mineta, const TArrayF* maxeta) const ;
1c5acb87	34
8a2dbbff	35	Bool_t IsInFiducialCut (TLorentzVector lv, TString det) const ;
1c5acb87	36
b960c7eb	37	void DoCTSFiducialCut (Bool_t b) { fCTSFiducialCut = b ; }
	38	void DoEMCALFiducialCut(Bool_t b) { fEMCALFiducialCut = b ; }
	39	void DoPHOSFiducialCut (Bool_t b) { fPHOSFiducialCut = b ; }
c5693f62	40
b960c7eb	41	Bool_t GetCTSFiducialCutStatus() const { return fCTSFiducialCut ; }
	42	Bool_t GetEMCALFiducialCut() const { return fEMCALFiducialCut ; }
	43	Bool_t GetPHOSFiducialCutStatus() const { return fPHOSFiducialCut ; }
c5693f62	44
8a2dbbff	45	void SetSimpleCTSFiducialCut (Float_t abseta, Float_t phimin, Float_t phimax) ;
	46	void SetSimpleEMCALFiducialCut(Float_t abseta, Float_t phimin, Float_t phimax) ;
	47	void SetSimplePHOSFiducialCut (Float_t abseta, Float_t phimin, Float_t phimax) ;
c5693f62	48
c5693f62	49	void Print(const Option_t * opt)const;
c5693f62	50
	51	void AddCTSFidCutMaxEtaArray(Int_t size, Float_t* array)
	52	{ fCTSFidCutMaxEta->Set(size,array) ; }
	53	TArrayF * GetCTSFidCutMaxEtaArray() const { return fCTSFidCutMaxEta ; }
	54
	55	void AddCTSFidCutMaxPhiArray(Int_t size, Float_t* array)
	56	{ fCTSFidCutMaxPhi->Set(size,array) ; }
	57	TArrayF * GetCTSFidCutMaxPhiArray() const { return fCTSFidCutMaxPhi ; }
	58
	59	void AddCTSFidCutMinEtaArray(Int_t size, Float_t* array)
	60	{ fCTSFidCutMinEta->Set(size,array) ; }
	61	TArrayF * GetCTSFidCutMinEtaArray() const { return fCTSFidCutMinEta ; }
	62
	63	void AddCTSFidCutMinPhiArray(Int_t size, Float_t* array)
	64	{ fCTSFidCutMinPhi->Set(size,array) ; }
	65	TArrayF * GetCTSFidCutMinPhiArray() const { return fCTSFidCutMinPhi ; }
	66
	67	void AddEMCALFidCutMaxEtaArray(Int_t size, Float_t* array)
	68	{ fEMCALFidCutMaxEta->Set(size,array) ; }
	69	TArrayF * GetEMCALFidCutMaxEtaArray() const { return fEMCALFidCutMaxEta ; }
	70
	71	void AddEMCALFidCutMaxPhiArray(Int_t size, Float_t* array)
	72	{ fEMCALFidCutMaxPhi->Set(size,array) ; }
	73	TArrayF * GetEMCALFidCutMaxPhiArray() const { return fEMCALFidCutMaxPhi ; }
	74
	75	void AddEMCALFidCutMinEtaArray(Int_t size, Float_t* array)
	76	{ fEMCALFidCutMinEta->Set(size,array) ; }
	77	TArrayF * GetEMCALFidCutMinEtaArray() const { return fEMCALFidCutMinEta ; }
	78
	79	void AddEMCALFidCutMinPhiArray(Int_t size, Float_t* array)
	80	{ fEMCALFidCutMinPhi->Set(size,array) ; }
	81	TArrayF * GetEMCALFidCutMinPhiArray() const { return fEMCALFidCutMinPhi ; }
	82
	83	void AddPHOSFidCutMaxEtaArray(Int_t size, Float_t* array)
	84	{ fPHOSFidCutMaxEta->Set(size,array) ; }
	85	TArrayF * GetPHOSFidCutMaxEtaArray() const { return fPHOSFidCutMaxEta ; }
	86
	87	void AddPHOSFidCutMaxPhiArray(Int_t size, Float_t* array)
	88	{ fPHOSFidCutMaxPhi->Set(size,array) ; }
	89	TArrayF * GetPHOSFidCutMaxPhiArray() const { return fPHOSFidCutMaxPhi ; }
	90
	91	void AddPHOSFidCutMinEtaArray(Int_t size, Float_t* array)
	92	{ fPHOSFidCutMinEta->Set(size,array) ; }
	93	TArrayF * GetPHOSFidCutMinEtaArray() const { return fPHOSFidCutMinEta ; }
1c5acb87	94
c5693f62	95	void AddPHOSFidCutMinPhiArray(Int_t size, Float_t* array)
	96	{ fPHOSFidCutMinPhi->Set(size,array) ; }
	97	TArrayF * GetPHOSFidCutMinPhiArray() const { return fPHOSFidCutMinPhi ; }
	98
b960c7eb	99	private:
c5693f62	100
	101	//Detector acceptance cuts
	102	Bool_t fEMCALFiducialCut ; // Apply fiducial cuts to EMCAL clusters
b960c7eb	103	Bool_t fPHOSFiducialCut ; // Apply fiducial cuts to PHOS clusters
b960c7eb	104	Bool_t fCTSFiducialCut ; // Apply fiducial cuts to CTS tracks
c5693f62	105
	106	TArrayF * fCTSFidCutMinEta ; // Take particles in CTS with eta > fCTSFidCutMinEta
	107	TArrayF * fCTSFidCutMinPhi ; // Take particles in CTS with phi > fCTSFidCutMinPhi
	108	TArrayF * fCTSFidCutMaxEta ; // Take particles in CTS with eta < fCTSFidCutMaxEta
	109	TArrayF * fCTSFidCutMaxPhi ; // Take particles in CTS with phi > fCTSFidCutMaxPhi
	110
	111	TArrayF * fEMCALFidCutMinEta ; // Take particles in EMCAL with eta > fEMCALFidCutMinEta
	112	TArrayF * fEMCALFidCutMinPhi ; // Take particles in EMCAL with phi > fEMCALFidCutMinPhi
	113	TArrayF * fEMCALFidCutMaxEta ; // Take particles in EMCAL with eta < fEMCALFidCutMaxEta
	114	TArrayF * fEMCALFidCutMaxPhi ; // Take particles in EMCAL with phi > fEMCALFidCutMaxPhi
	115
	116	TArrayF * fPHOSFidCutMinEta ; // Take particles in PHOS with eta > fPHOSFidCutMinEta
	117	TArrayF * fPHOSFidCutMinPhi ; // Take particles in PHOS with phi > fPHOSFidCutMinPhi
	118	TArrayF * fPHOSFidCutMaxEta ; // Take particles in PHOS with eta < fPHOSFidCutMaxEta
	119	TArrayF * fPHOSFidCutMaxPhi ; // Take particles in PHOS with phi > fPHOSFidCutMaxPhi
	120
b960c7eb	121	AliFiducialCut( const AliFiducialCut & g) ; // cpy ctor
b960c7eb	122	AliFiducialCut & operator = (const AliFiducialCut & g) ; // cpy assignment
1c5acb87	123
ff45398a	124	ClassDef(AliFiducialCut,1)
c5693f62	125
c5693f62	126	} ;
1c5acb87	127
1c5acb87	128
ff45398a	129	#endif //ALIFIDUCIALCUT_H
1c5acb87	130
	131
	132