/*
 *  gstvaapifeienc_h264.h - FEI Enc abstract
 *
 *  Copyright (C) 2016-2018 Intel Corporation
 *    Author: Leilei Shang <leilei.shang@intel.com>
 *    Author: Sreerenj Balachandran <sreerenj.balachandran@intel.com>
 *
 *  This library is free software; you can redistribute it and/or
 *  modify it under the terms of the GNU Lesser General Public License
 *  as published by the Free Software Foundation; either version 2.1
 *  of the License, or (at your option) any later version.
 *
 *  This library is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 *  Lesser General Public License for more details.
 *
 *  You should have received a copy of the GNU Lesser General Public
 *  License along with this library; if not, write to the Free
 *  Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
 *  Boston, MA 02110-1301 USA
 */

#ifndef GST_VAAPI_FEI_H264_ENC_H
#define GST_VAAPI_FEI_H264_ENC_H

#include <gst/vaapi/gstvaapiencoder.h>
#include <gst/vaapi/gstvaapiutils_h264.h>
#include <gst/vaapi/gstvaapifeiutils_h264.h>
#include <gst/vaapi/gstvaapifei_objects_priv.h>
G_BEGIN_DECLS

#define GST_TYPE_VAAPI_FEI_ENC_H264 \
    (gst_vaapi_feienc_h264_get_type ())
#define GST_VAAPI_FEI_ENC_H264(encoder) \
    (G_TYPE_CHECK_INSTANCE_CAST ((encoder), GST_TYPE_VAAPI_FEI_ENC_H264, GstVaapiFeiEncH264))
#define GST_IS_VAAPI_FEI_ENC_H264(encoder) \
    (G_TYPE_CHECK_INSTANCE_TYPE ((encoder), GST_TYPE_VAAPI_FEI_ENC_H264))

typedef struct _GstVaapiFeiEncH264 GstVaapiFeiEncH264;
typedef struct _GstVaapiFeiEncH264Class GstVaapiFeiEncH264Class;

/**
 * GstVaapiFeiEncH264Prop:
 * @GST_VAAPI_FEI_H264_ENC_PROP_MAX_BFRAMES: Number of B-frames between I
 *   and P (uint).
 * @GST_VAAPI_FEI_H264_ENC_PROP_INIT_QP: Initial quantizer value (uint).
 * @GST_VAAPI_FEI_H264_ENC_PROP_MIN_QP: Minimal quantizer value (uint).
 * @GST_VAAPI_FEI_H264_ENC_PROP_NUM_SLICES: Number of slices per frame (uint).
 * @GST_VAAPI_FEI_H264_ENC_PROP_CABAC: Enable CABAC entropy coding mode (bool).
 * @GST_VAAPI_FEI_H264_ENC_PROP_DCT8X8: Enable adaptive use of 8x8
 *   transforms in I-frames (bool).
 * @GST_VAAPI_FEI_H264_ENC_PROP_CPB_LENGTH: Length of the CPB buffer
 *   in milliseconds (uint).
 * @GST_VAAPI_FEI_H264_ENC_PROP_NUM_VIEWS: Number of views per frame.
 * @GST_VAAPI_FEI_H264_ENC_PROP_VIEW_IDS: View IDs
 *
 * The set of FEI Enc specific configurable properties.
 */
typedef enum
{
  GST_VAAPI_FEI_H264_ENC_PROP_MAX_BFRAMES = -1,
  GST_VAAPI_FEI_H264_ENC_PROP_INIT_QP = -2,
  GST_VAAPI_FEI_H264_ENC_PROP_MIN_QP = -3,
  GST_VAAPI_FEI_H264_ENC_PROP_NUM_SLICES = -4,
  GST_VAAPI_FEI_H264_ENC_PROP_CABAC = -5,
  GST_VAAPI_FEI_H264_ENC_PROP_DCT8X8 = -6,
  GST_VAAPI_FEI_H264_ENC_PROP_CPB_LENGTH = -7,
  GST_VAAPI_FEI_H264_ENC_PROP_NUM_VIEWS = -8,
  GST_VAAPI_FEI_H264_ENC_PROP_VIEW_IDS = -9,
  GST_VAAPI_FEI_H264_ENC_PROP_NUM_REF = -10,
  GST_VAAPI_FEI_H264_ENC_PROP_FEI_ENABLE = -11,
  GST_VAAPI_FEI_H264_ENC_PROP_NUM_MV_PREDICT_L0 = -12,
  GST_VAAPI_FEI_H264_ENC_PROP_NUM_MV_PREDICT_L1 = -13,
  GST_VAAPI_FEI_H264_ENC_PROP_SEARCH_WINDOW = -14,
  GST_VAAPI_FEI_H264_ENC_PROP_LEN_SP = -15,
  GST_VAAPI_FEI_H264_ENC_PROP_SEARCH_PATH = -16,
  GST_VAAPI_FEI_H264_ENC_PROP_REF_WIDTH = -17,
  GST_VAAPI_FEI_H264_ENC_PROP_REF_HEIGHT = -18,
  GST_VAAPI_FEI_H264_ENC_PROP_SUBMB_MASK = -19,
  GST_VAAPI_FEI_H264_ENC_PROP_SUBPEL_MODE = -20,
  GST_VAAPI_FEI_H264_ENC_PROP_INTRA_PART_MASK = -21,
  GST_VAAPI_FEI_H264_ENC_PROP_INTRA_SAD = -22,
  GST_VAAPI_FEI_H264_ENC_PROP_INTER_SAD = -23,
  GST_VAAPI_FEI_H264_ENC_PROP_ADAPT_SEARCH = -24,
  GST_VAAPI_FEI_H264_ENC_PROP_MULTI_PRED_L0 = -25,
  GST_VAAPI_FEI_H264_ENC_PROP_MULTI_PRED_L1 = -26,
  GST_VAAPI_FEI_H264_ENC_PROP_ENABLE_STATS_OUT = -27,
} GstVaapiFeiEncH264Prop;

GType
gst_vaapi_feienc_h264_get_type (void) G_GNUC_CONST;

GstVaapiEncoder *
gst_vaapi_feienc_h264_new (GstVaapiDisplay * display);

GPtrArray *
gst_vaapi_feienc_h264_get_default_properties (void);

gboolean
gst_vaapi_feienc_h264_set_max_profile (GstVaapiFeiEncH264 * feienc,
    GstVaapiProfile profile);

gboolean
gst_vaapi_feienc_h264_set_ref_pool (GstVaapiFeiEncH264 * feienc, gpointer ref_pool_ptr);

GstVaapiEncoderStatus
gst_vaapi_feienc_h264_encode (GstVaapiEncoder * base_encoder,
    GstVaapiEncPicture * picture, GstVaapiSurfaceProxy * reconstruct,
    GstVaapiCodedBufferProxy * codedbuf_proxy, GstVaapiFeiInfoToPakH264 *info_to_pak);

GstVaapiEncoderStatus
gst_vaapi_feienc_h264_flush (GstVaapiEncoder * base_encoder);

GstVaapiEncoderStatus
gst_vaapi_feienc_h264_reordering (GstVaapiEncoder * base_encoder,
    GstVideoCodecFrame * frame, GstVaapiEncPicture ** output);

GstVaapiEncoderStatus
gst_vaapi_feienc_h264_reconfigure (GstVaapiEncoder * base_encoder);

GstVaapiEncoderStatus
gst_vaapi_feienc_h264_set_property (GstVaapiEncoder * base_encoder,
    gint prop_id, const GValue * value);

gboolean
gst_vaapi_feienc_h264_get_profile_and_idc (GstVaapiFeiEncH264 * feienc,
    GstVaapiProfile * out_profile_ptr, guint8 * out_profile_idc_ptr);

G_END_DECLS
#endif /*GST_VAAPI_FEI_H264_ENC_H */