TriggerVariables.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/variables/TriggerVariables.h>
#include <analysis/dataobjects/Particle.h>
 
// trigger dataobjects
#include <mdst/dataobjects/TRGSummary.h>
#include <mdst/dataobjects/SoftwareTriggerResult.h>
 
// trigger dbobjects
#include <mdst/dbobjects/TRGGDLDBFTDLBits.h>
#include <mdst/dbobjects/TRGGDLDBPrescales.h>
 
// HLT dbobjects
#include <mdst/dbobjects/DBRepresentationOfSoftwareTriggerCut.h>
 
// framework
#include <framework/logging/Logger.h>
#include <framework/datastore/StoreObjPtr.h>
#include <framework/database/DBObjPtr.h>
#include <framework/utilities/Conversion.h>
 
// boost
#include <boost/algorithm/string.hpp>
 
// C++
#include <limits>
#include <stdexcept>
 
namespace Belle2 {
  namespace {
    std::string fullFormatIdentifier(const std::string& identifier)
    {
      std::string out = identifier;
      boost::replace_all(out, " ", "&");
      if (identifier.substr(0, 21) != "software_trigger_cut&")
        out = "software_trigger_cut&" + out;
      return out;
    }
 
    double extractSoftwareTriggerResultImplementation(bool nonPrescaled, const std::string& triggerIdentifier, const Particle*)
    {
      // get trigger result object
      StoreObjPtr<SoftwareTriggerResult> swtr;
      if (!swtr)
        return std::numeric_limits<double>::quiet_NaN();
 
      // check that the trigger ID provided by the user exists in the SWTR
      SoftwareTriggerCutResult swtcr;
      try {
        if (nonPrescaled) {
          swtcr = swtr->getNonPrescaledResult(fullFormatIdentifier(triggerIdentifier));
        } else {
          swtcr = swtr->getResult(fullFormatIdentifier(triggerIdentifier));
        }
      } catch (const std::out_of_range&) {
        // then the trigger identifier is wrong -- silently return nan
        return std::numeric_limits<double>::quiet_NaN();
      }
      return double(swtcr); // see mdst/dataobjects/include/SoftwareTriggerResult.h
    };
  }
 
  namespace Variable {
 
    double L1Trigger(const Particle*)
    {
      StoreObjPtr<TRGSummary> trg;
      if (!trg)
        return std::numeric_limits<double>::quiet_NaN();
      return trg->test();
    }
 
    Manager::FunctionPtr L1PSNM(const std::vector<std::string>& arguments)
    {
      if (arguments.size() == 1) {
        auto name = arguments[0];
        auto func = [name](const Particle*) -> double {
          StoreObjPtr<TRGSummary> trg;
          if (!trg)
            return std::numeric_limits<double>::quiet_NaN();
          try {
            return trg->testPsnm(name);
          } catch (const std::exception&)
          {
            // Something went wrong, return NaN.
            return std::numeric_limits<double>::quiet_NaN();
          }
        };
        return func;
      } else {
        B2FATAL("Wrong number of arguments for L1PSNM function. The only argument must be the name of the PSNM trigger bit.");
      }
    }
 
    Manager::FunctionPtr L1PSNMBit(const std::vector<std::string>& arguments)
    {
      if (arguments.size() == 1) {
        int testBit;
        try {
          testBit = Belle2::convertString<int>(arguments[0]);
        } catch (const std::invalid_argument&) {
          B2FATAL("Invalid argument for L1PSNMBit function. The argument must be an integer representing the PSNM trigger bit.");
        }
        auto func = [testBit](const Particle*) -> double {
          StoreObjPtr<TRGSummary> trg;
          if (!trg)
            return std::numeric_limits<double>::quiet_NaN();
          try {
            return trg->testPsnm(testBit);
          } catch (const std::exception&)
          {
            // Something went wrong, return NaN.
            return std::numeric_limits<double>::quiet_NaN();
          }
        };
        return func;
      } else {
        B2FATAL("Wrong number of arguments for L1PSNMBit function. The only argument must be the number of the PSNM trigger bit.");
      }
    }
 
    Manager::FunctionPtr L1FTDL(const std::vector<std::string>& arguments)
    {
      if (arguments.size() == 1) {
        auto name = arguments[0];
        auto func = [name](const Particle*) -> double {
          StoreObjPtr<TRGSummary> trg;
          if (!trg)
            return std::numeric_limits<double>::quiet_NaN();
          try {
            return trg->testFtdl(name);
          } catch (const std::exception&)
          {
            // Something went wrong, return NaN.
            return std::numeric_limits<double>::quiet_NaN();
          }
        };
        return func;
      } else {
        B2FATAL("Wrong number of arguments for L1FTDL function. The only argument must be the name of the FTDL trigger bit.");
      }
    }
 
    Manager::FunctionPtr L1FTDLBit(const std::vector<std::string>& arguments)
    {
      if (arguments.size() == 1) {
        int testBit;
        try {
          testBit = Belle2::convertString<int>(arguments[0]);
        } catch (const std::invalid_argument&) {
          B2FATAL("Invalid argument for L1FTDLBit function. The argument must be an integer representing the FTDL trigger bit.");
        }
        auto func = [testBit](const Particle*) -> double {
          StoreObjPtr<TRGSummary> trg;
          if (!trg)
            return std::numeric_limits<double>::quiet_NaN();
          try {
            return trg->testFtdl(testBit);
          } catch (const std::exception&)
          {
            // Something went wrong, return NaN.
            return std::numeric_limits<double>::quiet_NaN();
          }
        };
        return func;
      } else {
        B2FATAL("Wrong number of arguments for L1FTDLBit function. The only argument must be the number of the FTDL trigger bit.");
      }
    }
 
    Manager::FunctionPtr L1Input(const std::vector<std::string>& arguments)
    {
      if (arguments.size() == 1) {
        auto name = arguments[0];
        auto func = [name](const Particle*) -> double {
          StoreObjPtr<TRGSummary> trg;
          if (!trg)
            return std::numeric_limits<double>::quiet_NaN();
          try {
            return trg->testInput(name);
          } catch (const std::exception&)
          {
            // Something went wrong, return NaN.
            return std::numeric_limits<double>::quiet_NaN();
          }
        };
        return func;
      } else {
        B2FATAL("Wrong number of arguments for L1Input function. The only argument must be the name of the input trigger bit.");
      }
    }
 
    Manager::FunctionPtr L1InputBit(const std::vector<std::string>& arguments)
    {
      if (arguments.size() == 1) {
        int testBit;
        try {
          testBit = Belle2::convertString<int>(arguments[0]);
        } catch (const std::invalid_argument&) {
          B2FATAL("Invalid argument for L1InputBit function. The argument must be an integer representing the input trigger bit.");
        }
        auto func = [testBit](const Particle*) -> double {
          StoreObjPtr<TRGSummary> trg;
          if (!trg)
            return std::numeric_limits<double>::quiet_NaN();
          try {
            return trg->testInput(testBit);
          } catch (const std::exception&)
          {
            // Something went wrong, return NaN.
            return std::numeric_limits<double>::quiet_NaN();
          }
        };
        return func;
      } else {
        B2FATAL("Wrong number of arguments for L1Input function. The only argument must be the number of the input trigger bit.");
      }
    }
 
    Manager::FunctionPtr L1PSNMPrescale(const std::vector<std::string>& arguments)
    {
      if (arguments.size() == 1) {
        auto name = arguments[0];
        auto func = [name](const Particle*) -> double {
          static DBObjPtr<TRGGDLDBFTDLBits> ftdlBits;
          if (!ftdlBits.isValid())
            return std::numeric_limits<double>::quiet_NaN();
          static DBObjPtr<TRGGDLDBPrescales> prescales;
          if (!prescales.isValid())
            return std::numeric_limits<double>::quiet_NaN();
          for (unsigned int bit = 0; bit < TRGSummary::c_trgWordSize * TRGSummary::c_ntrgWords; bit++)
          {
            if (std::string(ftdlBits->getoutbitname((int)bit)) == name)
              return prescales->getprescales(bit);
          }
          return std::numeric_limits<double>::quiet_NaN();
        };
        return func;
      } else {
        B2FATAL("Wrong number of arguments for L1Prescale function. The only argument must be the name of the PSNM trigger bit.");
      }
    }
 
    Manager::FunctionPtr L1PSNMBitPrescale(const std::vector<std::string>& arguments)
    {
      if (arguments.size() == 1) {
        int testBit;
        try {
          testBit = Belle2::convertString<int>(arguments[0]);
        } catch (const std::invalid_argument&) {
          B2FATAL("Invalid argument for L1PSNMBitPrescale function. The argument must be an integer representing the PSNM trigger bit.");
        }
        auto func = [testBit](const Particle*) -> double {
          if (testBit < 0 or testBit >= (int)TRGSummary::c_trgWordSize * (int)TRGSummary::c_ntrgWords)
            return std::numeric_limits<double>::quiet_NaN();
          static DBObjPtr<TRGGDLDBPrescales> prescales;
          if (!prescales.isValid())
            return std::numeric_limits<double>::quiet_NaN();
          return prescales->getprescales(testBit);
        };
        return func;
      } else {
        B2FATAL("Wrong number of arguments for L1BitPrescale function. The only argument must be the number of the PSNM trigger bit.");
      }
    }
 
    double L1TimeType(const Particle*)
    {
      StoreObjPtr<TRGSummary> trg;
      if (!trg)
        return std::numeric_limits<double>::quiet_NaN();
      return trg->getTimType();
    }
 
    double L1TimeQuality(const Particle*)
    {
      StoreObjPtr<TRGSummary> trg;
      if (!trg)
        return std::numeric_limits<double>::quiet_NaN();
      return trg->getTimQuality();
    }
 
    double isPoissonInInjectionVeto(const Particle*)
    {
      StoreObjPtr<TRGSummary> trg;
      if (!trg)
        return std::numeric_limits<double>::quiet_NaN();
      return trg->isPoissonInInjectionVeto();
    }
 
    Manager::FunctionPtr softwareTriggerResult(const std::vector<std::string>& args)
    {
      /* The analyst has to know the name of the trigger she wants
       * (after having looked this up from the trigger db payload)
       *
       * This workflow will probably improve later: but for now parse the args
       * to check we have one name, then check the name does not throw an
       * exception when we ask for it from the SWTR (std::map)
       */
      if (args.size() != 1)
        B2FATAL("Wrong number of arguments for the function softwareTriggerResult");
      std::string triggerIdentifier = args[0];
 
      using namespace std::placeholders;
      return std::bind(extractSoftwareTriggerResultImplementation, false, triggerIdentifier, _1);
    }
 
    Manager::FunctionPtr softwareTriggerResultNonPrescaled(const std::vector<std::string>& args)
    {
      /* The analyst has to know the name of the trigger she wants
       * (after having looked this up from the trigger db payload)
       *
       * This workflow will probably improve later: but for now parse the args
       * to check we have one name, then check the name does not throw an
       * exception when we ask for it from the SWTR (std::map)
       */
      if (args.size() != 1)
        B2FATAL("Wrong number of arguments for the function softwareTriggerResultNonPrescaled");
      std::string triggerIdentifier = args[0];
 
      using namespace std::placeholders;
      return std::bind(extractSoftwareTriggerResultImplementation, true, triggerIdentifier, _1);
    }
 
    double passesAnyHighLevelTrigger(const Particle* p)
    {
      // for HLT, a c_accept is a pass and all other cases are fail
      // see mdst/dataobjects/include/SoftwareTriggerResult.h
      std::vector<std::string> hardcodedname
        = { "software_trigger_cut&filter&total_result" };
      double swtcr = softwareTriggerResult(hardcodedname)(p);
      if (swtcr > 0.5) return 1.0; // 1
      else             return 0.0; // 0 or -1
    }
 
    Manager::FunctionPtr softwareTriggerPrescaling(const std::vector<std::string>& args)
    {
      /* The analyst has to know the name of the trigger she wants
       * (after having looked this up from the trigger db payload)
       *
       * This workflow will probably improve later: but for now parse the args
       * to check we have one name, then check the database object is valid.
       * If not return NAN.
       */
      if (args.size() != 1)
        B2FATAL("Wrong number of arguments for the function softwareTriggerPrescaling");
      std::string triggerIdentifier = args[0];
 
      auto outputFunction = [triggerIdentifier](const Particle*) -> double {
 
        DBObjPtr<DBRepresentationOfSoftwareTriggerCut> downloadedCut(fullFormatIdentifier(triggerIdentifier));
        if (not downloadedCut)
          return std::numeric_limits<double>::quiet_NaN();
        return double(downloadedCut->getPreScaleFactor());
      };
 
      return outputFunction;
    }
 
    //-------------------------------------------------------------------------
    VARIABLE_GROUP("L1 Trigger");
    REGISTER_VARIABLE("L1Trigger", L1Trigger ,
                      "[Eventbased] Returns 1 if at least one PSNM L1 trigger bit is true.");
    REGISTER_VARIABLE("L1PSNM(name)", L1PSNM ,
                      "[Eventbased] Returns the PSNM (Prescale And Mask, after prescale) status of the trigger bit with the given name.");
    REGISTER_VARIABLE("L1FTDL(name)", L1FTDL ,
                      "[Eventbased] Returns the FTDL (Final Trigger Decision Logic, before prescale) status of the trigger bit with the given name.");
    REGISTER_VARIABLE("L1Input(name)", L1Input,
                      "[Eventbased] Returns the input bit status of the trigger bit with the given name.");
    REGISTER_VARIABLE("L1Prescale(name)", L1PSNMPrescale,
                      "[Eventbased] Returns the PSNM (prescale and mask) prescale of the trigger bit with the given name.");
    REGISTER_VARIABLE("L1PSNMBit(i)", L1PSNMBit,
                      "[Eventbased] Returns the PSNM (Prescale And Mask, after prescale) status of i-th trigger bit.");
    REGISTER_VARIABLE("L1FTDLBit(i)", L1FTDLBit,
                      "[Eventbased] Returns the FTDL (Final Trigger Decision Logic, before prescale) status of i-th trigger bit.");
    REGISTER_VARIABLE("L1InputBit(i)", L1InputBit,
                      "[Eventbased] Returns the input bit status of the i-th input trigger bit.");
    REGISTER_VARIABLE("L1PSNMBitPrescale(i)", L1PSNMBitPrescale,
                      "[Eventbased] Returns the PSNM (prescale and mask) prescale of i-th trigger bit.");
    REGISTER_VARIABLE("L1TimeType", L1TimeType,
                      "[Eventbased] Returns ETimingType time type.");
    REGISTER_VARIABLE("L1TimeQuality", L1TimeQuality,
                      "[Eventbased] Returns ETimingQuality time quality.");
    REGISTER_VARIABLE("isPoissonTriggerInInjectionVeto", isPoissonInInjectionVeto,
                      "[Eventbased] Returns 1 if the poisson random trigger is within the injection veto window.");
    //-------------------------------------------------------------------------
    VARIABLE_GROUP("Software Trigger");
    REGISTER_VARIABLE("SoftwareTriggerResult(triggerIdentifier)", softwareTriggerResult, R"DOC(
[Eventbased] [Expert] returns the SoftwareTriggerCutResult, defined as reject (-1), accept (1), or noResult (0). 
If the trigger identifier is not found, returns NaN.
 
For example:
 
.. code-block:: 
 
    SoftwareTriggerResult(filter 1_Estargt1_GeV_cluster_no_other_cluster_Estargt0.3_GeV)
 
which is equivalent to
 
.. code-block::
 
    SoftwareTriggerResult(software_trigger_cut&filter&1_Estargt1_GeV_cluster_no_other_cluster_Estargt0.3_GeV)
 
 
.. warning:: the meanings of these change depending if using trigger or the skim stage, hence expert.
 
.. seealso:: ``b2hlt_triggers`` for possible triggerIdentifiers.
 
        )DOC");
 
    REGISTER_VARIABLE("SoftwareTriggerResultNonPrescaled(triggerIdentifier)", softwareTriggerResultNonPrescaled,
                      "[Eventbased] [Expert] returns the SoftwareTriggerCutResult, "
                      "if this trigger would not be prescaled."
                      "Please note, this is not the final HLT decision! "
                      "It is defined as reject (-1), accept (1), or noResult (0). Note "
                      "that the meanings of these change depending if using trigger "
                      "or the skim stage, hence expert."
                      "If the trigger identifier is not found, returns NaN.");
    REGISTER_VARIABLE("HighLevelTrigger", passesAnyHighLevelTrigger,
                      "[Eventbased] 1.0 if event passes the HLT trigger, 0.0 if not");
    REGISTER_VARIABLE("SoftwareTriggerPrescaling(triggerIdentifier)", softwareTriggerPrescaling,
                      "[Eventbased] return the prescaling for the specific software trigger identifier. "
                      "Please note, this prescaling is taken from the currently setup database. It only corresponds "
                      "to the correct HLT prescale if you are using the online database!"
                      "If the trigger identifier is not found, returns NaN.");
    //-------------------------------------------------------------------------
  }
}