text: /* 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-2019, 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     IntraPrediction.h
      \brief    prediction class (header)
  */
  
  #ifndef __INTRAPREDICTION__
  #define __INTRAPREDICTION__
  
  
  // Include files
  #include "Unit.h"
  #include "Buffer.h"
  #include "Picture.h"
  
  #if JVET_N0217_MATRIX_INTRAPRED
  #include "MatrixIntraPrediction.h"
  #endif
  
  //! \ingroup CommonLib
  //! \{
  
  // ====================================================================================================================
  // Class definition
  // ====================================================================================================================
  
  /// prediction class
  enum PredBuf
  {
    PRED_BUF_UNFILTERED = 0,
    PRED_BUF_FILTERED   = 1,
    NUM_PRED_BUF        = 2
  };
  
  static const uint32_t MAX_INTRA_FILTER_DEPTHS=8;
  
  class IntraPrediction
  {
  private:
  
    Pel* m_piYuvExt[MAX_NUM_COMPONENT][NUM_PRED_BUF];
    int  m_iYuvExtSize;
  
    Pel* m_yuvExt2[MAX_NUM_COMPONENT][4];
    int  m_yuvExtSize2;
  
    static const uint8_t m_aucIntraFilter[MAX_NUM_CHANNEL_TYPE][MAX_INTRA_FILTER_DEPTHS];
  
    struct IntraPredParam //parameters of Intra Prediction
    {
      bool refFilterFlag;
      bool applyPDPC;
      bool isModeVer;
      int  multiRefIndex;
      int  whRatio;
      int  hwRatio;
      int  intraPredAngle;
      int  invAngle;
      bool interpolationFlag;
  
      IntraPredParam() :
        refFilterFlag     ( false                           ),
        applyPDPC         ( false                           ),
        isModeVer         ( false                           ),
        multiRefIndex     ( -1                              ),
        whRatio           ( 0                               ),
        hwRatio           ( 0                               ),
        intraPredAngle    ( std::numeric_limits<int>::max() ),
        invAngle          ( std::numeric_limits<int>::max() ),
        interpolationFlag ( false                           ) {}
    };
  
    IntraPredParam m_ipaParam;
  
    Pel* m_piTemp;
    Pel* m_pMdlmTemp; // for MDLM mode
  #if JVET_N0217_MATRIX_INTRAPRED
    MatrixIntraPrediction m_matrixIntraPred;
  #endif
  
  protected:
  
    ChromaFormat  m_currChromaFormat;
  
    int m_topRefLength;
    int m_leftRefLength;
    // prediction
    void xPredIntraPlanar           ( const CPelBuf &pSrc, PelBuf &pDst );
    void xPredIntraDc               ( const CPelBuf &pSrc, PelBuf &pDst, const ChannelType channelType, const bool enableBoundaryFilter = true );
    void xPredIntraAng              ( const CPelBuf &pSrc, PelBuf &pDst, const ChannelType channelType, const ClpRng& clpRng);
  
    void initPredIntraParams        ( const PredictionUnit & pu,  const CompArea compArea, const SPS& sps );
  
  #if JVET_N0435_WAIP_HARMONIZATION
    static bool isIntegerSlope(const int absAng) { return (0 == (absAng & 0x1F)); }
  #else
    static bool isIntegerSlope      ( const int absAng ) { return (0 == (absAng & 0x1F)) && absAng <=32; }  //  integer-slope modes 2, DIA_IDX and VDIA_IDX.  "absAng <=32" restricts wide-angle integer modes
  #endif
  
  #if JVET_N0413_RDPCM
    void xPredIntraBDPCM            ( const CPelBuf &pSrc, PelBuf &pDst, const uint32_t dirMode, const ClpRng& clpRng );
  #endif
    Pel  xGetPredValDc              ( const CPelBuf &pSrc, const Size &dstSize );
  
    void xFillReferenceSamples      ( const CPelBuf &recoBuf,      Pel* refBufUnfiltered, const CompArea &area, const CodingUnit &cu );
    void xFilterReferenceSamples    ( const Pel* refBufUnfiltered, Pel* refBufFiltered, const CompArea &area, const SPS &sps
      , int multiRefIdx
    );
  
    static int getWideAngle         ( int width, int height, int predMode );
    void setReferenceArrayLengths   ( const CompArea &area );
  
    void destroy                    ();
  
    void xGetLMParameters(const PredictionUnit &pu, const ComponentID compID, const CompArea& chromaArea, int& a, int& b, int& iShift);
  public:
    IntraPrediction();
    virtual ~IntraPrediction();
  
    void init                       (ChromaFormat chromaFormatIDC, const unsigned bitDepthY);
  
    // Angular Intra
    void predIntraAng               ( const ComponentID compId, PelBuf &piPred, const PredictionUnit &pu);
    Pel* getPredictorPtr            ( const ComponentID compId ) { return m_piYuvExt[compId][m_ipaParam.refFilterFlag ? PRED_BUF_FILTERED : PRED_BUF_UNFILTERED]; }
  
    // Cross-component Chroma
    void predIntraChromaLM(const ComponentID compID, PelBuf &piPred, const PredictionUnit &pu, const CompArea& chromaArea, int intraDir);
    void xGetLumaRecPixels(const PredictionUnit &pu, CompArea chromaArea);
    /// set parameters from CU data for accessing intra data
    void initIntraPatternChType     (const CodingUnit &cu, const CompArea &area, const bool forceRefFilterFlag = false); // use forceRefFilterFlag to get both filtered and unfiltered buffers
  
  #if JVET_N0217_MATRIX_INTRAPRED
    // Matrix-based intra prediction
    void initIntraMip               (const PredictionUnit &pu);
    void predIntraMip               (const ComponentID compId, PelBuf &piPred, const PredictionUnit &pu);
  #endif
  
    static bool useDPCMForFirstPassIntraEstimation(const PredictionUnit &pu, const uint32_t &uiDirMode);
  
    void geneWeightedPred           (const ComponentID compId, PelBuf &pred, const PredictionUnit &pu, Pel *srcBuf);
    Pel* getPredictorPtr2           (const ComponentID compID, uint32_t idx) { return m_yuvExt2[compID][idx]; }
    void switchBuffer               (const PredictionUnit &pu, ComponentID compID, PelBuf srcBuff, Pel *dst);
    void geneIntrainterPred         (const CodingUnit &cu);
  };
  
  //! \}
  
  #endif // __INTRAPREDICTION__
  