FloodFill

Flood Fill을 쓸 일이 있어서 퍼다 쓸까 했었는데.. 최적화에 문제가 좀 있더군요.

그래서 최적화 적용해서 만든 김에 공유합니다.

PixelColor 관련된 부분과 비트맵의 폭, 높이 관련된 부분만 적절히 수정하면 바로 실제 코드에 적용할 수 있도록 해두었습니다.

void FloodFill( CPixelColor* pBmp, const int nPosX, const int nPosY, CPixelColor colorTarget, CPixelColor colorReplace ) { std::queue<CPixel*> qNode; int nStrife = ::GetStrifeFromBitmap(); int nHeight = ::GetHeightFromBitmap(); CPixelColor* pNode = pBmp + ( nPosX + ( nHeight - nPosY - 1 ) * nStrife ); auto IsTarget = [&] ( CPixelColor* pPixel ) -> bool { return ( *pPixel == colorTarget ); }; if( IsTarget( pNode )){ qNode.push( pNode ); size_t nSize = 0; do { nSize = qNode.size(); for( size_t i = nSize; i > 0; --i ){ auto pNode = qNode.front(); qNode.pop(); bool bLastUp = false; bool bLastDown = false; CPixelColor* pCur = pNode; bool bFirstUp = IsTarget( pCur - nStrife ); bool bFirstDown = IsTarget( pCur + nStrife ); auto FloodFillUpAndDown = [&]() { *pCur = colorReplace; if( IsTarget( pCur - nStrife )){ if( !bLastUp ){ qNode.push( pCur - nStrife ); } bLastUp = true; } else { bLastUp = false; } if( IsTarget( pCur + nStrife )){ if( !bLastDown ){ qNode.push( pCur + nStrife ); } bLastDown = true; } else { bLastDown = false; } }; for( pCur = pNode; IsTarget( pCur ); pCur-- ){ FloodFillUpAndDown(); } bLastUp = bFirstUp; bLastDown = bFirstDown; for( pCur = pNode + 1; IsTarget( pCur ); pCur++ ){ FloodFillUpAndDown(); } } } while( qNode.size() > 0 ); } }