source: 3DVCSoftware/branches/HTM-9.0r1-F0122/source/Lib/TLibCommon/TComPicSym.cpp @ 744

Last change on this file since 744 was 655, checked in by tech, 11 years ago

Merged 8.1-Cleanup@654

  • Property svn:eol-style set to native
File size: 9.5 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-2013, 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{};
70
71
72Void TComPicSym::create  ( Int iPicWidth, Int iPicHeight, UInt uiMaxWidth, UInt uiMaxHeight, UInt uiMaxDepth )
73{
74  UInt i;
75
76  m_uhTotalDepth      = uiMaxDepth;
77  m_uiNumPartitions   = 1<<(m_uhTotalDepth<<1);
78 
79  m_uiMaxCUWidth      = uiMaxWidth;
80  m_uiMaxCUHeight     = uiMaxHeight;
81 
82  m_uiMinCUWidth      = uiMaxWidth  >> m_uhTotalDepth;
83  m_uiMinCUHeight     = uiMaxHeight >> m_uhTotalDepth;
84 
85  m_uiNumPartInWidth  = m_uiMaxCUWidth  / m_uiMinCUWidth;
86  m_uiNumPartInHeight = m_uiMaxCUHeight / m_uiMinCUHeight;
87 
88  m_uiWidthInCU       = ( iPicWidth %m_uiMaxCUWidth  ) ? iPicWidth /m_uiMaxCUWidth  + 1 : iPicWidth /m_uiMaxCUWidth;
89  m_uiHeightInCU      = ( iPicHeight%m_uiMaxCUHeight ) ? iPicHeight/m_uiMaxCUHeight + 1 : iPicHeight/m_uiMaxCUHeight;
90 
91  m_uiNumCUsInFrame   = m_uiWidthInCU * m_uiHeightInCU;
92  m_apcTComDataCU     = new TComDataCU*[m_uiNumCUsInFrame];
93 
94  if (m_uiNumAllocatedSlice>0)
95  {
96    for ( i=0; i<m_uiNumAllocatedSlice ; i++ )
97    {
98      delete m_apcTComSlice[i];
99    }
100    delete [] m_apcTComSlice;
101  }
102  m_apcTComSlice      = new TComSlice*[m_uiNumCUsInFrame*m_uiNumPartitions]; 
103  m_apcTComSlice[0]   = new TComSlice;
104  m_uiNumAllocatedSlice = 1;
105  for ( i=0; i<m_uiNumCUsInFrame ; i++ )
106  {
107    m_apcTComDataCU[i] = new TComDataCU;
108    m_apcTComDataCU[i]->create( m_uiNumPartitions, m_uiMaxCUWidth, m_uiMaxCUHeight, false, m_uiMaxCUWidth >> m_uhTotalDepth
109#if ADAPTIVE_QP_SELECTION
110      , true
111#endif     
112      );
113  }
114
115  m_puiCUOrderMap = new UInt[m_uiNumCUsInFrame+1];
116  m_puiTileIdxMap = new UInt[m_uiNumCUsInFrame];
117  m_puiInverseCUOrderMap = new UInt[m_uiNumCUsInFrame+1];
118
119  for( i=0; i<m_uiNumCUsInFrame; i++ )
120  {
121    m_puiCUOrderMap[i] = i;
122    m_puiInverseCUOrderMap[i] = i;
123  }
124  m_saoParam = NULL;
125}
126
127Void TComPicSym::destroy()
128{
129  if (m_uiNumAllocatedSlice>0)
130  {
131    for (Int i = 0; i<m_uiNumAllocatedSlice ; i++ )
132    {
133      delete m_apcTComSlice[i];
134    }
135    delete [] m_apcTComSlice;
136  }
137  m_apcTComSlice = NULL;
138 
139  for (Int i = 0; i < m_uiNumCUsInFrame; i++)
140  {
141    m_apcTComDataCU[i]->destroy();
142    delete m_apcTComDataCU[i];
143    m_apcTComDataCU[i] = NULL;
144  }
145  delete [] m_apcTComDataCU;
146  m_apcTComDataCU = NULL;
147
148  for(Int i = 0; i < (m_iNumColumnsMinus1+1)*(m_iNumRowsMinus1+1); i++ )
149  {
150    delete m_apcTComTile[i];
151  }
152  delete [] m_apcTComTile;
153
154  m_apcTComTile = NULL;
155
156  delete [] m_puiCUOrderMap;
157  m_puiCUOrderMap = NULL;
158
159  delete [] m_puiTileIdxMap;
160  m_puiTileIdxMap = NULL;
161
162  delete [] m_puiInverseCUOrderMap;
163  m_puiInverseCUOrderMap = NULL;
164 
165  if (m_saoParam)
166  {
167    TComSampleAdaptiveOffset::freeSaoParam(m_saoParam);
168    delete m_saoParam;
169    m_saoParam = 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::allocSaoParam(TComSampleAdaptiveOffset *sao)
311{
312  m_saoParam = new SAOParam;
313  sao->allocSaoParam(m_saoParam);
314}
315
316TComTile::TComTile()
317{
318}
319
320TComTile::~TComTile()
321{
322}
323//! \}
Note: See TracBrowser for help on using the repository browser.