common/cmd_ui.c - u-boot - Git at Google

 /*
 * Copyright (C) 2017 Amlogic, Inc. All rights reserved.
 * *
 This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 * *
 This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
 * more details.
 * *
 You should have received a copy of the GNU General Public License along
 * with this program; if not, write to the Free Software Foundation, Inc.,
 * 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
 * *
 Description:
 */

 #include <common.h>
 #include <command.h>
 #include <malloc.h>
 #include <video_fb.h>
 #include <video.h>


 int screen_init(void);
 void screen_uninit(void);
 int gr_init_ext_font(const char* font, GRFont** dest);
 int surface_loadbmp(GRSurface** surface, const char* filename);
 void surface_disaplay(GRSurface* surface, int sx, int sy, int dx, int dy);
 void screen_setcolor(unsigned int color);
 void screen_drawtextline(const GRFont* font, int x, int y, const char *s, bool bold);
 void screen_fillrect(int x, int y, int w, int h);
 void screen_update(void);
 const GRFont* gr_sys_font(void);
 void set_fastboot_flag(int flag);

 static int ui_inited = 0;
 extern unsigned int ui_log_level;

 typedef struct {
 	GRFont* font;
 	const char *font_name;
 } LIB_FONT;

 typedef struct {
 	GRSurface* surface;
 	const char *pic_name;
 } PIC_SURFACE;

 LIB_FONT lib_font[] = {{NULL, "font"}};
 PIC_SURFACE pic_surface[] = {{NULL, "bootup"}, {NULL, "upgrade_error"}, {NULL, "upgrade_fail"}, {NULL, "upgrade_success"}};

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

 	if (!ui_inited) {
 		set_fastboot_flag(1);

 		screen_init();

 		for (i =0; i < ARRAY_SIZE(lib_font); i++) {
 			gr_init_ext_font(lib_font[i].font_name, &lib_font[i].font);
 		}

 		for (i =0; i < ARRAY_SIZE(pic_surface); i++) {
 			surface_loadbmp(&pic_surface[i].surface, pic_surface[i].pic_name);
 		}
 	}

 	ui_inited = 1;

 	return 0;
 }

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

 	if (!ui_inited) {
 		printf("error: ui init first\n");
 		return -1;
 	} else if (argc != 6) {
 		return -1;
 	} else {
 		for (i =0; i < ARRAY_SIZE(pic_surface); i++) {
 			if (!strcmp(argv[1], pic_surface[i].pic_name))
 				if (pic_surface[i].surface) {
 					surface_disaplay(pic_surface[i].surface,
 							simple_strtoul(argv[2], NULL, 10), simple_strtoul(argv[3], NULL, 10),
 							simple_strtoul(argv[4], NULL, 10), simple_strtoul(argv[5], NULL, 10));
 					break;
 				}
 		}
 	}

 	if (i == ARRAY_SIZE(pic_surface)) {
 		printf("error: blit pic %s to framebuffer fail\n", argv[1]);
 		return -1;
 	}

 	return 0;
 }

 static int do_ui_setcolor(cmd_tbl_t *cmdtp, int flag, int argc,
 		       char *const argv[])
 {
 	if (!ui_inited) {
 		printf("error: ui init first\n");
 		return -1;
 	} else {
 		if (argc != 2)
 			return -1;
 		else
 			screen_setcolor(simple_strtoul(argv[1], NULL, 16));
 	}

 	return 0;
 }

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

 	if (!ui_inited) {
 		printf("error: ui init first\n");
 		return -1;
 	} else if (argc < 5) {
 			return -1;
 	}	else {
 		for (i =0; i < ARRAY_SIZE(lib_font); i++) {
 			if (!strcmp(argv[1], lib_font[i].font_name))
 				if (lib_font[i].font) {
 					screen_drawtextline(lib_font[i].font,
 								simple_strtoul(argv[3], NULL, 10), simple_strtoul(argv[4], NULL, 10),
 								argv[2], (argc >= 6 ? (simple_strtoul(argv[5], NULL, 10) == 1 ? 1 : 0) : 0));
 					break;
 				}
 		}
 	}

 	if (i == ARRAY_SIZE(lib_font)) {
 		printf("error: use external font lib %s fail, use default font lib\n", argv[1]);
 		screen_drawtextline(gr_sys_font(),
 					simple_strtoul(argv[3], NULL, 10), simple_strtoul(argv[4], NULL, 10),
 					argv[2], (argc >= 6 ? (simple_strtoul(argv[5], NULL, 10) == 1 ? 1 : 0) : 0));
 	}

 	return 0;
 }

 static int do_ui_rect(cmd_tbl_t *cmdtp, int flag, int argc,
 		       char *const argv[])
 {
 	if (!ui_inited) {
 		printf("error: ui init first\n");
 		return -1;
 	} else if (argc != 5){
 		return -1;
 	} else {
 		screen_fillrect(simple_strtoul(argv[1], NULL, 10), simple_strtoul(argv[2], NULL, 10),
 				simple_strtoul(argv[3], NULL, 10), simple_strtoul(argv[4], NULL, 10));
 	}

 	return 0;
 }

 static int do_ui_update(cmd_tbl_t *cmdtp, int flag, int argc,
 		       char *const argv[])
 {
 	if (!ui_inited) {
 		printf("error: ui init first\n");
 		return -1;
 	} else {
 		screen_update();
 	}

 	return 0;
 }

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

 	if (!ui_inited) {
 		printf("error: ui init first\n");
 		return -1;
 	} else {
 		for (i =0; i < ARRAY_SIZE(lib_font); i++) {
 			if (lib_font[i].font) {
 				if (lib_font[i].font->texture) {
 					free(lib_font[i].font->texture);
 					lib_font[i].font->texture = NULL;
 				}
 				free(lib_font[i].font);
 				lib_font[i].font = NULL;
 			}
 		}

 		for (i =0; i < ARRAY_SIZE(pic_surface); i++) {
 			if (pic_surface[i].surface) {
 				free(pic_surface[i].surface);
 				pic_surface[i].surface = NULL;
 			}
 		}

 		set_fastboot_flag(0);
 		screen_uninit();

 		ui_inited = 0;
 	}

 	return 0;
 }

 static int do_ui_debug(cmd_tbl_t *cmdtp, int flag, int argc,
 		       char *const argv[])
 {
 	if (argc != 2)
 		return -1;
 	else
 		ui_log_level = simple_strtoul(argv[1], NULL, 10);

 	return 0;
 }


 static cmd_tbl_t cmd_ui_sub[] = {
 	U_BOOT_CMD_MKENT(init, 2, 0, do_ui_init, "", ""),
 	U_BOOT_CMD_MKENT(pic, 9, 0, do_ui_pic, "", ""),
 	U_BOOT_CMD_MKENT(setcolor, 3, 0, do_ui_setcolor, "", ""),
 	U_BOOT_CMD_MKENT(text, 7, 0, do_ui_text, "", ""),
 	U_BOOT_CMD_MKENT(rect, 6, 0, do_ui_rect, "", ""),
 	U_BOOT_CMD_MKENT(update, 2, 0, do_ui_update, "", ""),
 	U_BOOT_CMD_MKENT(uinit, 2, 0, do_ui_uinit, "", ""),
 	U_BOOT_CMD_MKENT(debug, 2, 0, do_ui_debug, "", ""),
 };

 static int do_ui(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_ui_sub[0], ARRAY_SIZE(cmd_ui_sub));

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

 U_BOOT_CMD(
 	ui,	9,	1,	do_ui,
 	"ui sub-system",
 	"init                                        - init device and init default font lib\n"
 	"ui pic <name> <srcX srcY> <destX destY>        - blit pic to framebuffer\n"
 	"ui setcolor  <color>                           - set color\n"
 	"ui text <lib_font> <string> <x y> [bold]       - fill text to framebuffer\n"
 	"ui rect <x y> <w h>                            - fill rect to framebuffer\n"
 	"ui update                                      - update framebuffer to screen\n"
 	"uinit                                          - uinit device and free space\n"
 	"ui debug <level>                               - set log level\n"
 );
	/*
	* Copyright (C) 2017 Amlogic, Inc. All rights reserved.
	* *
	This program is free software; you can redistribute it and/or modify
	* it under the terms of the GNU General Public License as published by
	* the Free Software Foundation; either version 2 of the License, or
	* (at your option) any later version.
	* *
	This program is distributed in the hope that it will be useful, but WITHOUT
	* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
	* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
	* more details.
	* *
	You should have received a copy of the GNU General Public License along
	* with this program; if not, write to the Free Software Foundation, Inc.,
	* 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
	* *
	Description:
	*/

	#include <common.h>
	#include <command.h>
	#include <malloc.h>
	#include <video_fb.h>
	#include <video.h>


	int screen_init(void);
	void screen_uninit(void);
	int gr_init_ext_font(const char* font, GRFont** dest);
	int surface_loadbmp(GRSurface** surface, const char* filename);
	void surface_disaplay(GRSurface* surface, int sx, int sy, int dx, int dy);
	void screen_setcolor(unsigned int color);
	void screen_drawtextline(const GRFont* font, int x, int y, const char *s, bool bold);
	void screen_fillrect(int x, int y, int w, int h);
	void screen_update(void);
	const GRFont* gr_sys_font(void);
	void set_fastboot_flag(int flag);

	static int ui_inited = 0;
	extern unsigned int ui_log_level;

	typedef struct {
	GRFont* font;
	const char *font_name;
	} LIB_FONT;

	typedef struct {
	GRSurface* surface;
	const char *pic_name;
	} PIC_SURFACE;

	LIB_FONT lib_font[] = {{NULL, "font"}};
	PIC_SURFACE pic_surface[] = {{NULL, "bootup"}, {NULL, "upgrade_error"}, {NULL, "upgrade_fail"}, {NULL, "upgrade_success"}};

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

	if (!ui_inited) {
	set_fastboot_flag(1);

	screen_init();

	for (i =0; i < ARRAY_SIZE(lib_font); i++) {
	gr_init_ext_font(lib_font[i].font_name, &lib_font[i].font);
	}

	for (i =0; i < ARRAY_SIZE(pic_surface); i++) {
	surface_loadbmp(&pic_surface[i].surface, pic_surface[i].pic_name);
	}
	}

	ui_inited = 1;

	return 0;
	}

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

	if (!ui_inited) {
	printf("error: ui init first\n");
	return -1;
	} else if (argc != 6) {
	return -1;
	} else {
	for (i =0; i < ARRAY_SIZE(pic_surface); i++) {
	if (!strcmp(argv[1], pic_surface[i].pic_name))
	if (pic_surface[i].surface) {
	surface_disaplay(pic_surface[i].surface,
	simple_strtoul(argv[2], NULL, 10), simple_strtoul(argv[3], NULL, 10),
	simple_strtoul(argv[4], NULL, 10), simple_strtoul(argv[5], NULL, 10));
	break;
	}
	}
	}

	if (i == ARRAY_SIZE(pic_surface)) {
	printf("error: blit pic %s to framebuffer fail\n", argv[1]);
	return -1;
	}

	return 0;
	}

	static int do_ui_setcolor(cmd_tbl_t *cmdtp, int flag, int argc,
	char *const argv[])
	{
	if (!ui_inited) {
	printf("error: ui init first\n");
	return -1;
	} else {
	if (argc != 2)
	return -1;
	else
	screen_setcolor(simple_strtoul(argv[1], NULL, 16));
	}

	return 0;
	}

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

	if (!ui_inited) {
	printf("error: ui init first\n");
	return -1;
	} else if (argc < 5) {
	return -1;
	} else {
	for (i =0; i < ARRAY_SIZE(lib_font); i++) {
	if (!strcmp(argv[1], lib_font[i].font_name))
	if (lib_font[i].font) {
	screen_drawtextline(lib_font[i].font,
	simple_strtoul(argv[3], NULL, 10), simple_strtoul(argv[4], NULL, 10),
	argv[2], (argc >= 6 ? (simple_strtoul(argv[5], NULL, 10) == 1 ? 1 : 0) : 0));
	break;
	}
	}
	}

	if (i == ARRAY_SIZE(lib_font)) {
	printf("error: use external font lib %s fail, use default font lib\n", argv[1]);
	screen_drawtextline(gr_sys_font(),
	simple_strtoul(argv[3], NULL, 10), simple_strtoul(argv[4], NULL, 10),
	argv[2], (argc >= 6 ? (simple_strtoul(argv[5], NULL, 10) == 1 ? 1 : 0) : 0));
	}

	return 0;
	}

	static int do_ui_rect(cmd_tbl_t *cmdtp, int flag, int argc,
	char *const argv[])
	{
	if (!ui_inited) {
	printf("error: ui init first\n");
	return -1;
	} else if (argc != 5){
	return -1;
	} else {
	screen_fillrect(simple_strtoul(argv[1], NULL, 10), simple_strtoul(argv[2], NULL, 10),
	simple_strtoul(argv[3], NULL, 10), simple_strtoul(argv[4], NULL, 10));
	}

	return 0;
	}

	static int do_ui_update(cmd_tbl_t *cmdtp, int flag, int argc,
	char *const argv[])
	{
	if (!ui_inited) {
	printf("error: ui init first\n");
	return -1;
	} else {
	screen_update();
	}

	return 0;
	}

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

	if (!ui_inited) {
	printf("error: ui init first\n");
	return -1;
	} else {
	for (i =0; i < ARRAY_SIZE(lib_font); i++) {
	if (lib_font[i].font) {
	if (lib_font[i].font->texture) {
	free(lib_font[i].font->texture);
	lib_font[i].font->texture = NULL;
	}
	free(lib_font[i].font);
	lib_font[i].font = NULL;
	}
	}

	for (i =0; i < ARRAY_SIZE(pic_surface); i++) {
	if (pic_surface[i].surface) {
	free(pic_surface[i].surface);
	pic_surface[i].surface = NULL;
	}
	}

	set_fastboot_flag(0);
	screen_uninit();

	ui_inited = 0;
	}

	return 0;
	}

	static int do_ui_debug(cmd_tbl_t *cmdtp, int flag, int argc,
	char *const argv[])
	{
	if (argc != 2)
	return -1;
	else
	ui_log_level = simple_strtoul(argv[1], NULL, 10);

	return 0;
	}


	static cmd_tbl_t cmd_ui_sub[] = {
	U_BOOT_CMD_MKENT(init, 2, 0, do_ui_init, "", ""),
	U_BOOT_CMD_MKENT(pic, 9, 0, do_ui_pic, "", ""),
	U_BOOT_CMD_MKENT(setcolor, 3, 0, do_ui_setcolor, "", ""),
	U_BOOT_CMD_MKENT(text, 7, 0, do_ui_text, "", ""),
	U_BOOT_CMD_MKENT(rect, 6, 0, do_ui_rect, "", ""),
	U_BOOT_CMD_MKENT(update, 2, 0, do_ui_update, "", ""),
	U_BOOT_CMD_MKENT(uinit, 2, 0, do_ui_uinit, "", ""),
	U_BOOT_CMD_MKENT(debug, 2, 0, do_ui_debug, "", ""),
	};

	static int do_ui(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_ui_sub[0], ARRAY_SIZE(cmd_ui_sub));

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

	U_BOOT_CMD(
	ui, 9, 1, do_ui,
	"ui sub-system",
	"init - init device and init default font lib\n"
	"ui pic <name> <srcX srcY> <destX destY> - blit pic to framebuffer\n"
	"ui setcolor <color> - set color\n"
	"ui text <lib_font> <string> <x y> [bold] - fill text to framebuffer\n"
	"ui rect <x y> <w h> - fill rect to framebuffer\n"
	"ui update - update framebuffer to screen\n"
	"uinit - uinit device and free space\n"
	"ui debug <level> - set log level\n"
	);