NodeFeatureList.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
#ifndef MUONINFERENCEINTERACES_GRAPHFEATURELIST_H
#define MUONINFERENCEINTERACES_GRAPHFEATURELIST_H
 
#include "MuonInferenceInterfaces/NodeFeature.h"
#include "MuonInferenceInterfaces/NodeConnector.h"
 
class MsgStream;
 
namespace MuonML {
    struct GraphRawData;
    class NodeFeatureList {
        public:
            
            using Feature_t = std::shared_ptr<const NodeFeature>;
            using Connector_t = std::shared_ptr<const NodeConnector>;
 
            using Bucket_t = NodeFeature::Bucket_t;
            NodeFeatureList() = default;
            bool operator==(const NodeFeatureList& other) const;
            bool isValid() const;
            size_t numFeatures() const;
            std::vector<std::string> featureNames() const;
            void fillInData(const Bucket_t& bucket, 
                            GraphRawData& graphData) const;
            bool addFeature(const std::string& featName, MsgStream& msg);
            bool addFeature(const Feature_t& featPtr,  MsgStream& msg);

            bool setConnector(const std::string& conName, MsgStream& msg);
            void setConnector(const std::string& conName, NodeConnector::Evaluator_t evalFunc);
 
            
        private:
            std::vector<Feature_t> m_features{};
            Connector_t m_connector{};
            
    };
}
 
 
#endif