CalibrationDbIOTool.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
// this
#include "MuonCalibDbOperations/CalibrationDbIOTool.h"
 
#include "MuonCalibDbOperations/CalibHeadOperations.h"
#include "MuonCalibDbOperations/CalibRtDbOperations.h"
#include "MuonCalibDbOperations/CalibT0DbOperations.h"
// gaudi
#include "GaudiKernel/MsgStream.h"
 
// MdtCalibUtils
#include "MdtCalibUtils/RtDataFromFile.h"
#include "MdtCalibUtils/TubeDataFromFile.h"
 
// MdtCalibData
#include "MdtCalibData/IRtResolution.h"
#include "MdtCalibData/MdtTubeFitContainer.h"
#include "MdtCalibData/RtChebyshev.h"
#include "MdtCalibData/RtFromPoints.h"
#include "MdtCalibData/RtRelationLookUp.h"
#include "MdtCalibData/RtResolutionFromPoints.h"
#include "MdtCalibData/RtResolutionLookUp.h"
#include "MdtCalibData/RtSpline.h"
 
// MdtCalibRt
#include "MdtCalibRt/RtCalibrationOutput.h"
 
// MuonCalibStandAloneBase
#include "MuonCalibStandAloneBase/MdtStationT0Container.h"
#include "MuonCalibStandAloneBase/NtupleStationId.h"
 
// MuonCalibMath
#include "MuonCalibMath/SamplePoint.h"
 
// c - c++
#include <dirent.h>
#include <sys/types.h>
 
#include "MuonCalibStandAloneBase/RegionSelectionSvc.h"
#include "fstream"
#include "iostream"
#include "string"
 
namespace MuonCalib {
    CalibrationDbIOTool::~CalibrationDbIOTool() = default;
    CalibrationDbIOTool::CalibrationDbIOTool(const std::string &t, const std::string &n, const IInterface *p) :
        AthAlgTool(t, n, p), m_db_ConnectionString(""), m_db_WorkingSchema(""), m_sitename("") {
        declareInterface<ICalibrationIOTool>(this);
        declareProperty("ConnectionString", m_db_ConnectionString);
        declareProperty("WriterConnectionString", m_writer_connection_string);
        declareProperty("WorkingSchema", m_db_WorkingSchema);
        m_headid = -1;
        declareProperty("HeadId", m_headid);
        declareProperty("SiteName", m_sitename);
        m_max_rt_points = 100;
        declareProperty("MaxRtPoints", m_max_rt_points);
        m_use_validated_t0 = true;
        declareProperty("UseValidaedT0", m_use_validated_t0);
        m_use_validated_t0 = false;
        declareProperty("UseValidaedRt", m_use_validated_rt);
        declareProperty("ReaderAccount", m_reader_account);
        declareProperty("ReaderPassword", m_reader_password);
        declareProperty("WriterAccount", m_writer_account);
        declareProperty("WriterPassword", m_writer_password);
    }
 
    StatusCode CalibrationDbIOTool::initialize() {
        ServiceHandle<RegionSelectionSvc> reg_sel_svc{"RegionSelectionSvc", name()};
        ATH_CHECK(reg_sel_svc.retrieve());
        m_region_ids = reg_sel_svc->GetStationsInRegions();
        ATH_MSG_INFO(" CalibrationDbIOTool::initialize() - number of selected regions: " << m_region_ids.size());
 
        ATH_MSG_INFO("open connection");
        m_connection = std::make_unique<CalibDbConnection>(m_db_ConnectionString, m_db_WorkingSchema);
        ATH_MSG_INFO(".");
        if (!m_reader_account.empty()) { m_connection->SetLogin(m_reader_account, m_reader_password); }
        if (!m_connection->OpenConnection()) {
            ATH_MSG_FATAL("Cannot open connection to database!");
            return StatusCode::FAILURE;
        }
        ATH_MSG_INFO("..");
        m_head_ops = std::make_unique<CalibHeadOperations>(*m_connection);
        if (m_headid < 0) {
            m_headid = m_head_ops->GetLatestHeadId();
            if (m_headid < 0) {
                ATH_MSG_FATAL("Cannot get latest head_id for site " << m_sitename);
                return StatusCode::FAILURE;
            }
        }
        return StatusCode::SUCCESS;
    }
 
    StatusCode CalibrationDbIOTool::WriteT0(const MdtTubeFitContainer *t0, const NtupleStationId &it, int /*iov_start*/, int /*iov_end*/) {
        ATH_MSG_INFO("Writing to database");
        if (!m_connection->OpenConnection()) {
            ATH_MSG_FATAL("Open Connection failed!");
            return StatusCode::FAILURE;
        } else {
            if (!get_connection(1)) { return StatusCode::FAILURE; }
            m_data_connections[1]->OpenConnection();
            CalibT0DbOperations t0_op(*m_data_connections[1]);
            std::vector<int> val(0);
            if (!t0_op.WriteT0Chamber(it, t0, val, m_headid, m_sitename)) {
                ATH_MSG_FATAL("Writing t0 failed!");
                return StatusCode::FAILURE;
            }
        }
        return StatusCode::SUCCESS;
    }
 
    StatusCode CalibrationDbIOTool::LoadT0(std::map<NtupleStationId, MdtStationT0Container *> &t0s, int /*iov_id*/) {
        if (!m_connection->OpenConnection()) {
            ATH_MSG_FATAL("Open Connection failed!");
            return StatusCode::FAILURE;
        } else {
            if (!get_connection(0)) { return StatusCode::FAILURE; }
            m_data_connections[0]->OpenConnection();
            CalibT0DbOperations t0_op(*m_data_connections[0]);
            t0s.clear();
 
            // loop on all the ids in the selected calibration region
            for (std::vector<MuonCalib::NtupleStationId>::iterator it = m_region_ids.begin(); it != m_region_ids.end(); ++it) {
                ATH_MSG_INFO("Reading t0s for region " << it->regionId());
                MdtStationT0Container *t0;
                t0 = t0_op.LoadT0Calibration(*it, m_headid, m_sitename);
                if (t0 != nullptr) t0s[*it] = t0;
            }
        }
        return StatusCode::SUCCESS;
    }
 
    StatusCode CalibrationDbIOTool::WriteRt(const RtCalibrationOutput *rt_relation, std::shared_ptr<const IRtResolution> resolution,
                                            const NtupleStationId &station_id, int /*iov_start*/, int /*iov_end*/, bool /*real_rt*/,
                                            bool /*real resolution*/) {
        std::vector<SamplePoint> points;
        // fill rt poits
        auto rtrelationrt = rt_relation->rt();
        fillRtPoints(rtrelationrt.get(), points);
        // fill resolution points
        fillResPoints(*resolution, points);
        // fill multilayer rt difference as pseudo-point
        if (rtrelationrt->hasTmaxDiff()) {
            SamplePoint point(rt_relation->rt()->GetTmaxDiff(), -99.9, 0.0);
            points.push_back(point);
        }
        if (!m_connection->OpenConnection()) {
            ATH_MSG_FATAL("Open Connection failed!");
            return StatusCode::FAILURE;
        } else {
            if (!get_connection(1)) { return StatusCode::FAILURE; }
            m_data_connections[1]->OpenConnection();
            CalibRtDbOperations rt_op(*m_data_connections[1]);
            if (!rt_op.WriteUpdateRt(station_id, m_headid, m_sitename, points, 0, rt_relation->fullInfo().get())) {
                ATH_MSG_FATAL("Writing rt failed!");
                return StatusCode::FAILURE;
            }
        }
        return StatusCode::SUCCESS;
    }
 
    StatusCode CalibrationDbIOTool::LoadRt(std::map<NtupleStationId, IRtRelation *> &rts, std::map<NtupleStationId, IRtResolution *> &res,
                                           int /*iov_id*/) {
        if (!m_connection->OpenConnection()) {
            ATH_MSG_FATAL("Open Connection failed!");
            return StatusCode::FAILURE;
        } else {
            if (!get_connection(0)) { return StatusCode::FAILURE; }
            m_data_connections[0]->OpenConnection();
            CalibRtDbOperations rt_op(*m_data_connections[0]);
            // loop on all the ids in the selected calibration region
            RtFromPoints rt_from_points;
            RtResolutionFromPoints res_from_points;
            for (std::vector<MuonCalib::NtupleStationId>::iterator it = m_region_ids.begin(); it != m_region_ids.end(); ++it) {
                std::vector<SamplePoint> in_points;
                RtFullInfo full_info;
                if (!rt_op.LoadRt(*it, m_headid, m_use_validated_rt, m_sitename, in_points, &full_info)) continue;
                ATH_MSG_INFO("Reading rts for region " << it->regionId());
                std::vector<SamplePoint> outpoints;
                float tmax_diff(-9e9);
                for (auto & in_point : in_points) {
                    if (in_point.x2() < -99) {
                        tmax_diff = in_point.x1();
                        continue;
                    }
                    SamplePoint outpoint(in_point.x1(), in_point.x2(), 1.0);
                    outpoints.push_back(outpoint);
                }
                rts[*it] = rt_from_points.getRtRelationLookUp(outpoints).release();
                if (tmax_diff > -8e8) { rts[*it]->SetTmaxDiff(tmax_diff); }
                outpoints.clear();
                for (auto & in_point : in_points) {
                    if (in_point.x2() < -99) { continue; }
                    SamplePoint outpoint(in_point.x1(), in_point.error(), 1.0);
                    outpoints.push_back(outpoint);
                }
                res[*it] = new RtResolutionLookUp(res_from_points.getRtResolutionLookUp(outpoints));
            }
        }
        return StatusCode::SUCCESS;
    }
 
    void CalibrationDbIOTool::fillRtPoints(const IRtRelation* rt, std::vector<SamplePoint> &points) {
        points.clear();
        // for rt relations based on support points, write points
        auto rt_lookup = dynamic_cast<const RtRelationLookUp*>(rt);
        const CalibFunc::ParVec &rt_param = rt->parameters();
        if (rt_lookup && static_cast<int>(rt_lookup->nPar() - 2) <= m_max_rt_points) {
            double t_min(rt_param[0]);
            double bin_size = rt_param[1];
            unsigned int nb_points(rt_lookup->nPar() - 2);
            for (unsigned int k = 0; k < nb_points; k++) {
                double radius(rt_param[k + 2]);
                if (std::isnan(radius)) {
                    ATH_MSG_WARNING("Skipping NAN");
                    continue;
                }
                SamplePoint point(t_min + bin_size * k, rt_param[k + 2], -1);
                points.push_back(point);
            }
            return;
        }
        auto rt_spline = dynamic_cast<const RtSpline*>(rt);
        if (rt_spline && static_cast<int>(rt_param.size() / 2) <= m_max_rt_points) {
            for (unsigned int i = 0; i < rt_param.size(); i += 2) {
                SamplePoint point(rt_param[i], rt_param[i + 1], -1);
                points.push_back(point);
            }
            return;
        }
        // for all others, or if the number of support points is too high, calculate new support points
        for (int i = 0; i < m_max_rt_points; i++) {
            double t = rt->tLower() + i * ((rt->tUpper() - rt->tLower()) / (m_max_rt_points - 1));
            SamplePoint point(t, rt->radius(t), -1.0);
        }
    }
 
    void CalibrationDbIOTool::fillResPoints(const IRtResolution& rt_resolution, std::vector<SamplePoint> &points) {
        for (auto & point : points) { point.set_error(rt_resolution.resolution(point.x1())); }
    }
 
    CalibDbConnection *CalibrationDbIOTool::get_connection(int write) {
        if (m_data_connections[write]) { return m_data_connections[write].get(); }
        m_data_connections[write] = std::unique_ptr<CalibDbConnection>{m_head_ops->GetDataConnection(
            m_headid, static_cast<bool>(write), m_writer_connection_string, m_writer_account, m_writer_password)};
        return m_data_connections[write].get();
    }
 
}  // namespace MuonCalib