/*
	Copyright (C) 2005-2007 Feeling Software Inc.
	Portions of the code are:
	Copyright (C) 2005-2007 Sony Computer Entertainment America
	
	MIT License: http://www.opensource.org/licenses/mit-license.php
*/

#include "StdAfx.h"
#include "FCDocument/FCDocument.h"
#include "FCDocument/FCDAsset.h"
#include "FCDocument/FCDLibrary.h"
#include "FCDocument/FCDCamera.h"
#include "FCDocument/FCDEntityInstance.h"
#include "FCDocument/FCDExternalReferenceManager.h"
#include "FCDocument/FCDSceneNode.h"
#include "FUtils/FUTestBed.h"

TESTSUITE_START(FCTestXRefTree)

TESTSUITE_TEST(0, Export)
	// None of the previous tests should be leaving dangling documents.
	PassIf(FCollada::GetTopDocumentCount() == 0);

	FCDocument* topDoc = FCollada::NewTopDocument();
	FCDocument* midDoc = FCollada::NewTopDocument();
	FCDocument* botDoc = FCollada::NewTopDocument();

	// Create a camera at the bottom document.
	// This camera is referenced by the visual scene of the middle document.
	// The visual scene node of the middle document is referenced by the top document.
	FCDCamera* camera = botDoc->GetCameraLibrary()->AddEntity();
	camera->SetProjectionType(FCDCamera::ORTHOGRAPHIC);
	camera->SetMagX(2.0f);
	FCDSceneNode* midNode = midDoc->AddVisualScene();
	midNode = midNode->AddChildNode();
	midNode->AddInstance(camera);
	FCDSceneNode* topNode = topDoc->AddVisualScene();
	topNode = topNode->AddChildNode();
	topNode->AddInstance(midNode);

	// Save the files, out of order, for fun.
	midDoc->SetFileUrl(FS("XRefDocMid.dae"));
	topDoc->SetFileUrl(FS("XRefDocTop.dae"));
	FCollada::SaveDocument(botDoc, FC("XRefDocBot.dae"));
	SAFE_RELEASE(botDoc);
	FCollada::SaveDocument(midDoc, FC("XRefDocMid.dae"));
	SAFE_RELEASE(midDoc);
	FCollada::SaveDocument(topDoc, FC("XRefDocTop.dae"));
	SAFE_RELEASE(topDoc);

TESTSUITE_TEST(1, ReimportTopOnly)
	// None of the previous tests should be leaving dangling documents.
	PassIf(FCollada::GetTopDocumentCount() == 0);

	FCollada::SetDereferenceFlag(true);
	FCDocument* topDoc = FCollada::NewTopDocument();
	PassIf(FCollada::LoadDocumentFromFile(topDoc, FC("XRefDocTop.dae")));
	FCDSceneNode* node = topDoc->GetVisualSceneInstance();
	FailIf(node == NULL || node->GetChildrenCount() == 0);
	node = node->GetChild(0);
	FailIf(node == NULL || node->GetInstanceCount() == 0);
	FCDEntityInstance* instance = node->GetInstance(0);
	FailIf(instance == NULL || !instance->IsExternalReference());
	PassIf(instance->GetEntityType() == FCDEntity::SCENE_NODE);
	node = (FCDSceneNode*) instance->GetEntity();
	FailIf(node == NULL);
	PassIf(node->GetInstanceCount() == 1);
	instance = node->GetInstance(0);
	PassIf(instance->IsExternalReference());
	PassIf(instance->GetEntityType() == FCDEntity::CAMERA);
	FCDCamera* camera = (FCDCamera*) instance->GetEntity();
	PassIf(camera != NULL);
	PassIf(camera->GetProjectionType() == FCDCamera::ORTHOGRAPHIC);
	PassIf(IsEquivalent(camera->GetMagX(), 2.0f));

	SAFE_RELEASE(topDoc);

TESTSUITE_TEST(2, ReimportTopAndBot)
	// None of the previous tests should be leaving dangling documents.
	PassIf(FCollada::GetTopDocumentCount() == 0);

	FUErrorSimpleHandler errorHandler;
	FCollada::SetDereferenceFlag(true);
	FCDocument* topDoc = FCollada::NewTopDocument();
	PassIf(FCollada::LoadDocumentFromFile(topDoc, FC("XRefDocTop.dae")));
	FCDocument* botDoc = FCollada::NewTopDocument();
	PassIf(FCollada::LoadDocumentFromFile(botDoc, FC("XRefDocBot.dae")));
	PassIf(topDoc != NULL);
	PassIf(botDoc != NULL);
	FailIf(!errorHandler.IsSuccessful());

	FCDSceneNode* node = topDoc->GetVisualSceneInstance();
	FailIf(node == NULL || node->GetChildrenCount() == 0);
	node = node->GetChild(0);
	FailIf(node == NULL || node->GetInstanceCount() == 0);
	FCDEntityInstance* instance = node->GetInstance(0);
	FailIf(instance == NULL || !instance->IsExternalReference());
	PassIf(instance->GetEntityType() == FCDEntity::SCENE_NODE);
	node = (FCDSceneNode*) instance->GetEntity();
	FailIf(node == NULL);
	PassIf(node->GetInstanceCount() == 1);
	instance = node->GetInstance(0);
	PassIf(instance->IsExternalReference());
	PassIf(instance->GetEntityType() == FCDEntity::CAMERA);
	FCDCamera* camera = (FCDCamera*) instance->GetEntity();
	PassIf(camera != NULL);
	PassIf(camera->GetDocument() == botDoc);
	PassIf(camera->GetProjectionType() == FCDCamera::ORTHOGRAPHIC);
	PassIf(IsEquivalent(camera->GetMagX(), 2.0f));

	SAFE_RELEASE(topDoc);
	SAFE_RELEASE(botDoc);

TESTSUITE_END