align.C

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
 
    ROOT alignment macro
*/
 
// ==============================================================
// COMMON HEADER SECTION BEGIN : INSERT FROM COMPILE VERSION HERE
 
#undef BMN_TEST_ALGORITHM   // task type selection
 
// headers below are for IDE syntax to work and they will be
// turned off during execution in ROOT macro mode
#ifndef __CLING__
#include "BmnMeasureModel.h"
#include "BmnRun8cgDetModel.h"
#ifdef BMN_TEST_ALGORITHM
#include "BmnATestDetector.h"
#include "BmnATestHit.h"
#include "BmnAlignerTest.h"
 
#include <filesystem>
#else   // non-test mode
#include "BmnAligner.h"
#endif
#endif
 
#include "fairlogger/Logger.h"
 
#ifdef BMN_TEST_ALGORITHM
#define TEST_DETECTOR_STEP 20.0
#define TEST_TRACK_COUNT 30000
#define TEST_DETECTORS_FILE "cfg/test_detectors.json"
#define TEST_TRACKS_FILE "cfg/test_tracks.json"
#endif
#define FILE_LIST_PATH "cfg/dst7466noa.txt"
#define SOLUTION_PATH "cfg/new_solution.json"
 
constexpr const char *START_APPROX_PATH{nullptr},   // "cfg/fix_last.json"
    *CONSTRAINTS_PATH{nullptr};                     // "cfg/test_constraints.json"
 
// COMMON HEADER SECTION END : INSERT FROM COMPILE VERSION UTIL HERE
// =================================================================
 
int align(int argc = 0, char** argv = nullptr)
{
    // =======================================================
    // COMMON SECTION BEGIN : INSERT FROM COMPILE VERSION HERE
 
    fair::Logger::SetConsoleColor(true);
    fair::Logger::SetConsoleSeverity("info");
    // fair::Logger::InitFileSink("debug", "bmn-align.log");
 
    BmnMeasureModel measureModel;
 
#ifdef BMN_TEST_ALGORITHM
    // TEST MODE ===============================================================================
 
    // generate or load test detectors
    BmnATestDetector detModel{std::filesystem::exists(TEST_DETECTORS_FILE)
                                  ? BmnATestDetector(TEST_DETECTORS_FILE)
                                  : BmnATestDetector(TEST_DETECTOR_STEP   //,
                                                                          // TEST_DETECTORS_FILE
                                                     )};
 
    // init aligner: dynamic object is mandatory for ROOT macro mode compatibility
    BmnAlignerTest pAligner = new BmnAlignerTest(&measureModel, &detModel, TEST_TRACK_COUNT, constraintsPath, 16);
 
#else   // not BMN_TEST_ALGORITHM
    // REAL DATA MODE ==========================================================================
 
    BmnRun8cgDetModel detModel;
    detModel.Init();
 
    // init aligner: dynamic object is mandatory for ROOT macro mode compatibility
    BmnAligner<>* pAligner = new BmnAligner<>(&measureModel, &detModel, FILE_LIST_PATH, CONSTRAINTS_PATH, 16);
 
#endif   // BMN_TEST_ALGORITHM
    // =========================================================================================
 
    Bool_t success;
    success = pAligner->Initialized();
    if (!success) {
        LOG(error) << " <!> final failure: initialization unsuccessful";
        return -1;
    }
    success = pAligner->PrepareData();
    if (!success) {
        LOG(error) << " <!> final failure: data preparation unsuccessful";
        return -1;
    }
    if (START_APPROX_PATH) {
        success = pAligner->LoadSolution(START_APPROX_PATH);
        if (!success) {
            LOG(error) << " <!> final failure: calibration solution loading unsuccessful";
            return -1;
        }
    }
    success = pAligner->Align();
    if (!success) {
        LOG(error) << " <!> final failure: alignment unsuccessful";
        return -1;
    }
 
    success = pAligner->SaveSolution(SOLUTION_PATH);
    if (!success) {
        LOG(error) << " <!> final failure: solution saving unsuccessful";
        return -1;
    }
 
    // COMMON SECTION END : INSERT FROM COMPILE VERSION UTIL HERE
    // ==========================================================
 
    pAligner->Draw();   // works in ROOT macro mode only
 
    // failed to implement object cleanup in ROOT macro mode
    // delete pAligner;
    return 0;
}