SignalSideParticleFilterModule.cc

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2013 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Anze Zupanc                                              *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#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);
  bool inTheLists = false;
 
  StoreObjPtr<RestOfEvent> roe("RestOfEvent");
  if (roe.isValid()) {
    const Particle* particle = roe->getRelated<Particle>();
    if (particle) {
      if (m_cut->check(particle)) {
        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)) {
            inTheLists = true; break;
          }
        }
      }
    }
  }
  setReturnValue(inTheLists);
}