Moved from TDPMjet.

[u/mrichter/AliRoot.git] / PWG1 / AliMCInfoCuts.h

#ifndef ALIMCINFOCUTS_H\r
#define ALIMCINFOCUTS_H\r
\r
//------------------------------------------------------------------------------\r
// Class to keep selection cuts for MC tracks. \r
// \r
// Author: J.Otwinowski 04/02/2008 \r
//------------------------------------------------------------------------------\r
\r
#include "AliAnalysisCuts.h"\r
\r
class TArrayI;\r
\r
class AliMCInfoCuts : public AliAnalysisCuts\r
{\r
public:\r
  AliMCInfoCuts(const Char_t* name ="AliMCInfoCuts", const Char_t *title ="");\r
  virtual ~AliMCInfoCuts(); \r
 \r
  // setters \r
  void SetMinRowsWithDigits(const Int_t min=0) {fMinRowsWithDigits = min;}\r
  void SetMaxR(const Float_t max=1e99)         {fMaxR = max;}\r
  void SetMaxVz(const Float_t max=1e99)        {fMaxVz = max;}\r
  void SetRangeTPCSignal(const Float_t min=0, const Float_t max=1e99)  {fMinTPCSignal = min; fMaxTPCSignal = max;}\r
\r
  // getters \r
  Int_t GetMinRowsWithDigits() const {return fMinRowsWithDigits;}\r
  Float_t GetMaxR()            const {return fMaxR;}\r
  Float_t GetMaxVz()           const {return fMaxVz;}\r
  Float_t GetMinTPCSignal()    const {return fMinTPCSignal;}\r
  Float_t GetMaxTPCSignal()    const {return fMaxTPCSignal;}\r
\r
  Float_t GetEP()  const       {return ep;}\r
  Float_t GetEM()  const       {return em;}\r
  Float_t GetMuP() const       {return mup;}\r
  Float_t GetMuM() const       {return mum;}\r
  Float_t GetPiP() const       {return pip;}\r
  Float_t GetPiM() const       {return pim;}\r
  Float_t GetKP()  const       {return kp;}\r
  Float_t GetKM()  const       {return km;}\r
  Float_t GetProt() const      {return prot;}\r
  Float_t GetProtBar() const   {return protbar;}\r
 \r
  // cuts init function\r
  void Init();\r
\r
  // check MC tracks\r
  Bool_t IsSelected(TObject *) {return kTRUE;}\r
\r
  // add particle to array \r
  void AddPdgParticle(Int_t idx=-1, Int_t pdgcode=0) const;\r
\r
  // check particle in array \r
  Bool_t IsPdgParticle(Int_t pdgcode=0) const;\r
\r
  // check particle in array \r
  Bool_t IsPosPdgParticle(Int_t pdgcode=0) const;\r
\r
  // Merge output objects (needed by PROOF) \r
  virtual Long64_t Merge(TCollection* list);\r
\r
private:\r
  Int_t fMinRowsWithDigits; // min. number of TPC raws with digits  \r
  Float_t fMaxR;            // max. R distance from MC vertex \r
  Float_t fMaxVz;           // max. Z distance from MC vertex\r
  Float_t fMinTPCSignal;    // min. TPC Signal calculated from Bethe Bloch formula\r
  Float_t fMaxTPCSignal;    // max. TPC Signal calculated from Bethe Bloch formula\r
 \r
  TArrayI* aTrackParticles; // array of tracked particles \r
\r
  // PDG tracked particles (later added to aTrackParticles)\r
  enum enumData {\r
    kNParticles = 10, // number of particles below\r
    ep = -11,\r
    em = 11,\r
    mup = -13,\r
    mum = 13,\r
    pip = 211,\r
    pim = -211,\r
    kp = 321,\r
    km = -321,\r
    prot = 2212,\r
    protbar = -2212\r
  };\r
\r
  AliMCInfoCuts(const AliMCInfoCuts&); // not implemented\r
  AliMCInfoCuts& operator=(const AliMCInfoCuts&); // not implemented\r
\r
  ClassDef(AliMCInfoCuts, 1)\r
};\r
\r
#endif // ALIMCINFOCUTS_H\r