ProngShwVars.cxx

Go to the documentation of this file.

#include "SliceLID/art/var_dict_builder/ProngShwVars.h"

namespace SliceLID
{

void ProngShwVars::fillVars(
    const art::FindOneP<rb::Shower> &prongShw_fop, size_t pngIdx
)
{
    if (! prongShw_fop.isValid()) {
        return;
    }

    art::Ptr<rb::Shower> shw = prongShw_fop.at(pngIdx);
    if (! shw) {
        return;
    }

    vvars.fillVars(*shw, pngIdx);
}

void ProngShwVars::init(
    const art::Event &evt, const BasicProngVars &pngVars
)
{
    vvars.initVars(pngVars.prongs.size(), SHW_FALLBACK_VALUE);

    if (! pngVars.hasProngs) {
        return;
    }

    art::FindOneP<rb::Shower> prongShw_fop(
        pngVars.prongs, evt, prongShwLabel
    );

    for (size_t pngIdx = 0; pngIdx < pngVars.prongs.size(); pngIdx++) {
        fillVars(prongShw_fop, pngIdx);
    }
}

void ProngShwVars::saveToDict(
    std::unordered_map<std::string, std::vector<double>> &dict,
    const std::string &prefix
) const
{
    vvars.saveToDict(dict, prefix);
}

ProngShwVars::ProngShwVars(
    const art::Event &evt,
    const BasicProngVars &pngVars,
    const std::string &prongShwLabel
) : vvars(SHW_VAR_DEFS),
    prongShwLabel(prongShwLabel)
{
    init(evt, pngVars);
}

}