TeeggInputModule.h

 
/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#ifndef TEEGGINPUTMODULE_H
#define TEEGGINPUTMODULE_H
 
#include <framework/core/Module.h>
#include <framework/logging/Logger.h>
 
#include <generators/teegg/Teegg.h>
#include <generators/utilities/InitialParticleGeneration.h>
 
#include <mdst/dataobjects/MCParticleGraph.h>
 
#include <vector>
 
namespace Belle2 {
  class TeeggInputModule : public Module {
 
  public:
 
    TeeggInputModule();
 
    virtual ~TeeggInputModule();
 
    virtual void initialize() override;
 
    virtual void event() override;
 
    virtual void terminate() override;
 
  protected:
 
    std::string m_sVACPOL; 
    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; 
    std::string m_sRADCOR; 
    std::string m_sCONFIG; 
    std::string m_sMATRIX; 
    std::string m_sMTRXGG; 
    int m_UNWGHT; 
    Teegg m_generator;   
    MCParticleGraph m_mcGraph; 
    template <typename T>
    std::vector<T> make_vector(T const& t1, T const& t2);  
    template <typename T>
    std::pair<T, T> vectorToPair(std::vector<T>& vec, const std::string& name = "");
 
  private:
 
    void initializeGenerator();
 
    bool m_initialized{false}; 
    DBObjPtr<BeamParameters> m_beamParams; 
    InitialParticleGeneration m_initial; 
  }; 
  template <typename T>
  inline std::vector<T> TeeggInputModule::make_vector(T const& t1, T const& t2)
  {
    std::vector<T> v;
    v.push_back(t1);
    v.push_back(t2);
    return v; 
  }
 
  template <typename T>
  inline std::pair<T, T> TeeggInputModule::vectorToPair(std::vector<T>& vec, const std::string& name)
  {
    std::pair<T, T> p;
    if (vec.size() != 2) {
      B2ERROR("The parameter " + name + " has to have exactly 2 elements!");
    } else {
      p.first  = vec[0];
      p.second = vec[1];
    }
    return p;  
  }
 
} // end namespace Belle2
 
#endif /* TEEGGINPUTMODULE_H */