HepmcInputModule.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2019  Belle II Collaboration                              *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Jo-Frrederik Krohn                                       *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
#pragma once
 
#include <framework/core/Module.h>
 
#include <string>
 
#include <mdst/dataobjects/MCParticleGraph.h>
 
#include <generators/hepmc/HepMCReader.h>
#include <generators/utilities/InitialParticleGeneration.h>
 
namespace Belle2 {
  class HepMCInputModule : public Module {
 
  public:
 
    HepMCInputModule();
 
    virtual ~HepMCInputModule() {}
 
    virtual void initialize() override;
 
    virtual void event() override;
 
    virtual void terminate() override;
 
  protected:
 
    std::vector <std::string>  m_inputFileNames; 
    std::string m_inputFileName;   
    uint m_iFile;                     
    int m_nVirtual;                  
    std::unique_ptr<HepMCReader> m_hepmcreader;           
    MCParticleGraph m_mcParticleGraph;             
    bool m_useWeights;               
    bool m_boost2Lab;                
    bool m_wrongSignPz;              
    bool m_ignorereadEventNr;               
    int m_runNum;                    
    int m_expNum;                    
    int m_evtNum;                    
    int m_minEvent;                    
    int m_maxEvent;                    
    int m_totalEvents;                    
    //int m_skipNEvents;                    /**< skip events in the file */
    DBObjPtr<BeamParameters> m_beamParams; 
  private:
    InitialParticleGeneration m_initial; 
    StoreObjPtr<EventMetaData> m_eventMetaDataPtr; 
  };
 
} // end namespace Belle2