// Fill out your copyright notice in the Description page of Project Settings.


#include "VRM4U_VMCSubsystem.h"
#include "VrmVMCObject.h"

#include "Engine/Engine.h"
#include "UObject/StrongObjectPtr.h"
#include "Misc/ScopeLock.h"
#include "OSCManager.h"
#include "OSCServer.h"

bool UVRM4U_VMCSubsystem::CopyVMCData(FVMCData &data, FString ServerAddress, int port) {
	for (int i = 0; i < VMCObjectList.Num(); ++i) {
		auto a = VMCObjectList[i].Get();
		if (a == nullptr) continue;

		if (a->ServerName == ServerAddress && a->port == port) {
			a->CopyVMCData(data);
			return true;
		}
	}
	return false;
}

bool UVRM4U_VMCSubsystem::GetVMCData(TMap<FString, FTransform>& BoneData, TMap<FString, float>& CurveData, FString ServerAddress, int port) {
	FVMCData d;
	if (CopyVMCData(d, ServerAddress, port) == false) {
		return false;
	}

	BoneData = d.BoneData;
	CurveData = d.CurveData;

	return true;
}


UVrmVMCObject* UVRM4U_VMCSubsystem::FindOrAddServer(const FString ServerAddress, int port) {

	bool bFound = false;
	for (int i = 0; i < VMCObjectList.Num(); ++i) {
		auto a = VMCObjectList[i].Get();
		if (a == nullptr) continue;

		if (a->ServerName == ServerAddress && a->port == port) {
			bFound = true;
			return a;
		}
	}
	if (bFound == false) {
		int ind = VMCObjectList.AddDefaulted();
		VMCObjectList[ind].Reset(NewObject<UVrmVMCObject>());
		VMCObjectList[ind]->CreateServer(ServerAddress, port);
		return VMCObjectList[ind].Get();
	}
	return nullptr;
}

void UVRM4U_VMCSubsystem::DestroyVMCServer(const FString ServerAddress, int port) {
	for (int i = 0; i < VMCObjectList.Num(); ++i) {
		auto a = VMCObjectList[i].Get();
		if (a == nullptr) continue;

		if (a->ServerName == ServerAddress && a->port == port) {
			a->DestroyServer();
			VMCObjectList[i].Reset(nullptr);
			VMCObjectList.RemoveAt(i);
			return;
		}
	}
}
void UVRM4U_VMCSubsystem::DestroyVMCServerAll() {
	while (VMCObjectList.Num()) {
		auto a = VMCObjectList[0].Get();
		if (a) {
			a->DestroyServer();
		}
		VMCObjectList[0].Reset(nullptr);
		VMCObjectList.RemoveAt(0);
	}
}


bool UVRM4U_VMCSubsystem::CreateVMCServer(const FString ServerAddress, int port) {
	return (FindOrAddServer(ServerAddress, port) != nullptr);
}

void UVRM4U_VMCSubsystem::ClearData(const FString ServerAddress, int port) {
	for (int i = 0; i < VMCObjectList.Num(); ++i) {
		auto a = VMCObjectList[i].Get();
		if (a == nullptr) continue;

		if (a->ServerName == ServerAddress && a->port == port) {
			VMCObjectList[i]->ClearVMCData();
			return;
		}
	}
}

void UVRM4U_VMCSubsystem::Initialize(FSubsystemCollectionBase& Collection) {
	Super::Initialize(Collection);

#if WITH_EDITOR
	FEditorDelegates::BeginStandaloneLocalPlay.AddLambda([&](const uint32 processID) {
		this->DestroyVMCServerAll();
	});
#endif

}