#include "stdafx.h"
|
#include "BLOB_Tool.h"
|
|
CBlobTool::CBlobTool()
|
{
|
|
}
|
|
|
CBlobTool::~CBlobTool()
|
{
|
|
}
|
|
bool CBlobTool::BlobAnalysis(BYTE *pImageData, int nWidth, int nHeight, int nStep, ListEdgeBlobData &blobList, int nMergeRange)
|
{
|
if (pImageData==NULL) return false;
|
|
int nIndex;
|
int i, j, y, x;
|
ListEdgeBlobData pixelList;
|
|
for (i=0; i<nHeight; i++)
|
{
|
for (j=0; j<nWidth; j++)
|
{
|
nIndex = (i*nStep)+j;
|
if (pImageData[nIndex] == 255)
|
{
|
// push pixel
|
CEdgeBlobData *pPos = new CEdgeBlobData(j, i);
|
pixelList.push_back(pPos);
|
pImageData[nIndex] = 0;
|
|
int nPixelCount = 0;
|
int nTotalX = 0;
|
int nTotalY = 0;
|
int nMaxX, nMaxY;
|
int nMinX, nMinY;
|
|
nMinX = nMinY = INT_MAX;
|
nMaxX = nMaxY = INT_MIN;
|
|
ListEdgeBlobDataIt it;
|
while (pixelList.size()!=0)
|
{
|
// pop pixel
|
it = pixelList.begin();
|
CEdgeBlobData *pCurPos = *it;
|
|
x = (int)pCurPos->fCenterX;
|
y = (int)pCurPos->fCenterY;
|
|
int nTmpIndex;
|
for (int sy=(-nMergeRange); sy<(nMergeRange+1); sy++)
|
{
|
for (int sx=(-nMergeRange); sx<(nMergeRange+1); sx++)
|
{
|
if (((sx+x) > nWidth-1) || ((sx+x) < 0)) continue;
|
if (((sy+y) > nHeight-1) || ((sy+y) < 0)) continue;
|
if ((sx==0 && sy==0) || sx==sy || sx==(-sy)) continue;
|
|
nTmpIndex = ((y+sy)*nStep)+(x+sx);
|
|
if (pImageData[nTmpIndex] == 255)
|
{
|
// push pixel
|
CEdgeBlobData *pNode1 = new CEdgeBlobData(x+sx, y+sy);
|
pixelList.push_back(pNode1);
|
pImageData[nTmpIndex] = 0;
|
} // end if
|
} // end for sx
|
} // end for sy
|
|
// blob info
|
nMinX = (nMinX > x) ? x : nMinX;
|
nMinY = (nMinY > y) ? y : nMinY;
|
nMaxX = (nMaxX < x) ? x : nMaxX;
|
nMaxY = (nMaxY < y) ? y : nMaxY;
|
nTotalX += x;
|
nTotalY += y;
|
nPixelCount++;
|
|
// delete pixel
|
delete pCurPos;
|
pixelList.erase(it);
|
|
} // end while
|
pixelList.clear();
|
|
// push blob
|
CEdgeBlobData *pBlob = new CEdgeBlobData(nMinX, nMinY, nMaxX, nMaxY);
|
pBlob->fMassCenterX = float(nTotalX) / float(nPixelCount);
|
pBlob->fMassCenterY = float(nTotalY) / float(nPixelCount);
|
pBlob->nPixelCount = nPixelCount;
|
blobList.push_back(pBlob);
|
|
}
|
}
|
}
|
|
return true;
|
}
|
