MuonPhaseII/MuonConditions/MuonCondAlgR4/src/MdtCalibDbAlg.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/

 
#ifndef MUONCALIBR4_MDTCALIBDBALGR4_H
#define MUONCALIBR4_MDTCALIBDBALGR4_H
 
#include "AthenaBaseComps/AthCondAlgorithm.h"
 
#include "MuonIdHelpers/IMuonIdHelperSvc.h"
#include "MdtCalibData/MdtCalibDataContainer.h"
#include "MuonReadoutGeometryR4/MuonDetectorManager.h"
#include "AthenaPoolUtilities/CondAttrListCollection.h"
 
#include "StoreGate/ReadCondHandleKey.h"
#include "StoreGate/WriteCondHandleKey.h"
 
#include <nlohmann/json.hpp>
 
 
class TTree;
 
namespace MuonCalibR4 {
    class MdtCalibDbAlg : public AthCondAlgorithm {
        public:
            
            using AthCondAlgorithm::AthCondAlgorithm;
            virtual ~MdtCalibDbAlg() = default;
 
            virtual StatusCode initialize() override;
            virtual StatusCode execute(const EventContext& ctx) const override;
 
        private:
 
            using CorrectionPtr = MuonCalib::MdtCalibDataContainer::CorrectionPtr;
            using RtRelationPtr = MuonCalib::MdtCalibDataContainer::RtRelationPtr;
            using TubeContainerPtr = MuonCalib::MdtCalibDataContainer::TubeContainerPtr;
 
            /***   @brief Translates a r-t JSON payload into Mdt calibration constants
               *   @param rtBlob: JSON payload to translated
               *   @param outContainer: Container into which the Rt relations are appended */
            StatusCode parseRtPayload(const nlohmann::json& rtBlob,
                                      MuonCalib::MdtCalibDataContainer& outContainer) const;

            StatusCode parseRtPayload(TTree& rtTree,
                                      MuonCalib::MdtCalibDataContainer& outContainer) const;
            MuonCalib::IRtRelationPtr makeRt(const std::string& rtType,
                                             const std::vector<double>& pars) const;
            MuonCalib::ITrRelationPtr makeTr(const std::string& trType,
                                             const std::vector<double>& pars) const;
            MuonCalib::IRtResolutionPtr makeReso(const std::string& resoType,
                                                 const std::vector<double>& pars,
                                                  MuonCalib::IRtRelationPtr rt) const;
            StatusCode parseT0Payload(const nlohmann::json& t0Blob,
                                      MuonCalib::MdtCalibDataContainer& outContainer) const;

            StatusCode parseT0Payload(TTree& t0Tree,
                                      MuonCalib::MdtCalibDataContainer& outContainer) const;
 
 
            ServiceHandle<Muon::IMuonIdHelperSvc> m_idHelperSvc{this, "MuonIdHelperSvc", "Muon::MuonIdHelperSvc/MuonIdHelperSvc"};
            SG::WriteCondHandleKey<MuonCalib::MdtCalibDataContainer> m_writeKey{this, "WriteKey",  "MdtCalibConstants",
                                                                                            "Conditions object containing the calibrations"};   
                                                                
            Gaudi::Property<std::string> m_rtJSON{this, "RtJSON", ""};
            Gaudi::Property<std::string> m_t0JSON{this, "TubeT0JSON", ""};
 
            Gaudi::Property<std::string> m_rtRootFile{this, "RtROOT", "", 
                                                      "Path to an external root file to read the Rt relation from"};
            Gaudi::Property<std::string> m_t0RootFile{this, "TubeT0ROOT", "", 
                                                      "Path to an external root file to read the T0 constants from"};
 
            Gaudi::Property<std::string> m_rtTreeName{this, "RtTreeName", "RtCalibConstants"};
            Gaudi::Property<std::string> m_t0TreeName{this, "T0TreeName", "T0CalibConstants"};
 
            SG::ReadCondHandleKey<CondAttrListCollection> m_readKeyRt{this, "ReadKeyRt", "/MDT/RTJSON", "DB folder containing the RT calibrations"};
            SG::ReadCondHandleKey<CondAttrListCollection> m_readKeyTube{this, "ReadKeyTube", "/MDT/T0JSON", "DB folder containing the tube constants"};
            Gaudi::Property<std::string> m_dbPayloadType {this, "dbPayloadType","TTree", "specify the format of the payload in the database. Can be TTree or JSON"};
 
            Gaudi::Property<double> m_prop_beta{this, "PropagationSpeedBeta", 1., "Speed of the signal propagation"};
            Gaudi::Property<unsigned> m_t0CalibPrec{this, "T0CalibPrecision", 4};
 
            Gaudi::Property<bool> m_create_b_field_function{this, "CreateBFieldFunctions", false,
                "If set to true, the B-field correction functions are initialized for each rt-relation that is loaded."};
 
            Gaudi::Property<bool> m_createSlewingFunction{this, "CreateSlewingFunctions", false,
                "If set to true, the slewing correction functions are initialized for each rt-relation that is loaded."};
 
 
    };
}
#endif