histogramspecify.cpp

一些自己做的关于图象处理的程序（如傅立叶变换

// HistogramSpecify.cpp: implementation of the CHistogramSpecify class.
//
//////////////////////////////////////////////////////////////////////
/**********************************************************************
* Copyright (c) 2003, Medical Image Processing Lab, Sichuan University 
* All rights reserved.
* 
* 文件名称：Equalize.cpp
* 摘    要：图像直方图规定化类CHistogramSpecify源文件
* 
* 当前版本：1.1 
* 作    者：jian wei zhang	
* 完成日期：2004年11月2日
************************************************************************/

#include "stdafx.h"
#include "photostar.h"
#include "HistogramSpecify.h"
#include "math.h"
#ifdef _DEBUG
#undef THIS_FILE
static char THIS_FILE[]=__FILE__;
#define new DEBUG_NEW
#endif

//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////

CHistogramSpecify::CHistogramSpecify()
{

}

CHistogramSpecify::~CHistogramSpecify()
{

}

BOOL CHistogramSpecify::OnHistogramSpecify()
{
	if (m_pImageObject == NULL)
	{
		return FALSE;
	}
	
	int iWidth = m_pImageObject->GetWidth();
	int iHeight = m_pImageObject->GetHeight();
	int iNumbits=m_pImageObject->GetNumBits();
	/******************************the old image*****************************/
	unsigned char *pOldBuffer, *pNewBuffer, *pOldBits, *pNewBits;
	BITMAPFILEHEADER *pOldBFH, *pNewBFH;
	BITMAPINFOHEADER *pOldBIH, *pNewBIH;
	RGBQUAD *pOldPalette, *pNewPalette;
	int nWidthBytes, nNumColors;
	
	pOldBuffer = (unsigned char *) m_pImageObject->GetDIBPointer(&nWidthBytes, m_pImageObject->GetNumBits());
	if (pOldBuffer == NULL) 
	{
		return FALSE;
	}
	
	pOldBFH = (BITMAPFILEHEADER *) pOldBuffer;
	pOldBIH = (BITMAPINFOHEADER *) &pOldBuffer[sizeof(BITMAPFILEHEADER)];
	nNumColors = m_pImageObject->GetNumColors();
	pOldPalette = (RGBQUAD *)  &pOldBuffer[sizeof(BITMAPFILEHEADER)+sizeof(BITMAPINFOHEADER)];
	pOldBits = (unsigned char *) &pOldBuffer[sizeof(BITMAPFILEHEADER)
		+sizeof(BITMAPINFOHEADER)+nNumColors*sizeof(RGBQUAD)];
	
	/*****************************the new image*****************************/
	DWORD dwNewSize;
	HGLOBAL hNewDib;
	
	//Allocate a global memory block for the new image
	dwNewSize = sizeof(BITMAPFILEHEADER) + sizeof(BITMAPINFOHEADER) 
		+ nNumColors * sizeof(RGBQUAD)
		+ nWidthBytes * (m_pImageObject->GetHeight());
	hNewDib = ::GlobalAlloc(GMEM_MOVEABLE | GMEM_ZEROINIT, dwNewSize);
	if (hNewDib == NULL)
	{
		m_pImageObject->m_nLastError = IMAGELIB_MEMORY_ALLOCATION_ERROR;
		::GlobalUnlock(m_pImageObject->GetDib());
		return FALSE;
	}
	
	//Get the pointer to the new memory block
	pNewBuffer = (unsigned char *) ::GlobalLock(hNewDib);
	if (pNewBuffer == NULL)
	{
		::GlobalFree( hNewDib );
		m_pImageObject->m_nLastError = IMAGELIB_MEMORY_LOCK_ERROR;
		::GlobalUnlock(m_pImageObject->GetDib());
		return FALSE;
	}
	
	pNewBFH = (BITMAPFILEHEADER *) pNewBuffer;
	pNewBIH = (BITMAPINFOHEADER *) &pNewBuffer[sizeof(BITMAPFILEHEADER)];
	pNewPalette = (RGBQUAD *) &pNewBuffer[sizeof(BITMAPFILEHEADER)+sizeof(BITMAPINFOHEADER)];
	pNewBits = (unsigned char *) &pNewBuffer[sizeof(BITMAPFILEHEADER)+sizeof(BITMAPINFOHEADER)+nNumColors*sizeof(RGBQUAD)];
	
	//Make a copy of the old image
	memcpy(pNewBuffer,pOldBuffer,dwNewSize);
    /*******************************Histogram Specify****************************/
	//部分数据的初始化
	int nSize=1<<iNumbits;
	int k=0,i,j,n;
	long* plCount=NULL;
	long* splCount=NULL;
	long totalcount;
	double u=0;
	//开始直方图规定
	switch(iNumbits)
	{
	case 8 :
      plCount=new long[nSize];
	  splCount=new long[nSize+1];//暂时分配物理内存
	  if (plCount==NULL||splCount==NULL)
	  {
		  return FALSE;
	  }
	  //清零
	  for(k=0;k<256;k++)
	  {
		  plCount[k]=0;
		  
	  }
	  for(k=0;k<257;k++)
	  {
         splCount[k]=0;
	  }
	  for(j=0;j<iHeight;j++)
	  { 
		  for(i=0;i<iWidth;i++)
		  {
			  plCount[pOldBits[(iHeight-1-j)*nWidthBytes+i]]++;
			  //注意取图像中(i, j )点像素的方式!!!!!
		  }
	  }	  
      for(j=0;j<iHeight;j++)
	  {  
		  for(i=0;i<iWidth;i++)
		  {
                k=pOldBits[(iHeight-1-j)*nWidthBytes+i];
				for(n=0;n<k+1;n++)
				{
				  totalcount+=plCount[n];//各灰度级象素点的累加
				}
				for(n=1;n<nSize+1;n++)
				{   
				   u=1.0/3.0*pow(n-1,-2.0/3.0)/(pow(255,1.0/3.0)-pow(0,1.0/3.0))*255;
                   splCount[n]=splCount[n-1]+(int)u;
				   if(splCount[n]==totalcount*255/iHeight/iWidth)
					{
					    pNewBits[(iHeight-1-j)*nWidthBytes+i]=n;
					}
				}
				for(k=0;k<257;k++)
				{
				   splCount[k]=0;
				} 
				totalcount=0;//清零
		  }
	  }
		break;
	case 16:
		break;
	default:
		break;
	}
    //Free the memory for the old image
	::GlobalUnlock(m_pImageObject->GetDib());
	::GlobalFree(m_pImageObject->GetDib());
	
	//Update the DIB in the object pointed by m_pImaeObject
	::GlobalUnlock(hNewDib); 
	m_pImageObject->SetDib(hNewDib);
	
	return TRUE;
}