BmnNdetCuts.h

Go to the documentation of this file.

#ifndef BMN_NDET_CUTS_H
#define BMN_NDET_CUTS_H
 
#include "BmnHgndDigi.h"
#include "BmnNdetDigi.h"
#include "BmnNdetDigit.h"
 
#include <fstream>
#include <iostream>
#include <nlohmann/json.hpp>
 
using json = nlohmann::json;
 
struct BmnNdetCuts
{
 
    // --- internal helper ---
    struct Range
    {
        double min = std::numeric_limits<double>::min(), max = std::numeric_limits<double>::max();
        bool hasMin = false, hasMax = false;
        inline bool Pass(double v) const
        {
            if (hasMin && v <= min)
                return false;
            if (hasMax && v >= max)
                return false;
            return true;
        }
    };
 
    // --- physics fields ---
    Range Time;   // fTime
 
    // --- methods ---
    inline bool PassSim(const BmnNdetDigit* d) const { return true; }
 
    inline bool PassPrototype(const BmnNdetDigi* d) const { return true; }
 
    inline bool PassExp(const BmnHgndDigi* d) const
    {
        if (!Time.Pass(d->GetTime()))
            return false;
        return true;
    }
 
    static BmnNdetCuts Load(const std::string& file, bool isExp)
    {
        std::ifstream f(file);
        if (!f.is_open()) {
            std::cerr << "[BmnNdetCuts] Cannot open config: " << file << std::endl;
            return {};
        }
 
        json j;
        try {
            f >> j;
        } catch (const std::exception& e) {
            std::cerr << "[BmnNdetCuts] JSON parse error in " << file << ": " << e.what() << std::endl;
            return {};
        }
 
        BmnNdetCuts c;
        const json& sec = isExp ? j["experiment"] : j["simulation"];
 
        auto readRange = [](Range& r, const json& js, const char* key) {
            const std::string kmin = std::string(key) + "_min";
            const std::string kmax = std::string(key) + "_max";
            if (js.contains(kmin)) {
                r.min = js[kmin].get<double>();
                r.hasMin = true;
            }
            if (js.contains(kmax)) {
                r.max = js[kmax].get<double>();
                r.hasMax = true;
            }
        };
 
        if (isExp) {
            readRange(c.Time, sec, "fTime");
        }
 
        std::cout << "[BmnNdetCuts] Loaded cuts for " << (isExp ? "experiment" : "simulation") << " from " << file
                  << std::endl;
 
        return c;
    }
};
 
#endif   // BMN_NDET_CUTS_H