BabayagaNLOInputModule.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/logging/Logger.h>
 
#include <generators/modules/GeneratorBaseModule.h>
#include <generators/babayaganlo/BabayagaNLO.h>
#include <generators/utilities/InitialParticleGeneration.h>
 
#include <mdst/dataobjects/MCParticleGraph.h>
 
#include <vector>
#include <TFile.h>
#include <TH1D.h>
 
 
namespace Belle2 {

  class BabayagaNLOInputModule : public GeneratorBaseModule {
 
  public:

    BabayagaNLOInputModule();//: Module(), m_initial(BeamParameters::c_smearALL) {}

    virtual ~BabayagaNLOInputModule();

    virtual void generatorInitialize() override;

    virtual void generatorEvent() override;

    virtual void terminate() override;

    static double generateCMSEnergy()
    {
      return s_initial.generate().getMass();
    }
 
  protected:
 
    double getBeamEnergyCM(double e1, double e2, double angle); 

    std::string m_order; 
    std::string m_model; 
    std::string m_mode; 
    std::string m_vacPol; 
    std::string m_finalState; 
    double m_eMinFrac; 
    double m_eMin; 
    double m_epsilon; 
    double m_maxAcollinearity; 
    double m_fMax; 
    int m_nSearchMax; 
    int m_nPhot; 
    std::vector<double> m_ScatteringAngleRange; 
    std::string m_fileNameExtraInfo; 
    double m_Spread; 
    bool m_Uncertainty;   
    std::string m_NSKDataFile; 
 
    std::string m_userMode; 
    double m_eemin; 
    double m_temin; 
    double m_egmin; 
    double m_tgmin; 
    double m_eeveto; 
    double m_teveto; 
    double m_egveto; 
    double m_tgveto; 
    double m_maxprescale; 

    BabayagaNLO m_generator;   
    MCParticleGraph m_mcGraph; 
    TFile* m_fileExtraInfo;  
    TH1D* m_th1dSDif;   
 
    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; 
 
    static InitialParticleGeneration
    s_initial;   
  };
 
 
  template <typename T>
  inline std::vector<T> BabayagaNLOInputModule::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> BabayagaNLOInputModule::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