DalitzBWBoxSet.h

Go to the documentation of this file.

#ifndef DALITZ_BW_BOX_SET_HH
#define DALITZ_BW_BOX_SET_HH
// author: Jonas Rademacker (Jonas.Rademacker@bristol.ac.uk)
// status:  Mon 9 Feb 2009 19:17:58 GMT

#include "Mint/DalitzBWBox.h"

#include "Mint/IDalitzEvent.h"
#include "Mint/IUnweightedEventGenerator.h"

#include "Mint/IReturnRealForEvent.h"
#include "Mint/DalitzEventPtrList.h"

#include "Mint/counted_ptr.h"
#include "Mint/DalitzPhaseSpaceBox.h"

#include <vector>
#include <iostream>

#include "TRandom.h"


class DalitzBWBoxSet : public MINT::PolymorphVector<DalitzBWBox>
, virtual public MINT::IUnweightedEventGenerator<IDalitzEvent>{
 protected:
  static double __phaseSpaceFracDefaultValue;

  DalitzEventPtrList _eventPtrList;
  //DalitzEventList _eventList;

  double _maxWeightEstimate;
  double _maxWeightInSample;

  MINT::IReturnRealForEvent<IDalitzEvent>* _ampSum;

  bool _ready;
  std::vector<double> _volumeProbs;
  TRandom* _rnd;

  double _phaseSpaceFrac;
  mutable double _phaseSpaceIntegral;

  DalitzPhaseSpaceBox _psbox;
  double _pick_ps_prob;
  
  void getReady();
  void findMax();
  int justThrowAwayData(double maxValue
               , MINT::IReturnRealForEvent<IDalitzEvent>* amps
               );

  MINT::counted_ptr<DalitzEvent> popEventFromList();

  void set_psbox_height_and_weight();
  void setup_psbox();
  
  double VolumeSum()const;
  double heightSum()const;
  void makeVolumeProbIntervals();
  int pickRandomVolume();
  double calc_pick_ps_prob() const;
  double pick_ps_prob();

  bool checkIntegration()const;

  bool ready()const {return _ready;}

  double findMaxInList(double& sampleMax);

  // these are good for x-checks only, therefore not public:
  virtual MINT::counted_ptr<DalitzEvent> makeWeightedApproxEventForOwner();
  virtual MINT::counted_ptr<DalitzEvent> makeWeightedApproxEventForOwner(int& nTries);

 public:
  DalitzBWBoxSet(MINT::IReturnRealForEvent<IDalitzEvent>* amps=0
         , TRandom* r=gRandom);
  DalitzBWBoxSet(TRandom* r);
  DalitzBWBoxSet(const DalitzBWBoxSet& other);
  virtual ~DalitzBWBoxSet(){}

  void add(DalitzBWBox& box);
  void add(DalitzBWBoxSet& boxes);

  void setPhaseSpaceFrac(double ps){_phaseSpaceFrac = ps; getReady();}
  double phaseSpaceFrac() const{return _phaseSpaceFrac;}

  void setUnWeightPs(bool doSo=true);

  virtual MINT::counted_ptr<DalitzEvent> makeEventForOwner();
  virtual MINT::counted_ptr<DalitzEvent> makeEventForOwner(int& nTries);
  virtual MINT::counted_ptr<DalitzEvent> makeWeightedEventForOwner();
  virtual MINT::counted_ptr<DalitzEvent> makeWeightedEventForOwner(int& nTries);
  virtual MINT::counted_ptr<DalitzEvent> tryEventForOwner();

  virtual MINT::counted_ptr<DalitzEvent> phaseSpaceEvent();

  virtual MINT::counted_ptr<IDalitzEvent> newEvent();
  virtual MINT::counted_ptr<IDalitzEvent> newUnweightedEvent();

  virtual bool exhausted() const{return false;}

  double phaseSpaceIntegral() const;

  double fullPdf(DalitzEvent& evt);// (|sum A|^2, w/o phase space factor)

  double genValueNoPs(const DalitzEvent& evt) const;
  double genValuePs(const DalitzEvent& evt) const;
  double genValueWithLoop(DalitzEvent& evt) const;
  double genValue(const DalitzEvent& evt) const;

  void print(std::ostream& os)const;

  bool ensureFreshEvents();
  bool setRnd(TRandom* rnd=gRandom);


  // these two are for some sanity checks:
  bool am_I_generating_what_I_think_I_am_generating(int Nevents=1000000);
  bool compareGenerationMethodsForFullPDF(int Nevents=1000000);
};

std::ostream& operator<<(std::ostream& os, const DalitzBWBoxSet& box);
#endif
//