diff -r 27259810da6d -r ee5758ddb6d2 src/documentmanager.cpp --- a/src/documentmanager.cpp Sat Jun 11 14:30:30 2022 +0300 +++ b/src/documentmanager.cpp Sat Jun 11 15:20:24 2022 +0300 @@ -20,6 +20,7 @@ #include #include #include +#include #include "documentmanager.h" #include "parser.h" @@ -255,51 +256,49 @@ .arg(missingString); } -/** - * @brief Cleans up and erases models that are no longer required. - */ -void DocumentManager::prune() +template +void removeFromSet(std::set& set, K&& valueToRemove) { - for (auto it = this->openModels.begin(); it != this->openModels.end(); ++it) - { - // Find models that are not edited by the user and are not needed by any other model - if (true - and it->second.opentype == OpenType::AutomaticallyOpened - and not this->isReferencedByAnything(it->first) - ) { - // Remove the model - this->openModels.erase(it); - // We need to start over now. It is possible that other models that - // previously were referenced by the model we just erased have - // become prunable. Moreover, our iterator is invalid now and we - // cannot continue in this loop. - this->prune(); - break; - } + const auto it = std::lower_bound(set.begin(), set.end(), valueToRemove); + if (it != set.end() and *it == valueToRemove) { + set.erase(it); } } -/** - * @brief Finds out whether the specified model id is referenced by any other model - * @param modelId - * @returns bool - */ -bool DocumentManager::isReferencedByAnything(const ModelId modelId) const +//! @brief Cleans up and erases models that are no longer required. +void DocumentManager::prune() { - for (auto& haystackModelPair : this->openModels) - { - if (haystackModelPair.first != modelId) - { - for (auto& dependencyPair : haystackModelPair.second.dependencies) - { - if (dependencyPair.second == modelId) - { - return true; - } + Graph dependencyGraph; + forValueInMap(this->openModels, [&dependencyGraph](const ModelInfo& info) { + forValueInMap(info.dependencies, [&dependencyGraph, &info](ModelId dep){ + dependencyGraph.push_back({.from = info.id, .to = dep}); + }); + }); + std::set autoOpened; + forValueInMap(this->openModels, [&autoOpened](const ModelInfo& info) { + if (info.opentype == OpenType::AutomaticallyOpened) { + autoOpened.insert(info.id); + } + }); + bool repeat = true; + while (repeat) { + repeat = false; + std::set prunable = autoOpened; + for (const auto& pair : dependencyGraph) { + removeFromSet(prunable, pair.to); + } + for (ModelId idToPrune : prunable) { + auto it = this->openModels.find(idToPrune); + if (it != this->openModels.end()) { + this->openModels.erase(it); } + removeFromSet(autoOpened, idToPrune); + std::erase_if(dependencyGraph, [&idToPrune](const GraphEdge& edge) { + return edge.from == idToPrune; + }); + repeat = true; } } - return false; } void DocumentManager::makePolygonCacheForModel(const ModelId modelId)