MdtCalibDbAlgTest.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
#include "MdtCalibDbAlgTest.h"
 
#include "GaudiKernel/PhysicalConstants.h"
#include <MuonReadoutGeometryR4/MdtReadoutElement.h>
#include "MdtCalibData/MdtFullCalibData.h"
#include "GaudiKernel/ITHistSvc.h"
#include "TH2D.h"
#include "TCanvas.h"
 
using namespace MuonValR4;
 
StatusCode MdtCalibDbAlgTest::initialize() {
    ATH_MSG_VERBOSE("Initializing MdtCalibDbAlgTest");
    ATH_CHECK(m_MdtKey.initialize());
    ATH_CHECK(m_geoCtxKey.initialize());
    ATH_CHECK(m_calibrationTool.retrieve());
    ATH_CHECK(m_idHelperSvc.retrieve());
    ATH_CHECK(m_tree.init(this));
    ATH_CHECK(book(TH2D("DriftRadiusVsTime", "DriftRadiusVsTime", 680, 0., 680., 100, 0., 15.), "MdtCalibDbAlgTest", "MdtCalibDbAlgTest"));
    ATH_CHECK(book(TH2D("DriftVelocityVsTime", "DriftVelocityVsTime", 680, 0., 680., 140, -0.1, 0.6), "MdtCalibDbAlgTest", "MdtCalibDbAlgTest"));
    ATH_CHECK(book(TH2D("DriftTimeVsRadius", "DriftTimeVsRadius", 100, 0., 15., 680, 0., 680.), "MdtCalibDbAlgTest", "MdtCalibDbAlgTest"));
    ATH_CHECK(book(TH2D("DriftTimeClosure", "DriftTimeClosure", 680, 0., 680., 680, 0., 680.), "MdtCalibDbAlgTest", "MdtCalibDbAlgTest"));
 
    return StatusCode::SUCCESS;
}
 
StatusCode MdtCalibDbAlgTest::finalize() {
    ATH_MSG_VERBOSE("Finalizing MdtCalibDbAlgTest");
    ATH_CHECK(m_tree.write());
    return StatusCode::SUCCESS;
}
 
StatusCode MdtCalibDbAlgTest::execute(const EventContext& ctx) {
        
    const xAOD::MdtDriftCircleContainer* mdtContainer{nullptr};
    const ActsTrk::GeometryContext* geoCtx{nullptr};
 
    ATH_CHECK(SG::get(geoCtx, m_geoCtxKey, ctx));
    ATH_CHECK(SG::get(mdtContainer, m_MdtKey, ctx));
    
    constexpr double inversePropSpeed = 1. / Gaudi::Units::c_light;
    for(const xAOD::MdtDriftCircle* mdt : *mdtContainer) {
        if (mdt->status() != Muon::MdtDriftCircleStatus::MdtStatusDriftTime){
            continue;
        }
        const MuonGMR4::MdtReadoutElement* mdtRE = mdt->readoutElement();
        const Amg::Vector3D mdtGlobalTubePos = mdtRE->globalTubePos(*geoCtx, mdt->measurementHash());
        const float tdcAdj = IMdtCalibrationTool::tdcBinSize * mdt->tdc()  - inversePropSpeed * (mdtGlobalTubePos.norm() - 0.5 * mdtRE->activeTubeLength(mdt->measurementHash()));
        m_out_tdcAdj = tdcAdj;
        m_out_tdc = mdt->tdc();
        m_out_driftRadius = mdt->driftRadius();
        const float driftV = m_calibrationTool->getCalibConstants(ctx, mdt->identify())->rtRelation->rt()->driftVelocity(tdcAdj);
        std::optional<double> driftTime = m_calibrationTool->getCalibConstants(ctx, mdt->identify())->rtRelation->tr()->driftTime(mdt->driftRadius());
        const float f_driftTime = static_cast<float>(driftTime.value_or(0.));
        m_out_driftVelocity = driftV;
 
        m_out_stIndex = mdtRE->stationName();
        m_out_stEta = mdtRE->stationEta();
        m_out_stPhi = mdtRE->stationPhi();
        m_out_ml = mdtRE->multilayer();
        m_out_tl = mdt->tubeLayer();
        m_out_tube = mdt->driftTube();
        m_out_globalPos = mdtGlobalTubePos.norm();
        m_out_globalPosX = mdtGlobalTubePos.x();
        m_out_globalPosY = mdtGlobalTubePos.y();
        m_out_globalPosZ = mdtGlobalTubePos.z();
        m_out_tubeLength = mdtRE->activeTubeLength(mdt->measurementHash());
        m_out_driftTime = f_driftTime;
        hist("DriftRadiusVsTime")->Fill(tdcAdj, mdt->driftRadius());
        hist("DriftVelocityVsTime")->Fill(tdcAdj, driftV);
        hist("DriftTimeVsRadius")->Fill(mdt->driftRadius(), f_driftTime);
        hist("DriftTimeClosure")->Fill(tdcAdj, f_driftTime);
        m_tree.fill(ctx);
    }
 
    return StatusCode::SUCCESS;
}