path: root/src/openvic-simulation/types/IdentifierRegistry.hpp

#pragma once

#include <map>
#include <vector>

#include "openvic-simulation/dataloader/NodeTools.hpp"
#include "openvic-simulation/utility/Logger.hpp"

namespace OpenVic {
   /*
    * Base class for objects with a non-empty string identifier,
    * uniquely named instances of which can be entered into an
    * IdentifierRegistry instance.
    */
   class HasIdentifier {
      const std::string identifier;

   protected:
      HasIdentifier(const std::string_view new_identifier);

   public:
      HasIdentifier(HasIdentifier const&) = delete;
      HasIdentifier(HasIdentifier&&) = default;
      HasIdentifier& operator=(HasIdentifier const&) = delete;
      HasIdentifier& operator=(HasIdentifier&&) = delete;

      std::string const& get_identifier() const;
   };

   std::ostream& operator<<(std::ostream& stream, HasIdentifier const& obj);
   std::ostream& operator<<(std::ostream& stream, HasIdentifier const* obj);

   /*
    * Base class for objects with associated colour information.
    */
   class HasColour {
      const colour_t colour;

   protected:
      HasColour(const colour_t new_colour, bool can_be_null, bool can_have_alpha);

   public:
      HasColour(HasColour const&) = delete;
      HasColour(HasColour&&) = default;
      HasColour& operator=(HasColour const&) = delete;
      HasColour& operator=(HasColour&&) = delete;

      colour_t get_colour() const;
      std::string colour_to_hex_string() const;
   };

   /*
    * Base class for objects with a unique string identifier
    * and associated colour information.
    */
   class HasIdentifierAndColour : public HasIdentifier, public HasColour {
   protected:
      HasIdentifierAndColour(const std::string_view new_identifier, const colour_t new_colour, bool can_be_null, bool can_have_alpha);

   public:
      HasIdentifierAndColour(HasIdentifierAndColour const&) = delete;
      HasIdentifierAndColour(HasIdentifierAndColour&&) = default;
      HasIdentifierAndColour& operator=(HasIdentifierAndColour const&) = delete;
      HasIdentifierAndColour& operator=(HasIdentifierAndColour&&) = delete;
   };

   using distribution_t = std::map<HasIdentifierAndColour const*, float>;

   distribution_t::value_type get_largest_item(distribution_t const& dist);

   /*
    * Template for a list of objects with unique string identifiers that can
    * be locked to prevent any further additions. The template argument T is
    * the type of object that the registry will store, and the second part ensures
    * that HasIdentifier is a base class of T.
    */
   template<typename T>
   requires(std::derived_from<T, HasIdentifier>)
   class IdentifierRegistry {
      using identifier_index_map_t = std::map<std::string, size_t, std::less<void>>;

      const std::string name;
      const bool log_lock;
      std::vector<T> items;
      bool locked = false;
      identifier_index_map_t identifier_index_map;

   public:
      IdentifierRegistry(const std::string_view new_name, bool new_log_lock = true)
         : name { new_name }, log_lock { new_log_lock } {}

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

      bool add_item(T&& item) {
         if (locked) {
            Logger::error("Cannot add item to the ", name, " registry - locked!");
            return false;
         }
         T const* old_item = get_item_by_identifier(item.get_identifier());
         if (old_item != nullptr) {
            Logger::error("Cannot add item to the ", name, " registry - an item with the identifier \"", item.get_identifier(), "\" already exists!");
            return false;
         }
         identifier_index_map[item.get_identifier()] = items.size();
         items.push_back(std::move(item));
         return true;
      }

      void lock() {
         if (locked) {
            Logger::error("Failed to lock ", name, " registry - already locked!");
         } else {
            locked = true;
            if (log_lock) Logger::info("Locked ", name, " registry after registering ", size(), " items");
         }
      }

      bool is_locked() const {
         return locked;
      }

      void reset() {
         identifier_index_map.clear();
         items.clear();
         locked = false;
      }

      size_t size() const {
         return items.size();
      }

      bool empty() const {
         return items.empty();
      }

      void reserve(size_t size) {
         if (locked) {
            Logger::error("Failed to reserve space for ", size, " items in ", name, " registry - already locked!");
         } else {
            items.reserve(size);
         }
      }

      T* get_item_by_identifier(const std::string_view identifier) {
         const identifier_index_map_t::const_iterator it = identifier_index_map.find(identifier);
         if (it != identifier_index_map.end()) return &items[it->second];
         return nullptr;
      }

      T const* get_item_by_identifier(const std::string_view identifier) const {
         const identifier_index_map_t::const_iterator it = identifier_index_map.find(identifier);
         if (it != identifier_index_map.end()) return &items[it->second];
         return nullptr;
      }

      T* get_item_by_index(size_t index) {
         return index < items.size() ? &items[index] : nullptr;
      }

      T const* get_item_by_index(size_t index) const {
         return index < items.size() ? &items[index] : nullptr;
      }

      std::vector<T>& get_items() {
         return items;
      }

      std::vector<T> const& get_items() const {
         return items;
      }

      std::vector<std::string_view> get_item_identifiers() const {
         std::vector<std::string_view> identifiers;
         identifiers.reserve(items.size());
         for (identifier_index_map_t::value_type const& entry : identifier_index_map) {
            identifiers.push_back(entry.first);
         }
         return identifiers;
      }

      NodeTools::node_callback_t expect_item_identifier(NodeTools::callback_t<T&> callback) {
         return NodeTools::expect_identifier(
            [this, callback](std::string_view identifier) -> bool {
               T* item = get_item_by_identifier(identifier);
               if (item != nullptr) return callback(*item);
               Logger::error("Invalid ", name, ": ", identifier);
               return false;
            }
         );
      }

      NodeTools::node_callback_t expect_item_identifier(NodeTools::callback_t<T const&> callback) const {
         return NodeTools::expect_identifier(
            [this, callback](std::string_view identifier) -> bool {
               T const* item = get_item_by_identifier(identifier);
               if (item != nullptr) return callback(*item);
               Logger::error("Invalid ", name, ": ", identifier);
               return false;
            }
         );
      }

      NodeTools::node_callback_t expect_item_dictionary(NodeTools::callback_t<T&, ast::NodeCPtr> callback) {
         return NodeTools::expect_dictionary([this, callback](std::string_view key, ast::NodeCPtr value) -> bool {
            T* item = get_item_by_identifier(key);
            if (item != nullptr) {
               return callback(*item, value);
            }
            Logger::error("Invalid ", name, " identifier: ", key);
            return false;
         });
      }

      NodeTools::node_callback_t expect_item_dictionary(NodeTools::callback_t<T const&, ast::NodeCPtr> callback) const {
         return NodeTools::expect_dictionary([this, callback](std::string_view key, ast::NodeCPtr value) -> bool {
            T const* item = get_item_by_identifier(key);
            if (item != nullptr) {
               return callback(*item, value);
            }
            Logger::error("Invalid ", name, " identifier: ", key);
            return false;
         });
      }

      NodeTools::node_callback_t expect_item_decimal_map(NodeTools::callback_t<std::map<T const*, fixed_point_t>&&> callback) const {
         return [this, callback](ast::NodeCPtr node) -> bool {
            std::map<T const*, fixed_point_t> map;
            bool ret = expect_item_dictionary([&map](T const& key, ast::NodeCPtr value) -> bool {
               fixed_point_t val;
               const bool ret = NodeTools::expect_fixed_point(NodeTools::assign_variable_callback(val))(value);
               map.emplace(&key, val);
               return ret;
            })(node);
            ret &= callback(std::move(map));
            return ret;
         };
      }
   };

#define IDENTIFIER_REGISTRY_ACCESSORS_CUSTOM_PLURAL(type, singular, plural) \
   void lock_##plural() { plural.lock(); } \
   type const* get_##singular##_by_identifier(const std::string_view identifier) const { \
      return plural.get_item_by_identifier(identifier); } \
   size_t get_##singular##_count() const { \
      return plural.size(); } \
   std::vector<type> const& get_##plural() const { \
      return plural.get_items(); } \
   std::vector<std::string_view> get_##singular##_identifiers() const { \
      return plural.get_item_identifiers(); } \
   NodeTools::node_callback_t expect_##singular##_identifier(NodeTools::callback_t<type const&> callback) const { \
      return plural.expect_item_identifier(callback); } \
   NodeTools::node_callback_t expect_##singular##_dictionary(NodeTools::callback_t<type const&, ast::NodeCPtr> callback) const { \
      return plural.expect_item_dictionary(callback); } \
   NodeTools::node_callback_t expect_##singular##_decimal_map(NodeTools::callback_t<std::map<type const*, fixed_point_t>&&> callback) const { \
      return plural.expect_item_decimal_map(callback); }

#define IDENTIFIER_REGISTRY_NON_CONST_ACCESSORS_CUSTOM_PLURAL(type, singular, plural) \
   type* get_##singular##_by_identifier(const std::string_view identifier) { \
      return plural.get_item_by_identifier(identifier); } \
   NodeTools::node_callback_t expect_##singular##_identifier(NodeTools::callback_t<type&> callback) { \
      return plural.expect_item_identifier(callback); } \
   NodeTools::node_callback_t expect_##singular##_dictionary(NodeTools::callback_t<type&, ast::NodeCPtr> callback) { \
      return plural.expect_item_dictionary(callback); }

#define IDENTIFIER_REGISTRY_ACCESSORS(type, name) IDENTIFIER_REGISTRY_ACCESSORS_CUSTOM_PLURAL(type, name, name##s)
#define IDENTIFIER_REGISTRY_NON_CONST_ACCESSORS(type, name) IDENTIFIER_REGISTRY_NON_CONST_ACCESSORS_CUSTOM_PLURAL(type, name, name##s)
}