/*
 * Copyright (c) 2015 Google, Inc
 * (C) Copyright 2001-2015
 * DENX Software Engineering -- wd@denx.de
 * Compulab Ltd - http://compulab.co.il/
 * Bernecker & Rainer Industrieelektronik GmbH - http://www.br-automation.com
 *
 * SPDX-License-Identifier:     GPL-2.0+
 */

#include <common.h>
#include <dm.h>
#include <video.h>
#include <video_console.h>
#include <video_font.h>         /* Get font data, width and height */

/* By default we scroll by a single line */
#ifndef CONFIG_CONSOLE_SCROLL_LINES
#define CONFIG_CONSOLE_SCROLL_LINES 1
#endif

int vidconsole_putc_xy(struct udevice *dev, uint x, uint y, char ch)
{
        struct vidconsole_ops *ops = vidconsole_get_ops(dev);

        if (!ops->putc_xy)
                return -ENOSYS;
        return ops->putc_xy(dev, x, y, ch);
}

int vidconsole_move_rows(struct udevice *dev, uint rowdst, uint rowsrc,
                         uint count)
{
        struct vidconsole_ops *ops = vidconsole_get_ops(dev);

        if (!ops->move_rows)
                return -ENOSYS;
        return ops->move_rows(dev, rowdst, rowsrc, count);
}

int vidconsole_set_row(struct udevice *dev, uint row, int clr)
{
        struct vidconsole_ops *ops = vidconsole_get_ops(dev);

        if (!ops->set_row)
                return -ENOSYS;
        return ops->set_row(dev, row, clr);
}

static int vidconsole_entry_start(struct udevice *dev)
{
        struct vidconsole_ops *ops = vidconsole_get_ops(dev);

        if (!ops->entry_start)
                return -ENOSYS;
        return ops->entry_start(dev);
}

/* Move backwards one space */
static int vidconsole_back(struct udevice *dev)
{
        struct vidconsole_priv *priv = dev_get_uclass_priv(dev);
        struct vidconsole_ops *ops = vidconsole_get_ops(dev);
        int ret;

        if (ops->backspace) {
                ret = ops->backspace(dev);
                if (ret != -ENOSYS)
                        return ret;
        }

        priv->xcur_frac -= VID_TO_POS(priv->x_charsize);
        if (priv->xcur_frac < priv->xstart_frac) {
                priv->xcur_frac = (priv->cols - 1) *
                        VID_TO_POS(priv->x_charsize);
                priv->ycur -= priv->y_charsize;
                if (priv->ycur < 0)
                        priv->ycur = 0;
        }

        return 0;
}

/* Move to a newline, scrolling the display if necessary */
static void vidconsole_newline(struct udevice *dev)
{
        struct vidconsole_priv *priv = dev_get_uclass_priv(dev);
        struct udevice *vid_dev = dev->parent;
        struct video_priv *vid_priv = dev_get_uclass_priv(vid_dev);
        const int rows = CONFIG_CONSOLE_SCROLL_LINES;
        int i;

        priv->xcur_frac = priv->xstart_frac;
        priv->ycur += priv->y_charsize;

        /* Check if we need to scroll the terminal */
        if ((priv->ycur + priv->y_charsize) / priv->y_charsize > priv->rows) {
                vidconsole_move_rows(dev, 0, rows, priv->rows - rows);
                for (i = 0; i < rows; i++)
                        vidconsole_set_row(dev, priv->rows - i - 1,
                                           vid_priv->colour_bg);
                priv->ycur -= rows * priv->y_charsize;
        }
        priv->last_ch = 0;

        video_sync(dev->parent);
}

int vidconsole_put_char(struct udevice *dev, char ch)
{
        struct vidconsole_priv *priv = dev_get_uclass_priv(dev);
        int ret;

        switch (ch) {
        case '\a':
                /* beep */
                break;
        case '\r':
                priv->xcur_frac = priv->xstart_frac;
                break;
        case '\n':
                vidconsole_newline(dev);
                vidconsole_entry_start(dev);
                break;
        case '\t':      /* Tab (8 chars alignment) */
                priv->xcur_frac = ((priv->xcur_frac / priv->tab_width_frac)
                                + 1) * priv->tab_width_frac;

                if (priv->xcur_frac >= priv->xsize_frac)
                        vidconsole_newline(dev);
                break;
        case '\b':
                vidconsole_back(dev);
                priv->last_ch = 0;
                break;
        default:
                /*
                 * Failure of this function normally indicates an unsupported
                 * colour depth. Check this and return an error to help with
                 * diagnosis.
                 */
                ret = vidconsole_putc_xy(dev, priv->xcur_frac, priv->ycur, ch);
                if (ret == -EAGAIN) {
                        vidconsole_newline(dev);
                        ret = vidconsole_putc_xy(dev, priv->xcur_frac,
                                                 priv->ycur, ch);
                }
                if (ret < 0)
                        return ret;
                priv->xcur_frac += ret;
                priv->last_ch = ch;
                if (priv->xcur_frac >= priv->xsize_frac)
                        vidconsole_newline(dev);
                break;
        }

        return 0;
}

static void vidconsole_putc(struct stdio_dev *sdev, const char ch)
{
        struct udevice *dev = sdev->priv;

        vidconsole_put_char(dev, ch);
}

static void vidconsole_puts(struct stdio_dev *sdev, const char *s)
{
        struct udevice *dev = sdev->priv;

        while (*s)
                vidconsole_put_char(dev, *s++);
        video_sync(dev->parent);
}

/* Set up the number of rows and colours (rotated drivers override this) */
static int vidconsole_pre_probe(struct udevice *dev)
{
        struct vidconsole_priv *priv = dev_get_uclass_priv(dev);
        struct udevice *vid = dev->parent;
        struct video_priv *vid_priv = dev_get_uclass_priv(vid);

        priv->xsize_frac = VID_TO_POS(vid_priv->xsize);

        return 0;
}

/* Register the device with stdio */
static int vidconsole_post_probe(struct udevice *dev)
{
        struct vidconsole_priv *priv = dev_get_uclass_priv(dev);
        struct stdio_dev *sdev = &priv->sdev;
        int ret;

        if (!priv->tab_width_frac)
                priv->tab_width_frac = VID_TO_POS(priv->x_charsize) * 8;

        if (dev->seq) {
                snprintf(sdev->name, sizeof(sdev->name), "vidconsole%d",
                         dev->seq);
        } else {
                strcpy(sdev->name, "vidconsole");
        }

        sdev->flags = DEV_FLAGS_OUTPUT;
        sdev->putc = vidconsole_putc;
        sdev->puts = vidconsole_puts;
        sdev->priv = dev;
        ret = stdio_register(sdev);
        if (ret)
                return ret;

        return 0;
}

UCLASS_DRIVER(vidconsole) = {
        .id             = UCLASS_VIDEO_CONSOLE,
        .name           = "vidconsole0",
        .pre_probe      = vidconsole_pre_probe,
        .post_probe     = vidconsole_post_probe,
        .per_device_auto_alloc_size     = sizeof(struct vidconsole_priv),
};

void vidconsole_position_cursor(struct udevice *dev, unsigned col, unsigned row)
{
        struct vidconsole_priv *priv = dev_get_uclass_priv(dev);
        struct udevice *vid_dev = dev->parent;
        struct video_priv *vid_priv = dev_get_uclass_priv(vid_dev);

        priv->xcur_frac = VID_TO_POS(min_t(short, col, vid_priv->xsize - 1));
        priv->ycur = min_t(short, row, vid_priv->ysize - 1);
}

static int do_video_setcursor(cmd_tbl_t *cmdtp, int flag, int argc,
                              char *const argv[])
{
        unsigned int col, row;
        struct udevice *dev;

        if (argc != 3)
                return CMD_RET_USAGE;

        if (uclass_first_device_err(UCLASS_VIDEO_CONSOLE, &dev))
                return CMD_RET_FAILURE;
        col = simple_strtoul(argv[1], NULL, 10);
        row = simple_strtoul(argv[2], NULL, 10);
        vidconsole_position_cursor(dev, col, row);

        return 0;
}

static int do_video_puts(cmd_tbl_t *cmdtp, int flag, int argc,
                         char *const argv[])
{
        struct udevice *dev;
        const char *s;

        if (argc != 2)
                return CMD_RET_USAGE;

        if (uclass_first_device_err(UCLASS_VIDEO_CONSOLE, &dev))
                return CMD_RET_FAILURE;
        for (s = argv[1]; *s; s++)
                vidconsole_put_char(dev, *s);

        return 0;
}

U_BOOT_CMD(
        setcurs, 3,     1,      do_video_setcursor,
        "set cursor position within screen",
        "    <col> <row> in character"
);

U_BOOT_CMD(
        lcdputs, 2,     1,      do_video_puts,
        "print string on video framebuffer",
        "    <string>"
);