flower/Source/Engine/AssetSystem/AssetSystem.h

#pragma once
#include "Pch.h"
#include "../RuntimeModule.h"
#include "../Engine.h"
#include "AssetCommon.h"
#include "../Project.h"

namespace Flower
{
	struct ImportTextureOptions
	{
		bool bSrgb = true;
		bool bBuildMipmap = true;
		float cutoff = 1.0f;
		bool bHdr = false;
	};

	struct ImportOptions
	{
		std::optional<ImportTextureOptions> texOptions { };
	};

	class LRUAssetInterface;
	class RegistryEntry;
	class AssetSystem : public IRuntimeModule
	{
	protected:
		bool m_bProjectSetup = false;
		std::filesystem::path m_projectPath;

		std::filesystem::path m_projectAssetEntryPath;
		std::filesystem::path m_projectAssetHeaderFolderPath;
		std::filesystem::path m_projectAssetSceneFolderPath;
		std::filesystem::path m_projectAssetBinFolderPath;

		// Call gpu lru cache shrink next tick.
		std::atomic<bool> m_bCallGPULRUCacheShrink = false;

		std::vector<std::shared_ptr<struct AssetLoadTask>> m_uploadTasks;
		std::vector<std::shared_ptr<LRUAssetInterface>> m_unusedAseets;

		std::atomic<uint64_t> m_runtimeAssetVersion = 0;

	private:
		void engineAssetInit();

		void submitAllUploadTask();

	public:
		bool projectAlreadySet() const 
		{ 
			return m_bProjectSetup;
		}

		void setupProject(const std::filesystem::path& path);

		const auto& getProjectPath() const
		{
			return m_projectPath;
		}

		const auto& getProjectBinFolderPath() const
		{
			return m_projectAssetBinFolderPath;
		}

		const auto& getProjectSceneFolderPath() const
		{
			return m_projectAssetSceneFolderPath;
		}

		auto getProjectPathFilePath() const
		{
			return m_projectPath / ProjectContext::get()->project.getNameWithSuffix();
		}

		const auto& getProjectHeaderFolderPath() const
		{
			return m_projectAssetHeaderFolderPath;
		}

		const auto& getProjectAssetEntryPath() const
		{
			return m_projectAssetEntryPath;
		}

		UUID importAsset(const std::filesystem::path& path, EAssetType type, std::shared_ptr<RegistryEntry> entry, const ImportOptions& inOption = {});

		void markCallGPULRUCacheShrink()
		{
			m_bCallGPULRUCacheShrink = true;
		}

		void addUnusedAsset(std::shared_ptr<LRUAssetInterface> asset);

		void addUploadTask(std::shared_ptr<AssetLoadTask> inTask);

		void flushUploadTask();

	public:
		AssetSystem(ModuleManager* in, std::string name = "AssetSystem");

		virtual bool init() override;
		virtual void release() override;
		virtual void tick(const RuntimeModuleTickData& tickData) override;

	};
}