Thu, 03 Mar 2022 21:13:16 +0200
Clean up Model
24 | 1 | /* |
2 | * LDForge: LDraw parts authoring CAD | |
3 | * Copyright (C) 2013 - 2020 Teemu Piippo | |
4 | * | |
5 | * This program is free software: you can redistribute it and/or modify | |
6 | * it under the terms of the GNU General Public License as published by | |
7 | * the Free Software Foundation, either version 3 of the License, or | |
8 | * (at your option) any later version. | |
9 | * | |
10 | * This program is distributed in the hope that it will be useful, | |
11 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
12 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
13 | * GNU General Public License for more details. | |
14 | * | |
15 | * You should have received a copy of the GNU General Public License | |
16 | * along with this program. If not, see <http://www.gnu.org/licenses/>. | |
17 | */ | |
18 | ||
3 | 19 | #pragma once |
8
44679e468ba9
major update with many things
Teemu Piippo <teemu@hecknology.net>
parents:
6
diff
changeset
|
20 | #include <QAbstractListModel> |
3 | 21 | #include <memory> |
22 | #include "main.h" | |
23 | #include "header.h" | |
14 | 24 | #include "linetypes/object.h" |
141 | 25 | #include "linetypes/metacommand.h" |
21 | 26 | #include "gl/common.h" |
3 | 27 | |
8
44679e468ba9
major update with many things
Teemu Piippo <teemu@hecknology.net>
parents:
6
diff
changeset
|
28 | enum class HeaderProperty |
44679e468ba9
major update with many things
Teemu Piippo <teemu@hecknology.net>
parents:
6
diff
changeset
|
29 | { |
44679e468ba9
major update with many things
Teemu Piippo <teemu@hecknology.net>
parents:
6
diff
changeset
|
30 | Name |
44679e468ba9
major update with many things
Teemu Piippo <teemu@hecknology.net>
parents:
6
diff
changeset
|
31 | }; |
44679e468ba9
major update with many things
Teemu Piippo <teemu@hecknology.net>
parents:
6
diff
changeset
|
32 | |
44679e468ba9
major update with many things
Teemu Piippo <teemu@hecknology.net>
parents:
6
diff
changeset
|
33 | class Model : public QAbstractListModel |
3 | 34 | { |
35 | Q_OBJECT | |
36 | public: | |
37 | class EditContext; | |
21 | 38 | Model(QObject* parent = nullptr); |
5 | 39 | Model(const Model&) = delete; |
150
b6cbba6e29a1
extract polygon cache out of Model
Teemu Piippo <teemu@hecknology.net>
parents:
148
diff
changeset
|
40 | |
3 | 41 | int size() const; |
133
e39326ee48dc
Begin work on edit history
Teemu Piippo <teemu@hecknology.net>
parents:
117
diff
changeset
|
42 | ldraw::id_t at(int index) const; |
3 | 43 | EditContext edit(); |
8
44679e468ba9
major update with many things
Teemu Piippo <teemu@hecknology.net>
parents:
6
diff
changeset
|
44 | int rowCount(const QModelIndex&) const override; |
44679e468ba9
major update with many things
Teemu Piippo <teemu@hecknology.net>
parents:
6
diff
changeset
|
45 | QVariant data(const QModelIndex& index, int role) const override; |
151 | 46 | ldraw::Object* findObjectById(const ldraw::id_t id); |
47 | const ldraw::Object* findObjectById(const ldraw::id_t id) const; | |
150
b6cbba6e29a1
extract polygon cache out of Model
Teemu Piippo <teemu@hecknology.net>
parents:
148
diff
changeset
|
48 | QModelIndex find(ldraw::id_t id) const; |
b6cbba6e29a1
extract polygon cache out of Model
Teemu Piippo <teemu@hecknology.net>
parents:
148
diff
changeset
|
49 | ldraw::id_t idAt(const QModelIndex& index) const; |
141 | 50 | template<typename R> |
116 | 51 | const R* get(ldraw::Id<R> id) const; |
52 | template<typename R> | |
53 | struct Get2Result | |
54 | { | |
55 | QModelIndex index; | |
56 | const R* object; | |
57 | }; | |
58 | template<typename R> | |
59 | Get2Result<R> get2(ldraw::Id<R> id) const; | |
150
b6cbba6e29a1
extract polygon cache out of Model
Teemu Piippo <teemu@hecknology.net>
parents:
148
diff
changeset
|
60 | ldraw::Object* operator[](int index); |
151 | 61 | const ldraw::Object* operator[](int index) const; |
112 | 62 | Q_SIGNALS: |
150
b6cbba6e29a1
extract polygon cache out of Model
Teemu Piippo <teemu@hecknology.net>
parents:
148
diff
changeset
|
63 | void objectAdded(int position); |
b6cbba6e29a1
extract polygon cache out of Model
Teemu Piippo <teemu@hecknology.net>
parents:
148
diff
changeset
|
64 | void objectModified(int position); |
b6cbba6e29a1
extract polygon cache out of Model
Teemu Piippo <teemu@hecknology.net>
parents:
148
diff
changeset
|
65 | void objectRemoved(ldraw::id_t id); |
3 | 66 | private: |
35
98906a94732f
renamed the linetypes namespace to ldraw namespace and added more structures to it
Teemu Piippo <teemu@hecknology.net>
parents:
26
diff
changeset
|
67 | using ModelObjectPointer = std::unique_ptr<ldraw::Object>; |
3 | 68 | template<typename T, typename... Args> |
73
97df974b5ed5
ldraw::Id is now templated for extra type safety
Teemu Piippo <teemu@hecknology.net>
parents:
51
diff
changeset
|
69 | ldraw::Id<T> append(Args&&... args); |
8
44679e468ba9
major update with many things
Teemu Piippo <teemu@hecknology.net>
parents:
6
diff
changeset
|
70 | void append(ModelObjectPointer&& object); |
3 | 71 | template<typename T, typename... Args> |
76
7c4a63a02632
finished splitQuadrilateral theoretically (untested)
Teemu Piippo <teemu@hecknology.net>
parents:
73
diff
changeset
|
72 | ldraw::Id<T> insert(std::size_t position, Args&&... args); |
7c4a63a02632
finished splitQuadrilateral theoretically (untested)
Teemu Piippo <teemu@hecknology.net>
parents:
73
diff
changeset
|
73 | void remove(int position); |
133
e39326ee48dc
Begin work on edit history
Teemu Piippo <teemu@hecknology.net>
parents:
117
diff
changeset
|
74 | void editFinished(); |
e39326ee48dc
Begin work on edit history
Teemu Piippo <teemu@hecknology.net>
parents:
117
diff
changeset
|
75 | void objectModified(ldraw::id_t id); |
3 | 76 | bool modified = false; |
77 | std::vector<ModelObjectPointer> body; | |
73
97df974b5ed5
ldraw::Id is now templated for extra type safety
Teemu Piippo <teemu@hecknology.net>
parents:
51
diff
changeset
|
78 | std::map<ldraw::id_t, ldraw::Object*> objectsById; |
136 | 79 | /** |
80 | * @brief Amount of model edit contexts active | |
81 | */ | |
133
e39326ee48dc
Begin work on edit history
Teemu Piippo <teemu@hecknology.net>
parents:
117
diff
changeset
|
82 | int editCounter = 0; |
3 | 83 | }; |
84 | ||
150
b6cbba6e29a1
extract polygon cache out of Model
Teemu Piippo <teemu@hecknology.net>
parents:
148
diff
changeset
|
85 | void makeUnofficial(Model& model); |
151 | 86 | void save(const Model& model, QIODevice *device); |
141 | 87 | |
88 | /** | |
138 | 89 | * @brief Calls the specified function to all matching objects in the model |
90 | * @tparam R Type of LDraw line type object to filter by | |
91 | * @param fn Function to call. | |
117 | 92 | */ |
93 | template<typename R, typename Fn> | |
151 | 94 | void applyToModel(const Model& model, Fn&& f) |
117 | 95 | { |
151 | 96 | for (int i = 0; i < model.size(); i += 1) |
117 | 97 | { |
151 | 98 | const ldraw::Object* object = model[i]; |
99 | const R* subobject = dynamic_cast<const R*>(object); | |
117 | 100 | if (subobject != nullptr) |
101 | { | |
102 | f(subobject); | |
103 | } | |
104 | } | |
105 | } | |
106 | ||
3 | 107 | template<typename T, typename... Args> |
73
97df974b5ed5
ldraw::Id is now templated for extra type safety
Teemu Piippo <teemu@hecknology.net>
parents:
51
diff
changeset
|
108 | ldraw::Id<T> Model::append(Args&&... args) |
3 | 109 | { |
111
1f42c03fafca
Draw tool actually adds objects now
Teemu Piippo <teemu@hecknology.net>
parents:
81
diff
changeset
|
110 | const int position = static_cast<int>(this->body.size()); |
112 | 111 | Q_EMIT beginInsertRows({}, position, position); |
3 | 112 | this->body.push_back(std::make_unique<T>(args...)); |
35
98906a94732f
renamed the linetypes namespace to ldraw namespace and added more structures to it
Teemu Piippo <teemu@hecknology.net>
parents:
26
diff
changeset
|
113 | ldraw::Object* pointer = this->body.back().get(); |
6 | 114 | this->objectsById[pointer->id] = pointer; |
150
b6cbba6e29a1
extract polygon cache out of Model
Teemu Piippo <teemu@hecknology.net>
parents:
148
diff
changeset
|
115 | Q_EMIT objectAdded(static_cast<int>(this->body.size() - 1)); |
112 | 116 | Q_EMIT endInsertRows(); |
76
7c4a63a02632
finished splitQuadrilateral theoretically (untested)
Teemu Piippo <teemu@hecknology.net>
parents:
73
diff
changeset
|
117 | return ldraw::Id<T>{pointer->id.value}; |
3 | 118 | } |
119 | ||
120 | template<typename T, typename... Args> | |
111
1f42c03fafca
Draw tool actually adds objects now
Teemu Piippo <teemu@hecknology.net>
parents:
81
diff
changeset
|
121 | ldraw::Id<T> Model::insert(const std::size_t position, Args&&... args) |
3 | 122 | { |
112 | 123 | Q_EMIT beginInsertRows({}, position, position); |
76
7c4a63a02632
finished splitQuadrilateral theoretically (untested)
Teemu Piippo <teemu@hecknology.net>
parents:
73
diff
changeset
|
124 | this->body.insert(std::begin(this->body) + position, std::make_unique<T>(args...)); |
35
98906a94732f
renamed the linetypes namespace to ldraw namespace and added more structures to it
Teemu Piippo <teemu@hecknology.net>
parents:
26
diff
changeset
|
125 | ldraw::Object* pointer = this->body[position].get(); |
6 | 126 | this->objectsById[pointer->id] = pointer; |
150
b6cbba6e29a1
extract polygon cache out of Model
Teemu Piippo <teemu@hecknology.net>
parents:
148
diff
changeset
|
127 | Q_EMIT objectAdded(static_cast<int>(position)); |
112 | 128 | Q_EMIT endInsertRows(); |
76
7c4a63a02632
finished splitQuadrilateral theoretically (untested)
Teemu Piippo <teemu@hecknology.net>
parents:
73
diff
changeset
|
129 | return ldraw::Id<T>{pointer->id.value}; |
3 | 130 | } |
73
97df974b5ed5
ldraw::Id is now templated for extra type safety
Teemu Piippo <teemu@hecknology.net>
parents:
51
diff
changeset
|
131 | |
97df974b5ed5
ldraw::Id is now templated for extra type safety
Teemu Piippo <teemu@hecknology.net>
parents:
51
diff
changeset
|
132 | template<typename R> |
116 | 133 | const R* Model::get(ldraw::Id<R> id) const |
73
97df974b5ed5
ldraw::Id is now templated for extra type safety
Teemu Piippo <teemu@hecknology.net>
parents:
51
diff
changeset
|
134 | { |
116 | 135 | return this->get2(id).object; |
136 | } | |
137 | ||
138 | template<typename R> | |
139 | Model::Get2Result<R> Model::get2(const ldraw::Id<R> id) const | |
140 | { | |
141 | Get2Result<R> result; | |
150
b6cbba6e29a1
extract polygon cache out of Model
Teemu Piippo <teemu@hecknology.net>
parents:
148
diff
changeset
|
142 | result.index = this->find(id); |
116 | 143 | if (result.index.isValid()) |
76
7c4a63a02632
finished splitQuadrilateral theoretically (untested)
Teemu Piippo <teemu@hecknology.net>
parents:
73
diff
changeset
|
144 | { |
151 | 145 | result.object = static_cast<const R*>((*this)[result.index.row()]); |
81
62373840e33a
object editor widgets start to form up
Teemu Piippo <teemu@hecknology.net>
parents:
76
diff
changeset
|
146 | } |
62373840e33a
object editor widgets start to form up
Teemu Piippo <teemu@hecknology.net>
parents:
76
diff
changeset
|
147 | else |
62373840e33a
object editor widgets start to form up
Teemu Piippo <teemu@hecknology.net>
parents:
76
diff
changeset
|
148 | { |
116 | 149 | result.object = nullptr; |
81
62373840e33a
object editor widgets start to form up
Teemu Piippo <teemu@hecknology.net>
parents:
76
diff
changeset
|
150 | } |
116 | 151 | return result; |
73
97df974b5ed5
ldraw::Id is now templated for extra type safety
Teemu Piippo <teemu@hecknology.net>
parents:
51
diff
changeset
|
152 | } |