/*
 * Copyright (c) 2017-2018, NVIDIA CORPORATION.  All rights reserved.
 *
 * NVIDIA CORPORATION and its licensors retain all intellectual property
 * and proprietary rights in and to this software, related documentation
 * and any modifications thereto.  Any use, reproduction, disclosure or
 * distribution of this software and related documentation without an express
 * license agreement from NVIDIA CORPORATION is strictly prohibited.
 */

#ifndef INCLUDE_CAMRTC_ISP5_TILING_H
#define INCLUDE_CAMRTC_ISP5_TILING_H

#include "camrtc-common.h"
#include "camrtc-capture.h"

struct isp5_tile_width {
	uint16_t tile_width_first;
	uint16_t tile_width_middle;
	uint16_t tiles_in_slice;
};

struct isp5_slice_height {
	uint16_t slice_height;
	uint16_t vi_first_slice_height;
	uint16_t slices_in_image;
};

#define ISP5_MIN_TILE_WIDTH	U16_C(128)
#define ISP5_MAX_TILE_WIDTH	U16_C(1024)
#define ISP5_MIN_SLICE_HEIGHT	U16_C(128)
#define ISP5_MAX_SLICE_HEIGHT	U16_C(540)

static inline uint16_t isp5_min_u16(uint16_t a, uint16_t b)
{
	if (a < b)
		return a;
	else
		return b;
}

static inline uint16_t isp5_max_u16(uint16_t a, uint16_t b)
{
	if (a > b)
		return a;
	else
		return b;
}

static inline uint16_t isp5_align_down(uint16_t val, uint16_t alignment)
{
	uint16_t rem = val % alignment;

	return (rem > 0U) ? (val - rem) : val;
}

static inline uint16_t isp5_align_up(uint16_t val, uint16_t alignment)
{
	uint16_t rem = val % alignment;

	return (rem > 0U) ? (isp5_align_down(val, alignment) + alignment) : val;
}

static inline uint16_t isp5_div_round_up(uint16_t x, uint16_t y)
{
	return (x + y - 1U) / y;
}

/**
 * Calculate suitable tile width for given capture descriptor and ISP program
 */
__attribute__((unused))
static bool isp5_find_tile_width(const struct isp5_program *prg,
						const struct isp_capture_descriptor *cd,
						struct isp5_tile_width *tiling)
{
	const uint16_t img_width = cd->surface_configs.mr_width;
	const uint16_t alignment = prg->overfetch.alignment;

	if (img_width <= ISP5_MAX_TILE_WIDTH) {
		tiling->tile_width_first = img_width;
		tiling->tile_width_middle = 0U;
		tiling->tiles_in_slice = 1;
		return true;
	}

	if (alignment == 0) {
		return false;
	}

	const uint16_t max_width_first = isp5_align_down(ISP5_MAX_TILE_WIDTH -
							prg->overfetch.right +
							prg->overfetch.pru_ovf_h, alignment) -
					prg->overfetch.right + prg->overfetch.pru_ovf_h;

	const uint16_t max_width_middle = isp5_align_down(ISP5_MAX_TILE_WIDTH -
							prg->overfetch.right -
							prg->overfetch.left,
							alignment);

	/* Last tile right edge does not need to be aligned */
	const uint16_t max_width_last = ISP5_MAX_TILE_WIDTH - prg->overfetch.left;
	const uint16_t min_width = isp5_max_u16(ISP5_MIN_TILE_WIDTH, prg->overfetch.right);

	uint16_t tile_count = 2;

	if (img_width > max_width_first + max_width_last) {
		const uint16_t pixels_left = img_width - max_width_first - max_width_last;
		const uint16_t middle_tiles = isp5_div_round_up(pixels_left,
							isp5_min_u16(max_width_middle, max_width_first));
		tile_count += middle_tiles;
	}

	/* Divide image into roughly evenly spaced aligned tiles */
	uint16_t tile_width = (isp5_div_round_up(img_width, alignment) / tile_count) * alignment;

	/*
	 * The right edge of a tile as seen by AP must be aligned
	 * correctly for CAR filter.  When first tile width fulfills
	 * this condition, the rest of tiles are simple to andle by
	 * just aligning their active width
	 */
	uint16_t first_width = isp5_min_u16(max_width_first,
					isp5_align_down(
					tile_width + prg->overfetch.right -
					prg->overfetch.pru_ovf_h, alignment) -
					prg->overfetch.right + prg->overfetch.pru_ovf_h);
	uint16_t middle_width = (tile_count > 2) ? isp5_min_u16(max_width_middle, tile_width) : 0U;
	uint16_t last_width = img_width - first_width - (tile_count - 2) * middle_width;

	/*
	 * Ensure that last tile is wide enough. Width of the first
	 * tile a this point is guaranteed to be greater than:
	 *
	 * ((max_tile_width - total overfetch - 2*alignment) / 2) - alignment >= 407 pixels
	 *
	 * So there is no risk that this correction will cause it to
	 * be too narrow.
	 */
	if (last_width < min_width) {
		uint16_t corr = isp5_align_up(min_width-last_width, alignment);

		first_width -= corr;
		last_width  += corr;

	} else if (last_width > max_width_last) {
		const uint16_t corr = last_width - max_width_last;

		/* Try first increasing middle tile width */
		if (tile_count > 2) {
			const uint16_t max_middle_corr = max_width_middle - middle_width;
			const uint16_t middle_corr =
					isp5_min_u16(max_middle_corr,
						isp5_align_up(isp5_div_round_up(corr, tile_count-2),
							alignment));
			middle_width += middle_corr;
			last_width -= middle_corr * (tile_count-2);
		}

		if (last_width > max_width_last) {
			const uint16_t first_corr = isp5_align_up(last_width - max_width_last, alignment);

			first_width += first_corr;
			last_width -= first_corr;
		}
	}

	if (first_width < min_width) {
		return false;
	}

	if (first_width > max_width_first) {
		return false;
	}

	if (last_width < min_width) {
		return false;
	}

	if (last_width > max_width_last) {
		return false;
	}

	if (tile_count > 2U) {

		if (middle_width < min_width) {
			return false;
		}

		if (middle_width > max_width_middle) {
			return false;
		}
	}

	tiling->tile_width_first = first_width;
	tiling->tile_width_middle = middle_width;
	tiling->tiles_in_slice = tile_count;

	return true;
}

__attribute__((unused))
static bool isp5_find_tile_width_dpcm(const struct isp5_program *prg,
					const struct isp_capture_descriptor *cd,
					struct isp5_tile_width *tiling)
{
	const uint16_t alignment = isp5_max_u16(prg->overfetch.alignment, 8);

	if (alignment == 0) {
		return false;
	}

	const uint16_t max_width_first = isp5_align_down(ISP5_MAX_TILE_WIDTH - prg->overfetch.right,
						alignment);
	const uint16_t max_width_middle = isp5_align_down(ISP5_MAX_TILE_WIDTH - prg->overfetch.right -
						prg->overfetch.left,
						alignment);
	const uint16_t max_width_last = ISP5_MAX_TILE_WIDTH - prg->overfetch.left;
	const uint16_t min_width = isp5_max_u16(ISP5_MIN_TILE_WIDTH, prg->overfetch.right);

	if (cd->surface_configs.chunk_width_middle > max_width_middle) {
		return false;
	}

	tiling->tile_width_middle = cd->surface_configs.chunk_width_middle;

	/*
	 * Width of first tile must set so that left overfetch area of
	 * 2nd tile fits into 2nd chunk.
	 */
	tiling->tile_width_first = isp5_align_up(cd->surface_configs.chunk_width_first +
						prg->overfetch.left +
						prg->overfetch.right -
						prg->overfetch.pru_ovf_h, alignment) -
					prg->overfetch.right +
					prg->overfetch.pru_ovf_h;

	if (tiling->tile_width_first < min_width) {
		return false;
	}

	if (tiling->tile_width_first > max_width_first) {
		return false;
	}

	if (tiling->tile_width_first + prg->overfetch.right >
		cd->surface_configs.chunk_width_first + cd->surface_configs.chunk_overfetch_width) {
		return false;
	}

	tiling->tiles_in_slice = 1U + isp5_div_round_up(cd->surface_configs.mr_width -
						cd->surface_configs.chunk_width_first,
						cd->surface_configs.chunk_width_middle);
	const uint16_t last_width = cd->surface_configs.mr_width -
					tiling->tile_width_first -
					(tiling->tiles_in_slice - 1) * tiling->tile_width_middle;

	if (last_width < min_width) {
		return false;
	}

	if (last_width > max_width_last) {
		return false;
	}

	return true;
}

__attribute__((unused))
static bool isp5_find_slice_height(uint16_t img_height,
				struct isp5_slice_height *slicing)
{
	if (img_height < ISP5_MIN_SLICE_HEIGHT) {
		return false;
	}

	if (img_height % 2 != 0U) {
		return false;
	}

	if (img_height <= ISP5_MAX_SLICE_HEIGHT) {
		slicing->slices_in_image = U16_C(1);
		slicing->slice_height = img_height;
		slicing->vi_first_slice_height = img_height;
		return true;
	}

	uint16_t slice_height = ISP5_MAX_SLICE_HEIGHT;
	uint16_t slice_count = isp5_div_round_up(img_height, ISP5_MAX_SLICE_HEIGHT);
	uint16_t last_height = img_height - ISP5_MAX_SLICE_HEIGHT * (slice_count - 1);

	if (last_height < ISP5_MIN_SLICE_HEIGHT) {
		const uint16_t corr = ISP5_MIN_SLICE_HEIGHT - last_height;
		const uint16_t slice_corr = isp5_align_up(isp5_div_round_up(corr, slice_count - 1), 2U);

		slice_height -= slice_corr;
	}

	slicing->slice_height = slice_height;
	slicing->slices_in_image = slice_count;
	slicing->vi_first_slice_height = (slice_count == 1U) ? slice_height : slice_height + 18U;

	return true;
}

#endif /* INCLUDE_CAMRTC_ISP5_TILING_H */