SignalSideParticleFilterModule.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/SignalSideParticleFilter/SignalSideParticleFilterModule.h>
 
#include <framework/datastore/StoreArray.h>
#include <framework/datastore/StoreObjPtr.h>
 
#include <analysis/dataobjects/Particle.h>
#include <analysis/dataobjects/ParticleList.h>
#include <analysis/dataobjects/RestOfEvent.h>
 
using namespace Belle2;
 
//-----------------------------------------------------------------
//                 Register the Module
//-----------------------------------------------------------------
REG_MODULE(SignalSideParticleFilter);
 
//-----------------------------------------------------------------
//                 Implementation
//-----------------------------------------------------------------
 
SignalSideParticleFilterModule::SignalSideParticleFilterModule() : Module()
{
  // Set module properties
  setDescription("The module returns true if the current RestOfEvent object is related to\n"
                 "any of the Particles from the input ParticleList and passes selection criteria.\n"
                 "The module should be executed only in the for_each ROE path.");
 
  // Parameter definitions
  addParam("particleLists", m_particleLists, "Input ParticleList name", std::vector<std::string>());
  addParam("selection", m_selection, "Additional selection criteria", std::string(""));
}
 
void SignalSideParticleFilterModule::initialize()
{
  StoreArray<Particle>().isRequired();
 
  for (auto& iParticleListName : m_particleLists) {
 
    StoreObjPtr<ParticleList> iParticlelist;
    iParticlelist.isRequired(iParticleListName);
 
  }
 
  m_cut = Variable::Cut::compile(m_selection);
 
}
 
void SignalSideParticleFilterModule::event()
{
  setReturnValue(false);
 
  StoreObjPtr<RestOfEvent> roe("RestOfEvent");
  if (!roe.isValid())
    return;
 
  const Particle* particle = roe->getRelatedFrom<Particle>();
  if (!particle)
    return;
 
  for (auto& iParticleListName : m_particleLists) {
    StoreObjPtr<ParticleList> iParticlelist(iParticleListName);
 
    if (!iParticlelist) {
      B2WARNING("Input list " << iParticlelist.getName() << " was not created?");
      continue;
    }
 
    const unsigned int numParticles = iParticlelist->getListSize();
    if (numParticles == 0)
      continue;
 
    if (iParticlelist->contains(particle)) {
      // cut should be called only if the particle is in the ParticleList
      if (m_cut->check(particle)) {
        setReturnValue(true);
        return;
      }
    }
  }
 
}