/* -----------------------------------------------------------------------------

The copyright in this software is being made available under the Clear BSD

License, included below. No patent rights, trademark rights and/or 

other Intellectual Property Rights other than the copyrights concerning 

the Software are granted under this license.

The Clear BSD License

Copyright (c) 2019-2026, Fraunhofer-Gesellschaft zur Förderung der angewandten Forschung e.V. & The VVenC Authors.

All rights reserved.

Redistribution and use in source and binary forms, with or without modification,

are permitted (subject to the limitations in the disclaimer below) 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 copyright holder nor the names of its

     contributors may be used to endorse or promote products derived from this

     software without specific prior written permission.

NO EXPRESS OR IMPLIED LICENSES TO ANY PARTY'S PATENT RIGHTS ARE GRANTED BY

THIS LICENSE. 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     EncGOP.h

    \brief    GOP encoder class (header)

*/

#pragma once

#include "EncSampleAdaptiveOffset.h"

#include "VLCWriter.h"

#include "SEIwrite.h"

#include "SEIEncoder.h"

#include "Analyze.h"

#include "EncPicture.h"

#include "CommonLib/Picture.h"

#include "CommonLib/CommonDef.h"

#include "CommonLib/Nal.h"

#include "EncHRD.h"

#include "EncStage.h"

#include "SEIFilmGrainAnalyzer.h"

#include <vector>

#include <list>

#include <stdlib.h>

#include <atomic>

#include "Utilities/NoMallocThreadPool.h"

#include <mutex>

#include <condition_variable>

//! \ingroup EncoderLib

//! \{

namespace vvenc {

// ====================================================================================================================

class InputByteStream;

class DecLib;

class EncHRD;

class MsgLog;

class GOPCfg;

// ====================================================================================================================

class EncGOP;

struct FinishTaskParam {

  EncGOP*     gopEncoder;

  EncPicture* picEncoder;

  Picture*    pic;

  FinishTaskParam()                                          : gopEncoder( nullptr ), picEncoder( nullptr ), pic( nullptr ) {}

  FinishTaskParam( EncGOP* _g, EncPicture* _e, Picture* _p ) : gopEncoder( _g ),      picEncoder( _e ),      pic( _p )      {}

};

// ====================================================================================================================

struct RateCapParam {

  unsigned accumActualBits = 0;

  unsigned accumTargetBits = 0;

  unsigned accumGopCounter = 0;

  double   nonRateCapEstim = 0.0;

  int      gopAdaptedQPAdj = 0;

  void reset()

  {

    accumActualBits = 0;

    accumTargetBits = 0;

    accumGopCounter = 0;

  }

};

// ====================================================================================================================

// film grain parameters estimated by the film grain analysis, as signalled in the FGC SEI

struct FilmGrainParam {

  int               log2ScaleFactor = 0;

  SeiFgc::CompModel compModel[ MAX_NUM_COMP ];

  FilmGrainParam() = default;

  explicit FilmGrainParam( FGAnalyzer& fgAnalyzer )

    : log2ScaleFactor( fgAnalyzer.getLog2scaleFactor() )

  {

    for( int compIdx = 0; compIdx < MAX_NUM_COMP; compIdx++ )

    {

      compModel[ compIdx ] = fgAnalyzer.getCompModel( compIdx );

    }

  }

};

class EncGOP : public EncStage

{

private:

  MsgLog&                   msg;

  Analyze                   m_AnalyzeAll;

  Analyze                   m_AnalyzeI;

  Analyze                   m_AnalyzeP;

  Analyze                   m_AnalyzeB;

  std::function<void( void*, vvencYUVBuffer* )> m_recYuvBufFunc;

  void*                                         m_recYuvBufCtx;

  NoMallocThreadPool*       m_threadPool;

  std::mutex                m_gopEncMutex;

  std::condition_variable   m_gopEncCond;

  XUCache                   m_shrdUnitCache;

  std::mutex                m_unitCacheMutex;

  const VVEncCfg*           m_pcEncCfg;

  const GOPCfg*             m_gopCfg;

  RateCtrl*                 m_pcRateCtrl;

  HLSWriter                 m_HLSWriter;

  SEIWriter                 m_seiWriter;

  SEIEncoder                m_seiEncoder;

  BlkStat                   m_BlkStat;

  ParameterSetMap<SPS>      m_spsMap;

  ParameterSetMap<PPS>      m_ppsMap;

  EncHRD                    m_EncHRD;

  VPS                       m_VPS;

  DCI                       m_DCI;

  bool                      m_isPreAnalysis;

  bool                      m_bFirstWrite;

  bool                      m_bRefreshPending;

  int                       m_lastCodingNum;

  int                       m_numPicsCoded;

  int                       m_numPicsInMissing;

  int                       m_numPicsOutOffset;

  uint64_t                  m_lastCts;

  int                       m_pocRecOut;

  int                       m_ticksPerFrameMul4;

  int                       m_lastIDR;

  int                       m_lastRasPoc;

  int                       m_pocCRA;

  int                       m_associatedIRAPPOC;

  vvencNalUnitType          m_associatedIRAPType;

  RateCapParam              m_rcap;

  ThreadRsrcPool            m_threadRsrc;          ///< CU encoders of the worker threads, shared by all picture encoders

  std::list<EncPicture*>    m_freePicEncoderList;

  std::list<Picture*>       m_gopEncListInput;

  std::list<Picture*>       m_gopEncListOutput;

  std::list<Picture*>       m_procList;

  std::list<Picture*>       m_rcUpdateList;

  std::list<Picture*>       m_rcInputReorderList;  // used in RC in IFP lines synchro mode

  std::deque<PicApsGlobal*> m_globalApsList;

  std::vector<int>          m_globalCtuQpVector;

  bool                      m_forceSCC;

  FGAnalyzer                m_fgAnalyzer;

  std::deque<std::pair<int, FilmGrainParam>> m_fgaResults;  ///< analysis results by coding number, waiting for the analyzed picture's access unit

  FilmGrainParam            m_fgcParam;                     ///< film grain parameters signalled in the access units being written

public:

  EncGOP( MsgLog& msglog );

  virtual ~EncGOP();

  void setRecYUVBufferCallback( void* ctx, std::function<void( void*, vvencYUVBuffer* )> func );

  void init               ( const VVEncCfg& encCfg, const GOPCfg* gopCfg, RateCtrl& rateCtrl, NoMallocThreadPool* threadPool, bool isPreAnalysis );

  void printOutSummary    ( const bool printMSEBasedSNR, const bool printSequenceMSE, const bool printHexPsnr );

  void getParameterSets   ( AccessUnitList& accessUnit );

protected:

  virtual void initPicture    ( Picture* pic );

  virtual void processPictures( const PicList& picList, AccessUnitList& auList, PicList& doneList, PicList& freeList );

  virtual void waitForFreeEncoders();

private:

  void xUpdateRasInit                 ( Slice* slice );

  void xProcessPictures               ( AccessUnitList& auList, PicList& doneList );

  void xEncodePicture                 ( Picture* pic, EncPicture* picEncoder );

  void xOutputRecYuv                  ( const PicList& picList );

  void xReleasePictures               ( const PicList& picList, PicList& freeList );

  void xInitVPS                       ( VPS &vps ) const;

  void xInitDCI                       ( DCI &dci, const SPS &sps, const int dciId ) const;

  void xInitConstraintInfo            ( ConstraintInfo &ci ) const;

  void xInitSPS                       ( SPS &sps ) const;

  void xInitPPS                       ( PPS &pps, const SPS &sps ) const;

  void xInitPPSforTiles               ( PPS &pps, const SPS &sps ) const;

  void xInitRPL                       ( SPS &sps ) const;

  void xInitHrdParameters             ( SPS &sps );

  vvencNalUnitType xGetNalUnitType    ( const GOPEntry* _gopEntry ) const;

  bool xIsSliceTemporalSwitchingPoint ( const Slice* slice, const PicList& picList ) const;

  void xSetupPicAps                   ( Picture* pic );

  void xAnalyzeFilmGrain              ( Picture& pic );

  void xInitPicsInCodingOrder         ( const PicList& picList );

  void xGetProcessingLists            ( std::list<Picture*>& procList, std::list<Picture*>& rcUpdateList, const bool lockStepMode );

  void xUpdateVAStartOfLastGop        ( Picture& keyPic ) const;

  void xInitGopQpCascade              ( Picture& keyPic, PicList::const_iterator picItr, const PicList& picList );

  void xInitFirstSlice                ( Picture& pic, const PicList& picList, bool isEncodeLtRef );

  void xInitSliceTMVPFlag             ( PicHeader* picHeader, const Slice* slice );

  void xUpdateRPRtmvp                 ( PicHeader* picHeader, Slice* slice );

  void xInitSliceMvdL1Zero            ( PicHeader* picHeader, const Slice* slice );

  void xSelectReferencePictureList    ( Slice* slice ) const;

  void xSyncAlfAps                    ( Picture& pic );

  void xUpdateRcIfp                   ();

  void xWritePicture                  ( Picture& pic, AccessUnitList& au, bool isEncodeLtRef );

  int  xWriteParameterSets            ( Picture& pic, AccessUnitList& accessUnit, HLSWriter& hlsWriter );

  int  xWritePictureSlices            ( Picture& pic, AccessUnitList& accessUnit, HLSWriter& hlsWriter );

  void xWriteLeadingSEIs              ( const Picture& pic, AccessUnitList& accessUnit );

  void xWriteTrailingSEIs             ( const Picture& pic, AccessUnitList& accessUnit, std::string& digestStr );

  int  xWriteVPS                      ( AccessUnitList &accessUnit, const VPS *vps, HLSWriter& hlsWriter );

  int  xWriteDCI                      ( AccessUnitList &accessUnit, const DCI *dci, HLSWriter& hlsWriter );

  int  xWriteSPS                      ( AccessUnitList &accessUnit, const SPS *sps, HLSWriter& hlsWriter );

  int  xWritePPS                      ( AccessUnitList &accessUnit, const PPS *pps, const SPS *sps, HLSWriter& hlsWriter );

  int  xWriteAPS                      ( AccessUnitList &accessUnit, const APS *aps, HLSWriter& hlsWriter, vvencNalUnitType eNalUnitType );

  void xWriteAccessUnitDelimiter      ( AccessUnitList &accessUnit, Slice* slice, bool IrapOrGdr, HLSWriter& hlsWriter );

  void xWriteSEI                      ( vvencNalUnitType naluType, SEIMessages& seiMessages, AccessUnitList &accessUnit, AccessUnitList::iterator &auPos, int temporalId, const SPS *sps );

  void xWriteSEISeparately            ( vvencNalUnitType naluType, SEIMessages& seiMessages, AccessUnitList &accessUnit, AccessUnitList::iterator &auPos, int temporalId, const SPS *sps );

  void xAttachSliceDataToNalUnit      ( OutputNALUnit& rNalu, const OutputBitstream* pcBitstreamRedirect );

  void xCabacZeroWordPadding          ( const Picture& pic, const Slice* slice, uint32_t binCountsInNalUnits, uint32_t numBytesInVclNalUnits, std::ostringstream &nalUnitData );

  void xUpdateRateCap();

  void xUpdateRateCapBits             ( const Picture* pic, const uint32_t uibits );

  void xAddPSNRStats( const Picture* pic, CPelUnitBuf cPicD, AccessUnitList&, bool printFrameMSE, double* PSNR_Y, bool isEncodeLtRef );

  uint64_t xFindDistortionPlane       ( const CPelBuf& pic0, const CPelBuf& pic1, uint32_t rshift ) const;

  void xPrintPictureInfo              ( const Picture& pic, AccessUnitList& accessUnit, const std::string& digestStr, bool printFrameMSE, bool isEncodeLtRef );

  inline bool xEncodersFinished       () { return ( int ) m_freePicEncoderList.size() >= std::max(1, m_pcEncCfg->m_maxParallelFrames); }

  inline bool xLockStepPicsFinished   ()

  {

    std::lock_guard<std::mutex> lock( m_gopEncMutex );

    return ( int ) m_freePicEncoderList.size() >= std::max(1, m_pcEncCfg->m_maxParallelFrames); 

  }

  void xForceScc                      ( Picture& pic );

};// END CLASS DEFINITION EncGOP

} // namespace vvenc

//! \}