[u/mrichter/AliRoot.git] / PWG3 / dielectron / AliDielectronTrackCuts.h

#ifndef ALIDIELECTRONTRACKCUTS_H
#define ALIDIELECTRONTRACKCUTS_H

/* Copyright(c) 1998-2009, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

//#############################################################
//#                                                           # 
//#         Class AliDielectronTrackCuts                     #
//#                                                           #
//#  Authors:                                                 #
//#   Anton     Andronic, GSI / A.Andronic@gsi.de             #
//#   Ionut C.  Arsene,   GSI / I.C.Arsene@gsi.de             #
//#   Julian    Book,     Uni Ffm / Julian.Book@cern.ch       #
//#   Frederick Kramer,   Uni Ffm, / Frederick.Kramer@cern.ch #
//#   Magnus    Mager,    CERN / Magnus.Mager@cern.ch         #
//#   WooJin J. Park,     GSI / W.J.Park@gsi.de               #
//#   Jens      Wiechula, Uni Tü / Jens.Wiechula@cern.ch      #
//#                                                           #
//#############################################################

#include <AliPID.h>
#include <AliAnalysisCuts.h>

class AliDielectronTrackCuts : public AliAnalysisCuts {
public:
  enum ITSClusterRequirement { kOff = 0, kNone, kAny, kFirst, kOnlyFirst, kSecond, kOnlySecond, kBoth };
  enum Detector { kSPD = 0, kSDD, kSSD };

  enum {
    kITSin=0x0001,kITSout=0x0002,kITSrefit=0x0004,kITSpid=0x0008,
      kTPCin=0x0010,kTPCout=0x0020,kTPCrefit=0x0040,kTPCpid=0x0080,
      kTRDin=0x0100,kTRDout=0x0200,kTRDrefit=0x0400,kTRDpid=0x0800,
      kTOFin=0x1000,kTOFout=0x2000,kTOFrefit=0x4000,kTOFpid=0x8000,
      kHMPIDout=0x10000,kHMPIDpid=0x20000
  };
  
  AliDielectronTrackCuts();
  AliDielectronTrackCuts(const char*name, const char* title);

  virtual ~AliDielectronTrackCuts();

  void SetV0DaughterCut(AliPID::EParticleType type);
  void SetClusterRequirementITS(Detector det, ITSClusterRequirement req = kOff) { fCutClusterRequirementITS[det] = req; }
  
  void SetRequireITSRefit(Bool_t req) { fRequireITSRefit=req; }
  void SetRequireTPCRefit(Bool_t req) { fRequireTPCRefit=req; }
  
  Int_t GetV0DaughterCut() const { return fV0DaughterCut; }
  ITSClusterRequirement GetClusterRequirementITS(Detector det) const { return fCutClusterRequirementITS[det]; }
  
  //
  //Analysis cuts interface
  //
  virtual Bool_t IsSelected(TObject* track);
  virtual Bool_t IsSelected(TList*   /* list */ ) {return kFALSE;}
  

private:

  AliDielectronTrackCuts(const AliDielectronTrackCuts &c);
  AliDielectronTrackCuts &operator=(const AliDielectronTrackCuts &c);

  Int_t fV0DaughterCut;                                // Bit for V0 daughter selection
  ITSClusterRequirement fCutClusterRequirementITS[3];  // detailed ITS cluster requirements for (SPD, SDD, SSD)

  Bool_t fRequireITSRefit;                             // require ITS refit
  Bool_t fRequireTPCRefit;                             // require TPC refit
  
  Bool_t CheckITSClusterRequirement(ITSClusterRequirement req, Bool_t clusterL1, Bool_t clusterL2) const;
  
  ClassDef(AliDielectronTrackCuts,1)         // Dielectron TrackCuts
};


#endif
Commit	Line	Data
164bfb53	1	#ifndef ALIDIELECTRONTRACKCUTS_H
	2	#define ALIDIELECTRONTRACKCUTS_H
	3
	4	/* Copyright(c) 1998-2009, ALICE Experiment at CERN, All rights reserved. *
	5	* See cxx source for full Copyright notice */
	6
	7	//#############################################################
	8	//# #
	9	//# Class AliDielectronTrackCuts #
	10	//# #
	11	//# Authors: #
	12	//# Anton Andronic, GSI / A.Andronic@gsi.de #
	13	//# Ionut C. Arsene, GSI / I.C.Arsene@gsi.de #
	14	//# Julian Book, Uni Ffm / Julian.Book@cern.ch #
	15	//# Frederick Kramer, Uni Ffm, / Frederick.Kramer@cern.ch #
	16	//# Magnus Mager, CERN / Magnus.Mager@cern.ch #
	17	//# WooJin J. Park, GSI / W.J.Park@gsi.de #
	18	//# Jens Wiechula, Uni Tü / Jens.Wiechula@cern.ch #
	19	//# #
	20	//#############################################################
	21
	22	#include <AliPID.h>
	23	#include <AliAnalysisCuts.h>
	24
	25	class AliDielectronTrackCuts : public AliAnalysisCuts {
	26	public:
	27	enum ITSClusterRequirement { kOff = 0, kNone, kAny, kFirst, kOnlyFirst, kSecond, kOnlySecond, kBoth };
	28	enum Detector { kSPD = 0, kSDD, kSSD };
	29
	30	enum {
	31	kITSin=0x0001,kITSout=0x0002,kITSrefit=0x0004,kITSpid=0x0008,
	32	kTPCin=0x0010,kTPCout=0x0020,kTPCrefit=0x0040,kTPCpid=0x0080,
	33	kTRDin=0x0100,kTRDout=0x0200,kTRDrefit=0x0400,kTRDpid=0x0800,
	34	kTOFin=0x1000,kTOFout=0x2000,kTOFrefit=0x4000,kTOFpid=0x8000,
	35	kHMPIDout=0x10000,kHMPIDpid=0x20000
	36	};
	37
	38	AliDielectronTrackCuts();
	39	AliDielectronTrackCuts(const charname, const char title);
	40
	41	virtual ~AliDielectronTrackCuts();
	42
	43	void SetV0DaughterCut(AliPID::EParticleType type);
	44	void SetClusterRequirementITS(Detector det, ITSClusterRequirement req = kOff) { fCutClusterRequirementITS[det] = req; }
	45
	46	void SetRequireITSRefit(Bool_t req) { fRequireITSRefit=req; }
	47	void SetRequireTPCRefit(Bool_t req) { fRequireTPCRefit=req; }
	48
	49	Int_t GetV0DaughterCut() const { return fV0DaughterCut; }
	50	ITSClusterRequirement GetClusterRequirementITS(Detector det) const { return fCutClusterRequirementITS[det]; }
	51
	52	//
	53	//Analysis cuts interface
	54	//
	55	virtual Bool_t IsSelected(TObject* track);
	56	virtual Bool_t IsSelected(TList* /* list */ ) {return kFALSE;}
	57
	58
	59	private:
	60
	61	AliDielectronTrackCuts(const AliDielectronTrackCuts &c);
	62	AliDielectronTrackCuts &operator=(const AliDielectronTrackCuts &c);
	63
	64	Int_t fV0DaughterCut; // Bit for V0 daughter selection
65	ITSClusterRequirement fCutClusterRequirementITS[3]; // detailed ITS cluster requirements for (SPD, SDD, SSD)
66
67	Bool_t fRequireITSRefit; // require ITS refit
68	Bool_t fRequireTPCRefit; // require TPC refit
69
70	Bool_t CheckITSClusterRequirement(ITSClusterRequirement req, Bool_t clusterL1, Bool_t clusterL2) const;
71
72	ClassDef(AliDielectronTrackCuts,1) // Dielectron TrackCuts
73	};
74
75
76
77	#endif