File size: 3,777 Bytes
985c397 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 | // SPDX-License-Identifier: LGPL-2.1-or-later
/***************************************************************************
* 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/>. *
* *
**************************************************************************/
#ifndef MATERIAL_FOLDERTREE_H
#define MATERIAL_FOLDERTREE_H
#include <map>
#include <memory>
#include <QString>
namespace Materials
{
template<class T>
class FolderTreeNode
{
public:
enum class NodeType
{
UnknownNode,
DataNode,
FolderNode
};
FolderTreeNode()
: _type(NodeType::UnknownNode)
, _oldFormat(false)
, _readOnly(false)
{}
virtual ~FolderTreeNode() = default;
NodeType getType() const
{
// assert(_type == NodeType::DataNode || _type == NodeType::FolderNode);
return _type;
}
void setType(NodeType type)
{
_type = type;
}
std::shared_ptr<std::map<QString, std::shared_ptr<FolderTreeNode<T>>>> getFolder() const
{
assert(_type == NodeType::FolderNode);
return _folder;
}
std::shared_ptr<std::map<QString, std::shared_ptr<FolderTreeNode<T>>>> getFolder()
{
assert(_type == NodeType::FolderNode);
return _folder;
}
std::shared_ptr<T> getData() const
{
assert(_type == NodeType::DataNode);
return _data;
}
QString getUUID() const
{
assert(_type == NodeType::DataNode);
return _uuid;
}
bool isOldFormat() const
{
return _oldFormat;
}
bool isReadOnly() const
{
return _readOnly;
}
void setFolder(std::shared_ptr<std::map<QString, std::shared_ptr<FolderTreeNode<T>>>> folder)
{
setType(NodeType::FolderNode);
_folder = folder;
}
void setData(std::shared_ptr<T> data)
{
setType(NodeType::DataNode);
_data = data;
}
void setUUID(const QString& uuid)
{
setType(NodeType::DataNode);
_uuid = uuid;
}
void setOldFormat(bool oldFormat)
{
_oldFormat = oldFormat;
}
void setReadOnly(bool readOnly)
{
_readOnly = readOnly;
}
private:
NodeType _type;
std::shared_ptr<std::map<QString, std::shared_ptr<FolderTreeNode<T>>>> _folder;
QString _uuid;
std::shared_ptr<T> _data;
bool _oldFormat;
bool _readOnly;
};
} // namespace Materials
#endif // MATERIAL_FOLDERTREE_H
|