cmd/amlogic/cmd_osd.c

// SPDX-License-Identifier: (GPL-2.0+ OR MIT)
/*
 * Copyright (c) 2019 Amlogic, Inc. All rights reserved.
 */

#include <common.h>
#include <cpu_func.h>
#include <amlogic/lcd.h>
#include <command.h>
#include <asm/byteorder.h>
#include <malloc.h>
#include <splash.h>
#include <image.h>
#include <amlogic/video_fb.h>
#include <amlogic/video.h>
#include <amlogic/fb.h>
#ifdef CONFIG_AML_HDMITX
#include <amlogic/media/vout/hdmitx21/hdmitx_ext.h>
#endif

int osd_enabled = 0;
/* Graphic Device */
static GraphicDevice *gdev = NULL;

static int do_osd_open(cmd_tbl_t *cmdtp, int flag, int argc,
		       char *const argv[])
{
	char *s;
	uint fb_for_4k2k = 0;

	s = env_get("fb_for_4k2k");
	if (s) {
		fb_for_4k2k = simple_strtoul(s, NULL, 10) ? 1 : 0;
		printf("[OSD]using fb_for_4k2k %d\n", fb_for_4k2k);
	}
	osd_set_4k2k_fb_mode_hw(fb_for_4k2k);

	gdev = video_hw_init(RECT_MODE);
	if (gdev == NULL) {
		printf("Initialize video device failed!\n");
		return 1;
	}
	osd_enabled = 1;
	return 0;
}

static int do_osd_enable(cmd_tbl_t *cmdtp, int flag, int argc,
			char *const argv[])
{
	int index = 0;

	index = get_osd_layer();
	if (index >= 0)
		osd_enable_hw(index, 1);

	return 0;
}

static int do_osd_close(cmd_tbl_t *cmdtp, int flag, int argc,
			char *const argv[])
{

	uint index = 0;

	index = get_osd_layer();

	if (gdev == NULL)
		return 1;

	gdev = NULL;
	if (is_vppx(index)) {
		osd_enable_hw(index, 0);
		if (get_osd_viux_scale_cap())
			osd_set_free_scale_enable_hw(index, 0);
	} else {
		osd_enable_hw(OSD1, 0);
		osd_enable_hw(OSD2, 0);
		osd_set_free_scale_enable_hw(OSD1, 0);
		osd_set_free_scale_enable_hw(OSD2, 0);
	}

	osd_enabled = 0;

	return 0;
}

static int do_osd_clear(cmd_tbl_t *cmdtp, int flag, int argc,
			char *const argv[])
{
#ifdef OSD_SCALE_ENABLE
	uint index = 0;
	ulong fb_addr;
	ulong fb_len;
#endif
	if (gdev == NULL) {
		printf("Please enable osd device first!\n");
		return 1;
	}

#ifdef OSD_SCALE_ENABLE
	index = get_osd_layer();

	fb_addr = (ulong)gdev->frameAdrs + get_fb_offset(index);
	fb_len = get_fb_len(index);
	memset((void *)fb_addr, 0, fb_len);
	flush_cache(fb_addr, fb_len);
#else
	fb_len = CANVAS_ALIGNED(gdev->winSizeX * gdev->gdfBytesPP) *
			gdev->winSizeY;
	memset((void *)(long long)(gdev->frameAdrs), 0, fb_len);

	flush_cache(gdev->frameAdrs, fb_len);
#endif
	return 0;
}

static int do_osd_debug(cmd_tbl_t *cmdtp, int flag, int argc,
			char *const argv[])
{
	int ret = 0;
	int level = 0;

	switch (argc) {
	case 1:
		osd_debug();
		break;
	case 2:
		level = simple_strtoul(argv[1], NULL, 10);
		osd_set_log_level(level);
		break;
	default:
		return CMD_RET_USAGE;
	}

	return ret;
}

static int do_osd_test(cmd_tbl_t *cmdtp, int flag, int argc, char *const argv[])
{
	int ret = 0;
	switch (argc) {
	case 1:
		osd_test();
		break;
	case 2:
		osd_rma_test(simple_strtoul(argv[1], NULL, 10));
		break;
	case 3:
		osd_rma_test_with_addr(simple_strtoul(argv[1], NULL, 16),
			simple_strtoul(argv[2], NULL, 16));
		break;
	default:
		return CMD_RET_USAGE;
	}
	return ret;
}

static int do_osd_display(cmd_tbl_t *cmdtp, int flag, int argc,
			  char *const argv[])
{
	int ret = 0;
	int x = 0, y = 0;
	ulong addr;

	if (gdev == NULL) {
		printf("do_osd_display, enable osd device first!\n");
		return 1;
	}

	splash_get_pos(&x, &y);

	switch (argc) {
	case 1:		/* use load_addr as default address */
		addr = image_load_addr;
		break;
	case 2:		/* use argument */
		addr = simple_strtoul(argv[1], NULL, 16);
		break;
	case 4:
		addr = simple_strtoul(argv[1], NULL, 16);
		x = simple_strtoul(argv[2], NULL, 10);
		y = simple_strtoul(argv[3], NULL, 10);
		break;
	default:
		return CMD_RET_USAGE;
	}

	ret = video_display_bitmap((unsigned long)addr, x, y);

	return ret;
}

static int do_osd_set(cmd_tbl_t *cmdtp, int flag, int argc,
			  char *const argv[])
{
	int i;
	ulong osdID;
	char *str = NULL;
	char *hist_env_key[12] = {"hist_max_min_osd0","hist_spl_val_osd0","hist_spl_pix_cnt_osd0","hist_cheoma_sum_osd0",
	                         "hist_max_min_osd1","hist_spl_val_osd1","hist_spl_pix_cnt_osd1","hist_cheoma_sum_osd1",
							 "hist_max_min_osd2","hist_spl_val_osd2","hist_spl_pix_cnt_osd2","hist_cheoma_sum_osd2"};
	if (argc != 6) {
		return CMD_RET_USAGE;
	}
	osdID = simple_strtoul(argv[1], NULL, 10);

	if (osdID > 2) {
		printf("=== osdID is wrong. ===\n");
		return 1;
	}

	for (i = osdID * 4; i < (osdID + 1) * 4; i++) {
		str = env_get(hist_env_key[i]);
		if (str) {
			env_set(hist_env_key[i], argv[i%4+2]);
			printf("set %s : %s\n", hist_env_key[i], env_get(hist_env_key[i]));
		}
	}
	return 0;
}

static int do_osd_get(cmd_tbl_t *cmdtp, int flag, int argc,
			  char *const argv[])
{
	int i;
	char *str = NULL;
	char *hist_env_key[12] = {"hist_max_min_osd0","hist_spl_val_osd0","hist_spl_pix_cnt_osd0","hist_cheoma_sum_osd0",
	                         "hist_max_min_osd1","hist_spl_val_osd1","hist_spl_pix_cnt_osd1","hist_cheoma_sum_osd1",
							 "hist_max_min_osd2","hist_spl_val_osd2","hist_spl_pix_cnt_osd2","hist_cheoma_sum_osd2"};
	for (i = 0; i < 12; i++) {
		str = env_get(hist_env_key[i]);
		if (str)
			printf("%s : %s\n", hist_env_key[i], str);
	}

	return 0;
}

static int do_osd_dual_logo(cmd_tbl_t *cmdtp, int flag, int argc,
			    char *const argv[])
{
#ifdef CONFIG_AML_HDMITX
	int st = 0;

	/* detect hdmi plugin or not */
	st = hdmitx_get_hpd_state_ext();
	printf("osd: hpd_state=%c\n", st ? '1' : '0');

	if (env_get("set_logo_on") && strchr(env_get("set_logo_on"), '3'))
		env_set("logo3_on", "on");
	else
		env_set("logo3_on", "off");
	if (env_get("set_logo_on") && strchr(env_get("set_logo_on"), '2'))
		env_set("logo2_on", "on");
	else
		env_set("logo2_on", "off");
	if (env_get("set_logo_on") && strchr(env_get("set_logo_on"), '1'))
		env_set("logo1_on", "on");
	else
		env_set("logo1_on", "off");

	if (st) {
		/* hdmi plugin, dual logo display
		 * CONFIG_RECOVERY_DUAL_LOGO is given priority in recovery
		 */
		if (env_get("reboot_mode") && !strncmp(env_get("reboot_mode"),
					"factory_reset", 13)) {
		#if defined(CONFIG_RECOVERY_DUAL_LOGO)
			run_command(CONFIG_RECOVERY_DUAL_LOGO, 0);
		#else
		#if defined(CONFIG_DUAL_LOGO)
			printf("osd: use dual logo cmd macro in recovery mode\n");
			run_command(CONFIG_DUAL_LOGO, 0);
		#else
			printf("osd: dual logo cmd macro is not defined in recovery mode\n");
		#endif
		#endif
		} else {
		#if defined(CONFIG_DUAL_LOGO)
			run_command(CONFIG_DUAL_LOGO, 0);
		#else
			printf("osd: dual logo cmd macro is not defined\n");
		#endif
		}
	} else {
		/* hdmi plugout, single logo display */
	#if defined(CONFIG_SINGLE_LOGO)
		run_command(CONFIG_SINGLE_LOGO, 0);
	#else
		printf("osd: single logo cmd macro is not defined\n");
	#endif
	}
#else
	printf("osd: no hdmitx_device defined\n");
	#if defined(CONFIG_SINGLE_LOGO)
		run_command(CONFIG_SINGLE_LOGO, 0);
	#else
		printf("osd: single logo cmd macro is not defined\n");
	#endif
#endif

	return 0;
}

static cmd_tbl_t cmd_osd_sub[] = {
	U_BOOT_CMD_MKENT(open, 2, 0, do_osd_open, "", ""),
	U_BOOT_CMD_MKENT(enable, 2, 0, do_osd_enable, "", ""),
	U_BOOT_CMD_MKENT(close, 2, 0, do_osd_close, "", ""),
	U_BOOT_CMD_MKENT(clear, 2, 0, do_osd_clear, "", ""),
	U_BOOT_CMD_MKENT(debug, 2, 0, do_osd_debug, "", ""),
	U_BOOT_CMD_MKENT(test, 2, 0, do_osd_test, "", ""),
	U_BOOT_CMD_MKENT(display, 5, 0, do_osd_display, "", ""),
	U_BOOT_CMD_MKENT(set, 7, 0, do_osd_set, "", ""),
	U_BOOT_CMD_MKENT(get, 2, 0, do_osd_get, "", ""),
	U_BOOT_CMD_MKENT(dual_logo, 2, 0, do_osd_dual_logo, "", ""),
};

static int do_osd(cmd_tbl_t *cmdtp, int flag, int argc, char *const argv[])
{
	cmd_tbl_t *c;

	/* Strip off leading 'osd' command argument */
	argc--;
	argv++;

	c = find_cmd_tbl(argv[0], &cmd_osd_sub[0], ARRAY_SIZE(cmd_osd_sub));

	if (c)
		return  c->cmd(cmdtp, flag, argc, argv);
	else
		return CMD_RET_USAGE;
}

U_BOOT_CMD(
	osd,	7,	1,	do_osd,
	"osd sub-system",
	"open                         - open osd device\n"
	"osd enable                       - enable osd device\n"
	"osd close                        - close osd device\n"
	"osd clear                        - clear osd framebuffer\n"
	"osd debug                        - debug osd device\n"
	"osd test [osdID]                 - test osd device\n"
	"osd display <imageAddr> [x y]    - display image\n"
	"osd set <osdID> <a> <b> <c> <d>  - set Hist GoldenData in env\n"
	"                                        a for hist_max_min\n"
	"                                        b for hist_spl_val\n"
	"                                        c for hist_spl_pix_cnt\n"
	"                                        d for hist_cheoma_sum\n"
	"osd get                          - get Hist GoldenData from env\n"
	"osd dual_logo                    - detect hdmi hpd, then display logo"
);