BHWideInputModule.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.                  *
 **************************************************************************/
 
#pragma once
 
#include <framework/core/Module.h>
#include <framework/logging/Logger.h>
 
#include <generators/bhwide/BHWide.h>
 
#include <mdst/dataobjects/MCParticleGraph.h>
 
#include <generators/utilities/InitialParticleGeneration.h>
 
#include <vector>
 
namespace Belle2 {

  class BHWideInputModule : public Module {
 
  public:

    BHWideInputModule();

    virtual ~BHWideInputModule();

    virtual void initialize() override;

    virtual void event() override;

    virtual void terminate() override;
 
  protected:
 
    double getBeamEnergyCM(double e1, double e2, double angle); 

    std::vector<double>
    m_ScatteringAngleRangePositron; 
    std::vector<double>
    m_ScatteringAngleRangeElectron; 
 
    bool m_weakCorrections;  
    std::string m_vacPolString; 
    BHWide::PhotonVacPolarization m_vacPol; 
    double m_eMin; 
    double m_maxAcollinearity; 
    double m_wtMax; 

    BHWide 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> BHWideInputModule::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> BHWideInputModule::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