Coverage Report

Created: 2026-07-16 06:32

next uncovered line (L), next uncovered region (R), next uncovered branch (B)
/work/vvenc/source/Lib/CommonLib/ContextModelling.h
Line
Count
Source
1
/* -----------------------------------------------------------------------------
2
The copyright in this software is being made available under the Clear BSD
3
License, included below. No patent rights, trademark rights and/or 
4
other Intellectual Property Rights other than the copyrights concerning 
5
the Software are granted under this license.
6
7
The Clear BSD License
8
9
Copyright (c) 2019-2026, Fraunhofer-Gesellschaft zur Förderung der angewandten Forschung e.V. & The VVenC Authors.
10
All rights reserved.
11
12
Redistribution and use in source and binary forms, with or without modification,
13
are permitted (subject to the limitations in the disclaimer below) provided that
14
the following conditions are met:
15
16
     * Redistributions of source code must retain the above copyright notice,
17
     this list of conditions and the following disclaimer.
18
19
     * Redistributions in binary form must reproduce the above copyright
20
     notice, this list of conditions and the following disclaimer in the
21
     documentation and/or other materials provided with the distribution.
22
23
     * Neither the name of the copyright holder nor the names of its
24
     contributors may be used to endorse or promote products derived from this
25
     software without specific prior written permission.
26
27
NO EXPRESS OR IMPLIED LICENSES TO ANY PARTY'S PATENT RIGHTS ARE GRANTED BY
28
THIS LICENSE. THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND
29
CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
30
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A
31
PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR
32
CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
33
EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
34
PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR
35
BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER
36
IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
37
ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
38
POSSIBILITY OF SUCH DAMAGE.
39
40
41
------------------------------------------------------------------------------------------- */
42
/** \file     ContextModelling.h
43
 *  \brief    Classes providing probability descriptions and contexts (header)
44
 */
45
46
#pragma once
47
48
#include "CommonDef.h"
49
#include "Contexts.h"
50
#include "Slice.h"
51
#include "Unit.h"
52
#include "UnitPartitioner.h"
53
54
#include <bitset>
55
56
//! \ingroup CommonLib
57
//! \{
58
59
namespace vvenc
60
{
61
62
struct CtxTpl
63
{
64
  // lower 5 bits are absSum1, upper 3 bits are numPos
65
  uint8_t ctxTpl;
66
};
67
68
struct CoeffCodingContext
69
{
70
public:
71
  CoeffCodingContext( const TransformUnit& tu, ComponentID component, bool signHide, bool bdpcm = false, CtxTpl* tplBuf = nullptr );
72
public:
73
  void  initSubblock    ( int SubsetId, bool sigGroupFlag = false );
74
public:
75
11.6k
  void  resetSigGroup   ()                      { m_sigCoeffGroupFlag.reset( m_subSetPos ); }
76
466k
  void  setSigGroup     ()                      { m_sigCoeffGroupFlag.set( m_subSetPos ); }
77
0
  bool  noneSigGroup    ()                      { return m_sigCoeffGroupFlag.none(); }
78
450k
  int   lastSubSet      ()                      { return ( maxNumCoeff() - 1 ) >> log2CGSize(); }
79
382k
  bool  isLastSubSet    ()                      { return lastSubSet() == m_subSetId; }
80
67.8k
  bool  only1stSigGroup ()                      { return m_sigCoeffGroupFlag.count()-m_sigCoeffGroupFlag[lastSubSet()]==0; }
81
898k
  void  setScanPosLast  ( int       posLast )   { m_scanPosLast = posLast; }
82
public:
83
0
  ComponentID     compID          ()                        const { return m_compID; }
84
633k
  int             subSetId        ()                        const { return m_subSetId; }
85
0
  int             subSetPos       ()                        const { return m_subSetPos; }
86
46.1k
  int             cgPosY          ()                        const { return m_subSetPosY; }
87
46.1k
  int             cgPosX          ()                        const { return m_subSetPosX; }
88
1.79M
  unsigned        width           ()                        const { return m_width; }
89
0
  unsigned        height          ()                        const { return m_height; }
90
6.04k
  unsigned        log2CGWidth     ()                        const { return m_log2CGWidth; }
91
0
  unsigned        log2CGHeight    ()                        const { return m_log2CGHeight; }
92
6.45M
  unsigned        log2CGSize      ()                        const { return m_log2CGSize; }
93
5.43M
  int             maxLog2TrDRange ()                        const { return m_maxLog2TrDynamicRange; }
94
7.24M
  unsigned        maxNumCoeff     ()                        const { return m_maxNumCoeff; }
95
12.8M
  int             scanPosLast     ()                        const { return m_scanPosLast; }
96
11.5M
  int             minSubPos       ()                        const { return m_minSubPos; }
97
409k
  int             maxSubPos       ()                        const { return m_maxSubPos; }
98
926k
  bool            isLast          ()                        const { return ( ( m_scanPosLast >> m_log2CGSize ) == m_subSetId ); }
99
46.5k
  bool            isNotFirst      ()                        const { return ( m_subSetId != 0 ); }
100
0
  bool            isSigGroup      ( int scanPosCG )         const { return m_sigCoeffGroupFlag[m_scanCG[scanPosCG].idx]; }
101
1.07M
  bool            isSigGroup      ()                        const { return m_sigCoeffGroupFlag[ m_subSetPos ]; }
102
9.67k
  bool            signHiding      ()                        const { return m_signHiding; }
103
  bool            hideSign        ( int       posFirst,
104
917k
                                    int       posLast   )   const { return ( m_signHiding && ( posLast - posFirst >= SBH_THRESHOLD ) ); }
105
57.3M
  unsigned        blockPos        ( int scanPos )           const { return m_scan[scanPos].idx; }
106
9.72M
  unsigned        posX            ( int scanPos )           const { return m_scan[scanPos].x; }
107
9.72M
  unsigned        posY            ( int scanPos )           const { return m_scan[scanPos].y; }
108
898k
  unsigned        maxLastPosX     ()                        const { return m_maxLastPosX; }
109
898k
  unsigned        maxLastPosY     ()                        const { return m_maxLastPosY; }
110
2.05M
  unsigned        lastXCtxId      ( unsigned  posLastX  )   const { return m_CtxSetLastX( m_lastOffsetX + ( posLastX >> m_lastShiftX ) ); }
111
2.11M
  unsigned        lastYCtxId      ( unsigned  posLastY  )   const { return m_CtxSetLastY( m_lastOffsetY + ( posLastY >> m_lastShiftY ) ); }
112
1.01M
  unsigned        sigGroupCtxId   ( bool ts = false     )   const { return ts ? m_sigGroupCtxIdTS : m_sigGroupCtxId; }
113
6.43M
  bool            bdpcm           ()                        const { return m_bdpcm; }
114
115
  unsigned sigCtxIdAbs( int scanPos, const TCoeffSig* coeff, const int state )
116
0
  {
117
0
    const uint32_t   posY      = m_scan[scanPos].y;
118
0
    const uint32_t   posX      = m_scan[scanPos].x;
119
0
    const TCoeffSig* pData     = coeff + posX + posY * m_width;
120
0
    const int        diag      = posX + posY;
121
0
    int              numPos    = 0;
122
0
    int              sumAbs    = 0;
123
0
#define UPDATE(x) {int a=abs(x);sumAbs+=std::min(4+(a&1),a);numPos+=!!a;}
124
0
    if( posX < m_width-1 )
125
0
    {
126
0
      UPDATE( pData[1] );
127
0
      if( posX < m_width-2 )
128
0
      {
129
0
        UPDATE( pData[2] );
130
0
      }
131
0
      if( posY < m_height-1 )
132
0
      {
133
0
        UPDATE( pData[m_width+1] );
134
0
      }
135
0
    }
136
0
    if( posY < m_height-1 )
137
0
    {
138
0
      UPDATE( pData[m_width] );
139
0
      if( posY < m_height-2 )
140
0
      {
141
0
        UPDATE( pData[m_width<<1] );
142
0
      }
143
0
    }
144
0
#undef UPDATE
145
146
0
    int ctxOfs = std::min((sumAbs+1)>>1, 3) + ( diag < 2 ? 4 : 0 );
147
148
0
    if( m_chType == CH_L )
149
0
    {
150
0
      ctxOfs += diag < 5 ? 4 : 0;
151
0
    }
152
153
0
    m_tmplCpDiag = diag;
154
0
    m_tmplCpSum1 = sumAbs - numPos;
155
0
    return m_sigFlagCtxSet[std::max( 0, state-1 )]( ctxOfs );
156
0
  }
157
158
  unsigned sigCtxIdAbsWithAcc( const int scanPos, const int state )
159
6.39M
  {
160
6.39M
    const auto scanEl       = m_scan[scanPos];
161
6.39M
    const uint32_t posY     = scanEl.y;
162
6.39M
    const uint32_t posX     = scanEl.x;
163
6.39M
    const int32_t blkPos    = scanEl.idx;
164
6.39M
    const int      diag     = posX + posY;
165
6.39M
    const int      tplVal   = m_tplBuf[-blkPos].ctxTpl;
166
6.39M
    const int      numPos   = tplVal >> 5u;
167
6.39M
    const int      sumAbs   = tplVal & 31;
168
169
6.39M
    int ctxOfs = std::min( ( sumAbs + 1 ) >> 1, 3 ) + ( diag < 2 ? 4 : 0 );
170
171
6.39M
    if( isLuma( m_chType ) )
172
448k
    {
173
448k
      ctxOfs += diag < 5 ? 4 : 0;
174
448k
    }
175
6.39M
    m_tmplCpDiag = diag;
176
6.39M
    m_tmplCpSum1 = sumAbs - numPos;
177
6.39M
    return m_sigFlagCtxSet[std::max( 0, state-1 )]( ctxOfs );
178
6.39M
  }
179
180
  void absVal1stPass( const int scanPos, const TCoeffSig absLevel1 )
181
6.73M
  {
182
6.73M
    CHECKD( !absLevel1, "Shound not be called if '0'!" );
183
184
6.73M
    const auto scanEl     = m_scan[scanPos];
185
6.73M
    const uint32_t posY   = scanEl.y;
186
6.73M
    const uint32_t posX   = scanEl.x;
187
6.73M
    const int32_t  blkPos = scanEl.idx;
188
189
6.73M
    auto update_deps = [&]( int offset )
190
17.6M
    {
191
17.6M
      auto& ctx   = m_tplBuf[-blkPos + offset];
192
17.6M
      ctx.ctxTpl += uint8_t( 32 + absLevel1 );
193
17.6M
    };
194
195
6.73M
    if( posY > 1 ) update_deps( 2 * m_width );
196
6.73M
    if( posY > 0
197
4.53M
     && posX > 0 ) update_deps( m_width + 1 );
198
6.73M
    if( posY > 0 ) update_deps( m_width );
199
6.73M
    if( posX > 1 ) update_deps( 2 );
200
6.73M
    if( posX > 0 ) update_deps( 1 );
201
6.73M
  }
202
  
203
204
  void remAbsVal1stPass( const int scanPos, const TCoeffSig absLevel1 )
205
0
  {
206
0
    CHECKD( !absLevel1, "Shound not be called if '0'!" );
207
208
0
    const auto scanEl     = m_scan[scanPos];
209
0
    const uint32_t posY   = scanEl.y;
210
0
    const uint32_t posX   = scanEl.x;
211
0
    const int32_t  blkPos = scanEl.idx;
212
213
0
    auto update_deps = [&]( int offset )
214
0
    {
215
0
      auto& ctx   = m_tplBuf[-blkPos + offset];
216
0
      ctx.ctxTpl -= uint8_t( 32 + absLevel1 );
217
0
    };
218
219
0
    if( posY > 1 ) update_deps( 2 * m_width );
220
0
    if( posY > 0
221
0
     && posX > 0 ) update_deps( m_width + 1 );
222
0
    if( posY > 0 ) update_deps( m_width );
223
0
    if( posX > 1 ) update_deps( 2 );
224
0
    if( posX > 0 ) update_deps( 1 );
225
0
  }
226
227
  uint8_t ctxOffsetAbs()
228
6.73M
  {
229
6.73M
    int offset = 0;
230
6.73M
    if( m_tmplCpDiag != -1 )
231
5.83M
    {
232
5.83M
      offset  = std::min( m_tmplCpSum1, 4 ) + 1;
233
5.83M
      offset += ( !m_tmplCpDiag ? ( m_chType == CH_L ? 15 : 5 ) : m_chType == CH_L ? m_tmplCpDiag < 3 ? 10 : ( m_tmplCpDiag < 10 ? 5 : 0 ) : 0 );
234
5.83M
    }
235
6.73M
    return uint8_t(offset);
236
6.73M
  }
237
238
5.88M
  unsigned parityCtxIdAbs   ( uint8_t offset )  const { return m_parFlagCtxSet   ( offset ); }
239
6.73M
  unsigned greater1CtxIdAbs ( uint8_t offset )  const { return m_gtxFlagCtxSet[1]( offset ); }
240
5.88M
  unsigned greater2CtxIdAbs ( uint8_t offset )  const { return m_gtxFlagCtxSet[0]( offset ); }
241
242
  unsigned templateAbsSum( int scanPos, const TCoeffSig* coeff, int baseLevel )
243
4.87M
  {
244
4.87M
    const uint32_t   posY = m_scan[scanPos].y;
245
4.87M
    const uint32_t   posX = m_scan[scanPos].x;
246
4.87M
    const TCoeffSig* pData = coeff + posX + posY * m_width;
247
4.87M
    int              sum   = 0;
248
4.87M
    if (posX < m_width - 1)
249
4.77M
    {
250
4.77M
      sum += abs(pData[1]);
251
4.77M
      if (posX < m_width - 2)
252
4.58M
      {
253
4.58M
        sum += abs(pData[2]);
254
4.58M
      }
255
4.77M
      if (posY < m_height - 1)
256
4.62M
      {
257
4.62M
        sum += abs(pData[m_width + 1]);
258
4.62M
      }
259
4.77M
    }
260
4.87M
    if (posY < m_height - 1)
261
4.72M
    {
262
4.72M
      sum += abs(pData[m_width]);
263
4.72M
      if (posY < m_height - 2)
264
4.48M
      {
265
4.48M
        sum += abs(pData[m_width << 1]);
266
4.48M
      }
267
4.72M
    }
268
4.87M
    return std::max(std::min(sum - 5 * baseLevel, 31), 0);
269
4.87M
  }
270
271
  unsigned sigCtxIdAbsTS( int scanPos, const TCoeffSig* coeff )
272
12.3M
  {
273
12.3M
    const uint32_t   posY   = m_scan[scanPos].y;
274
12.3M
    const uint32_t   posX   = m_scan[scanPos].x;
275
12.3M
    const TCoeffSig* posC   = coeff + posX + posY * m_width;
276
12.3M
    int              numPos = 0;
277
21.8M
#define UPDATE(x) {int a=abs(x);numPos+=!!a;}
278
12.3M
    if( posX > 0 )
279
11.0M
    {
280
11.0M
      UPDATE( posC[-1] );
281
11.0M
    }
282
12.3M
    if( posY > 0 )
283
10.7M
    {
284
10.7M
      UPDATE( posC[-(int)m_width] );
285
10.7M
    }
286
12.3M
#undef UPDATE
287
288
12.3M
    return m_tsSigFlagCtxSet( numPos );
289
12.3M
  }
290
291
10.6M
  unsigned parityCtxIdAbsTS   ()                  const { return m_tsParFlagCtxSet(      0 ); }
292
3.74M
  unsigned greaterXCtxIdAbsTS ( uint8_t offset )  const { return m_tsGtxFlagCtxSet( offset ); }
293
294
  unsigned lrg1CtxIdAbsTS(int scanPos, const TCoeffSig* coeff, int bdpcm)
295
10.6M
  {
296
10.6M
    const uint32_t   posY = m_scan[scanPos].y;
297
10.6M
    const uint32_t   posX = m_scan[scanPos].x;
298
10.6M
    const TCoeffSig* posC = coeff + posX + posY * m_width;
299
300
10.6M
    int             numPos = 0;
301
10.6M
#define UPDATE(x) {int a=abs(x);numPos+=!!a;}
302
303
10.6M
    if (bdpcm)
304
10.1M
    {
305
10.1M
      numPos = 3;
306
10.1M
    }
307
497k
    else
308
497k
    {
309
497k
      if (posX > 0)
310
459k
      {
311
459k
        UPDATE(posC[-1]);
312
459k
      }
313
497k
      if (posY > 0)
314
458k
      {
315
458k
        UPDATE(posC[-(int)m_width]);
316
458k
      }
317
497k
    }
318
319
10.6M
#undef UPDATE
320
10.6M
    return m_tsLrg1FlagCtxSet(numPos);
321
10.6M
  }
322
323
  unsigned signCtxIdAbsTS(int scanPos, const TCoeffSig* coeff, int bdpcm)
324
10.6M
  {
325
10.6M
    const uint32_t   posY  = m_scan[scanPos].y;
326
10.6M
    const uint32_t   posX  = m_scan[scanPos].x;
327
10.6M
    const TCoeffSig* pData = coeff + posX + posY * m_width;
328
329
10.6M
    int rightSign = 0, belowSign = 0;
330
10.6M
    unsigned signCtx = 0;
331
332
10.6M
    if (posX > 0)
333
9.60M
    {
334
9.60M
      rightSign = pData[-1];
335
9.60M
    }
336
10.6M
    if (posY > 0)
337
9.07M
    {
338
9.07M
      belowSign = pData[-(int)m_width];
339
9.07M
    }
340
341
10.6M
    if ((rightSign == 0 && belowSign == 0) || ((rightSign*belowSign) < 0))
342
9.25M
    {
343
9.25M
      signCtx = 0;
344
9.25M
    }
345
1.39M
    else if (rightSign >= 0 && belowSign >= 0)
346
662
    {
347
662
      signCtx = 1;
348
662
    }
349
1.39M
    else
350
1.39M
    {
351
1.39M
      signCtx = 2;
352
1.39M
    }
353
10.6M
    if (bdpcm)
354
10.1M
    {
355
10.1M
      signCtx += 3;
356
10.1M
    }
357
10.6M
    return m_tsSignFlagCtxSet(signCtx);
358
10.6M
  }
359
360
  void neighTS(int& rightPixel, int& belowPixel, int scanPos, const TCoeffSig* coeff)
361
15.2M
  {
362
15.2M
    const uint32_t   posY = m_scan[scanPos].y;
363
15.2M
    const uint32_t   posX = m_scan[scanPos].x;
364
15.2M
    const TCoeffSig* data = coeff + posX + posY * m_width;
365
366
15.2M
    rightPixel = belowPixel = 0;
367
368
15.2M
    if (posX > 0)
369
13.6M
    {
370
13.6M
      rightPixel = data[-1];
371
13.6M
    }
372
15.2M
    if (posY > 0)
373
12.4M
    {
374
12.4M
      belowPixel = data[-(int)m_width];
375
12.4M
    }
376
15.2M
  }
377
378
  int deriveModCoeff(int rightPixel, int belowPixel, int absCoeff, int bdpcm = 0)
379
7.94M
  {
380
7.94M
    if (absCoeff == 0)
381
3.87M
      return 0;
382
383
4.06M
    int pred1, absBelow = abs(belowPixel), absRight = abs(rightPixel);
384
385
4.06M
    int absCoeffMod = absCoeff;
386
387
4.06M
    if (bdpcm == 0)
388
528k
    {
389
528k
      pred1 = std::max(absBelow, absRight);
390
391
528k
      if (absCoeff == pred1)
392
0
      {
393
0
        absCoeffMod = 1;
394
0
      }
395
528k
      else
396
528k
      {
397
528k
        absCoeffMod = absCoeff < pred1 ? absCoeff + 1 : absCoeff;
398
528k
      }
399
528k
    }
400
401
4.06M
    return(absCoeffMod);
402
7.94M
  }
403
404
  int decDeriveModCoeff(int rightPixel, int belowPixel, int absCoeff)
405
0
  {
406
0
    if (absCoeff == 0)
407
0
      return 0;
408
0
409
0
    int pred1, absBelow = abs(belowPixel), absRight = abs(rightPixel);
410
0
    pred1 = std::max(absBelow, absRight);
411
0
412
0
    int absCoeffMod;
413
0
414
0
    if (absCoeff == 1 && pred1 > 0)
415
0
    {
416
0
      absCoeffMod = pred1;
417
0
    }
418
0
    else
419
0
    {
420
0
      absCoeffMod = absCoeff - (absCoeff <= pred1);
421
0
    }
422
0
    return(absCoeffMod);
423
0
  }
424
425
  //unsigned templateAbsSumTS( int scanPos, const TCoeffSig* coeff )
426
  //{
427
  //  return 1;
428
  //}
429
430
  int                       remRegBins;
431
432
private:
433
  // constant
434
  const ComponentID         m_compID;
435
  const ChannelType         m_chType;
436
  const unsigned            m_width;
437
  const unsigned            m_height;
438
  const unsigned            m_log2CGWidth;
439
  const unsigned            m_log2CGHeight;
440
  const unsigned            m_log2CGSize;
441
  const unsigned            m_widthInGroups;
442
  const unsigned            m_heightInGroups;
443
  const unsigned            m_log2WidthInGroups;
444
  const unsigned            m_log2BlockWidth;
445
  const unsigned            m_log2BlockHeight;
446
  const unsigned            m_maxNumCoeff;
447
  const bool                m_signHiding;
448
  const int                 m_maxLog2TrDynamicRange;
449
  const ScanElement *       m_scan;
450
  const ScanElement *       m_scanCG;
451
  const CtxSet              m_CtxSetLastX;
452
  const CtxSet              m_CtxSetLastY;
453
  const unsigned            m_maxLastPosX;
454
  const unsigned            m_maxLastPosY;
455
  const int                 m_lastOffsetX;
456
  const int                 m_lastOffsetY;
457
  const int                 m_lastShiftX;
458
  const int                 m_lastShiftY;
459
  // modified
460
  int                       m_scanPosLast;
461
  int                       m_subSetId;
462
  int                       m_subSetPos;
463
  int                       m_subSetPosX;
464
  int                       m_subSetPosY;
465
  int                       m_minSubPos;
466
  int                       m_maxSubPos;
467
  unsigned                  m_sigGroupCtxId;
468
  int                       m_tmplCpSum1;
469
  int                       m_tmplCpDiag;
470
  CtxSet                    m_sigFlagCtxSet[3];
471
  CtxSet                    m_parFlagCtxSet;
472
  CtxSet                    m_gtxFlagCtxSet[2];
473
  unsigned                  m_sigGroupCtxIdTS;
474
  CtxSet                    m_tsSigFlagCtxSet;
475
  CtxSet                    m_tsParFlagCtxSet;
476
  CtxSet                    m_tsGtxFlagCtxSet;
477
  CtxSet                    m_tsLrg1FlagCtxSet;
478
  CtxSet                    m_tsSignFlagCtxSet;
479
  std::bitset<MLS_GRP_NUM>  m_sigCoeffGroupFlag;
480
  const bool                m_bdpcm;
481
  CtxTpl*                   m_tplBuf;
482
};
483
484
485
struct CUCtx
486
{
487
441k
  CUCtx()              : isDQPCoded(false), isChromaQpAdjCoded(false),
488
441k
                         qgStart(false)
489
441k
                         {
490
441k
                           violatesLfnstConstrained[CH_L] = false;
491
441k
                           violatesLfnstConstrained[CH_C] = false;
492
441k
                           violatesMtsCoeffConstraint      = false;
493
441k
                           mtsLastScanPos                  = false;
494
441k
                           lfnstLastScanPos                = false;
495
441k
                         }
496
15.1k
  CUCtx(int _qp)       : isDQPCoded(false), isChromaQpAdjCoded(false),
497
15.1k
                         qgStart(false),
498
15.1k
                         qp(_qp)
499
15.1k
                         {
500
15.1k
                           violatesLfnstConstrained[CH_L] = false;
501
15.1k
                           violatesLfnstConstrained[CH_C] = false;
502
15.1k
                           violatesMtsCoeffConstraint      = false;
503
15.1k
                           mtsLastScanPos                  = false;
504
15.1k
                           lfnstLastScanPos                = false;
505
15.1k
                         }
506
507
  bool      isDQPCoded;
508
  bool      isChromaQpAdjCoded;
509
  bool      qgStart;
510
  bool      lfnstLastScanPos;
511
  int8_t    qp;                   // used as a previous(last) QP and for QP prediction
512
  bool      violatesLfnstConstrained[MAX_NUM_CH];
513
  bool      violatesMtsCoeffConstraint;
514
  bool      mtsLastScanPos;
515
};
516
517
struct MergeCtx
518
{
519
806k
  MergeCtx() : numValidMergeCand( 0 ), hasMergedCandList( false ) { for( unsigned i = 0; i < MRG_MAX_NUM_CANDS; i++ ) mrgTypeNeighbours[i] = MRG_TYPE_DEFAULT_N; }
520
521
  void          setMmvdMergeCandiInfo ( CodingUnit& cu, const MmvdIdx candIdx) const;
522
  void          setMergeInfo          ( CodingUnit& cu, const int     candIdx) const;
523
  void          getMmvdDeltaMv        ( const Slice& slice, const MmvdIdx candIdx, Mv deltaMv[NUM_REF_PIC_LIST_01] ) const;
524
525
  MvField       mvFieldNeighbours [MRG_MAX_NUM_CANDS][NUM_REF_PIC_LIST_01];
526
  uint8_t       BcwIdx            [MRG_MAX_NUM_CANDS];
527
  unsigned char interDirNeighbours[MRG_MAX_NUM_CANDS];
528
  MergeType     mrgTypeNeighbours [MRG_MAX_NUM_CANDS];
529
  int           numValidMergeCand;
530
  bool          hasMergedCandList;
531
532
  MvField       mmvdBaseMv        [MMVD_BASE_MV_NUM][NUM_REF_PIC_LIST_01];
533
  bool          mmvdUseAltHpelIf  [MMVD_BASE_MV_NUM];
534
  bool          useAltHpelIf      [MRG_MAX_NUM_CANDS];
535
};
536
537
struct AffineMergeCtx
538
{
539
0
  AffineMergeCtx() : numValidMergeCand( 0 ) { for ( unsigned i = 0; i < AFFINE_MRG_MAX_NUM_CANDS; i++ ) affineType[i] = AFFINEMODEL_4PARAM; }
540
541
  MvField       mvFieldNeighbours [AFFINE_MRG_MAX_NUM_CANDS][NUM_REF_PIC_LIST_01][3];
542
  unsigned char interDirNeighbours[AFFINE_MRG_MAX_NUM_CANDS];
543
  EAffineModel  affineType        [AFFINE_MRG_MAX_NUM_CANDS];
544
  uint8_t       BcwIdx            [AFFINE_MRG_MAX_NUM_CANDS];
545
  int           numValidMergeCand;
546
  int           maxNumMergeCand;
547
548
  MergeType     mergeType[AFFINE_MRG_MAX_NUM_CANDS];
549
  MotionBuf     subPuMvpMiBuf;
550
};
551
552
553
struct DeriveCtx
554
{
555
  const CodingUnit* cuRestrictedLeft[MAX_NUM_CH];
556
  const CodingUnit* cuRestrictedAbove[MAX_NUM_CH];
557
558
  void determineNeighborCus       ( const CodingStructure& cs, const UnitArea& ua, const ChannelType ch, const TreeType treeType );
559
560
3.72M
  inline static unsigned CtxQtCbf ( const ComponentID compID, const bool prevCbf = false, const int ispIdx = 0 )   { return ( ispIdx && isLuma( compID ) ) ? (2 + (int)prevCbf) : (((compID == COMP_Cr) && prevCbf) ? 1 : 0); }
561
  void     CtxSplit               ( const Partitioner& partitioner, unsigned& ctxSpl, unsigned& ctxQt, unsigned& ctxHv, unsigned& ctxHorBt, unsigned& ctxVerBt, const bool canSplit[6] ) const;
562
  unsigned CtxMipFlag             ( const CodingUnit& cu ) const;
563
0
  unsigned CtxInterDir            ( const CodingUnit& cu ) const { return ( 7 - ((Log2(cu.lumaSize().area()) + 1) >> 1) ); }
564
565
  unsigned CtxModeConsFlag() const
566
0
  {
567
0
    const CodingUnit* cuLeft  = cuRestrictedLeft[CH_L]; const CodingUnit* cuAbove = cuRestrictedAbove[CH_L];
568
0
    return ((cuAbove && cuAbove->predMode == MODE_INTRA) || (cuLeft && cuLeft->predMode == MODE_INTRA)) ? 1 : 0;
569
0
  }
570
571
  unsigned CtxAffineFlag() const
572
0
  {
573
0
    const CodingUnit *cuLeft = cuRestrictedLeft[CH_L]; const CodingUnit *cuAbove = cuRestrictedAbove[CH_L];
574
0
    return (( cuLeft && cuLeft->affine ) ? 1 : 0) + (( cuAbove && cuAbove->affine ) ? 1 : 0);
575
0
  }
576
577
  unsigned CtxSkipFlag() const
578
95.7k
  {
579
95.7k
    const CodingUnit *cuLeft = cuRestrictedLeft[CH_L]; const CodingUnit *cuAbove = cuRestrictedAbove[CH_L];
580
95.7k
    return (( cuLeft && cuLeft->skip ) ? 1 : 0) + (( cuAbove && cuAbove->skip ) ? 1 : 0);
581
95.7k
  }
582
583
  unsigned CtxPredModeFlag() const
584
0
  {
585
0
    const CodingUnit *cuLeft = cuRestrictedLeft[CH_L]; const CodingUnit *cuAbove = cuRestrictedAbove[CH_L];
586
0
    return ((cuAbove && cuAbove->predMode == MODE_INTRA) || (cuLeft && cuLeft->predMode == MODE_INTRA)) ? 1 : 0;
587
0
  }
588
589
  unsigned CtxIBCFlag(const CodingUnit& cu) const
590
280k
  {
591
280k
    const CodingUnit *cuLeft = cuRestrictedLeft[cu.chType]; const CodingUnit *cuAbove = cuRestrictedAbove[cu.chType];
592
280k
    return ((cuLeft && cuLeft->predMode == MODE_IBC) ? 1 : 0) + ((cuAbove && cuAbove->predMode == MODE_IBC) ? 1 : 0);
593
280k
  }
594
595
};
596
597
} // namespace vvenc
598
599
//! \}
600