C++ 直方图匹配算法代码

C++ 直方图匹配算法代码

分类：
VC++ 数字图像处理
2009-10-12 08:17
1907人阅读 评论(3)
收藏
举报

//////////////////////////////////////////////////////////////////////////

//函数名称: histeq()

//传入参数: 

//   BYTE*dstData  要匹配的灰度图像内存空间

//   double *srcArray 模版的直方图累积，并进行归一化，大小为256

//   int m_Width   匹配内存空间的宽度

//   int m_Height  匹配内存空间的高度

//   int m_pitch   匹配内存空间的每行所在内存大小

//////////////////////////////////////////////////////////////////////////

void histeq(BYTE*dstData,double *srcArray,int m_Width,int m_Height,int m_pitch)

{

 double dstHist[256];

 memset(dstHist,0,256 * sizeof(double));

 int i,j;

 double dstArray[256];

 memset(dstArray,0,256 * sizeof(double));

 //统计直方图

 for (i = 0;i < m_Height;i++)

 {

  for (j = 0;j < m_Width;j++)

  {

   dstHist[(int)dstData[i * m_pitch + j]]++;

  }

 }

 //计算直方图累积

 double m_Bytes = m_Width * m_Height;

 dstArray[0] = dstHist[0];

 for (i = 1;i < 256;i++)

 {

  dstArray[i] = dstArray[i - 1] + dstHist[i];

 }

 //直方图累积归一化

 for (i = 0;i < 256;i++)

 {

  dstArray[i] /= m_Bytes;

 }

 //直方图匹配

 double m_diffA,m_diffB;

 int k = 0;

 BYTE mapPixel[256];

 memset(mapPixel,0,256 * sizeof(BYTE));

 for (i = 0;i < 256;i++)

 {

  m_diffB = 1;

  for (j = k; j < 256;j++)

  {

   m_diffA = abs(dstArray[i] - srcArray[j]);

   if (m_diffA - m_diffB < 1.0E-5)

   {

    m_diffB = m_diffA;

    k = j;

   }

   else

   {

    k = j - 1;

    break;

   }

  }

  if (k == 255)

  {

   for (int l = i;l < 256;l++)

   {

    mapPixel[l] = (BYTE) k;

   }

   break;

  }

  mapPixel[i] = (BYTE) k;

 }

 //目标图像查找索引表

 for (i = 0;i < m_Height;i++)

 {

  for (j = 0;j < m_Width;j++)

  {

   dstData[i * m_pitch + j] = mapPixel[dstData[i * m_pitch + j]];

  }

 }

}