/*==LICENSE==*

CyanWorlds.com Engine - MMOG client, server and tools
Copyright (C) 2011  Cyan Worlds, Inc.

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>.

You can contact Cyan Worlds, Inc. by email legal@cyan.com
 or by snail mail at:
      Cyan Worlds, Inc.
      14617 N Newport Hwy
      Mead, WA   99021

*==LICENSE==*/
#include "HeadSpin.h"
#include "max.h"
#include "resource.h"
//#include "hsUtils.h"
#include "hsTemplates.h"
#include "../plResMgr/plKeyFinder.h"
#include "../plResMgr/plPageInfo.h"
#include "hsResMgr.h"
#include "../MaxMain/plMaxNode.h"
#include "plClothingComponent.h"
#include "plComponentReg.h"
#include "../MaxPlasmaMtls/Materials/plClothingMtl.h"
#include "../pnMessage/plRefMsg.h"
#include "../plAvatar/plAvatarClothing.h"
#include "../pnSceneObject/plSceneObject.h"
#include "../MaxConvert/hsMaterialConverter.h"
#include "../MaxConvert/plMeshConverter.h"
#include "plPickMaterialMap.h"
#include "../MaxMain/plMtlCollector.h"
#include "plAvatarComponent.h"
#include "../MaxMain/plPlasmaRefMsgs.h"			
#include "../plDrawable/plSharedMesh.h"
#include "../plDrawable/plDrawableSpans.h"
#include "../plDrawable/plMorphSequence.h"
#include "../plScene/plSceneNode.h"
#include "../plDrawable/plGeometrySpan.h"

void DummyCodeIncludeFuncClothing()
{
}

CLASS_DESC(plClothingComponent, gClothingDesc, "Avatar Clothing",  "AvatarClothing", COMP_TYPE_MISC, CLOTHING_COMPONENT_CLASS_ID)

static plClothingComponentProc gClothingComponentProc;

class plClothingAccessor : public PBAccessor
{
public:

	//! Public Accessor Class, used in ParamBlock2 processing.
	/*!
	Workhorse for this Accessor Class (derived from Max's PBAccessor).

	When one of our parameters that is a ref changes, send out the component ref
	changed message.  Normally, messages from component refs are ignored since
	they pass along all the messages of the ref, which generates a lot of false
	converts.
	*/

	void Set(PB2Value& v, ReferenceMaker* owner, ParamID id, int tabIndex, TimeValue t)
	{
		if (id == plClothingComponent::kMeshNodeAddBtn)
		{
			plClothingComponent *comp = (plClothingComponent *)owner;
			IParamBlock2 *pb = comp->GetParamBlockByID(plClothingComponent::kBlkComp);
			int state = pb->GetInt(plClothingComponent::kLODState);
				
			INode *node = pb->GetINode(plClothingComponent::kMeshNodeAddBtn);
			if (node)
				pb->SetValue(plClothingComponent::kMeshNodeTab, 0, node, state);
		}
			
		if (id == plClothingComponent::kMeshNodeTab)
		{
			plComponentBase *comp = (plComponentBase*)owner;
			comp->NotifyDependents(FOREVER, PART_ALL, REFMSG_USER_COMP_REF_CHANGED);
		}
	}
};

plClothingAccessor gClothingAccessor;

ParamBlockDesc2 gClothingBk
(	

	plComponent::kBlkComp, _T("Clothing"), 0, &gClothingDesc, P_AUTO_CONSTRUCT + P_AUTO_UI, plComponent::kRefComp,

	//Roll out
	IDD_COMP_CLOTHING, IDS_COMP_CLOTHING, 0, 0, &gClothingComponentProc,

	plClothingComponent::kMaterials,	_T("ClothingMaterials"),	TYPE_MTL_TAB, 0,	0, 0,	
		end,
		
	plClothingComponent::kGroup,	_T("ClothingGroup"),	TYPE_INT, 0, 0,
		end,
		
	plClothingComponent::kType,		_T("ClothingType"),		TYPE_INT, 0, 0,
		end,

	plClothingComponent::kLODState, _T("LODState"),		TYPE_INT,	0, 0,
		p_default, 0,
		end,

	plClothingComponent::kMeshNodeTab,	_T("MeshObject"),	TYPE_INODE_TAB, plLODAvatarComponent::kMaxNumLODLevels,		0, 0,
		p_accessor,		&gClothingAccessor,
		end,

	plClothingComponent::kMeshNodeAddBtn, _T("MshNodePicker"),	TYPE_INODE,		0, 0,
		p_ui,	TYPE_PICKNODEBUTTON, IDC_COMP_LOD_CLOTHING_MESH_PICKB,
		p_sclassID,	GEOMOBJECT_CLASS_ID,
		p_prompt, IDS_COMP_AVATAR_PROXYS,
		p_accessor, &gClothingAccessor,
		end,

	end
);

plClothingComponent::plClothingComponent()
{
	fClassDesc = &gClothingDesc;
	fClassDesc->MakeAutoParamBlocks(this);
}

hsBool plClothingComponent::SetupProperties(plMaxNode* node, plErrorMsg* pErrMsg)
{
	int i;
	for (i = 0; i < fCompPB->Count(kMeshNodeTab); i++)
	{
		plMaxNode *LODNode = (plMaxNode *)fCompPB->GetINode(kMeshNodeTab, 0, i);
		if (LODNode != nil)
		{
			char *dbgNodeName = LODNode->GetName();
			//LODNode->SetCanConvert(false);
			LODNode->SetDrawable(false);
			LODNode->SetForceShadow(true);
			LODNode->SetForceMatShade(true);
			if (!LODNode->GetSwappableGeom())
				LODNode->SetSwappableGeom(new plSharedMesh);

			//UInt32 targetID = fCompPB->GetInt(kType);
			//((plMaxNode *)LODNode->GetParentNode())->SetSwappableGeomTarget(targetID);
		}
	}
	
	return true;
}

hsBool plClothingComponent::PreConvert(plMaxNode *node, plErrorMsg *pErrMsg)
{
	node->SetupBoneHierarchyPalette();

	return true;
}

hsBool plClothingComponent::Convert(plMaxNode *node, plErrorMsg *pErrMsg)
{
	int i, j;
	hsTArray<plGeometrySpan*> spanArray;
	hsTArray<plKey> keys;
	plMaxNode *LODNode = nil;
	plMaxNode *locationNode = nil;


	if (fCompPB->Count(plClothingComponent::kMaterials) <= 0)
		return true;

	for (i = 0; i < fCompPB->Count(kMeshNodeTab); i++)
	{
		spanArray.Reset();
		//plSharedMesh *mesh = TRACKED_NEW plSharedMesh;
		LODNode = (plMaxNode *)fCompPB->GetINode(kMeshNodeTab, 0, i);
		if (LODNode != nil)
		{
			char *dbgNodeName = LODNode->GetName();
			keys.Append(LODNode->GetSwappableGeom()->GetKey());
			locationNode = LODNode;

			if (fCompPB->GetInt(ParamID(kType)) != plClothingMgr::kTypeFace)
			{
				// We only save state for the face node.
				LODNode->GetSwappableGeom()->fFlags |= plSharedMesh::kDontSaveMorphState;
			}
			else
			{
				// The face's weight for the first layer (0) is to be applied to all
				// meshes on that node.
				LODNode->GetSwappableGeom()->fFlags |= plSharedMesh::kLayer0GlobalToMod;
			}
		}
		else
		{
			keys.Append(nil);
			//delete mesh;
		}
	}

	const plPageInfo* thisInfo = plKeyFinder::Instance().GetLocationInfo(locationNode ? locationNode->GetLocation() : node->GetLocation());
	const plLocation &loc = plKeyFinder::Instance().FindLocation("GlobalClothing", thisInfo->GetPage());

	for (i = 0; i < fCompPB->Count(plClothingComponent::kMaterials); i++)
	{
		plClothingMtl *mtl = (plClothingMtl *)fCompPB->GetMtl(ParamID(plClothingComponent::kMaterials), 0, i);
		plClothingItem *cloth = hsMaterialConverter::Instance().GenerateClothingItem(mtl, loc);
		cloth->fGroup = fCompPB->GetInt(ParamID(kGroup));
		cloth->fType = fCompPB->GetInt(ParamID(kType));

		plGenRefMsg *refMsg;
		for (j = 0; j < keys.GetCount(); j++)
		{
			if (keys[j] != nil)
			{
				refMsg = TRACKED_NEW plGenRefMsg(cloth->GetKey(), plRefMsg::kOnCreate, j, -1);
				hsgResMgr::ResMgr()->AddViaNotify(keys[j], refMsg, plRefFlags::kActiveRef);
			}
		}
	}
	return true;
}

////////////////////////////////////////////////////////////////////////////////////////

BOOL plClothingComponentProc::DlgProc(TimeValue t, IParamMap2 *pm, HWND hWnd, UINT msg, WPARAM wParam, LPARAM lParam)
{
	IParamBlock2 *pb = pm->GetParamBlock();
	HWND hList = GetDlgItem(hWnd, IDC_CLOTHING_LIST);
	HWND hGroup = GetDlgItem(hWnd, IDC_CLOTHING_GROUP);
	HWND hType = GetDlgItem(hWnd, IDC_CLOTHING_TYPE);
	HWND hLOD = GetDlgItem(hWnd, IDC_COMP_LOD_CLOTHING_STATE);
	switch (msg)
	{
	case WM_INITDIALOG:
		{
			ListBox_ResetContent(hList);
			int i;
			for (i = 0; i < pb->Count(plClothingComponent::kMaterials); i++)
				ListBox_AddString(hList, pb->GetMtl(ParamID(plClothingComponent::kMaterials), 0, i)->GetName());

			ListBox_SetCurSel(hList, -1);

			for (i = 0; i < plClothingMgr::kMaxGroup; i++)
				ComboBox_AddString(hGroup, plClothingMgr::GroupStrings[i]);
			ComboBox_SetCurSel(hGroup, pb->GetInt(plClothingComponent::kGroup));

			for (i = 0; i < plClothingMgr::kMaxType; i++)
				ComboBox_AddString(hType, plClothingMgr::TypeStrings[i]);
			ComboBox_SetCurSel(hType, pb->GetInt(plClothingComponent::kType));

			ComboBox_AddString(hLOD, "High");
			ComboBox_AddString(hLOD, "Medium");
			ComboBox_AddString(hLOD, "Low");
			ComboBox_SetCurSel(hLOD, pb->GetInt(plClothingComponent::kLODState));
		}
		return TRUE;


	case WM_COMMAND:
		if (HIWORD(wParam) == BN_CLICKED)
		{
			if (LOWORD(wParam) == IDC_CLOTHING_ADD)
			{
				Mtl *pickedMtl = plPickMaterialMap::PickMaterial(plMtlCollector::kClothingMtlOnly);
				if (pickedMtl != nil)
				{
					LRESULT stringIdx = ListBox_FindStringExact(hList, -1, pickedMtl->GetName());
					if (stringIdx == LB_ERR) // It's not already there, go and add it
					{
						pb->Append(ParamID(plClothingComponent::kMaterials), 1, &pickedMtl, 0);
						ListBox_AddString(hList, pickedMtl->GetName());
					}
				}
				return TRUE;
			}
			// Remove the currently selected material
			else if (LOWORD(wParam) == IDC_CLOTHING_REMOVE)
			{
				int sel = ListBox_GetCurSel(hList);
				if (sel != LB_ERR)
				{
					pb->Delete(plClothingComponent::kMaterials, sel, 1);
					ListBox_DeleteString(hList, sel);
				}
				return TRUE;
			}
			else if( LOWORD( wParam ) == IDC_CLOTHING_CLEARMESH )
			{
				int state = pb->GetInt(plClothingComponent::kLODState);
				pb->SetValue(plClothingComponent::kMeshNodeTab, 0, (INode*)nil, state );
				pb->Reset(plClothingComponent::kMeshNodeAddBtn);
			}
		}
		else if (LOWORD(wParam) == IDC_CLOTHING_GROUP)
		{
			int setIdx = ComboBox_GetCurSel(hGroup);
			pb->SetValue(plClothingComponent::kGroup, 0, setIdx);

			return TRUE;
		}
		else if (LOWORD(wParam) == IDC_CLOTHING_TYPE)
		{
			int setIdx = ComboBox_GetCurSel(hType);
			pb->SetValue(plClothingComponent::kType, 0, setIdx);

			return TRUE;
		}
		else
		{
			int state = pb->GetInt(plClothingComponent::kLODState);
				
			INode *node = pb->GetINode(plClothingComponent::kMeshNodeAddBtn);
			if (node)
				pb->SetValue(plClothingComponent::kMeshNodeTab, 0, node, state);

			if(LOWORD(wParam) == IDC_COMP_LOD_CLOTHING_STATE && HIWORD(wParam) == CBN_SELCHANGE)
			{
				int idx = SendMessage((HWND)lParam, CB_GETCURSEL, 0, 0);
				pb->SetValue(plClothingComponent::kLODState, 0, idx);
			
				node = pb->GetINode(plClothingComponent::kMeshNodeTab, 0, idx);
				if (node)
					pb->SetValue(plClothingComponent::kMeshNodeAddBtn, 0, node);
				else
					pb->Reset(plClothingComponent::kMeshNodeAddBtn);

				return TRUE;
			}
		}
	}

	return FALSE;
}