BmnATestDetector.cxx

Go to the documentation of this file.

/*
    BM@N alignment routine
    BM@N experiment at NICA complex, JINR, 2025
 
    Department: Math & Soft Group of HEP lab
    Author:     Igor Polev, polev@jinr.ru
 
    BmnATestDetector implementation
*/
 
#include "BmnATestDetector.h"
 
#include "fairlogger/Logger.h"
 
#include <fstream>
#include <nlohmann/json.hpp>
 
BmnATestDetector::BmnATestDetector(Double_t detectorStep, const char* dumpPath)
    : BmnSimpleDetModel(BMN_TEST_MODULE_COUNT)
{
    LOG(info) << "  > Generating " << BMN_TEST_MODULE_COUNT << " test detectors planes";
 
    Double_t nominalZ = detectorStep;
    for (Int_t i = 0; i < BMN_TEST_MODULE_COUNT; i++) {   // first detector is fixed
        for (Int_t j = iX; j <= iZ; j++) {
            if (i > 0)
                fMisalignments[i](j) = BMN_RND.Gaus(0.0, BMN_ALIGN_PARAM_TOLERENCE);
            // if (i == 2 && j == iX)
            //     fMisalignments[i](j) = 0.1;
        }
        fNominalZ[i] = nominalZ;
        nominalZ += detectorStep;
    }
 
    if (dumpPath)
        Save(dumpPath);
    RebuildDetectors();
}
 
Bool_t BmnATestDetector::Save(const char* path) const
{
    LOG(info) << "    Saving test detectors planes to " << path;
    std::ofstream ofs(path);
    if (!ofs.good()) {
        LOG(error) << "  <!> failed to open JSON file " << path;
        return kFALSE;
    }
 
    nlohmann::json j;
    j["nominal_z"] = fNominalZ;
    j["misalignments"] = nlohmann::json::array();
    for (Int_t i = 0; i < BMN_TEST_MODULE_COUNT; i++) {
        j["misalignments"].push_back({fMisalignments[i](iX), fMisalignments[i](iY), fMisalignments[i](iZ)});
    }
 
    ofs << j.dump(2);
    return kTRUE;
}
 
Bool_t BmnATestDetector::Load(const char* path)
{
    LOG(info) << "  > Loading test detectors planes from " << path;
    std::ifstream ifs(path);
    if (!ifs.good()) {
        LOG(error) << "  <!> failed to open JSON file " << path;
        return kFALSE;
    }
    nlohmann::json j;
    ifs >> j;
 
    const auto& jz = j["nominal_z"];
    const auto& jm = j["misalignments"];
 
    if (jz.size() != jm.size()) {
        LOG(error) << "  <!> inconsistent number of detector planes in JSON file " << path;
        return kFALSE;
    }
 
    for (Int_t i = 0; i < BMN_TEST_MODULE_COUNT; i++) {
        fNominalZ[i] = jz[i].get<double>();
        fMisalignments[i] =
            SVectGL(jm[i][static_cast<Int_t>(iX)].get<double>(), jm[i][static_cast<Int_t>(iY)].get<double>(),
                    jm[i][static_cast<Int_t>(iZ)].get<double>());
    }
 
    RebuildDetectors();
    return kTRUE;
}
 
void BmnATestDetector::RebuildDetectors()
{
    for (Int_t i = 0; i < BMN_TEST_MODULE_COUNT; i++)
        fDetectors[i] = BmnATestDetectorPlane(fNominalZ[i], fMisalignments[i], i);
}