/*! * \copy *     Copyright (c)  2009-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    picture_handle.c * * \brief   picture pData handling * * \date    5/20/2009 Created * *************************************************************************************/#include "picture_handle.h"#include "svc_motion_estimate.h" namespace WelsEnc {/*! * \brief   alloc picture pData with borders for each plane based width and height of picture * \param   cx              width of picture in pixels * \param   cy              height of picture in pixels * \param   need_data       need pData allocation * \pram    need_expand     need borders expanding * \return  successful if effective picture pointer returned, otherwise failed with NULL */SPicture* AllocPicture (CMemoryAlign* pMa, const int32_t kiWidth , const int32_t kiHeight,                        bool bNeedMbInfo, int32_t iNeedFeatureStorage) {  SPicture* pPic = NULL;  int32_t iPicWidth = 0;  int32_t iPicHeight = 0;   int32_t iPicChromaWidth       = 0;  int32_t iPicChromaHeight      = 0;  int32_t iLumaSize             = 0;  int32_t iChromaSize           = 0;   pPic = static_cast<SPicture*> (pMa->WelsMallocz (sizeof (SPicture), "pPic"));   WELS_VERIFY_RETURN_IF (NULL, NULL == pPic);   iPicWidth         = WELS_ALIGN (kiWidth, MB_WIDTH_LUMA) + (PADDING_LENGTH << 1);  // with width of horizon  iPicHeight        = WELS_ALIGN (kiHeight, MB_HEIGHT_LUMA) + (PADDING_LENGTH << 1);        // with height of vertical  iPicChromaWidth   = iPicWidth >> 1;  iPicChromaHeight  = iPicHeight >> 1;  iPicWidth         = WELS_ALIGN (iPicWidth,                                  32);  // 32(or 16 for chroma below) to match original imp. here instead of cache_line_size  iPicChromaWidth   = WELS_ALIGN (iPicChromaWidth, 16);  iLumaSize         = iPicWidth * iPicHeight;  iChromaSize       = iPicChromaWidth * iPicChromaHeight;   pPic->pBuffer = (uint8_t*)pMa->WelsMalloc (iLumaSize /* luma */                  + (iChromaSize << 1) /* Cb,Cr */                  , "pPic->pBuffer");  WELS_VERIFY_RETURN_PROC_IF (NULL, NULL == pPic->pBuffer, FreePicture (pMa, &pPic));  pPic->iLineSize[0]    = iPicWidth;  pPic->iLineSize[1]    = pPic->iLineSize[2]    = iPicChromaWidth;  pPic->pData[0]        = pPic->pBuffer + (1 + pPic->iLineSize[0]) * PADDING_LENGTH;  pPic->pData[1]        = pPic->pBuffer + iLumaSize + (((1 + pPic->iLineSize[1]) * PADDING_LENGTH) >> 1);  pPic->pData[2]        = pPic->pBuffer + iLumaSize + iChromaSize + (((1 + pPic->iLineSize[2]) * PADDING_LENGTH) >> 1);   pPic->iWidthInPixel   = kiWidth;  pPic->iHeightInPixel  = kiHeight;  pPic->iFrameNum       = -1;   pPic->bIsLongRef      = false;  pPic->iLongTermPicNum = -1;  pPic->uiRecieveConfirmed = 0;  pPic->iMarkFrameNum   = -1;   if (bNeedMbInfo) {    const uint32_t kuiCountMbNum = ((15 + kiWidth) >> 4) * ((15 + kiHeight) >> 4);     pPic->uiRefMbType = (uint32_t*)pMa->WelsMallocz (kuiCountMbNum * sizeof (uint32_t), "pPic->uiRefMbType");    WELS_VERIFY_RETURN_PROC_IF (NULL, NULL == pPic->uiRefMbType, FreePicture (pMa, &pPic));     pPic->pRefMbQp = (uint8_t*)pMa->WelsMallocz (kuiCountMbNum * sizeof (uint8_t), "pPic->pRefMbQp");    WELS_VERIFY_RETURN_PROC_IF (NULL, NULL == pPic->pRefMbQp, FreePicture (pMa, &pPic));     pPic->sMvList           = static_cast<SMVUnitXY*> (pMa->WelsMallocz (kuiCountMbNum * sizeof (SMVUnitXY),                              "pPic->sMvList"));    WELS_VERIFY_RETURN_PROC_IF (NULL, NULL == pPic->sMvList, FreePicture (pMa, &pPic));     pPic->pMbSkipSad       = (int32_t*)pMa->WelsMallocz (kuiCountMbNum * sizeof (int32_t), "pPic->pMbSkipSad");    WELS_VERIFY_RETURN_PROC_IF (NULL, NULL == pPic->pMbSkipSad, FreePicture (pMa, &pPic));  }   if (iNeedFeatureStorage) {    pPic->pScreenBlockFeatureStorage = static_cast<SScreenBlockFeatureStorage*> (pMa->WelsMallocz (sizeof (                                         SScreenBlockFeatureStorage), "pScreenBlockFeatureStorage"));    int32_t iReturn = RequestScreenBlockFeatureStorage (pMa, kiWidth,  kiHeight, iNeedFeatureStorage,                      pPic->pScreenBlockFeatureStorage);    WELS_VERIFY_RETURN_PROC_IF (NULL, ENC_RETURN_SUCCESS != iReturn, FreePicture (pMa, &pPic));  } else {    pPic->pScreenBlockFeatureStorage = NULL;  }  return pPic;} /*! * \brief   free picture pData planes * \param   pPic        picture pointer to be destoryed * \return  none */void FreePicture (CMemoryAlign* pMa, SPicture** ppPic) {  if (NULL != ppPic && NULL != *ppPic) {    SPicture* pPic = *ppPic;     if (NULL != pPic->pBuffer) {      pMa->WelsFree (pPic->pBuffer, "pPic->pBuffer");      pPic->pBuffer = NULL;    }    pPic->pBuffer          = NULL;    pPic->pData[0]         =      pPic->pData[1]       =        pPic->pData[2]     = NULL;    pPic->iLineSize[0]     =      pPic->iLineSize[1]   =        pPic->iLineSize[2] = 0;     pPic->iWidthInPixel         = 0;    pPic->iHeightInPixel        = 0;    pPic->iFrameNum             = -1;     pPic->bIsLongRef            = false;    pPic->uiRecieveConfirmed    = 0;    pPic->iLongTermPicNum       = -1;    pPic->iMarkFrameNum         = -1;     if (pPic->uiRefMbType) {      pMa->WelsFree (pPic->uiRefMbType, "pPic->uiRefMbType");      pPic->uiRefMbType = NULL;    }    if (pPic->pRefMbQp) {      pMa->WelsFree (pPic->pRefMbQp, "pPic->pRefMbQp");      pPic->pRefMbQp = NULL;    }     if (pPic->sMvList) {      pMa->WelsFree (pPic->sMvList, "pPic->sMvList");      pPic->sMvList = NULL;    }    if (pPic->pMbSkipSad) {      pMa->WelsFree (pPic->pMbSkipSad, "pPic->pMbSkipSad");      pPic->pMbSkipSad = NULL;    }     if (pPic->pScreenBlockFeatureStorage) {      ReleaseScreenBlockFeatureStorage (pMa, pPic->pScreenBlockFeatureStorage);      pMa->WelsFree (pPic->pScreenBlockFeatureStorage, "pPic->pScreenBlockFeatureStorage");      pPic->pScreenBlockFeatureStorage = NULL;    }     pMa->WelsFree (*ppPic, "pPic");    *ppPic = NULL;  }} } // namespace WelsEnc