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

 

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

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

http://blog.youkuaiyun.com/zhenxiaohui/article/details/4656597

//
//函数名称: 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]];
  }
 }
}