Teegg.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2015  Belle II Collaboration                              *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Torben Ferber                                            *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
#include <mdst/dataobjects/MCParticleGraph.h>
#include <TLorentzRotation.h>
 
namespace Belle2 {
  class Teegg {
  public:
 
    Teegg();
 
    ~Teegg();
 
    void setDefaultSettings();
 
    void setVACPOL(const std::string& vacpol = "HLMNT") {m_sVACPOL = vacpol;}
 
    void setTEVETO(double teveto = 666.66) { m_TEVETO = teveto; }
 
    void setTEMIN(double temin = 666.66) { m_TEMIN = temin; }
 
    void setTGMIN(double tgmin = 666.66) { m_TGMIN = tgmin; }
 
    void setTGVETO(double tgveto = 666.66) { m_TGVETO = tgveto; }
 
    void setEEMIN(double eemin = 666.66) { m_EEMIN = eemin; }
 
    void setEGMIN(double egmin = 666.66) { m_EGMIN = egmin; }
 
    void setPEGMIN(double pegmin = 666.66) { m_PEGMIN = pegmin; }
 
    void setEEVETO(double eeveto = 666.66) { m_EEVETO = eeveto; }
 
    void setEGVETO(double egveto = 666.66) { m_EGVETO = egveto; }
 
    void setPHVETO(double phveto = 666.66) { m_PHVETO = phveto; }
 
    void setCUTOFF(double cutoff = 666.66) { m_CUTOFF = cutoff; }
 
    void setEPS(double eps = 666.66) { m_EPS = eps; }
 
    void setFRAPHI(double fraphi = 666.66) { m_FRAPHI = fraphi; }
 
    void setEPSPHI(double epsphi = 666.66) { m_EPSPHI = epsphi; }
 
    void setWGHT1M(double wght1m = 666.66) { m_WGHT1M = wght1m; }
 
    void setWGHTMX(double wghtmx = 666.66) { m_WGHTMX = wghtmx; }
 
    void setRADCOR(const std::string& radcor = "NONE") {m_sRADCOR = radcor;}
 
    void setCONFIG(const std::string& config = "NONE") {m_sCONFIG = config;}
 
    void setMATRIX(const std::string& matrix = "NONE") {m_sMATRIX = matrix;}
 
    void setMTRXGG(const std::string& mtrxgg = "NONE") {m_sMTRXGG = mtrxgg;}
 
    void setUNWGHT(int unwght = 1) { m_UNWGHT = unwght; }
 
    void setCMSEnergy(double cmsEnergy) { m_cmsEnergy = cmsEnergy; }
 
    void init(); 
    void initExtraInfo();
 
    void generateEvent(MCParticleGraph& mcGraph, TVector3 vertex, TLorentzRotation boost);
 
    double getT() {return m_t;};
 
    double getW2() {return m_w2;};
 
    double getWeight() {return m_weight;};
 
    void term();
 
  protected:
    double m_pi;                    
    double m_conversionFactor;      
    double m_alphaQED0;             
    double m_massElectron;          
    double m_cmsEnergy;             
    double m_TEVETO; 
    double m_TEMIN;  
    double m_TGMIN;  
    double m_TGVETO; 
    double m_EEMIN;  
    double m_EGMIN;  
    double m_PEGMIN; 
    double m_EEVETO; 
    double m_EGVETO; 
    double m_PHVETO; 
    double m_CUTOFF; 
    double m_EPS; 
    double m_FRAPHI; 
    double m_EPSPHI; 
    double m_WGHT1M; 
    double m_WGHTMX; 
    int m_VACPOL; 
    int m_RADCOR; 
    int m_CONFIG; 
    int m_MATRIX; 
    int m_MTRXGG; 
    std::string m_sVACPOL;  
    std::string m_sRADCOR; 
    std::string m_sCONFIG; 
    std::string m_sMATRIX; 
    std::string m_sMTRXGG; 
    int m_UNWGHT; 
    double m_t;  
    double m_w2;  
    double m_weight;  
    double m_vp2;  
    void applySettings();
 
    void storeParticle(MCParticleGraph& mcGraph, const double* mom, int pdg, TVector3 vertex, TLorentzRotation boost,
                       bool isVirtual = false, bool isInitial = false);
 
  private:
 
    int    m_npar[100];  
    double m_xpar[100];  
  };
}