TimeDependentGenerator.h

Go to the documentation of this file.

#ifndef __TIMEDEPENDENTGENERATOR_H__
#define __TIMEDEPENDENTGENERATOR_H__

#include <Mint/FitAmpSum.h>
#include <Mint/DalitzEvent.h>
#include <Mint/SignalGenerator.h>
#include <Mint/FitAmpIncoherentSum.h>
#include <memory>
#include <list>
#include <map>
#include <complex>
#include <string>
#include <TSpline.h>
#include <Mint/Eff3piSymmetric.h>
#include <TH1F.h>

class TRandom3 ;

// Class to generate time dependent phase space events.
class TimeDependentGenerator {

public :
  // Class to hold the tag (production flavour), decay time and
  // Dalitz event that's generated.
  class GenTimeEvent : public DalitzEvent {
  public :
    GenTimeEvent(const IDalitzEvent&, const int, const double,
         const double smeareddecaytime = -999.) ;
    GenTimeEvent(const IDalitzEvent&) ;
    int getTag() const ;
    void setTag(int) ;
    double getDecayTime() const ;
    void setDecayTime(double) ;
    double getSmearedDecayTime() const ;
    void setSmearedDecayTime(double) ;
    enum InfoIndex{ITAG = 0, IDECAYTIME, ISMEAREDDECAYTIME} ;
    static std::map<std::string, unsigned> infoNames ;
  private :
    static std::map<std::string, unsigned> makeInfoNames() ;
  } ;

  static DalitzEventPattern anti(DalitzEventPattern pat) ;

  TimeDependentGenerator(const DalitzEventPattern& pattern, double width, double deltam,
             double deltagamma, double qoverp, double phi,
             TRandom3* rndm, TH1F* h_efficiency = NULL,
                         float resWidth = 0.05, bool addExpEffects = false,
                         Eff3piSymmetric* sEfficiency = NULL) ;
  TimeDependentGenerator(MINT::counted_ptr<FitAmpSum> model, MINT::counted_ptr<FitAmpSum> cpmodel,
             double width, double deltam,
             double deltagamma, double qoverp, double phi,
             TRandom3*, TH1F* h_efficiency = NULL,
                         float resWidth = 0.05, bool addExpEffects = false,
                         Eff3piSymmetric* sEfficiency = NULL) ;

  std::pair<double, double> generate_decay_time() const ;

  MINT::counted_ptr<IDalitzEvent> generate_event() ;

  MINT::counted_ptr<IDalitzEvent> generate_event(const double s13, const double s23) ;

  bool accept_or_reject(const int tag, const double decaytime, IDalitzEvent&) ;

  double get_scale() const ;
  float get_gen_efficiency() const ;

  double pdf_value(int, double, IDalitzEvent&) ;
  double pdf_value(IDalitzEvent&) ;
  double envelope_value(const double, IDalitzEvent&) ;

  typedef std::pair<std::complex<double>, std::complex<double> > AmpPair ;
  AmpPair amplitude_coefficients(const int tag, const double decaytime) ;

  TH1F draw_pdf_vs_time(IDalitzEvent&, unsigned, float, float, const std::string& name = "pdf_vs_time") ;
  TH1F draw_envelope_vs_time(IDalitzEvent&, unsigned, float, float, const std::string& name = "envelope_vs_time") ;

private :
  void init() ;

  TRandom3* m_rndm ;
  const DalitzEventPattern m_pattern ;
  const DalitzEventPattern m_cppattern ;
  MINT::counted_ptr<FitAmpSum> m_model ;
  MINT::counted_ptr<FitAmpSum> m_cpmodel ;
  FitAmpIncoherentSum m_bothmodel ;
  SignalGenerator m_generator ;

  const double m_width ;
  const double m_deltam ;
  const double m_deltagamma ;
  const std::complex<double> m_qoverp ;
  double m_scale ;
  unsigned m_ngen ;
  unsigned m_naccept ;

  TH1F* m_h_efficiency;
  TSpline3 m_efficiencyFit;

  float m_resWidth;
  bool m_addExpEffects;

  Eff3piSymmetric* m_sEfficiency;
} ;

#endif