updated

[u/mrichter/AliRoot.git] / PWGPP / AliFilteredTreeAcceptanceCuts.h

#ifndef ALIFILTEREDTREEACCEPTANCECUTS_H\r
#define ALIFILTEREDTREEACCEPTANCECUTS_H\r
\r
//------------------------------------------------------------------------------\r
// Class to keep selection cuts for MC tracks. \r
// \r
// Author: J.Otwinowski 03/11/2008 \r
// last change: 2011-04-04 by M.Knichel\r
//------------------------------------------------------------------------------\r
\r
class TParticle;\r
class AliESDtrack;\r
class AliExternalTrackParam;\r
\r
#include "AliAnalysisCuts.h"\r
\r
class AliFilteredTreeAcceptanceCuts : public AliAnalysisCuts\r
{\r
public:\r
  AliFilteredTreeAcceptanceCuts(const Char_t* name ="AliFilteredTreeAcceptanceCuts", const Char_t *title ="");\r
  virtual ~AliFilteredTreeAcceptanceCuts(); \r
 \r
  // setters \r
  void SetEtaRange(Float_t min=-1e99, Float_t max=1e99)  { fMinEta=min; fMaxEta=max; }\r
  void SetPhiRange(Float_t min=-1e99, Float_t max=1e99)  { fMinPhi=min; fMaxPhi=max;}\r
  void SetPtRange(Float_t min=-1e99, Float_t max=1e99)   { fMinPt=min;  fMaxPt=max;}\r
  void SetExcludeEtaPhiRange(Float_t etaMin, Float_t etaMax, Float_t phiMin, Float_t phiMax)\r
  	{ fExcludeMinEta = etaMin; fExcludeMaxEta = etaMax; fExcludeMinPhi = phiMin; fExcludeMaxPhi = phiMax; fCheckRange=kTRUE; }\r
  void SetExcludeEtaPhiRange2(Float_t etaMin, Float_t etaMax, Float_t phiMin, Float_t phiMax)\r
  	{ fExcludeMinEta2 = etaMin; fExcludeMaxEta2 = etaMax; fExcludeMinPhi2 = phiMin; fExcludeMaxPhi2 = phiMax; fCheckRange=kTRUE; }  	\r
\r
  void SetMaxDCAr(Float_t max=1e99) { fMaxDCAr=max;}\r
  void SetMaxDCAz(Float_t max=1e99) { fMaxDCAz=max;}\r
\r
  // getters \r
  Float_t GetMinEta() const {return fMinEta;}\r
  Float_t GetMaxEta() const {return fMaxEta;}\r
  Float_t GetMinPhi() const {return fMinPhi;}\r
  Float_t GetMaxPhi() const {return fMaxPhi;}\r
  Float_t GetMinPt() const {return fMinPt;}\r
  Float_t GetMaxPt() const {return fMaxPt;}\r
  \r
  Bool_t  GetCheckRange() const { return fCheckRange; }\r
  Float_t GetExcludeMinEta() const { return fExcludeMinEta; }\r
  Float_t GetExcludeMaxEta() const { return fExcludeMaxEta; }\r
  Float_t GetExcludeMinPhi() const { return fExcludeMinPhi; }\r
  Float_t GetExcludeMaxPhi() const { return fExcludeMaxPhi; }  \r
\r
  Float_t GetMaxDCAr() const {return fMaxDCAr;}\r
  Float_t GetMaxDCAz() const {return fMaxDCAz;}\r
\r
  // cuts init function\r
  void Init();\r
\r
  // check MC tracks\r
  virtual Bool_t IsSelected(TObject *) {return kTRUE;}\r
  virtual Bool_t IsSelected(TList *) {return kTRUE;}\r
\r
  //\r
  Bool_t AcceptTrack(AliESDtrack *track);\r
  Bool_t AcceptTrackLocalTPC(AliESDtrack *track);\r
  Bool_t AcceptTrack(AliExternalTrackParam *track);\r
  Bool_t AcceptTrack(TParticle *particle);\r
  \r
  // Merge output objects (needed by PROOF) \r
  virtual Long64_t Merge(TCollection* list);\r
\r
private:\r
  Float_t fMinEta; // min pseudorapidity \r
  Float_t fMaxEta; // max pseudorapidity\r
  Float_t fMinPhi; // min azimuthal angle (rad)\r
  Float_t fMaxPhi; // max azimuthal angle (rad)\r
  Float_t fMinPt;  // min pt\r
  Float_t fMaxPt;  // max pt\r
  \r
  Float_t fExcludeMinEta;\r
  Float_t fExcludeMaxEta;\r
  Float_t fExcludeMinPhi;\r
  Float_t fExcludeMaxPhi;\r
  Float_t fExcludeMinEta2;\r
  Float_t fExcludeMaxEta2;\r
  Float_t fExcludeMinPhi2;\r
  Float_t fExcludeMaxPhi2;  \r
  Bool_t  fCheckRange;\r
\r
  // max DCAr and DCAz with respect\r
  // to nominal vertex position\r
  Float_t fMaxDCAr; // min DCAr\r
  Float_t fMaxDCAz; // max DCAz\r
 \r
  AliFilteredTreeAcceptanceCuts(const AliFilteredTreeAcceptanceCuts&); // not implemented\r
  AliFilteredTreeAcceptanceCuts& operator=(const AliFilteredTreeAcceptanceCuts&); // not implemented\r
\r
  ClassDef(AliFilteredTreeAcceptanceCuts, 1)\r
};\r
\r
#endif // \r