#include <u.h>
#include <libc.h>
#include <bio.h>

#include "pci.h"
#include "vga.h"

/*
* IBM RGB52x and compatibles.
* High Performance Palette DAC.
*/
uchar (*rgb524mnxi)(Vga*, int);
void (*rgb524mnxo)(Vga*, int, uchar);

enum {                                          /* index registers */
       MiscClock       = 0x02,
       SyncControl     = 0x03,
       HSyncControl    = 0x04,
       PowerMgmnt      = 0x05,
       PaletteControl  = 0x07,
       SYSCLKControl   = 0x08,
       PixelFormat     = 0x0A,
       Pixel8Control   = 0x0B,
       Pixel16Control  = 0x0C,
       Pixel32Control  = 0x0E,
       PLLControl1     = 0x10,
       PLLControl2     = 0x11,
       SYSCLKN         = 0x15,
       SYSCLKM         = 0x16,
       M0              = 0x20,
       N0              = 0x21,
       MiscControl1    = 0x70,
       MiscControl2    = 0x71,
};

static void
clock(Vga* vga, Ctlr*, ulong fref, ulong maxpclk)
{
       int d, mind;
       ulong df, f, m, n, vrf;

       mind = vga->f[0]+1;
       for(df = 0; df < 4; df++){
               for(m = 2; m < 64; m++){
                       for(n = 2; n < 32; n++){
                               f = (fref*(m+65))/n;
                               switch(df){
                               case 0:
                                       vrf = fref/(n*2);
                                       if(vrf > maxpclk/4 || vrf < 1000000)
                                               continue;
                                       f /= 8;
                                       break;
                               case 1:
                                       vrf = fref/(n*2);
                                       if(vrf > maxpclk/2 || vrf < 1000000)
                                               continue;
                                       f /= 4;
                                       break;
                               case 2:
                                       vrf = fref/(n*2);
                                       if(vrf > maxpclk || vrf < 1000000)
                                               continue;
                                       f /= 2;
                                       break;
                               case 3:
                                       vrf = fref/n;
                                       if(vrf > maxpclk || vrf < 1000000)
                                               continue;
                                       break;
                               }
                               if(f > maxpclk)
                                       continue;

                               d = vga->f[0] - f;
                               if(d < 0)
                                       d = -d;
                               if(d <= mind){
                                       vga->m[0] = m;
                                       vga->n[0] = n;
                                       vga->d[0] = df;
                                       mind = d;
                               }
                       }
               }
       }
}

static void
init(Vga* vga, Ctlr* ctlr)
{
       ulong fref, maxpclk;
       char *p, *val;

       /*
        * Part comes in at least a -170MHz speed-grade.
        */
       maxpclk = 170000000;
       if(p = strrchr(ctlr->name, '-'))
               maxpclk = strtoul(p+1, 0, 0) * 1000000;

       /*
        * If we don't already have a desired pclk,
        * take it from the mode.
        * Check it's within range.
        */
       if(vga->f[0] == 0)
               vga->f[0] = vga->mode->frequency;
       if(vga->f[0] > maxpclk)
               error("%s: invalid pclk - %ld\n", ctlr->name, vga->f[0]);
       if(val = dbattr(vga->attr, "rgb524mnrefclk"))
               fref = strtol(val, 0, 0);
       else
               fref = RefFreq;

       /*
        * Initialise the PLL parameters.
        * Use m/n pair 2.
        */
       clock(vga, ctlr, fref, maxpclk);
       vga->i[0] = 2;

       ctlr->flag |= Finit;
}

static void
load(Vga* vga, Ctlr* ctlr)
{
       char *val;
       int hsyncdelay, x;

       if(rgb524mnxi == nil && rgb524mnxo == nil)
               error("%s->load: no access routines\n", ctlr->name);

       /*
        * Set the m/n values for the desired frequency and
        * set pixel control to use compatibility mode with
        * internal frequency select using the specified set
        * of m/n registers.
        */
       rgb524mnxo(vga, M0+vga->i[0]*2, vga->d[0]<<6|vga->m[0]);
       rgb524mnxo(vga, N0+vga->i[0]*2, vga->n[0]);
       rgb524mnxo(vga, PLLControl2, vga->i[0]);
       rgb524mnxo(vga, PLLControl1, 0x03);

       /*
        * Enable pixel programming in MiscClock;
        * nothing to do in MiscControl1;
        * set internal PLL clock and !vga in MiscControl2;
        */
       x = rgb524mnxi(vga, MiscClock) & ~0x01;
       x |= 0x01;
       rgb524mnxo(vga, MiscClock, x);

       x = rgb524mnxi(vga, MiscControl2) & ~0x41;
       x |= 0x41;
       rgb524mnxo(vga, MiscControl2, x);

       /*
        * Syncs.
        */
       x = 0;
       if(vga->mode->hsync == '+')
               x |= 0x10;
       if(vga->mode->vsync == '+')
               x |= 0x20;
       rgb524mnxo(vga, SyncControl, x);
       if(val = dbattr(vga->mode->attr, "hsyncdelay"))
               hsyncdelay = strtol(val, 0, 0);
       else switch(vga->mode->z){
       default:
       case 8:
               hsyncdelay = 1;
               break;
       case 15:
       case 16:
               hsyncdelay = 5;
               break;
       case 32:
               hsyncdelay = 7;
               break;
       }
       rgb524mnxo(vga, HSyncControl, hsyncdelay);

rgb524mnxo(vga, SYSCLKM, 0x50);
rgb524mnxo(vga, SYSCLKN, 0x08);
sleep(50);
//rgb524mnxo(vga, SYSCLKM, 0x6F);
//rgb524mnxo(vga, SYSCLKN, 0x0F);
//sleep(500);

       /*
        * Set the palette for the desired format.
        * ****NEEDS WORK FOR OTHER THAN 8-BITS****
        */
       rgb524mnxo(vga, PaletteControl, 0x00);
       switch(vga->mode->z){
       case 8:
               rgb524mnxo(vga, PixelFormat, 0x03);
               rgb524mnxo(vga, Pixel8Control, 0x00);
               break;
       case 15:
               rgb524mnxo(vga, PixelFormat, 0x04);
               rgb524mnxo(vga, Pixel16Control, 0xC4);
       case 16:
               rgb524mnxo(vga, PixelFormat, 0x04);
               rgb524mnxo(vga, Pixel16Control, 0xC6);
               break;
       case 32:
               rgb524mnxo(vga, PixelFormat, 0x06);
               rgb524mnxo(vga, Pixel32Control, 0x03);
               break;
       }
}

static void
dumpclock(Vga*, Ctlr* ctlr, ulong fref, ulong m, ulong n, char* name)
{
       ulong df, f;

       df = (m>>6) & 0x03;
       m &= 0x3F;
       n &= 0x1F;
       if(m == 0 || n == 0)
               return;
       f = (fref*(m+65))/n;
       switch(df){
       case 0:
               f /= 8;
               break;
       case 1:
               f /= 4;
               break;
       case 2:
               f /= 2;
               break;
       case 3:
               break;
       }
       printitem(ctlr->name, name);
       Bprint(&stdout, "%12lud\n", f);
}

static void
dump(Vga* vga, Ctlr* ctlr)
{
       int i;
       char *val;
       uchar x[256];
       ulong fref, fs;

       if(rgb524mnxi == nil && rgb524mnxo == nil)
               error("%s->dump: no access routines\n", ctlr->name);

       printitem(ctlr->name, "index00");
       for(i = 0x00; i < 0x0F; i++){
               x[i] = rgb524mnxi(vga, i);
               printreg(x[i]);
       }
       printitem(ctlr->name, "index10");
       for(i = 0x10; i < 0x18; i++){
               x[i] = rgb524mnxi(vga, i);
               printreg(x[i]);
       }
       printitem(ctlr->name, "index20");
       for(i = 0x20; i < 0x30; i++){
               x[i] = rgb524mnxi(vga, i);
               printreg(x[i]);
       }
       printitem(ctlr->name, "index30");
       for(i = 0x30; i < 0x39; i++){
               x[i] = rgb524mnxi(vga, i);
               printreg(x[i]);
       }
       printitem(ctlr->name, "index40");
       for(i = 0x40; i < 0x49; i++){
               x[i] = rgb524mnxi(vga, i);
               printreg(x[i]);
       }
       printitem(ctlr->name, "index60");
       for(i = 0x60; i < 0x63; i++){
               x[i] = rgb524mnxi(vga, i);
               printreg(x[i]);
       }
       printitem(ctlr->name, "index70");
       for(i = 0x70; i < 0x73; i++){
               x[i] = rgb524mnxi(vga, i);
               printreg(x[i]);
       }
       printitem(ctlr->name, "index8E");
       for(i = 0x8E; i < 0x92; i++){
               x[i] = rgb524mnxi(vga, i);
               printreg(x[i]);
       }

       if(val = dbattr(vga->attr, "rgb524mnrefclk"))
               fref = strtol(val, 0, 0);
       else
               fref = RefFreq;
       if(!(x[SYSCLKControl] & 0x04))
               dumpclock(vga, ctlr, fref, x[0x16], x[0x15], "sysclk");
       fs = x[PLLControl1] & 0x07;
       if(fs == 0x01 || fs == 0x03){
               if(fs == 0x01)
                       i = ((vga->misc>>2) & 0x03)*2;
               else
                       i = x[PLLControl2] & 0x07;
               dumpclock(vga, ctlr, fref, x[M0+i*2], x[N0+i*2], "pllclk");
       }
}

Ctlr rgb524mn = {
       "rgb524mn",                     /* name */
       0,                              /* snarf */
       0,                              /* options */
       init,                           /* init */
       load,                           /* load */
       dump,                           /* dump */
};