/* SPDX-License-Identifier: LGPL-2.1 */
/*
 * Copyright 2018 Cisco Systems, Inc. and/or its affiliates. All rights reserved.
 */

#ifndef CODEC_V4L2_FWHT_H
#define CODEC_V4L2_FWHT_H

#include "codec-fwht.h"

struct v4l2_fwht_pixfmt_info {
 u32 id;
 unsigned int bytesperline_mult;
 unsigned int sizeimage_mult;
 unsigned int sizeimage_div;
 unsigned int luma_alpha_step;
 unsigned int chroma_step;
 /* Chroma plane subsampling */
 unsigned int width_div;
 unsigned int height_div;
 unsigned int components_num;
 unsigned int planes_num;
 unsigned int pixenc;
};

struct v4l2_fwht_state {
 const struct v4l2_fwht_pixfmt_info *info;
 unsigned int visible_width;
 unsigned int visible_height;
 unsigned int coded_width;
 unsigned int coded_height;
 unsigned int stride;
 unsigned int ref_stride;
 unsigned int gop_size;
 unsigned int gop_cnt;
 u16 i_frame_qp;
 u16 p_frame_qp;

 enum v4l2_colorspace colorspace;
 enum v4l2_ycbcr_encoding ycbcr_enc;
 enum v4l2_xfer_func xfer_func;
 enum v4l2_quantization quantization;

 struct fwht_raw_frame ref_frame;
 struct fwht_cframe_hdr header;
 u8 *compressed_frame;
 u64 ref_frame_ts;
};

const struct v4l2_fwht_pixfmt_info *v4l2_fwht_find_pixfmt(u32 pixelformat);
const struct v4l2_fwht_pixfmt_info *v4l2_fwht_get_pixfmt(u32 idx);
bool v4l2_fwht_validate_fmt(const struct v4l2_fwht_pixfmt_info *info,
       u32 width_div, u32 height_div, u32 components_num,
       u32 pixenc);
const struct v4l2_fwht_pixfmt_info *v4l2_fwht_find_nth_fmt(u32 width_div,
         u32 height_div,
         u32 components_num,
         u32 pixenc,
         unsigned int start_idx);

int v4l2_fwht_encode(struct v4l2_fwht_state *state, u8 *p_in, u8 *p_out);
int v4l2_fwht_decode(struct v4l2_fwht_state *state, u8 *p_in, u8 *p_out);

#endif