patch_manager.cpp

/*---------------------------------------------------------------------------*\
 *
 *  bitpit
 *
 *  Copyright (C) 2015-2021 OPTIMAD engineering Srl
 *
 *  -------------------------------------------------------------------------
 *  License
 *  This file is part of bitpit.
 *
 *  bitpit is free software: you can redistribute it and/or modify it
 *  under the terms of the GNU Lesser General Public License v3 (LGPL)
 *  as published by the Free Software Foundation.
 *
 *  bitpit is distributed in the hope that it will be useful, but WITHOUT
 *  ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 *  FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public
 *  License for more details.
 *
 *  You should have received a copy of the GNU Lesser General Public License
 *  along with bitpit. If not, see <http://www.gnu.org/licenses/>.
 *
\*---------------------------------------------------------------------------*/
 
#include "patch_kernel.hpp"
#include "patch_manager.hpp"
 
namespace bitpit {
 
const int PatchManager::AUTOMATIC_ID = IndexGenerator<int>::NULL_ID;
 
/*
    Initialize logger manager instance.
*/
std::unique_ptr<PatchManager> PatchManager::m_manager = nullptr;
 
PatchManager & PatchManager::manager()
{
    if (!m_manager) {
        m_manager = std::unique_ptr<PatchManager>(new PatchManager());
    }
 
    return *m_manager;
}
 
PatchKernel * PatchManager::get(int id)
{
    for (const auto &entry : m_patchIds) {
        long patchId = entry.second;
        if (patchId == id) {
            return entry.first;
        }
    }
 
    return nullptr;
}
 
int PatchManager::registerPatch(PatchKernel *patch, int id)
{
    if (id >= 0) {
        if (m_idGenerator.isAssigned(id)) {
            throw std::runtime_error ("A patch with the same id already exists");
        }
 
        m_idGenerator.setAssigned(id);
    } else {
        id = m_idGenerator.generate();
    }
 
    patch->_setId(id);
    m_patchIds[patch] = id;
    m_patchOrder.push_back(patch);
 
    return id;
}
 
void PatchManager::unregisterPatch(PatchKernel *patch)
{
    auto iterator = m_patchIds.find(patch);
    if (iterator == m_patchIds.end()) {
        throw std::runtime_error ("The patch to be unregistered does not exist");
    }
 
    int id = iterator->second;
    m_idGenerator.trash(id);
 
    m_patchIds.erase(iterator);
    for (auto itr = m_patchOrder.begin(); itr != m_patchOrder.end(); ++itr) {
        if (*itr == patch) {
            m_patchOrder.erase(itr);
            break;
        }
    }
}
 
PatchManager::PatchManager()
{
}
 
void PatchManager::dump(std::ostream &stream)
{
    m_idGenerator.dump(stream);
}
 
void PatchManager::restore(std::istream &stream)
{
    m_idGenerator.restore(stream);
}
 
void PatchManager::dumpAll(std::ostream &stream)
{
    for (PatchKernel *patch : m_patchOrder) {
        patch->dump(stream);
    }
 
    dump(stream);
}
 
void PatchManager::restoreAll(std::istream &stream)
{
    m_idGenerator.reset();
 
    for (PatchKernel *patch : m_patchOrder) {
        patch->restore(stream);
    }
 
    restore(stream);
}
 
// Patch manager global functions
namespace patch {
 
    // Generic global functions
 
    PatchManager & manager()
    {
        return PatchManager::manager();
    }
 
}
 
}