/*      $NetBSD: rcons_subr.c,v 1.18 2009/03/14 21:04:22 dsl Exp $ */

/*
 * Copyright (c) 1991, 1993
 *      The Regents of the University of California.  All rights reserved.
 *
 * This software was developed by the Computer Systems Engineering group
 * at Lawrence Berkeley Laboratory under DARPA contract BG 91-66 and
 * contributed to Berkeley.
 *
 * All advertising materials mentioning features or use of this software
 * must display the following acknowledgement:
 *      This product includes software developed by the University of
 *      California, Lawrence Berkeley Laboratory.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 * 3. Neither the name of the University nor the names of its contributors
 *    may be used to endorse or promote products derived from this software
 *    without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 *
 *      @(#)rcons_subr.c        8.1 (Berkeley) 6/11/93
 */

#include <sys/cdefs.h>
__KERNEL_RCSID(0, "$NetBSD: rcons_subr.c,v 1.18 2009/03/14 21:04:22 dsl Exp $");

#include <sys/param.h>
#ifdef _KERNEL
#include <sys/device.h>
#include <sys/systm.h>
#else
#include "myfbdevice.h"
#endif

#include <dev/rcons/rcons.h>
#include <dev/wscons/wsdisplayvar.h>

extern void rcons_bell(struct rconsole *);

#if 0
#define RCONS_ISPRINT(c) ((((c) >= ' ') && ((c) <= '~')) || ((c) > 160))
#else
#define RCONS_ISPRINT(c) (((((c) >= ' ') && ((c) <= '~'))) || ((c) > 127))
#endif
#define RCONS_ISDIGIT(c) ((c) >= '0' && (c) <= '9')

/* Initialize our operations set */
void
rcons_init_ops(struct rconsole *rc)
{
        long tmp;
        int i, m;

        m = sizeof(rc->rc_charmap) / sizeof(rc->rc_charmap[0]);

        for (i = 0; i < m; i++)
                rc->rc_ops->mapchar(rc->rc_cookie, i, rc->rc_charmap + i);

        /* Determine which attributes the device supports. */
#ifdef RASTERCONSOLE_FGCOL
        rc->rc_deffgcolor = RASTERCONSOLE_FGCOL;
#endif
#ifdef RASTERCONSOLE_BGCOL
        rc->rc_defbgcolor = RASTERCONSOLE_BGCOL;
#endif
        rc->rc_fgcolor = rc->rc_deffgcolor;
        rc->rc_bgcolor = rc->rc_defbgcolor;
        rc->rc_supwsflg = 0;

        for (i = 1; i < 256; i <<= 1)
                if (rc->rc_ops->allocattr(rc->rc_cookie, 0, 0, i, &tmp) == 0)
                        rc->rc_supwsflg |= i;

        /* Allocate kernel output attribute */
        rc->rc_wsflg = WSATTR_HILIT;
        rcons_setcolor(rc, rc->rc_deffgcolor, rc->rc_defbgcolor);
        rc->rc_kern_attr = rc->rc_attr;

        rc->rc_wsflg = 0;
        rcons_setcolor(rc, rc->rc_deffgcolor, rc->rc_defbgcolor);
        rc->rc_defattr = rc->rc_attr;
}

/* Output (or at least handle) a string sent to the console */
void
rcons_puts(struct rconsole *rc, const unsigned char *str, int n)
{
        int c, i, j;
        const unsigned char *cp;

        /* Jump scroll */
        /* XXX maybe this should be an option? */
        if ((rc->rc_bits & FB_INESC) == 0) {
                /* Count newlines up to an escape sequence */
                i = 0;
                j = 0;
                for (cp = str; j++ < n && *cp != '\033'; ++cp) {
                        if (*cp == '\n')
                                ++i;
                        else if (*cp == '\013')
                                --i;
                }

                /* Only jump scroll two or more rows */
                if (rc->rc_row + i > rc->rc_maxrow + 1) {
                        /* Erase the cursor (if necessary) */
                        if (rc->rc_bits & FB_CURSOR)
                                rcons_cursor(rc);

                        rcons_scroll(rc, i);
                }
        }

        /* Process characters */
        while (--n >= 0) {
                c = *str;
                if (c == '\033') {
                        /* Start an escape (perhaps aborting one in progress) */
                        rc->rc_bits |= FB_INESC | FB_P0_DEFAULT | FB_P1_DEFAULT;
                        rc->rc_bits &= ~(FB_P0 | FB_P1);

                        /* Most parameters default to 1 */
                        rc->rc_p0 = rc->rc_p1 = 1;
                } else if (rc->rc_bits & FB_INESC) {
                        rcons_esc(rc, c);
                } else {
                        /* Erase the cursor (if necessary) */
                        if (rc->rc_bits & FB_CURSOR)
                                rcons_cursor(rc);

                        /* Display the character */
                        if (RCONS_ISPRINT(c)) {
                                /* Try to output as much as possible */
                                j = rc->rc_maxcol - rc->rc_col;
                                if (j > n)
                                        j = n;
                                for (i = 1; i < j && RCONS_ISPRINT(str[i]); ++i)
                                        continue;
                                rcons_text(rc, str, i);
                                --i;
                                str += i;
                                n -= i;
                        } else
                                rcons_pctrl(rc, c);
                }
                ++str;
        }
        /* Redraw the cursor (if necessary) */
        if ((rc->rc_bits & FB_CURSOR) == 0)
                rcons_cursor(rc);
}


/* Handle a control character sent to the console */
void
rcons_pctrl(struct rconsole *rc, int c)
{

        switch (c) {
        case '\r':      /* Carriage return */
                rc->rc_col = 0;
                break;

        case '\b':      /* Backspace */
                if (rc->rc_col > 0)
                        (rc->rc_col)--;
                break;

        case '\v':      /* Vertical tab */
                if (rc->rc_row > 0)
                        (rc->rc_row)--;
                break;

        case '\f':      /* Formfeed */
                rc->rc_row = rc->rc_col = 0;
                rcons_clear2eop(rc);
                break;

        case '\n':      /* Linefeed */
                (rc->rc_row)++;
                if (rc->rc_row >= rc->rc_maxrow)
                        rcons_scroll(rc, 1);
                break;

        case '\a':      /* Bell */
                rcons_bell(rc);
                break;

        case '\t':      /* Horizontal tab */
                rc->rc_col = (rc->rc_col + 8) & ~7;
                if (rc->rc_col >= rc->rc_maxcol)
                        rc->rc_col = rc->rc_maxcol;
                break;
        }
}

/* Handle the next character in an escape sequence */
void
rcons_esc(struct rconsole *rc, int c)
{

        if (c == '[') {
                /* Parameter 0 */
                rc->rc_bits &= ~FB_P1;
                rc->rc_bits |= FB_P0;
        } else if (c == ';') {
                /* Parameter 1 */
                rc->rc_bits &= ~FB_P0;
                rc->rc_bits |= FB_P1;
        } else if (RCONS_ISDIGIT(c)) {
                /* Add a digit to a parameter */
                if (rc->rc_bits & FB_P0) {
                        /* Parameter 0 */
                        if (rc->rc_bits & FB_P0_DEFAULT) {
                                rc->rc_bits &= ~FB_P0_DEFAULT;
                                rc->rc_p0 = 0;
                        }
                        rc->rc_p0 *= 10;
                        rc->rc_p0 += c - '0';
                } else if (rc->rc_bits & FB_P1) {
                        /* Parameter 1 */
                        if (rc->rc_bits & FB_P1_DEFAULT) {
                                rc->rc_bits &= ~FB_P1_DEFAULT;
                                rc->rc_p1 = 0;
                        }
                        rc->rc_p1 *= 10;
                        rc->rc_p1 += c - '0';
                }
        } else {
                /* Erase the cursor (if necessary) */
                if (rc->rc_bits & FB_CURSOR)
                        rcons_cursor(rc);

                /* Process the completed escape sequence */
                rcons_doesc(rc, c);
                rc->rc_bits &= ~FB_INESC;
        }
}


/* Handle an SGR (Select Graphic Rendition) escape */
void
rcons_sgresc(struct rconsole *rc, int c)
{

        switch (c) {
        /* Clear all attributes || End underline */
        case 0:
                rc->rc_wsflg = 0;
                rc->rc_fgcolor = rc->rc_deffgcolor;
                rc->rc_bgcolor = rc->rc_defbgcolor;
                rc->rc_attr = rc->rc_defattr;
                break;

        /* ANSI foreground color */
        case 30: case 31: case 32: case 33:
        case 34: case 35: case 36: case 37:
                rcons_setcolor(rc, c - 30, rc->rc_bgcolor);
                break;

        /* ANSI background color */
        case 40: case 41: case 42: case 43:
        case 44: case 45: case 46: case 47:
                rcons_setcolor(rc, rc->rc_fgcolor, c - 40);
                break;

        /* Begin reverse */
        case 7:
                rc->rc_wsflg |= WSATTR_REVERSE;
                rcons_setcolor(rc, rc->rc_fgcolor, rc->rc_bgcolor);
                break;

        /* Begin bold */
        case 1:
                rc->rc_wsflg |= WSATTR_HILIT;
                rcons_setcolor(rc, rc->rc_fgcolor, rc->rc_bgcolor);
                break;

        /* Begin underline */
        case 4:
                rc->rc_wsflg |= WSATTR_UNDERLINE;
                rcons_setcolor(rc, rc->rc_fgcolor, rc->rc_bgcolor);
                break;
        }
}


/* Process a complete escape sequence */
void
rcons_doesc(struct rconsole *rc, int c)
{

#ifdef notdef
        /* XXX add escape sequence to enable visual (and audible) bell */
        rc->rc_bits = FB_VISBELL;
#endif

        switch (c) {

        case '@':
                /* Insert Character (ICH) */
                rcons_insertchar(rc, rc->rc_p0);
                break;

        case 'A':
                /* Cursor Up (CUU) */
                if (rc->rc_row >= rc->rc_p0)
                        rc->rc_row -= rc->rc_p0;
                else
                        rc->rc_row = 0;
                break;

        case 'B':
                /* Cursor Down (CUD) */
                rc->rc_row += rc->rc_p0;
                if (rc->rc_row >= rc->rc_maxrow)
                        rc->rc_row = rc->rc_maxrow - 1;
                break;

        case 'C':
                /* Cursor Forward (CUF) */
                rc->rc_col += rc->rc_p0;
                if (rc->rc_col >= rc->rc_maxcol)
                        rc->rc_col = rc->rc_maxcol - 1;
                break;

        case 'D':
                /* Cursor Backward (CUB) */
                if (rc->rc_col >= rc->rc_p0)
                        rc->rc_col -= rc->rc_p0;
                else
                        rc->rc_col = 0;
                break;

        case 'E':
                /* Cursor Next Line (CNL) */
                rc->rc_col = 0;
                rc->rc_row += rc->rc_p0;
                if (rc->rc_row >= rc->rc_maxrow)
                        rc->rc_row = rc->rc_maxrow - 1;
                break;

        case 'f':
                /* Horizontal And Vertical Position (HVP) */
        case 'H':
                /* Cursor Position (CUP) */
                rc->rc_col = MIN(MAX(rc->rc_p1, 1), rc->rc_maxcol) - 1;
                rc->rc_row = MIN(MAX(rc->rc_p0, 1), rc->rc_maxrow) - 1;
                break;

        case 'J':
                /* Erase in Display (ED) */
                rcons_clear2eop(rc);
                break;

        case 'K':
                /* Erase in Line (EL) */
                rcons_clear2eol(rc);
                break;

        case 'L':
                /* Insert Line (IL) */
                rcons_insertline(rc, rc->rc_p0);
                break;

        case 'M':
                /* Delete Line (DL) */
                rcons_delline(rc, rc->rc_p0);
                break;

        case 'P':
                /* Delete Character (DCH) */
                rcons_delchar(rc, rc->rc_p0);
                break;

        case 'm':
                /* Select Graphic Rendition (SGR) */
                /* (defaults to zero) */
                if (rc->rc_bits & FB_P0_DEFAULT)
                        rc->rc_p0 = 0;

                if (rc->rc_bits & FB_P1_DEFAULT)
                        rc->rc_p1 = 0;

                rcons_sgresc(rc, rc->rc_p0);

                if (rc->rc_bits & FB_P1)
                        rcons_sgresc(rc, rc->rc_p1);

                break;

        /*
         * XXX: setting SUNBOW and SUNWOB should probably affect
         * deffgcolor, defbgcolor and defattr too.
         */
        case 'p':
                /* Black On White (SUNBOW) */
                rcons_setcolor(rc, WSCOL_BLACK, WSCOL_WHITE);
                break;

        case 'q':
                /* White On Black (SUNWOB) */
                rcons_setcolor(rc, WSCOL_WHITE, WSCOL_BLACK);
                break;

        case 'r':
                /* Set scrolling (SUNSCRL) */
                /* (defaults to zero) */
                if (rc->rc_bits & FB_P0_DEFAULT)
                        rc->rc_p0 = 0;
                /* XXX not implemented yet */
                rc->rc_scroll = rc->rc_p0;
                break;

        case 's':
                /* Reset terminal emulator (SUNRESET) */
                rc->rc_wsflg = 0;
                rc->rc_scroll = 0;
                rc->rc_bits &= ~FB_NO_CURSOR;
                rc->rc_fgcolor = rc->rc_deffgcolor;
                rc->rc_bgcolor = rc->rc_defbgcolor;
                rc->rc_attr = rc->rc_defattr;

                if (rc->rc_bits & FB_INVERT)
                        rcons_invert(rc, 0);
                break;
#ifdef notyet
        /*
         * XXX following two read \E[?25h and \E[?25l. rcons
         * can't currently handle the '?'.
         */
        case 'h':
                /* Normal/very visible cursor */
                if (rc->rc_p0 == 25) {
                        rc->rc_bits &= ~FB_NO_CURSOR;

                        if (rc->rc_bits & FB_CURSOR) {
                                rc->rc_bits ^= FB_CURSOR;
                                rcons_cursor(rc);
                        }
                }
                break;

        case 'l':
                /* Invisible cursor */
                if (rc->rc_p0 == 25 && (rc->rc_bits & FB_NO_CURSOR) == 0) {
                        if (rc->rc_bits & FB_CURSOR)
                                rcons_cursor(rc);

                        rc->rc_bits |= FB_NO_CURSOR;
                }
                break;
#endif
        }
}

/* Set ANSI colors */
void
rcons_setcolor(struct rconsole *rc, int fg, int bg)
{
        int flg;

        if (fg > WSCOL_WHITE || fg < 0)
                return;

        if (bg > WSCOL_WHITE || bg < 0)
                return;

#ifdef RASTERCONS_WONB
        flg = bg;
        bg = fg;
        fg = flg;
#endif

        /* Emulate WSATTR_REVERSE attribute if it's not supported */
        if ((rc->rc_wsflg & WSATTR_REVERSE) &&
            !(rc->rc_supwsflg & WSATTR_REVERSE)) {
                flg = bg;
                bg = fg;
                fg = flg;
        }

        /*
         * Mask out unsupported flags and get attribute
         * XXX - always ask for WSCOLORS if supported (why shouldn't we?)
         */
        flg = (rc->rc_wsflg | WSATTR_WSCOLORS) & rc->rc_supwsflg;
        rc->rc_bgcolor = bg;
        rc->rc_fgcolor = fg;
        rc->rc_ops->allocattr(rc->rc_cookie, fg, bg, flg, &rc->rc_attr);
}


/* Actually write a string to the frame buffer */
void
rcons_text(struct rconsole *rc, const unsigned char *str, int n)
{
        u_int uc;

        while (n--) {
                uc = rc->rc_charmap[*str++ & 255];
                rc->rc_ops->putchar(rc->rc_cookie, rc->rc_row, rc->rc_col++,
                    uc, rc->rc_attr);
        }

        if (rc->rc_col >= rc->rc_maxcol) {
                rc->rc_col = 0;
                rc->rc_row++;
        }

        if (rc->rc_row >= rc->rc_maxrow)
                rcons_scroll(rc, 1);
}

/* Paint (or unpaint) the cursor */
void
rcons_cursor(struct rconsole *rc)
{
        rc->rc_bits ^= FB_CURSOR;

        if (rc->rc_bits & FB_NO_CURSOR)
                return;

        rc->rc_ops->cursor(rc->rc_cookie, rc->rc_bits & FB_CURSOR,
            rc->rc_row, rc->rc_col);
}

/* Possibly change to SUNWOB or SUNBOW mode */
void
rcons_invert(struct rconsole *rc, int wob)
{

        rc->rc_bits ^= FB_INVERT;
        /* XXX how do we do we invert the framebuffer?? */
}

/* Clear to the end of the page */
void
rcons_clear2eop(struct rconsole *rc)
{
        if (rc->rc_col || rc->rc_row) {
                rcons_clear2eol(rc);

                if (rc->rc_row < (rc->rc_maxrow - 1))
                        rc->rc_ops->eraserows(rc->rc_cookie, rc->rc_row + 1,
                            rc->rc_maxrow, rc->rc_attr);
        } else
                rc->rc_ops->eraserows(rc->rc_cookie, 0, rc->rc_maxrow,
                    rc->rc_attr);
}

/* Clear to the end of the line */
void
rcons_clear2eol(struct rconsole *rc)
{
        rc->rc_ops->erasecols(rc->rc_cookie, rc->rc_row, rc->rc_col,
            rc->rc_maxcol - rc->rc_col, rc->rc_attr);
}


/* Scroll up */
void
rcons_scroll(struct rconsole *rc, int n)
{
        /* Can't scroll more than the whole screen */
        if (n > rc->rc_maxrow)
                n = rc->rc_maxrow;

        /* Calculate new row */
        if (rc->rc_row >= n)
                rc->rc_row -= n;
        else 
                rc->rc_row = 0;

        rc->rc_ops->copyrows(rc->rc_cookie, n, 0, rc->rc_maxrow - n);
        rc->rc_ops->eraserows(rc->rc_cookie, rc->rc_maxrow - n, n,  rc->rc_attr);
}

/* Delete characters */
void
rcons_delchar(struct rconsole *rc, int n)
{
        /* Can't delete more chars than there are */
        if (n > rc->rc_maxcol - rc->rc_col)
                n = rc->rc_maxcol - rc->rc_col;

        rc->rc_ops->copycols(rc->rc_cookie, rc->rc_row, rc->rc_col + n,
            rc->rc_col, rc->rc_maxcol - rc->rc_col - n);

        rc->rc_ops->erasecols(rc->rc_cookie, rc->rc_row,
            rc->rc_maxcol - n, n, rc->rc_attr);
}

/* Delete a number of lines */
void
rcons_delline(struct rconsole *rc, int n)
{
        /* Can't delete more lines than there are */
        if (n > rc->rc_maxrow - rc->rc_row)
                n = rc->rc_maxrow - rc->rc_row;

        rc->rc_ops->copyrows(rc->rc_cookie, rc->rc_row + n, rc->rc_row,
            rc->rc_maxrow - rc->rc_row - n);

        rc->rc_ops->eraserows(rc->rc_cookie, rc->rc_maxrow - n, n,
            rc->rc_attr);
}

/* Insert some characters */
void
rcons_insertchar(struct rconsole *rc, int n)
{
        /* Can't insert more chars than can fit */
        if (n > rc->rc_maxcol - rc->rc_col)
                n = rc->rc_maxcol - rc->rc_col - 1;

        rc->rc_ops->copycols(rc->rc_cookie, rc->rc_row, rc->rc_col,
            rc->rc_col + n, rc->rc_maxcol - rc->rc_col - n - 1);

        rc->rc_ops->erasecols(rc->rc_cookie, rc->rc_row, rc->rc_col,
            n, rc->rc_attr);
}

/* Insert some lines */
void
rcons_insertline(struct rconsole *rc, int n)
{
        /* Can't insert more lines than can fit */
        if (n > rc->rc_maxrow - rc->rc_row)
                n = rc->rc_maxrow - rc->rc_row;

        rc->rc_ops->copyrows(rc->rc_cookie, rc->rc_row, rc->rc_row + n,
            rc->rc_maxrow - rc->rc_row - n);

        rc->rc_ops->eraserows(rc->rc_cookie, rc->rc_row, n,
            rc->rc_attr);
}

/* end of rcons_subr.c */