// Dib.h: interface for the CDib class.
//
//////////////////////////////////////////////////////////////////////

#if !defined(AFX_DIB_H__AC952C3A_9B6B_4319_8D6E_E7F509348A88__INCLUDED_)
#define AFX_DIB_H__AC952C3A_9B6B_4319_8D6E_E7F509348A88__INCLUDED_

#if _MSC_VER > 1000
#pragma once
#endif // _MSC_VER > 1000

class CDib : public CObject 
{
public:
 CDib();
 virtual ~CDib();
//operations
public:
 // 用于操作DIB的函數聲明
 BOOL   ConstructPalette(HGLOBAL,CPalette* );         //構造邏輯調色板
 LPSTR  GetBits(LPSTR);                               //取得位圖數據的入口地址
 DWORD  GetWidth(LPSTR);                              //取得位圖的寬度
 DWORD  GetHeight(LPSTR);                             //取得位圖的高度
 WORD   GetPalSize(LPSTR);                            //取得調色板的大小
 WORD   GetColorNum(LPSTR);                            //取得位圖包含的顏色數目
 WORD   GetBitCount(LPSTR);                            //取得位圖的顏色深度
 HGLOBAL   LoadFile(CFile&);                          //從文件中加載位圖
 BOOL SaveFile(HGLOBAL hDib, CFile& file);

// 在對圖象進行處理時，針對位圖的字節寬度必須是4的倍數的這一要求，
// 我們設計了函數GetRequireWidth，來處理這種比較特殊的情況
 int     GetReqByteWidth(int );                     //轉換后的字節數GetRequireByteWidth
 long    GetRectWidth(LPCRECT );                    //取得區域的寬度
 long    GetRectHeight(LPCRECT);                    //取得區域的高度
public:
 void ClearMemory();
 void InitMembers();
public:
 LPBITMAPINFO        lpbminfo; // 指向BITMAPINFO結構的指針
 LPBITMAPINFOHEADER  lpbmihrd; //指向BITMAPINFOHEADER結構的指針
 BITMAPFILEHEADER bmfHeader;  //BITMAPFILEHEADER結構
 LPSTR    lpdib;      //指向DIB的指針
 LPSTR    lpDIBBits;  // DIB像素指針
 DWORD    dwDIBSize;  //DIB大小
 
 HGLOBAL    m_hDib;//DIB對象的句柄
 CPalette* m_palDIB;//調色板指針
};

#endif // !defined(AFX_DIB_H__AC952C3A_9B6B_4319_8D6E_E7F509348A88__INCLUDED_)

/////////////////////////////////////////////////////////////////////實現文件//////////////////////////////////////////////////////////////////////////////
// Dib.cpp: implementation of the CDib class.
//
//////////////////////////////////////////////////////////////////////

#include "stdafx.h"
#include "Dib.h"

#include <math.h>

#ifdef _DEBUG
#undef THIS_FILE
static char THIS_FILE[]=__FILE__;
#define new DEBUG_NEW
#endif

//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////
CDib::CDib()
{
 InitMembers();
}

CDib::~CDib()
{
 ClearMemory();
}

/*************************************************************************
 * 函數名稱：ConstructPalette(HGLOBAL hDIB, CPalette* pPal)
 * 函數參數:
 *   HGLOBAL hDIB，DIB對象的句柄
 *   CPalette* pPal，調色板的指針
 * 函數類型:BOOL
 * 函數說明:該函數按照DIB創建一個邏輯調色板 
 ************************************************************************/

BOOL CDib::ConstructPalette(HGLOBAL hDIB, CPalette* pPal)
{ 
 
 HANDLE hLogPal;// 邏輯調色板的句柄 
 int iLoop;// 循環變量 
 BOOL bSuccess = FALSE;// 創建結果 
 if (hDIB == NULL)//判斷是否是有效的DIB對象
 {  
  return FALSE;// 返回FALSE
 } 
 lpdib = (LPSTR) ::GlobalLock((HGLOBAL) hDIB);// 鎖定DIB
 lpbminfo= (LPBITMAPINFO)lpdib;   
 long wNumColors =GetColorNum(lpdib);// 獲取DIB中顏色表中的顏色數目 
 if (wNumColors != 0)
 {  
  hLogPal = ::GlobalAlloc(GHND, sizeof(LOGPALETTE)// 分配為邏輯調色板內存
         + sizeof(PALETTEENTRY)
         * wNumColors); 
  if (hLogPal == 0)// 如果失敗則退出
  {   
   ::GlobalUnlock((HGLOBAL) hDIB);// 解除鎖定   
   return FALSE;
  }  
  LPLOGPALETTE lpPal = (LPLOGPALETTE) ::GlobalLock((HGLOBAL) hLogPal);  
  
  lpPal->palVersion = 0x300;// 設置調色板版本號    
  lpPal->palNumEntries = (WORD)wNumColors;// 設置顏色數目  
  for (iLoop=0; iLoop<(int)wNumColors;iLoop++)// 讀取調色板
  {   
   lpPal->palPalEntry[iLoop].peRed =lpbminfo->bmiColors[iLoop].rgbRed;// 讀取三原色分量      
   lpPal->palPalEntry[iLoop].peGreen =lpbminfo->bmiColors[iLoop].rgbGreen;       
   lpPal->palPalEntry[iLoop].peBlue =lpbminfo->bmiColors[iLoop].rgbBlue;        
   lpPal->palPalEntry[iLoop].peFlags =0;// 保留位   
  }  
  bSuccess=pPal->CreatePalette(lpPal);// 按照邏輯調色板創建調色板，并返回指針   
  ::GlobalUnlock((HGLOBAL) hLogPal);// 解除鎖定    
  ::GlobalFree((HGLOBAL) hLogPal);// 釋放邏輯調色板
 }  
 ::GlobalUnlock((HGLOBAL) hDIB);// 解除鎖定  
 return bSuccess;// 返回結果
}

/*************************************************************************
 * 函數名稱：GetBits(LPSTR lpdib)
 * 函數參數:
 *   LPSTR lpdib，指向DIB對象的指針
 * 函數類型:LPSTR
 * 函數功能:計算DIB像素的起始位置，并返回指向它的指針
 ************************************************************************/

LPSTR CDib::GetBits(LPSTR lpdib)
{
 return (lpdib + ((LPBITMAPINFOHEADER)lpdib)->biSize+GetPalSize(lpdib));
// return (lpdib + *(LPDWORD)lpdib+GetPalSize(lpdib));
}

/*************************************************************************
 * 函數名稱：GetWidth(LPSTR lpdib)
 * 函數參數:
 *   LPSTR lpdib，指向DIB對象的指針
 * 函數類型:DWORD
 * 函數功能:該函數返回DIB中圖象的寬度
 ************************************************************************/

DWORD CDib::GetWidth(LPSTR lpdib)
{
 return ((LPBITMAPINFOHEADER)lpdib)->biWidth;//返回DIB寬度
}

/*************************************************************************
 * 函數名稱：GetHeight(LPSTR lpdib)
 * 函數參數:
 *   LPSTR lpdib ，指向DIB對象的指針
 * 函數類型:DWORD
 * 函數功能:該函數返回DIB中圖象的高度
 ************************************************************************/

DWORD CDib::GetHeight(LPSTR lpdib)
{
 return ((LPBITMAPINFOHEADER)lpdib)->biHeight;//返回DIB高度

}
/*************************************************************************
 * 函數名稱：GetPalSize(LPSTR lpdib)
 * 函數參數:
 *   LPSTR lpdib，指向DIB對象的指針
 * 函數類型:WORD
 * 函數功能:該函數返回DIB中調色板的大小
 ************************************************************************/

WORD CDib::GetPalSize(LPSTR lpdib)
{ 
 return (WORD)(GetColorNum(lpdib) * sizeof(RGBQUAD));// 計算DIB中調色板的大小
}

/*************************************************************************
 * 函數名稱：GetColorNum(LPSTR lpdib)
 * 函數參數:
 *   LPSTR lpdib，指向DIB對象的指針
 * 函數類型:WORD
 * 函數功能:該函數返回DIB中調色板的顏色的種數
 ************************************************************************/
WORD CDib::GetColorNum(LPSTR lpdib)
{ 
 long dwClrUsed = ((LPBITMAPINFOHEADER)lpdib)->biClrUsed; // 讀取dwClrUsed值 
 
 if (dwClrUsed != 0)
 {   
  return (WORD)dwClrUsed;// 如果dwClrUsed不為0，直接返回該值
 }
 else
 {
  dwClrUsed=(long)pow(2,24);
  return (WORD)dwClrUsed;
 }
}

/*************************************************************************
 * 函數名稱：GetBitCount(LPSTR lpdib)
 * 函數參數:
 *   LPSTR lpdib，指向DIB對象的指針
 * 函數類型:WORD
 * 函數功能:該函數返回DIBBitCount
 ************************************************************************/
WORD CDib::GetBitCount(LPSTR lpdib)
{
 return ((LPBITMAPINFOHEADER)lpdib)->biBitCount;// 返回位寬
}

/*************************************************************************
 * 函數名稱：LoadFile(CFile& file)
 * 函數參數:
 *   CFile& file，要讀取得文件文件CFile
 * 函數類型:HGLOBAL
 * 函數功能:將指定的文件中的DIB對象讀到指定的內存區域中
 *************************************************************************/

HGLOBAL CDib::LoadFile(CFile& file)
{

 DWORD dwFileSize;
 
 dwFileSize= file.GetLength();//獲取文件大小 
 if (file.Read((LPSTR)&bmfHeader, sizeof(bmfHeader)) != sizeof(bmfHeader))// 讀取DIB文件頭
 {  
  return NULL;// 大小不一致，返回NULL
 }
 // 如果文件類型不是"BM"，其16進制值為0x4d42，則返回并進行相應錯誤處理
 if (bmfHeader.bfType !=0x4d42)
 {  
  return NULL;// 如果不是則返回NULL
 } 
 m_hDib= (HGLOBAL) ::GlobalAlloc(GMEM_MOVEABLE | GMEM_ZEROINIT, dwFileSize-sizeof(BITMAPFILEHEADER));// 分配DIB內存
 if (m_hDib==NULL)
 {  
  return NULL;// 分配失敗，返回NULL
 }
 /////////////////////////////////////////////////////////////////////////
 //給CDib類的成員變量賦值 
 lpdib = (LPSTR) ::GlobalLock((HGLOBAL) m_hDib);// 鎖定 
 if (file.ReadHuge(lpdib, dwFileSize - sizeof(BITMAPFILEHEADER)) !=// 讀像素
  dwFileSize - sizeof(BITMAPFILEHEADER) )//大小不一致
 {     
  ::GlobalUnlock((HGLOBAL) m_hDib); // 解除鎖定   
  ::GlobalFree((HGLOBAL) m_hDib); // 釋放內存
  return NULL;
 }  
 ::GlobalUnlock((HGLOBAL) m_hDib);// 解除鎖定  
 return m_hDib;// 返回DIB句柄
}

/*************************************************************************
 * 函數名稱：GetReqByteWidth(int bits)
 * 函數參數:
 *   int bits，位數
 * 函數類型:int
 * 函數功能:獲取需要的行字節數，應為4的倍數
 *************************************************************************/
int  CDib::GetReqByteWidth(int bits)
{
 int getBytes=(bits + 31) / 32 * 4;
 return  getBytes;
}

/*************************************************************************
 * 函數名稱：GetRectWidth(LPCRECT lpRect)
 * 函數參數:
 *   LPCRECT lpRect，指向矩形區域的指針
 * 函數類型:long
 * 函數功能:獲取矩形區域的寬度
 *************************************************************************/

long  CDib::GetRectWidth(LPCRECT lpRect)
{
 long nWidth=lpRect->right - lpRect->left;
 return nWidth;
}
/*************************************************************************
 * 函數名稱：GetRectHeight(LPCRECT lpRect)
 * 函數參數:
 *   LPCRECT lpRect，指向矩形區域的指針
 * 函數類型:long
 * 函數功能:獲取矩形區域的高度
 *************************************************************************/
long  CDib::GetRectHeight(LPCRECT lpRect)
{
 long nHeight=lpRect->bottom - lpRect->top;
 return nHeight;
}

/*************************************************************************
 * 函數名稱：InitMembers()
 * 函數類型: void
 * 函數功能:初始化類的成員變量
 *************************************************************************/
void CDib::InitMembers()
{
 m_hDib=NULL;
 lpdib=NULL;
 lpDIBBits=NULL;
}

/*************************************************************************
 * 函數名稱：ClearMemory()
 * 函數類型: void
 * 函數功能:復位類的成員變量
 *************************************************************************/
void CDib::ClearMemory()
{
 if(m_hDib!=NULL)
  ::GlobalFree(m_hDib);
 lpdib=NULL;
}

/*************************************************************************
 * 函數名稱：SaveFile(HGLOBAL hDib, CFile& file)
 * 函數參數:
 *   HGLOBAL hDib，要保存的DIB
 *   CFile& file，保存文件CFile
 * 函數類型:BOOL
 * 函數功能:將指定的DIB對象保存到指定的CFile中
 *************************************************************************/

BOOL CDib::SaveFile(HGLOBAL hDib, CFile& file)
{
 if (hDib == NULL)
 {  
  return FALSE;// 如果DIB為空，返回FALSE
 } 
 lpbmihrd = (LPBITMAPINFOHEADER) ::GlobalLock((HGLOBAL) hDib);// 讀取BITMAPINFO結構，并鎖定 
 if (lpbmihrd == NULL)
 {  
  return FALSE;// 為空，返回FALSE
 }  
 bmfHeader.bfType ='BM'; // 填充文件頭 
 dwDIBSize = *(LPDWORD)lpbmihrd + GetPalSize((LPSTR)lpbmihrd);// 文件頭大小＋顏色表大小   
 DWORD dwBmBitsSize;// 像素的大小  
 dwBmBitsSize =GetReqByteWidth((lpbmihrd->biWidth)*((DWORD)lpbmihrd->biBitCount)) * lpbmihrd->biHeight;// 大小為Width * Height  
 dwDIBSize += dwBmBitsSize;// 計算后DIB每行字節數為4的倍數時的大小
 lpbmihrd->biSizeImage = dwBmBitsSize;// 更新biSizeImage 
 bmfHeader.bfSize = dwDIBSize + sizeof(BITMAPFILEHEADER);// 文件大小  
 bmfHeader.bfReserved1 = 0;// 兩個保留字
 bmfHeader.bfReserved2 = 0;
 bmfHeader.bfOffBits = (DWORD)sizeof(BITMAPFILEHEADER) + lpbmihrd->biSize// 計算偏移量bfOffBits
       + GetPalSize((LPSTR)lpbmihrd);
 file.Write(&bmfHeader, sizeof(BITMAPFILEHEADER));// 寫文件頭    
 file.WriteHuge(lpbmihrd, dwDIBSize);// 寫DIB頭和像素
 
 ::GlobalUnlock((HGLOBAL) hDib);// 解除鎖定 
 return TRUE;// 返回TRUE
}