MuonPRDCacheCreator.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
#include "MuonPRDCacheCreator.h"
 
#include "AthViews/View.h"
#include "MuonDigitContainer/TgcDigit.h"

MuonPRDCacheCreator::MuonPRDCacheCreator(const std::string& name, ISvcLocator* pSvcLocator) :
    IDCCacheCreatorBase(name, pSvcLocator),
    m_CscCacheKey(""),
    m_CscStripCacheKey(""),
    m_MdtCacheKey(""),
    m_RpcCacheKey(""),
    m_sTgcCacheKey(""),
    m_MmCacheKey(""),
    m_RpcCoinCacheKey("") {
    declareProperty("CscCacheKey", m_CscCacheKey);
    declareProperty("CscStripCacheKey", m_CscStripCacheKey);
    declareProperty("MdtCacheKey", m_MdtCacheKey);
    declareProperty("RpcCacheKey", m_RpcCacheKey);
    declareProperty("sTgcCacheKey", m_sTgcCacheKey);
    declareProperty("MmCacheKey", m_MmCacheKey);
    declareProperty("RpcCoinCacheKey", m_RpcCoinCacheKey);
    declareProperty("DisableViewWarning", m_disableWarning);
}
 
StatusCode MuonPRDCacheCreator::initialize() {
    ATH_CHECK(m_CscCacheKey.initialize(SG::AllowEmpty));
    ATH_CHECK(m_CscStripCacheKey.initialize(SG::AllowEmpty));
    ATH_CHECK(m_MdtCacheKey.initialize(SG::AllowEmpty));
    ATH_CHECK(m_RpcCacheKey.initialize(SG::AllowEmpty));
    ATH_CHECK(m_sTgcCacheKey.initialize(SG::AllowEmpty));
    ATH_CHECK(m_MmCacheKey.initialize(SG::AllowEmpty));
    ATH_CHECK(m_RpcCoinCacheKey.initialize(SG::AllowEmpty));
 
    // build the TGC PRD and Coin cache keys (multiple due to reading out multiple BC)
    const bool doTgcPrdCache = not m_tgcPrdCacheKeyStr.empty();
    if (doTgcPrdCache) {
        m_TgcCacheKeys.resize(TgcDigit::BC_NEXT + 1);
        for (int ibc = 0; ibc < TgcDigit::BC_NEXT + 1; ibc++) {
            const int bcTag = ibc + 1;
            std::ostringstream location;
            location << m_tgcPrdCacheKeyStr.value() << (bcTag == TgcDigit::BC_PREVIOUS ? "PriorBC" : "")
                     << (bcTag == TgcDigit::BC_NEXT ? "NextBC" : "") << (bcTag == (TgcDigit::BC_NEXT + 1) ? "AllBCs" : "");
            m_TgcCacheKeys.at(ibc) = location.str();
            ATH_MSG_INFO("Setting next TGC PRD Cache to " << location.str());
        }
    }
    const bool doTgcCoinCache = not m_tgcCoinCacheKeyStr.empty();
    if (doTgcCoinCache) {
        m_TgcCoinCacheKeys.resize(TgcDigit::BC_NEXTNEXT);
        for (int ibc = 0; ibc < TgcDigit::BC_NEXTNEXT; ibc++) {
            const int bcTag = ibc + 1;
            std::ostringstream location;
            location << m_tgcCoinCacheKeyStr.value() << (bcTag == TgcDigit::BC_PREVIOUS ? "PriorBC" : "")
                     << (bcTag == TgcDigit::BC_NEXT ? "NextBC" : "") << (bcTag == TgcDigit::BC_NEXTNEXT ? "NextNextBC" : "");
            m_TgcCoinCacheKeys.at(ibc) = location.str();
            ATH_MSG_INFO("Setting next TGC Coin Cache to " << location.str());
        }  // BC loop
    }
    ATH_CHECK(m_TgcCacheKeys.initialize(doTgcPrdCache));
    ATH_CHECK(m_TgcCoinCacheKeys.initialize(doTgcCoinCache));
 
    ATH_CHECK(m_idHelperSvc.retrieve());
 
    // Check we have the tools to allow us to setup cache
    if (!m_idHelperSvc->hasCSC() && !m_CscCacheKey.key().empty()) {
        ATH_MSG_WARNING("CSC ID Helper is not available and CSC PRD cache was requested - This will not be created");
    }
    if (!m_idHelperSvc->hasMDT() && !m_MdtCacheKey.key().empty()) {
        ATH_MSG_WARNING("MDT ID Helper is not available and MDT PRD cache was requested - This will not be created");
    }
    if (!m_idHelperSvc->hasRPC() && !m_RpcCacheKey.key().empty()) {
        ATH_MSG_WARNING("RPC ID Helper is not available and RPC PRD cache was requested - This will not be created");
    }
    if (!m_idHelperSvc->hasTGC() && (doTgcPrdCache || doTgcCoinCache)) {
        ATH_MSG_WARNING("TGC ID Helper is not available and TGC cache was requested - This will not be created");
    }
    if (!m_idHelperSvc->hasSTGC() && !m_sTgcCacheKey.key().empty()) {
        ATH_MSG_WARNING("STGC ID Helper is not available and STGC PRD cache was requested - This will not be created");
    }
    if (!m_idHelperSvc->hasMM() && !m_MmCacheKey.key().empty()) {
        ATH_MSG_WARNING("MM ID Helper is not available and MM PRD cache was requested - This will not be created");
    }
    if (m_disableWarning) m_disableWarningCheck.test_and_set(std::memory_order_relaxed);
    return StatusCode::SUCCESS;
}
 
StatusCode MuonPRDCacheCreator::execute(const EventContext& ctx) const {
    ATH_CHECK(checkInsideViewOnce(ctx));
 
    // Create all the cache containers (if the tools are available)
    // CSC
    if (m_idHelperSvc->hasCSC()) {
        ATH_CHECK(createContainer(m_CscCacheKey, m_idHelperSvc->cscIdHelper().module_hash_max(), ctx));
        ATH_CHECK(createContainer(m_CscStripCacheKey, m_idHelperSvc->cscIdHelper().module_hash_max(), ctx));
    }
 
    // MDT
    if (m_idHelperSvc->hasMDT()) {
        auto maxHashMDTs = m_idHelperSvc->mdtIdHelper().stationNameIndex("BME") != -1
                               ? m_idHelperSvc->mdtIdHelper().detectorElement_hash_max()
                               : m_idHelperSvc->mdtIdHelper().module_hash_max();
        ATH_CHECK(createContainer(m_MdtCacheKey, maxHashMDTs, ctx));
    }
 
    // RPC
    if (m_idHelperSvc->hasRPC()) {
        ATH_CHECK(createContainer(m_RpcCacheKey, m_idHelperSvc->rpcIdHelper().module_hash_max(), ctx));
        ATH_CHECK(createContainer(m_RpcCoinCacheKey, m_idHelperSvc->rpcIdHelper().module_hash_max(), ctx));
    }
 
    // TGC
    if (m_idHelperSvc->hasTGC()) {
        for (const auto& tgcCacheKey : m_TgcCacheKeys) {
            ATH_CHECK(createContainer(tgcCacheKey, m_idHelperSvc->tgcIdHelper().module_hash_max(), ctx));
        }
        for (const auto& tgcCoinCacheKey : m_TgcCoinCacheKeys) {
            ATH_CHECK(createContainer(tgcCoinCacheKey, m_idHelperSvc->tgcIdHelper().module_hash_max(), ctx));
        }
    }
 
    // NSW STGC
    if (m_idHelperSvc->hasSTGC()) { ATH_CHECK(createContainer(m_sTgcCacheKey, m_idHelperSvc->stgcIdHelper().module_hash_max(), ctx)); }
 
    // NSW MM
    if (m_idHelperSvc->hasMM()) { ATH_CHECK(createContainer(m_MmCacheKey, m_idHelperSvc->mmIdHelper().module_hash_max(), ctx)); }
 
    return StatusCode::SUCCESS;
}