BmnMetadataTest.C

Go to the documentation of this file.

 
#include <ROOT/RDF/RInterface.hxx>
#include <ROOT/RDataFrame.hxx>
#include <filesystem>
 
namespace fs = std::filesystem;
using SysClock = std::chrono::system_clock;
using SysPoint = std::chrono::time_point<SysClock>;
 
SysPoint fRunStartTime;
SysPoint fRunEndTime;
UInt_t fRunId;
UInt_t fPeriodId;
bool get_run_interval_from_file = false;
unique_ptr<DigiRunHeader> digiRunHeader;
vector<string> fDigiFileNames;
 
void GatherStatusBlocksFromDigis()
{
    // try get run interval from DB
    if (!get_run_interval_from_file) {
        UniRun* pRun = UniRun::GetRun(fPeriodId, fRunId);
        if (!pRun) {
            get_run_interval_from_file = true;
        } else {
            TDatime fRunStartDate = pRun->GetStartDatetime();
            TDatime* fRunEndDate = pRun->GetEndDatetime();
            fRunStartTime = SysClock::from_time_t(fRunStartDate.Convert(true)) - 3h;   // crutch for zone shift
            if (fRunEndDate)
                fRunEndTime = SysClock::from_time_t(fRunEndDate->Convert(true)) - 3h;
            else
                get_run_interval_from_file = true;
            printf("Set Run from DB:\n\tstart ts: %s\n\tstop  ts: %s\n",
                   BmnFunctionSet::TimePoint2String(fRunStartTime).c_str(),
                   BmnFunctionSet::TimePoint2String(fRunEndTime).c_str());
            delete pRun;
        }
    }
}
 
void BmnMetadataTest(vector<string> files, TString meta_file, UInt_t run_id, UInt_t period = 8)
{
    gSystem->ExpandPathName(meta_file);
    //        ROOT::EnableImplicitMT();
    FairLogger::GetLogger()->SetLogToScreen(true);
    FairLogger::GetLogger()->SetLogScreenLevel("info");   // for info write "INFO"
 
    TStopwatch timer;
    timer.Start();
    fDigiFileNames = files;
    fRunId = run_id;
    fPeriodId = period;
    string tar_dir = "/ncx/eos/nica/bmn/exp/digi/run8/24.04.0/0_technichal_info/";
    // string dig_dir = "/ncx/eos/nica/bmn/exp/digi/run8/24.04.0/";
    //  string dig_dir = "/home/ilnur/filesbmn/";
    //      string dig_dir = "/ncx/eos/nica/bmn/users/ilnur/spill_test/";
    string dig_dir = "/home/ilnur/spill_test/";
    //    string dig_dir = "/data/spill_test/";
    //      string dig_dir = "/mnt/ddc/ceph/bmn/work/ilnur/spill_test/";
    // string dig_dir = "/mnt/ddc/eos/lpp/nica/bmn/users/ilnur/spill_test/";
 
    if (files.size() == 0) {
        string re_str(Form(".*%d(_ev\\d+_p\\d+)[^\\/]*\\.root", run_id));
        TPRegexp re_dig(re_str);
        // Check if the path exists and is a directory
        //        if (fs::exists(dig_dir) && fs::is_directory(dig_dir)) {
        std::cout << "Target files in a directory: " << dig_dir << std::endl;
        // Iterate through the directory
        for (const auto& entry : std::filesystem::directory_iterator(dig_dir)) {
            string fileName = entry.path();
            if (re_dig.MatchB(fileName)) {
                std::cout << entry.path().filename() << std::endl;
                files.push_back(fileName.data());
            }
        }
    }
    GatherStatusBlocksFromDigis();
    // count activated triggers in events
    TFile* file = new TFile(meta_file, "READ");
    if (file->IsOpen() == false) {
        cout << "Cannot open file " << meta_file << " !" << endl;
        return;
    }
    DigiRunHeader* rh = static_cast<DigiRunHeader*>(file->Get("DigiRunHeader"));
    BmnMetadataRaw* metadata = static_cast<BmnMetadataRaw*>(file->Get("RawRunMetadata"));
    map<string, uint16_t>& tr_map = rh->TrcBitMap();
    //    map<string, uint16_t> tr_map = {
    //        std::pair("BC1L", 0),
    //        std::pair("pBT", 1),
    //        std::pair("BT", 2),
    //        std::pair("NiT", 3),
    //        std::pair("CCT1", 4),
    //        std::pair("MBT", 5),
    //        std::pair("BT&nBUSY", 6),
    //        std::pair("CCT2", 7)
    //    };
    ROOT::RDataFrame df{"bmndata", files};
    ROOT::RDataFrame df_spill{"spill", meta_file};
    auto df_run = df;
    auto df_spill_run = df_spill;
    printf("Evt_cnt total %7u\n", rh->GetNEvents());
    for (auto& el : tr_map) {
        const string& tr_name = el.first;
        printf("Evt_cnt %-12s %7lu\n", tr_name.c_str(), rh->EventTriggerCnt()[tr_name]);
    }
    auto df_sp_BT_CCT2 = df_run.Filter(
        [&](BmnEventHeader& eh) {
            return ((eh.GetInputSignalsAR() & BIT(tr_map["BT"])) && (eh.GetInputSignalsAR() & BIT(tr_map["CCT2"])));
        },
        {"BmnEventHeader."});
    ULong64_t df_sp_BT_CCT2_cnt = *(df_sp_BT_CCT2.Count());
    printf("Evt_cnt BT_CCT2 %7llu \n", df_sp_BT_CCT2_cnt);
    printf("Evt_cnt (BT_CCT2/BT) %5.4f \n", df_sp_BT_CCT2_cnt / (Double_t)rh->EventTriggerCnt()["BT"]);
 
    printf("T0 CCT2 = %10llu\n", rh->GetRawT0Counter("CCT2"));
    printf("T0 BT   = %10llu\n", rh->GetRawT0Counter("BT"));
    printf("T0      (CCT2/BT)    %5.4f \n", rh->GetRawT0Counter("CCT2") / (Double_t)rh->GetRawT0Counter("BT"));
 
    timer.Stop();
    printf("\tMacro finished successfully!\n\n");   // marker of successfully execution
    Double_t rtime = timer.RealTime();
    Double_t ctime = timer.CpuTime();
    printf("All steps: Real time %f s, CPU time %f s\n", rtime, ctime);
}