BansheeEngine/Source/SBansheeEditor/Source/BsScriptPrefabUtility.cpp

//********************************** Banshee Engine (www.banshee3d.com) **************************************************//
//**************** Copyright (c) 2016 Marko Pintera ([email protected]). All rights reserved. **********************//
#include "BsScriptPrefabUtility.h"
#include "BsMonoManager.h"
#include "BsMonoClass.h"
#include "BsPrefabUtility.h"
#include "BsScriptSceneObject.h"
#include "BsSceneObject.h"
#include "BsPrefab.h"
#include "BsResources.h"
#include "BsMonoUtil.h"
#include "BsScriptGameObjectManager.h"
#include "BsScriptPrefab.h"

namespace BansheeEngine
{
	ScriptPrefabUtility::ScriptPrefabUtility(MonoObject* instance)
		:ScriptObject(instance)
	{ }

	void ScriptPrefabUtility::initRuntimeData()
	{
		metaData.scriptClass->addInternalCall("Internal_BreakPrefab", &ScriptPrefabUtility::internal_breakPrefab);
		metaData.scriptClass->addInternalCall("Internal_ApplyPrefab", &ScriptPrefabUtility::internal_applyPrefab);
		metaData.scriptClass->addInternalCall("Internal_RevertPrefab", &ScriptPrefabUtility::internal_revertPrefab);
		metaData.scriptClass->addInternalCall("Internal_HasPrefabLink", &ScriptPrefabUtility::internal_hasPrefabLink);
		metaData.scriptClass->addInternalCall("Internal_GetPrefabParent", &ScriptPrefabUtility::internal_getPrefabParent);
		metaData.scriptClass->addInternalCall("Internal_GetPrefabUUID", &ScriptPrefabUtility::internal_GetPrefabUUID);
		metaData.scriptClass->addInternalCall("Internal_UpdateFromPrefab", &ScriptPrefabUtility::internal_UpdateFromPrefab);
		metaData.scriptClass->addInternalCall("Internal_RecordPrefabDiff", &ScriptPrefabUtility::internal_RecordPrefabDiff);
	}

	void ScriptPrefabUtility::internal_breakPrefab(ScriptSceneObject* soPtr)
	{
		if (ScriptSceneObject::checkIfDestroyed(soPtr))
			return;

		soPtr->getNativeSceneObject()->breakPrefabLink();
	}

	void ScriptPrefabUtility::internal_applyPrefab(ScriptSceneObject* soPtr, ScriptPrefab* prefabPtr)
	{
		if (ScriptSceneObject::checkIfDestroyed(soPtr))
			return;

		prefabPtr->getHandle()->update(soPtr->getNativeSceneObject());
	}

	void ScriptPrefabUtility::internal_revertPrefab(ScriptSceneObject* soPtr)
	{
		if (ScriptSceneObject::checkIfDestroyed(soPtr))
			return;

		PrefabUtility::revertToPrefab(soPtr->getNativeSceneObject());
	}

	bool ScriptPrefabUtility::internal_hasPrefabLink(ScriptSceneObject* soPtr)
	{
		if (ScriptSceneObject::checkIfDestroyed(soPtr))
			return false;

		return !soPtr->getNativeSceneObject()->getPrefabLink(true).empty();
	}

	MonoObject* ScriptPrefabUtility::internal_getPrefabParent(ScriptSceneObject* soPtr)
	{
		if (ScriptSceneObject::checkIfDestroyed(soPtr))
			return nullptr;

		HSceneObject so = soPtr->getNativeSceneObject();
		HSceneObject parent = so->getPrefabParent();

		if (parent != nullptr)
		{
			ScriptSceneObject* scriptParent = ScriptGameObjectManager::instance().getOrCreateScriptSceneObject(parent);
			return scriptParent->getManagedInstance();
		}

		return nullptr;
	}

	MonoString* ScriptPrefabUtility::internal_GetPrefabUUID(ScriptSceneObject* soPtr)
	{
		if (ScriptSceneObject::checkIfDestroyed(soPtr))
			return nullptr;

		HSceneObject so = soPtr->getNativeSceneObject();

		String prefabUUID = soPtr->getNativeSceneObject()->getPrefabLink(true);
		if (prefabUUID.empty())
			return nullptr;

		return MonoUtil::stringToMono(prefabUUID);
	}

	void ScriptPrefabUtility::internal_UpdateFromPrefab(ScriptSceneObject* soPtr)
	{
		if (ScriptSceneObject::checkIfDestroyed(soPtr))
			return;

		HSceneObject so = soPtr->getNativeSceneObject();
		HSceneObject prefabParent = so->getPrefabParent();

		if(prefabParent != nullptr)
			PrefabUtility::updateFromPrefab(prefabParent);
	}

	void ScriptPrefabUtility::internal_RecordPrefabDiff(ScriptSceneObject* soPtr)
	{
		if (ScriptSceneObject::checkIfDestroyed(soPtr))
			return;

		HSceneObject so = soPtr->getNativeSceneObject();
		PrefabUtility::recordPrefabDiff(so);
	}
}