brintos

brintos / linux-shallow public Read only

0
0
Text · 43.0 KiB · 6289166 Raw
1335 lines · c
1// SPDX-License-Identifier: GPL-2.0-only2/*3 * Copyright (c) 2020, The Linux Foundation. All rights reserved.4 */5#include <linux/kernel.h>6#include <linux/sizes.h>7#include <linux/videodev2.h>8 9#include "hfi.h"10#include "hfi_plat_bufs.h"11#include "helpers.h"12 13#define MIN_INPUT_BUFFERS				414#define MIN_ENC_OUTPUT_BUFFERS				415 16#define NV12_UBWC_Y_TILE_WIDTH				3217#define NV12_UBWC_Y_TILE_HEIGHT				818#define NV12_UBWC_UV_TILE_WIDTH				1619#define NV12_UBWC_UV_TILE_HEIGHT			820#define TP10_UBWC_Y_TILE_WIDTH				4821#define TP10_UBWC_Y_TILE_HEIGHT				422#define METADATA_STRIDE_MULTIPLE			6423#define METADATA_HEIGHT_MULTIPLE			1624#define HFI_DMA_ALIGNMENT				25625 26#define MAX_FE_NBR_CTRL_LCU64_LINE_BUFFER_SIZE		6427#define MAX_FE_NBR_CTRL_LCU32_LINE_BUFFER_SIZE		6428#define MAX_FE_NBR_CTRL_LCU16_LINE_BUFFER_SIZE		6429#define MAX_FE_NBR_DATA_LUMA_LINE_BUFFER_SIZE		64030#define MAX_FE_NBR_DATA_CB_LINE_BUFFER_SIZE		32031#define MAX_FE_NBR_DATA_CR_LINE_BUFFER_SIZE		32032 33#define MAX_SE_NBR_CTRL_LCU64_LINE_BUFFER_SIZE		(128 / 8)34#define MAX_SE_NBR_CTRL_LCU32_LINE_BUFFER_SIZE		(128 / 8)35#define MAX_SE_NBR_CTRL_LCU16_LINE_BUFFER_SIZE		(128 / 8)36 37#define MAX_PE_NBR_DATA_LCU64_LINE_BUFFER_SIZE		(64 * 2 * 3)38#define MAX_PE_NBR_DATA_LCU32_LINE_BUFFER_SIZE		(32 * 2 * 3)39#define MAX_PE_NBR_DATA_LCU16_LINE_BUFFER_SIZE		(16 * 2 * 3)40 41#define MAX_TILE_COLUMNS				32 /* 8K/256 */42 43#define VPP_CMD_MAX_SIZE				BIT(20)44#define NUM_HW_PIC_BUF					3245#define BIN_BUFFER_THRESHOLD				(1280 * 736)46#define H264D_MAX_SLICE					180047/* sizeof(h264d_buftab_t) aligned to 256 */48#define SIZE_H264D_BUFTAB_T				25649/* sizeof(h264d_hw_pic_t) aligned to 32 */50#define SIZE_H264D_HW_PIC_T				BIT(11)51#define SIZE_H264D_BSE_CMD_PER_BUF			(32 * 4)52#define SIZE_H264D_VPP_CMD_PER_BUF			51253 54/* Line Buffer definitions, One for Luma and 1/2 for each Chroma */55#define SIZE_H264D_LB_FE_TOP_DATA(width, height)	\56	(MAX_FE_NBR_DATA_LUMA_LINE_BUFFER_SIZE * ALIGN((width), 16) * 3)57 58#define SIZE_H264D_LB_FE_TOP_CTRL(width, height)	\59	(MAX_FE_NBR_CTRL_LCU64_LINE_BUFFER_SIZE * (((width) + 15) >> 4))60 61#define SIZE_H264D_LB_FE_LEFT_CTRL(width, height)	\62	(MAX_FE_NBR_CTRL_LCU64_LINE_BUFFER_SIZE * (((height) + 15) >> 4))63 64#define SIZE_H264D_LB_SE_TOP_CTRL(width, height)	\65	(MAX_SE_NBR_CTRL_LCU64_LINE_BUFFER_SIZE * (((width) + 15) >> 4))66 67#define SIZE_H264D_LB_SE_LEFT_CTRL(width, height)	\68	(MAX_SE_NBR_CTRL_LCU64_LINE_BUFFER_SIZE * (((height) + 15) >> 4))69 70#define SIZE_H264D_LB_PE_TOP_DATA(width, height)	\71	(MAX_PE_NBR_DATA_LCU64_LINE_BUFFER_SIZE * (((width) + 15) >> 4))72 73#define SIZE_H264D_LB_VSP_TOP(width, height)	(((((width) + 15) >> 4) << 7))74 75#define SIZE_H264D_LB_RECON_DMA_METADATA_WR(width, height)	\76	(ALIGN((height), 16) * 32)77 78#define SIZE_H264D_QP(width, height)	\79	((((width) + 63) >> 6) * (((height) + 63) >> 6) * 128)80 81#define SIZE_HW_PIC(size_per_buf)	(NUM_HW_PIC_BUF * (size_per_buf))82 83#define H264_CABAC_HDR_RATIO_HD_TOT	184#define H264_CABAC_RES_RATIO_HD_TOT	385 86/*87 * Some content need more bin buffer, but limit buffer88 * size for high resolution89 */90#define NUM_SLIST_BUF_H264		(256 + 32)91#define SIZE_SLIST_BUF_H264		51292#define LCU_MAX_SIZE_PELS		6493#define LCU_MIN_SIZE_PELS		1694#define SIZE_SEI_USERDATA		409695 96#define H265D_MAX_SLICE			360097#define SIZE_H265D_HW_PIC_T		SIZE_H264D_HW_PIC_T98#define SIZE_H265D_BSE_CMD_PER_BUF	(16 * sizeof(u32))99#define SIZE_H265D_VPP_CMD_PER_BUF	256100 101#define SIZE_H265D_LB_FE_TOP_DATA(width, height)	\102	(MAX_FE_NBR_DATA_LUMA_LINE_BUFFER_SIZE * (ALIGN(width, 64) + 8) * 2)103 104#define SIZE_H265D_LB_FE_TOP_CTRL(width, height)	\105	(MAX_FE_NBR_CTRL_LCU64_LINE_BUFFER_SIZE *	\106	(ALIGN(width, LCU_MAX_SIZE_PELS) / LCU_MIN_SIZE_PELS))107 108#define SIZE_H265D_LB_FE_LEFT_CTRL(width, height)	\109	(MAX_FE_NBR_CTRL_LCU64_LINE_BUFFER_SIZE *	\110	(ALIGN(height, LCU_MAX_SIZE_PELS) / LCU_MIN_SIZE_PELS))111 112#define SIZE_H265D_LB_SE_TOP_CTRL(width, height)	\113	((LCU_MAX_SIZE_PELS / 8 * (128 / 8)) * (((width) + 15) >> 4))114 115static inline u32 size_h265d_lb_se_left_ctrl(u32 width, u32 height)116{117	u32 x, y, z;118 119	x = ((height + 16 - 1) / 8) * MAX_SE_NBR_CTRL_LCU16_LINE_BUFFER_SIZE;120	y = ((height + 32 - 1) / 8) * MAX_SE_NBR_CTRL_LCU32_LINE_BUFFER_SIZE;121	z = ((height + 64 - 1) / 8) * MAX_SE_NBR_CTRL_LCU64_LINE_BUFFER_SIZE;122 123	return max3(x, y, z);124}125 126#define SIZE_H265D_LB_PE_TOP_DATA(width, height)	\127	(MAX_PE_NBR_DATA_LCU64_LINE_BUFFER_SIZE *	\128	(ALIGN(width, LCU_MIN_SIZE_PELS) / LCU_MIN_SIZE_PELS))129 130#define SIZE_H265D_LB_VSP_TOP(width, height)	((((width) + 63) >> 6) * 128)131 132#define SIZE_H265D_LB_VSP_LEFT(width, height)	((((height) + 63) >> 6) * 128)133 134#define SIZE_H265D_LB_RECON_DMA_METADATA_WR(width, height)	\135	SIZE_H264D_LB_RECON_DMA_METADATA_WR(width, height)136 137#define SIZE_H265D_QP(width, height)	SIZE_H264D_QP(width, height)138 139#define H265_CABAC_HDR_RATIO_HD_TOT	2140#define H265_CABAC_RES_RATIO_HD_TOT	2141 142/*143 * Some content need more bin buffer, but limit buffer size144 * for high resolution145 */146#define SIZE_SLIST_BUF_H265	BIT(10)147#define NUM_SLIST_BUF_H265	(80 + 20)148#define H265_NUM_TILE_COL	32149#define H265_NUM_TILE_ROW	128150#define H265_NUM_TILE		(H265_NUM_TILE_ROW * H265_NUM_TILE_COL + 1)151 152static inline u32 size_vpxd_lb_fe_left_ctrl(u32 width, u32 height)153{154	u32 x, y, z;155 156	x = ((height + 15) >> 4) * MAX_FE_NBR_CTRL_LCU16_LINE_BUFFER_SIZE;157	y = ((height + 31) >> 5) * MAX_FE_NBR_CTRL_LCU32_LINE_BUFFER_SIZE;158	z = ((height + 63) >> 6) * MAX_FE_NBR_CTRL_LCU64_LINE_BUFFER_SIZE;159 160	return max3(x, y, z);161}162 163#define SIZE_VPXD_LB_FE_TOP_CTRL(width, height)		\164	(((ALIGN(width, 64) + 8) * 10 * 2)) /* small line */165#define SIZE_VPXD_LB_SE_TOP_CTRL(width, height) \166	((((width) + 15) >> 4) * MAX_FE_NBR_CTRL_LCU16_LINE_BUFFER_SIZE)167 168static inline u32 size_vpxd_lb_se_left_ctrl(u32 width, u32 height)169{170	u32 x, y, z;171 172	x = ((height + 15) >> 4) * MAX_SE_NBR_CTRL_LCU16_LINE_BUFFER_SIZE;173	y = ((height + 31) >> 5) * MAX_SE_NBR_CTRL_LCU32_LINE_BUFFER_SIZE;174	z = ((height + 63) >> 6) * MAX_SE_NBR_CTRL_LCU64_LINE_BUFFER_SIZE;175 176	return max3(x, y, z);177}178 179#define SIZE_VPXD_LB_RECON_DMA_METADATA_WR(width, height)	\180	ALIGN((ALIGN(height, 16) / (4 / 2)) * 64, 32)181#define SIZE_VP8D_LB_FE_TOP_DATA(width, height)			\182	((ALIGN(width, 16) + 8) * 10 * 2)183#define SIZE_VP9D_LB_FE_TOP_DATA(width, height)			\184	((ALIGN(ALIGN(width, 16), 64) + 8) * 10 * 2)185#define SIZE_VP8D_LB_PE_TOP_DATA(width, height)			\186	((ALIGN(width, 16) >> 4) * 64)187#define SIZE_VP9D_LB_PE_TOP_DATA(width, height)			\188	((ALIGN(ALIGN(width, 16), 64) >> 6) * 176)189#define SIZE_VP8D_LB_VSP_TOP(width, height)			\190	(((ALIGN(width, 16) >> 4) * 64 / 2) + 256)191#define SIZE_VP9D_LB_VSP_TOP(width, height)			\192	(((ALIGN(ALIGN(width, 16), 64) >> 6) * 64 * 8) + 256)193 194#define HFI_IRIS2_VP9D_COMV_SIZE				\195	((((8192 + 63) >> 6) * ((4320 + 63) >> 6) * 8 * 8 * 2 * 8))196 197#define VPX_DECODER_FRAME_CONCURENCY_LVL		2198#define VPX_DECODER_FRAME_BIN_HDR_BUDGET_RATIO_NUM	1199#define VPX_DECODER_FRAME_BIN_HDR_BUDGET_RATIO_DEN	2200#define VPX_DECODER_FRAME_BIN_RES_BUDGET_RATIO_NUM	3201#define VPX_DECODER_FRAME_BIN_RES_BUDGET_RATIO_DEN	2202 203#define VP8_NUM_FRAME_INFO_BUF			(5 + 1)204#define VP9_NUM_FRAME_INFO_BUF			32205#define VP8_NUM_PROBABILITY_TABLE_BUF		VP8_NUM_FRAME_INFO_BUF206#define VP9_NUM_PROBABILITY_TABLE_BUF		(VP9_NUM_FRAME_INFO_BUF + 4)207#define VP8_PROB_TABLE_SIZE			3840208#define VP9_PROB_TABLE_SIZE			3840209 210#define VP9_UDC_HEADER_BUF_SIZE			(3 * 128)211#define MAX_SUPERFRAME_HEADER_LEN		34212#define CCE_TILE_OFFSET_SIZE			ALIGN(32 * 4 * 4, 32)213 214#define QMATRIX_SIZE				(sizeof(u32) * 128 + 256)215#define MP2D_QPDUMP_SIZE			115200216#define HFI_IRIS2_ENC_PERSIST_SIZE		204800217#define HFI_MAX_COL_FRAME			6218#define HFI_VENUS_VENC_TRE_WB_BUFF_SIZE		(65 << 4) /* in Bytes */219#define HFI_VENUS_VENC_DB_LINE_BUFF_PER_MB	512220#define HFI_VENUS_VPPSG_MAX_REGISTERS		2048221#define HFI_VENUS_WIDTH_ALIGNMENT		128222#define HFI_VENUS_WIDTH_TEN_BIT_ALIGNMENT	192223#define HFI_VENUS_HEIGHT_ALIGNMENT		32224 225#define SYSTEM_LAL_TILE10	192226#define NUM_MBS_720P		(((1280 + 15) >> 4) * ((720 + 15) >> 4))227#define NUM_MBS_4K		(((4096 + 15) >> 4) * ((2304 + 15) >> 4))228#define MB_SIZE_IN_PIXEL	(16 * 16)229#define HDR10PLUS_PAYLOAD_SIZE		1024230#define HDR10_HIST_EXTRADATA_SIZE	4096231 232static u32 size_vpss_lb(u32 width, u32 height, u32 num_vpp_pipes)233{234	u32 vpss_4tap_top_buffer_size, vpss_div2_top_buffer_size;235	u32 vpss_4tap_left_buffer_size, vpss_div2_left_buffer_size;236	u32 opb_wr_top_line_luma_buf_size, opb_wr_top_line_chroma_buf_size;237	u32 opb_lb_wr_llb_y_buffer_size, opb_lb_wr_llb_uv_buffer_size;238	u32 macrotiling_size;239	u32 size = 0;240 241	vpss_4tap_top_buffer_size = 0;242	vpss_div2_top_buffer_size = 0;243	vpss_4tap_left_buffer_size = 0;244	vpss_div2_left_buffer_size = 0;245 246	macrotiling_size = 32;247	opb_wr_top_line_luma_buf_size =248		ALIGN(width, macrotiling_size) / macrotiling_size * 256;249	opb_wr_top_line_luma_buf_size =250		ALIGN(opb_wr_top_line_luma_buf_size, HFI_DMA_ALIGNMENT) +251		(MAX_TILE_COLUMNS - 1) * 256;252	opb_wr_top_line_luma_buf_size =253		max(opb_wr_top_line_luma_buf_size, (32 * ALIGN(height, 16)));254	opb_wr_top_line_chroma_buf_size = opb_wr_top_line_luma_buf_size;255	opb_lb_wr_llb_y_buffer_size = ALIGN((ALIGN(height, 16) / 2) * 64, 32);256	opb_lb_wr_llb_uv_buffer_size = opb_lb_wr_llb_y_buffer_size;257	size = num_vpp_pipes *258		2 * (vpss_4tap_top_buffer_size + vpss_div2_top_buffer_size) +259		2 * (vpss_4tap_left_buffer_size + vpss_div2_left_buffer_size) +260		opb_wr_top_line_luma_buf_size +261		opb_wr_top_line_chroma_buf_size +262		opb_lb_wr_llb_uv_buffer_size +263		opb_lb_wr_llb_y_buffer_size;264 265	return size;266}267 268static u32 size_h264d_hw_bin_buffer(u32 width, u32 height)269{270	u32 size_yuv, size_bin_hdr, size_bin_res;271	u32 size = 0;272	u32 product;273 274	product = width * height;275	size_yuv = (product <= BIN_BUFFER_THRESHOLD) ?276		((BIN_BUFFER_THRESHOLD * 3) >> 1) : ((product * 3) >> 1);277 278	size_bin_hdr = size_yuv * H264_CABAC_HDR_RATIO_HD_TOT;279	size_bin_res = size_yuv * H264_CABAC_RES_RATIO_HD_TOT;280	size_bin_hdr = ALIGN(size_bin_hdr, HFI_DMA_ALIGNMENT);281	size_bin_res = ALIGN(size_bin_res, HFI_DMA_ALIGNMENT);282	size = size_bin_hdr + size_bin_res;283 284	return size;285}286 287static u32 h264d_scratch_size(u32 width, u32 height, bool is_interlaced)288{289	u32 aligned_width = ALIGN(width, 16);290	u32 aligned_height = ALIGN(height, 16);291	u32 size = 0;292 293	if (!is_interlaced)294		size = size_h264d_hw_bin_buffer(aligned_width, aligned_height);295 296	return size;297}298 299static u32 size_h265d_hw_bin_buffer(u32 width, u32 height)300{301	u32 size_yuv, size_bin_hdr, size_bin_res;302	u32 size = 0;303	u32 product;304 305	product = width * height;306	size_yuv = (product <= BIN_BUFFER_THRESHOLD) ?307		((BIN_BUFFER_THRESHOLD * 3) >> 1) : ((product * 3) >> 1);308	size_bin_hdr = size_yuv * H265_CABAC_HDR_RATIO_HD_TOT;309	size_bin_res = size_yuv * H265_CABAC_RES_RATIO_HD_TOT;310	size_bin_hdr = ALIGN(size_bin_hdr, HFI_DMA_ALIGNMENT);311	size_bin_res = ALIGN(size_bin_res, HFI_DMA_ALIGNMENT);312	size = size_bin_hdr + size_bin_res;313 314	return size;315}316 317static u32 h265d_scratch_size(u32 width, u32 height, bool is_interlaced)318{319	u32 aligned_width = ALIGN(width, 16);320	u32 aligned_height = ALIGN(height, 16);321	u32 size = 0;322 323	if (!is_interlaced)324		size = size_h265d_hw_bin_buffer(aligned_width, aligned_height);325 326	return size;327}328 329static u32 vpxd_scratch_size(u32 width, u32 height, bool is_interlaced)330{331	u32 aligned_width = ALIGN(width, 16);332	u32 aligned_height = ALIGN(height, 16);333	u32 size_yuv = aligned_width * aligned_height * 3 / 2;334	u32 size = 0;335 336	if (!is_interlaced) {337		u32 binbuffer1_size, binbufer2_size;338 339		binbuffer1_size = max_t(u32, size_yuv,340					((BIN_BUFFER_THRESHOLD * 3) >> 1));341		binbuffer1_size *= VPX_DECODER_FRAME_CONCURENCY_LVL *342				   VPX_DECODER_FRAME_BIN_HDR_BUDGET_RATIO_NUM /343				   VPX_DECODER_FRAME_BIN_HDR_BUDGET_RATIO_DEN;344		binbufer2_size = max_t(u32, size_yuv,345				       ((BIN_BUFFER_THRESHOLD * 3) >> 1));346		binbufer2_size *= VPX_DECODER_FRAME_CONCURENCY_LVL *347				  VPX_DECODER_FRAME_BIN_RES_BUDGET_RATIO_NUM /348				  VPX_DECODER_FRAME_BIN_RES_BUDGET_RATIO_DEN;349		size = ALIGN(binbuffer1_size + binbufer2_size,350			     HFI_DMA_ALIGNMENT);351	}352 353	return size;354}355 356static u32 mpeg2d_scratch_size(u32 width, u32 height, bool is_interlaced)357{358	return 0;359}360 361static u32 calculate_enc_output_frame_size(u32 width, u32 height, u32 rc_type)362{363	u32 aligned_width, aligned_height;364	u32 mbs_per_frame;365	u32 frame_size;366 367	/*368	 * Encoder output size calculation: 32 Align width/height369	 * For resolution < 720p : YUVsize * 4370	 * For resolution > 720p & <= 4K : YUVsize / 2371	 * For resolution > 4k : YUVsize / 4372	 * Initially frame_size = YUVsize * 2;373	 */374	aligned_width = ALIGN(width, 32);375	aligned_height = ALIGN(height, 32);376	mbs_per_frame = (ALIGN(aligned_height, 16) *377			 ALIGN(aligned_width, 16)) / 256;378	frame_size = width * height * 3;379 380	if (mbs_per_frame < NUM_MBS_720P)381		frame_size = frame_size << 1;382	else if (mbs_per_frame <= NUM_MBS_4K)383		frame_size = frame_size >> 2;384	else385		frame_size = frame_size >> 3;386 387	if (rc_type == HFI_RATE_CONTROL_OFF || rc_type == HFI_RATE_CONTROL_CQ)388		frame_size = frame_size << 1;389 390	/*391	 * In case of opaque color format bitdepth will be known392	 * with first ETB, buffers allocated already with 8 bit393	 * won't be sufficient for 10 bit394	 * calculate size considering 10-bit by default395	 * For 10-bit cases size = size * 1.25396	 */397	frame_size *= 5;398	frame_size /= 4;399 400	return ALIGN(frame_size, SZ_4K);401}402 403static u32 calculate_enc_scratch_size(u32 width, u32 height, u32 work_mode,404				      u32 lcu_size, u32 num_vpp_pipes,405				      u32 rc_type)406{407	u32 aligned_width, aligned_height, bitstream_size;408	u32 total_bitbin_buffers, size_single_pipe, bitbin_size;409	u32 sao_bin_buffer_size, padded_bin_size, size;410 411	aligned_width = ALIGN(width, lcu_size);412	aligned_height = ALIGN(height, lcu_size);413	bitstream_size =414		calculate_enc_output_frame_size(width, height, rc_type);415 416	bitstream_size = ALIGN(bitstream_size, HFI_DMA_ALIGNMENT);417 418	if (work_mode == VIDC_WORK_MODE_2) {419		total_bitbin_buffers = 3;420		bitbin_size = bitstream_size * 17 / 10;421		bitbin_size = ALIGN(bitbin_size, HFI_DMA_ALIGNMENT);422	} else {423		total_bitbin_buffers = 1;424		bitstream_size = aligned_width * aligned_height * 3;425		bitbin_size = ALIGN(bitstream_size, HFI_DMA_ALIGNMENT);426	}427 428	if (num_vpp_pipes > 2)429		size_single_pipe = bitbin_size / 2;430	else431		size_single_pipe = bitbin_size;432 433	size_single_pipe = ALIGN(size_single_pipe, HFI_DMA_ALIGNMENT);434	sao_bin_buffer_size =435		(64 * (((width + 32) * (height + 32)) >> 10)) + 384;436	padded_bin_size = ALIGN(size_single_pipe, HFI_DMA_ALIGNMENT);437	size_single_pipe = sao_bin_buffer_size + padded_bin_size;438	size_single_pipe = ALIGN(size_single_pipe, HFI_DMA_ALIGNMENT);439	bitbin_size = size_single_pipe * num_vpp_pipes;440	size = ALIGN(bitbin_size, HFI_DMA_ALIGNMENT) *441		total_bitbin_buffers + 512;442 443	return size;444}445 446static u32 h264e_scratch_size(u32 width, u32 height, u32 work_mode,447			      u32 num_vpp_pipes, u32 rc_type)448{449	return calculate_enc_scratch_size(width, height, work_mode, 16,450					  num_vpp_pipes, rc_type);451}452 453static u32 h265e_scratch_size(u32 width, u32 height, u32 work_mode,454			      u32 num_vpp_pipes, u32 rc_type)455{456	return calculate_enc_scratch_size(width, height, work_mode, 32,457					  num_vpp_pipes, rc_type);458}459 460static u32 vp8e_scratch_size(u32 width, u32 height, u32 work_mode,461			     u32 num_vpp_pipes, u32 rc_type)462{463	return calculate_enc_scratch_size(width, height, work_mode, 16,464					  num_vpp_pipes, rc_type);465}466 467static u32 hfi_iris2_h264d_comv_size(u32 width, u32 height,468				     u32 yuv_buf_min_count)469{470	u32 frame_width_in_mbs = ((width + 15) >> 4);471	u32 frame_height_in_mbs = ((height + 15) >> 4);472	u32 col_mv_aligned_width = (frame_width_in_mbs << 7);473	u32 col_zero_aligned_width = (frame_width_in_mbs << 2);474	u32 col_zero_size = 0, size_colloc = 0, comv_size = 0;475 476	col_mv_aligned_width = ALIGN(col_mv_aligned_width, 16);477	col_zero_aligned_width = ALIGN(col_zero_aligned_width, 16);478	col_zero_size =479		col_zero_aligned_width * ((frame_height_in_mbs + 1) >> 1);480	col_zero_size = ALIGN(col_zero_size, 64);481	col_zero_size <<= 1;482	col_zero_size = ALIGN(col_zero_size, 512);483	size_colloc = col_mv_aligned_width * ((frame_height_in_mbs + 1) >> 1);484	size_colloc = ALIGN(size_colloc, 64);485	size_colloc <<= 1;486	size_colloc = ALIGN(size_colloc, 512);487	size_colloc += (col_zero_size + SIZE_H264D_BUFTAB_T * 2);488	comv_size = size_colloc * yuv_buf_min_count;489	comv_size += 512;490 491	return comv_size;492}493 494static u32 size_h264d_bse_cmd_buf(u32 height)495{496	u32 aligned_height = ALIGN(height, 32);497 498	return min_t(u32, (((aligned_height + 15) >> 4) * 3 * 4),499		     H264D_MAX_SLICE) * SIZE_H264D_BSE_CMD_PER_BUF;500}501 502static u32 size_h264d_vpp_cmd_buf(u32 height)503{504	u32 aligned_height = ALIGN(height, 32);505	u32 size;506 507	size = min_t(u32, (((aligned_height + 15) >> 4) * 3 * 4),508		     H264D_MAX_SLICE) * SIZE_H264D_VPP_CMD_PER_BUF;509	if (size > VPP_CMD_MAX_SIZE)510		size = VPP_CMD_MAX_SIZE;511 512	return size;513}514 515static u32 hfi_iris2_h264d_non_comv_size(u32 width, u32 height,516					 u32 num_vpp_pipes)517{518	u32 size_bse, size_vpp, size;519 520	size_bse = size_h264d_bse_cmd_buf(height);521	size_vpp = size_h264d_vpp_cmd_buf(height);522	size =523		ALIGN(size_bse, HFI_DMA_ALIGNMENT) +524		ALIGN(size_vpp, HFI_DMA_ALIGNMENT) +525		ALIGN(SIZE_HW_PIC(SIZE_H264D_HW_PIC_T), HFI_DMA_ALIGNMENT) +526		ALIGN(SIZE_H264D_LB_FE_TOP_DATA(width, height),527		      HFI_DMA_ALIGNMENT) +528		ALIGN(SIZE_H264D_LB_FE_TOP_CTRL(width, height),529		      HFI_DMA_ALIGNMENT) +530		ALIGN(SIZE_H264D_LB_FE_LEFT_CTRL(width, height),531		      HFI_DMA_ALIGNMENT) * num_vpp_pipes +532		ALIGN(SIZE_H264D_LB_SE_TOP_CTRL(width, height),533		      HFI_DMA_ALIGNMENT) +534		ALIGN(SIZE_H264D_LB_SE_LEFT_CTRL(width, height),535		      HFI_DMA_ALIGNMENT) * num_vpp_pipes +536		ALIGN(SIZE_H264D_LB_PE_TOP_DATA(width, height),537		      HFI_DMA_ALIGNMENT) +538		ALIGN(SIZE_H264D_LB_VSP_TOP(width, height), HFI_DMA_ALIGNMENT) +539		ALIGN(SIZE_H264D_LB_RECON_DMA_METADATA_WR(width, height),540		      HFI_DMA_ALIGNMENT) * 2 +541		ALIGN(SIZE_H264D_QP(width, height), HFI_DMA_ALIGNMENT);542 543	return ALIGN(size, HFI_DMA_ALIGNMENT);544}545 546static u32 size_h265d_bse_cmd_buf(u32 width, u32 height)547{548	u32 size;549 550	size = (ALIGN(width, LCU_MAX_SIZE_PELS) / LCU_MIN_SIZE_PELS) *551	       (ALIGN(height, LCU_MAX_SIZE_PELS) / LCU_MIN_SIZE_PELS) *552	       NUM_HW_PIC_BUF;553	size = min_t(u32, size, H265D_MAX_SLICE + 1);554	size = 2 * size * SIZE_H265D_BSE_CMD_PER_BUF;555 556	return ALIGN(size, HFI_DMA_ALIGNMENT);557}558 559static u32 size_h265d_vpp_cmd_buf(u32 width, u32 height)560{561	u32 size;562 563	size = (ALIGN(width, LCU_MAX_SIZE_PELS) / LCU_MIN_SIZE_PELS) *564	       (ALIGN(height, LCU_MAX_SIZE_PELS) / LCU_MIN_SIZE_PELS) *565	       NUM_HW_PIC_BUF;566	size = min_t(u32, size, H265D_MAX_SLICE + 1);567	size = ALIGN(size, 4);568	size = 2 * size * SIZE_H265D_VPP_CMD_PER_BUF;569	size = ALIGN(size, HFI_DMA_ALIGNMENT);570	if (size > VPP_CMD_MAX_SIZE)571		size = VPP_CMD_MAX_SIZE;572 573	return size;574}575 576static u32 hfi_iris2_h265d_comv_size(u32 width, u32 height,577				     u32 yuv_buf_count_min)578{579	u32 size;580 581	size = ALIGN(((((width + 15) >> 4) * ((height + 15) >> 4)) << 8), 512);582	size *= yuv_buf_count_min;583	size += 512;584 585	return size;586}587 588static u32 hfi_iris2_h265d_non_comv_size(u32 width, u32 height,589					 u32 num_vpp_pipes)590{591	u32 size_bse, size_vpp, size;592 593	size_bse = size_h265d_bse_cmd_buf(width, height);594	size_vpp = size_h265d_vpp_cmd_buf(width, height);595	size =596		ALIGN(size_bse, HFI_DMA_ALIGNMENT) +597		ALIGN(size_vpp, HFI_DMA_ALIGNMENT) +598		ALIGN(NUM_HW_PIC_BUF * 20 * 22 * 4, HFI_DMA_ALIGNMENT) +599		ALIGN(2 * sizeof(u16) *600		(ALIGN(width, LCU_MAX_SIZE_PELS) / LCU_MIN_SIZE_PELS) *601		(ALIGN(height, LCU_MAX_SIZE_PELS) / LCU_MIN_SIZE_PELS),602		       HFI_DMA_ALIGNMENT) +603		ALIGN(SIZE_HW_PIC(SIZE_H265D_HW_PIC_T), HFI_DMA_ALIGNMENT) +604		ALIGN(SIZE_H265D_LB_FE_TOP_DATA(width, height),605		      HFI_DMA_ALIGNMENT) +606		ALIGN(SIZE_H265D_LB_FE_TOP_CTRL(width, height),607		      HFI_DMA_ALIGNMENT) +608		ALIGN(SIZE_H265D_LB_FE_LEFT_CTRL(width, height),609		      HFI_DMA_ALIGNMENT) * num_vpp_pipes +610		ALIGN(size_h265d_lb_se_left_ctrl(width, height),611		      HFI_DMA_ALIGNMENT) * num_vpp_pipes +612		ALIGN(SIZE_H265D_LB_SE_TOP_CTRL(width, height),613		      HFI_DMA_ALIGNMENT) +614		ALIGN(SIZE_H265D_LB_PE_TOP_DATA(width, height),615		      HFI_DMA_ALIGNMENT) +616		ALIGN(SIZE_H265D_LB_VSP_TOP(width, height), HFI_DMA_ALIGNMENT) +617		ALIGN(SIZE_H265D_LB_VSP_LEFT(width, height),618		      HFI_DMA_ALIGNMENT) * num_vpp_pipes +619		ALIGN(SIZE_H265D_LB_RECON_DMA_METADATA_WR(width, height),620		      HFI_DMA_ALIGNMENT)621			* 4 +622		ALIGN(SIZE_H265D_QP(width, height), HFI_DMA_ALIGNMENT);623 624	return ALIGN(size, HFI_DMA_ALIGNMENT);625}626 627static u32 hfi_iris2_vp8d_comv_size(u32 width, u32 height,628				    u32 yuv_min_buf_count)629{630	return (((width + 15) >> 4) * ((height + 15) >> 4) * 8 * 2);631}632 633static u32 h264d_scratch1_size(u32 width, u32 height, u32 min_buf_count,634			       bool split_mode_enabled, u32 num_vpp_pipes)635{636	u32 co_mv_size, nonco_mv_size, vpss_lb_size = 0;637 638	co_mv_size = hfi_iris2_h264d_comv_size(width, height, min_buf_count);639	nonco_mv_size = hfi_iris2_h264d_non_comv_size(width, height,640						      num_vpp_pipes);641	if (split_mode_enabled)642		vpss_lb_size = size_vpss_lb(width, height, num_vpp_pipes);643 644	return co_mv_size + nonco_mv_size + vpss_lb_size;645}646 647static u32 h265d_scratch1_size(u32 width, u32 height, u32 min_buf_count,648			       bool split_mode_enabled, u32 num_vpp_pipes)649{650	u32 co_mv_size, nonco_mv_size, vpss_lb_size = 0;651 652	co_mv_size = hfi_iris2_h265d_comv_size(width, height, min_buf_count);653	nonco_mv_size = hfi_iris2_h265d_non_comv_size(width, height,654						      num_vpp_pipes);655	if (split_mode_enabled)656		vpss_lb_size = size_vpss_lb(width, height, num_vpp_pipes);657 658	return co_mv_size + nonco_mv_size + vpss_lb_size +659		HDR10_HIST_EXTRADATA_SIZE;660}661 662static u32 vp8d_scratch1_size(u32 width, u32 height, u32 min_buf_count,663			      bool split_mode_enabled, u32 num_vpp_pipes)664{665	u32 vpss_lb_size = 0, size;666 667	size = hfi_iris2_vp8d_comv_size(width, height, 0);668	size += ALIGN(size_vpxd_lb_fe_left_ctrl(width, height),669		      HFI_DMA_ALIGNMENT) * num_vpp_pipes +670		ALIGN(size_vpxd_lb_se_left_ctrl(width, height),671		      HFI_DMA_ALIGNMENT) * num_vpp_pipes +672		ALIGN(SIZE_VP8D_LB_VSP_TOP(width, height), HFI_DMA_ALIGNMENT) +673		ALIGN(SIZE_VPXD_LB_FE_TOP_CTRL(width, height),674		      HFI_DMA_ALIGNMENT) +675		2 * ALIGN(SIZE_VPXD_LB_RECON_DMA_METADATA_WR(width, height),676			  HFI_DMA_ALIGNMENT) +677		ALIGN(SIZE_VPXD_LB_SE_TOP_CTRL(width, height),678		      HFI_DMA_ALIGNMENT) +679		ALIGN(SIZE_VP8D_LB_PE_TOP_DATA(width, height),680		      HFI_DMA_ALIGNMENT) +681		ALIGN(SIZE_VP8D_LB_FE_TOP_DATA(width, height),682		      HFI_DMA_ALIGNMENT);683	if (split_mode_enabled)684		vpss_lb_size = size_vpss_lb(width, height, num_vpp_pipes);685 686	size += vpss_lb_size;687 688	return size;689}690 691static u32 vp9d_scratch1_size(u32 width, u32 height, u32 min_buf_count,692			      bool split_mode_enabled, u32 num_vpp_pipes)693{694	u32 vpss_lb_size = 0;695	u32 size;696 697	size =698		ALIGN(size_vpxd_lb_fe_left_ctrl(width, height),699		      HFI_DMA_ALIGNMENT) * num_vpp_pipes +700		ALIGN(size_vpxd_lb_se_left_ctrl(width, height),701		      HFI_DMA_ALIGNMENT) * num_vpp_pipes +702		ALIGN(SIZE_VP9D_LB_VSP_TOP(width, height), HFI_DMA_ALIGNMENT) +703		ALIGN(SIZE_VPXD_LB_FE_TOP_CTRL(width, height),704		      HFI_DMA_ALIGNMENT) +705		2 * ALIGN(SIZE_VPXD_LB_RECON_DMA_METADATA_WR(width, height),706			  HFI_DMA_ALIGNMENT) +707		ALIGN(SIZE_VPXD_LB_SE_TOP_CTRL(width, height),708		      HFI_DMA_ALIGNMENT) +709		ALIGN(SIZE_VP9D_LB_PE_TOP_DATA(width, height),710		      HFI_DMA_ALIGNMENT) +711		ALIGN(SIZE_VP9D_LB_FE_TOP_DATA(width, height),712		      HFI_DMA_ALIGNMENT);713 714	if (split_mode_enabled)715		vpss_lb_size = size_vpss_lb(width, height, num_vpp_pipes);716 717	size += vpss_lb_size + HDR10_HIST_EXTRADATA_SIZE;718 719	return size;720}721 722static u32 mpeg2d_scratch1_size(u32 width, u32 height, u32 min_buf_count,723				bool split_mode_enabled, u32 num_vpp_pipes)724{725	u32 vpss_lb_size = 0;726	u32 size;727 728	size =729		ALIGN(size_vpxd_lb_fe_left_ctrl(width, height),730		      HFI_DMA_ALIGNMENT) * num_vpp_pipes +731		ALIGN(size_vpxd_lb_se_left_ctrl(width, height),732		      HFI_DMA_ALIGNMENT) * num_vpp_pipes +733		ALIGN(SIZE_VP8D_LB_VSP_TOP(width, height), HFI_DMA_ALIGNMENT) +734		ALIGN(SIZE_VPXD_LB_FE_TOP_CTRL(width, height),735		      HFI_DMA_ALIGNMENT) +736		2 * ALIGN(SIZE_VPXD_LB_RECON_DMA_METADATA_WR(width, height),737			  HFI_DMA_ALIGNMENT) +738		ALIGN(SIZE_VPXD_LB_SE_TOP_CTRL(width, height),739		      HFI_DMA_ALIGNMENT) +740		ALIGN(SIZE_VP8D_LB_PE_TOP_DATA(width, height),741		      HFI_DMA_ALIGNMENT) +742		ALIGN(SIZE_VP8D_LB_FE_TOP_DATA(width, height),743		      HFI_DMA_ALIGNMENT);744 745	if (split_mode_enabled)746		vpss_lb_size = size_vpss_lb(width, height, num_vpp_pipes);747 748	size += vpss_lb_size;749 750	return size;751}752 753static u32754calculate_enc_scratch1_size(u32 width, u32 height, u32 lcu_size, u32 num_ref,755			    bool ten_bit, u32 num_vpp_pipes, bool is_h265)756{757	u32 line_buf_ctrl_size, line_buf_data_size, leftline_buf_ctrl_size;758	u32 line_buf_sde_size, sps_pps_slice_hdr, topline_buf_ctrl_size_FE;759	u32 leftline_buf_ctrl_size_FE, line_buf_recon_pix_size;760	u32 leftline_buf_recon_pix_size, lambda_lut_size, override_buffer_size;761	u32 col_mv_buf_size, vpp_reg_buffer_size, ir_buffer_size;762	u32 vpss_line_buf, leftline_buf_meta_recony, h265e_colrcbuf_size;763	u32 h265e_framerc_bufsize, h265e_lcubitcnt_bufsize;764	u32 h265e_lcubitmap_bufsize, se_stats_bufsize;765	u32 bse_reg_buffer_size, bse_slice_cmd_buffer_size, slice_info_bufsize;766	u32 line_buf_ctrl_size_buffid2, slice_cmd_buffer_size;767	u32 width_lcu_num, height_lcu_num, width_coded, height_coded;768	u32 frame_num_lcu, linebuf_meta_recon_uv, topline_bufsize_fe_1stg_sao;769	u32 size, bit_depth, num_lcu_mb;770	u32 vpss_line_buffer_size_1;771 772	width_lcu_num = (width + lcu_size - 1) / lcu_size;773	height_lcu_num = (height + lcu_size - 1) / lcu_size;774	frame_num_lcu = width_lcu_num * height_lcu_num;775	width_coded = width_lcu_num * lcu_size;776	height_coded = height_lcu_num * lcu_size;777	num_lcu_mb = (height_coded / lcu_size) *778		     ((width_coded + lcu_size * 8) / lcu_size);779	slice_info_bufsize = 256 + (frame_num_lcu << 4);780	slice_info_bufsize = ALIGN(slice_info_bufsize, HFI_DMA_ALIGNMENT);781	line_buf_ctrl_size = ALIGN(width_coded, HFI_DMA_ALIGNMENT);782	line_buf_ctrl_size_buffid2 = ALIGN(width_coded, HFI_DMA_ALIGNMENT);783 784	bit_depth = ten_bit ? 10 : 8;785	line_buf_data_size =786		(((((bit_depth * width_coded + 1024) +787		(HFI_DMA_ALIGNMENT - 1)) & (~(HFI_DMA_ALIGNMENT - 1))) * 1) +788		(((((bit_depth * width_coded + 1024) >> 1) +789		(HFI_DMA_ALIGNMENT - 1)) & (~(HFI_DMA_ALIGNMENT - 1))) * 2));790 791	leftline_buf_ctrl_size = is_h265 ?792		((height_coded + 32) / 32 * 4 * 16) :793		((height_coded + 15) / 16 * 5 * 16);794 795	if (num_vpp_pipes > 1) {796		leftline_buf_ctrl_size += 512;797		leftline_buf_ctrl_size =798			ALIGN(leftline_buf_ctrl_size, 512) * num_vpp_pipes;799	}800 801	leftline_buf_ctrl_size =802		ALIGN(leftline_buf_ctrl_size, HFI_DMA_ALIGNMENT);803	leftline_buf_recon_pix_size = (((ten_bit + 1) * 2 *804		(height_coded) + HFI_DMA_ALIGNMENT) +805		(HFI_DMA_ALIGNMENT << (num_vpp_pipes - 1)) - 1) &806		(~((HFI_DMA_ALIGNMENT << (num_vpp_pipes - 1)) - 1)) * 1;807 808	topline_buf_ctrl_size_FE = is_h265 ? (64 * (width_coded >> 5)) :809		(HFI_DMA_ALIGNMENT + 16 * (width_coded >> 4));810	topline_buf_ctrl_size_FE =811		ALIGN(topline_buf_ctrl_size_FE, HFI_DMA_ALIGNMENT);812	leftline_buf_ctrl_size_FE =813		(((HFI_DMA_ALIGNMENT + 64 * (height_coded >> 4)) +814		(HFI_DMA_ALIGNMENT << (num_vpp_pipes - 1)) - 1) &815		(~((HFI_DMA_ALIGNMENT << (num_vpp_pipes - 1)) - 1)) * 1) *816		num_vpp_pipes;817	leftline_buf_meta_recony = (HFI_DMA_ALIGNMENT + 64 *818		((height_coded) / (8 * (ten_bit ? 4 : 8))));819	leftline_buf_meta_recony =820		ALIGN(leftline_buf_meta_recony, HFI_DMA_ALIGNMENT);821	leftline_buf_meta_recony = leftline_buf_meta_recony * num_vpp_pipes;822	linebuf_meta_recon_uv = (HFI_DMA_ALIGNMENT + 64 *823		((height_coded) / (4 * (ten_bit ? 4 : 8))));824	linebuf_meta_recon_uv = ALIGN(linebuf_meta_recon_uv, HFI_DMA_ALIGNMENT);825	linebuf_meta_recon_uv = linebuf_meta_recon_uv * num_vpp_pipes;826	line_buf_recon_pix_size = ((ten_bit ? 3 : 2) * width_coded);827	line_buf_recon_pix_size =828		ALIGN(line_buf_recon_pix_size, HFI_DMA_ALIGNMENT);829	slice_cmd_buffer_size = ALIGN(20480, HFI_DMA_ALIGNMENT);830	sps_pps_slice_hdr = 2048 + 4096;831	col_mv_buf_size = is_h265 ? (16 * ((frame_num_lcu << 2) + 32)) :832		(3 * 16 * (width_lcu_num * height_lcu_num + 32));833	col_mv_buf_size =834		ALIGN(col_mv_buf_size, HFI_DMA_ALIGNMENT) * (num_ref + 1);835	h265e_colrcbuf_size =836		(((width_lcu_num + 7) >> 3) * 16 * 2 * height_lcu_num);837	if (num_vpp_pipes > 1)838		h265e_colrcbuf_size =839			ALIGN(h265e_colrcbuf_size, HFI_DMA_ALIGNMENT) *840			num_vpp_pipes;841 842	h265e_colrcbuf_size = ALIGN(h265e_colrcbuf_size, HFI_DMA_ALIGNMENT) *843				HFI_MAX_COL_FRAME;844	h265e_framerc_bufsize = (is_h265) ? (256 + 16 *845		(14 + (((height_coded >> 5) + 7) >> 3))) :846		(256 + 16 * (14 + (((height_coded >> 4) + 7) >> 3)));847	h265e_framerc_bufsize *= 6;   /* multiply by max numtilescol */848	if (num_vpp_pipes > 1)849		h265e_framerc_bufsize =850			ALIGN(h265e_framerc_bufsize, HFI_DMA_ALIGNMENT) *851			num_vpp_pipes;852 853	h265e_framerc_bufsize = ALIGN(h265e_framerc_bufsize, 512) *854				HFI_MAX_COL_FRAME;855	h265e_lcubitcnt_bufsize = 256 + 4 * frame_num_lcu;856	h265e_lcubitcnt_bufsize =857		ALIGN(h265e_lcubitcnt_bufsize, HFI_DMA_ALIGNMENT);858	h265e_lcubitmap_bufsize = 256 + (frame_num_lcu >> 3);859	h265e_lcubitmap_bufsize =860		ALIGN(h265e_lcubitmap_bufsize, HFI_DMA_ALIGNMENT);861	line_buf_sde_size = 256 + 16 * (width_coded >> 4);862	line_buf_sde_size = ALIGN(line_buf_sde_size, HFI_DMA_ALIGNMENT);863	if ((width_coded * height_coded) > (4096 * 2160))864		se_stats_bufsize = 0;865	else if ((width_coded * height_coded) > (1920 * 1088))866		se_stats_bufsize = (40 * 4 * frame_num_lcu + 256 + 256);867	else868		se_stats_bufsize = (1024 * frame_num_lcu + 256 + 256);869 870	se_stats_bufsize = ALIGN(se_stats_bufsize, HFI_DMA_ALIGNMENT) * 2;871	bse_slice_cmd_buffer_size = (((8192 << 2) + 7) & (~7)) * 6;872	bse_reg_buffer_size = (((512 << 3) + 7) & (~7)) * 4;873	vpp_reg_buffer_size =874		(((HFI_VENUS_VPPSG_MAX_REGISTERS << 3) + 31) & (~31)) * 10;875	lambda_lut_size = 256 * 11;876	override_buffer_size = 16 * ((num_lcu_mb + 7) >> 3);877	override_buffer_size =878		ALIGN(override_buffer_size, HFI_DMA_ALIGNMENT) * 2;879	ir_buffer_size = (((frame_num_lcu << 1) + 7) & (~7)) * 3;880	vpss_line_buffer_size_1 = (((8192 >> 2) << 5) * num_vpp_pipes) + 64;881	vpss_line_buf =882		(((((max(width_coded, height_coded) + 3) >> 2) << 5) + 256) *883		16) + vpss_line_buffer_size_1;884	topline_bufsize_fe_1stg_sao = 16 * (width_coded >> 5);885	topline_bufsize_fe_1stg_sao =886		ALIGN(topline_bufsize_fe_1stg_sao, HFI_DMA_ALIGNMENT);887 888	size =889		line_buf_ctrl_size + line_buf_data_size +890		line_buf_ctrl_size_buffid2 + leftline_buf_ctrl_size +891		vpss_line_buf + col_mv_buf_size + topline_buf_ctrl_size_FE +892		leftline_buf_ctrl_size_FE + line_buf_recon_pix_size +893		leftline_buf_recon_pix_size +894		leftline_buf_meta_recony + linebuf_meta_recon_uv +895		h265e_colrcbuf_size + h265e_framerc_bufsize +896		h265e_lcubitcnt_bufsize + h265e_lcubitmap_bufsize +897		line_buf_sde_size +898		topline_bufsize_fe_1stg_sao + override_buffer_size +899		bse_reg_buffer_size + vpp_reg_buffer_size + sps_pps_slice_hdr +900		slice_cmd_buffer_size + bse_slice_cmd_buffer_size +901		ir_buffer_size + slice_info_bufsize + lambda_lut_size +902		se_stats_bufsize + 1024;903 904	return size;905}906 907static u32 h264e_scratch1_size(u32 width, u32 height, u32 num_ref, bool ten_bit,908			       u32 num_vpp_pipes)909{910	return calculate_enc_scratch1_size(width, height, 16, num_ref, ten_bit,911					   num_vpp_pipes, false);912}913 914static u32 h265e_scratch1_size(u32 width, u32 height, u32 num_ref, bool ten_bit,915			       u32 num_vpp_pipes)916{917	return calculate_enc_scratch1_size(width, height, 32, num_ref, ten_bit,918					   num_vpp_pipes, true);919}920 921static u32 vp8e_scratch1_size(u32 width, u32 height, u32 num_ref, bool ten_bit,922			      u32 num_vpp_pipes)923{924	return calculate_enc_scratch1_size(width, height, 16, num_ref, ten_bit,925					   1, false);926}927 928static u32 ubwc_metadata_plane_stride(u32 width, u32 metadata_stride_multi,929				      u32 tile_width_pels)930{931	return ALIGN(((width + (tile_width_pels - 1)) / tile_width_pels),932			metadata_stride_multi);933}934 935static u32 ubwc_metadata_plane_bufheight(u32 height, u32 metadata_height_multi,936					 u32 tile_height_pels)937{938	return ALIGN(((height + (tile_height_pels - 1)) / tile_height_pels),939			metadata_height_multi);940}941 942static u32 ubwc_metadata_plane_buffer_size(u32 metadata_stride,943					   u32 metadata_buf_height)944{945	return ALIGN(metadata_stride * metadata_buf_height, SZ_4K);946}947 948static u32 enc_scratch2_size(u32 width, u32 height, u32 num_ref, bool ten_bit)949{950	u32 aligned_width, aligned_height, chroma_height, ref_buf_height;951	u32 luma_size, chroma_size;952	u32 metadata_stride, meta_buf_height, meta_size_y, meta_size_c;953	u32 ref_luma_stride_bytes, ref_chroma_height_bytes;954	u32 ref_buf_size, ref_stride;955	u32 size;956 957	if (!ten_bit) {958		aligned_height = ALIGN(height, HFI_VENUS_HEIGHT_ALIGNMENT);959		chroma_height = height >> 1;960		chroma_height = ALIGN(chroma_height,961				      HFI_VENUS_HEIGHT_ALIGNMENT);962		aligned_width = ALIGN(width, HFI_VENUS_WIDTH_ALIGNMENT);963		metadata_stride =964			ubwc_metadata_plane_stride(width, 64,965						   NV12_UBWC_Y_TILE_WIDTH);966		meta_buf_height =967			ubwc_metadata_plane_bufheight(height, 16,968						      NV12_UBWC_Y_TILE_HEIGHT);969		meta_size_y = ubwc_metadata_plane_buffer_size(metadata_stride,970							      meta_buf_height);971		meta_size_c = ubwc_metadata_plane_buffer_size(metadata_stride,972							      meta_buf_height);973		size = (aligned_height + chroma_height) * aligned_width +974			meta_size_y + meta_size_c;975		size = (size * (num_ref + 3)) + 4096;976	} else {977		ref_buf_height = (height + (HFI_VENUS_HEIGHT_ALIGNMENT - 1))978					& (~(HFI_VENUS_HEIGHT_ALIGNMENT - 1));979		ref_luma_stride_bytes =980			((width + SYSTEM_LAL_TILE10 - 1) / SYSTEM_LAL_TILE10) *981			SYSTEM_LAL_TILE10;982		ref_stride = 4 * (ref_luma_stride_bytes / 3);983		ref_stride = (ref_stride + (128 - 1)) & (~(128 - 1));984		luma_size = ref_buf_height * ref_stride;985		ref_chroma_height_bytes = (((height + 1) >> 1) +986			(32 - 1)) & (~(32 - 1));987		chroma_size = ref_stride * ref_chroma_height_bytes;988		luma_size = (luma_size + (SZ_4K - 1)) & (~(SZ_4K - 1));989		chroma_size = (chroma_size + (SZ_4K - 1)) & (~(SZ_4K - 1));990		ref_buf_size = luma_size + chroma_size;991		metadata_stride =992			ubwc_metadata_plane_stride(width,993						   METADATA_STRIDE_MULTIPLE,994						   TP10_UBWC_Y_TILE_WIDTH);995		meta_buf_height =996			ubwc_metadata_plane_bufheight(height,997						      METADATA_HEIGHT_MULTIPLE,998						      TP10_UBWC_Y_TILE_HEIGHT);999		meta_size_y = ubwc_metadata_plane_buffer_size(metadata_stride,1000							      meta_buf_height);1001		meta_size_c = ubwc_metadata_plane_buffer_size(metadata_stride,1002							      meta_buf_height);1003		size = ref_buf_size + meta_size_y + meta_size_c;1004		size = (size * (num_ref + 3)) + 4096;1005	}1006 1007	return size;1008}1009 1010static u32 enc_persist_size(void)1011{1012	return HFI_IRIS2_ENC_PERSIST_SIZE;1013}1014 1015static u32 h264d_persist1_size(void)1016{1017	return ALIGN((SIZE_SLIST_BUF_H264 * NUM_SLIST_BUF_H2641018		     + NUM_HW_PIC_BUF * SIZE_SEI_USERDATA), HFI_DMA_ALIGNMENT);1019}1020 1021static u32 h265d_persist1_size(void)1022{1023	return ALIGN((SIZE_SLIST_BUF_H265 * NUM_SLIST_BUF_H265 + H265_NUM_TILE1024			* sizeof(u32) + NUM_HW_PIC_BUF * SIZE_SEI_USERDATA), HFI_DMA_ALIGNMENT);1025}1026 1027static u32 vp8d_persist1_size(void)1028{1029	return ALIGN(VP8_NUM_PROBABILITY_TABLE_BUF * VP8_PROB_TABLE_SIZE,1030			HFI_DMA_ALIGNMENT);1031}1032 1033static u32 vp9d_persist1_size(void)1034{1035	return1036		ALIGN(VP9_NUM_PROBABILITY_TABLE_BUF * VP9_PROB_TABLE_SIZE,1037		      HFI_DMA_ALIGNMENT) +1038		ALIGN(HFI_IRIS2_VP9D_COMV_SIZE, HFI_DMA_ALIGNMENT) +1039		ALIGN(MAX_SUPERFRAME_HEADER_LEN, HFI_DMA_ALIGNMENT) +1040		ALIGN(VP9_UDC_HEADER_BUF_SIZE, HFI_DMA_ALIGNMENT) +1041		ALIGN(VP9_NUM_FRAME_INFO_BUF * CCE_TILE_OFFSET_SIZE,1042		      HFI_DMA_ALIGNMENT);1043}1044 1045static u32 mpeg2d_persist1_size(void)1046{1047	return QMATRIX_SIZE + MP2D_QPDUMP_SIZE;1048}1049 1050struct dec_bufsize_ops {1051	u32 (*scratch)(u32 width, u32 height, bool is_interlaced);1052	u32 (*scratch1)(u32 width, u32 height, u32 min_buf_count,1053			bool split_mode_enabled, u32 num_vpp_pipes);1054	u32 (*persist1)(void);1055};1056 1057struct enc_bufsize_ops {1058	u32 (*scratch)(u32 width, u32 height, u32 work_mode, u32 num_vpp_pipes,1059		       u32 rc_type);1060	u32 (*scratch1)(u32 width, u32 height, u32 num_ref, bool ten_bit,1061			u32 num_vpp_pipes);1062	u32 (*scratch2)(u32 width, u32 height, u32 num_ref, bool ten_bit);1063	u32 (*persist)(void);1064};1065 1066static const struct dec_bufsize_ops dec_h264_ops = {1067	.scratch = h264d_scratch_size,1068	.scratch1 = h264d_scratch1_size,1069	.persist1 = h264d_persist1_size,1070};1071 1072static const struct dec_bufsize_ops dec_h265_ops = {1073	.scratch = h265d_scratch_size,1074	.scratch1 = h265d_scratch1_size,1075	.persist1 = h265d_persist1_size,1076};1077 1078static const struct dec_bufsize_ops dec_vp8_ops = {1079	.scratch = vpxd_scratch_size,1080	.scratch1 = vp8d_scratch1_size,1081	.persist1 = vp8d_persist1_size,1082};1083 1084static const struct dec_bufsize_ops dec_vp9_ops = {1085	.scratch = vpxd_scratch_size,1086	.scratch1 = vp9d_scratch1_size,1087	.persist1 = vp9d_persist1_size,1088};1089 1090static const struct dec_bufsize_ops dec_mpeg2_ops = {1091	.scratch = mpeg2d_scratch_size,1092	.scratch1 = mpeg2d_scratch1_size,1093	.persist1 = mpeg2d_persist1_size,1094};1095 1096static const struct enc_bufsize_ops enc_h264_ops = {1097	.scratch = h264e_scratch_size,1098	.scratch1 = h264e_scratch1_size,1099	.scratch2 = enc_scratch2_size,1100	.persist = enc_persist_size,1101};1102 1103static const struct enc_bufsize_ops enc_h265_ops = {1104	.scratch = h265e_scratch_size,1105	.scratch1 = h265e_scratch1_size,1106	.scratch2 = enc_scratch2_size,1107	.persist = enc_persist_size,1108};1109 1110static const struct enc_bufsize_ops enc_vp8_ops = {1111	.scratch = vp8e_scratch_size,1112	.scratch1 = vp8e_scratch1_size,1113	.scratch2 = enc_scratch2_size,1114	.persist = enc_persist_size,1115};1116 1117static u321118calculate_dec_input_frame_size(u32 width, u32 height, u32 codec,1119			       u32 max_mbs_per_frame, u32 buffer_size_limit)1120{1121	u32 frame_size, num_mbs;1122	u32 div_factor = 1;1123	u32 base_res_mbs = NUM_MBS_4K;1124 1125	/*1126	 * Decoder input size calculation:1127	 * If clip is 8k buffer size is calculated for 8k : 8k mbs/41128	 * For 8k cases we expect width/height to be set always.1129	 * In all other cases size is calculated for 4k:1130	 * 4k mbs for VP8/VP9 and 4k/2 for remaining codecs1131	 */1132	num_mbs = (ALIGN(height, 16) * ALIGN(width, 16)) / 256;1133	if (num_mbs > NUM_MBS_4K) {1134		div_factor = 4;1135		base_res_mbs = max_mbs_per_frame;1136	} else {1137		base_res_mbs = NUM_MBS_4K;1138		if (codec == V4L2_PIX_FMT_VP9)1139			div_factor = 1;1140		else1141			div_factor = 2;1142	}1143 1144	frame_size = base_res_mbs * MB_SIZE_IN_PIXEL * 3 / 2 / div_factor;1145 1146	/* multiply by 10/8 (1.25) to get size for 10 bit case */1147	if (codec == V4L2_PIX_FMT_VP9 || codec == V4L2_PIX_FMT_HEVC)1148		frame_size = frame_size + (frame_size >> 2);1149 1150	if (buffer_size_limit && buffer_size_limit < frame_size)1151		frame_size = buffer_size_limit;1152 1153	return ALIGN(frame_size, SZ_4K);1154}1155 1156static int output_buffer_count(u32 session_type, u32 codec)1157{1158	u32 output_min_count;1159 1160	if (session_type == VIDC_SESSION_TYPE_DEC) {1161		switch (codec) {1162		case V4L2_PIX_FMT_MPEG2:1163		case V4L2_PIX_FMT_VP8:1164			output_min_count = 6;1165			break;1166		case V4L2_PIX_FMT_VP9:1167			output_min_count = 11;1168			break;1169		case V4L2_PIX_FMT_H264:1170		case V4L2_PIX_FMT_HEVC:1171		default:1172			output_min_count = 18;1173			break;1174		}1175	} else {1176		output_min_count = MIN_ENC_OUTPUT_BUFFERS;1177	}1178 1179	return output_min_count;1180}1181 1182static int bufreq_dec(struct hfi_plat_buffers_params *params, u32 buftype,1183		      struct hfi_buffer_requirements *bufreq)1184{1185	enum hfi_version version = params->version;1186	u32 codec = params->codec;1187	u32 width = params->width, height = params->height, out_min_count;1188	u32 out_width = params->out_width, out_height = params->out_height;1189	const struct dec_bufsize_ops *dec_ops;1190	bool is_secondary_output = params->dec.is_secondary_output;1191	bool is_interlaced = params->dec.is_interlaced;1192	u32 max_mbs_per_frame = params->dec.max_mbs_per_frame;1193	u32 buffer_size_limit = params->dec.buffer_size_limit;1194	u32 num_vpp_pipes = params->num_vpp_pipes;1195 1196	switch (codec) {1197	case V4L2_PIX_FMT_H264:1198		dec_ops = &dec_h264_ops;1199		break;1200	case V4L2_PIX_FMT_HEVC:1201		dec_ops = &dec_h265_ops;1202		break;1203	case V4L2_PIX_FMT_VP8:1204		dec_ops = &dec_vp8_ops;1205		break;1206	case V4L2_PIX_FMT_VP9:1207		dec_ops = &dec_vp9_ops;1208		break;1209	case V4L2_PIX_FMT_MPEG2:1210		dec_ops = &dec_mpeg2_ops;1211		break;1212	default:1213		return -EINVAL;1214	}1215 1216	out_min_count = output_buffer_count(VIDC_SESSION_TYPE_DEC, codec);1217	/* Max of driver and FW count */1218	out_min_count = max(out_min_count, hfi_bufreq_get_count_min(bufreq, version));1219 1220	bufreq->type = buftype;1221	bufreq->region_size = 0;1222	bufreq->count_actual = 1;1223	hfi_bufreq_set_count_min(bufreq, version, 1);1224	hfi_bufreq_set_hold_count(bufreq, version, 1);1225	bufreq->contiguous = 1;1226	bufreq->alignment = 256;1227 1228	if (buftype == HFI_BUFFER_INPUT) {1229		hfi_bufreq_set_count_min(bufreq, version, MIN_INPUT_BUFFERS);1230		bufreq->size =1231			calculate_dec_input_frame_size(width, height, codec,1232						       max_mbs_per_frame,1233						       buffer_size_limit);1234	} else if (buftype == HFI_BUFFER_OUTPUT || buftype == HFI_BUFFER_OUTPUT2) {1235		hfi_bufreq_set_count_min(bufreq, version, out_min_count);1236		bufreq->size =1237			venus_helper_get_framesz_raw(params->hfi_color_fmt,1238						     out_width, out_height);1239		if (buftype == HFI_BUFFER_OUTPUT &&1240		    params->dec.is_secondary_output)1241			bufreq->size =1242				venus_helper_get_framesz_raw(params->hfi_dpb_color_fmt,1243							     out_width, out_height);1244	} else if (buftype == HFI_BUFFER_INTERNAL_SCRATCH(version)) {1245		bufreq->size = dec_ops->scratch(width, height, is_interlaced);1246	} else if (buftype == HFI_BUFFER_INTERNAL_SCRATCH_1(version)) {1247		bufreq->size = dec_ops->scratch1(width, height, VB2_MAX_FRAME,1248						 is_secondary_output,1249						 num_vpp_pipes);1250	} else if (buftype == HFI_BUFFER_INTERNAL_PERSIST_1) {1251		bufreq->size = dec_ops->persist1();1252	} else {1253		bufreq->size = 0;1254	}1255 1256	return 0;1257}1258 1259static int bufreq_enc(struct hfi_plat_buffers_params *params, u32 buftype,1260		      struct hfi_buffer_requirements *bufreq)1261{1262	enum hfi_version version = params->version;1263	const struct enc_bufsize_ops *enc_ops;1264	u32 width = params->width;1265	u32 height = params->height;1266	bool is_tenbit = params->enc.is_tenbit;1267	u32 num_bframes = params->enc.num_b_frames;1268	u32 codec = params->codec;1269	u32 work_mode = params->enc.work_mode;1270	u32 rc_type = params->enc.rc_type;1271	u32 num_vpp_pipes = params->num_vpp_pipes;1272	u32 num_ref, count_min;1273 1274	switch (codec) {1275	case V4L2_PIX_FMT_H264:1276		enc_ops = &enc_h264_ops;1277		break;1278	case V4L2_PIX_FMT_HEVC:1279		enc_ops = &enc_h265_ops;1280		break;1281	case V4L2_PIX_FMT_VP8:1282		enc_ops = &enc_vp8_ops;1283		break;1284	default:1285		return -EINVAL;1286	}1287 1288	num_ref = num_bframes > 0 ? num_bframes + 1 : 1;1289 1290	bufreq->type = buftype;1291	bufreq->region_size = 0;1292	bufreq->count_actual = 1;1293	hfi_bufreq_set_count_min(bufreq, version, 1);1294	hfi_bufreq_set_hold_count(bufreq, version, 1);1295	bufreq->contiguous = 1;1296	bufreq->alignment = 256;1297 1298	if (buftype == HFI_BUFFER_INPUT) {1299		hfi_bufreq_set_count_min(bufreq, version, MIN_INPUT_BUFFERS);1300		bufreq->size =1301			venus_helper_get_framesz_raw(params->hfi_color_fmt,1302						     width, height);1303	} else if (buftype == HFI_BUFFER_OUTPUT ||1304		   buftype == HFI_BUFFER_OUTPUT2) {1305		count_min = output_buffer_count(VIDC_SESSION_TYPE_ENC, codec);1306		hfi_bufreq_set_count_min(bufreq, version, count_min);1307		bufreq->size = calculate_enc_output_frame_size(width, height,1308							       rc_type);1309	} else if (buftype == HFI_BUFFER_INTERNAL_SCRATCH(version)) {1310		bufreq->size = enc_ops->scratch(width, height, work_mode,1311						num_vpp_pipes, rc_type);1312	} else if (buftype == HFI_BUFFER_INTERNAL_SCRATCH_1(version)) {1313		bufreq->size = enc_ops->scratch1(width, height, num_ref,1314						 is_tenbit, num_vpp_pipes);1315	} else if (buftype == HFI_BUFFER_INTERNAL_SCRATCH_2(version)) {1316		bufreq->size = enc_ops->scratch2(width, height, num_ref,1317						 is_tenbit);1318	} else if (buftype == HFI_BUFFER_INTERNAL_PERSIST) {1319		bufreq->size = enc_ops->persist();1320	} else {1321		bufreq->size = 0;1322	}1323 1324	return 0;1325}1326 1327int hfi_plat_bufreq_v6(struct hfi_plat_buffers_params *params, u32 session_type,1328		       u32 buftype, struct hfi_buffer_requirements *bufreq)1329{1330	if (session_type == VIDC_SESSION_TYPE_DEC)1331		return bufreq_dec(params, buftype, bufreq);1332	else1333		return bufreq_enc(params, buftype, bufreq);1334}1335