FragmentationModule.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2010 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Ami Rostomyan, Torben Ferber                             *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
#include <framework/core/Module.h>
#include <framework/datastore/StoreArray.h>
#include <mdst/dataobjects/MCParticle.h>
#include <mdst/dataobjects/MCParticleGraph.h>
 
#include <Pythia8/Pythia.h>
#include <generators/modules/fragmentation/EvtGenDecays.h>
 
#include <string>
 
namespace Belle2 {
  class FragmentationRndm : public Pythia8::RndmEngine {
 
  public:
    // Constructor
    FragmentationRndm();
 
    // in PYTHIA, there is one pure virtual method in RndmEngine,
    // to generate one random number flat in the range between 0 and 1:
    double flat(); 
  private:
 
  };
 
 
  class FragmentationModule : public Module {
 
  public:
    FragmentationModule();
 
    virtual ~FragmentationModule();
 
    virtual void event() override;      
    virtual void initialize() override; 
    virtual void terminate() override;  
  protected:
 
    void loadEvtGenParticleData(Pythia8::Pythia* pythia);
 
    int addParticleToPYTHIA(const MCParticle& mcParticle);
 
    Pythia8::Pythia* m_Pythia = nullptr;
 
    Pythia8::Event* m_PythiaEvent = nullptr;
 
    EvtGenDecays* evtgen; 
    std::string m_parameterfile; 
    int m_listEvent;  
    int m_useEvtGen;  
    std::string m_DecFile;  
    std::string m_UserDecFile;  
    bool m_coherentMixing;  
    MCParticleGraph mcParticleGraph; 
    int nAll;   
    int nGood;   
    int nAdded;  
    int nQuarks; 
    int nVpho; 
    std::string m_particleList; 
    StoreArray<MCParticle> m_mcparticles; 
  };
 
} // end namespace Belle2