HadronicParameters.h

Go to the documentation of this file.

#ifndef __HADRONICPARAMETERS_H__
#define __HADRONICPARAMETERS_H__

#include <Mint/DalitzEvent.h>
#include <Mint/counted_ptr.h>
#include <Mint/FitAmpSum.h>
#include <complex>
#include <deque>
#include <utility>
#include <iostream>

class TRandom3 ;

class HadronicParameters {
 public :
  typedef MINT::counted_ptr<FitAmpSum> ModelPtr ;

  class EventBinInfo {
  public :
    // Should I use a counted_ptr?
    IDalitzEvent* evt ;
    std::complex<double> amp ;
    std::complex<double> ampBar ;
    double F ;
    double Fbar ;
    int binNumber ;

    int binSign() const ;
    bool isPlus() const ;
  } ;
  class PhaseBinningBase {
  public :
    PhaseBinningBase(unsigned nBins) ;
    virtual EventBinInfo binInfo(IDalitzEvent&) const = 0 ;
    virtual std::string type() const = 0 ;
    virtual void Print(const std::string&, std::ostream& os = std::cout) const = 0 ;
    virtual ~PhaseBinningBase() {} ;
    const unsigned nBins ;
    int binNumber(IDalitzEvent&) const ;
    virtual bool operator==(const PhaseBinningBase&) const ;
    bool operator!=(const PhaseBinningBase&) const ;
  } ;
  class ModelPhaseBinning : public PhaseBinningBase {
  public :
    ModelPhaseBinning(ModelPtr, ModelPtr, unsigned) ;
    virtual ~ModelPhaseBinning() {} ;
    virtual EventBinInfo binInfo(IDalitzEvent&) const override ;
    virtual std::string type() const override ;
    virtual void Print(const std::string&, std::ostream& os = std::cout) const override ;
    static MINT::counted_ptr<PhaseBinningBase> 
      fromConfig(const std::string&, const std::string& fname = "") ;
    // TODO: implement this - not really sure how best to check that the models are the same.
    //virtual bool operator==(const PhaseBinningBase&) const override ;
  protected :
    virtual bool isFavoured(const double, const double, IDalitzEvent&) const ;
    ModelPtr m_model ;
    ModelPtr m_cpmodel ;
  } ;
  class ModelBinning3Body : public ModelPhaseBinning {
  public :
    ModelBinning3Body(ModelPtr, ModelPtr, unsigned) ;
    virtual ~ModelBinning3Body() {} ;
    virtual std::string type() const override ;
    static MINT::counted_ptr<PhaseBinningBase> 
      fromConfig(const std::string&, const std::string& fname = "") ;
  protected :
    virtual bool isFavoured(const double, const double, IDalitzEvent&) const override ;
  } ;
  class Bin {
  private :
    double m_Fplus ;
    double m_Fminus ;
    std::complex<double> m_X ;
    double m_Fbarplus ;
    double m_Fbarminus ;
    std::complex<double> m_Xbar ;
    double m_sumw ;
    double m_sumw2 ;
    double m_norm ;
    double m_normBar ;

    double _R(double, double, double, const std::complex<double>&, const std::complex<double>&,
          double, double, const std::complex<double>&, const std::complex<double>&) const ;
  public :
    Bin(double, double, const std::complex<double>&, double, double, const std::complex<double>&,
    double norm = 1., double normbar = 1., double sumw = 1., double sumw2 = 1.) ;
    Bin() ;
    Bin(const std::string&, unsigned, const std::string& fname = "") ;

    void add(const EventBinInfo&, const EventBinInfo&, double weight = 1.) ;
    double Fplus() const ;
    double Fminus() const ;
    std::complex<double> Xplus() const ;
    std::complex<double> Xminus() const ;
    double Fbarplus() const ;
    double Fbarminus() const ;
    std::complex<double> Xbarplus() const ;
    std::complex<double> Xbarminus() const ;
    double getNorm() const ;
    void setNorm(double) ;
    double getNormBar() const ;
    void setNormBar(double) ;
    void Print(const std::string& name, unsigned, std::ostream& os=std::cout) const ;
    static std::string getName(const std::string&, unsigned) ;

    double R(double, double, double, const std::complex<double>&, const std::complex<double>&) const ;
    double Rbar(double, double, double, const std::complex<double>&, const std::complex<double>&) const ;
    
  } ;

  typedef std::deque<Bin> Bins ;
  typedef MINT::counted_ptr<PhaseBinningBase> BinningPtr ;

  HadronicParameters(const Bins&, BinningPtr) ;
  HadronicParameters(BinningPtr) ;
  HadronicParameters(const std::string&, const std::string& fname = "") ;

  const PhaseBinningBase& binning() const ;
  HadronicParameters::BinningPtr binningPtr() const ;

  const Bin& bin(IDalitzEvent&) const ;
  const Bin& bin(unsigned) const ;

  void add(IDalitzEvent&, double weight = 1.) ;
  void add(const DalitzEventPattern&, TRandom3&, unsigned) ;
  void add(const DalitzEventPattern&, TRandom3&) ;

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

  std::pair<double, double> normalise(double norm = 1., double normBar = 1.) ;
  void Print(const std::string&, std::ostream& os = std::cout) const ;
  void write(const std::string&, const std::string&) const ;

  static BinningPtr getPhaseBinning(const std::string&, const std::string& fname = "") ;

 private :
  Bins m_bins ;
  BinningPtr m_phaseBinning ;
} ;

#endif