NtupleStationId.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
#include "MuonCalibStandAloneBase/NtupleStationId.h"
 
#include "AthenaKernel/getMessageSvc.h"
#include "GaudiKernel/MsgStream.h"
#include "Identifier/IdContext.h"
#include "MuonIdHelpers/MdtIdHelper.h"
#include "MuonReadoutGeometry/MdtReadoutElement.h"
#include "MuonReadoutGeometry/MuonDetectorManager.h"
 
namespace MuonCalib {
 
    bool NtupleStationId::Initialize(const std::string& station, const int& eta, const int& phi, const int& ml, const int& author) {
        MuonFixedId id;
        if (station == "ANY") {
            m_station = -1;
        } else {
            m_station = id.stationStringToFixedStationNumber(station);
            if (m_station < 0) return false;
        }
        m_eta = eta;
        m_phi = phi;
        m_ml = ml;
        m_author = author;
        m_geom_ok = false;
        return true;
    }
 
    void NtupleStationId::SetStation(const std::string& station) {
        MuonFixedId id;
        m_station = id.stationStringToFixedStationNumber(station);
    }
 
    bool NtupleStationId::InitializeGeometry(const MdtIdHelper& mdtIdHelper, const MuonGM::MuonDetectorManager* detMgr) {
        MuonFixedId fid;
        MsgStream log(Athena::getMessageSvc(), "NtupleStationId");
        if (m_station == -1 || m_phi < 0 || m_eta == -99) {
            log << MSG::WARNING << "NtupleStationId::InitializeGeometry: Cannot initialize geometry for multi station id" << endmsg;
            m_geom_ok = false;
            return false;
        }
        Identifier id(mdtIdHelper.elementID(fid.stationNumberToFixedStationString(m_station), m_eta, m_phi));
        m_n_ml = mdtIdHelper.numberOfMultilayers(id);
        // loop on multilayers
        for (int i = 0; i < m_n_ml; i++) {
            const MuonGM::MdtReadoutElement* detEl_ml = detMgr->getMdtReadoutElement(mdtIdHelper.channelID(id, 1 + i, 1, 1));
            m_layer_min[i] = 1;
            if (!detEl_ml) {
                log << MSG::WARNING << regionId() << " ml " << i << "does not exist in current geometry" << endmsg;
                return false;
            }
            m_layer_max[i] = detEl_ml->getNLayers();
            m_n_layer[i] = m_layer_max[i] - m_layer_min[i] + 1;
            m_tube_min[i] = 1;
            m_tube_max[i] = detEl_ml->getNtubesperlayer();
            m_n_tubes[i] = m_tube_max[i] - m_tube_min[i] + 1;
        }
        MdtBasicRegionHash hash;
        IdContext idCont = mdtIdHelper.module_context();
        mdtIdHelper.get_hash(id, hash, &idCont);
        m_region_hash = static_cast<int>(hash);
        m_geom_ok = true;
        return true;
    }
 
    std::string NtupleStationId::regionId() const {
        MuonFixedId id;
        std::ostringstream id_stream;
        if (m_station < 0) {
            id_stream << "ANY";
        } else {
            id_stream << id.stationNumberToFixedStationString(m_station);
        }
        id_stream << "_";
        if (m_phi >= 0) {
            id_stream << m_phi;
        } else {
            id_stream << "ANY";
        }
        id_stream << "_";
        if (m_eta != -99) {
            id_stream << m_eta;
        } else {
            id_stream << "ANY";
        }
        if (m_ml > 0) { id_stream << "_" << m_ml; }
        return id_stream.str();
    }
 
    int NtupleStationId::FixedId() const {
        if (m_station < 0 || m_eta == -99 || m_phi < 0) return -1;
        MuonFixedId id;
        id.set_mdt();
        if (!id.setStationName(m_station)) return -1;
        if (!id.setStationEta(m_eta)) return -1;
        if (!id.setStationPhi(m_phi)) return -1;
        return id.mdtChamberId().getIdInt();
    }
 
}  // namespace MuonCalib