VLCWriter.h 7.30 KiB
/* The copyright in this software is being made available under the BSD
* License, included below. This software may be subject to other third party
* and contributor rights, including patent rights, and no such rights are
* granted under this license.
*
* Copyright (c) 2010-2023, ITU/ISO/IEC
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are met:
*
* * Redistributions of source code must retain the above copyright notice,
* this list of conditions and the following disclaimer.
* * Redistributions in binary form must reproduce the above copyright notice,
* this list of conditions and the following disclaimer in the documentation
* and/or other materials provided with the distribution.
* * Neither the name of the ITU/ISO/IEC nor the names of its contributors may
* be used to endorse or promote products derived from this software without
* specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS
* BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
* THE POSSIBILITY OF SUCH DAMAGE.
*/
/** \file VLCWriter.h
* \brief Writer for high level syntax
*/
#ifndef __VLCWRITER__
#define __VLCWRITER__
#include "CommonLib/CommonDef.h"
#include "CommonLib/BitStream.h"
#include "CommonLib/Rom.h"
#include "CommonLib/Slice.h"
#include "CABACWriter.h"
//! \ingroup EncoderLib
//! \{
#if ENABLE_TRACING
#define WRITE_SCODE( value, length, name) xWriteSCodeTr ( value, length, name )
#define WRITE_CODE( value, length, name) xWriteCodeTr ( value, length, name )
#define WRITE_UVLC( value, name) xWriteUvlcTr ( value, name )
#define WRITE_SVLC( value, name) xWriteSvlcTr ( value, name )
#define WRITE_FLAG( value, name) xWriteFlagTr ( value, name )
extern bool g_HLSTraceEnable;
#else
#define WRITE_SCODE( value, length, name) xWriteSCode ( value, length )
#define WRITE_CODE( value, length, name) xWriteCode ( value, length )
#define WRITE_UVLC( value, name) xWriteUvlc ( value )
#define WRITE_SVLC( value, name) xWriteSvlc ( value )
#define WRITE_FLAG( value, name) xWriteFlag ( value )
#endif
class VLCWriter{
protected:
OutputBitstream* m_pcBitIf;
VLCWriter() : m_pcBitIf(NULL) {}
virtual ~VLCWriter() {}
void setBitstream ( OutputBitstream* p ) { m_pcBitIf = p; }
#if JVET_S0266_VUI_length
OutputBitstream* getBitstream( ) { return m_pcBitIf; }
#endif
void xWriteSCode ( int code, uint32_t length );
void xWriteCode ( uint32_t uiCode, uint32_t uiLength );
void xWriteUvlc ( uint32_t uiCode );
void xWriteSvlc ( int iCode );
void xWriteFlag ( uint32_t uiCode );
#if ENABLE_TRACING
void xWriteSCodeTr ( int value, uint32_t length, const char *pSymbolName);
void xWriteCodeTr ( uint32_t value, uint32_t length, const char *pSymbolName);
void xWriteUvlcTr ( uint32_t value, const char *pSymbolName);
void xWriteSvlcTr ( int value, const char *pSymbolName);
void xWriteFlagTr ( uint32_t value, const char *pSymbolName);
#endif
void xWriteRbspTrailingBits();
bool isByteAligned() { return (m_pcBitIf->getNumBitsUntilByteAligned() == 0); } ;
};
class AUDWriter : public VLCWriter
{
public:
AUDWriter() {};
virtual ~AUDWriter() {};
void codeAUD(OutputBitstream& bs, const bool audIrapOrGdrAuFlag, const int pictureType);
};
class HLSWriter : public VLCWriter
{
public:
HLSWriter() {}
virtual ~HLSWriter() {}
private:
void xCodeRefPicList( const ReferencePictureList* rpl, bool isLongTermPresent, uint32_t ltLsbBitsCount, const bool isForbiddenZeroDeltaPoc, int rplIdx);
bool xFindMatchingLTRP ( Slice* pcSlice, uint32_t *ltrpsIndex, int ltrpPOC, bool usedFlag );
void xCodePredWeightTable ( Slice* pcSlice );
void xCodePredWeightTable ( PicHeader *picHeader, const SPS *sps );
void xCodeScalingList ( const ScalingList* scalingList, uint32_t scalinListId, bool isPredictor);
public:
void setBitstream ( OutputBitstream* p ) { m_pcBitIf = p; }
uint32_t getNumberOfWrittenBits () { return m_pcBitIf->getNumberOfWrittenBits(); }
void codeVUI ( const VUI *pcVUI, const SPS* pcSPS );
void codeSPS ( const SPS* pcSPS );
void codePPS ( const PPS* pcPPS );
#if EMBEDDED_APS
void codeAPS ( APS* pcAPS, const bool writeRbspTrailingBits );
#else
void codeAPS ( APS* pcAPS );
#endif
void codeAlfAps ( APS* pcAPS );
void codeLmcsAps ( APS* pcAPS );
void codeScalingListAps ( APS* pcAPS );
void codeVPS ( const VPS* pcVPS );
void codeDCI ( const DCI* dci );
#if EMBEDDED_APS
void codePictureHeader ( PicHeader* picHeader, const bool writeRbspTrailingBits, std::vector<APS>& aps );
#else
void codePictureHeader ( PicHeader* picHeader, bool writeRbspTrailingBits );
#endif
#if EMBEDDED_APS
void codeSliceHeader ( std::vector<APS>& aps, Slice* pcSlice );
#else
void codeSliceHeader ( Slice* pcSlice );
#endif
#if JVET_AI0084_ALF_RESIDUALS_SCALING
void writeScaleAlf ( const SPS* sps, ScaleAlf& curScaleAlfParam, const int nbCorr );
void writeScaleAlf ( Slice* pcSlice );
#endif
void codeConstraintInfo ( const ConstraintInfo* cinfo );
void codeProfileTierLevel ( const ProfileTierLevel* ptl, bool profileTierPresentFlag, int maxNumSubLayersMinus1 );
void codeOlsHrdParameters(const GeneralHrdParams * generalHrd, const OlsHrdParams *olsHrd , const uint32_t firstSubLayer, const uint32_t maxNumSubLayersMinus1);
void codeGeneralHrdparameters(const GeneralHrdParams *hrd);
void codeTilesWPPEntryPoint ( Slice* pSlice );
#if JVET_R0433
void codeScalingList(const ScalingList &scalingList, bool aps_chromaPresentFlag);
#else
void codeScalingList ( const ScalingList &scalingList );
#endif
#if JVET_W0066_CCSAO
void codeCcSao ( Slice* pcSlice, PicHeader* picHeader, const SPS* sps, const CcSaoComParam& ccSaoParam );
#endif
#if ALF_IMPROVEMENT
void alfFilter( const AlfParam& alfParam, const bool isChroma, const int altIdx, int order0, int order1 );
void alfGolombEncode(const int coeff, const int k, const bool signed_coeff = true);
#else
void alfFilter( const AlfParam& alfParam, const bool isChroma, const int altIdx );
#endif
void dpb_parameters(int maxSubLayersMinus1, bool subLayerInfoFlag, const SPS *pcSPS);
private:
};
//! \}
#endif