Acceptance.h

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file Acceptance.h

#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

typedef bool (*cutfunc)(const erhic::VirtualParticle&);

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;
  };

  // Allows any function that takes a VirtualParticle
  // and a parameter vectior (mins and maxes, e.g.) and returns a boolean
  class CustomCutFunction {
    // typedef bool (*cutfunc)(const erhic::VirtualParticle& p, std::vector<double> params);
  public:
    virtual ~CustomCutFunction();
    CustomCutFunction( cutfunc func);
    // bool Contains(const erhic::VirtualParticle& p) const;
  protected:
    // std::vector<double> params;
    cutfunc func;
  };

  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_