source: 3DVCSoftware/branches/HTM-10.2-dev3-MediaTek/source/Lib/TLibCommon/TComPicSym.cpp @ 904

Last change on this file since 904 was 872, checked in by tech, 11 years ago

Merged HTM-10.0-dev0@871. (MV-HEVC 7 HLS)
  • Property svn:eol-style set to native
File size: 14.8 KB
Line 
1/* The copyright in this software is being made available under the BSD
2 * License, included below. This software may be subject to other third party
3 * and contributor rights, including patent rights, and no such rights are
4 * granted under this license. 
5 *
6* Copyright (c) 2010-2014, ITU/ISO/IEC
7 * All rights reserved.
8 *
9 * Redistribution and use in source and binary forms, with or without
10 * modification, are permitted provided that the following conditions are met:
11 *
12 *  * Redistributions of source code must retain the above copyright notice,
13 *    this list of conditions and the following disclaimer.
14 *  * Redistributions in binary form must reproduce the above copyright notice,
15 *    this list of conditions and the following disclaimer in the documentation
16 *    and/or other materials provided with the distribution.
17 *  * Neither the name of the ITU/ISO/IEC nor the names of its contributors may
18 *    be used to endorse or promote products derived from this software without
19 *    specific prior written permission.
20 *
21 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
22 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS
25 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
26 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
27 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
28 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
29 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
30 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
31 * THE POSSIBILITY OF SUCH DAMAGE.
32 */
33
34/** \file     TComPicSym.cpp
35    \brief    picture symbol class
36*/
37
38#include "TComPicSym.h"
39#include "TComSampleAdaptiveOffset.h"
40
41//! \ingroup TLibCommon
42//! \{
43
44// ====================================================================================================================
45// Constructor / destructor / create / destroy
46// ====================================================================================================================
47
48TComPicSym::TComPicSym()
49:m_uiWidthInCU(0)
50,m_uiHeightInCU(0)
51,m_uiMaxCUWidth(0)
52,m_uiMaxCUHeight(0)
53,m_uiMinCUWidth(0)
54,m_uiMinCUHeight(0)
55,m_uhTotalDepth(0)
56,m_uiNumPartitions(0)
57,m_uiNumPartInWidth(0)
58,m_uiNumPartInHeight(0)
59,m_uiNumCUsInFrame(0)
60,m_apcTComSlice(NULL)
61,m_uiNumAllocatedSlice (0)
62,m_apcTComDataCU (NULL)
63,m_iNumColumnsMinus1 (0)
64,m_iNumRowsMinus1(0)
65,m_apcTComTile(NULL)
66,m_puiCUOrderMap(0)
67,m_puiTileIdxMap(NULL)
68,m_puiInverseCUOrderMap(NULL)
69,m_saoBlkParams(NULL)
70{};
71
72
73Void TComPicSym::create  ( Int iPicWidth, Int iPicHeight, UInt uiMaxWidth, UInt uiMaxHeight, UInt uiMaxDepth )
74{
75  UInt i;
76
77  m_uhTotalDepth      = uiMaxDepth;
78  m_uiNumPartitions   = 1<<(m_uhTotalDepth<<1);
79 
80  m_uiMaxCUWidth      = uiMaxWidth;
81  m_uiMaxCUHeight     = uiMaxHeight;
82 
83  m_uiMinCUWidth      = uiMaxWidth  >> m_uhTotalDepth;
84  m_uiMinCUHeight     = uiMaxHeight >> m_uhTotalDepth;
85 
86  m_uiNumPartInWidth  = m_uiMaxCUWidth  / m_uiMinCUWidth;
87  m_uiNumPartInHeight = m_uiMaxCUHeight / m_uiMinCUHeight;
88 
89  m_uiWidthInCU       = ( iPicWidth %m_uiMaxCUWidth  ) ? iPicWidth /m_uiMaxCUWidth  + 1 : iPicWidth /m_uiMaxCUWidth;
90  m_uiHeightInCU      = ( iPicHeight%m_uiMaxCUHeight ) ? iPicHeight/m_uiMaxCUHeight + 1 : iPicHeight/m_uiMaxCUHeight;
91 
92  m_uiNumCUsInFrame   = m_uiWidthInCU * m_uiHeightInCU;
93  m_apcTComDataCU     = new TComDataCU*[m_uiNumCUsInFrame];
94 
95  if (m_uiNumAllocatedSlice>0)
96  {
97    for ( i=0; i<m_uiNumAllocatedSlice ; i++ )
98    {
99      delete m_apcTComSlice[i];
100    }
101    delete [] m_apcTComSlice;
102  }
103  m_apcTComSlice      = new TComSlice*[m_uiNumCUsInFrame*m_uiNumPartitions]; 
104  m_apcTComSlice[0]   = new TComSlice;
105  m_uiNumAllocatedSlice = 1;
106  for ( i=0; i<m_uiNumCUsInFrame ; i++ )
107  {
108    m_apcTComDataCU[i] = new TComDataCU;
109    m_apcTComDataCU[i]->create( m_uiNumPartitions, m_uiMaxCUWidth, m_uiMaxCUHeight, false, m_uiMaxCUWidth >> m_uhTotalDepth
110#if ADAPTIVE_QP_SELECTION
111      , true
112#endif     
113      );
114  }
115
116  m_puiCUOrderMap = new UInt[m_uiNumCUsInFrame+1];
117  m_puiTileIdxMap = new UInt[m_uiNumCUsInFrame];
118  m_puiInverseCUOrderMap = new UInt[m_uiNumCUsInFrame+1];
119
120  for( i=0; i<m_uiNumCUsInFrame; i++ )
121  {
122    m_puiCUOrderMap[i] = i;
123    m_puiInverseCUOrderMap[i] = i;
124  }
125
126  m_saoBlkParams = new SAOBlkParam[m_uiNumCUsInFrame];
127}
128
129Void TComPicSym::destroy()
130{
131  if (m_uiNumAllocatedSlice>0)
132  {
133    for (Int i = 0; i<m_uiNumAllocatedSlice ; i++ )
134    {
135      delete m_apcTComSlice[i];
136    }
137    delete [] m_apcTComSlice;
138  }
139  m_apcTComSlice = NULL;
140 
141  for (Int i = 0; i < m_uiNumCUsInFrame; i++)
142  {
143    m_apcTComDataCU[i]->destroy();
144    delete m_apcTComDataCU[i];
145    m_apcTComDataCU[i] = NULL;
146  }
147  delete [] m_apcTComDataCU;
148  m_apcTComDataCU = NULL;
149
150  for(Int i = 0; i < (m_iNumColumnsMinus1+1)*(m_iNumRowsMinus1+1); i++ )
151  {
152    delete m_apcTComTile[i];
153  }
154  delete [] m_apcTComTile;
155
156  m_apcTComTile = NULL;
157
158  delete [] m_puiCUOrderMap;
159  m_puiCUOrderMap = NULL;
160
161  delete [] m_puiTileIdxMap;
162  m_puiTileIdxMap = NULL;
163
164  delete [] m_puiInverseCUOrderMap;
165  m_puiInverseCUOrderMap = NULL;
166 
167  if(m_saoBlkParams)
168  {
169    delete[] m_saoBlkParams; m_saoBlkParams = NULL;
170  }
171}
172
173Void TComPicSym::allocateNewSlice()
174{
175  m_apcTComSlice[m_uiNumAllocatedSlice ++] = new TComSlice;
176  if (m_uiNumAllocatedSlice>=2)
177  {
178    m_apcTComSlice[m_uiNumAllocatedSlice-1]->copySliceInfo( m_apcTComSlice[m_uiNumAllocatedSlice-2] );
179    m_apcTComSlice[m_uiNumAllocatedSlice-1]->initSlice();
180  }
181}
182
183Void TComPicSym::clearSliceBuffer()
184{
185  UInt i;
186  for (i = 1; i < m_uiNumAllocatedSlice; i++)
187  {
188    delete m_apcTComSlice[i];
189  }
190  m_uiNumAllocatedSlice = 1;
191}
192
193UInt TComPicSym::getPicSCUEncOrder( UInt SCUAddr )
194{ 
195  return getInverseCUOrderMap(SCUAddr/m_uiNumPartitions)*m_uiNumPartitions + SCUAddr%m_uiNumPartitions; 
196}
197
198UInt TComPicSym::getPicSCUAddr( UInt SCUEncOrder )
199{
200  return getCUOrderMap(SCUEncOrder/m_uiNumPartitions)*m_uiNumPartitions + SCUEncOrder%m_uiNumPartitions;
201}
202
203Void TComPicSym::xCreateTComTileArray()
204{
205  m_apcTComTile = new TComTile*[(m_iNumColumnsMinus1+1)*(m_iNumRowsMinus1+1)];
206  for( UInt i=0; i<(m_iNumColumnsMinus1+1)*(m_iNumRowsMinus1+1); i++ )
207  {
208    m_apcTComTile[i] = new TComTile;
209  }
210}
211
212Void TComPicSym::xInitTiles()
213{
214  UInt  uiTileIdx;
215  UInt  uiColumnIdx = 0;
216  UInt  uiRowIdx = 0;
217  UInt  uiRightEdgePosInCU;
218  UInt  uiBottomEdgePosInCU;
219  Int   i, j;
220
221  //initialize each tile of the current picture
222  for( uiRowIdx=0; uiRowIdx < m_iNumRowsMinus1+1; uiRowIdx++ )
223  {
224    for( uiColumnIdx=0; uiColumnIdx < m_iNumColumnsMinus1+1; uiColumnIdx++ )
225    {
226      uiTileIdx = uiRowIdx * (m_iNumColumnsMinus1+1) + uiColumnIdx;
227
228      //initialize the RightEdgePosInCU for each tile
229      uiRightEdgePosInCU = 0;
230      for( i=0; i <= uiColumnIdx; i++ )
231      {
232        uiRightEdgePosInCU += this->getTComTile(uiRowIdx * (m_iNumColumnsMinus1+1) + i)->getTileWidth();
233      }
234      this->getTComTile(uiTileIdx)->setRightEdgePosInCU(uiRightEdgePosInCU-1);
235
236      //initialize the BottomEdgePosInCU for each tile
237      uiBottomEdgePosInCU = 0;
238      for( i=0; i <= uiRowIdx; i++ )
239      {
240        uiBottomEdgePosInCU += this->getTComTile(i * (m_iNumColumnsMinus1+1) + uiColumnIdx)->getTileHeight();
241      }
242      this->getTComTile(uiTileIdx)->setBottomEdgePosInCU(uiBottomEdgePosInCU-1);
243
244      //initialize the FirstCUAddr for each tile
245      this->getTComTile(uiTileIdx)->setFirstCUAddr( (this->getTComTile(uiTileIdx)->getBottomEdgePosInCU() - this->getTComTile(uiTileIdx)->getTileHeight() +1)*m_uiWidthInCU + 
246        this->getTComTile(uiTileIdx)->getRightEdgePosInCU() - this->getTComTile(uiTileIdx)->getTileWidth() + 1);
247    }
248  }
249
250  //initialize the TileIdxMap
251  for( i=0; i<m_uiNumCUsInFrame; i++)
252  {
253    for(j=0; j < m_iNumColumnsMinus1+1; j++)
254    {
255      if(i % m_uiWidthInCU <= this->getTComTile(j)->getRightEdgePosInCU())
256      {
257        uiColumnIdx = j;
258        j = m_iNumColumnsMinus1+1;
259      }
260    }
261    for(j=0; j < m_iNumRowsMinus1+1; j++)
262    {
263      if(i/m_uiWidthInCU <= this->getTComTile(j*(m_iNumColumnsMinus1 + 1))->getBottomEdgePosInCU())
264      {
265        uiRowIdx = j;
266        j = m_iNumRowsMinus1 + 1;
267      }
268    }
269    m_puiTileIdxMap[i] = uiRowIdx * (m_iNumColumnsMinus1 + 1) + uiColumnIdx;
270  }
271
272}
273
274UInt TComPicSym::xCalculateNxtCUAddr( UInt uiCurrCUAddr )
275{
276  UInt  uiNxtCUAddr;
277  UInt  uiTileIdx;
278 
279  //get the tile index for the current LCU
280  uiTileIdx = this->getTileIdxMap(uiCurrCUAddr);
281
282  //get the raster scan address for the next LCU
283  if( uiCurrCUAddr % m_uiWidthInCU == this->getTComTile(uiTileIdx)->getRightEdgePosInCU() && uiCurrCUAddr / m_uiWidthInCU == this->getTComTile(uiTileIdx)->getBottomEdgePosInCU() )
284  //the current LCU is the last LCU of the tile
285  {
286    if(uiTileIdx == (m_iNumColumnsMinus1+1)*(m_iNumRowsMinus1+1)-1)
287    {
288      uiNxtCUAddr = m_uiNumCUsInFrame;
289    }
290    else
291    {
292      uiNxtCUAddr = this->getTComTile(uiTileIdx+1)->getFirstCUAddr();
293    }
294  } 
295  else //the current LCU is not the last LCU of the tile
296  {
297    if( uiCurrCUAddr % m_uiWidthInCU == this->getTComTile(uiTileIdx)->getRightEdgePosInCU() )  //the current LCU is on the rightmost edge of the tile
298    {
299      uiNxtCUAddr = uiCurrCUAddr + m_uiWidthInCU - this->getTComTile(uiTileIdx)->getTileWidth() + 1;
300    }
301    else
302    {
303      uiNxtCUAddr = uiCurrCUAddr + 1;
304    }
305  }
306
307  return uiNxtCUAddr;
308}
309
310Void TComPicSym::deriveLoopFilterBoundaryAvailibility(Int ctu,
311                                                      Bool& isLeftAvail,
312                                                      Bool& isRightAvail,
313                                                      Bool& isAboveAvail,
314                                                      Bool& isBelowAvail,
315                                                      Bool& isAboveLeftAvail,
316                                                      Bool& isAboveRightAvail,
317                                                      Bool& isBelowLeftAvail,
318                                                      Bool& isBelowRightAvail
319                                                      )
320{
321
322  isLeftAvail      = (ctu % m_uiWidthInCU != 0);
323  isRightAvail     = (ctu % m_uiWidthInCU != m_uiWidthInCU-1);
324  isAboveAvail     = (ctu >= m_uiWidthInCU );
325  isBelowAvail     = (ctu <  m_uiNumCUsInFrame - m_uiWidthInCU);
326  isAboveLeftAvail = (isAboveAvail && isLeftAvail);
327  isAboveRightAvail= (isAboveAvail && isRightAvail);
328  isBelowLeftAvail = (isBelowAvail && isLeftAvail);
329  isBelowRightAvail= (isBelowAvail && isRightAvail);
330
331  Bool isLoopFiltAcrossTilePPS = getCU(ctu)->getSlice()->getPPS()->getLoopFilterAcrossTilesEnabledFlag();
332
333  {
334    TComDataCU* ctuCurr  = getCU(ctu);
335    TComDataCU* ctuLeft  = isLeftAvail ?getCU(ctu-1):NULL;
336    TComDataCU* ctuRight = isRightAvail?getCU(ctu+1):NULL;
337    TComDataCU* ctuAbove = isAboveAvail?getCU(ctu-m_uiWidthInCU):NULL;
338    TComDataCU* ctuBelow = isBelowAvail?getCU(ctu+m_uiWidthInCU):NULL;
339    TComDataCU* ctuAboveLeft  = isAboveLeftAvail ? getCU(ctu-m_uiWidthInCU-1):NULL;
340    TComDataCU* ctuAboveRigtht= isAboveRightAvail? getCU(ctu-m_uiWidthInCU+1):NULL;
341    TComDataCU* ctuBelowLeft  = isBelowLeftAvail ? getCU(ctu+m_uiWidthInCU-1):NULL;
342    TComDataCU* ctuBelowRight = isBelowRightAvail? getCU(ctu+m_uiWidthInCU+1):NULL;
343
344    {
345      //left
346      if(ctuLeft != NULL)
347      {
348        isLeftAvail = (ctuCurr->getSlice()->getSliceCurStartCUAddr() != ctuLeft->getSlice()->getSliceCurStartCUAddr())?ctuCurr->getSlice()->getLFCrossSliceBoundaryFlag():true;
349      }
350      //above
351      if(ctuAbove != NULL)
352      {
353        isAboveAvail = (ctuCurr->getSlice()->getSliceCurStartCUAddr() != ctuAbove->getSlice()->getSliceCurStartCUAddr())?ctuCurr->getSlice()->getLFCrossSliceBoundaryFlag():true;
354      }
355      //right
356      if(ctuRight != NULL)
357      {
358        isRightAvail = (ctuCurr->getSlice()->getSliceCurStartCUAddr() != ctuRight->getSlice()->getSliceCurStartCUAddr())?ctuRight->getSlice()->getLFCrossSliceBoundaryFlag():true;
359      }
360      //below
361      if(ctuBelow != NULL)
362      {
363        isBelowAvail = (ctuCurr->getSlice()->getSliceCurStartCUAddr() != ctuBelow->getSlice()->getSliceCurStartCUAddr())?ctuBelow->getSlice()->getLFCrossSliceBoundaryFlag():true;
364      }
365      //above-left
366      if(ctuAboveLeft != NULL)
367      {
368        isAboveLeftAvail = (ctuCurr->getSlice()->getSliceCurStartCUAddr() != ctuAboveLeft->getSlice()->getSliceCurStartCUAddr())?ctuCurr->getSlice()->getLFCrossSliceBoundaryFlag():true;
369      }
370      //below-right
371      if(ctuBelowRight != NULL)
372{
373        isBelowRightAvail = (ctuCurr->getSlice()->getSliceCurStartCUAddr() != ctuBelowRight->getSlice()->getSliceCurStartCUAddr())?ctuBelowRight->getSlice()->getLFCrossSliceBoundaryFlag():true;
374      }
375
376
377      //above-right
378      if(ctuAboveRigtht != NULL)
379      {
380        Int curSliceStartEncOrder  = ctuCurr->getSlice()->getSliceCurStartCUAddr();
381        Int aboveRigthtSliceStartEncOrder = ctuAboveRigtht->getSlice()->getSliceCurStartCUAddr();
382
383        isAboveRightAvail = (curSliceStartEncOrder == aboveRigthtSliceStartEncOrder)?(true):
384          (
385          (curSliceStartEncOrder > aboveRigthtSliceStartEncOrder)?(ctuCurr->getSlice()->getLFCrossSliceBoundaryFlag())
386          :(ctuAboveRigtht->getSlice()->getLFCrossSliceBoundaryFlag())
387          );         
388      }
389      //below-left
390      if(ctuBelowLeft != NULL)
391      {
392        Int curSliceStartEncOrder  = ctuCurr->getSlice()->getSliceCurStartCUAddr();
393        Int belowLeftSliceStartEncOrder = ctuBelowLeft->getSlice()->getSliceCurStartCUAddr();
394
395        isBelowLeftAvail = (curSliceStartEncOrder == belowLeftSliceStartEncOrder)?(true):
396          (
397          (curSliceStartEncOrder > belowLeftSliceStartEncOrder)?(ctuCurr->getSlice()->getLFCrossSliceBoundaryFlag())
398          :(ctuBelowLeft->getSlice()->getLFCrossSliceBoundaryFlag())
399          );
400      }       
401    }
402
403    if(!isLoopFiltAcrossTilePPS)
404    {     
405      isLeftAvail      = (!isLeftAvail      ) ?false:(getTileIdxMap( ctuLeft->getAddr()         ) == getTileIdxMap( ctu ));
406      isAboveAvail     = (!isAboveAvail     ) ?false:(getTileIdxMap( ctuAbove->getAddr()        ) == getTileIdxMap( ctu ));
407      isRightAvail     = (!isRightAvail     ) ?false:(getTileIdxMap( ctuRight->getAddr()        ) == getTileIdxMap( ctu ));
408      isBelowAvail     = (!isBelowAvail     ) ?false:(getTileIdxMap( ctuBelow->getAddr()        ) == getTileIdxMap( ctu ));
409      isAboveLeftAvail = (!isAboveLeftAvail ) ?false:(getTileIdxMap( ctuAboveLeft->getAddr()    ) == getTileIdxMap( ctu ));
410      isAboveRightAvail= (!isAboveRightAvail) ?false:(getTileIdxMap( ctuAboveRigtht->getAddr()  ) == getTileIdxMap( ctu ));
411      isBelowLeftAvail = (!isBelowLeftAvail ) ?false:(getTileIdxMap( ctuBelowLeft->getAddr()    ) == getTileIdxMap( ctu ));
412      isBelowRightAvail= (!isBelowRightAvail) ?false:(getTileIdxMap( ctuBelowRight->getAddr()   ) == getTileIdxMap( ctu ));
413    }
414  }
415
416}
417
418TComTile::TComTile()
419{
420}
421
422TComTile::~TComTile()
423{
424}
425//! \}
Note: See TracBrowser for help on using the repository browser.