RestOfEventInterpreterModule.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 <analysis/modules/RestOfEventInterpreter/RestOfEventInterpreterModule.h>
 
#include <analysis/dataobjects/Particle.h>
#include <analysis/dataobjects/RestOfEvent.h>
 
#include <framework/logging/Logger.h>
#include <framework/core/ModuleParam.templateDetails.h>
 
#include <iostream>
 
using namespace std;
 
namespace Belle2 {
  //-----------------------------------------------------------------
  //                 Register the Module
  //-----------------------------------------------------------------
  REG_MODULE(RestOfEventInterpreter)
 
  //-----------------------------------------------------------------
  //                 Implementation
  //-----------------------------------------------------------------
 
  RestOfEventInterpreterModule::RestOfEventInterpreterModule() : Module()
  {
    // Set module properties
    setDescription("Creates a mask (vector of boolean values) for tracks and clusters in RestOfEvent.");
    setPropertyFlags(c_ParallelProcessingCertified);
 
    // Add parameters
    std::vector<std::tuple<std::string, std::string, std::string, std::string>> emptyROEMask;
 
    addParam("particleList", m_particleListName, "Name of the ParticleList");
 
    addParam("ROEMasks", m_ROEMasks,
             "List of (maskName, trackSelectionCut, eclClusterSelectionCut) tuples that specify all ROE masks of a specific particle to be created.",
             emptyROEMask);
 
    addParam("update", m_update, "Set true for updating a-priori charged stable fractions used in calculation of ROE 4-momentum",
             false);
 
  }
 
  void RestOfEventInterpreterModule::initialize()
  {
    // input
    m_plist.isRequired(m_particleListName);
 
    for (auto ROEMask : m_ROEMasks) {
      // parsing of the input tuple (maskName, trackSelectionCut, eclClusterSelectionCut, fractions)
 
      std::string maskName = get<0>(ROEMask);
      std::string trackSelection = get<1>(ROEMask);
      std::string eclClusterSelection = get<2>(ROEMask);
      std::string klmClusterSelection = get<3>(ROEMask);
 
      std::shared_ptr<Variable::Cut> trackCut = std::shared_ptr<Variable::Cut>(Variable::Cut::compile(trackSelection));
      std::shared_ptr<Variable::Cut> eclClusterCut = std::shared_ptr<Variable::Cut>(Variable::Cut::compile(eclClusterSelection));
      std::shared_ptr<Variable::Cut> klmClusterCut = std::shared_ptr<Variable::Cut>(Variable::Cut::compile(klmClusterSelection));
 
      m_maskNames.push_back(maskName);
      m_trackCuts.insert(stringAndCutMap::value_type(maskName, trackCut));
      m_eclClusterCuts.insert(stringAndCutMap::value_type(maskName, eclClusterCut));
      m_klmClusterCuts.insert(stringAndCutMap::value_type(maskName, klmClusterCut));
 
      B2INFO("RestOfEventInterpreter added ROEMask with specific fractions under name \'" << maskName << "\' with track cuts: " <<
             trackSelection << " and eclCluster cuts: " << eclClusterSelection);
    }
  }
 
  void RestOfEventInterpreterModule::event()
  {
    unsigned int nParts = m_plist->getListSize();
 
    for (unsigned i = 0; i < nParts; i++) {
      const Particle* particle = m_plist->getParticle(i);
      auto* roe = particle->getRelatedTo<RestOfEvent>("ALL");
      for (auto& maskName : m_maskNames) {
        if (!m_update) {
          roe->initializeMask(maskName, "ROEInterpreterModule");
        }
        roe->updateMaskWithCuts(maskName, m_trackCuts.at(maskName), m_eclClusterCuts.at(maskName), m_klmClusterCuts.at(maskName), m_update);
      }
    }
  }
 
}