DYTSrouce/src/workspace/WorkSpace.h

#pragma once

#include <map>
#include <memory>
#include <cstdint>
#include <QObject>

#include <osgEarth/Viewpoint>

#include "scene/OsgScene.h"
#include "scene/OEScene.h"
#include "config.h"
#include "common/SpdLogger.h"
#include "workspace/FileEntry.h"

//#include "../ui/chartPlot/DYTChart.h"

class WorkSpaceItem;
class CommandManager;

class WorkSpace : public QObject {
	Q_OBJECT

public:
	explicit WorkSpace(QObject* parent = nullptr) noexcept;
    WorkSpace(const QString& path, QObject* parent = nullptr);
	~WorkSpace() override = default;

	inline void SetName(const QString& name) {
		name_ = name;
	}
	inline const QString& GetName() const {
		return name_;
	}
	inline const QString& GetPath() const {
        return path_;
    }
	const QString GetDir() const;

    inline void SetUUid(const QString& uuid) {
        uuid_ = uuid;
    }
    inline const QString& GetUUid() const {
        return uuid_;
    }
//	static WorkSpace* Create(const QString& name);
    inline void SetDescribe(const QString& describe) {
		describe_ = describe;
    }
	inline const QString& GetDescribe() const {
		return describe_;
	}

	void SetCommondFilePath(const QString& path);
	const QString GetCommondFilePath() const {
		return commondPath_;
	}

    // Execute command xml according to trigger
    enum class CommandWhen { OnCreate, OnLoad };
    void ExecuteCommands(CommandWhen when);

    // Deprecated path APIs removed: SimMatlab/MatlabParam/Wave/Report/RD

	// Files list API (per-type, max 9 per type)
    enum class FileEntryResult { Ok, LimitExceeded, Duplicate, CopyFailed, TypeMismatch, InvalidFile };
    
    // New unified file entry management
    FileEntryResult SetFileEntry(std::shared_ptr<FileEntry> fileEntry, bool is_copy = true);
    
	std::vector<std::shared_ptr<FileEntry>> GetFileEntries(FileEntryType type) const;

    // Manage grouped file entries
    bool SetFileEntryCount(FileEntryType type, int count);
    QString GetFileEntryAbsPath(FileEntryType type, int index) const;

	inline void SetHomeViewpoint(const osgEarth::Viewpoint& viewpoint) {
        homeViewpoint_ = viewpoint;
		homeViewpoint_.setHeading(0.0); // Ensure heading is set to 0.0
	}
    inline const osgEarth::Viewpoint& GetHomeViewpoint() const {
        return homeViewpoint_;
    }

	const std::vector<class Entity*>& GetEntities() const {
		return entities_;
	}
    std::vector<class Entity*>& GetEntities() {
        return entities_;
    }

	void SetActiveScene(OEScene* scene);

	OEScene* GetActiveScene() const {
		dyt_check(nullptr != scene_);
		return scene_;
	}
	bool SetTimestep(class Timestep* timestep);
	bool SetTimestepPath(const QString& path);
    class Timestep* GetTimestep() const {
		return timestep_;
    }

    bool SetLampStatus(class LampStatus* timestep);
    bool SetLampPath(const QString& path);
    class LampStatus* GetLampStatus() const {
        return lampStatus_;
    }

	void AddEntity(class Entity* entity);
	void RemoveEntity(class Entity* entity);
	bool TrackEntity(class Entity* entity);
	void UntrackEntity();
    class Entity* GetTrackEntity() const {
        return trackedEntity_;
    }

	bool Save(const QString& path);
	bool Save();

	bool Load(const QString& dyt);
	void Unlaod();

	void Begin();
	void OnFrame(double dt);
	void End();

    void OnLoaded();

    // 通知：某类型的文件条目属性已更新（不改变数量，仅触发刷新）
    void NotifyFileEntryUpdated(FileEntryType type, std::shared_ptr<FileEntry> fileEntry = nullptr);

Q_SIGNALS:
    void EntityAdded(class Entity* entity);
    void EntityRemoved(class Entity* entity);
    void TimestepChanged(class Timestep* timestep);
    void LampStatusChanged(class LampStatus* lampStatus);
    void FilesChanged(FileEntryType type, std::shared_ptr<FileEntry> fileEntry);

private:
    QString name_;
    QString uuid_;
    QString describe_;
    QString path_;
    QString commondPath_;
    // Removed deprecated path members: simMatlabPath_, waveFile_, reportFile_, rdFile_, matlabParamPath_

    osgEarth::Viewpoint homeViewpoint_;

	bool leaded_{ false };
    std::vector<class Entity*> entities_;
    OEScene* scene_{ nullptr };
    class Timestep* timestep_{ nullptr };
    class LampStatus* lampStatus_{ nullptr }; 
    class Entity* trackedEntity_{ nullptr };
    // Stored as file entries under workspace dir, keyed by type
    std::map<FileEntryType, std::vector<std::shared_ptr<FileEntry>>> files_;
    // Monotonic sequence for file entries changes, used to trigger UI refresh
    std::uint64_t filesSeq_{ 0 };
    // Executor for command XML actions
    std::unique_ptr<CommandManager> cmdMgr_;
public:
    std::uint64_t GetFilesSeq() const { return filesSeq_; }
	friend class WorkSpaceXMLWrite;
	friend class WorkSpaceXMLParse;
};