////////////////////////////////////////////////////////////////////////////
//
//  Crytek Engine Source File.
//  Copyright (C), Crytek Studios, 2002.
// -------------------------------------------------------------------------
//  File name:   MaterialDialog.h
//  Version:     v1.00
//  Created:     22/1/2003 by Timur.
//  Compilers:   Visual Studio.NET
//  Description: 
// -------------------------------------------------------------------------
//  History:
//
////////////////////////////////////////////////////////////////////////////

#ifndef __materialdialogdialog_h__
#define __materialdialogdialog_h__
#pragma once

#include "BaseLibraryDialog.h"
#include "Controls\SplitterWndEx.h"
#include "Controls\TreeCtrlEx.h"
#include "Controls\PropertyCtrl.h"
#include "Controls\PreviewModelCtrl.h"

class CMaterial;
class CMaterialManager;

/** Dialog which hosts entity prototype library.
*/
class CMaterialDialog : public CBaseLibraryDialog
{
	DECLARE_DYNAMIC(CMaterialDialog)
public:
	CMaterialDialog( CWnd *pParent );
	~CMaterialDialog();

	// Called every frame.
	void Update();

	virtual UINT GetDialogMenuID();

public:
	afx_msg void OnAssignMaterialToSelection();
	afx_msg void OnResetMaterialOnSelection();
	afx_msg void OnGetMaterialFromSelection();

protected:
	IStatObj* GetGeometryFromObject( CBaseObject *pObject );
	ICryCharInstance* GetCharacterFromObject( CBaseObject *pObject );
	void DoDataExchange(CDataExchange* pDX);
	BOOL OnInitDialog();
	
	afx_msg void OnDestroy();
	afx_msg void OnSize(UINT nType, int cx, int cy);
	afx_msg void OnMouseMove(UINT nFlags, CPoint point);
	afx_msg void OnLButtonUp(UINT nFlags, CPoint point);
	
	afx_msg void OnAddItem();
	afx_msg void OnPlay();
	afx_msg void OnUpdatePlay( CCmdUI* pCmdUI );
	afx_msg void OnDrawSelection();
	afx_msg void OnDrawBox();
	afx_msg void OnDrawSphere();
	afx_msg void OnDrawTeapot();
	afx_msg void OnAddSubMtl();
	afx_msg void OnDelSubMtl();
	afx_msg void OnUpdateMtlSelected( CCmdUI* pCmdUI );
	afx_msg void OnUpdateObjectSelected( CCmdUI* pCmdUI );
	afx_msg void OnUpdateAssignMtlToSelection( CCmdUI *pCmdUI );
	afx_msg void OnBeginDrag(NMHDR* pNMHDR, LRESULT* pResult);
	afx_msg void OnNotifyMtlTreeRClick(NMHDR* pNMHDR, LRESULT* pResult);
	afx_msg void OnPickMtl();
	afx_msg void OnUpdatePickMtl( CCmdUI* pCmdUI );
	afx_msg void OnCopy();
	afx_msg void OnPaste();
	afx_msg void OnGenCubemap();
	afx_msg void OnSelectAssignedObjects();

	//////////////////////////////////////////////////////////////////////////
	// Some functions can be overriden to modify standart functionality.
	//////////////////////////////////////////////////////////////////////////
	virtual void InitToolbar();
	virtual HTREEITEM InsertItemToTree( CBaseLibraryItem *pItem,HTREEITEM hParent );
	virtual void SelectItem( CBaseLibraryItem *item,bool bForceReload=false );
	virtual void DeleteItem( CBaseLibraryItem *pItem );

	//////////////////////////////////////////////////////////////////////////
	CMaterial* GetSelectedMaterial();
	void OnUpdateProperties( IVariable *var );

	
	void LoadGeometry( const CString &filename );
	void ReleaseGeometry();
	void AssignMtlToGeometry();

	//void SetTextureVars( CVariableArray *texVar,CMaterial *mtl,int id,const CString &name );
	void SetMaterialVars( CMaterial *mtl );

	void DropToItem( HTREEITEM hItem,HTREEITEM hSrcItem,CMaterial *pMtl );

	//////////////////////////////////////////////////////////////////////////
	// IDocListener listener implementation
	//////////////////////////////////////////////////////////////////////////
	virtual	void OnNewDocument();
	virtual	void OnLoadDocument();
	virtual void OnCloseDocument();
	//////////////////////////////////////////////////////////////////////////

	enum EDrawType
	{
		DRAW_BOX,
		DRAW_SPHERE,
		DRAW_TEAPOT,
		DRAW_SELECTION,
	};
	

	DECLARE_MESSAGE_MAP()

	CSplitterWndEx m_wndHSplitter;
	CSplitterWndEx m_wndVSplitter;
	
	CPreviewModelCtrl m_previewCtrl;
	CPropertyCtrl m_propsCtrl;
	CImageList m_imageList;

	CImageList *m_dragImage;

	// Object to render.
	CString m_visualObject;
	IStatObj *m_pGeometry;
	IEntityRender *m_pEntityRender;

	bool m_bRealtimePreviewUpdate;
	bool m_bOwnGeometry;

	// Material manager.
	CMaterialManager *m_pMatManager;

	CVarBlockPtr m_vars;
	CVarBlockPtr m_publicVars;
	CPropertyItem *m_publicVarsItems;
	CVarBlockPtr m_shaderGenParamsVars;
	CPropertyItem *m_shaderGenParamsVarsItem;

	EDrawType m_drawType;
	CString m_geometryFile;

	HTREEITEM m_hDropItem;
	HTREEITEM m_hDraggedItem;

	TSmartPtr<CMaterial> m_pDraggedMtl;
};

#endif // __materialdialogdialog_h__