a644f75d53
The class MaterialLibrary has a shared pointer to its Material objects and the class Material has a shared pointer to the MaterialLibrary. The class MaterialManager owns all MaterialLibrary and Material objects in static containers. To resolve the cyclic references the method cleanup() has been added. The class ModelLibrary has a shared pointer to its Model objects and the class Model has a shared pointer to the ModelLibrary. The class ModelManager owns all ModelLibrary and Model objects in static containers. To resolve the cyclic references the method cleanup() has been added. The Materials module registers a function to App::CleanupProcess that calls the above cleanup() methods. NOTE: This registration is only done in debug mode mainly to satisfy memory checkers
158 lines
4.6 KiB
C++
158 lines
4.6 KiB
C++
/***************************************************************************
|
|
* Copyright (c) 2023 David Carter <dcarter@david.carter.ca> *
|
|
* *
|
|
* This file is part of FreeCAD. *
|
|
* *
|
|
* FreeCAD is free software: you can redistribute it and/or modify it *
|
|
* under the terms of the GNU Lesser General Public License as *
|
|
* published by the Free Software Foundation, either version 2.1 of the *
|
|
* License, or (at your option) any later version. *
|
|
* *
|
|
* FreeCAD 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 FreeCAD. If not, see *
|
|
* <https://www.gnu.org/licenses/>. *
|
|
* *
|
|
**************************************************************************/
|
|
|
|
#include "PreCompiled.h"
|
|
#ifndef _PreComp_
|
|
#endif
|
|
|
|
#include <QDirIterator>
|
|
#include <QMutexLocker>
|
|
|
|
#include <Base/Console.h>
|
|
|
|
#include "Model.h"
|
|
#include "ModelLoader.h"
|
|
#include "ModelManager.h"
|
|
|
|
|
|
using namespace Materials;
|
|
|
|
std::shared_ptr<std::list<std::shared_ptr<ModelLibrary>>> ModelManager::_libraryList = nullptr;
|
|
std::shared_ptr<std::map<QString, std::shared_ptr<Model>>> ModelManager::_modelMap = nullptr;
|
|
QMutex ModelManager::_mutex;
|
|
|
|
TYPESYSTEM_SOURCE(Materials::ModelManager, Base::BaseClass)
|
|
|
|
ModelManager::ModelManager()
|
|
{
|
|
initLibraries();
|
|
}
|
|
|
|
void ModelManager::initLibraries()
|
|
{
|
|
QMutexLocker locker(&_mutex);
|
|
|
|
if (_modelMap == nullptr) {
|
|
_modelMap = std::make_shared<std::map<QString, std::shared_ptr<Model>>>();
|
|
if (_libraryList == nullptr) {
|
|
_libraryList = std::make_shared<std::list<std::shared_ptr<ModelLibrary>>>();
|
|
}
|
|
|
|
// Load the libraries
|
|
ModelLoader loader(_modelMap, _libraryList);
|
|
}
|
|
}
|
|
|
|
bool ModelManager::isModel(const QString& file)
|
|
{
|
|
// if (!fs::is_regular_file(p))
|
|
// return false;
|
|
// check file extension
|
|
if (file.endsWith(QString::fromStdString(".yml"))) {
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
void ModelManager::cleanup()
|
|
{
|
|
if (_libraryList) {
|
|
_libraryList->clear();
|
|
}
|
|
|
|
if (_modelMap) {
|
|
for (auto& it : *_modelMap) {
|
|
// This is needed to resolve cyclic dependencies
|
|
it.second->setLibrary(nullptr);
|
|
}
|
|
_modelMap->clear();
|
|
}
|
|
}
|
|
|
|
void ModelManager::refresh()
|
|
{
|
|
_modelMap->clear();
|
|
_libraryList->clear();
|
|
|
|
// Load the libraries
|
|
ModelLoader loader(_modelMap, _libraryList);
|
|
}
|
|
|
|
std::shared_ptr<Model> ModelManager::getModel(const QString& uuid) const
|
|
{
|
|
try {
|
|
if (_modelMap == nullptr) {
|
|
throw Uninitialized();
|
|
}
|
|
|
|
return _modelMap->at(uuid);
|
|
}
|
|
catch (std::out_of_range const&) {
|
|
throw ModelNotFound();
|
|
}
|
|
}
|
|
|
|
std::shared_ptr<Model> ModelManager::getModelByPath(const QString& path) const
|
|
{
|
|
QString cleanPath = QDir::cleanPath(path);
|
|
|
|
for (auto& library : *_libraryList) {
|
|
if (cleanPath.startsWith(library->getDirectory())) {
|
|
return library->getModelByPath(cleanPath);
|
|
}
|
|
}
|
|
|
|
throw MaterialNotFound();
|
|
}
|
|
|
|
std::shared_ptr<Model> ModelManager::getModelByPath(const QString& path, const QString& lib) const
|
|
{
|
|
auto library = getLibrary(lib); // May throw LibraryNotFound
|
|
return library->getModelByPath(path); // May throw ModelNotFound
|
|
}
|
|
|
|
std::shared_ptr<ModelLibrary> ModelManager::getLibrary(const QString& name) const
|
|
{
|
|
for (auto& library : *_libraryList) {
|
|
if (library->getName() == name) {
|
|
return library;
|
|
}
|
|
}
|
|
|
|
throw LibraryNotFound();
|
|
}
|
|
|
|
bool ModelManager::passFilter(ModelFilter filter, Model::ModelType modelType)
|
|
{
|
|
switch (filter) {
|
|
case ModelFilter_None:
|
|
return true;
|
|
|
|
case ModelFilter_Physical:
|
|
return (modelType == Model::ModelType_Physical);
|
|
|
|
case ModelFilter_Appearance:
|
|
return (modelType == Model::ModelType_Appearance);
|
|
}
|
|
|
|
return false;
|
|
}
|