CalibrationDbIOTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef CalibrationDbIOTool_H
#define CalibrationDbIOTool_H
 
class MdtIdHelper;
namespace MuonGM {
    class MuonDetectorManager;
}
 
// MuonCalibStandAloneBase
#include <array>
#include <memory>
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "MdtCalibUtils/RtDataFromFile.h"
#include "MuonCalibDbOperations/CalibDbConnection.h"
#include "MuonCalibDbOperations/CalibHeadOperations.h"
#include "MuonCalibMath/SamplePoint.h"
#include "MuonCalibStandAloneBase/ICalibrationIOTool.h"
 
namespace MuonCalib {
 
    // class CalibDbConnection;
    class IRtRelation;
 
    class CalibrationDbIOTool : public AthAlgTool, virtual public ICalibrationIOTool {
    public:
        CalibrationDbIOTool(const std::string &t, const std::string &n, const IInterface *p);
        virtual ~CalibrationDbIOTool();
        StatusCode initialize() override;
        StatusCode WriteT0(const MdtTubeFitContainer *t0_output, const NtupleStationId &station_id, int iov_start, int iov_end) override;
        StatusCode LoadT0(std::map<NtupleStationId, MdtStationT0Container *> &t0s, int /*iov_id*/) override;
        StatusCode 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*/) override;
        StatusCode LoadRt(std::map<NtupleStationId, IRtRelation *> &rts, std::map<NtupleStationId, IRtResolution *> &res,
                          int /*iov_id*/) override;
 
    private:
        std::string m_calib_dir;
        bool m_use_validated_t0, m_use_validated_rt = false;
        // url of oracle data-base
        std::string m_db_ConnectionString;
        std::string m_writer_connection_string;
        std::string m_db_WorkingSchema;
        std::string m_reader_account, m_reader_password;
        std::string m_writer_account, m_writer_password;
        // username and password - job options
        int m_max_rt_points;
        int m_headid;
        std::string m_sitename;
        std::vector<MuonCalib::NtupleStationId> m_region_ids;
        std::unique_ptr<CalibDbConnection> m_connection;
        std::array<std::unique_ptr<CalibDbConnection>, 2> m_data_connections;
        std::unique_ptr<CalibHeadOperations> m_head_ops;
        void fillRtPoints(const std::shared_ptr<const IRtRelation>& rt_relation, std::vector<SamplePoint> &points);
        void fillResPoints(const std::shared_ptr<const IRtResolution>& rt_resolution, std::vector<SamplePoint> &points);
        CalibDbConnection *get_connection(int write);
    };
 
}  // namespace MuonCalib
#endif