Skip to content
Snippets Groups Projects
Slice.h 163 KiB
Newer Older
  • Learn to ignore specific revisions
  •   bool              m_bNoAffineMotionConstraintFlag;
      bool              m_bNoGbiConstraintFlag;
      bool              m_bNoMhIntraConstraintFlag;
      bool              m_bNoTriangleConstraintFlag;
    
      bool              m_bNoCurrPicRefConstraintFlag;
      bool              m_bNoQpDeltaConstraintFlag;
    
      bool              m_bNoDepQuantConstraintFlag;
      bool              m_bNoSignDataHidingConstraintFlag;
    
      bool              m_affineAmvrEnabledFlag;
    
      bool              m_DMVR;
    
    #if HEVC_VPS
      int               m_VPSId;
    #endif
      ChromaFormat      m_chromaFormatIdc;
    
      uint32_t              m_uiMaxTLayers;           // maximum number of temporal layers
    
      // Structure
      uint32_t              m_picWidthInLumaSamples;
      uint32_t              m_picHeightInLumaSamples;
    
      int               m_log2MinCodingBlockSize;
      int               m_log2DiffMaxMinCodingBlockSize;
    
      unsigned    m_CTUSize;
      unsigned    m_partitionOverrideEnalbed;       // enable partition constraints override function
      unsigned    m_minQT[3];   // 0: I slice luma; 1: P/B slice; 2: I slice chroma
      unsigned    m_maxBTDepth[3];
      unsigned    m_maxBTSize[3];
      unsigned    m_maxTTSize[3];
      unsigned    m_dualITree;
    
      uint32_t              m_uiMaxCUWidth;
      uint32_t              m_uiMaxCUHeight;
      uint32_t              m_uiMaxCodingDepth; ///< Total CU depth, relative to the smallest possible transform block size.
    
      Window            m_conformanceWindow;
    
      RPSList           m_RPSList;
      bool              m_bLongTermRefsPresent;
      bool              m_SPSTemporalMVPEnabledFlag;
      int               m_numReorderPics[MAX_TLAYER];
    
      // Tool list
    
      bool                  m_pcmEnabledFlag;
    
      uint32_t              m_pcmLog2MaxSize;
      uint32_t              m_uiPCMLog2MinSize;
    
      // Parameter
      BitDepths         m_bitDepths;
      int               m_qpBDOffset[MAX_NUM_CHANNEL_TYPE];
      int               m_pcmBitDepths[MAX_NUM_CHANNEL_TYPE];
      bool              m_bPCMFilterDisableFlag;
    
    
      bool              m_sbtmvpEnabledFlag;
      bool              m_bdofEnabledFlag;
    
      bool              m_disFracMmvdEnabledFlag;
    
      uint32_t              m_uiBitsForPOC;
      uint32_t              m_numLongTermRefPicSPS;
      uint32_t              m_ltRefPicPocLsbSps[MAX_NUM_LONG_TERM_REF_PICS];
      bool              m_usedByCurrPicLtSPSFlag[MAX_NUM_LONG_TERM_REF_PICS];
    
    #if MAX_TB_SIZE_SIGNALLING
      uint32_t          m_log2MaxTbSize;
    #endif
    
      bool              m_saoEnabledFlag;
    
    
      bool              m_bTemporalIdNestingFlag; // temporal_id_nesting_flag
    
    #if HEVC_USE_SCALING_LISTS
      bool              m_scalingListEnabledFlag;
      bool              m_scalingListPresentFlag;
      ScalingList       m_scalingList;
    #endif
      uint32_t              m_uiMaxDecPicBuffering[MAX_TLAYER];
      uint32_t              m_uiMaxLatencyIncreasePlus1[MAX_TLAYER];
    
    #if HEVC_USE_INTRA_SMOOTHING_T32 || HEVC_USE_INTRA_SMOOTHING_T64
      bool              m_useStrongIntraSmoothing;
    #endif
    
      bool              m_vuiParametersPresentFlag;
      VUI               m_vuiParameters;
    
      SPSRExt           m_spsRangeExtension;
    
      static const int  m_winUnitX[NUM_CHROMA_FORMAT];
      static const int  m_winUnitY[NUM_CHROMA_FORMAT];
    
    #if !JVET_M0101_HLS
    
    #else
      ProfileTierLevel  m_profileTierLevel;
    #endif
    
      bool              m_alfEnabledFlag;
    
      bool              m_wrapAroundEnabledFlag;
    
      unsigned          m_wrapAroundOffset;
    
    Yu Han's avatar
    Yu Han committed
      unsigned          m_IBCFlag;
    
    Taoran Lu's avatar
    Taoran Lu committed
      bool              m_lumaReshapeEnable;
    
      bool              m_AMVREnabledFlag;
    
      bool              m_cclmCollocatedChromaFlag;
      bool              m_MTS;
      bool              m_IntraMTS;                   // 18
      bool              m_InterMTS;                   // 19
      bool              m_Affine;
      bool              m_AffineType;
      bool              m_GBi;                        //
      bool              m_MHIntra;
      bool              m_Triangle;
    #if LUMA_ADAPTIVE_DEBLOCKING_FILTER_QP_OFFSET
      bool              m_LadfEnabled;
      int               m_LadfNumIntervals;
      int               m_LadfQpOffset[MAX_LADF_INTERVALS];
      int               m_LadfIntervalLowerBound[MAX_LADF_INTERVALS];
    #endif
    
    public:
    
      SPS();
      virtual                 ~SPS();
    
    #if HEVC_VPS
      int                     getVPSId() const                                                                { return m_VPSId;                                                      }
      void                    setVPSId(int i)                                                                 { m_VPSId = i;                                                         }
    
    #if !JVET_M0101_HLS
    
      bool                    getIntraOnlyConstraintFlag() const                                              { return m_bIntraOnlyConstraintFlag;                                   }
      void                    setIntraOnlyConstraintFlag(bool bVal)                                           { m_bIntraOnlyConstraintFlag = bVal;                                   }
      uint32_t                getMaxBitDepthConstraintIdc() const                                             { return m_maxBitDepthConstraintIdc;                                   }
      void                    setMaxBitDepthConstraintIdc(uint32_t u)                                         { m_maxBitDepthConstraintIdc = u;                                      }
      uint32_t                getMaxChromaFormatConstraintIdc() const                                         { return m_maxChromaFormatConstraintIdc;                               }
      void                    setMaxChromaFormatConstraintIdc(uint32_t u)                                     { m_maxChromaFormatConstraintIdc = u;                                  }
    
      bool                    getFrameConstraintFlag() const                                                  { return m_bFrameConstraintFlag;                                       }
      void                    setFrameConstraintFlag(bool bVal)                                               { m_bFrameConstraintFlag = bVal;                                       }
    
      bool                    getNoQtbttDualTreeIntraConstraintFlag() const                                   { return m_bNoQtbttDualTreeIntraConstraintFlag;                        }
      void                    setNoQtbttDualTreeIntraConstraintFlag(bool bVal)                                { m_bNoQtbttDualTreeIntraConstraintFlag = bVal;                        }
      bool                    getNoSaoConstraintFlag() const                                                  { return m_bNoSaoConstraintFlag;                                       }
      void                    setNoSaoConstraintFlag(bool bVal)                                               { m_bNoSaoConstraintFlag = bVal;                                       }
      bool                    getNoAlfConstraintFlag() const                                                  { return m_bNoAlfConstraintFlag;                                       }
      void                    setNoAlfConstraintFlag(bool bVal)                                               { m_bNoAlfConstraintFlag = bVal;                                       }
      bool                    getNoPcmConstraintFlag() const                                                  { return m_bNoPcmConstraintFlag;                                       }
      void                    setNoPcmConstraintFlag(bool bVal)                                               { m_bNoPcmConstraintFlag = bVal;                                       }
    
      bool                    getNoRefWraparoundConstraintFlag() const                                        { return m_bNoRefWraparoundConstraintFlag;                             }
      void                    setNoRefWraparoundConstraintFlag(bool bVal)                                     { m_bNoRefWraparoundConstraintFlag= bVal;                              }
    
      bool                    getNoTemporalMvpConstraintFlag() const                                          { return m_bNoTemporalMvpConstraintFlag;                               }
      void                    setNoTemporalMvpConstraintFlag(bool bVal)                                       { m_bNoTemporalMvpConstraintFlag = bVal;                               }
      bool                    getNoSbtmvpConstraintFlag() const                                               { return m_bNoSbtmvpConstraintFlag;                                    }
      void                    setNoSbtmvpConstraintFlag(bool bVal)                                            { m_bNoSbtmvpConstraintFlag = bVal;                                    }
      bool                    getNoAmvrConstraintFlag() const                                                 { return m_bNoAmvrConstraintFlag;                                      }
      void                    setNoAmvrConstraintFlag(bool bVal)                                              { m_bNoAmvrConstraintFlag = bVal;                                      }
    
      bool                    getNoBdofConstraintFlag() const                                                 { return m_bNoBdofConstraintFlag;                                      }
      void                    setNoBdofConstraintFlag(bool bVal)                                              { m_bNoBdofConstraintFlag = bVal;                                      }
      bool                    getNoCclmConstraintFlag() const                                                 { return m_bNoCclmConstraintFlag;                                      }
      void                    setNoCclmConstraintFlag(bool bVal)                                              { m_bNoCclmConstraintFlag = bVal;                                      }
    
      bool                    getNoMtsConstraintFlag() const                                                  { return m_bNoMtsConstraintFlag;                                       }
      void                    setNoMtsConstraintFlag(bool bVal)                                               { m_bNoMtsConstraintFlag = bVal;                                       }
    
      bool                    getNoAffineMotionConstraintFlag() const                                         { return m_bNoAffineMotionConstraintFlag;                              }
      void                    setNoAffineMotionConstraintFlag(bool bVal)                                      { m_bNoAffineMotionConstraintFlag = bVal;                              }
      bool                    getNoGbiConstraintFlag() const                                                  { return m_bNoGbiConstraintFlag;                                       }
      void                    setNoGbiConstraintFlag(bool bVal)                                               { m_bNoGbiConstraintFlag = bVal;                                       }
      bool                    getNoMhIntraConstraintFlag() const                                              { return m_bNoMhIntraConstraintFlag;                                   }
      void                    setNoMhIntraConstraintFlag(bool bVal)                                           { m_bNoMhIntraConstraintFlag = bVal;                                   }
      bool                    getNoTriangleConstraintFlag() const                                             { return m_bNoTriangleConstraintFlag;                                  }
      void                    setNoTriangleConstraintFlag(bool bVal)                                          { m_bNoTriangleConstraintFlag = bVal;                                  }
    
      bool                    getNoLadfConstraintFlag() const                                                 { return m_bNoLadfConstraintFlag;                                      }
      void                    setNoLadfConstraintFlag(bool bVal)                                              { m_bNoLadfConstraintFlag = bVal;                                      }
    
      bool                    getNoCurrPicRefConstraintFlag() const                                           { return m_bNoCurrPicRefConstraintFlag;                                }
      void                    setNoCurrPicRefConstraintFlag(bool bVal)                                        { m_bNoCurrPicRefConstraintFlag = bVal;                                }
      bool                    getNoQpDeltaConstraintFlag() const                                              { return m_bNoQpDeltaConstraintFlag;                                   }
      void                    setNoQpDeltaConstraintFlag(bool bVal)                                           { m_bNoQpDeltaConstraintFlag = bVal;                                   }
    
      bool                    getNoDepQuantConstraintFlag() const                                             { return m_bNoDepQuantConstraintFlag;                                  }
      void                    setNoDepQuantConstraintFlag(bool bVal)                                          { m_bNoDepQuantConstraintFlag = bVal;                                  }
      bool                    getNoSignDataHidingConstraintFlag() const                                       { return m_bNoSignDataHidingConstraintFlag;                            }
      void                    setNoSignDataHidingConstraintFlag(bool bVal)                                    { m_bNoSignDataHidingConstraintFlag = bVal;                            }
    
      int                     getSPSId() const                                                                { return m_SPSId;                                                      }
      void                    setSPSId(int i)                                                                 { m_SPSId = i;                                                         }
      ChromaFormat            getChromaFormatIdc () const                                                     { return m_chromaFormatIdc;                                            }
      void                    setChromaFormatIdc (ChromaFormat i)                                             { m_chromaFormatIdc = i;                                               }
    
      static int              getWinUnitX (int chromaFormatIdc)                                               { CHECK(chromaFormatIdc < 0 || chromaFormatIdc >= NUM_CHROMA_FORMAT, "Invalid chroma format parameter"); return m_winUnitX[chromaFormatIdc]; }
      static int              getWinUnitY (int chromaFormatIdc)                                               { CHECK(chromaFormatIdc < 0 || chromaFormatIdc >= NUM_CHROMA_FORMAT, "Invalid chroma format parameter"); return m_winUnitY[chromaFormatIdc]; }
    
      // structure
      void                    setPicWidthInLumaSamples( uint32_t u )                                              { m_picWidthInLumaSamples = u;                                         }
      uint32_t                    getPicWidthInLumaSamples() const                                                { return  m_picWidthInLumaSamples;                                     }
      void                    setPicHeightInLumaSamples( uint32_t u )                                             { m_picHeightInLumaSamples = u;                                        }
      uint32_t                    getPicHeightInLumaSamples() const                                               { return  m_picHeightInLumaSamples;                                    }
    
      Window&                 getConformanceWindow()                                                          { return  m_conformanceWindow;                                         }
      const Window&           getConformanceWindow() const                                                    { return  m_conformanceWindow;                                         }
      void                    setConformanceWindow(Window& conformanceWindow )                                { m_conformanceWindow = conformanceWindow;                             }
    
      uint32_t                    getNumLongTermRefPicSPS() const                                                 { return m_numLongTermRefPicSPS;                                       }
      void                    setNumLongTermRefPicSPS(uint32_t val)                                               { m_numLongTermRefPicSPS = val;                                        }
    
      uint32_t                    getLtRefPicPocLsbSps(uint32_t index) const                                          { CHECK( index >= MAX_NUM_LONG_TERM_REF_PICS, "Index exceeds boundary" ); return m_ltRefPicPocLsbSps[index]; }
      void                    setLtRefPicPocLsbSps(uint32_t index, uint32_t val)                                      { CHECK( index >= MAX_NUM_LONG_TERM_REF_PICS, "Index exceeds boundary" ); m_ltRefPicPocLsbSps[index] = val;  }
    
      bool                    getUsedByCurrPicLtSPSFlag(int i) const                                          { CHECK( i >= MAX_NUM_LONG_TERM_REF_PICS, "Index exceeds boundary" ); return m_usedByCurrPicLtSPSFlag[i];    }
      void                    setUsedByCurrPicLtSPSFlag(int i, bool x)                                        { CHECK( i >= MAX_NUM_LONG_TERM_REF_PICS, "Index exceeds boundary" ); m_usedByCurrPicLtSPSFlag[i] = x;       }
    
      int                     getLog2MinCodingBlockSize() const                                               { return m_log2MinCodingBlockSize;                                     }
      void                    setLog2MinCodingBlockSize(int val)                                              { m_log2MinCodingBlockSize = val;                                      }
      int                     getLog2DiffMaxMinCodingBlockSize() const                                        { return m_log2DiffMaxMinCodingBlockSize;                              }
      void                    setLog2DiffMaxMinCodingBlockSize(int val)                                       { m_log2DiffMaxMinCodingBlockSize = val;                               }
    
      void                    setCTUSize(unsigned    ctuSize)                                                 { m_CTUSize = ctuSize; }
      unsigned                getCTUSize()                                                              const { return  m_CTUSize; }
      void                    setSplitConsOverrideEnabledFlag(bool b)                                         { m_partitionOverrideEnalbed = b; }
      bool                    getSplitConsOverrideEnabledFlag()                                         const { return m_partitionOverrideEnalbed; }
      void                    setMinQTSizes(unsigned*   minQT)                                                { m_minQT[0] = minQT[0]; m_minQT[1] = minQT[1]; m_minQT[2] = minQT[2]; }
      unsigned                getMinQTSize(SliceType   slicetype,
                                           ChannelType chType = CHANNEL_TYPE_LUMA)
                                                                                                        const { return slicetype == I_SLICE ? (chType == CHANNEL_TYPE_LUMA ? m_minQT[0] : m_minQT[2]) : m_minQT[1]; }
      void                    setMaxBTDepth(unsigned    maxBTDepth,
                                            unsigned    maxBTDepthI,
                                            unsigned    maxBTDepthIChroma)
                                                                                                              { m_maxBTDepth[1] = maxBTDepth; m_maxBTDepth[0] = maxBTDepthI; m_maxBTDepth[2] = maxBTDepthIChroma; }
      unsigned                getMaxBTDepth()                                                           const { return m_maxBTDepth[1]; }
      unsigned                getMaxBTDepthI()                                                          const { return m_maxBTDepth[0]; }
      unsigned                getMaxBTDepthIChroma()                                                    const { return m_maxBTDepth[2]; }
      void                    setMaxBTSize(unsigned    maxBTSize,
                                           unsigned    maxBTSizeI,
    
                                           unsigned    maxBTSizeC)
    
                                                                                                              { m_maxBTSize[1] = maxBTSize; m_maxBTSize[0] = maxBTSizeI; m_maxBTSize[2] = maxBTSizeC; }
      unsigned                getMaxBTSize()                                                            const { return m_maxBTSize[1]; }
      unsigned                getMaxBTSizeI()                                                           const { return m_maxBTSize[0]; }
      unsigned                getMaxBTSizeIChroma()                                                     const { return m_maxBTSize[2]; }
      void                    setMaxTTSize(unsigned    maxTTSize,
                                           unsigned    maxTTSizeI,
                                           unsigned    maxTTSizeC)
                                                                                                              { m_maxTTSize[1] = maxTTSize; m_maxTTSize[0] = maxTTSizeI; m_maxTTSize[2] = maxTTSizeC; }
      unsigned                getMaxTTSize()                                                            const { return m_maxTTSize[1]; }
      unsigned                getMaxTTSizeI()                                                           const { return m_maxTTSize[0]; }
      unsigned                getMaxTTSizeIChroma()                                                     const { return m_maxTTSize[2]; }
    
      void                    setUseDualITree(bool b) { m_dualITree = b; }
      bool                    getUseDualITree()                                      const { return m_dualITree; }
    
    
      void                    setMaxCUWidth( uint32_t u )                                                         { m_uiMaxCUWidth = u;                                                  }
      uint32_t                    getMaxCUWidth() const                                                           { return  m_uiMaxCUWidth;                                              }
      void                    setMaxCUHeight( uint32_t u )                                                        { m_uiMaxCUHeight = u;                                                 }
      uint32_t                    getMaxCUHeight() const                                                          { return  m_uiMaxCUHeight;                                             }
      void                    setMaxCodingDepth( uint32_t u )                                                     { m_uiMaxCodingDepth = u;                                              }
      uint32_t                    getMaxCodingDepth() const                                                       { return  m_uiMaxCodingDepth;                                          }
    
      void                    setPCMEnabledFlag( bool b )                                                         { m_pcmEnabledFlag = b;                                                }
      bool                    getPCMEnabledFlag() const                                                           { return m_pcmEnabledFlag;                                             }
    
      void                    setPCMLog2MaxSize( uint32_t u )                                                     { m_pcmLog2MaxSize = u;                                                }
      uint32_t                    getPCMLog2MaxSize() const                                                       { return  m_pcmLog2MaxSize;                                            }
      void                    setPCMLog2MinSize( uint32_t u )                                                     { m_uiPCMLog2MinSize = u;                                              }
      uint32_t                    getPCMLog2MinSize() const                                                       { return  m_uiPCMLog2MinSize;                                          }
      void                    setBitsForPOC( uint32_t u )                                                         { m_uiBitsForPOC = u;                                                  }
      uint32_t                    getBitsForPOC() const                                                           { return m_uiBitsForPOC;                                               }
      void                    setNumReorderPics(int i, uint32_t tlayer)                                           { m_numReorderPics[tlayer] = i;                                        }
      int                     getNumReorderPics(uint32_t tlayer) const                                            { return m_numReorderPics[tlayer];                                     }
      void                    createRPSList( int numRPS );
      const RPSList*          getRPSList() const                                                              { return &m_RPSList;                                                   }
      RPSList*                getRPSList()                                                                    { return &m_RPSList;                                                   }
      bool                    getLongTermRefsPresent() const                                                  { return m_bLongTermRefsPresent;                                       }
      void                    setLongTermRefsPresent(bool b)                                                  { m_bLongTermRefsPresent=b;                                            }
      bool                    getSPSTemporalMVPEnabledFlag() const                                            { return m_SPSTemporalMVPEnabledFlag;                                  }
      void                    setSPSTemporalMVPEnabledFlag(bool b)                                            { m_SPSTemporalMVPEnabledFlag=b;                                       }
    
    #if MAX_TB_SIZE_SIGNALLING
      void                    setLog2MaxTbSize( uint32_t u )                                                  { m_log2MaxTbSize = u;                                                 }
      uint32_t                getLog2MaxTbSize() const                                                        { return  m_log2MaxTbSize;                                             }
      uint32_t                getMaxTbSize() const                                                            { return  1 << m_log2MaxTbSize;                                        }
    #endif
    
      // Bit-depth
      int                     getBitDepth(ChannelType type) const                                             { return m_bitDepths.recon[type];                                      }
      void                    setBitDepth(ChannelType type, int u )                                           { m_bitDepths.recon[type] = u;                                         }
      const BitDepths&        getBitDepths() const                                                            { return m_bitDepths;                                                  }
      int                     getMaxLog2TrDynamicRange(ChannelType channelType) const                         { return getSpsRangeExtension().getExtendedPrecisionProcessingFlag() ? std::max<int>(15, int(m_bitDepths.recon[channelType] + 6)) : 15; }
    
      int                     getDifferentialLumaChromaBitDepth() const                                       { return int(m_bitDepths.recon[CHANNEL_TYPE_LUMA]) - int(m_bitDepths.recon[CHANNEL_TYPE_CHROMA]); }
      int                     getQpBDOffset(ChannelType type) const                                           { return m_qpBDOffset[type];                                           }
      void                    setQpBDOffset(ChannelType type, int i)                                          { m_qpBDOffset[type] = i;                                              }
    
    
      void                    setSAOEnabledFlag(bool bVal)                                                    { m_saoEnabledFlag = bVal;                                                    }
      bool                    getSAOEnabledFlag() const                                                       { return m_saoEnabledFlag;                                                    }
    
      bool                    getALFEnabledFlag() const                                                       { return m_alfEnabledFlag; }
      void                    setALFEnabledFlag( bool b )                                                     { m_alfEnabledFlag = b; }
    
      bool                    getSBTMVPEnabledFlag() const                                                    { return m_sbtmvpEnabledFlag; }
      void                    setSBTMVPEnabledFlag(bool b)                                                    { m_sbtmvpEnabledFlag = b; }
    
      void                    setBDOFEnabledFlag(bool b)                                                      { m_bdofEnabledFlag = b; }
      bool                    getBDOFEnabledFlag() const                                                      { return m_bdofEnabledFlag; }
    
    
      bool                    getDisFracMmvdEnabledFlag() const                                               { return m_disFracMmvdEnabledFlag; }
      void                    setDisFracMmvdEnabledFlag( bool b )                                             { m_disFracMmvdEnabledFlag = b;    }
    
      bool                    getUseDMVR()const                                                               { return m_DMVR; }
      void                    setUseDMVR(bool b)                                                              { m_DMVR = b;    }
    
      uint32_t                getMaxTLayers() const                                                           { return m_uiMaxTLayers; }
      void                    setMaxTLayers( uint32_t uiMaxTLayers )                                          { CHECK( uiMaxTLayers > MAX_TLAYER, "Invalid number T-layers" ); m_uiMaxTLayers = uiMaxTLayers; }
    
    
      bool                    getTemporalIdNestingFlag() const                                                { return m_bTemporalIdNestingFlag;                                     }
      void                    setTemporalIdNestingFlag( bool bValue )                                         { m_bTemporalIdNestingFlag = bValue;                                   }
      uint32_t                    getPCMBitDepth(ChannelType type) const                                          { return m_pcmBitDepths[type];                                         }
      void                    setPCMBitDepth(ChannelType type, uint32_t u)                                        { m_pcmBitDepths[type] = u;                                            }
      void                    setPCMFilterDisableFlag( bool bValue )                                          { m_bPCMFilterDisableFlag = bValue;                                    }
      bool                    getPCMFilterDisableFlag() const                                                 { return m_bPCMFilterDisableFlag;                                      }
    
    #if HEVC_USE_SCALING_LISTS
      bool                    getScalingListFlag() const                                                      { return m_scalingListEnabledFlag;                                     }
      void                    setScalingListFlag( bool b )                                                    { m_scalingListEnabledFlag  = b;                                       }
      bool                    getScalingListPresentFlag() const                                               { return m_scalingListPresentFlag;                                     }
      void                    setScalingListPresentFlag( bool b )                                             { m_scalingListPresentFlag  = b;                                       }
      ScalingList&            getScalingList()                                                                { return m_scalingList; }
      const ScalingList&      getScalingList() const                                                          { return m_scalingList; }
    #endif
      uint32_t                    getMaxDecPicBuffering(uint32_t tlayer) const                                        { return m_uiMaxDecPicBuffering[tlayer];                               }
      void                    setMaxDecPicBuffering( uint32_t ui, uint32_t tlayer )                                   { CHECK(tlayer >= MAX_TLAYER, "Invalid T-layer"); m_uiMaxDecPicBuffering[tlayer] = ui;    }
      uint32_t                    getMaxLatencyIncreasePlus1(uint32_t tlayer) const                                   { return m_uiMaxLatencyIncreasePlus1[tlayer];                          }
      void                    setMaxLatencyIncreasePlus1( uint32_t ui , uint32_t tlayer)                              { m_uiMaxLatencyIncreasePlus1[tlayer] = ui;                            }
    
    #if HEVC_USE_INTRA_SMOOTHING_T32 || HEVC_USE_INTRA_SMOOTHING_T64
      void                    setUseStrongIntraSmoothing(bool bVal)                                           { m_useStrongIntraSmoothing = bVal;                                    }
      bool                    getUseStrongIntraSmoothing() const                                              { return m_useStrongIntraSmoothing;                                    }
    
    
    #endif
      void                    setAffineAmvrEnabledFlag( bool val )                                            { m_affineAmvrEnabledFlag = val;                                       }
      bool                    getAffineAmvrEnabledFlag() const                                                { return m_affineAmvrEnabledFlag;                                      }
    
      bool                    getVuiParametersPresentFlag() const                                             { return m_vuiParametersPresentFlag;                                   }
      void                    setVuiParametersPresentFlag(bool b)                                             { m_vuiParametersPresentFlag = b;                                      }
      VUI*                    getVuiParameters()                                                              { return &m_vuiParameters;                                             }
      const VUI*              getVuiParameters() const                                                        { return &m_vuiParameters;                                             }
    
    #if !JVET_M0101_HLS
    
      const PTL*              getPTL() const                                                                  { return &m_pcPTL;                                                     }
      PTL*                    getPTL()                                                                        { return &m_pcPTL;                                                     }
    
    #else
      const ProfileTierLevel* getProfileTierLevel() const                                                     { return &m_profileTierLevel; }
      ProfileTierLevel*       getProfileTierLevel()                                                           { return &m_profileTierLevel; }
    #endif
    
    
      const SPSRExt&          getSpsRangeExtension() const                                                    { return m_spsRangeExtension;                                          }
      SPSRExt&                getSpsRangeExtension()                                                          { return m_spsRangeExtension;                                          }
    
    
      void                    setWrapAroundEnabledFlag(bool b)                                                { m_wrapAroundEnabledFlag = b;                                         }
      bool                    getWrapAroundEnabledFlag() const                                                { return m_wrapAroundEnabledFlag;                                      }
    
      void                    setWrapAroundOffset(unsigned offset)                                            { m_wrapAroundOffset = offset;                                         }
      unsigned                getWrapAroundOffset() const                                                     { return m_wrapAroundOffset;                                           }
    
    Taoran Lu's avatar
    Taoran Lu committed
      void                    setUseReshaper(bool b)                                                          { m_lumaReshapeEnable = b;                                                   }
      bool                    getUseReshaper() const                                                          { return m_lumaReshapeEnable;                                                }
    
    Yu Han's avatar
    Yu Han committed
      void                    setIBCFlag(unsigned IBCFlag)                                                    { m_IBCFlag = IBCFlag; }
      unsigned                getIBCFlag() const                                                              { return m_IBCFlag; }
    
      void                    setUseSBT( bool b )                                                             { m_SBT = b; }
      bool                    getUseSBT() const                                                               { return m_SBT; }
      void                    setMaxSbtSize( uint8_t val )                                                    { m_MaxSbtSize = val; }
      uint8_t                 getMaxSbtSize() const                                                           { return m_MaxSbtSize; }
    
      void      setAMVREnabledFlag    ( bool b )                                        { m_AMVREnabledFlag = b; }
      bool      getAMVREnabledFlag    ()                                      const     { return m_AMVREnabledFlag; }
    
      void      setUseAffine          ( bool b )                                        { m_Affine = b; }
      bool      getUseAffine          ()                                      const     { return m_Affine; }
      void      setUseAffineType      ( bool b )                                        { m_AffineType = b; }
      bool      getUseAffineType      ()                                      const     { return m_AffineType; }
      void      setUseLMChroma        ( bool b )                                        { m_LMChroma = b; }
      bool      getUseLMChroma        ()                                      const     { return m_LMChroma; }
      void      setCclmCollocatedChromaFlag( bool b )                                   { m_cclmCollocatedChromaFlag = b; }
      bool      getCclmCollocatedChromaFlag()                                 const     { return m_cclmCollocatedChromaFlag; }
      void      setUseMTS             ( bool b )                                        { m_MTS = b; }
      bool      getUseMTS             ()                                      const     { return m_MTS; }
      bool      getUseImplicitMTS     ()                                      const     { return m_MTS && !m_IntraMTS && !m_InterMTS; }
      void      setUseIntraMTS        ( bool b )                                        { m_IntraMTS = b; }
      bool      getUseIntraMTS        ()                                      const     { return m_IntraMTS; }
      void      setUseInterMTS        ( bool b )                                        { m_InterMTS = b; }
      bool      getUseInterMTS        ()                                      const     { return m_InterMTS; }
      void      setUseGBi             ( bool b )                                        { m_GBi = b; }
      bool      getUseGBi             ()                                      const     { return m_GBi; }
    #if LUMA_ADAPTIVE_DEBLOCKING_FILTER_QP_OFFSET
      void      setLadfEnabled        ( bool b )                                        { m_LadfEnabled = b; }
      bool      getLadfEnabled        ()                                      const     { return m_LadfEnabled; }
      void      setLadfNumIntervals   ( int i )                                         { m_LadfNumIntervals = i; }
      int       getLadfNumIntervals   ()                                      const     { return m_LadfNumIntervals; }
      void      setLadfQpOffset       ( int value, int idx )                            { m_LadfQpOffset[ idx ] = value; }
      int       getLadfQpOffset       ( int idx )                             const     { return m_LadfQpOffset[ idx ]; }
      void      setLadfIntervalLowerBound( int value, int idx )                         { m_LadfIntervalLowerBound[ idx ] = value; }
      int       getLadfIntervalLowerBound( int idx )                          const     { return m_LadfIntervalLowerBound[ idx ]; }
    #endif
    
      void      setUseMHIntra         ( bool b )                                        { m_MHIntra = b; }
      bool      getUseMHIntra         ()                                      const     { return m_MHIntra; }
      void      setUseTriangle        ( bool b )                                        { m_Triangle = b; }
      bool      getUseTriangle        ()                                      const     { return m_Triangle; }
    
    };
    
    
    /// Reference Picture Lists class
    
    class RefPicListModification
    {
    private:
      bool m_refPicListModificationFlagL0;
      bool m_refPicListModificationFlagL1;
      uint32_t m_RefPicSetIdxL0[REF_PIC_LIST_NUM_IDX];
      uint32_t m_RefPicSetIdxL1[REF_PIC_LIST_NUM_IDX];
    
    public:
              RefPicListModification();
      virtual ~RefPicListModification();
    
      bool    getRefPicListModificationFlagL0() const        { return m_refPicListModificationFlagL0;                                  }
      void    setRefPicListModificationFlagL0(bool flag)     { m_refPicListModificationFlagL0 = flag;                                  }
      bool    getRefPicListModificationFlagL1() const        { return m_refPicListModificationFlagL1;                                  }
      void    setRefPicListModificationFlagL1(bool flag)     { m_refPicListModificationFlagL1 = flag;                                  }
      uint32_t    getRefPicSetIdxL0(uint32_t idx) const              { CHECK(idx>=REF_PIC_LIST_NUM_IDX, "Invalid ref-pic-list index"); return m_RefPicSetIdxL0[idx];         }
      void    setRefPicSetIdxL0(uint32_t idx, uint32_t refPicSetIdx) { CHECK(idx>=REF_PIC_LIST_NUM_IDX, "Invalid ref-pic-list index"); m_RefPicSetIdxL0[idx] = refPicSetIdx; }
      uint32_t    getRefPicSetIdxL1(uint32_t idx) const              { CHECK(idx>=REF_PIC_LIST_NUM_IDX, "Invalid ref-pic-list index"); return m_RefPicSetIdxL1[idx];         }
      void    setRefPicSetIdxL1(uint32_t idx, uint32_t refPicSetIdx) { CHECK(idx>=REF_PIC_LIST_NUM_IDX, "Invalid ref-pic-list index"); m_RefPicSetIdxL1[idx] = refPicSetIdx; }
    };
    
    
    
    /// PPS RExt class
    class PPSRExt // Names aligned to text specification
    {
    private:
      int              m_log2MaxTransformSkipBlockSize;
      bool             m_crossComponentPredictionEnabledFlag;
    
      // Chroma QP Adjustments
    
      int              m_cuChromaQpOffsetSubdiv;
    
      int              m_chromaQpOffsetListLen; // size (excludes the null entry used in the following array).
      ChromaQpAdj      m_ChromaQpAdjTableIncludingNullEntry[1+MAX_QP_OFFSET_LIST_SIZE]; //!< Array includes entry [0] for the null offset used when cu_chroma_qp_offset_flag=0, and entries [cu_chroma_qp_offset_idx+1...] otherwise
    
      uint32_t             m_log2SaoOffsetScale[MAX_NUM_CHANNEL_TYPE];
    
    public:
      PPSRExt();
    
      bool settingsDifferFromDefaults(const bool bTransformSkipEnabledFlag) const
      {
        return (bTransformSkipEnabledFlag && (getLog2MaxTransformSkipBlockSize() !=2))
            || (getCrossComponentPredictionEnabledFlag() )
            || (getChromaQpOffsetListEnabledFlag() )
            || (getLog2SaoOffsetScale(CHANNEL_TYPE_LUMA) !=0 )
            || (getLog2SaoOffsetScale(CHANNEL_TYPE_CHROMA) !=0 );
      }
    
      uint32_t                   getLog2MaxTransformSkipBlockSize() const                         { return m_log2MaxTransformSkipBlockSize;         }
      void                   setLog2MaxTransformSkipBlockSize( uint32_t u )                       { m_log2MaxTransformSkipBlockSize  = u;           }
    
      bool                   getCrossComponentPredictionEnabledFlag() const                   { return m_crossComponentPredictionEnabledFlag;   }
      void                   setCrossComponentPredictionEnabledFlag(bool value)               { m_crossComponentPredictionEnabledFlag = value;  }
    
      void                   clearChromaQpOffsetList()                                        { m_chromaQpOffsetListLen = 0;                    }
    
    
      uint32_t               getCuChromaQpOffsetSubdiv () const                               { return m_cuChromaQpOffsetSubdiv;                }
      void                   setCuChromaQpOffsetSubdiv ( uint32_t u )                         { m_cuChromaQpOffsetSubdiv = u;                   }
    
    
      bool                   getChromaQpOffsetListEnabledFlag() const                         { return getChromaQpOffsetListLen()>0;            }
      int                    getChromaQpOffsetListLen() const                                 { return m_chromaQpOffsetListLen;                 }
    
      const ChromaQpAdj&     getChromaQpOffsetListEntry( int cuChromaQpOffsetIdxPlus1 ) const
      {
        CHECK(cuChromaQpOffsetIdxPlus1 >= m_chromaQpOffsetListLen+1, "Invalid chroma QP offset");
        return m_ChromaQpAdjTableIncludingNullEntry[cuChromaQpOffsetIdxPlus1]; // Array includes entry [0] for the null offset used when cu_chroma_qp_offset_flag=0, and entries [cu_chroma_qp_offset_idx+1...] otherwise
      }
    
      void                   setChromaQpOffsetListEntry( int cuChromaQpOffsetIdxPlus1, int cbOffset, int crOffset )
      {
        CHECK(cuChromaQpOffsetIdxPlus1 == 0 || cuChromaQpOffsetIdxPlus1 > MAX_QP_OFFSET_LIST_SIZE, "Invalid chroma QP offset");
        m_ChromaQpAdjTableIncludingNullEntry[cuChromaQpOffsetIdxPlus1].u.comp.CbOffset = cbOffset; // Array includes entry [0] for the null offset used when cu_chroma_qp_offset_flag=0, and entries [cu_chroma_qp_offset_idx+1...] otherwise
        m_ChromaQpAdjTableIncludingNullEntry[cuChromaQpOffsetIdxPlus1].u.comp.CrOffset = crOffset;
        m_chromaQpOffsetListLen = std::max(m_chromaQpOffsetListLen, cuChromaQpOffsetIdxPlus1);
      }
    
      // Now: getPpsRangeExtension().getLog2SaoOffsetScale and getPpsRangeExtension().setLog2SaoOffsetScale
      uint32_t                   getLog2SaoOffsetScale(ChannelType type) const                    { return m_log2SaoOffsetScale[type];             }
      void                   setLog2SaoOffsetScale(ChannelType type, uint32_t uiBitShift)         { m_log2SaoOffsetScale[type] = uiBitShift;       }
    
    };
    
    
    /// PPS class
    class PPS
    {
    private:
      int              m_PPSId;                    // pic_parameter_set_id
      int              m_SPSId;                    // seq_parameter_set_id
      int              m_picInitQPMinus26;
      bool             m_useDQP;
      bool             m_bConstrainedIntraPred;    // constrained_intra_pred_flag
      bool             m_bSliceChromaQpFlag;       // slicelevel_chroma_qp_flag
    
      // access channel
    
      uint32_t         m_cuQpDeltaSubdiv;           // cu_qp_delta_subdiv
    
    #if JVET_N0054_JOINT_CHROMA
      int              m_chromaCbCrQpOffset;
    #endif
    
    
      uint32_t             m_numRefIdxL0DefaultActive;
      uint32_t             m_numRefIdxL1DefaultActive;
    
      bool             m_bUseWeightPred;                    //!< Use of Weighting Prediction (P_SLICE)
      bool             m_useWeightedBiPred;                 //!< Use of Weighting Bi-Prediction (B_SLICE)
      bool             m_OutputFlagPresentFlag;             //!< Indicates the presence of output_flag in slice header
      bool             m_TransquantBypassEnabledFlag;       //!< Indicates presence of cu_transquant_bypass_flag in CUs.
      bool             m_useTransformSkip;
    #if HEVC_DEPENDENT_SLICES
      bool             m_dependentSliceSegmentsEnabledFlag; //!< Indicates the presence of dependent slices
    #endif
      bool             m_tilesEnabledFlag;                  //!< Indicates the presence of tiles
      bool             m_entropyCodingSyncEnabledFlag;      //!< Indicates the presence of wavefronts
    
      bool             m_loopFilterAcrossTilesEnabledFlag;
      bool             m_uniformSpacingFlag;
      int              m_numTileColumnsMinus1;
      int              m_numTileRowsMinus1;
      std::vector<int> m_tileColumnWidth;
      std::vector<int> m_tileRowHeight;
    
      bool             m_cabacInitPresentFlag;
    
      bool             m_sliceHeaderExtensionPresentFlag;
      bool             m_loopFilterAcrossSlicesEnabledFlag;
      bool             m_deblockingFilterControlPresentFlag;
      bool             m_deblockingFilterOverrideEnabledFlag;
      bool             m_ppsDeblockingFilterDisabledFlag;
      int              m_deblockingFilterBetaOffsetDiv2;    //< beta offset for deblocking filter
      int              m_deblockingFilterTcOffsetDiv2;      //< tc offset for deblocking filter
    #if HEVC_USE_SCALING_LISTS
      bool             m_scalingListPresentFlag;
      ScalingList      m_scalingList;                       //!< ScalingList class
    #endif
      bool             m_listsModificationPresentFlag;
      uint32_t             m_log2ParallelMergeLevelMinus2;
      int              m_numExtraSliceHeaderBits;
    
      PPSRExt          m_ppsRangeExtension;
    
    public:
      PreCalcValues   *pcv;
    
    public:
                             PPS();
      virtual                ~PPS();
    
      int                    getPPSId() const                                                 { return m_PPSId;                               }
      void                   setPPSId(int i)                                                  { m_PPSId = i;                                  }
      int                    getSPSId() const                                                 { return m_SPSId;                               }
      void                   setSPSId(int i)                                                  { m_SPSId = i;                                  }
    
      int                    getPicInitQPMinus26() const                                      { return  m_picInitQPMinus26;                   }
      void                   setPicInitQPMinus26( int i )                                     { m_picInitQPMinus26 = i;                       }
      bool                   getUseDQP() const                                                { return m_useDQP;                              }
      void                   setUseDQP( bool b )                                              { m_useDQP   = b;                               }
      bool                   getConstrainedIntraPred() const                                  { return  m_bConstrainedIntraPred;              }
      void                   setConstrainedIntraPred( bool b )                                { m_bConstrainedIntraPred = b;                  }
      bool                   getSliceChromaQpFlag() const                                     { return  m_bSliceChromaQpFlag;                 }
      void                   setSliceChromaQpFlag( bool b )                                   { m_bSliceChromaQpFlag = b;                     }
    
    
      void                   setCuQpDeltaSubdiv( uint32_t u )                                 { m_cuQpDeltaSubdiv = u;                         }
      uint32_t               getCuQpDeltaSubdiv() const                                       { return m_cuQpDeltaSubdiv;                      }
    
    
      void                   setQpOffset(ComponentID compID, int i )
      {
        if      (compID==COMPONENT_Cb)
        {
          m_chromaCbQpOffset = i;
        }
        else if (compID==COMPONENT_Cr)
        {
          m_chromaCrQpOffset = i;
        }
    
    #if JVET_N0054_JOINT_CHROMA
        else if (compID==JOINT_CbCr)
        {
          m_chromaCbCrQpOffset = i;
        }
    #endif
    
        else
        {
          THROW( "Invalid chroma QP offset" );
        }
      }
      int                    getQpOffset(ComponentID compID) const
      {
    
    #if JVET_N0054_JOINT_CHROMA
        return (compID==COMPONENT_Y) ? 0 : (compID==COMPONENT_Cb ? m_chromaCbQpOffset : compID==COMPONENT_Cr ? m_chromaCrQpOffset : m_chromaCbCrQpOffset );
    #else
    
        return (compID==COMPONENT_Y) ? 0 : (compID==COMPONENT_Cb ? m_chromaCbQpOffset : m_chromaCrQpOffset );
    
      }
    
      void                   setNumRefIdxL0DefaultActive(uint32_t ui)                             { m_numRefIdxL0DefaultActive=ui;                }
      uint32_t                   getNumRefIdxL0DefaultActive() const                              { return m_numRefIdxL0DefaultActive;            }
      void                   setNumRefIdxL1DefaultActive(uint32_t ui)                             { m_numRefIdxL1DefaultActive=ui;                }
      uint32_t                   getNumRefIdxL1DefaultActive() const                              { return m_numRefIdxL1DefaultActive;            }
    
      bool                   getUseWP() const                                                 { return m_bUseWeightPred;                      }
      bool                   getWPBiPred() const                                              { return m_useWeightedBiPred;                   }
      void                   setUseWP( bool b )                                               { m_bUseWeightPred = b;                         }
      void                   setWPBiPred( bool b )                                            { m_useWeightedBiPred = b;                      }
    
      void                   setOutputFlagPresentFlag( bool b )                               { m_OutputFlagPresentFlag = b;                  }
      bool                   getOutputFlagPresentFlag() const                                 { return m_OutputFlagPresentFlag;               }
      void                   setTransquantBypassEnabledFlag( bool b )                         { m_TransquantBypassEnabledFlag = b;            }
      bool                   getTransquantBypassEnabledFlag() const                           { return m_TransquantBypassEnabledFlag;         }
    
      bool                   getUseTransformSkip() const                                      { return m_useTransformSkip;                    }
      void                   setUseTransformSkip( bool b )                                    { m_useTransformSkip  = b;                      }
    
      void                   setLoopFilterAcrossTilesEnabledFlag(bool b)                      { m_loopFilterAcrossTilesEnabledFlag = b;       }
      bool                   getLoopFilterAcrossTilesEnabledFlag() const                      { return m_loopFilterAcrossTilesEnabledFlag;    }
    #if HEVC_DEPENDENT_SLICES
      bool                   getDependentSliceSegmentsEnabledFlag() const                     { return m_dependentSliceSegmentsEnabledFlag;   }
      void                   setDependentSliceSegmentsEnabledFlag(bool val)                   { m_dependentSliceSegmentsEnabledFlag = val;    }
    #endif
      bool                   getEntropyCodingSyncEnabledFlag() const                          { return m_entropyCodingSyncEnabledFlag;        }
      void                   setEntropyCodingSyncEnabledFlag(bool val)                        { m_entropyCodingSyncEnabledFlag = val;         }
    
      void                   setTilesEnabledFlag(bool val)                                    { m_tilesEnabledFlag = val;                     }
      bool                   getTilesEnabledFlag() const                                      { return m_tilesEnabledFlag;                    }
      void                   setTileUniformSpacingFlag(bool b)                                { m_uniformSpacingFlag = b;                     }
      bool                   getTileUniformSpacingFlag() const                                { return m_uniformSpacingFlag;                  }
      void                   setNumTileColumnsMinus1(int i)                                   { m_numTileColumnsMinus1 = i;                   }
      int                    getNumTileColumnsMinus1() const                                  { return m_numTileColumnsMinus1;                }
      void                   setTileColumnWidth(const std::vector<int>& columnWidth )         { m_tileColumnWidth = columnWidth;              }
      uint32_t                   getTileColumnWidth(uint32_t columnIdx) const                         { return  m_tileColumnWidth[columnIdx];         }
      void                   setNumTileRowsMinus1(int i)                                      { m_numTileRowsMinus1 = i;                      }
      int                    getNumTileRowsMinus1() const                                     { return m_numTileRowsMinus1;                   }
      void                   setTileRowHeight(const std::vector<int>& rowHeight)              { m_tileRowHeight = rowHeight;                  }
      uint32_t                   getTileRowHeight(uint32_t rowIdx) const                              { return m_tileRowHeight[rowIdx];               }
    
      void                   setCabacInitPresentFlag( bool flag )                             { m_cabacInitPresentFlag = flag;                }
      bool                   getCabacInitPresentFlag() const                                  { return m_cabacInitPresentFlag;                }
      void                   setDeblockingFilterControlPresentFlag( bool val )                { m_deblockingFilterControlPresentFlag = val;   }
      bool                   getDeblockingFilterControlPresentFlag() const                    { return m_deblockingFilterControlPresentFlag;  }
      void                   setDeblockingFilterOverrideEnabledFlag( bool val )               { m_deblockingFilterOverrideEnabledFlag = val;  }
      bool                   getDeblockingFilterOverrideEnabledFlag() const                   { return m_deblockingFilterOverrideEnabledFlag; }
      void                   setPPSDeblockingFilterDisabledFlag(bool val)                     { m_ppsDeblockingFilterDisabledFlag = val;      } //!< set offset for deblocking filter disabled
      bool                   getPPSDeblockingFilterDisabledFlag() const                       { return m_ppsDeblockingFilterDisabledFlag;     } //!< get offset for deblocking filter disabled
      void                   setDeblockingFilterBetaOffsetDiv2(int val)                       { m_deblockingFilterBetaOffsetDiv2 = val;       } //!< set beta offset for deblocking filter
      int                    getDeblockingFilterBetaOffsetDiv2() const                        { return m_deblockingFilterBetaOffsetDiv2;      } //!< get beta offset for deblocking filter
      void                   setDeblockingFilterTcOffsetDiv2(int val)                         { m_deblockingFilterTcOffsetDiv2 = val;         } //!< set tc offset for deblocking filter
      int                    getDeblockingFilterTcOffsetDiv2() const                          { return m_deblockingFilterTcOffsetDiv2;        } //!< get tc offset for deblocking filter
    #if HEVC_USE_SCALING_LISTS
      bool                   getScalingListPresentFlag() const                                { return m_scalingListPresentFlag;              }
      void                   setScalingListPresentFlag( bool b )                              { m_scalingListPresentFlag  = b;                }
      ScalingList&           getScalingList()                                                 { return m_scalingList;                         }
      const ScalingList&     getScalingList() const                                           { return m_scalingList;                         }
    #endif
      bool                   getListsModificationPresentFlag() const                          { return m_listsModificationPresentFlag;        }
      void                   setListsModificationPresentFlag( bool b )                        { m_listsModificationPresentFlag = b;           }
      uint32_t                   getLog2ParallelMergeLevelMinus2() const                          { return m_log2ParallelMergeLevelMinus2;        }
      void                   setLog2ParallelMergeLevelMinus2(uint32_t mrgLevel)                   { m_log2ParallelMergeLevelMinus2 = mrgLevel;    }
      int                    getNumExtraSliceHeaderBits() const                               { return m_numExtraSliceHeaderBits;             }
      void                   setNumExtraSliceHeaderBits(int i)                                { m_numExtraSliceHeaderBits = i;                }
      void                   setLoopFilterAcrossSlicesEnabledFlag( bool bValue )              { m_loopFilterAcrossSlicesEnabledFlag = bValue; }
      bool                   getLoopFilterAcrossSlicesEnabledFlag() const                     { return m_loopFilterAcrossSlicesEnabledFlag;   }
      bool                   getSliceHeaderExtensionPresentFlag() const                       { return m_sliceHeaderExtensionPresentFlag;     }
      void                   setSliceHeaderExtensionPresentFlag(bool val)                     { m_sliceHeaderExtensionPresentFlag = val;      }
    
      const PPSRExt&         getPpsRangeExtension() const                                     { return m_ppsRangeExtension;                   }
      PPSRExt&               getPpsRangeExtension()                                           { return m_ppsRangeExtension;                   }
    };
    
    
    Hendry's avatar
    Hendry committed
    class APS
    {
    private:
      int                    m_APSId;                    // adaptation_parameter_set_id
      AlfSliceParam          m_alfAPSParam;
    
    public:
      APS();
      virtual                ~APS();
    
      int                    getAPSId() const                                                 { return m_APSId;                               }
      void                   setAPSId(int i)                                                  { m_APSId = i;                                  }
    
      void                   setAlfAPSParam(AlfSliceParam& alfAPSParam)                       { m_alfAPSParam = alfAPSParam;                  }
    
      const AlfSliceParam&   getAlfAPSParam() const                                           { return m_alfAPSParam;                         }
    
    Hendry's avatar
    Hendry committed
    };
    
    struct WPScalingParam
    {
      // Explicit weighted prediction parameters parsed in slice header,
      // or Implicit weighted prediction parameters (8 bits depth values).
      bool bPresentFlag;
      uint32_t uiLog2WeightDenom;
      int  iWeight;
      int  iOffset;
    
      // Weighted prediction scaling values built from above parameters (bitdepth scaled):
      int  w;
      int  o;
      int  offset;
      int  shift;
      int  round;
    
    };
    struct WPACDCParam
    {
      int64_t iAC;
      int64_t iDC;
    };
    
    
    
    /// slice header class
    class Slice
    {
    
    private:
      //  Bitstream writing
      bool                       m_saoEnabledFlag[MAX_NUM_CHANNEL_TYPE];
      int                        m_iPPSId;               ///< picture parameter set ID
      bool                       m_PicOutputFlag;        ///< pic_output_flag
      int                        m_iPOC;
      int                        m_iLastIDR;
      int                        m_iAssociatedIRAP;
      NalUnitType                m_iAssociatedIRAPType;
      const ReferencePictureSet* m_pRPS;             //< pointer to RPS, either in the SPS or the local RPS in the same slice header
      ReferencePictureSet        m_localRPS;             //< RPS when present in slice header
      int                        m_rpsIdx;               //< index of used RPS in the SPS or -1 for local RPS in the slice header
      RefPicListModification     m_RefPicListModification;
      NalUnitType                m_eNalUnitType;         ///< Nal unit type for the slice
      SliceType                  m_eSliceType;
      int                        m_iSliceQp;
      int                        m_iSliceQpBase;
    #if HEVC_DEPENDENT_SLICES
      bool                       m_dependentSliceSegmentFlag;
    #endif
      bool                       m_ChromaQpAdjEnabled;
      bool                       m_deblockingFilterDisable;
      bool                       m_deblockingFilterOverrideFlag;      //< offsets for deblocking filter inherit from PPS
      int                        m_deblockingFilterBetaOffsetDiv2;    //< beta offset for deblocking filter
      int                        m_deblockingFilterTcOffsetDiv2;      //< tc offset for deblocking filter
      int                        m_list1IdxToList0Idx[MAX_NUM_REF];
      int                        m_aiNumRefIdx   [NUM_REF_PIC_LIST_01];    //  for multiple reference of current slice
      bool                       m_pendingRasInit;
    
      bool                       m_depQuantEnabledFlag;
    #if HEVC_USE_SIGN_HIDING
      bool                       m_signDataHidingEnabledFlag;
    #endif
      bool                       m_bCheckLDC;
    
    
      int                        m_symRefIdx[2];
    
    
    #if JVET_N0054_JOINT_CHROMA
      int                        m_iSliceChromaQpDelta[MAX_NUM_COMPONENT+1];
    #else
    
      int                        m_iSliceChromaQpDelta[MAX_NUM_COMPONENT];
    
      Picture*                   m_apcRefPicList [NUM_REF_PIC_LIST_01][MAX_NUM_REF+1];
      int                        m_aiRefPOCList  [NUM_REF_PIC_LIST_01][MAX_NUM_REF+1];
      bool                       m_bIsUsedAsLongTerm[NUM_REF_PIC_LIST_01][MAX_NUM_REF+1];
      int                        m_iDepth;
    
    
      // access channel
    #if HEVC_VPS
      const VPS*                 m_pcVPS;
    #endif
      const SPS*                 m_pcSPS;
      const PPS*                 m_pcPPS;
      Picture*                   m_pcPic;
      bool                       m_colFromL0Flag;  // collocated picture from List0 flag
    
      bool                       m_noOutputPriorPicsFlag;
      bool                       m_noRaslOutputFlag;
    
      bool                       m_handleCraAsCvsStartFlag;
    
      uint32_t                   m_maxNumAffineMergeCand;
    
      double                     m_lambdas[MAX_NUM_COMPONENT];
    
      bool                       m_abEqualRef  [NUM_REF_PIC_LIST_01][MAX_NUM_REF][MAX_NUM_REF];
      uint32_t                       m_uiTLayer;
      bool                       m_bTLayerSwitchingFlag;
    
      SliceConstraint            m_sliceMode;
      uint32_t                       m_sliceArgument;
      uint32_t                       m_sliceCurStartCtuTsAddr;
      uint32_t                       m_sliceCurEndCtuTsAddr;
      uint32_t                       m_independentSliceIdx;
    #if HEVC_DEPENDENT_SLICES
      uint32_t                       m_sliceSegmentIdx;
      SliceConstraint            m_sliceSegmentMode;
      uint32_t                       m_sliceSegmentArgument;
      uint32_t                       m_sliceSegmentCurStartCtuTsAddr;
      uint32_t                       m_sliceSegmentCurEndCtuTsAddr;
    #endif
      bool                       m_nextSlice;
    #if HEVC_DEPENDENT_SLICES
      bool                       m_nextSliceSegment;
    #endif
      uint32_t                       m_sliceBits;
    #if HEVC_DEPENDENT_SLICES
      uint32_t                       m_sliceSegmentBits;
    #endif
      bool                       m_bFinalized;
    
      bool                       m_bTestWeightPred;
      bool                       m_bTestWeightBiPred;
      WPScalingParam             m_weightPredTable[NUM_REF_PIC_LIST_01][MAX_NUM_REF][MAX_NUM_COMPONENT]; // [REF_PIC_LIST_0 or REF_PIC_LIST_1][refIdx][0:Y, 1:U, 2:V]
      WPACDCParam                m_weightACDCParam[MAX_NUM_COMPONENT];
      ClpRngs                    m_clpRngs;
      std::vector<uint32_t>          m_substreamSizes;
    
      bool                       m_cabacInitFlag;
    
      bool                       m_bLMvdL1Zero;
    
    #if !JVET_M0101_HLS
    
      bool                       m_LFCrossSliceBoundaryFlag;
    
      bool                       m_enableTMVPFlag;
    
    
      SliceType                  m_encCABACTableIdx;           // Used to transmit table selection across slices.
    
      clock_t                    m_iProcessingStartTime;
      double                     m_dProcessingTime;
    
      bool                       m_splitConsOverrideFlag;
      uint32_t                   m_uiMinQTSize;
      uint32_t                   m_uiMaxBTDepth;
      uint32_t                   m_uiMaxTTSize;
    
      uint32_t                   m_uiMinQTSizeIChroma;
      uint32_t                   m_uiMaxBTDepthIChroma;
      uint32_t                   m_uiMaxBTSizeIChroma;
      uint32_t                   m_uiMaxTTSizeIChroma;
    
      bool                       m_tileGroupAlfEnabledFlag;
    
    #if JVET_N0329_IBC_SEARCH_IMP 
      bool                       m_disableSATDForRd;
    #endif
    
    Taoran Lu's avatar
    Taoran Lu committed
      SliceReshapeInfo           m_sliceReshapeInfo;
    
    public:
                                  Slice();
      virtual                     ~Slice();
      void                        initSlice();
      int                         getRefIdx4MVPair( RefPicList eCurRefPicList, int nCurRefIdx );
    #if HEVC_VPS
      void                        setVPS( VPS* pcVPS )                                   { m_pcVPS = pcVPS;                                              }
      const VPS*                  getVPS() const                                         { return m_pcVPS;                                               }
    #endif
      void                        setSPS( const SPS* pcSPS )                             { m_pcSPS = pcSPS;                                              }
      const SPS*                  getSPS() const                                         { return m_pcSPS;                                               }
    
      void                        setPPS( const PPS* pcPPS )                             { m_pcPPS = pcPPS; m_iPPSId = (pcPPS) ? pcPPS->getPPSId() : -1; }
      const PPS*                  getPPS() const                                         { return m_pcPPS;                                               }
    
      void                        setPPSId( int PPSId )                                  { m_iPPSId = PPSId;                                             }
      int                         getPPSId() const                                       { return m_iPPSId;                                              }
    
      void                        setAPS(APS* aps)                                     { m_aps = aps; m_apsId = (aps) ? aps->getAPSId() : -1; }
      APS*                        getAPS()                                               { return m_aps;                                               }
      void                        setAPSId(int apsId)                                    { m_apsId = apsId;                                             }
      int                         getAPSId() const                                       { return m_apsId;                                              }
    
      void                        setPicOutputFlag( bool b   )                           { m_PicOutputFlag = b;                                          }
      bool                        getPicOutputFlag() const                               { return m_PicOutputFlag;                                       }
      void                        setSaoEnabledFlag(ChannelType chType, bool s)          {m_saoEnabledFlag[chType] =s;                                   }
      bool                        getSaoEnabledFlag(ChannelType chType) const            { return m_saoEnabledFlag[chType];                              }
      void                        setRPS( const ReferencePictureSet *pcRPS )             { m_pRPS = pcRPS;                                               }
      const ReferencePictureSet*  getRPS()                                               { return m_pRPS;                                                }
      ReferencePictureSet*        getLocalRPS()                                          { return &m_localRPS;                                           }
    
      void                        setRPSidx( int rpsIdx )                                { m_rpsIdx = rpsIdx;                                            }
      int                         getRPSidx() const                                      { return m_rpsIdx;                                              }
      RefPicListModification*     getRefPicListModification()                            { return &m_RefPicListModification;                             }
      void                        setLastIDR(int iIDRPOC)                                { m_iLastIDR = iIDRPOC;                                         }
      int                         getLastIDR() const                                     { return m_iLastIDR;                                            }
      void                        setAssociatedIRAPPOC(int iAssociatedIRAPPOC)           { m_iAssociatedIRAP = iAssociatedIRAPPOC;                       }
      int                         getAssociatedIRAPPOC() const                           { return m_iAssociatedIRAP;                                     }
      void                        setAssociatedIRAPType(NalUnitType associatedIRAPType)  { m_iAssociatedIRAPType = associatedIRAPType;                   }
      NalUnitType                 getAssociatedIRAPType() const                          { return m_iAssociatedIRAPType;                                 }
      SliceType                   getSliceType() const                                   { return m_eSliceType;                                          }
      int                         getPOC() const                                         { return m_iPOC;                                                }
      int                         getSliceQp() const                                     { return m_iSliceQp;                                            }
      bool                        getUseWeightedPrediction() const                       { return( (m_eSliceType==P_SLICE && testWeightPred()) || (m_eSliceType==B_SLICE && testWeightBiPred()) ); }
    #if HEVC_DEPENDENT_SLICES
      bool                        getDependentSliceSegmentFlag() const                   { return m_dependentSliceSegmentFlag;                           }
      void                        setDependentSliceSegmentFlag(bool val)                 { m_dependentSliceSegmentFlag = val;                            }
    #endif
      int                         getSliceQpDelta() const                                { return m_iSliceQpDelta;                                       }
      int                         getSliceChromaQpDelta(ComponentID compID) const        { return isLuma(compID) ? 0 : m_iSliceChromaQpDelta[compID];    }
      bool                        getUseChromaQpAdj() const                              { return m_ChromaQpAdjEnabled;                                  }
      bool                        getDeblockingFilterDisable() const                     { return m_deblockingFilterDisable;                             }
      bool                        getDeblockingFilterOverrideFlag() const                { return m_deblockingFilterOverrideFlag;                        }
      int                         getDeblockingFilterBetaOffsetDiv2()const               { return m_deblockingFilterBetaOffsetDiv2;                      }
      int                         getDeblockingFilterTcOffsetDiv2() const                { return m_deblockingFilterTcOffsetDiv2;                        }
      bool                        getPendingRasInit() const                              { return m_pendingRasInit;                                      }
      void                        setPendingRasInit( bool val )                          { m_pendingRasInit = val;                                       }
    
      int                         getNumRefIdx( RefPicList e ) const                     { return m_aiNumRefIdx[e];                                      }
      Picture*                    getPic()                                               { return m_pcPic;                                               }
      const Picture*              getPic() const                                         { return m_pcPic;                                               }
      const Picture*              getRefPic( RefPicList e, int iRefIdx) const            { return m_apcRefPicList[e][iRefIdx];                           }
      int                         getRefPOC( RefPicList e, int iRefIdx) const            { return m_aiRefPOCList[e][iRefIdx];                            }
      int                         getDepth() const                                       { return m_iDepth;                                              }
      bool                        getColFromL0Flag() const                               { return m_colFromL0Flag;                                       }
      uint32_t                        getColRefIdx() const                                   { return m_colRefIdx;                                           }
      void                        checkColRefIdx(uint32_t curSliceSegmentIdx, const Picture* pic);
      bool                        getIsUsedAsLongTerm(int i, int j) const                { return m_bIsUsedAsLongTerm[i][j];                             }
      void                        setIsUsedAsLongTerm(int i, int j, bool value)          { m_bIsUsedAsLongTerm[i][j] = value;                            }
      bool                        getCheckLDC() const                                    { return m_bCheckLDC;                                           }
      bool                        getMvdL1ZeroFlag() const                               { return m_bLMvdL1Zero;                                         }
      int                         getNumRpsCurrTempList() const;
      int                         getList1IdxToList0Idx( int list1Idx ) const            { return m_list1IdxToList0Idx[list1Idx];                        }
    
    #if !JVET_M0101_HLS
    
      bool                        isReferenceNalu() const                                { return ((getNalUnitType() <= NAL_UNIT_RESERVED_VCL_R15) && (getNalUnitType()%2 != 0)) || ((getNalUnitType() >= NAL_UNIT_CODED_SLICE_BLA_W_LP) && (getNalUnitType() <= NAL_UNIT_RESERVED_IRAP_VCL23) ); }
    
      void                        setPOC( int i )                                        { m_iPOC              = i;                                      }
      void                        setNalUnitType( NalUnitType e )                        { m_eNalUnitType      = e;                                      }
      NalUnitType                 getNalUnitType() const                                 { return m_eNalUnitType;                                        }
      bool                        getRapPicFlag() const;
      bool                        getIdrPicFlag() const                                  { return getNalUnitType() == NAL_UNIT_CODED_SLICE_IDR_W_RADL || getNalUnitType() == NAL_UNIT_CODED_SLICE_IDR_N_LP; }
    
    #if !JVET_M0101_HLS
    
      bool                        isIRAP() const                                         { return (getNalUnitType() >= 16) && (getNalUnitType() <= 23);  }
      bool                        isIDRorBLA() const                                      { return (getNalUnitType() >= 16) && (getNalUnitType() <= 20);  }
    
    #else
      bool                        isIRAP() const { return (getNalUnitType() >= NAL_UNIT_CODED_SLICE_IDR_W_RADL) && (getNalUnitType() <= NAL_UNIT_RESERVED_IRAP_VCL13); }
      bool                        isIDRorBLA() const { return (getNalUnitType() >= NAL_UNIT_CODED_SLICE_IDR_W_RADL) && (getNalUnitType() <= NAL_UNIT_CODED_SLICE_IDR_N_LP); }
    #endif
    
      void                        checkCRA(const ReferencePictureSet *pReferencePictureSet, int& pocCRA, NalUnitType& associatedIRAPType, PicList& rcListPic);
      void                        decodingRefreshMarking(int& pocCRA, bool& bRefreshPending, PicList& rcListPic, const bool bEfficientFieldIRAPEnabled);
      void                        setSliceType( SliceType e )                            { m_eSliceType        = e;                                      }
      void                        setSliceQp( int i )                                    { m_iSliceQp          = i;                                      }
      void                        setSliceQpDelta( int i )                               { m_iSliceQpDelta     = i;                                      }
      void                        setSliceChromaQpDelta( ComponentID compID, int i )     { m_iSliceChromaQpDelta[compID] = isLuma(compID) ? 0 : i;       }
      void                        setUseChromaQpAdj( bool b )                            { m_ChromaQpAdjEnabled = b;                                     }
      void                        setDeblockingFilterDisable( bool b )                   { m_deblockingFilterDisable= b;                                 }
      void                        setDeblockingFilterOverrideFlag( bool b )              { m_deblockingFilterOverrideFlag = b;                           }
      void                        setDeblockingFilterBetaOffsetDiv2( int i )             { m_deblockingFilterBetaOffsetDiv2 = i;                         }
      void                        setDeblockingFilterTcOffsetDiv2( int i )               { m_deblockingFilterTcOffsetDiv2 = i;                           }
    
      void                        setNumRefIdx( RefPicList e, int i )                    { m_aiNumRefIdx[e]    = i;                                      }
      void                        setPic( Picture* p )                                   { m_pcPic             = p;                                      }
      void                        setDepth( int iDepth )                                 { m_iDepth            = iDepth;                                 }
    
      void                        setRefPicList( PicList& rcListPic, bool checkNumPocTotalCurr = false, bool bCopyL0toL1ErrorCase = false );
      void                        setRefPOCList();
    
      void                        setColFromL0Flag( bool colFromL0 )                     { m_colFromL0Flag = colFromL0;                                  }
      void                        setColRefIdx( uint32_t refIdx)                             { m_colRefIdx = refIdx;                                         }
      void                        setCheckLDC( bool b )                                  { m_bCheckLDC = b;                                              }
      void                        setMvdL1ZeroFlag( bool b)                              { m_bLMvdL1Zero = b;                                            }
    
    
      void                        setBiDirPred( bool b, int refIdx0, int refIdx1 ) { m_biDirPred = b; m_symRefIdx[0] = refIdx0; m_symRefIdx[1] = refIdx1; }
      bool                        getBiDirPred() const { return m_biDirPred; }
    
      int                         getSymRefIdx( int refList ) const { return m_symRefIdx[refList]; }
    
    
      bool                        isIntra() const                                        { return m_eSliceType == I_SLICE;                               }
      bool                        isInterB() const                                       { return m_eSliceType == B_SLICE;                               }
      bool                        isInterP() const                                       { return m_eSliceType == P_SLICE;                               }
    
      void                        setLambdas( const double lambdas[MAX_NUM_COMPONENT] )  { for (int component = 0; component < MAX_NUM_COMPONENT; component++) m_lambdas[component] = lambdas[component]; }
      const double*               getLambdas() const                                     { return m_lambdas;                                             }
    
    
      void                        setSplitConsOverrideFlag(bool b)                       { m_splitConsOverrideFlag = b; }
      bool                        getSplitConsOverrideFlag() const                       { return m_splitConsOverrideFlag; }
      void                        setMinQTSize(int i)                                    { m_uiMinQTSize = i; }
      uint32_t                    getMinQTSize() const                                   { return m_uiMinQTSize; }
      void                        setMaxBTDepth(int i)                                   { m_uiMaxBTDepth = i; }
      uint32_t                    getMaxBTDepth() const                                  { return m_uiMaxBTDepth; }
      void                        setMaxTTSize(int i)                                    { m_uiMaxTTSize = i; }
      uint32_t                    getMaxTTSize() const                                   { return m_uiMaxTTSize; }
    
      void                        setMinQTSizeIChroma(int i)                             { m_uiMinQTSizeIChroma = i; }
      uint32_t                    getMinQTSizeIChroma() const                            { return m_uiMinQTSizeIChroma; }
      void                        setMaxBTDepthIChroma(int i)                            { m_uiMaxBTDepthIChroma = i; }
      uint32_t                    getMaxBTDepthIChroma() const                           { return m_uiMaxBTDepthIChroma; }
      void                        setMaxBTSizeIChroma(int i)                             { m_uiMaxBTSizeIChroma = i; }
      uint32_t                    getMaxBTSizeIChroma() const                            { return m_uiMaxBTSizeIChroma; }
      void                        setMaxTTSizeIChroma(int i)                             { m_uiMaxTTSizeIChroma = i; }
      uint32_t                    getMaxTTSizeIChroma() const                            { return m_uiMaxTTSizeIChroma; }
    
      void                        setMaxBTSize(int i)                                    { m_uiMaxBTSize = i; }
      uint32_t                        getMaxBTSize() const                                   { return m_uiMaxBTSize; }
    
      void                        setDepQuantEnabledFlag( bool b )                       { m_depQuantEnabledFlag = b; }
      bool                        getDepQuantEnabledFlag() const                         { return m_depQuantEnabledFlag; }
    #if HEVC_USE_SIGN_HIDING
      void                        setSignDataHidingEnabledFlag( bool b )                 { m_signDataHidingEnabledFlag = b;              }
      bool                        getSignDataHidingEnabledFlag() const                   { return m_signDataHidingEnabledFlag;           }
    #endif
    
      void                        initEqualRef();