CDIB.h

Go to the documentation of this file.

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

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__ */