TwoBodyDecayGen.hxx

Go to the documentation of this file.

#ifndef TWOBODYDECAYGEN_HXX
#define TWOBODYDECAYGEN_HXX

#include <string>
#include <vector>
#include <deque>

#include <TH1.h>
#include <TTree.h>
#include <TLorentzVector.h>
#include <TGenPhaseSpace.h>

#define NDAUS 2                 
class TwoBodyDecayGen {
public:

  typedef std::pair<std::vector<TwoBodyDecayGen*>, double> DauNode; 
  typedef std::vector<DauNode> DauNodeVec; 
  typedef std::pair<unsigned, double> chBFpair; 
  TwoBodyDecayGen(double mommass, double dau1mass, double dau2mass,
                  TwoBodyDecayGen *dau1=NULL,
                  TwoBodyDecayGen *dau2=NULL);

  TwoBodyDecayGen(double mommass, double *daumasses,
                  TwoBodyDecayGen *dau1=NULL,
                  TwoBodyDecayGen *dau2=NULL);

  TwoBodyDecayGen(double *masses, unsigned nparts);

  ~TwoBodyDecayGen() {}

  bool add_decay_channel(double *masses, unsigned nparts,
                         double brfr);

  TwoBodyDecayGen* get_daughter(unsigned chid, unsigned dauid);

  double get_brfr(unsigned chid);

  int find_leaf_nodes(std::vector<std::deque<chBFpair> > &brfrVec,
                       std::deque<chBFpair> &brfrQ);

  double generate(TLorentzVector &momp,
                  std::vector<TLorentzVector> &particle_lvs,
                  std::deque<chBFpair> chQ);

  TTree* get_event_tree(unsigned nevents, TH1 *hmomp);

  void print(unsigned indent=0);

private:

  void _printQ(std::string prefix, std::deque<chBFpair> queue);

  void _printQ(std::string prefix, std::vector<std::deque<chBFpair> > queue);

  static unsigned long long _count; 
  TGenPhaseSpace _generator;    
  double _mommass;              
  double _daumasses[NDAUS];     
  DauNodeVec _dauchannels;      
};

#endif  // TWOBODYDECAYGEN_HXX