ConnectorFederateManager.hpp

/*
Copyright (c) 2017-2024,
Battelle Memorial Institute; Lawrence Livermore National Security, LLC; Alliance for Sustainable
Energy, LLC.  See the top-level NOTICE for additional details. All rights reserved.
SPDX-License-Identifier: BSD-3-Clause
*/
#pragma once
 
#include "../common/GuardedTypes.hpp"
#include "Filters.hpp"
#include "Translator.hpp"
#include "gmlc/containers/StringMappedVector.hpp"
 
#include <memory>
#include <string>
 
namespace helics {
class Core;
class Federate;
class ConnectorFederateManager {
  public:
    ConnectorFederateManager(Core* coreObj,
                             Federate* fFed,
                             LocalFederateId fid,
                             bool singleThreaded);
    ~ConnectorFederateManager();
 
    Filter&
        registerFilter(std::string_view name, std::string_view type_in, std::string_view type_out);
 
    CloningFilter& registerCloningFilter(std::string_view name,
                                         std::string_view type_in,
                                         std::string_view type_out);
 
    Filter& registerFilter(FilterTypes type, std::string_view name);
 
    CloningFilter& registerCloningFilter(FilterTypes type, std::string_view name);
 
    Filter& getFilter(std::string_view name);
    const Filter& getFilter(std::string_view name) const;
    Filter& getFilter(int index);
    const Filter& getFilter(int index) const;
 
    Translator& registerTranslator(std::string_view name,
                                   std::string_view endpointType,
                                   std::string_view units);
 
    Translator& getTranslator(std::string_view name);
    const Translator& getTranslator(std::string_view name) const;
    Translator& getTranslator(int index);
    const Translator& getTranslator(int index) const;
 
    int getFilterCount() const;
    int getTranslatorCount() const;
    void closeAllConnectors();
    void disconnectAllConnectors();
    void disconnect();
 
  private:
    Core* coreObject{nullptr};
    shared_guarded_opt<gmlc::containers::StringMappedVector<std::unique_ptr<Filter>>> filters;
    shared_guarded_opt<gmlc::containers::StringMappedVector<Translator>> translators;
    Federate* fed = nullptr;  
    const LocalFederateId fedID;  
};
}  // namespace helics