ObjectPool.h

/*
 * Copyright (C) 2025, Robert Patterson
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 */
#pragma once
 
#include <string>
#include <map>
#include <unordered_map>
#include <vector>
#include <memory>
#include <tuple>
#include <variant>
#include <stdexcept>
#include <functional>
#include <limits>
 
#include "BaseClasses.h"
#include "Others.h"
#include "Details.h"
#include "Entries.h"
#include "SmartShape.h"
 
namespace musx {
namespace dom {
 
template <typename ObjectBaseType>
class ObjectPool
{
public:
    using ObjectPtr = std::shared_ptr<ObjectBaseType>;
    struct ObjectKey {
        std::string nodeId;             
        Cmper partId;                   
        std::optional<Cmper> cmper1;    
        std::optional<Cmper> cmper2;    
        std::optional<Inci> inci;       
 
        ObjectKey(const std::string n,
            Cmper p,
            std::optional<Cmper> c1 = std::nullopt,
            std::optional<Cmper> c2 = std::nullopt,
            std::optional<Inci> i = std::nullopt) : nodeId(n), partId(p), cmper1(c1), cmper2(c2), inci(i)
        {
        }
 
        bool operator<(const ObjectKey& other) const
        {
            if (nodeId != other.nodeId) {
                return nodeId < other.nodeId;
            }
            if (partId != other.partId) {
                return partId < other.partId;
            }
            if (cmper1 != other.cmper1) {
                return cmper1 < other.cmper1;
            }
            if (cmper2 != other.cmper2) {
                return cmper2 < other.cmper2;
            }
            return inci < other.inci;
        }
 
        std::string description() const
        {
            std::string result = nodeId + " part " + std::to_string(partId);
            if (cmper1) {
                if (cmper2) {
                    result += " cmpers [" + std::to_string(cmper1.value()) + ", " + std::to_string(cmper2.value()) + "]";
                } else {
                    result += " cmper " + std::to_string(cmper1.value());
                }
            }
            if (inci) {
                result += " inci " + std::to_string(inci.value());
            }
            return result;
        }
    };
 
    virtual ~ObjectPool() = default;
 
    void add(const ObjectKey& key, ObjectPtr object)
    {
        if (key.inci.has_value()) {
            ObjectKey noInciKey = key;
            noInciKey.inci = std::nullopt;
            auto currentIncis = getArray<ObjectBaseType>(noInciKey);
            if (key.inci.value() != int(currentIncis.size())) {
                MUSX_INTEGRITY_ERROR("Node " + key.nodeId + " has inci " + std::to_string(key.inci.value()) + " that is out of sequence.");
            }
        }
        auto [poolIt, emplaced] = m_pool.emplace(key, object);
        if (!emplaced) {
            MUSX_INTEGRITY_ERROR("Attempted to add same key more than once: " + key.description());
        }
        auto it = m_shareMode.find(key.nodeId);
        if (it == m_shareMode.end()) {
            m_shareMode.emplace(key.nodeId, object->getShareMode());
        } else if (object->getShareMode() != it->second && object->getShareMode() != Base::ShareMode::All) {
            if (it->second == Base::ShareMode::All) {
                m_shareMode[key.nodeId] = object->getShareMode();
            } else {
                MUSX_INTEGRITY_ERROR("Share mode for added " + key.nodeId + " object [" + std::to_string(int(object->getShareMode()))
                    + "] does not match previous [" + std::to_string(int(it->second)) + "]");                
            }
        }
    }
 
    template <typename T>
    std::vector<std::shared_ptr<T>> getArray(const ObjectKey& key) const
    {
        std::vector<std::shared_ptr<T>> result;
 
        auto rangeStart = m_pool.lower_bound(key);
        auto rangeEnd = m_pool.upper_bound(
            ObjectKey{
                key.nodeId,
                key.partId,
                key.cmper1.value_or((std::numeric_limits<Cmper>::max)()),
                key.cmper2.value_or((std::numeric_limits<Cmper>::max)()), 
                key.inci.value_or((std::numeric_limits<Inci>::max)())
            }
        );
 
        for (auto it = rangeStart; it != rangeEnd; ++it) {
            auto typedPtr = std::dynamic_pointer_cast<T>(it->second);
            assert(typedPtr);
            result.push_back(typedPtr);
        }
        return result;
    }
 
    template <typename T>
    std::vector<std::shared_ptr<T>> getArrayForPart(const ObjectKey& key) const
    {
        Base::ShareMode forShareMode = Base::ShareMode::All;
        if (key.partId != SCORE_PARTID) {
            auto it = m_shareMode.find(key.nodeId);
            // If the nodeId is not found in m_shareMode, it means the document contains no instances,
            // and every path will return an empty vector. We can safely ignore this case.
            if (it != m_shareMode.end()) {
                forShareMode = it->second;
            }
            if (forShareMode == Base::ShareMode::Partial) {
                if constexpr (std::is_base_of_v<OthersBase, T>) {
                    if (!key.cmper1.has_value()) {
                        throw std::invalid_argument("Array searches on partially shared Others must supply a cmper.");
                    }
                } else if constexpr (std::is_base_of_v<DetailsBase, T>) {
                    if (!key.cmper1.has_value() || !key.cmper2.has_value()) {
                        throw std::invalid_argument("Array searches on partially shared Details must supply both cmpers.");
                    }
                }
            }
        }
        auto keyResult = getArray<T>(key);
        if (!keyResult.empty() || key.partId == SCORE_PARTID || forShareMode == Base::ShareMode::None) {
            return keyResult;
        }
        ObjectKey scoreKey(key);
        scoreKey.partId = SCORE_PARTID;
        return getArray<T>(scoreKey);
    }
 
    template <typename T>
    std::shared_ptr<T> get(const ObjectKey& key) const
    {
        auto it = m_pool.find(key);
        if (it == m_pool.end()) {
            return nullptr;
        }
        auto typedPtr = std::dynamic_pointer_cast<T>(it->second);
        assert(typedPtr); // There is a program bug if the pointer cast fails.
        return typedPtr;
    }
 
    template <typename T>
    std::shared_ptr<T> getEffectiveForPart(const ObjectKey& key) const
    {
        if (auto partVersion = get<T>(key)) {
            return partVersion;
        }
        if (key.partId == SCORE_PARTID) {
            // if this is already the score version, there is nothing to return.
            return nullptr;
        }
        ObjectKey scoreKey(key);
        scoreKey.partId = SCORE_PARTID;
        return get<T>(scoreKey);
    }
 
protected:
    ObjectPool(const std::unordered_map<std::string, dom::Base::ShareMode>& knownShareModes = {})
        : m_shareMode(knownShareModes) {}
 
private:
    std::map<ObjectKey, ObjectPtr> m_pool;
    std::unordered_map<std::string, dom::Base::ShareMode> m_shareMode;
};
 
class OptionsPool : protected ObjectPool<OptionsBase>
{
public:
    void add(const std::string& nodeName, const std::shared_ptr<OptionsBase>& instance)
    {
        if (instance->getPartId()) {
            MUSX_INTEGRITY_ERROR("Options node " + nodeName + " hase non-zero part id [" + std::to_string(instance->getPartId()) + "]");
        }
        ObjectPool::add({ nodeName, instance->getPartId() }, instance);
    }
 
    template <typename T>
    std::vector<std::shared_ptr<T>> getArray() const
    {
        return ObjectPool::getArray<T>({ std::string(T::XmlNodeName), SCORE_PARTID });
    }
 
    template <typename T>
    std::shared_ptr<T> get() const
    {
        return ObjectPool::get<T>({ std::string(T::XmlNodeName), SCORE_PARTID });
    }
};
using OptionsPoolPtr = std::shared_ptr<OptionsPool>;
 
class OthersPool : public ObjectPool<OthersBase>
{
public:
    OthersPool() : ObjectPool({
        { std::string(others::BeatChartElement::XmlNodeName), Base::ShareMode::None },
        { std::string(others::InstrumentUsed::XmlNodeName), Base::ShareMode::None },
        { std::string(others::SystemLock::XmlNodeName), Base::ShareMode::None },
        { std::string(others::MultiStaffInstrumentGroup::XmlNodeName), Base::ShareMode::None },
        { std::string(others::MultimeasureRest::XmlNodeName), Base::ShareMode::None },
        { std::string(others::MultiStaffGroupId::XmlNodeName), Base::ShareMode::None },
        { std::string(others::Page::XmlNodeName), Base::ShareMode::None },
        { std::string(others::PartGlobals::XmlNodeName), Base::ShareMode::None },
        { std::string(others::StaffSystem::XmlNodeName), Base::ShareMode::None },
        { std::string(others::StaffStyleAssign::XmlNodeName), Base::ShareMode::None },
        // add other known sharemode none items as they are identified.
    }) {}
 
    void add(const std::string& nodeName, const std::shared_ptr<OthersBase>& instance)
    { ObjectPool::add({nodeName, instance->getPartId(), instance->getCmper(), std::nullopt, instance->getInci()}, instance); }
    
    template <typename T>
    std::vector<std::shared_ptr<T>> getArray(Cmper partId, std::optional<Cmper> cmper = std::nullopt) const
    { return ObjectPool::getArrayForPart<T>({ std::string(T::XmlNodeName), partId, cmper }); }
 
    template <typename T>
    std::shared_ptr<T> get(Cmper partId, Cmper cmper, std::optional<Inci> inci = std::nullopt) const
    { return ObjectPool::getEffectiveForPart<T>({std::string(T::XmlNodeName), partId, cmper, std::nullopt, inci}); }
};
using OthersPoolPtr = std::shared_ptr<OthersPool>;
 
class DetailsPool : protected ObjectPool<DetailsBase>
{
public:
    DetailsPool() : ObjectPool({
        { std::string(details::CenterShape::XmlNodeName), Base::ShareMode::None },
        { std::string(details::StaffGroup::XmlNodeName), Base::ShareMode::None },
        { std::string(details::StaffSize::XmlNodeName), Base::ShareMode::None },
        // add other known sharemode none items as they are identified.
    }) {}
 
    void add(const std::string& nodeName, const std::shared_ptr<DetailsBase>& instance)
    { ObjectPool::add({nodeName, instance->getPartId(), instance->getCmper1(), instance->getCmper2(), instance->getInci()}, instance); }
 
    template <typename T>
    std::vector<std::shared_ptr<T>> getArray(Cmper partId) const
    { return ObjectPool::template getArrayForPart<T>({ std::string(T::XmlNodeName), partId }); }
 
    template <typename T, typename std::enable_if_t<!std::is_base_of_v<EntryDetailsBase, T>, int> = 0>
    std::vector<std::shared_ptr<T>> getArray(Cmper partId, Cmper cmper1, std::optional<Cmper> cmper2 = std::nullopt) const
    { return ObjectPool::template getArrayForPart<T>({ std::string(T::XmlNodeName), partId, cmper1, cmper2 }); }
 
    template <typename T, typename std::enable_if_t<std::is_base_of_v<EntryDetailsBase, T>, int> = 0>
    std::vector<std::shared_ptr<T>> getArray(Cmper partId, EntryNumber entnum) const
    { return ObjectPool::template getArrayForPart<T>({ std::string(T::XmlNodeName), partId, Cmper(entnum >> 16), Cmper(entnum & 0xffff) }); }
 
    template <typename T, typename std::enable_if_t<!std::is_base_of_v<EntryDetailsBase, T>, int> = 0>
    std::shared_ptr<T> get(Cmper partId, Cmper cmper1, Cmper cmper2, std::optional<Inci> inci = std::nullopt) const
    { return ObjectPool::getEffectiveForPart<T>({std::string(T::XmlNodeName), partId, cmper1, cmper2, inci}); }
 
    template <typename T, typename std::enable_if_t<std::is_base_of_v<EntryDetailsBase, T>, int> = 0>
    std::shared_ptr<T> get(Cmper partId, EntryNumber entnum, std::optional<Inci> inci = std::nullopt) const
    { return ObjectPool::getEffectiveForPart<T>({std::string(T::XmlNodeName), partId, Cmper(entnum >> 16), Cmper(entnum & 0xffff), inci}); }
 
    template <typename T, typename std::enable_if_t<std::is_base_of_v<NoteDetailsBase, T>, int> = 0>
    std::shared_ptr<T> getForNote(const NoteInfoPtr& noteInfo, const std::optional<Cmper>& forPartId = std::nullopt)
    {
        auto details = getArray<T>(
            forPartId.value_or(noteInfo.getEntryInfo().getFrame()->getRequestedPartId()),
            noteInfo.getEntryInfo()->getEntry()->getEntryNumber()
        );
        for (const auto& detail : details) {
            if (detail->getNoteId() == noteInfo->getNoteId()) {
                return detail;
            }
        }
        return nullptr;
    }
};
 
using DetailsPoolPtr = std::shared_ptr<DetailsPool>;
 
class EntryPool // uses different implementation than other pools for more efficient access
{
public:
    void add(EntryNumber entryNumber, const std::shared_ptr<Entry>& instance)
    {
        auto [it, emplaced] = m_pool.emplace(entryNumber, instance);
        if (!emplaced) {
            MUSX_INTEGRITY_ERROR("Entry number " + std::to_string(entryNumber) + " added twice.");
        }
    }
 
    std::shared_ptr<Entry> get(EntryNumber entryNumber) const
    {
        const auto it = m_pool.find(entryNumber);
        if (it == m_pool.end()) {
            return nullptr;
        }
        return it->second;
    }
 
private:
    std::unordered_map<EntryNumber, std::shared_ptr<Entry>> m_pool;
};
using EntryPoolPtr = std::shared_ptr<EntryPool>;
 
class TextsPool : protected ObjectPool<TextsBase>
{
public:
    void add(const std::string& nodeName, const std::shared_ptr<TextsBase>& instance)
    {
        if (instance->getPartId()) {
            MUSX_INTEGRITY_ERROR("Texts node " + nodeName + " hase non-zero part id [" + std::to_string(instance->getPartId()) + "]");
        }
        ObjectPool::add({ nodeName, instance->getPartId(), instance->getTextNumber() }, instance);
    }
    
    template <typename T>
    std::vector<std::shared_ptr<T>> getArray(std::optional<Cmper> cmper = std::nullopt) const
    { return ObjectPool::getArray<T>({ std::string(T::XmlNodeName), SCORE_PARTID, cmper }); }
 
    template <typename T>
    std::shared_ptr<T> get(Cmper cmper) const
    { return ObjectPool::get<T>({ std::string(T::XmlNodeName), SCORE_PARTID, cmper, std::nullopt, std::nullopt }); }
};
using TextsPoolPtr = std::shared_ptr<TextsPool>;
 
} // namespace dom
} // namespace musx