// Copyright (c), Firelight Technologies Pty, Ltd. 2012-2016.

#include "FMODStudioEditorPrivatePCH.h"
#include "FMODEventEditor.h"
#include "FMODEvent.h"
#include "FMODStudioModule.h"
#include "FMODUtils.h"
#include "SFMODEventEditorPanel.h"
#include "SDockTab.h"
//#include "WorkspaceMenuStructureModule.h"
#include "fmod_studio.hpp"

#define LOCTEXT_NAMESPACE "FMODEventEditor"

DEFINE_LOG_CATEGORY_STATIC(LogFMODEventEditor, Log, All);

const FName FFMODEventEditor::EventEditorTabId(TEXT("FFMODEventEditor_EventView"));
const FName FFMODEventEditor::FMODEventEditorAppIdentifier(TEXT("FMODEventEditorApp"));

void FFMODEventEditor::RegisterTabSpawners(const TSharedRef<class FTabManager>& TabManager)
{
	WorkspaceMenuCategory = TabManager->AddLocalWorkspaceMenuCategory(LOCTEXT("WorkspaceMenu_FMODEventEditor", "FMOD Event Editor"));
	auto WorkspaceMenuCategoryRef = WorkspaceMenuCategory.ToSharedRef();

	FAssetEditorToolkit::RegisterTabSpawners(TabManager);

	TabManager->RegisterTabSpawner(
		EventEditorTabId,
		FOnSpawnTab::CreateSP(this, &FFMODEventEditor::SpawnTab_EventEditor))
		.SetDisplayName(LOCTEXT("EventTab", "FMOD Event"))
		.SetGroup(WorkspaceMenuCategoryRef);
}

void FFMODEventEditor::UnregisterTabSpawners(const TSharedRef<class FTabManager>& TabManager)
{
	FAssetEditorToolkit::UnregisterTabSpawners(TabManager);

	TabManager->UnregisterTabSpawner(EventEditorTabId);
}

FFMODEventEditor::FFMODEventEditor()
	: CurrentPreviewEventInstance(nullptr)
{
	IFMODStudioModule::Get().BanksReloadedEvent().AddRaw(this, &FFMODEventEditor::HandleBanksReloaded);
	BeginPIEDelegateHandle = FEditorDelegates::BeginPIE.AddRaw(this, &FFMODEventEditor::HandleBeginPIE);
}

FFMODEventEditor::~FFMODEventEditor()
{
	IFMODStudioModule::Get().BanksReloadedEvent().RemoveAll(this);
	FEditorDelegates::BeginPIE.Remove(BeginPIEDelegateHandle);

	CurrentPreviewEventInstance = nullptr;
}

UFMODEvent* FFMODEventEditor::GetEditedEvent() const
{
	return EditedEvent;
}

FMOD::Studio::EventDescription* FFMODEventEditor::GetEventDescription() const
{
	return IFMODStudioModule::Get().GetEventDescription(EditedEvent, EFMODSystemContext::Auditioning);
}

void FFMODEventEditor::PlayEvent()
{
	CurrentPreviewEventInstance = IFMODStudioModule::Get().CreateAuditioningInstance(EditedEvent);
	if (CurrentPreviewEventInstance != nullptr)
	{
		for (int32 ParamIdx = 0; ParamIdx < ParameterValues.Num(); ParamIdx++)
		{
			CurrentPreviewEventInstance->setParameterValueByIndex(ParamIdx, ParameterValues[ParamIdx]);
		}

		CurrentPreviewEventInstance->start();
	}
}

void FFMODEventEditor::PauseEvent()
{
	if (CurrentPreviewEventInstance != nullptr)
	{
		bool bIsPaused = false;
		CurrentPreviewEventInstance->getPaused(&bIsPaused);
		CurrentPreviewEventInstance->setPaused(!bIsPaused);
	}
}

void FFMODEventEditor::StopEvent()
{
	IFMODStudioModule::Get().StopAuditioningInstance();
}

void FFMODEventEditor::SetParameterValue(int32 ParameterIdx, float Value)
{
	ParameterValues[ParameterIdx] = Value;

	if (CurrentPreviewEventInstance != nullptr)
	{
		CurrentPreviewEventInstance->setParameterValueByIndex(ParameterIdx, Value);
	}
}

TArray<float>& FFMODEventEditor::GetParameterValues()
{
	return ParameterValues;
}

void FFMODEventEditor::InitFMODEventEditor(const EToolkitMode::Type Mode, const TSharedPtr< class IToolkitHost >& InitToolkitHost, UFMODEvent* Event)
{
	EditedEvent = Event;

	TSharedRef<FTabManager::FLayout> StandaloneDefaultLayout = FTabManager::NewLayout("Standalone_FMODEventEditor_Layout")
	->AddArea
	(
		FTabManager::NewPrimaryArea()->SetOrientation(Orient_Vertical)
		->Split
		(
			FTabManager::NewStack()
			->AddTab(EventEditorTabId, ETabState::OpenedTab)->SetHideTabWell(true)
		)
	);

	const bool bCreateDefaultStandaloneMenu = true;
	const bool bCreateDefaultToolbar = false;
	FAssetEditorToolkit::InitAssetEditor(Mode, InitToolkitHost, FFMODEventEditor::FMODEventEditorAppIdentifier, StandaloneDefaultLayout, bCreateDefaultStandaloneMenu, bCreateDefaultToolbar, Event);
}

FName FFMODEventEditor::GetToolkitFName() const
{
	return FName("FMODEventEditor");
}

FText FFMODEventEditor::GetBaseToolkitName() const
{
	return LOCTEXT("ToolkitName", "FMOD Event Editor");
}

FString FFMODEventEditor::GetWorldCentricTabPrefix() const
{
	return LOCTEXT("WorldCentricTabPrefix", "FMOD Event ").ToString();
}

FLinearColor FFMODEventEditor::GetWorldCentricTabColorScale() const
{
	return FLinearColor(0.0f, 0.0f, 0.5f, 0.5f);
}

void FFMODEventEditor::CreateInternalWidgets()
{
	FMODEventEditorPanel = SNew(SFMODEventEditorPanel)
		.FMODEventEditor(SharedThis(this));
}

TSharedRef<SDockTab> FFMODEventEditor::SpawnTab_EventEditor(const FSpawnTabArgs& Args)
{
	check(Args.GetTabId().TabType == EventEditorTabId);

	CreateInternalWidgets();

	return SAssignNew(OwnerTab, SDockTab)
		.Label(LOCTEXT("EventEditorTitle", "FMOD Event"))
		.TabColorScale(GetTabColorScale())
		[
			FMODEventEditorPanel.ToSharedRef()
		];
}

void FFMODEventEditor::HandleBanksReloaded()
{
	CurrentPreviewEventInstance = nullptr;

	CreateInternalWidgets();

	if (OwnerTab.IsValid())
	{
		OwnerTab->SetContent(FMODEventEditorPanel.ToSharedRef());
	}
}

void FFMODEventEditor::HandleBeginPIE(bool bSimulating)
{
	CurrentPreviewEventInstance = nullptr;
}

#undef LOCTEXT_NAMESPACE