cmd/amlogic/cmd_osd.c - yocto/uboot - Git Browser for ODROID

 // 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"
 );
	// 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"
	);