media_driver/agnostic/common/codec/hal/codechal_kernel_intra_dist.h

/*
* Copyright (c) 2017, Intel Corporation
*
* Permission is hereby granted, free of charge, to any person obtaining a
* copy of this software and associated documentation files (the "Software"),
* to deal in the Software without restriction, including without limitation
* the rights to use, copy, modify, merge, publish, distribute, sublicense,
* and/or sell copies of the Software, and to permit persons to whom the
* Software is furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included
* in all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
* OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
* THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR
* OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
* ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
* OTHER DEALINGS IN THE SOFTWARE.
*/
//!
//! \file     codechal_kernel_intra_dist.h
//! \brief    Defines the intra distortion kernel base
//! \details  Intra distortion base includes all common functions and definitions for intra distortion
//!
#ifndef __CODECHAL_KERNEL_INTRA_DIST_H__
#define __CODECHAL_KERNEL_INTRA_DIST_H__
#include "codechal_kernel_base.h"

//!
//! \class    CodechalKernelIntraDist
//! \brief    Codechal kernel intra dist
//!
class CodechalKernelIntraDist : public CodechalKernelBase
{
public:
    //!
    //! \enum     KernelIndex
    //! \brief    Kernel index
    //!
    enum KernelIndex
    {
        intraDistortion = 0
    };

    //!
    //! \enum     BindingTableOffset
    //! \brief    Binding table offset
    //!
    enum BindingTableOffset
    {
        intraDistCurrent4xY   = 0,
        intraDistOutputSurf   = 1,
        intraDistVmeIntraPred = 2,
        intraDistSurfaceNum   = 3
    };

    //!
    //! \class    CurbeParam
    //! \brief    Curbe parameter
    //!
    struct CurbeParam
    {
        uint32_t downScaledWidthInMb4x = 0;
        uint32_t downScaledHeightInMb4x = 0;
    };

    //!
    //! \class    SurfaceParams
    //! \brief    Surface parameters
    //!
    struct SurfaceParams
    {
        uint32_t     intraDistBottomFieldOffset = 0;
        PMOS_SURFACE input4xDsSurface = nullptr;
        PMOS_SURFACE intraDistSurface = nullptr;
        PMOS_SURFACE input4xDsVmeSurface = nullptr;
    };

    // clang-format off
    //!
    //! \class    Curbe
    //! \brief    Curbe
    //!
    class Curbe
    {
    public:
        Curbe()
        {
            memset(&m_data, 0, sizeof(CurbeData));
            m_data.DW1.interSAD = 2;
            m_data.DW1.intraSAD = 2;
            m_data.DW8.value    = 0xFFFFFFFF;
            m_data.DW9.value    = 0xFFFFFFFF;
            m_data.DW10.value   = 0xFFFFFFFF;
        }
        //!
        //! \struct    CurbeData
        //! \brief     Curbe data
        //!
        struct CurbeData
        {
            //DW0
            union
            {
                struct
                {
                    uint32_t picWidthInLumaSamples  : MOS_BITFIELD_RANGE(0, 15);
                    uint32_t picHeightInLumaSamples : MOS_BITFIELD_RANGE(16, 31);
                };
                uint32_t value;
            } DW0;

            //DW1
            union
            {
                struct
                {
                    uint32_t srcSize              : MOS_BITFIELD_RANGE(0, 1);
                    uint32_t reserved2            : MOS_BITFIELD_RANGE(2, 13);
                    uint32_t skipType             : MOS_BITFIELD_BIT(14);
                    uint32_t reserved15           : MOS_BITFIELD_BIT(15);
                    uint32_t intraChromaMode      : MOS_BITFIELD_BIT(16);
                    uint32_t ftSkipCheckEnable    : MOS_BITFIELD_BIT(17);
                    uint32_t reserved18           : MOS_BITFIELD_BIT(18);
                    uint32_t blockBasedSkipEnable : MOS_BITFIELD_BIT(19);
                    uint32_t interSAD             : MOS_BITFIELD_RANGE(20, 21);
                    uint32_t intraSAD             : MOS_BITFIELD_RANGE(22, 23);
                    uint32_t reserved24           : MOS_BITFIELD_RANGE(24, 31);
                };
                uint32_t value;
            } DW1;

            //DW2
            union
            {
                struct
                {
                    uint32_t intraPartMask     : MOS_BITFIELD_RANGE(0, 4);
                    uint32_t nonSkipZmvAdded   : MOS_BITFIELD_BIT(5);
                    uint32_t nonSkipModeAdded  : MOS_BITFIELD_BIT(6);
                    uint32_t intraCornerSwap   : MOS_BITFIELD_BIT(7);
                    uint32_t reserved8         : MOS_BITFIELD_RANGE(8, 15);
                    uint32_t mvCostScaleFactor : MOS_BITFIELD_RANGE(16, 17);
                    uint32_t bilinearEnable    : MOS_BITFIELD_BIT(18);
                    uint32_t reserved19        : MOS_BITFIELD_BIT(19);
                    uint32_t weightedSADHaar   : MOS_BITFIELD_BIT(20);
                    uint32_t acOnlyHaar        : MOS_BITFIELD_BIT(21);
                    uint32_t refIDCostMode     : MOS_BITFIELD_BIT(22);
                    uint32_t reserved23        : MOS_BITFIELD_BIT(23);
                    uint32_t skipCenterMask    : MOS_BITFIELD_RANGE(24, 31);
                };
                uint32_t value;
            } DW2;

            //DW3
            union
            {
                struct
                {
                    uint32_t reserved;
                };
                uint32_t value;
            } DW3;

            //DW4
            union
            {
                struct
                {
                    uint32_t reserved;
                };
                uint32_t value;
            } DW4;

            //DW5
            union
            {
                struct
                {
                    uint32_t reserved;
                };
                uint32_t value;
            } DW5;

            //DW6
            union
            {
                struct
                {
                    uint32_t reserved;
                };
                uint32_t value;
            } DW6;

            //DW7
            union
            {
                struct
                {
                    uint32_t reserved;
                };
                uint32_t value;
            } DW7;

            //DW8
            union
            {
                struct
                {
                    uint32_t currPic4xBTI;
                };
                uint32_t value;
            } DW8;

            //DW9
            union
            {
                struct
                {
                    uint32_t intraDistSurfaceBTI;
                };
                uint32_t value;
            } DW9;

            //DW10
            union
            {
                struct
                {
                    uint32_t vmeIntraPredSurfaceBTI;
                };
                uint32_t value;
            } DW10;
        } m_data;
        static const uint32_t m_curbeSize = sizeof(CurbeData);
    };
    // clang-format on

    CodechalKernelIntraDist(CodechalEncoderState *encoder);

    virtual ~CodechalKernelIntraDist();

    //!
    //! \brief  Execute Intra Distortion kernel
    //!
    //! \param  [in] curbeParam
    //!         Reference to CodechalKernelIntraDist::CurbeParam
    //! \param  [in] surfaceParam
    //!         Reference to CodechalKernelIntraDist::SurfaceParams
    //!
    //! \return MOS_STATUS
    //!         MOS_STATUS_SUCCESS if success
    //!
    MOS_STATUS Execute(CurbeParam &curbeParam, SurfaceParams &surfaceParam);

    uint32_t GetBTCount() { return BindingTableOffset::intraDistSurfaceNum; }

protected:
    MOS_STATUS AllocateResources();
    MOS_STATUS ReleaseResources();

    virtual uint32_t   GetCurbeSize() { return Curbe::m_curbeSize; }
    virtual MOS_STATUS SetCurbe(MHW_KERNEL_STATE *kernelState);
    virtual MOS_STATUS SendSurfaces(PMOS_COMMAND_BUFFER cmd, MHW_KERNEL_STATE *kernelState);
    virtual MOS_STATUS AddPerfTag();
    virtual MHW_KERNEL_STATE *GetActiveKernelState();
    virtual MOS_STATUS InitWalkerCodecParams(CODECHAL_WALKER_CODEC_PARAMS &walkerParam);
    virtual CODECHAL_MEDIA_STATE_TYPE GetMediaStateType();

    CurbeParam    m_curbeParam;    //!< curbe paramters
    SurfaceParams m_surfaceParam;  //! surface parameters
};

#endif /* __CODECHAL_KERNEL_INTRA_DIST_H__ */