BmnHitDumper.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:     Claude 4.7 (Opus)
    Supervisor: Igor Polev, polev@jinr.int
 
    BmnHitDumper class implementation
*/
 
#include "BmnHitDumper.h"
 
#include "BmnEventHeader.h"
#include "CbmStsHit.h"
#include "FairMCEventHeader.h"
#include "FairRootManager.h"
#include "TSystem.h"
#include "fairlogger/Logger.h"
 
BmnHitDumper::BmnHitDumper(const TString& dstFileName)
    : fCsvFilePath(dstFileName)
{
    // expand $VAR and ~ if present
    char* expanded = gSystem->ExpandPathName(fCsvFilePath.Data());
    fCsvFilePath = expanded;
    delete[] expanded;
    // replace extension with .csv (or append if none)
    Int_t dotIdx{fCsvFilePath.Last('.')};
    Int_t sepIdx{fCsvFilePath.Last('/')};
    if (dotIdx > sepIdx)
        fCsvFilePath.Replace(dotIdx, fCsvFilePath.Length() - dotIdx, ".csv");
    else
        fCsvFilePath.Append(".csv");
}
 
InitStatus BmnHitDumper::Init()
{
    if (fDetModel.IgnoresSome()) {
        LOG(error) << " <!> BmnHitDumper::Init() : detector model is incomplete (ignores some hits)"
                      " - cannot be used to dump hits consistently. Task disabled.";
        SetActive(kFALSE);
        return kERROR;
    }
    FairRootManager* ioman = FairRootManager::Instance();
    if (!ioman) {
        LOG(error) << " <!> BmnHitDumper::Init() : FairRootManager not found";
        SetActive(kFALSE);
        return kERROR;
    }
    fpStsHitsArray = (TClonesArray*)ioman->GetObject("StsHit");
    if (!fpStsHitsArray) {
        LOG(error) << " <!> BmnHitDumper::Init() : Failed to get StsHit array";
        SetActive(kFALSE);
        return kERROR;
    }
    // The "DstEventHeader." branch cannot be queried via ioman->GetObject(): the
    // output DstEventHeader (registered via FairRunAna::SetEventHeader) shadows
    // the input one in FairRootManager's folder lookup, and its BMN-specific
    // fEventId is never populated by FairFileSource/BmnFileSource (which only
    // touch FairEventHeader base fields). Read the per-event ID from the input
    // header instead: BmnEventHeader. for experimental data, MCEventHeader. for
    // simulation. Both are auto-updated each event via FairFileSource::ReadEvent.
    fpBmnHeader = (BmnEventHeader*)ioman->GetObject("BmnEventHeader.");
    fpMcHeader = (FairMCEventHeader*)ioman->GetObject("MCEventHeader.");
    if (!fpBmnHeader && !fpMcHeader) {
        LOG(error) << " <!> BmnHitDumper::Init() : "
                      "no BmnEventHeader. or MCEventHeader. branch in input";
        SetActive(kFALSE);
        return kERROR;
    }
    if (!fDetModel.Init()) {
        LOG(error) << " <!> BmnHitDumper::Init() : Failed to initialize detector model";
        SetActive(kFALSE);
        return kERROR;
    }
    fCsvFile.open(fCsvFilePath.Data(), std::ofstream::out);
    if (!fCsvFile.is_open()) {
        LOG(error) << " <!> BmnHitDumper::Init() : Failed to open output file " << fCsvFilePath.Data();
        SetActive(kFALSE);
        return kERROR;
    }
    fCsvFile << "eventID,detectorID,x,y,z,dx,dy\n";
    LOG(info) << "BmnHitDumper: writing hits to " << fCsvFilePath.Data();
    SetActive(kTRUE);
    fTimer.Start();
    return kSUCCESS;
}
 
void BmnHitDumper::Exec(Option_t* /*opt*/)
{
    UInt_t eventID{fpBmnHeader ? fpBmnHeader->GetEventId() : (UInt_t)fpMcHeader->GetEventID()};
    Int_t n{fpStsHitsArray->GetEntriesFast()};
    for (Int_t i = 0; i < n; i++) {
        CbmStsHit* hit = static_cast<CbmStsHit*>(fpStsHitsArray->At(i));
        fCsvFile << eventID << "," << fDetModel.IDfromHit(hit->GetDetectorID()) << "," << hit->GetX() << ","
                 << hit->GetY() << "," << hit->GetZ() << "," << hit->GetDx() << "," << hit->GetDy() << "\n";
    }
    LOG(info) << "BmnHitDumper: " << n << " hits written from event " << eventID;
}
 
void BmnHitDumper::Finish()
{
    if (fCsvFile.is_open())
        fCsvFile.close();
    fTimer.Stop();
    Int_t secs{static_cast<Int_t>(fTimer.RealTime())};
    LOG(info) << Form("BmnHitDumper: total time spent: %dh %02dm %02ds", secs / 3600, (secs % 3600) / 60, secs % 60);
}