V1.0.6

packages/artinchip/mpp/ve/encoder/jpeg/jpeg_enc_ctx.h

@@ -0,0 +1,145 @@
+/*
+ * Copyright (C) 2020-2024 ArtInChip Technology Co. Ltd
+ *
+ * SPDX-License-Identifier: Apache-2.0
+ *
+ * Author: <qi.xu@artinchip.com>
+ * Desc: jpeg enc context
+ */
+
+#ifndef JPEG_ENC_CTX_H
+#define JPEG_ENC_CTX_H
+
+#include <unistd.h>
+
+#include "put_bits.h"
+#include "jpeg_tables.h"
+#include "mpp_codec.h"
+
+#define QUANT_FIXED_POINT_BITS 19
+#define ALIGN_4K(x) (((x) + 4095) & ~(4095))
+
+/* JPEG marker codes */
+enum JpegMarker {
+	/* start of frame */
+	SOF0 = 0xc0,       /* baseline */
+	SOF1 = 0xc1,       /* extended sequential, huffman */
+	SOF2 = 0xc2,       /* progressive, huffman */
+	SOF3 = 0xc3,       /* lossless, huffman */
+
+	SOF5 = 0xc5,       /* differential sequential, huffman */
+	SOF6 = 0xc6,       /* differential progressive, huffman */
+	SOF7 = 0xc7,       /* differential lossless, huffman */
+	JPG = 0xc8,       /* reserved for JPEG extension */
+	SOF9 = 0xc9,       /* extended sequential, arithmetic */
+	SOF10 = 0xca,       /* progressive, arithmetic */
+	SOF11 = 0xcb,       /* lossless, arithmetic */
+
+	SOF13 = 0xcd,       /* differential sequential, arithmetic */
+	SOF14 = 0xce,       /* differential progressive, arithmetic */
+	SOF15 = 0xcf,       /* differential lossless, arithmetic */
+
+	DHT = 0xc4,       /* define huffman tables */
+
+	DAC = 0xcc,       /* define arithmetic-coding conditioning */
+
+	/* restart with modulo 8 count "m" */
+	RST0 = 0xd0,
+	RST1 = 0xd1,
+	RST2 = 0xd2,
+	RST3 = 0xd3,
+	RST4 = 0xd4,
+	RST5 = 0xd5,
+	RST6 = 0xd6,
+	RST7 = 0xd7,
+
+	SOI = 0xd8,       /* start of image */
+	EOI = 0xd9,       /* end of image */
+	SOS = 0xda,       /* start of scan */
+	DQT = 0xdb,       /* define quantization tables */
+	DNL = 0xdc,       /* define number of lines */
+	DRI = 0xdd,       /* define restart interval */
+	DHP = 0xde,       /* define hierarchical progression */
+	EXP = 0xdf,       /* expand reference components */
+
+	APP0 = 0xe0,
+	APP1 = 0xe1,
+	APP2 = 0xe2,
+	APP3 = 0xe3,
+	APP4 = 0xe4,
+	APP5 = 0xe5,
+	APP6 = 0xe6,
+	APP7 = 0xe7,
+	APP8 = 0xe8,
+	APP9 = 0xe9,
+	APP10 = 0xea,
+	APP11 = 0xeb,
+	APP12 = 0xec,
+	APP13 = 0xed,
+	APP14 = 0xee,
+	APP15 = 0xef,
+
+	JPG0 = 0xf0,
+	JPG1 = 0xf1,
+	JPG2 = 0xf2,
+	JPG3 = 0xf3,
+	JPG4 = 0xf4,
+	JPG5 = 0xf5,
+	JPG6 = 0xf6,
+	SOF48 = 0xf7,       ///< JPEG-LS
+	LSE = 0xf8,       ///< JPEG-LS extension parameters
+	JPG9 = 0xf9,
+	JPG10 = 0xfa,
+	JPG11 = 0xfb,
+	JPG12 = 0xfc,
+	JPG13 = 0xfd,
+
+	COM = 0xfe,   /* comment */
+
+	TEM = 0x01,  /* temporary private use for arithmetic coding */
+				/* 0x02 -> 0xbf reserved */
+};
+
+struct jpeg_ctx {
+	struct mpp_encoder encoder;
+	unsigned int luma_quant_table[64];
+	unsigned int chroma_quant_table[64];
+	uint8_t huff_size_dc_luminance[12];     ///< DC luminance Huffman table size.
+	uint16_t huff_code_dc_luminance[12];    ///< DC luminance Huffman table codes.
+	uint8_t huff_size_dc_chrominance[12];   ///< DC chrominance Huffman table size.
+	uint16_t huff_code_dc_chrominance[12];  ///< DC chrominance Huffman table codes.
+	uint8_t huff_size_ac_luminance[256];    ///< AC luminance Huffman table size.
+	uint16_t huff_code_ac_luminance[256];   ///< AC luminance Huffman table codes.
+	uint8_t huff_size_ac_chrominance[256];  ///< AC chrominance Huffman table size.
+	uint16_t huff_code_ac_chrominance[256]; ///< AC chrominance Huffman table codes.
+
+	int uv_interleave;
+	int h_count[4];
+	int v_count[4];
+	int comp_num;
+	int width;
+	int height;
+	int y_stride;
+	int quality;
+	int header_offset;
+
+	struct mpp_frame* cur_frame;
+
+	unsigned int phy_addr[3];
+
+	int encode_data_len;
+
+	int ve_fd;
+	unsigned long regs_base;
+	struct ve_buffer_allocator *alloc;
+
+	unsigned char* bitstream_vir_addr;
+	unsigned int bitstream_phy_addr;
+	int stream_num;
+	struct put_bit_ctx pb;
+	struct encode_config config;
+};
+
+int jpeg_hw_encode(struct jpeg_ctx *s);
+
+#endif