expdata.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 data export macro
*/
 
#ifndef __CLING__
#include "BmnRootIterator.h"
#include "BmnRun8DetModel.h"
#include "TStopwatch.h"
 
#include <fstream>
#endif
 
#define BMN_IN_FILELIST "cfg/dstDataExample.txt"
#define BMN_EXPORT_FILE "data.csv"
 
int expdata(int argc = 0, char** argv = nullptr)
{
    TStopwatch timer;
 
    // prepare ROOT file readers and histograms
    BmnRun8DetModel detModel;
    detModel.Init();
    BmnRootIterator data(BMN_IN_FILELIST, detModel);
 
    // prepare export file
    std::ofstream f(BMN_EXPORT_FILE, std::ofstream::out);
    f << "trackID,detectorID,x,y,z,wx,wy\n";
 
    // process ROOT data
    Int_t trackID{0}, hitCnt{0};
    data.ResetAll();
    while (!data.EndOfTracks()) {
        while (!data.EndOfHits()) {
            f << trackID << "," << data.HitDetectorID() << "," << data.HitX() << "," << data.HitY() << ","
              << data.HitZ() << "," << data.HitWx() << "," << data.HitWy() << "\n";
            hitCnt++;
            data.NextHit();
        }
        trackID++;
        data.NextTrack();
    }
    std::cout << "Tracks exported: " << trackID << std::endl;
    std::cout << "Hits exported:   " << hitCnt << std::endl;
 
    f.close();
    // no memory cleanup in ROOT macro mode
    std::cout << "Done in " << timer.RealTime() << "s" << std::endl;
    return 0;
}