NodeFeature.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
#ifndef MUONINFERENCEINTERACES_GRAPHFEATURE_H
#define MUONINFERENCEINTERACES_GRAPHFEATURE_H
 
 
#include <functional>
#include <string>
 
#include "MuonInferenceInterfaces/LayerBucket.h"
 
namespace MuonML {
    class NodeFeature {
        public:
            using Bucket_t = LayerSpBucket;

            using Func_t = std::function<double(const Bucket_t&, size_t)>;

            NodeFeature(const std::string& featName,
                        const Func_t& extractFunc):
                m_name{featName}, m_func{extractFunc}{}

            NodeFeature(NodeFeature&& other) = default;

            const std::string& name() const {
                return m_name;
            }

            double eval(const Bucket_t& bucket, size_t spIndex) const {
                return m_func(bucket, spIndex);
            }
        private:
            std::string m_name{};
            Func_t m_func{[](const Bucket_t& , size_t) { return 0.; }};
    };   
}
#endif