IfcOpenShell/src/bonsaiviewer/SessionState.h

// This file was generated with the assistance of an AI coding tool.
/********************************************************************************
 *                                                                              *
 * This file is part of Bonsai.                                                 *
 *                                                                              *
 * Bonsai is free software: you can redistribute it and/or modify               *
 * it under the terms of the 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.                                          *
 *                                                                              *
 * Bonsai 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 General Public License for more details.                                 *
 *                                                                              *
 * You should have received a copy of the GNU General Public License            *
 * along with this program. If not, see <http://www.gnu.org/licenses/>.         *
 *                                                                              *
 ********************************************************************************/

#ifndef IFCINTERFACE_SESSIONSTATE_H
#define IFCINTERFACE_SESSIONSTATE_H

#include <QHash>
#include <QObject>
#include <QString>
#include <QVariantMap>

class Federation;
class SceneLoader;
class ViewportWindow;

namespace bonsaiviewer {

class ElementRegistry;

namespace modules::connectors { class ConnectorRegistry; }

class SessionState : public QObject {
    Q_OBJECT

public:
    explicit SessionState(QObject* parent = nullptr);

    // Owned by SessionState once it can be tied to a viewport. Wires
    // loader → element registry signals internally. Call exactly once.
    void createLoader(ViewportWindow* viewport);

    Federation* federation() const { return federation_; }
    SceneLoader* loader() const { return loader_; }
    ElementRegistry* elementRegistry() const { return element_registry_; }
    modules::connectors::ConnectorRegistry* connectorRegistry() const { return connector_registry_; }
    QString statusMode() const { return status_mode_; }
    QString statusDetail() const { return status_detail_; }

    void setSelectedObjectId(uint32_t object_id);
    uint32_t selectedObjectId() const { return selected_object_id_; }
    void setStatusMessage(const QString& mode, const QString& detail);

    // Generic progress reporting for any long-running operation (load,
    // convert, export, ...). Subscribers (the status bar) react to the
    // signals; they do not need to know which operation is running.
    void beginProgress(const QString& label);
    void setProgress(int percent);
    void endProgress();

    void setModelMapping(const QString& model_id, uint32_t session_model_id);
    void removeModelMappingByModelId(const QString& model_id);
    void clearModelMappings();

    // Per-session cloud metadata returned by connectors (revision/date/
    // author/...). Not persisted to the .ifcfed; display only. Lifetime
    // is tied to the model_id — removeModelMappingByModelId and
    // clearModelMappings drop the matching entries.
    void setCloudMetadata(const QString& model_id, const QVariantMap& metadata);
    QVariantMap cloudMetadata(const QString& model_id) const;
    uint32_t sessionModelIdForModelId(const QString& model_id) const;
    QString modelIdForSessionModelId(uint32_t session_model_id) const;
    QList<uint32_t> sessionModelIds() const;

    // The active model — the single model the spatial hierarchy (and other
    // model-scoped views) operate on. Set by clicking a model in the models
    // panel; defaults to the first loaded model. Empty when no model is loaded.
    QString activeModelId() const { return active_model_id_; }
    void setActiveModelId(const QString& model_id);

    void notifySelectionChanged();
    void notifyModelsChanged();
    void notifyFederationChanged();
    void notifyVisibilityChanged();
    void notifyModelGeometryReady(uint32_t session_model_id);
    void notifyProjectOpened(const QString& path);
    void notifyProjectSaved(const QString& path);
    void notifyProjectReset();

signals:
    void projectOpened(const QString& path);
    void projectSaved(const QString& path);
    void projectReset();
    void modelsChanged();
    // Fires whenever a command has mutated the federation (groups, transforms,
    // origin, config). Always implies the project is now dirty; callers do not
    // emit this on save/load/reset — projectSaved/Opened/Reset cover those.
    void federationChanged();
    void visibilityChanged();
    // Fires when a model's geometry has been pushed to the viewport. Fires
    // for both sidecar-cache and stream loads; subscribers that just need to
    // re-derive view state (e.g. ViewportView::refresh) listen to this.
    void modelGeometryReady(uint32_t session_model_id);
    // Fires when a model's live IFC data source (the .ifc/.rdb, opened in the
    // background after a sidecar-cache hit) becomes available for queries —
    // e.g. so the spatial hierarchy can be built once the file is loaded.
    void modelDataSourceReady(uint32_t session_model_id);
    // Fires when the active model changes (empty model_id when cleared).
    void activeModelChanged(const QString& model_id);
    // Fires when SceneLoader reports a load failure. SessionState turns the
    // raw loader signal into a session-level one so views (e.g. the MessageBox)
    // can subscribe without touching the loader directly.
    void loadError(const QString& message);
    void selectionChanged(uint32_t object_id);
    void statusMessageChanged(const QString& mode, const QString& detail);
    void progressBegan(const QString& label);
    void progressChanged(int percent);
    void progressEnded();

private:
    Federation* federation_ = nullptr;
    SceneLoader* loader_ = nullptr;
    ElementRegistry* element_registry_ = nullptr;
    modules::connectors::ConnectorRegistry* connector_registry_ = nullptr;
    uint32_t selected_object_id_ = 0;
    QString status_mode_;
    QString status_detail_;
    QHash<QString, uint32_t> model_id_to_session_model_id_;
    QHash<uint32_t, QString> session_model_id_to_model_id_;
    QString active_model_id_;
    QHash<QString, QVariantMap> cloud_metadata_;
};

} // namespace bonsaiviewer

#endif