mirror of
https://github.com/IfcOpenShell/IfcOpenShell.git
synced 2026-08-28 15:53:00 +00:00
Dion Moult
4e4553201b
Handle streamer success, failure, and cancellation as distinct terminal states so failed or cancelled loads do not finalize as successful models. Clean up partial model/UI state in the full and minimal viewer apps when a load is cancelled or fails. Generated with the assistance of an AI coding tool.
92 lines
3.2 KiB
C++
92 lines
3.2 KiB
C++
/********************************************************************************
|
|
* *
|
|
* 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/>. *
|
|
* *
|
|
********************************************************************************/
|
|
|
|
#ifndef GEOMETRYSTREAMER_H
|
|
#define GEOMETRYSTREAMER_H
|
|
|
|
#include <QObject>
|
|
#include <QThread>
|
|
|
|
#include <string>
|
|
#include <vector>
|
|
#include <atomic>
|
|
#include <memory>
|
|
#include <mutex>
|
|
|
|
#include "../ifcparse/file.h"
|
|
#include "../ifcgeom/Iterator.h"
|
|
|
|
#include "InstancedGeometry.h"
|
|
|
|
struct ElementInfo {
|
|
uint32_t object_id;
|
|
uint32_t model_id;
|
|
int ifc_id;
|
|
std::string guid;
|
|
std::string name;
|
|
std::string type;
|
|
int parent_id;
|
|
};
|
|
|
|
class GeometryStreamer : public QObject {
|
|
Q_OBJECT
|
|
public:
|
|
explicit GeometryStreamer(QObject* parent = nullptr);
|
|
~GeometryStreamer();
|
|
|
|
void loadFile(const std::string& path, uint32_t start_object_id, uint32_t model_id, int num_threads = 0);
|
|
void cancel();
|
|
|
|
bool isRunning() const { return running_.load(); }
|
|
int progress() const { return progress_.load(); }
|
|
uint32_t lastObjectId() const { return next_object_id_; }
|
|
uint32_t modelId() const { return model_id_; }
|
|
|
|
ifcopenshell::file* ifcFile() const { return ifc_file_.get(); }
|
|
|
|
// Thread-safe access to discovered elements
|
|
std::vector<ElementInfo> drainElements();
|
|
|
|
signals:
|
|
void progressChanged(int percent);
|
|
void meshReady(MeshChunk chunk);
|
|
void instanceReady(InstanceChunk chunk);
|
|
void finished();
|
|
void cancelled();
|
|
void errorOccurred(const QString& message);
|
|
|
|
private:
|
|
void run(const std::string& path, int num_threads);
|
|
|
|
std::unique_ptr<ifcopenshell::file> ifc_file_;
|
|
std::unique_ptr<QThread> worker_thread_;
|
|
std::atomic<bool> running_{false};
|
|
std::atomic<bool> cancel_requested_{false};
|
|
std::atomic<bool> succeeded_{false};
|
|
std::atomic<int> progress_{0};
|
|
|
|
std::mutex elements_mutex_;
|
|
std::vector<ElementInfo> pending_elements_;
|
|
|
|
uint32_t next_object_id_ = 1;
|
|
uint32_t model_id_ = 0;
|
|
};
|
|
|
|
#endif // GEOMETRYSTREAMER_H
|