EvtGenInterface.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 <EvtGen/EvtGen.hh>
#include <EvtGenBase/EvtParticle.hh>
#include <EvtGenBase/EvtVector4R.hh>
#include <generators/evtgen/EvtGenFwRandEngine.h>
#include <mdst/dataobjects/MCParticleGraph.h>
 
#include <framework/utilities/IOIntercept.h>
 
#include <Math/Vector3D.h>
#include <string>
 
namespace Belle2 {
  class EvtGenInterface {
 
  public:
    static EvtGen* createEvtGen(const std::string& decayFileName, bool coherentMixing);
 
    EvtGenInterface():
      m_parent(0),
      m_Generator(0),
      m_pinit(0, 0, 0, 0),
      m_ParentInitialized(false),
      m_logCaptureDebug("EvtGen", LogConfig::c_Debug, LogConfig::c_Warning, 20, 20)
    {}
 
    ~EvtGenInterface();
 
    int setup(const std::string& decayFileName, const std::string& parentParticle,
              const std::string& userFileName = std::string(""), bool coherentMixing = true);
 
    int simulateEvent(MCParticleGraph& graph, ROOT::Math::PxPyPzEVector pParentParticle,
                      ROOT::Math::XYZVector pPrimaryVertex, int inclusiveType, const std::string& inclusiveParticle);
 
    int simulateDecay(MCParticleGraph& graph,
                      MCParticleGraph::GraphParticle& parent);
 
  private:
    int addParticles2Graph(EvtParticle* particle, MCParticleGraph& graph, ROOT::Math::XYZVector pPrimaryVertex,
                           MCParticleGraph::GraphParticle* parent, double timeOffset = 0);
 
    void updateGraphParticle(EvtParticle* eParticle, MCParticleGraph::GraphParticle* gParticle,
                             ROOT::Math::XYZVector pPrimaryVertex, double timeOffset = 0);
 
  protected:
    EvtParticle* m_parent;      
    static EvtGenFwRandEngine m_eng;   
    EvtGen* m_Generator;        
    EvtVector4R m_pinit;        
    EvtId m_ParentParticle;     
    bool m_ParentInitialized;   
    IOIntercept::OutputToLogMessages m_logCaptureDebug; 
  }; 
 
}