/*      $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 */