/**********************************************************************
 *<
	FILE: VertexPaint.h

	DESCRIPTION: Modifier definition

	CREATED BY:  Christer Janson, Nikolai Sander

	HISTORY:

 *>	Copyright (c) 1997, All Rights Reserved.
 **********************************************************************/

#ifndef __VERTEXPAINT__H
#define __VERTEXPAINT__H

#include "Max.h"
#include "resource.h"
#include "istdplug.h"
#include "modstack.h"


#define VERTEX_TREE_PAINT_CLASS_ID	Class_ID(0x40c7005e, 0x2a95082c)
#define CID_PAINT				(CID_USER+0x439c)
#define NUMPALETTES				7
#define NUMGRADIENTPALETTES		2
#define WM_POSTINIT				(WM_USER+0x1177)

extern TCHAR *GetString(int id);
extern ClassDesc* GetVertexPaintDesc();
extern HINSTANCE hInstance;

class NVert
{
public:
	NVert();
	NVert& operator= (NVert &nvert);
	Tab<int> faces;
	Tab<int> whichVertex;
};

class ColorData
{
public:
	ColorData(DWORD col);
	ColorData();

	COLORREF color;
};


class VertexPaintData : public LocalModData {
	
	friend class VertexPaintRestore;

public:

		enum	TComponent {Red=0, Green, Blue};

		VertexPaintData(Mesh& m);
		VertexPaintData();
		~VertexPaintData();
		LocalModData*	Clone();

		Mesh*	GetMesh();
		NVert&  GetNVert(int i);
		NVert&  GetNVCVert(int i)  ;
		void	SetCache(Mesh& m);
		void	FreeCache();

		COLORREF&	GetColor(int i);
		ColorData&  GetColorData(int i);
		// colorMask to enable/disable write to R/G/B
		void		SetColor(int i,float bary,COLORREF c, TComponent whichComp);
		void		SetColor(int i, const ColorData &c);
		int			GetNumColors();
		int			GetMaxNumColors();
		void		AllocColorData(int numcols);
		void		SynchVerts(Mesh &m);

	private:
		Mesh*	mesh;
		ColorData* colordata;
		int		numColors;
		NVert    *nverts;
		int      numnverts;
		NVert    *nvcverts;
		int      numnvcverts;

	};

class VertexPaint : public Modifier {
	
	friend class PaintMouseProc;
	friend BOOL CALLBACK VertexPaintDlgProc(HWND hWnd, UINT msg, WPARAM wParam, LPARAM lParam);
public:

		static IObjParam*		ip;
		static VertexPaint*		editMod;
		static HWND				hParams;
		static ICustButton*		iPaintButton;
		static ICustButton*		iPickButton;
		static IColorSwatch*	iColor;
		static COLORREF			lastWeightColor;
		static COLORREF			lastPhaseColor;
		static COLORREF			palColors[NUMPALETTES];

		// Yoyo: ZGradient
		static IColorSwatch*	iColorGradient[NUMGRADIENTPALETTES];
		static COLORREF			lastGradientColor[NUMGRADIENTPALETTES];
		float					fGradientBend;
		ISpinnerControl			*iGradientBend;


		//Constructor/Destructor
		VertexPaint();
		~VertexPaint();

		// From Animatable
		void DeleteThis() { delete this; }
		void GetClassName(TSTR& s) { s= TSTR(GetString(IDS_CLASS_NAME)); }  
		virtual Class_ID ClassID() { return VERTEX_TREE_PAINT_CLASS_ID;}		
		RefTargetHandle Clone(RemapDir& remap = DefaultRemapDir());
		TCHAR *GetObjectName() { return GetString(IDS_CLASS_NAME); }
		IOResult Load(ILoad *iload);
		IOResult Save(ISave *isave);
		IOResult LoadLocalData(ILoad *iload, LocalModData **pld);
		IOResult SaveLocalData(ISave *isave, LocalModData *ld);

		//From Modifier
		ChannelMask ChannelsUsed()  { return GEOM_CHANNEL|TOPO_CHANNEL|VERTCOLOR_CHANNEL|PART_SUBSEL_TYPE|SELECT_CHANNEL|PART_DISPLAY|TEXMAP_CHANNEL; }
		ChannelMask ChannelsChanged() { return VERTCOLOR_CHANNEL|TOPO_CHANNEL|GEOM_CHANNEL; }
		void ModifyObject(TimeValue t, ModContext &mc, ObjectState *os, INode *node);
		Class_ID InputType() {return triObjectClassID;}
		Interval LocalValidity(TimeValue t);
		void NotifyInputChanged(Interval changeInt, PartID partID, RefMessage message, ModContext *mc);


		// From BaseObject
		BOOL ChangeTopology() {return FALSE;}
		BOOL DependOnTopology(ModContext &mc);

		int GetParamBlockIndex(int id) {return id;}

		//From ReferenceMaker
		int NumRefs();
		RefTargetHandle GetReference(int i);
		void SetReference(int i, RefTargetHandle rtarg);
		
		int NumSubs();
		Animatable* SubAnim(int i);
		TSTR SubAnimName(int i);

		RefResult NotifyRefChanged( Interval changeInt,RefTargetHandle hTarget, 
		   PartID& partID, RefMessage message);

		
		CreateMouseCallBack* GetCreateMouseCallBack() {return NULL;}
		void BeginEditParams(IObjParam *ip, ULONG flags,Animatable *prev);
		void EndEditParams(IObjParam *ip, ULONG flags,Animatable *next);

		Interval GetValidity(TimeValue t);
		ParamDimension *GetParameterDim(int pbIndex);
		TSTR GetParameterName(int pbIndex);

		BOOL	ActivatePaint(BOOL bOnOff, BOOL bPick = FALSE);
		void	EnterMode();
		void	ExitMode();
		BOOL	IsValidNode(INode* node);
		ModContext* ModContextFromNode(INode* node);
		COLORREF GetActiveColor();
		void	TurnVCOn(BOOL shaded);

		// fill current selection with color.
		void	fillSelectionColor();

		// fill current selection with gradient color.
		void	fillSelectionGradientColor();

		// Edit what? Intensity, phaseLevel1, phaseLevel2.
		void	setEditionType(int editMode);
		int		getEditionType() const {return _EditType;}

		// Bkup/reload iColor color according to getEditionType()
		void	backupCurrentColor();
		void	reloadBkupColor();

		void	PaletteButton(HWND hWnd);
		void	InitPalettes();
		void	SavePalettes();
	
	private :

		void SetFlag (DWORD fl) { flags |= fl; }
		void ClearFlag (DWORD fl) { flags &= ~fl; }
		void SetFlag (DWORD fl, bool set) { if (set) SetFlag (fl); else ClearFlag (fl); }
		bool GetFlag (DWORD fl) { return (fl&flags) ? TRUE : FALSE; }

	private:
		ModContextList	modContexts;
		INodeTab		nodeTab;
		HWND			hPaletteWnd[NUMPALETTES];
		float			fTint;
		ISpinnerControl *iTint;
		DWORD flags;

		int				_EditType;
		bool			_LastPaintMode;
	};

/****************************************************************************
 *
 * 3D Paint Implementation
 *
 ***************************************************************************/

// Mouse proc
class PaintMouseProc : public MouseCallBack {
public:
	int		proc(HWND hWnd, int msg, int point, int flags, IPoint2 m); // Mouse callback
	void	SetModifier(VertexPaint* pMod) { pModifier = pMod; }
	void	SetPickMode(BOOL bPick){bPickMode = bPick;}
	BOOL	GetPickMode(){return bPickMode;}
	void	DoPainting(HWND hWnd, IPoint2 m);
	void	DoPickColor(HWND hWnd, IPoint2 m);
	void	MaybeStartHold();
	void	MaybeEndHold();
	BOOL	IsValidPickNode(INode *node);


private:
	VertexPaint*	pModifier;
	BOOL			wasHolding;
	BOOL			bPickMode;
};


class PaintCommandMode : public CommandMode {
	public:
		PaintCommandMode();

		PaintMouseProc mouseProc;

		int							Class();
		virtual int					ID();
		MouseCallBack*				MouseProc(int *numPoints);
		BOOL						ChangeFG(CommandMode *oldMode);
		ChangeForegroundCallback*	ChangeFGProc();
		void						EnterMode();
		void						ExitMode();

		void						SetInterface(Interface* ip) { iInterface = ip; }
		void						SetModifier(VertexPaint* pMod)		{ pModifier = pMod; mouseProc.SetModifier(pMod); }

private:
		Interface*					iInterface;
		VertexPaint*				pModifier;
};

class VertexPaintRestore : public RestoreObj
{
private:
	VertexPaintData *pPaintData;
	VertexPaint *pMod;

	ColorData *colordata;
	ColorData *redoColordata;
	int numcolors;
	int redonumcolors;

public:	

	VertexPaintRestore(VertexPaintData *pLocalData, VertexPaint *pVPaint);
	~VertexPaintRestore();
	void Restore(int isUndo);
	void Redo();
	int  Size();
};


#endif // __VERTEXPAINT__H