Interface mockup 5

src/interface/ElementRegistry.cpp

@@ -0,0 +1,82 @@
+// This file was generated with the assistance of an AI coding tool.
+/********************************************************************************
+ *                                                                              *
+ * This file is part of IfcOpenShell.                                           *
+ *                                                                              *
+ * IfcOpenShell is free software: you can redistribute it and/or modify         *
+ * it under the terms of the Lesser GNU General Public License as published by  *
+ * the Free Software Foundation, either version 3.0 of the License, or          *
+ * (at your option) any later version.                                          *
+ *                                                                              *
+ * IfcOpenShell 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                 *
+ * Lesser GNU General Public License for more details.                          *
+ *                                                                              *
+ * You should have received a copy of the Lesser GNU General Public License     *
+ * along with this program. If not, see <http://www.gnu.org/licenses/>.         *
+ *                                                                              *
+ ********************************************************************************/
+
+#include "ElementRegistry.h"
+
+#include "../ifcviewer/GeometryStreamer.h"
+#include "../ifcviewer/SceneLoader.h"
+#include "../ifcviewer/SidecarCache.h"
+
+namespace ifcinterface {
+
+ElementRegistry::ElementRegistry(QObject* parent)
+    : QObject(parent)
+{
+}
+
+void ElementRegistry::bindLoader(SceneLoader* loader) {
+    connect(loader, &SceneLoader::sidecarElementsReady,
+            this, &ElementRegistry::onSidecarElementsReady);
+    connect(loader, &SceneLoader::streamedElementsReady,
+            this, &ElementRegistry::onStreamedElementsReady);
+}
+
+std::optional<BasicElementInfo> ElementRegistry::find(uint32_t object_id) const {
+    auto it = elements_.find(object_id);
+    if (it == elements_.end()) return std::nullopt;
+    return it->second;
+}
+
+void ElementRegistry::onSidecarElementsReady(uint32_t /*mid*/,
+                                             std::vector<PackedElementInfo> elements,
+                                             std::string string_table) {
+    auto str = [&](uint32_t offset, uint32_t length) -> QString {
+        if (length == 0 || offset + length > string_table.size()) return {};
+        return QString::fromStdString(string_table.substr(offset, length));
+    };
+
+    for (const auto& pe : elements) {
+        BasicElementInfo info;
+        info.object_id = pe.object_id;
+        info.model_id = pe.model_id;
+        info.ifc_id = pe.ifc_id;
+        info.parent_id = pe.parent_id;
+        info.guid = str(pe.guid_offset, pe.guid_length);
+        info.name = str(pe.name_offset, pe.name_length);
+        info.type = str(pe.type_offset, pe.type_length);
+        elements_[info.object_id] = info;
+    }
+}
+
+void ElementRegistry::onStreamedElementsReady(uint32_t /*mid*/, std::vector<ElementInfo> elements) {
+    for (const auto& e : elements) {
+        BasicElementInfo info;
+        info.object_id = e.object_id;
+        info.model_id = e.model_id;
+        info.ifc_id = e.ifc_id;
+        info.parent_id = e.parent_id;
+        info.guid = QString::fromStdString(e.guid);
+        info.name = QString::fromStdString(e.name);
+        info.type = QString::fromStdString(e.type);
+        elements_[info.object_id] = info;
+    }
+}
+
+} // namespace ifcinterface