NodeConnector.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
#ifndef MUONINFERENCEINTERACES_NODECONNECTOR_H
#define MUONINFERENCEINTERACES_NODECONNECTOR_H
 
#include <string>
#include <functional>
 
#include <MuonInferenceInterfaces/NodeFeature.h>
namespace MuonML{
    class NodeConnector {
        public:
            using Bucket_t = NodeFeature::Bucket_t;
            using Evaluator_t = std::function<bool(const Bucket_t&, size_t, size_t)>;
            NodeConnector(const std::string& cName, const Evaluator_t conFunc):
                            m_name{cName}, m_func{conFunc} {} 

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

            bool connect(const Bucket_t& bucket, size_t i, size_t j) const {
                return m_func(bucket, i, j);
            }
 
        private:
            std::string m_name{};
            Evaluator_t m_func{[](const Bucket_t, size_t, size_t) { return false; }};
 
    };
}
#endif