Acceptance.h

Go to the documentation of this file.

 
#ifndef INCLUDE_EICSMEAR_SMEAR_ACCEPTANCE_H_
#define INCLUDE_EICSMEAR_SMEAR_ACCEPTANCE_H_
 
#include <set>
#include <vector>
 
#include <TFormula.h>
#include <TMath.h>
 
#include "eicsmear/smear/Smear.h"  // Definition of KinType
 
class TString;
 
namespace erhic {
 
class VirtualParticle;
 
}  // namespace erhic
 
namespace Smear {
 
class Acceptance {
 public:
  class CustomCut {
   public:
    virtual ~CustomCut();
    CustomCut();
    CustomCut(const TString&, double min, double max);
    virtual bool Contains(const erhic::VirtualParticle&) const;
   protected:
    TFormula mFormula;
    int dim;
    KinType Kin1;
    KinType Kin2;
    double Min;
    double Max;
  };
 
  class Zone {
   public:
    virtual ~Zone();
 
    Zone(double theta = 0., double = TMath::Pi(),
         double phi = 0., double = TMath::TwoPi(),
         double E = 0., double = TMath::Infinity(),
         double p = 0., double = TMath::Infinity(),
         double pt = 0., double = TMath::Infinity(),
         double pz = -TMath::Infinity(), double = TMath::Infinity());
 
    virtual void Add(const CustomCut&);
 
    virtual Bool_t Contains(const erhic::VirtualParticle&) const;
 
   protected:
    double thetaMin;
    double thetaMax;
    double phiMin;
    double phiMax;
    double EMin;
    double EMax;
    double PMin;
    double PMax;
    double pTMin;
    double pTMax;
    double pZMin;
    double pZMax;
    std::vector<Smear::Acceptance::CustomCut> CustomCuts;
    // We want to be able to write Acceptance objects to a ROOT file,
    // so nested classes need a dictionary as well.
    ClassDef(Smear::Acceptance::Zone, 1)
  };
 
  virtual ~Acceptance();
 
  explicit Acceptance(int genre = kAll);
 
  void AddZone(const Zone&);
 
  UInt_t GetNZones() const;
 
  Int_t GetGenre() const;
 
  void SetGenre(int genre);
 
  void SetCharge(ECharge charge);
 
  ECharge GetCharge() const;
 
  void AddParticle(int particle);
 
  bool Is(const erhic::VirtualParticle& prt) const;
 
 protected:
  int mGenre;
  ECharge mCharge;  // Particle charges accepted (neutral, charged or all)
  std::vector<Zone> mZones;
  std::set<int> mParticles;
 
  ClassDef(Smear::Acceptance, 1)
};
 
inline UInt_t Acceptance::GetNZones() const {
  return mZones.size();
}
 
inline Int_t Acceptance::GetGenre() const {
  return mGenre;
}
 
inline ECharge Acceptance::GetCharge() const {
  return mCharge;
}
 
}  // namespace Smear
 
#endif  // INCLUDE_EICSMEAR_SMEAR_ACCEPTANCE_H_