C++ 長條圖匹配演算法代碼

//////////////////////////////////////////////////////////////////////////
//函數名稱: 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]];
  }
 }
}