flower/source/editor/widgets/component_draw/component_pmx.cpp

#include "component_draw.h"
#include <imgui/ui.h>
#include <scene/scene.h>
#include "../editor/editor.h"
#include "../editor/widgets/project_content.h"
#include <scene/component/pmx.h>

using namespace engine;
using namespace engine::ui;

void drawPMXSelect(
	std::shared_ptr<SceneNode> node,
	std::shared_ptr<PMXComponent> comp)
{
	auto* assetSystem = Editor::get()->getAssetSystem();
	auto* context = Editor::get()->getContext();

	if (ImGui::BeginMenu("Project"))
	{
		const auto& map = assetSystem->getAssetMap(EAssetType::PMX);
		for (const auto& id : map)
		{
			const auto& asset = assetSystem->getAsset(id);
			if (ImGui::MenuItem((std::string("  ") + ICON_FA_CHESS_QUEEN"   " + asset->getRelativePathUtf8()).c_str()))
			{
				comp->setPMX(id);
			}
		}
		ImGui::EndMenu();
	}

}

void drawVMDSelect(
	std::shared_ptr<SceneNode> node,
	std::shared_ptr<PMXComponent> comp)
{
	auto* assetSystem = Editor::get()->getAssetSystem();
	auto* context = Editor::get()->getContext();

	if (ImGui::BeginMenu("Project"))
	{
		const auto& map = assetSystem->getAssetMap(EAssetType::VMD);
		for (const auto& id : map)
		{
			const auto& asset =std::dynamic_pointer_cast<AssetVMD>(assetSystem->getAsset(id));

			if (!asset->m_bCamera && ImGui::MenuItem((std::string("  ") + ICON_FA_PERSON_WALKING"   " + asset->getRelativePathUtf8()).c_str()))
			{
				// TODO: multi vmd file.
				comp->clearVmd();
				comp->addVmd(id);
			}
		}
		ImGui::EndMenu();
	}

}

void ComponentDrawer::drawMMDCamera(std::shared_ptr<engine::SceneNode> node)
{
	std::shared_ptr<MMDCameraComponent> comp = node->getComponent<MMDCameraComponent>();

	ProjectContentWidget* contentWidget = Editor::get()->getContentWidget();
	auto* assetSystem = Editor::get()->getAssetSystem();

	ImGui::Unindent();
	ui::beginGroupPanel("Vmd");
	{
		static const std::string selectVmdButtonName = " VMD Chose ";
		if (ImGui::Button(selectVmdButtonName.c_str()))
			ImGui::OpenPopup("VMDSelectPopUp");
		if (ImGui::BeginPopup("VMDSelectPopUp"))
		{
			ImGui::TextDisabled("Select VMD...");
			ImGui::Spacing();

			{
				auto* assetSystem = Editor::get()->getAssetSystem();
				auto* context = Editor::get()->getContext();

				if (ImGui::BeginMenu("Project"))
				{
					const auto& map = assetSystem->getAssetMap(EAssetType::VMD);
					for (const auto& id : map)
					{
						const auto& asset = std::dynamic_pointer_cast<AssetVMD>(assetSystem->getAsset(id));

						if (asset->m_bCamera && ImGui::MenuItem((std::string("  ") + ICON_FA_CAMERA"   " + asset->getRelativePathUtf8()).c_str()))
						{
							comp->setVmd(id);
						}
					}
					ImGui::EndMenu();
				}
			}
			ImGui::EndPopup();
		}
	}
	ui::endGroupPanel();
	ImGui::Indent();
}

void drawWaveSelect(
	std::shared_ptr<SceneNode> node,
	std::shared_ptr<PMXComponent> comp)
{
	auto* assetSystem = Editor::get()->getAssetSystem();
	auto* context = Editor::get()->getContext();

	if (ImGui::BeginMenu("Project"))
	{
		const auto& map = assetSystem->getAssetMap(EAssetType::Wave);
		for (const auto& id : map)
		{
			const auto& asset = std::dynamic_pointer_cast<AssetWave>(assetSystem->getAsset(id));

			if (ImGui::MenuItem((std::string("  ") + ICON_FA_MUSIC"   " + asset->getRelativePathUtf8()).c_str()))
			{
				comp->setSong(id);
			}
		}
		ImGui::EndMenu();
	}

}

void ComponentDrawer::drawPMX(std::shared_ptr<engine::SceneNode> node)
{
	std::shared_ptr<PMXComponent> comp = node->getComponent<PMXComponent>();

	ProjectContentWidget* contentWidget = Editor::get()->getContentWidget();
	auto* assetSystem = Editor::get()->getAssetSystem();

	ImGui::Unindent();
	ui::beginGroupPanel("Basic");
	{
		static const std::string selectButtonName = " PMX Chose ";
		if (ImGui::Button(selectButtonName.c_str()))
			ImGui::OpenPopup("PMXSelectPopUp");
		if (ImGui::BeginPopup("PMXSelectPopUp"))
		{
			ImGui::TextDisabled("Select PMX...");
			ImGui::Spacing();

			drawPMXSelect(node, comp);
			ImGui::EndPopup();
		}

		static const std::string selectVmdButtonName = " VMD Chose ";
		if (ImGui::Button(selectVmdButtonName.c_str()))
			ImGui::OpenPopup("VMDSelectPopUp");
		if (ImGui::BeginPopup("VMDSelectPopUp"))
		{
			ImGui::TextDisabled("Select VMD...");
			ImGui::Spacing();

			drawVMDSelect(node, comp);
			ImGui::EndPopup();
		}

		static const std::string selectWaveButtonName = " Wave Chose ";
		if (ImGui::Button(selectWaveButtonName.c_str()))
			ImGui::OpenPopup("WaveSelectPopUp");
		if (ImGui::BeginPopup("WaveSelectPopUp"))
		{
			ImGui::TextDisabled("Select Wave...");
			ImGui::Spacing();

			drawWaveSelect(node, comp);
			ImGui::EndPopup();
		}
	}
	ui::endGroupPanel();
	ImGui::Indent();
}