LHEInputModule.cc

/**************************************************************************
 * 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.                  *
 **************************************************************************/
 
#include <generators/modules/LHEInputModule.h>
 
#include <framework/logging/Logger.h>
#include <framework/datastore/DataStore.h>
#include <framework/datastore/StoreArray.h>
#include <framework/core/Environment.h>
 
using namespace std;
using namespace Belle2;
 
REG_MODULE(LHEInput);
 
LHEInputModule::LHEInputModule() : Module(),
  m_nInitial(0),
  m_nVirtual(0),
  m_evtNum(-1)
{
  //Set module properties
  setDescription("LHE file input. This module loads an event record from LHE format and store the content into the MCParticle collection. LHE format is a standard event record format to contain an event record in a Monte Carlo-independent format.");
  setPropertyFlags(c_Input);
 
  //Parameter definition
  addParam("inputFileList", m_inputFileNames, "List of names of LHE files");
  addParam("createEventMetaData", m_createEventMetaData,
           "Boolean to indicate whether the event numbers from input file should be used.", false);
  addParam("runNum", m_runNum, "Run number", -1);
  addParam("expNum", m_expNum, "Experiment number", -1);
  addParam("skipEvents", m_skipEventNumber, "Skip this number of events before starting.", 0);
  addParam("useWeights", m_useWeights, "Set to 'true' to if generator weights should be propagated.", false);
  addParam("nInitialParticles", m_nInitial, "Number of MCParticles at the beginning of the events that should be flagged c_Initial.",
           0);
  addParam("nVirtualParticles", m_nVirtual,
           "Number of MCParticles at the beginning of the events that should be flagged c_IsVirtual.", 0);
  addParam("wrongSignPz", m_wrongSignPz, "Boolean to signal that directions of HER and LER were switched", true);
}
 
void LHEInputModule::initialize()
{
  if (m_createEventMetaData) {
    if (m_expNum < 0 or m_runNum < 0)
      B2FATAL("The exp. and run numbers are not properly initialized: please set the 'expNum' and 'runNum' parameters of the LHEInput module.");
 
    m_eventMetaData.registerInDataStore(DataStore::c_ErrorIfAlreadyRegistered);
    B2INFO("LHEInput acts as input module for this process. This means the exp., run and event numbers will be set by this module.");
  }
 
  m_iFile = 0;
  if (m_inputFileNames.size() == 0) {
    //something is wrong with the file list.
    B2FATAL("Invalid list of input files, no entries found.");
  } else {
    //let's start with the first file:
    m_inputFileName = m_inputFileNames[m_iFile];
  }
 
  unsigned int totalEvents = 0;
  for (const auto& fileName : m_inputFileNames) {
    totalEvents += m_lhe.countEvents(fileName);
  }
  Environment::Instance().setNumberOfMCEvents(totalEvents);
 
  try {
    B2INFO("Opening first file: " << m_inputFileName);
    m_lhe.open(m_inputFileName);
    m_lhe.skipEvents(m_skipEventNumber);
  } catch (runtime_error& e) {
    B2FATAL(e.what());
  }
  m_lhe.setInitialIndex(m_nInitial);
  m_lhe.setVirtualIndex(m_nInitial + m_nVirtual);
  m_lhe.m_wrongSignPz = m_wrongSignPz;
 
  //Initialize MCParticle collection
  StoreArray<MCParticle> mcparticle;
  mcparticle.registerInDataStore();
}
 
void LHEInputModule::event()
{
  if (!m_eventMetaData)
    m_eventMetaData.create();
  try {
    m_pg.clear();
    double weight = 1;
    int id = m_lhe.getEvent(m_pg, weight);
    if (m_createEventMetaData) {
      if (id > -1) {
        m_evtNum = id;
      } else {
        id = ++m_evtNum;
      }
      m_eventMetaData->setExperiment(m_expNum);
      m_eventMetaData->setRun(m_runNum);
      m_eventMetaData->setEvent(id);
    }
    if (m_useWeights)
      m_eventMetaData->setGeneratedWeight(weight);
    m_pg.generateList("", MCParticleGraph::c_setDecayInfo | MCParticleGraph::c_checkCyclic);
  } catch (LHEReader::LHEEmptyEventError&) {
    B2DEBUG(100, "Reached end of LHE file.");
    m_lhe.closeCurrentInputFile();
    m_iFile++;
    if (m_iFile < m_inputFileNames.size()) {
      try {
        m_inputFileName = m_inputFileNames[m_iFile];
        B2INFO("Opening next file: " << m_inputFileName);
        m_lhe.open(m_inputFileName);
      } catch (runtime_error& e) {
        B2FATAL(e.what());
      }
    } else {
      m_eventMetaData->setEndOfData();
      B2DEBUG(100, "Reached end of all LHE files.");
    }
  } catch (runtime_error& e) {
    B2ERROR(e.what());
  }
}