ref: e413ed20a2c3346e7097c45f7cb5f2bb3b9c0849
parent: b2528a256aea033d80baa4241c2988a2d0d58ee7
parent: b2178aacc023a7df9ef872927814edf3c0cde9c4
author: Ethan Hugg <ethanhugg@gmail.com>
date: Wed Jan 29 02:04:19 EST 2014
Merge pull request #254 from mstorsjo/directory-typofix Fix a typo in a directory name
--- a/codec/processing/build/win32/WelsVP_2008.vcproj
+++ b/codec/processing/build/win32/WelsVP_2008.vcproj
@@ -730,11 +730,11 @@
Name="BackgroundDetection"
>
<File
- RelativePath="..\..\src\backgounddetection\BackgroundDetection.cpp"
+ RelativePath="..\..\src\backgrounddetection\BackgroundDetection.cpp"
>
</File>
<File
- RelativePath="..\..\src\backgounddetection\BackgroundDetection.h"
+ RelativePath="..\..\src\backgrounddetection\BackgroundDetection.h"
>
</File>
</Filter>
--- a/codec/processing/src/backgounddetection/BackgroundDetection.cpp
+++ /dev/null
@@ -1,388 +1,0 @@
-/*!
- * \copy
- * Copyright (c) 2013, Cisco Systems
- * All rights reserved.
- *
- * Redistribution and use in source and binary forms, with or without
- * modification, are permitted provided that the following conditions
- * are met:
- *
- * * Redistributions of source code must retain the above copyright
- * notice, this list of conditions and the following disclaimer.
- *
- * * Redistributions in binary form must reproduce the above copyright
- * notice, this list of conditions and the following disclaimer in
- * the documentation and/or other materials provided with the
- * distribution.
- *
- * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
- * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
- * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
- * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
- * COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
- * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
- * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
- * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
- * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
- * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
- * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
- * POSSIBILITY OF SUCH DAMAGE.
- *
- */
-
-#include "BackgroundDetection.h"
-
-WELSVP_NAMESPACE_BEGIN
-
-#define LOG2_BGD_OU_SIZE (4)
-#define LOG2_BGD_OU_SIZE_UV (LOG2_BGD_OU_SIZE-1)
-#define BGD_OU_SIZE (1<<LOG2_BGD_OU_SIZE)
-#define BGD_OU_SIZE_UV (BGD_OU_SIZE>>1)
-#define BGD_THD_SAD (2*BGD_OU_SIZE*BGD_OU_SIZE)
-#define BGD_THD_ASD_UV (4*BGD_OU_SIZE_UV)
-#define LOG2_MB_SIZE (4)
-#define OU_SIZE_IN_MB (BGD_OU_SIZE >> 4)
-#define Q_FACTOR (8)
-#define BGD_DELTA_QP_THD (3)
-
-#define OU_LEFT (0x01)
-#define OU_RIGHT (0x02)
-#define OU_TOP (0x04)
-#define OU_BOTTOM (0x08)
-
-CBackgroundDetection::CBackgroundDetection (int32_t iCpuFlag) {- m_eMethod = METHOD_BACKGROUND_DETECTION;
- WelsMemset (&m_BgdParam, 0, sizeof (m_BgdParam));
- m_iLargestFrameSize = 0;
-}
-
-CBackgroundDetection::~CBackgroundDetection() {- FreeOUArrayMemory();
-}
-
-EResult CBackgroundDetection::Process (int32_t iType, SPixMap* pSrcPixMap, SPixMap* pRefPixMap) {- EResult eReturn = RET_INVALIDPARAM;
-
- if (pSrcPixMap == NULL || pRefPixMap == NULL)
- return eReturn;
-
- m_BgdParam.pCur[0] = (uint8_t*)pSrcPixMap->pPixel[0];
- m_BgdParam.pCur[1] = (uint8_t*)pSrcPixMap->pPixel[1];
- m_BgdParam.pCur[2] = (uint8_t*)pSrcPixMap->pPixel[2];
- m_BgdParam.pRef[0] = (uint8_t*)pRefPixMap->pPixel[0];
- m_BgdParam.pRef[1] = (uint8_t*)pRefPixMap->pPixel[1];
- m_BgdParam.pRef[2] = (uint8_t*)pRefPixMap->pPixel[2];
- m_BgdParam.iBgdWidth = pSrcPixMap->sRect.iRectWidth;
- m_BgdParam.iBgdHeight = pSrcPixMap->sRect.iRectHeight;
- m_BgdParam.iStride[0] = pSrcPixMap->iStride[0];
- m_BgdParam.iStride[1] = pSrcPixMap->iStride[1];
- m_BgdParam.iStride[2] = pSrcPixMap->iStride[2];
-
- int32_t iCurFrameSize = m_BgdParam.iBgdWidth * m_BgdParam.iBgdHeight;
- if (m_BgdParam.pOU_array == NULL || iCurFrameSize > m_iLargestFrameSize) {- FreeOUArrayMemory();
- m_BgdParam.pOU_array = AllocateOUArrayMemory (m_BgdParam.iBgdWidth, m_BgdParam.iBgdHeight);
- m_iLargestFrameSize = iCurFrameSize;
- }
-
- if (m_BgdParam.pOU_array == NULL)
- return eReturn;
-
- BackgroundDetection (&m_BgdParam);
-
- return RET_SUCCESS;
-}
-
-EResult CBackgroundDetection::Set (int32_t iType, void* pParam) {- if (pParam == NULL) {- return RET_INVALIDPARAM;
- }
-
- SBGDInterface* pInterface = (SBGDInterface*)pParam;
-
- m_BgdParam.pBackgroundMbFlag = (int8_t*)pInterface->pBackgroundMbFlag;
- m_BgdParam.pCalcRes = pInterface->pCalcRes;
-
- return RET_SUCCESS;
-}
-
-inline SBackgroundOU* CBackgroundDetection::AllocateOUArrayMemory (int32_t iWidth, int32_t iHeight) {- int32_t iMaxOUWidth = (BGD_OU_SIZE - 1 + iWidth) >> LOG2_BGD_OU_SIZE;
- int32_t iMaxOUHeight = (BGD_OU_SIZE - 1 + iHeight) >> LOG2_BGD_OU_SIZE;
- return (SBackgroundOU*)WelsMalloc (iMaxOUWidth * iMaxOUHeight * sizeof (SBackgroundOU));
-}
-
-inline void CBackgroundDetection::FreeOUArrayMemory() {- _SafeFree (m_BgdParam.pOU_array);
-}
-
-void CBackgroundDetection::GetOUParameters (SVAACalcResult* sVaaCalcInfo, int32_t iMbIndex, int32_t iMbWidth,
- SBackgroundOU* pBgdOU) {- int32_t iSubSD[4];
- uint8_t iSubMAD[4];
- int32_t iSubSAD[4];
-
- uint8_t (*pMad8x8)[4];
- int32_t (*pSad8x8)[4];
- int32_t (*pSd8x8)[4];
-
- pSad8x8 = sVaaCalcInfo->pSad8x8;
- pMad8x8 = sVaaCalcInfo->pMad8x8;
- pSd8x8 = sVaaCalcInfo->pSumOfDiff8x8;
-
- iSubSAD[0] = pSad8x8[iMbIndex][0];
- iSubSAD[1] = pSad8x8[iMbIndex][1];
- iSubSAD[2] = pSad8x8[iMbIndex][2];
- iSubSAD[3] = pSad8x8[iMbIndex][3];
-
- iSubSD[0] = pSd8x8[iMbIndex][0];
- iSubSD[1] = pSd8x8[iMbIndex][1];
- iSubSD[2] = pSd8x8[iMbIndex][2];
- iSubSD[3] = pSd8x8[iMbIndex][3];
-
- iSubMAD[0] = pMad8x8[iMbIndex][0];
- iSubMAD[1] = pMad8x8[iMbIndex][1];
- iSubMAD[2] = pMad8x8[iMbIndex][2];
- iSubMAD[3] = pMad8x8[iMbIndex][3];
-
- pBgdOU->iSD = iSubSD[0] + iSubSD[1] + iSubSD[2] + iSubSD[3];
- pBgdOU->iSAD = iSubSAD[0] + iSubSAD[1] + iSubSAD[2] + iSubSAD[3];
- pBgdOU->iSD = WELS_ABS (pBgdOU->iSD);
-
- // get the max absolute difference (MAD) of OU and min value of the MAD of sub-blocks of OU
- pBgdOU->iMAD = WELS_MAX (WELS_MAX (iSubMAD[0], iSubMAD[1]), WELS_MAX (iSubMAD[2], iSubMAD[3]));
- pBgdOU->iMinSubMad = WELS_MIN (WELS_MIN (iSubMAD[0], iSubMAD[1]), WELS_MIN (iSubMAD[2], iSubMAD[3]));
-
- // get difference between the max and min SD of the SDs of sub-blocks of OU
- pBgdOU->iMaxDiffSubSd = WELS_MAX (WELS_MAX (iSubSD[0], iSubSD[1]), WELS_MAX (iSubSD[2], iSubSD[3])) -
- WELS_MIN (WELS_MIN (iSubSD[0], iSubSD[1]), WELS_MIN (iSubSD[2], iSubSD[3]));
-}
-
-void CBackgroundDetection::ForegroundBackgroundDivision (vBGDParam* pBgdParam) {- int32_t iPicWidthInOU = pBgdParam->iBgdWidth >> LOG2_BGD_OU_SIZE;
- int32_t iPicHeightInOU = pBgdParam->iBgdHeight >> LOG2_BGD_OU_SIZE;
- int32_t iPicWidthInMb = (15 + pBgdParam->iBgdWidth) >> 4;
-
- SBackgroundOU* pBackgroundOU = pBgdParam->pOU_array;
-
- for (int32_t j = 0; j < iPicHeightInOU; j ++) {- for (int32_t i = 0; i < iPicWidthInOU; i++) {- GetOUParameters (pBgdParam->pCalcRes, (j * iPicWidthInMb + i) << (LOG2_BGD_OU_SIZE - LOG2_MB_SIZE), iPicWidthInMb,
- pBackgroundOU);
-
- pBackgroundOU->iBackgroundFlag = 0;
- if (pBackgroundOU->iMAD > 63) {- pBackgroundOU++;
- continue;
- }
- if ((pBackgroundOU->iMaxDiffSubSd <= pBackgroundOU->iSAD >> 3
- || pBackgroundOU->iMaxDiffSubSd <= (BGD_OU_SIZE * Q_FACTOR))
- && pBackgroundOU->iSAD < (BGD_THD_SAD << 1)) { //BGD_OU_SIZE*BGD_OU_SIZE>>2- if (pBackgroundOU->iSAD <= BGD_OU_SIZE * Q_FACTOR) {- pBackgroundOU->iBackgroundFlag = 1;
- } else {- pBackgroundOU->iBackgroundFlag = pBackgroundOU->iSAD < BGD_THD_SAD ?
- (pBackgroundOU->iSD < (pBackgroundOU->iSAD * 3) >> 2) :
- (pBackgroundOU->iSD << 1 < pBackgroundOU->iSAD);
- }
- }
- pBackgroundOU++;
- }
- }
-}
-inline int32_t CBackgroundDetection::CalculateAsdChromaEdge (uint8_t* pOriRef, uint8_t* pOriCur, int32_t iStride) {- int32_t ASD = 0;
- int32_t idx;
- for (idx = 0; idx < BGD_OU_SIZE_UV; idx++) {- ASD += *pOriCur - *pOriRef;
- pOriRef += iStride;
- pOriCur += iStride;
- }
- return WELS_ABS (ASD);
-}
-
-inline bool_t CBackgroundDetection::ForegroundDilation23Luma (SBackgroundOU* pBackgroundOU,
- SBackgroundOU* pOUNeighbours[]) {- SBackgroundOU* pOU_L = pOUNeighbours[0];
- SBackgroundOU* pOU_R = pOUNeighbours[1];
- SBackgroundOU* pOU_U = pOUNeighbours[2];
- SBackgroundOU* pOU_D = pOUNeighbours[3];
-
- if (pBackgroundOU->iMAD > pBackgroundOU->iMinSubMad << 1) {- int32_t iMaxNbrForegroundMad;
- int32_t iMaxNbrBackgroundMad;
- int32_t aBackgroundMad[4];
- int32_t aForegroundMad[4];
-
- aForegroundMad[0] = (pOU_L->iBackgroundFlag - 1) & pOU_L->iMAD;
- aForegroundMad[1] = (pOU_R->iBackgroundFlag - 1) & pOU_R->iMAD;
- aForegroundMad[2] = (pOU_U->iBackgroundFlag - 1) & pOU_U->iMAD;
- aForegroundMad[3] = (pOU_D->iBackgroundFlag - 1) & pOU_D->iMAD;
- iMaxNbrForegroundMad = WELS_MAX (WELS_MAX (aForegroundMad[0], aForegroundMad[1]), WELS_MAX (aForegroundMad[2],
- aForegroundMad[3]));
-
- aBackgroundMad[0] = ((!pOU_L->iBackgroundFlag) - 1) & pOU_L->iMAD;
- aBackgroundMad[1] = ((!pOU_R->iBackgroundFlag) - 1) & pOU_R->iMAD;
- aBackgroundMad[2] = ((!pOU_U->iBackgroundFlag) - 1) & pOU_U->iMAD;
- aBackgroundMad[3] = ((!pOU_D->iBackgroundFlag) - 1) & pOU_D->iMAD;
- iMaxNbrBackgroundMad = WELS_MAX (WELS_MAX (aBackgroundMad[0], aBackgroundMad[1]), WELS_MAX (aBackgroundMad[2],
- aBackgroundMad[3]));
-
- return ((iMaxNbrForegroundMad > pBackgroundOU->iMinSubMad << 2) || (pBackgroundOU->iMAD > iMaxNbrBackgroundMad << 1
- && pBackgroundOU->iMAD <= (iMaxNbrForegroundMad * 3) >> 1));
- }
- return 0;
-}
-
-inline bool_t CBackgroundDetection::ForegroundDilation23Chroma (int8_t iNeighbourForegroundFlags,
- int32_t iStartSamplePos, int32_t iPicStrideUV, vBGDParam* pBgdParam) {- static const int8_t kaOUPos[4] = {OU_LEFT, OU_RIGHT, OU_TOP, OU_BOTTOM};- int32_t aEdgeOffset[4] = {0, BGD_OU_SIZE_UV - 1, 0, iPicStrideUV* (BGD_OU_SIZE_UV - 1)};- int32_t iStride[4] = {iPicStrideUV, iPicStrideUV, 1, 1};-
- // V component first, high probability because V stands for red color and human skin colors have more weight on this component
- for (int32_t i = 0; i < 4; i++) {- if (iNeighbourForegroundFlags & kaOUPos[i]) {- uint8_t* pRefC = pBgdParam->pRef[2] + iStartSamplePos + aEdgeOffset[i];
- uint8_t* pCurC = pBgdParam->pCur[2] + iStartSamplePos + aEdgeOffset[i];
- if (CalculateAsdChromaEdge (pRefC, pCurC, iStride[i]) > BGD_THD_ASD_UV) {- return 1;
- }
- }
- }
- // U component, which stands for blue color, low probability
- for (int32_t i = 0; i < 4; i++) {- if (iNeighbourForegroundFlags & kaOUPos[i]) {- uint8_t* pRefC = pBgdParam->pRef[1] + iStartSamplePos + aEdgeOffset[i];
- uint8_t* pCurC = pBgdParam->pCur[1] + iStartSamplePos + aEdgeOffset[i];
- if (CalculateAsdChromaEdge (pRefC, pCurC, iStride[i]) > BGD_THD_ASD_UV) {- return 1;
- }
- }
- }
-
- return 0;
-}
-
-inline void CBackgroundDetection::ForegroundDilation (SBackgroundOU* pBackgroundOU, SBackgroundOU* pOUNeighbours[],
- vBGDParam* pBgdParam, int32_t iChromaSampleStartPos) {- int32_t iPicStrideUV = pBgdParam->iStride[1];
- int32_t iSumNeighBackgroundFlags = pOUNeighbours[0]->iBackgroundFlag + pOUNeighbours[1]->iBackgroundFlag +
- pOUNeighbours[2]->iBackgroundFlag + pOUNeighbours[3]->iBackgroundFlag;
-
- if (pBackgroundOU->iSAD > BGD_OU_SIZE * Q_FACTOR) {- switch (iSumNeighBackgroundFlags) {- case 0:
- case 1:
- pBackgroundOU->iBackgroundFlag = 0;
- break;
- case 2:
- case 3:
- pBackgroundOU->iBackgroundFlag = !ForegroundDilation23Luma (pBackgroundOU, pOUNeighbours);
-
- // chroma component check
- if (pBackgroundOU->iBackgroundFlag == 1) {- int8_t iNeighbourForegroundFlags = !pOUNeighbours[0]->iBackgroundFlag | ((!pOUNeighbours[1]->iBackgroundFlag) << 1)
- | ((!pOUNeighbours[2]->iBackgroundFlag) << 2) | ((!pOUNeighbours[3]->iBackgroundFlag) << 3);
- pBackgroundOU->iBackgroundFlag = !ForegroundDilation23Chroma (iNeighbourForegroundFlags, iChromaSampleStartPos,
- iPicStrideUV, pBgdParam);
- }
- break;
- default:
- break;
- }
- }
-}
-inline void CBackgroundDetection::BackgroundErosion (SBackgroundOU* pBackgroundOU, SBackgroundOU* pOUNeighbours[]) {- if (pBackgroundOU->iMaxDiffSubSd <= (BGD_OU_SIZE * Q_FACTOR)) { //BGD_OU_SIZE*BGD_OU_SIZE>>2- int32_t iSumNeighBackgroundFlags = pOUNeighbours[0]->iBackgroundFlag + pOUNeighbours[1]->iBackgroundFlag +
- pOUNeighbours[2]->iBackgroundFlag + pOUNeighbours[3]->iBackgroundFlag;
- int32_t sumNbrBGsad = (pOUNeighbours[0]->iSAD & (-pOUNeighbours[0]->iBackgroundFlag)) + (pOUNeighbours[2]->iSAD &
- (-pOUNeighbours[2]->iBackgroundFlag))
- + (pOUNeighbours[1]->iSAD & (-pOUNeighbours[1]->iBackgroundFlag)) + (pOUNeighbours[3]->iSAD &
- (-pOUNeighbours[3]->iBackgroundFlag));
- if (pBackgroundOU->iSAD * iSumNeighBackgroundFlags <= (3 * sumNbrBGsad) >> 1) {- if (iSumNeighBackgroundFlags == 4) {- pBackgroundOU->iBackgroundFlag = 1;
- } else {- if ((pOUNeighbours[0]->iBackgroundFlag & pOUNeighbours[1]->iBackgroundFlag)
- || (pOUNeighbours[2]->iBackgroundFlag & pOUNeighbours[3]->iBackgroundFlag)) {- pBackgroundOU->iBackgroundFlag = !ForegroundDilation23Luma (pBackgroundOU, pOUNeighbours);
- }
- }
- }
- }
-}
-
-inline void CBackgroundDetection::SetBackgroundMbFlag (int8_t* pBackgroundMbFlag, int32_t iPicWidthInMb,
- int32_t iBackgroundMbFlag) {- *pBackgroundMbFlag = iBackgroundMbFlag;
-}
-
-inline void CBackgroundDetection::UpperOUForegroundCheck (SBackgroundOU* pCurOU, int8_t* pBackgroundMbFlag,
- int32_t iPicWidthInOU, int32_t iPicWidthInMb) {- if (pCurOU->iSAD > BGD_OU_SIZE * Q_FACTOR) {- SBackgroundOU* pOU_L = pCurOU - 1;
- SBackgroundOU* pOU_R = pCurOU + 1;
- SBackgroundOU* pOU_U = pCurOU - iPicWidthInOU;
- SBackgroundOU* pOU_D = pCurOU + iPicWidthInOU;
- if (pOU_L->iBackgroundFlag + pOU_R->iBackgroundFlag + pOU_U->iBackgroundFlag + pOU_D->iBackgroundFlag <= 1) {- SetBackgroundMbFlag (pBackgroundMbFlag, iPicWidthInMb, 0);
- pCurOU->iBackgroundFlag = 0;
- }
- }
-}
-
-void CBackgroundDetection::ForegroundDilationAndBackgroundErosion (vBGDParam* pBgdParam) {- int32_t iPicStrideUV = pBgdParam->iStride[1];
- int32_t iPicWidthInOU = pBgdParam->iBgdWidth >> LOG2_BGD_OU_SIZE;
- int32_t iPicHeightInOU = pBgdParam->iBgdHeight >> LOG2_BGD_OU_SIZE;
- int32_t iOUStrideUV = iPicStrideUV << (LOG2_BGD_OU_SIZE - 1);
- int32_t iPicWidthInMb = (15 + pBgdParam->iBgdWidth) >> 4;
-
- SBackgroundOU* pBackgroundOU = pBgdParam->pOU_array;
- int8_t* pVaaBackgroundMbFlag = (int8_t*)pBgdParam->pBackgroundMbFlag;
- SBackgroundOU* pOUNeighbours[4];//0: left; 1: right; 2: top; 3: bottom
-
- pBackgroundOU = pBgdParam->pOU_array;
- pOUNeighbours[2] = pBackgroundOU;//top OU
- for (int32_t j = 0; j < iPicHeightInOU; j ++) {- int8_t* pRowSkipFlag = pVaaBackgroundMbFlag;
- pOUNeighbours[0] = pBackgroundOU;//left OU
- pOUNeighbours[3] = pBackgroundOU + (iPicWidthInOU & ((j == iPicHeightInOU - 1) - 1)); //bottom OU
- for (int32_t i = 0; i < iPicWidthInOU; i++) {- pOUNeighbours[1] = pBackgroundOU + (i < iPicWidthInOU - 1); //right OU
-
- if (pBackgroundOU->iBackgroundFlag)
- ForegroundDilation (pBackgroundOU, pOUNeighbours, pBgdParam, j * iOUStrideUV + (i << LOG2_BGD_OU_SIZE_UV));
- else
- BackgroundErosion (pBackgroundOU, pOUNeighbours);
-
- // check the up OU
- if (j > 1 && i > 0 && i < iPicWidthInOU - 1 && pOUNeighbours[2]->iBackgroundFlag == 1) {- UpperOUForegroundCheck (pOUNeighbours[2], pRowSkipFlag - OU_SIZE_IN_MB * iPicWidthInMb, iPicWidthInOU, iPicWidthInMb);
- }
-
- SetBackgroundMbFlag (pRowSkipFlag, iPicWidthInMb, pBackgroundOU->iBackgroundFlag);
-
- // preparation for the next OU
- pRowSkipFlag += OU_SIZE_IN_MB;
- pOUNeighbours[0] = pBackgroundOU;
- pOUNeighbours[2]++;
- pOUNeighbours[3]++;
- pBackgroundOU++;
- }
- pOUNeighbours[2] = pBackgroundOU - iPicWidthInOU;
- pVaaBackgroundMbFlag += OU_SIZE_IN_MB * iPicWidthInMb;
- }
-}
-
-void CBackgroundDetection::BackgroundDetection (vBGDParam* pBgdParam) {- // 1st step: foreground/background coarse division
- ForegroundBackgroundDivision (pBgdParam);
-
- // 2nd step: foreground dilation and background erosion
- ForegroundDilationAndBackgroundErosion (pBgdParam);
-}
-
-WELSVP_NAMESPACE_END
--- a/codec/processing/src/backgounddetection/BackgroundDetection.h
+++ /dev/null
@@ -1,106 +1,0 @@
-/*!
- * \copy
- * Copyright (c) 2011-2013, Cisco Systems
- * All rights reserved.
- *
- * Redistribution and use in source and binary forms, with or without
- * modification, are permitted provided that the following conditions
- * are met:
- *
- * * Redistributions of source code must retain the above copyright
- * notice, this list of conditions and the following disclaimer.
- *
- * * Redistributions in binary form must reproduce the above copyright
- * notice, this list of conditions and the following disclaimer in
- * the documentation and/or other materials provided with the
- * distribution.
- *
- * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
- * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
- * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
- * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
- * COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
- * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
- * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
- * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
- * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
- * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
- * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
- * POSSIBILITY OF SUCH DAMAGE.
- *
- * \file : BackgroundDetection.h
- *
- * \brief : background detection class of wels video processor class
- *
- * \date : 2011/03/17
- *
- * \description : 1. rewrite the package code of background detection class
- *
- */
-
-#ifndef WELSVP_BACKGROUNDDETECTION_H
-#define WELSVP_BACKGROUNDDETECTION_H
-
-#include "../common/util.h"
-#include "../common/memory.h"
-#include "../common/WelsFrameWork.h"
-#include "../../interface/IWelsVP.h"
-
-WELSVP_NAMESPACE_BEGIN
-
-typedef struct {- int32_t iBackgroundFlag;
- int32_t iSAD;
- int32_t iSD;
- int32_t iMAD;
- int32_t iMinSubMad;
- int32_t iMaxDiffSubSd;
-} SBackgroundOU;
-
-class CBackgroundDetection : public IStrategy {- public:
- CBackgroundDetection (int32_t iCpuFlag);
- ~CBackgroundDetection();
-
- EResult Process (int32_t iType, SPixMap* pSrc, SPixMap* pRef);
- EResult Set (int32_t iType, void* pParam);
-
- private:
- struct vBGDParam {- uint8_t* pCur[3];
- uint8_t* pRef[3];
- int32_t iBgdWidth;
- int32_t iBgdHeight;
- int32_t iStride[3];
- SBackgroundOU* pOU_array;
- int8_t* pBackgroundMbFlag;
- SVAACalcResult* pCalcRes;
- } m_BgdParam;
-
- int32_t m_iLargestFrameSize;
-
- private:
- inline SBackgroundOU* AllocateOUArrayMemory (int32_t iWidth, int32_t iHeight);
- inline void FreeOUArrayMemory();
- inline int32_t CalculateAsdChromaEdge (uint8_t* pOriRef, uint8_t* pOriCur, int32_t iStride);
- inline bool_t ForegroundDilation23Luma (SBackgroundOU* pBackgroundOU,
- SBackgroundOU* pOUNeighbours[]); //Foreground_Dilation_2_3_Luma
- inline bool_t ForegroundDilation23Chroma (int8_t iNeighbourForegroundFlags, int32_t iStartSamplePos,
- int32_t iPicStrideUV, vBGDParam* pBgdParam);//Foreground_Dilation_2_3_Chroma
- inline void ForegroundDilation (SBackgroundOU* pBackgroundOU, SBackgroundOU* pOUNeighbours[], vBGDParam* pBgdParam,
- int32_t iChromaSampleStartPos);
- inline void BackgroundErosion (SBackgroundOU* pBackgroundOU, SBackgroundOU* pOUNeighbours[]);
- inline void SetBackgroundMbFlag (int8_t* pBackgroundMbFlag, int32_t iPicWidthInMb, int32_t iBackgroundMbFlag);
- inline void UpperOUForegroundCheck (SBackgroundOU* pCurOU, int8_t* pBackgroundMbFlag, int32_t iPicWidthInOU,
- int32_t iPicWidthInMb);
-
- void GetOUParameters (SVAACalcResult* sVaaCalcInfo, int32_t iMbIndex, int32_t iMbWidth,
- SBackgroundOU* pBackgroundOU);
- void ForegroundBackgroundDivision (vBGDParam* pBgdParam);
- void ForegroundDilationAndBackgroundErosion (vBGDParam* pBgdParam);
- void BackgroundDetection (vBGDParam* pBgdParam);
-};
-
-WELSVP_NAMESPACE_END
-
-#endif
--- /dev/null
+++ b/codec/processing/src/backgrounddetection/BackgroundDetection.cpp
@@ -1,0 +1,388 @@
+/*!
+ * \copy
+ * Copyright (c) 2013, Cisco Systems
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ *
+ * * Redistributions of source code must retain the above copyright
+ * notice, this list of conditions and the following disclaimer.
+ *
+ * * Redistributions in binary form must reproduce the above copyright
+ * notice, this list of conditions and the following disclaimer in
+ * the documentation and/or other materials provided with the
+ * distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+ * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+ * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
+ * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
+ * COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
+ * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
+ * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
+ * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+ * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
+ * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
+ * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
+ * POSSIBILITY OF SUCH DAMAGE.
+ *
+ */
+
+#include "BackgroundDetection.h"
+
+WELSVP_NAMESPACE_BEGIN
+
+#define LOG2_BGD_OU_SIZE (4)
+#define LOG2_BGD_OU_SIZE_UV (LOG2_BGD_OU_SIZE-1)
+#define BGD_OU_SIZE (1<<LOG2_BGD_OU_SIZE)
+#define BGD_OU_SIZE_UV (BGD_OU_SIZE>>1)
+#define BGD_THD_SAD (2*BGD_OU_SIZE*BGD_OU_SIZE)
+#define BGD_THD_ASD_UV (4*BGD_OU_SIZE_UV)
+#define LOG2_MB_SIZE (4)
+#define OU_SIZE_IN_MB (BGD_OU_SIZE >> 4)
+#define Q_FACTOR (8)
+#define BGD_DELTA_QP_THD (3)
+
+#define OU_LEFT (0x01)
+#define OU_RIGHT (0x02)
+#define OU_TOP (0x04)
+#define OU_BOTTOM (0x08)
+
+CBackgroundDetection::CBackgroundDetection (int32_t iCpuFlag) {+ m_eMethod = METHOD_BACKGROUND_DETECTION;
+ WelsMemset (&m_BgdParam, 0, sizeof (m_BgdParam));
+ m_iLargestFrameSize = 0;
+}
+
+CBackgroundDetection::~CBackgroundDetection() {+ FreeOUArrayMemory();
+}
+
+EResult CBackgroundDetection::Process (int32_t iType, SPixMap* pSrcPixMap, SPixMap* pRefPixMap) {+ EResult eReturn = RET_INVALIDPARAM;
+
+ if (pSrcPixMap == NULL || pRefPixMap == NULL)
+ return eReturn;
+
+ m_BgdParam.pCur[0] = (uint8_t*)pSrcPixMap->pPixel[0];
+ m_BgdParam.pCur[1] = (uint8_t*)pSrcPixMap->pPixel[1];
+ m_BgdParam.pCur[2] = (uint8_t*)pSrcPixMap->pPixel[2];
+ m_BgdParam.pRef[0] = (uint8_t*)pRefPixMap->pPixel[0];
+ m_BgdParam.pRef[1] = (uint8_t*)pRefPixMap->pPixel[1];
+ m_BgdParam.pRef[2] = (uint8_t*)pRefPixMap->pPixel[2];
+ m_BgdParam.iBgdWidth = pSrcPixMap->sRect.iRectWidth;
+ m_BgdParam.iBgdHeight = pSrcPixMap->sRect.iRectHeight;
+ m_BgdParam.iStride[0] = pSrcPixMap->iStride[0];
+ m_BgdParam.iStride[1] = pSrcPixMap->iStride[1];
+ m_BgdParam.iStride[2] = pSrcPixMap->iStride[2];
+
+ int32_t iCurFrameSize = m_BgdParam.iBgdWidth * m_BgdParam.iBgdHeight;
+ if (m_BgdParam.pOU_array == NULL || iCurFrameSize > m_iLargestFrameSize) {+ FreeOUArrayMemory();
+ m_BgdParam.pOU_array = AllocateOUArrayMemory (m_BgdParam.iBgdWidth, m_BgdParam.iBgdHeight);
+ m_iLargestFrameSize = iCurFrameSize;
+ }
+
+ if (m_BgdParam.pOU_array == NULL)
+ return eReturn;
+
+ BackgroundDetection (&m_BgdParam);
+
+ return RET_SUCCESS;
+}
+
+EResult CBackgroundDetection::Set (int32_t iType, void* pParam) {+ if (pParam == NULL) {+ return RET_INVALIDPARAM;
+ }
+
+ SBGDInterface* pInterface = (SBGDInterface*)pParam;
+
+ m_BgdParam.pBackgroundMbFlag = (int8_t*)pInterface->pBackgroundMbFlag;
+ m_BgdParam.pCalcRes = pInterface->pCalcRes;
+
+ return RET_SUCCESS;
+}
+
+inline SBackgroundOU* CBackgroundDetection::AllocateOUArrayMemory (int32_t iWidth, int32_t iHeight) {+ int32_t iMaxOUWidth = (BGD_OU_SIZE - 1 + iWidth) >> LOG2_BGD_OU_SIZE;
+ int32_t iMaxOUHeight = (BGD_OU_SIZE - 1 + iHeight) >> LOG2_BGD_OU_SIZE;
+ return (SBackgroundOU*)WelsMalloc (iMaxOUWidth * iMaxOUHeight * sizeof (SBackgroundOU));
+}
+
+inline void CBackgroundDetection::FreeOUArrayMemory() {+ _SafeFree (m_BgdParam.pOU_array);
+}
+
+void CBackgroundDetection::GetOUParameters (SVAACalcResult* sVaaCalcInfo, int32_t iMbIndex, int32_t iMbWidth,
+ SBackgroundOU* pBgdOU) {+ int32_t iSubSD[4];
+ uint8_t iSubMAD[4];
+ int32_t iSubSAD[4];
+
+ uint8_t (*pMad8x8)[4];
+ int32_t (*pSad8x8)[4];
+ int32_t (*pSd8x8)[4];
+
+ pSad8x8 = sVaaCalcInfo->pSad8x8;
+ pMad8x8 = sVaaCalcInfo->pMad8x8;
+ pSd8x8 = sVaaCalcInfo->pSumOfDiff8x8;
+
+ iSubSAD[0] = pSad8x8[iMbIndex][0];
+ iSubSAD[1] = pSad8x8[iMbIndex][1];
+ iSubSAD[2] = pSad8x8[iMbIndex][2];
+ iSubSAD[3] = pSad8x8[iMbIndex][3];
+
+ iSubSD[0] = pSd8x8[iMbIndex][0];
+ iSubSD[1] = pSd8x8[iMbIndex][1];
+ iSubSD[2] = pSd8x8[iMbIndex][2];
+ iSubSD[3] = pSd8x8[iMbIndex][3];
+
+ iSubMAD[0] = pMad8x8[iMbIndex][0];
+ iSubMAD[1] = pMad8x8[iMbIndex][1];
+ iSubMAD[2] = pMad8x8[iMbIndex][2];
+ iSubMAD[3] = pMad8x8[iMbIndex][3];
+
+ pBgdOU->iSD = iSubSD[0] + iSubSD[1] + iSubSD[2] + iSubSD[3];
+ pBgdOU->iSAD = iSubSAD[0] + iSubSAD[1] + iSubSAD[2] + iSubSAD[3];
+ pBgdOU->iSD = WELS_ABS (pBgdOU->iSD);
+
+ // get the max absolute difference (MAD) of OU and min value of the MAD of sub-blocks of OU
+ pBgdOU->iMAD = WELS_MAX (WELS_MAX (iSubMAD[0], iSubMAD[1]), WELS_MAX (iSubMAD[2], iSubMAD[3]));
+ pBgdOU->iMinSubMad = WELS_MIN (WELS_MIN (iSubMAD[0], iSubMAD[1]), WELS_MIN (iSubMAD[2], iSubMAD[3]));
+
+ // get difference between the max and min SD of the SDs of sub-blocks of OU
+ pBgdOU->iMaxDiffSubSd = WELS_MAX (WELS_MAX (iSubSD[0], iSubSD[1]), WELS_MAX (iSubSD[2], iSubSD[3])) -
+ WELS_MIN (WELS_MIN (iSubSD[0], iSubSD[1]), WELS_MIN (iSubSD[2], iSubSD[3]));
+}
+
+void CBackgroundDetection::ForegroundBackgroundDivision (vBGDParam* pBgdParam) {+ int32_t iPicWidthInOU = pBgdParam->iBgdWidth >> LOG2_BGD_OU_SIZE;
+ int32_t iPicHeightInOU = pBgdParam->iBgdHeight >> LOG2_BGD_OU_SIZE;
+ int32_t iPicWidthInMb = (15 + pBgdParam->iBgdWidth) >> 4;
+
+ SBackgroundOU* pBackgroundOU = pBgdParam->pOU_array;
+
+ for (int32_t j = 0; j < iPicHeightInOU; j ++) {+ for (int32_t i = 0; i < iPicWidthInOU; i++) {+ GetOUParameters (pBgdParam->pCalcRes, (j * iPicWidthInMb + i) << (LOG2_BGD_OU_SIZE - LOG2_MB_SIZE), iPicWidthInMb,
+ pBackgroundOU);
+
+ pBackgroundOU->iBackgroundFlag = 0;
+ if (pBackgroundOU->iMAD > 63) {+ pBackgroundOU++;
+ continue;
+ }
+ if ((pBackgroundOU->iMaxDiffSubSd <= pBackgroundOU->iSAD >> 3
+ || pBackgroundOU->iMaxDiffSubSd <= (BGD_OU_SIZE * Q_FACTOR))
+ && pBackgroundOU->iSAD < (BGD_THD_SAD << 1)) { //BGD_OU_SIZE*BGD_OU_SIZE>>2+ if (pBackgroundOU->iSAD <= BGD_OU_SIZE * Q_FACTOR) {+ pBackgroundOU->iBackgroundFlag = 1;
+ } else {+ pBackgroundOU->iBackgroundFlag = pBackgroundOU->iSAD < BGD_THD_SAD ?
+ (pBackgroundOU->iSD < (pBackgroundOU->iSAD * 3) >> 2) :
+ (pBackgroundOU->iSD << 1 < pBackgroundOU->iSAD);
+ }
+ }
+ pBackgroundOU++;
+ }
+ }
+}
+inline int32_t CBackgroundDetection::CalculateAsdChromaEdge (uint8_t* pOriRef, uint8_t* pOriCur, int32_t iStride) {+ int32_t ASD = 0;
+ int32_t idx;
+ for (idx = 0; idx < BGD_OU_SIZE_UV; idx++) {+ ASD += *pOriCur - *pOriRef;
+ pOriRef += iStride;
+ pOriCur += iStride;
+ }
+ return WELS_ABS (ASD);
+}
+
+inline bool_t CBackgroundDetection::ForegroundDilation23Luma (SBackgroundOU* pBackgroundOU,
+ SBackgroundOU* pOUNeighbours[]) {+ SBackgroundOU* pOU_L = pOUNeighbours[0];
+ SBackgroundOU* pOU_R = pOUNeighbours[1];
+ SBackgroundOU* pOU_U = pOUNeighbours[2];
+ SBackgroundOU* pOU_D = pOUNeighbours[3];
+
+ if (pBackgroundOU->iMAD > pBackgroundOU->iMinSubMad << 1) {+ int32_t iMaxNbrForegroundMad;
+ int32_t iMaxNbrBackgroundMad;
+ int32_t aBackgroundMad[4];
+ int32_t aForegroundMad[4];
+
+ aForegroundMad[0] = (pOU_L->iBackgroundFlag - 1) & pOU_L->iMAD;
+ aForegroundMad[1] = (pOU_R->iBackgroundFlag - 1) & pOU_R->iMAD;
+ aForegroundMad[2] = (pOU_U->iBackgroundFlag - 1) & pOU_U->iMAD;
+ aForegroundMad[3] = (pOU_D->iBackgroundFlag - 1) & pOU_D->iMAD;
+ iMaxNbrForegroundMad = WELS_MAX (WELS_MAX (aForegroundMad[0], aForegroundMad[1]), WELS_MAX (aForegroundMad[2],
+ aForegroundMad[3]));
+
+ aBackgroundMad[0] = ((!pOU_L->iBackgroundFlag) - 1) & pOU_L->iMAD;
+ aBackgroundMad[1] = ((!pOU_R->iBackgroundFlag) - 1) & pOU_R->iMAD;
+ aBackgroundMad[2] = ((!pOU_U->iBackgroundFlag) - 1) & pOU_U->iMAD;
+ aBackgroundMad[3] = ((!pOU_D->iBackgroundFlag) - 1) & pOU_D->iMAD;
+ iMaxNbrBackgroundMad = WELS_MAX (WELS_MAX (aBackgroundMad[0], aBackgroundMad[1]), WELS_MAX (aBackgroundMad[2],
+ aBackgroundMad[3]));
+
+ return ((iMaxNbrForegroundMad > pBackgroundOU->iMinSubMad << 2) || (pBackgroundOU->iMAD > iMaxNbrBackgroundMad << 1
+ && pBackgroundOU->iMAD <= (iMaxNbrForegroundMad * 3) >> 1));
+ }
+ return 0;
+}
+
+inline bool_t CBackgroundDetection::ForegroundDilation23Chroma (int8_t iNeighbourForegroundFlags,
+ int32_t iStartSamplePos, int32_t iPicStrideUV, vBGDParam* pBgdParam) {+ static const int8_t kaOUPos[4] = {OU_LEFT, OU_RIGHT, OU_TOP, OU_BOTTOM};+ int32_t aEdgeOffset[4] = {0, BGD_OU_SIZE_UV - 1, 0, iPicStrideUV* (BGD_OU_SIZE_UV - 1)};+ int32_t iStride[4] = {iPicStrideUV, iPicStrideUV, 1, 1};+
+ // V component first, high probability because V stands for red color and human skin colors have more weight on this component
+ for (int32_t i = 0; i < 4; i++) {+ if (iNeighbourForegroundFlags & kaOUPos[i]) {+ uint8_t* pRefC = pBgdParam->pRef[2] + iStartSamplePos + aEdgeOffset[i];
+ uint8_t* pCurC = pBgdParam->pCur[2] + iStartSamplePos + aEdgeOffset[i];
+ if (CalculateAsdChromaEdge (pRefC, pCurC, iStride[i]) > BGD_THD_ASD_UV) {+ return 1;
+ }
+ }
+ }
+ // U component, which stands for blue color, low probability
+ for (int32_t i = 0; i < 4; i++) {+ if (iNeighbourForegroundFlags & kaOUPos[i]) {+ uint8_t* pRefC = pBgdParam->pRef[1] + iStartSamplePos + aEdgeOffset[i];
+ uint8_t* pCurC = pBgdParam->pCur[1] + iStartSamplePos + aEdgeOffset[i];
+ if (CalculateAsdChromaEdge (pRefC, pCurC, iStride[i]) > BGD_THD_ASD_UV) {+ return 1;
+ }
+ }
+ }
+
+ return 0;
+}
+
+inline void CBackgroundDetection::ForegroundDilation (SBackgroundOU* pBackgroundOU, SBackgroundOU* pOUNeighbours[],
+ vBGDParam* pBgdParam, int32_t iChromaSampleStartPos) {+ int32_t iPicStrideUV = pBgdParam->iStride[1];
+ int32_t iSumNeighBackgroundFlags = pOUNeighbours[0]->iBackgroundFlag + pOUNeighbours[1]->iBackgroundFlag +
+ pOUNeighbours[2]->iBackgroundFlag + pOUNeighbours[3]->iBackgroundFlag;
+
+ if (pBackgroundOU->iSAD > BGD_OU_SIZE * Q_FACTOR) {+ switch (iSumNeighBackgroundFlags) {+ case 0:
+ case 1:
+ pBackgroundOU->iBackgroundFlag = 0;
+ break;
+ case 2:
+ case 3:
+ pBackgroundOU->iBackgroundFlag = !ForegroundDilation23Luma (pBackgroundOU, pOUNeighbours);
+
+ // chroma component check
+ if (pBackgroundOU->iBackgroundFlag == 1) {+ int8_t iNeighbourForegroundFlags = !pOUNeighbours[0]->iBackgroundFlag | ((!pOUNeighbours[1]->iBackgroundFlag) << 1)
+ | ((!pOUNeighbours[2]->iBackgroundFlag) << 2) | ((!pOUNeighbours[3]->iBackgroundFlag) << 3);
+ pBackgroundOU->iBackgroundFlag = !ForegroundDilation23Chroma (iNeighbourForegroundFlags, iChromaSampleStartPos,
+ iPicStrideUV, pBgdParam);
+ }
+ break;
+ default:
+ break;
+ }
+ }
+}
+inline void CBackgroundDetection::BackgroundErosion (SBackgroundOU* pBackgroundOU, SBackgroundOU* pOUNeighbours[]) {+ if (pBackgroundOU->iMaxDiffSubSd <= (BGD_OU_SIZE * Q_FACTOR)) { //BGD_OU_SIZE*BGD_OU_SIZE>>2+ int32_t iSumNeighBackgroundFlags = pOUNeighbours[0]->iBackgroundFlag + pOUNeighbours[1]->iBackgroundFlag +
+ pOUNeighbours[2]->iBackgroundFlag + pOUNeighbours[3]->iBackgroundFlag;
+ int32_t sumNbrBGsad = (pOUNeighbours[0]->iSAD & (-pOUNeighbours[0]->iBackgroundFlag)) + (pOUNeighbours[2]->iSAD &
+ (-pOUNeighbours[2]->iBackgroundFlag))
+ + (pOUNeighbours[1]->iSAD & (-pOUNeighbours[1]->iBackgroundFlag)) + (pOUNeighbours[3]->iSAD &
+ (-pOUNeighbours[3]->iBackgroundFlag));
+ if (pBackgroundOU->iSAD * iSumNeighBackgroundFlags <= (3 * sumNbrBGsad) >> 1) {+ if (iSumNeighBackgroundFlags == 4) {+ pBackgroundOU->iBackgroundFlag = 1;
+ } else {+ if ((pOUNeighbours[0]->iBackgroundFlag & pOUNeighbours[1]->iBackgroundFlag)
+ || (pOUNeighbours[2]->iBackgroundFlag & pOUNeighbours[3]->iBackgroundFlag)) {+ pBackgroundOU->iBackgroundFlag = !ForegroundDilation23Luma (pBackgroundOU, pOUNeighbours);
+ }
+ }
+ }
+ }
+}
+
+inline void CBackgroundDetection::SetBackgroundMbFlag (int8_t* pBackgroundMbFlag, int32_t iPicWidthInMb,
+ int32_t iBackgroundMbFlag) {+ *pBackgroundMbFlag = iBackgroundMbFlag;
+}
+
+inline void CBackgroundDetection::UpperOUForegroundCheck (SBackgroundOU* pCurOU, int8_t* pBackgroundMbFlag,
+ int32_t iPicWidthInOU, int32_t iPicWidthInMb) {+ if (pCurOU->iSAD > BGD_OU_SIZE * Q_FACTOR) {+ SBackgroundOU* pOU_L = pCurOU - 1;
+ SBackgroundOU* pOU_R = pCurOU + 1;
+ SBackgroundOU* pOU_U = pCurOU - iPicWidthInOU;
+ SBackgroundOU* pOU_D = pCurOU + iPicWidthInOU;
+ if (pOU_L->iBackgroundFlag + pOU_R->iBackgroundFlag + pOU_U->iBackgroundFlag + pOU_D->iBackgroundFlag <= 1) {+ SetBackgroundMbFlag (pBackgroundMbFlag, iPicWidthInMb, 0);
+ pCurOU->iBackgroundFlag = 0;
+ }
+ }
+}
+
+void CBackgroundDetection::ForegroundDilationAndBackgroundErosion (vBGDParam* pBgdParam) {+ int32_t iPicStrideUV = pBgdParam->iStride[1];
+ int32_t iPicWidthInOU = pBgdParam->iBgdWidth >> LOG2_BGD_OU_SIZE;
+ int32_t iPicHeightInOU = pBgdParam->iBgdHeight >> LOG2_BGD_OU_SIZE;
+ int32_t iOUStrideUV = iPicStrideUV << (LOG2_BGD_OU_SIZE - 1);
+ int32_t iPicWidthInMb = (15 + pBgdParam->iBgdWidth) >> 4;
+
+ SBackgroundOU* pBackgroundOU = pBgdParam->pOU_array;
+ int8_t* pVaaBackgroundMbFlag = (int8_t*)pBgdParam->pBackgroundMbFlag;
+ SBackgroundOU* pOUNeighbours[4];//0: left; 1: right; 2: top; 3: bottom
+
+ pBackgroundOU = pBgdParam->pOU_array;
+ pOUNeighbours[2] = pBackgroundOU;//top OU
+ for (int32_t j = 0; j < iPicHeightInOU; j ++) {+ int8_t* pRowSkipFlag = pVaaBackgroundMbFlag;
+ pOUNeighbours[0] = pBackgroundOU;//left OU
+ pOUNeighbours[3] = pBackgroundOU + (iPicWidthInOU & ((j == iPicHeightInOU - 1) - 1)); //bottom OU
+ for (int32_t i = 0; i < iPicWidthInOU; i++) {+ pOUNeighbours[1] = pBackgroundOU + (i < iPicWidthInOU - 1); //right OU
+
+ if (pBackgroundOU->iBackgroundFlag)
+ ForegroundDilation (pBackgroundOU, pOUNeighbours, pBgdParam, j * iOUStrideUV + (i << LOG2_BGD_OU_SIZE_UV));
+ else
+ BackgroundErosion (pBackgroundOU, pOUNeighbours);
+
+ // check the up OU
+ if (j > 1 && i > 0 && i < iPicWidthInOU - 1 && pOUNeighbours[2]->iBackgroundFlag == 1) {+ UpperOUForegroundCheck (pOUNeighbours[2], pRowSkipFlag - OU_SIZE_IN_MB * iPicWidthInMb, iPicWidthInOU, iPicWidthInMb);
+ }
+
+ SetBackgroundMbFlag (pRowSkipFlag, iPicWidthInMb, pBackgroundOU->iBackgroundFlag);
+
+ // preparation for the next OU
+ pRowSkipFlag += OU_SIZE_IN_MB;
+ pOUNeighbours[0] = pBackgroundOU;
+ pOUNeighbours[2]++;
+ pOUNeighbours[3]++;
+ pBackgroundOU++;
+ }
+ pOUNeighbours[2] = pBackgroundOU - iPicWidthInOU;
+ pVaaBackgroundMbFlag += OU_SIZE_IN_MB * iPicWidthInMb;
+ }
+}
+
+void CBackgroundDetection::BackgroundDetection (vBGDParam* pBgdParam) {+ // 1st step: foreground/background coarse division
+ ForegroundBackgroundDivision (pBgdParam);
+
+ // 2nd step: foreground dilation and background erosion
+ ForegroundDilationAndBackgroundErosion (pBgdParam);
+}
+
+WELSVP_NAMESPACE_END
--- /dev/null
+++ b/codec/processing/src/backgrounddetection/BackgroundDetection.h
@@ -1,0 +1,106 @@
+/*!
+ * \copy
+ * Copyright (c) 2011-2013, Cisco Systems
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ *
+ * * Redistributions of source code must retain the above copyright
+ * notice, this list of conditions and the following disclaimer.
+ *
+ * * Redistributions in binary form must reproduce the above copyright
+ * notice, this list of conditions and the following disclaimer in
+ * the documentation and/or other materials provided with the
+ * distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+ * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+ * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
+ * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
+ * COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
+ * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
+ * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
+ * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+ * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
+ * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
+ * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
+ * POSSIBILITY OF SUCH DAMAGE.
+ *
+ * \file : BackgroundDetection.h
+ *
+ * \brief : background detection class of wels video processor class
+ *
+ * \date : 2011/03/17
+ *
+ * \description : 1. rewrite the package code of background detection class
+ *
+ */
+
+#ifndef WELSVP_BACKGROUNDDETECTION_H
+#define WELSVP_BACKGROUNDDETECTION_H
+
+#include "../common/util.h"
+#include "../common/memory.h"
+#include "../common/WelsFrameWork.h"
+#include "../../interface/IWelsVP.h"
+
+WELSVP_NAMESPACE_BEGIN
+
+typedef struct {+ int32_t iBackgroundFlag;
+ int32_t iSAD;
+ int32_t iSD;
+ int32_t iMAD;
+ int32_t iMinSubMad;
+ int32_t iMaxDiffSubSd;
+} SBackgroundOU;
+
+class CBackgroundDetection : public IStrategy {+ public:
+ CBackgroundDetection (int32_t iCpuFlag);
+ ~CBackgroundDetection();
+
+ EResult Process (int32_t iType, SPixMap* pSrc, SPixMap* pRef);
+ EResult Set (int32_t iType, void* pParam);
+
+ private:
+ struct vBGDParam {+ uint8_t* pCur[3];
+ uint8_t* pRef[3];
+ int32_t iBgdWidth;
+ int32_t iBgdHeight;
+ int32_t iStride[3];
+ SBackgroundOU* pOU_array;
+ int8_t* pBackgroundMbFlag;
+ SVAACalcResult* pCalcRes;
+ } m_BgdParam;
+
+ int32_t m_iLargestFrameSize;
+
+ private:
+ inline SBackgroundOU* AllocateOUArrayMemory (int32_t iWidth, int32_t iHeight);
+ inline void FreeOUArrayMemory();
+ inline int32_t CalculateAsdChromaEdge (uint8_t* pOriRef, uint8_t* pOriCur, int32_t iStride);
+ inline bool_t ForegroundDilation23Luma (SBackgroundOU* pBackgroundOU,
+ SBackgroundOU* pOUNeighbours[]); //Foreground_Dilation_2_3_Luma
+ inline bool_t ForegroundDilation23Chroma (int8_t iNeighbourForegroundFlags, int32_t iStartSamplePos,
+ int32_t iPicStrideUV, vBGDParam* pBgdParam);//Foreground_Dilation_2_3_Chroma
+ inline void ForegroundDilation (SBackgroundOU* pBackgroundOU, SBackgroundOU* pOUNeighbours[], vBGDParam* pBgdParam,
+ int32_t iChromaSampleStartPos);
+ inline void BackgroundErosion (SBackgroundOU* pBackgroundOU, SBackgroundOU* pOUNeighbours[]);
+ inline void SetBackgroundMbFlag (int8_t* pBackgroundMbFlag, int32_t iPicWidthInMb, int32_t iBackgroundMbFlag);
+ inline void UpperOUForegroundCheck (SBackgroundOU* pCurOU, int8_t* pBackgroundMbFlag, int32_t iPicWidthInOU,
+ int32_t iPicWidthInMb);
+
+ void GetOUParameters (SVAACalcResult* sVaaCalcInfo, int32_t iMbIndex, int32_t iMbWidth,
+ SBackgroundOU* pBackgroundOU);
+ void ForegroundBackgroundDivision (vBGDParam* pBgdParam);
+ void ForegroundDilationAndBackgroundErosion (vBGDParam* pBgdParam);
+ void BackgroundDetection (vBGDParam* pBgdParam);
+};
+
+WELSVP_NAMESPACE_END
+
+#endif
--- a/codec/processing/src/common/WelsFrameWork.cpp
+++ b/codec/processing/src/common/WelsFrameWork.cpp
@@ -35,7 +35,7 @@
#include "../downsample/downsample.h"
#include "../scenechangedetection/SceneChangeDetection.h"
#include "../vaacalc/vaacalculation.h"
-#include "../backgounddetection/BackgroundDetection.h"
+#include "../backgrounddetection/BackgroundDetection.h"
#include "../adaptivequantization/AdaptiveQuantization.h"
#include "../complexityanalysis/ComplexityAnalysis.h"
#include "../imagerotate/imagerotate.h"
--- a/codec/processing/targets.mk
+++ b/codec/processing/targets.mk
@@ -1,7 +1,7 @@
PROCESSING_SRCDIR=codec/processing
PROCESSING_CPP_SRCS=\
$(PROCESSING_SRCDIR)/./src/adaptivequantization/AdaptiveQuantization.cpp\
- $(PROCESSING_SRCDIR)/./src/backgounddetection/BackgroundDetection.cpp\
+ $(PROCESSING_SRCDIR)/./src/backgrounddetection/BackgroundDetection.cpp\
$(PROCESSING_SRCDIR)/./src/common/memory.cpp\
$(PROCESSING_SRCDIR)/./src/common/thread.cpp\
$(PROCESSING_SRCDIR)/./src/common/WelsFrameWork.cpp\