OnlineEventT0CreatorModule.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.                  *
 **************************************************************************/
 
/* Own header. */
#include <hlt/modules/onlineEventT0/OnlineEventT0CreatorModule.h>
 
/* Belle 2 headers. */
#include <framework/core/Environment.h>
#include <framework/logging/LogConfig.h>
 
using namespace Belle2;
 
REG_MODULE(OnlineEventT0Creator)
 
OnlineEventT0CreatorModule::OnlineEventT0CreatorModule() : Module()
{
  setDescription("Module to write the EventT0s computed on the online systems (HLT). "
                 "By default, this module is run only on the online systems unless the user "
                 "specifically requires to execute it.");
  addParam("executeThisModule", m_executeThisModule,
           "Execute this module: if True, this module is executed durig the reconstruction, otherwise not.", false);
  setPropertyFlags(c_ParallelProcessingCertified);
}
 
void OnlineEventT0CreatorModule::initialize()
{
  // If we are running online, execute the module regardless of what the user specified.
  LogConfig::ELogRealm realm = Environment::Instance().getRealm();
  if (realm == LogConfig::c_Online)
    m_executeThisModule = true;
  // Run the usual checks only if the module is actually executed.
  if (m_executeThisModule) {
    m_onlineEventT0.registerInDataStore();
    m_eventT0.isRequired();
  }
}
 
void OnlineEventT0CreatorModule::event()
{
  if (!m_executeThisModule) {
    B2DEBUG(20, "OnlineEventT0 is not executed as requested");
    return;
  }
  if (!m_eventT0.isValid()) {
    B2DEBUG(20, "EventT0 object not created, cannot write OnlineEventT0");
    return;
  }
  if (m_onlineEventT0.getEntries() > 0) {
    B2DEBUG(20, "OnlineEventT0 object already present, do nothing");
    return;
  }
  // check if ECL hypothesis exists
  auto eclHypos = m_eventT0->getTemporaryEventT0s(Const::EDetector::ECL);
  if (eclHypos.size() == 0) {
    B2DEBUG(20, "No ECL EventT0 available");
  } else {
    // get the most accurate ECL evenT0 (smallest chi2/quality)
    auto eclBestT0 = std::min_element(eclHypos.begin(), eclHypos.end(), [](EventT0::EventT0Component c1,
    EventT0::EventT0Component c2) {return c1.quality < c2.quality;});
    m_onlineEventT0.appendNew(eclBestT0->eventT0, eclBestT0->eventT0Uncertainty, Const::EDetector::ECL);
  }
  // check if a CDC hypothesis exists
  auto cdcHypos = m_eventT0->getTemporaryEventT0s(Const::EDetector::CDC);
  if (cdcHypos.size() == 0) {
    B2DEBUG(20, "No CDC EventT0 available");
  } else {
    // get the most accurate CDC evenT0 (latest)
    const auto& cdcBestT0 = cdcHypos.back();
    m_onlineEventT0.appendNew(cdcBestT0.eventT0, cdcBestT0.eventT0Uncertainty, Const::EDetector::CDC);
  }
  // check if a TOP hypothesis exists
  auto topHypos = m_eventT0->getTemporaryEventT0s(Const::EDetector::TOP);
  if (topHypos.size() == 0) {
    B2DEBUG(20, "No TOP EventT0 available");
  } else {
    // get the most accurate TOP eventT0 (there is only one)
    const auto& topBestT0 = topHypos.back();
    m_onlineEventT0.appendNew(topBestT0.eventT0, topBestT0.eventT0Uncertainty, Const::EDetector::TOP);
  }
}