KoralW.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
 
/* Basf2 headers. */
#include <mdst/dataobjects/MCParticleGraph.h>
 
/* ROOT headers. */
#include <Math/LorentzRotation.h>
#include <Math/Vector3D.h>
 
/* C++ headers. */
#include <string>
 
namespace Belle2 {
  class KoralW {
 
  public:
 
    KoralW() :
      m_crossSection(0.0),
      m_crossSectionError(0.0),
      m_cmsEnergy(0.0),
      m_seed1(900000000),
      m_seed2(10000),
      m_seed3(1000)
    {for (int i = 0; i < m_numXPar; i++) m_XPar[i] = 0.0;}
 
 
    ~KoralW() {};
 
    void setCMSEnergy(double cmsEnergy) { m_cmsEnergy = cmsEnergy; }
 
    void init(const std::string& dataPath, const std::string& userDataFile);
 
    void generateEvent(MCParticleGraph& mcGraph, ROOT::Math::XYZVector vertex, ROOT::Math::LorentzRotation boost);
 
    void term();
 
    double getCrossSection() { return m_crossSection; }
 
    double getCrossSectionError() { return m_crossSectionError; }
 
 
  protected:
 
    double m_crossSection;      
    double m_crossSectionError; 
    double m_cmsEnergy; 
    void storeParticle(MCParticleGraph& mcGraph, const float* mom, const float* vtx, int pdg, ROOT::Math::XYZVector vertex,
                       ROOT::Math::LorentzRotation boost,
                       bool isVirtual = false, bool isInitial = false);
 
  private:
 
    static constexpr int m_numXPar = 10000; 
    double m_XPar[m_numXPar];  
    unsigned int m_seed1; 
    unsigned int m_seed2; 
    unsigned int m_seed3; 
  };
}