EnhancedBiasWeightCompAlg.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef TRIG_ENHANCEDBIASWEIGHTCOMPALG_H
#define TRIG_ENHANCEDBIASWEIGHTCOMPALG_H
 
#include "AthenaBaseComps/AthReentrantAlgorithm.h"
#include "TrigCompositeUtils/TrigCompositeUtils.h"
#include "TrigConfData/HLTMenu.h"
#include "TrigConfData/HLTPrescalesSet.h"
#include "TrigConfData/L1PrescalesSet.h"
#include "xAODTrigger/TrigCompositeContainer.h"
#include "TrigDecisionTool/TrigDecisionTool.h"
 
#include <vector>
#include <string>
 
#include <tbb/concurrent_hash_map.h>
#include <tbb/concurrent_unordered_map.h>
#include <tbb/concurrent_vector.h>

 
class EnhancedBiasWeightCompAlg : public AthReentrantAlgorithm {
  public:
 
    EnhancedBiasWeightCompAlg(const std::string& name, ISvcLocator* svcLoc);
 
    virtual StatusCode initialize() override;
    virtual StatusCode start() override;
    virtual StatusCode execute(const EventContext& context) const override;
    virtual StatusCode stop() override;
 
  private:
 
    class EBChainInfo {
      public:
        EBChainInfo(const HLT::Identifier & id) : m_id(id), m_totalPrescale(1.0) {}
 
        std::string getName() const {return m_id.name();}
        TrigCompositeUtils::DecisionID getId() const {return m_id;}
        double getTotalPrescale() const {return m_totalPrescale;}
        bool getIsRandom() const {return (getName().find("HLT_noalg_L1RD") != std::string::npos);}
        bool getIsNoPS() const {return (getName().find("noPS") != std::string::npos);}
        bool getIsDisabled() const {return (m_totalPrescale <= 0);}
 
        void setTotalPrescale(const double& prescale) {m_totalPrescale = prescale;}
 
      private:
        HLT::Identifier m_id;

        double m_totalPrescale;
    };

    struct EBResult {
      double weight {1.0};
      uint8_t isUnbiased {};
    };
 
    SG::ReadHandleKey<TrigConf::HLTMenu> m_HLTMenuKey{this, "HLTTriggerMenu", "DetectorStore+HLTTriggerMenu", "HLT Menu"};
    PublicToolHandle<Trig::TrigDecisionTool> m_tdt{this, "TrigDecisionTool", "Trig::TrigDecisionTool/TrigDecisionTool", "TrigDecisionTool"};
    SG::ReadHandleKey<TrigCompositeUtils::DecisionContainer> m_finalDecisionKey{ this, "FinalDecisionKey", "", "Final stage of all decisions" };
 
    SG::ReadCondHandleKey<TrigConf::HLTPrescalesSet> m_HLTPrescaleSetInputKey {this, "HLTPrescales", "HLTPrescales", "HLT prescales set"};
    SG::ReadCondHandleKey<TrigConf::L1PrescalesSet> m_L1PrescaleSetInputKey {this, "L1Prescales", "L1Prescales", "L1 prescales set"};
 
    Gaudi::Property<std::map<std::string, std::vector<std::string>>> m_chainToHLTSeed {this, "ChainToItemMap", {}, "Map for HLT seeded chains" };

    EBResult calculateEBWeight(const std::vector<EBChainInfo>& EBChains) const;

    StatusCode fillTotalPrescaleForChains(const EventContext& context, std::vector<EBChainInfo>& EBChains) const;

    std::vector<EBChainInfo> getPassedEBChains() const;

    bool checkIfTriggeredByRandomChain(const std::vector<EBChainInfo>& EBChain) const;

    std::vector<std::string> parseItems(const std::string& itemStr);

    std::vector<HLT::Identifier> m_EBChainIds;

    std::map<HLT::Identifier, std::vector<std::string>> m_EBChainIdToItem;

    mutable std::string m_outputFilename ATLAS_THREAD_SAFE;

    mutable tbb::concurrent_vector<std::pair<double, bool>> m_ebWeights ATLAS_THREAD_SAFE;

    mutable tbb::concurrent_unordered_map<int, double> m_eventToWeight ATLAS_THREAD_SAFE;
 
};
 
#endif //TRIG_ENHANCEDBIASWEIGHTCOMPALG_H