/*
===========================================================================

Doom 3 GPL Source Code
Copyright (C) 1999-2011 id Software LLC, a ZeniMax Media company. 

This file is part of the Doom 3 GPL Source Code (?Doom 3 Source Code?).  

Doom 3 Source Code 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.

Doom 3 Source Code 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 Doom 3 Source Code.  If not, see <http://www.gnu.org/licenses/>.

In addition, the Doom 3 Source Code is also subject to certain additional terms. You should have received a copy of these additional terms immediately following the terms and conditions of the GNU General Public License which accompanied the Doom 3 Source Code.  If not, please request a copy in writing from id Software at the address below.

If you have questions concerning this license or the applicable additional terms, you may contact in writing id Software LLC, c/o ZeniMax Media Inc., Suite 120, Rockville, Maryland 20850 USA.

===========================================================================
*/

#ifndef __CDIB__
#define __CDIB__

// Original ColorPicker/DIB source by Rajiv Ramachandran <rrajivram@hotmail.com>
// included with permission from the author

class CDIB  {
public:
	enum BitmapType {
		BMP,
		GIF,
		TIFF
	};
					CDIB( HANDLE hDib = NULL,int nBits = 8 );
	virtual			~CDIB();

	CDIB &			operator=( CDIB& dib );
	BOOL			IsValid() { return ( m_pVoid && Width() && Height() ); }
	void			UseGamma( float fg, BOOL bUse = TRUE );
	BOOL			CreateFromHandle( HANDLE hDib, int nBits );
	BOOL			Create( int width, int height, int bits = 24 );
	BOOL			Create( BITMAPINFOHEADER& bmInfo );
	BOOL			CopyDIB( CDIB& dib );
	BOOL			OpenDIB( CString &fileName );
	BOOL			SaveDIB( CString &fileName, BitmapType type );
	void			ReplaceColor(unsigned char oldColor,unsigned char newColor);
	HANDLE			GetDIBits(int nStartX=-1,int nStartY=-1,int nCx=-1,int nCy=-1);
	CBitmap *		GetBitmap(CDC& dc);
	CBitmap *		GetTempBitmap(CDC& dc);
	DWORD			GetDIBSize();
	int				GetPaletteSize(BITMAPINFOHEADER& bmInfo);
	int				GetPaletteSize();
	int				CountColors();
	int				EnumColors(BYTE *colors);
	void			InitDIB(COLORREF color);
	void			CopyLine(int source,int dest);
	void			DestroyDIB();
	void			SetPalette(unsigned char *palette);
	void			SetPalette(RGBQUAD *pRGB);
	COLORREF		PaletteColor(int index);
	void			SetPixel(int x,int y,COLORREF color);
	void			SetPixel8(int x,int y,unsigned char color);
	COLORREF		GetPixel(int x,int y);
	void			GetPixel(UINT x,UINT y,int& pixel);
	void			BitBlt(HDC hDest,int nXDest,int nYDest,int nWidth,int nHeight,int xSrc,int ySrc);
	void			BitBlt(int nXDest,int nYDest,int nWidth,int nHeight,CDIB& dibSrc,int nSrcX,int nSrcY,BYTE *colors=NULL);
	void			StretchBlt(HDC hDest,int nXDest,int nYDest,int nDWidth,int nDHeight,int xSrc,int ySrc,int  nSWidth,int nSHeight);
	void			StretchBlt(int nXDest,int nYDest,int nDWidth,int nDHeight,CDIB& dibSrc,int xSrc,int ySrc,int  nSWidth,int nSHeight);
	void			ExpandBlt(int nXDest,int nYDest,int xRatio,int yRatio,CDIB& dibSrc,int xSrc,int ySrc,int  nSWidth,int nSHeight);
	void			SetFlags(int flag) { m_nFlags = flag; }
	int				Height() { return height ; }
	int				Width() { return width ; }
	unsigned char *GetLinePtr(int line);	
	inline int		GetBitCount() { return m_pInfo->bmiHeader.biBitCount; }
	BOOL			Make8Bit( CDIB &dib );
	BOOL			SwitchFromOne( CDIB &dib );
	BOOL			SwitchFromFour( CDIB &dib );
	BOOL			SwitchFrom24( CDIB &dib );
	BOOL			SwitchPalette( CDIB &dib );
	int				ClosestColor(RGBQUAD *pRgb );
	LPBITMAPINFO	GetBitmapInfo() { return m_pInfo; }
	static unsigned int Distance( RGBQUAD& rgb1, RGBQUAD& rgb2 );

protected:
	HANDLE			DIBHandle();
	BOOL			OpenBMP( CString &csFileName );
	BOOL			OpenGIF( CString &csFileName );
	BOOL			OpenTIFF( CString &csFileName );
	BOOL			SaveBMP( CString &csFileName );
	BOOL			SaveGIF( CString &csFileName );
	BOOL			SaveTIFF( CString &csFileName );
	void			CreateGammaCurve();
	void			Expand( int nXDest, int nYDest, int xRatio, int yRatio, CDIB &dibSrc, int xSrc, int ySrc, int nSWidth, int nSHeight );

	unsigned char *	m_pBits;
	PBITMAPINFO		m_pInfo;
	RGBQUAD *		m_pRGB;
	void *			m_pVoid; 
	BYTE **			m_pLinePtr;
	int				height;
	int				bytes;
	int				width;
	int				m_nBits;
	int				m_nFlags;
	BOOL			m_bUseGamma;
	float			m_fGamma;
	float			m_fOldGamma;
	unsigned char	Gamma[256];
	RGBQUAD			CacheQuad[256];
	char			CachePtr[256];
};

#endif /* !__CDIB__ */