Simulation.h

// Copyright (c) 2013 Patrick Huck
#ifndef STROOT_BESCOCKTAIL_SIMULATION_H_
#define STROOT_BESCOCKTAIL_SIMULATION_H_

#include <string>
#include <vector>
#include <TRandom3.h>
#include <TLorentzVector.h>
#include <TF1.h>
#include <TH1D.h>
#include "StRoot/BesCocktail/Database.h"
#include "StRoot/BesCocktail/Functions.h"

class Simulation {
  private:
    std::string particle;
    double energy;

    std::vector<TRandom3*> rndm;
    double mass;  // hadron mass
    double mass_dec;  // daughter hadron mass
    double mPt;  // hadron pt
    double mEta;  // hadron eta
    double mPhi;  // hadron phi
    double mBR;  // fraction of dalitz in total branching ratio (ee+dalitz)
    double sigma0; // crystal ball gaussian width
    int mode;  // decay mode
    DatabaseManager* dbm;
    TLorentzVector* ep;
    TLorentzVector* em;
    TLorentzVector* dh;
    TLorentzVector* dlp;
    Functions* fp;
    TF1* fPt;  // input pt distribution
    TF1* fM;  // input mass distribution
    TF1* fRes;  // pt momentum resolution
    TF1* fCB;  // CrystalBall for momentum smearing
    TF1* fKW;  // KrollWada for dalitz decays
    //TF1* fRapJpsi;  // jpsi gaussian rapidity distribution
    std::vector<double> vfill;
    Float_t* afill;
    bool isDhKin;  // flag true if decay hadron kinematics
    //TH1D* hPtInv;  // test input pT histo
    TH1D* hPt;  // pT histo (hPtInv*pT@200)
    TH1D* hM;
    TH1D* hCB;
    TH1D* hKW;
    //TH1D* hRapJpsi;

    double getEta(const double&);
    void applyMomSmear(TLorentzVector&);
    void doTwoBodyDecay();
    void doDalitzDecay();
    void twoBodyKinematics(const double&, double&, double&, double&);
    void eeDecayVM(const double&);
    void hDecayVM(const double&);
    void pushNtVars();

  public:
    Simulation(const std::string&, const double&);
    virtual ~Simulation() {}

    void sampleInput();
    void decay();
    void smear();
    Float_t* getFillArray();
};
#endif  // STROOT_BESCOCKTAIL_SIMULATION_H_