BasicProngVars.cxx

Go to the documentation of this file.

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

namespace SliceLID
{

bool sortByLength(const art::Ptr<rb::Prong> &a, const art::Ptr<rb::Prong> &b) {
    return a->TotalLength() > b->TotalLength();
}

void BasicProngVars::reserveVectors(size_t nProngs)
{
    len .reserve(nProngs);
    calE.reserve(nProngs);
}

bool BasicProngVars::getProngs(
    const art::Event &evt,
    const std::vector<art::Ptr<rb::Vertex>> &elastics,
    const std::string &baseLabel,
    const std::string &assnLabel,
    std::vector<art::Ptr<rb::Prong>> &prongs
)
{
    auto png_fmp = art::FindManyP<rb::Prong>(
        elastics, evt, art::InputTag(baseLabel, assnLabel)
    );

    if (! png_fmp.isValid()) {
        return false;
    }

    prongs = png_fmp.at(0);
    std::sort(prongs.begin(), prongs.end(), sortByLength);

    return true;
}

void BasicProngVars::initBasicVars()
{
    if (! hasProngs) {
        return;
    }

    reserveVectors(prongs.size());

    for (size_t pngIdx = 0; pngIdx < prongs.size(); ++pngIdx)
    {
        const rb::Prong &prong = *prongs[pngIdx];
        len .push_back(prong.TotalLength());
        calE.push_back(prong.CalorimetricEnergy());
    }
}

void BasicProngVars::initProngs(
    const art::Event &evt,
    const std::vector<art::Ptr<rb::Vertex>> &elastics
)
{
    if (elastics.empty()) {
        return;
    }

    hasProngs = getProngs(
        evt, elastics, fuzzykLabel, fuzzykAssnLabel, prongs
    );
}

BasicProngVars::BasicProngVars(
    const art::Event &evt,
    const std::vector<art::Ptr<rb::Vertex>> &elastics,
    const std::string &fuzzykLabel,
    const std::string &fuzzykAssnLabel
)
    : fuzzykLabel(fuzzykLabel),
      fuzzykAssnLabel(fuzzykAssnLabel)
{
    initProngs(evt, elastics);
    initBasicVars();
}

void BasicProngVars::saveToDict(
    std::unordered_map<std::string, std::vector<double>> &dict,
    const std::string &prefix
) const
{
    dict.emplace(prefix + "len",  len);
    dict.emplace(prefix + "calE", calE);
}

}