/*
 *  F i l e N a m e  :  U L X B E N C H . C
 *
 ****************************************************************************
 *                   (C) CopyRight ALPS Electric Co., Ltd. 1996
 ****************************************************************************
 *  Version
 *
 *  ver     Date
 *  ----  --------
 *  1.01  '96.04.05  Changed DISKTYPE 4 -> 7.
 *  1.00  '95.12.20  Initial revision .
 ****************************************************************************
*/
/*===========================================*/
/* ﾚｵﾄﾞﾗｲﾌﾞ用ﾃｽﾄｿﾌﾄのﾍﾞﾝﾁﾏｰｸ測定ﾙｰﾁﾝ         */
/*===========================================*/

#include <ultra64.h>
#include "ulxmain.h"
#include "ulxlib.h"
#include "ulxutil.h"

#include "leodrive.h"

#define RDBLOCKS    0x32

const static LEOCmdRead pm_read_base =
      { { LEO_COMMAND_READ,           /* COMMAND        */
          0x00,                       /* RESERVE        */
          0x00,                       /* CONTROL        */
          0x00,                       /* RESERVE        */
          0x00,                       /* STATUS         */
          0x00,                       /* SENSE          */
          0x00,                       /* RESERVE        */
          0x00,                       /* RESERVE        */
          0x00                        /* POST QUEUE     */
        },
          0x00,                       /* LBA            */
          RDBLOCKS,                   /* BLOCKS         */
          (u64 *)&RDB,                /* BUFFER POINTER */
          0x00,                       /* READ BYTES     */
      };

const static LEOCmdSeek pm_seek_base =
      { { LEO_COMMAND_SEEK,           /* COMMAND        */
          0x00,                       /* RESERVE        */
          0x00,                       /* CONTROL        */
          0x00,                       /* RESERVE        */
          0x00,                       /* STATUS         */
          0x00,                       /* SENSE          */
          0x00,                       /* RESERVE        */
          0x00,                       /* RESERVE        */
          0x00                        /* POST QUEUE     */
        },
          0x00                        /* LBA            */
      };

extern u16    LEOVZONE_TBL[DISK_TYPES][VZONES_PER_DRV];
extern u16    LEOBYTE_TBL2[ZONES_PER_DRV];
extern u8     LEOdisk_type;
extern OSMesg LEOcur_command;

/*==========================================================*/

void    transfer_rate_test( void )
{
    u32    key;
    u32    test_lba, read_bytes;
    OSTime  cmd_time, tcount_pre,tcount_aft;
    u8     test_zone;
    volatile LEOCmdRead pm_read_cmd0;
    volatile LEOCmdRead pm_read_cmd1;
    volatile LEOCmdRead pm_read_cmd2;
    volatile LEOCmdRead pm_read_cmd3;
    volatile LEOCmdRead pm_read_cmd4;
    volatile LEOCmdSeek pm_seek_cmd;

    test_lba = (u32)0x00;

    ClrDisp();
    LEOcur_command = (OSMesg)&pm_read_cmd0;
    Disp_Ready(0, 1, CYAN);
    sprintf( STR, "TRANSFER RATE RESULT" );
    Disp_Fin();

    Disp_Ready(4,  4, WHITE);
    sprintf( STR, "TRANSFER RATE TESTING..        ");
    Disp_Fin();

    for (test_zone=0; test_zone<9; test_zone++)
    {
        /* test command set. */
        pm_read_cmd0 = pm_read_base;
        pm_read_cmd1 = pm_read_base;
        pm_read_cmd2 = pm_read_base;
        pm_read_cmd3 = pm_read_base;
        pm_read_cmd4 = pm_read_base;
        if (test_zone)
        {
            switch(LEOdisk_type)
            {
                case 0:
                    if (test_zone > 2)
                        test_lba = LEOVZONE_TBL[LEOdisk_type][test_zone + 1];
                    else
                        test_lba = LEOVZONE_TBL[LEOdisk_type][test_zone -1]; 
                    break;
                case 1:
                    if (test_zone > 3)
                        test_lba = LEOVZONE_TBL[LEOdisk_type][test_zone + 2];
                    else
                        test_lba = LEOVZONE_TBL[LEOdisk_type][test_zone -1]; 
                    break;
                case 2:
                    if (test_zone > 4)
                        test_lba = LEOVZONE_TBL[LEOdisk_type][test_zone + 3];
                    else
                        test_lba = LEOVZONE_TBL[LEOdisk_type][test_zone -1]; 
                    break;
                case 3:
                    if (test_zone > 5)
                        test_lba = LEOVZONE_TBL[LEOdisk_type][test_zone + 4];
                    else
                        test_lba = LEOVZONE_TBL[LEOdisk_type][test_zone -1]; 
                    break;
                case 4:
                    if (test_zone > 6)
                        test_lba = LEOVZONE_TBL[LEOdisk_type][test_zone + 5];
                    else
                        test_lba = LEOVZONE_TBL[LEOdisk_type][test_zone -1]; 
                    break;
                case 5:
                    if (test_zone > 7)
                        test_lba = LEOVZONE_TBL[LEOdisk_type][test_zone + 6];
                    else
                        test_lba = LEOVZONE_TBL[LEOdisk_type][test_zone -1]; 
                    break;
                case 6:
                    test_lba = LEOVZONE_TBL[LEOdisk_type][test_zone -1];
                default:
                    break;
            }
        }
        if (test_zone == 0)
            test_lba = (u32)0;
        else
            test_lba -= SYSTEM_LBAS;

        if (RezeroUnit() || SeekDrive(test_lba + (RDBLOCKS * 4)) || SeekDrive(test_lba) )
        {
            Disp_Ready(2, 13, WHITE);
            sprintf( STR, "STS:%02d  SC:%02d   ",Status,SCode);
            Disp_Fin();
            break;
        }

        pm_read_cmd0.lba = test_lba;
        pm_read_cmd1.lba = test_lba;
        pm_read_cmd2.lba = test_lba + RDBLOCKS;
        pm_read_cmd3.lba = test_lba + (RDBLOCKS * 2);
        pm_read_cmd4.lba = test_lba + (RDBLOCKS * 3);

        pm_read_cmd0.xfer_blks = 1;
        leoCommand( (void *)&pm_read_cmd0);
        do {} while (pm_read_cmd0.header.status == LEO_STATUS_BUSY);

        tcount_pre = osGetTime();  /* get start time */

        leoCommand( (void *)&pm_read_cmd1);  /* issue command */
        leoCommand( (void *)&pm_read_cmd2);
        leoCommand( (void *)&pm_read_cmd3);
        leoCommand( (void *)&pm_read_cmd4);

        if (InCtrlData() == CONT_B) goto abort_test;
        do {} while (pm_read_cmd1.header.status == LEO_STATUS_BUSY);
        if (InCtrlData() == CONT_B) goto abort_test;
        do {} while (pm_read_cmd2.header.status == LEO_STATUS_BUSY);
        if (InCtrlData() == CONT_B) goto abort_test;
        do {} while (pm_read_cmd3.header.status == LEO_STATUS_BUSY);
        if (InCtrlData() == CONT_B)
        {
abort_test:
            Disp_Ready(2, (6+test_zone), YELLOW);
            sprintf( STR, "   ABORT TEST            ");
            Disp_Fin();
            do {} while (pm_read_cmd4.header.status == LEO_STATUS_BUSY);
            goto test_end;
        }

        /* WAIT LAST READ COMMAND FINISH. */
        do {} while (pm_read_cmd4.header.status == LEO_STATUS_BUSY);

        tcount_aft = osGetTime();
        tcount_aft = (tcount_aft - tcount_pre);  /* calc test time */

        Disp_Ready(2,  3, GREEN);
        sprintf( STR, "ZONE   BYTE     mSEC    BYTE/SEC ");
        Disp_Fin();
        Disp_Ready(2,  4, GREEN);
        sprintf( STR, "-------------------------------- ");
        Disp_Fin();

        Disp_Ready(2, (5+test_zone), WHITE);

        if ((pm_read_cmd1.header.status == LEO_STATUS_GOOD) && (pm_read_cmd2.header.status == LEO_STATUS_GOOD) &&
            (pm_read_cmd3.header.status == LEO_STATUS_GOOD) && (pm_read_cmd4.header.status == LEO_STATUS_GOOD) &&
            (pm_read_cmd0.header.status == LEO_STATUS_GOOD) )
        {
            cmd_time = OS_CYCLES_TO_USEC(tcount_aft) / 1000;

            read_bytes = LEOBYTE_TBL2[test_zone] * RDBLOCKS * 4;
            sprintf( STR, " %X   %7d    %4Ld    %7Ld ",test_zone, read_bytes, cmd_time,((read_bytes * 1000)/cmd_time) );
        }
        else
        {
            if (pm_read_cmd4.header.status != LEO_STATUS_GOOD)
                sprintf( STR, " %X RD ERROR STS:%02d  SC:%02d ",test_zone,pm_read_cmd4.header.status,pm_read_cmd4.header.sense);
            if (pm_read_cmd3.header.status != LEO_STATUS_GOOD)
                sprintf( STR, " %X RD ERROR STS:%02d  SC:%02d ",test_zone,pm_read_cmd3.header.status,pm_read_cmd3.header.sense);
            if (pm_read_cmd2.header.status != LEO_STATUS_GOOD)
                sprintf( STR, " %X RD ERROR STS:%02d  SC:%02d ",test_zone,pm_read_cmd2.header.status,pm_read_cmd2.header.sense);
            if (pm_read_cmd1.header.status != LEO_STATUS_GOOD)
                sprintf( STR, " %X RD ERROR STS:%02d  SC:%02d ",test_zone,pm_read_cmd1.header.status,pm_read_cmd1.header.sense);
            if (pm_read_cmd0.header.status != LEO_STATUS_GOOD)
                sprintf( STR, " %X RD ERROR STS:%02d  SC:%02d ",test_zone,pm_read_cmd0.header.status,pm_read_cmd0.header.sense);
        }
        Disp_Fin();
#if 0
        key = InCtrlData();
        if ( key == CONT_B)
        {
            Disp_Ready(2, (6+test_zone), YELLOW);
            sprintf( STR, "   TEST IS ABORT         ");
            Disp_Fin();
            goto test_end;
        }
#endif
    }
test_end:
    Disp_Ready(1, 14, YELLOW);
    sprintf( STR, "Hit any key to return MENU");
    Disp_Fin();

    GetChr();

}

void seek_time_tk_test(u32 test_count,u32 slba )
{
    u32 elba;
    OSTime cmd_time;
    OSTime tcount_pre, tcount_aft;
    u32 cmd_count;
    volatile LEOCmdSeek pm_seek_cmd;

    cmd_count = 0;

    ClrDisp();
    LEOcur_command = (OSMesg)&pm_seek_cmd;

    elba = slba + 2;
    if (elba > MAX_L_LBA)
        elba = MAX_L_LBA;

    if (test_count == 0)
        test_count = 1;

    Disp_Ready(0, 1, CYAN);
    sprintf( STR, "TK-TK SEEK TIME (%u[times]) ",test_count);
    Disp_Fin();

    Disp_Ready(2,  4, WHITE);
    sprintf( STR, "TESTING..           ");
    Disp_Fin();

    if (RezeroUnit() || SeekDrive(0x341) || SeekDrive(0)
     || SeekDrive(elba) || SeekDrive(slba) )
    {
        Disp_Ready(2,  4, YELLOW);
        sprintf( STR, " SEEK NG. TEST ABORT. ");
        Disp_Fin();

        Disp_Ready(2, 6, WHITE);
        sprintf( STR, "STS:%02d  SC:%02d   ",Status,SCode);
        Disp_Fin();

        Disp_Ready(1, 14, YELLOW);
        sprintf( STR, "Hit any key to return MENU");
        Disp_Fin();

        GetChr();
        return;
    }
    /* test command set. */
    pm_seek_cmd     = pm_seek_base;
    pm_seek_cmd.lba = elba;

    tcount_pre = osGetTime();
    for (cmd_count=0; cmd_count<test_count; cmd_count++)
    {
        leoCommand( (void *)&pm_seek_cmd);

        if (pm_seek_cmd.lba == slba)
            pm_seek_cmd.lba = elba;
        else
            pm_seek_cmd.lba = slba;
        /* WAIT LAST READ COMMAND FINISH. */
        do {} while (pm_seek_cmd.header.status == LEO_STATUS_BUSY);
    }
    tcount_aft = osGetTime();

    cmd_time = OS_CYCLES_TO_USEC(tcount_aft-tcount_pre) / test_count;
    Disp_Ready(2, 4, WHITE);
    sprintf( STR, "AVE = %Ld [uSEC] ",cmd_time);
    Disp_Fin();

    Disp_Ready(2, 12, WHITE);
    sprintf( STR, "Include CMD Overhead.      ");
    Disp_Fin();

    Disp_Ready(1, 14, YELLOW);
    sprintf( STR, "Hit any key to return MENU");
    Disp_Fin();

    GetChr();
}

void seek_time_rand_test(u32 test_count, u32 slba, u32 elba )
{
    u32 lba ;
    OSTime cmd_time;
    OSTime tcount_pre, tcount_aft;
    u32 cmd_count;
    volatile LEOCmdSeek pm_seek_cmd;

    lba = cmd_count = 0;

    ClrDisp();
    LEOcur_command = (OSMesg)&pm_seek_cmd;

    /* Random Seek Mode.  */
    if (slba > elba)
    {
        lba  = elba;
        elba = slba;
        slba = lba;
    }

    if (test_count == 0)
        test_count = 0x01;

    Disp_Ready(0, 1, CYAN);
    sprintf( STR, "Rand. SEEK TIME  (%u[times]) ",test_count);
    Disp_Fin();

    Disp_Ready(2,  4, WHITE);
    sprintf( STR, "TESTING..           ");
    Disp_Fin();

    if (RezeroUnit() || SeekDrive(0x341) || SeekDrive(0)
     || SeekDrive(slba) || SeekDrive((u32)( Rand32() % (elba - slba + 1)) + slba) )
    {
        Disp_Ready(2,  4, YELLOW);
        sprintf( STR, " SEEK NG. TEST ABORT. ");
        Disp_Fin();

        Disp_Ready(2, 6, WHITE);
        sprintf( STR, "STS:%02d  SC:%02d   ",Status,SCode);
        Disp_Fin();

        Disp_Ready(1, 14, YELLOW);
        sprintf( STR, "Hit any key to return MENU");
        Disp_Fin();

        GetChr();

        return;
    }
    /* test command set. */
    pm_seek_cmd     = pm_seek_base;
    /*pm_seek_cmd.lba = slba;*/
    pm_seek_cmd.lba = (u32)( Rand32() % (elba - slba + 1)) + slba;

    tcount_pre = osGetTime();
    for (cmd_count=0; cmd_count<test_count; cmd_count++)
    {
        leoCommand( (void *)&pm_seek_cmd);

        pm_seek_cmd.lba = (u32)( Rand32() % (elba - slba + 1)) + slba;
        /* WAIT LAST READ COMMAND FINISH. */
        do {} while (pm_seek_cmd.header.status == LEO_STATUS_BUSY);
    }
    tcount_aft = osGetTime();

    cmd_time = OS_CYCLES_TO_USEC(tcount_aft-tcount_pre) / test_count;
    Disp_Ready(2, 4, WHITE);
    sprintf( STR, "AVE = %7Ld [uSEC] ",cmd_time);
    Disp_Fin();

    Disp_Ready(2, 12, WHITE);
    sprintf( STR, "Include CMD Overhead.      ");
    Disp_Fin();

    Disp_Ready(1, 14, YELLOW);
    sprintf( STR, "Hit any key to return MENU");
    Disp_Fin();

    GetChr();

}

void seek_time_full_test(u32 test_count )
{
    u32 slba, elba;
    OSTime cmd_time;
    OSTime tcount_pre, tcount_aft;
    u32 cmd_count;
    volatile LEOCmdSeek pm_seek_cmd;

    cmd_count = 0;

    ClrDisp();
    LEOcur_command = (OSMesg)&pm_seek_cmd;

    /* Full Stroke Seek Mode. */
    slba = (u32)0;
    switch(LEOdisk_type)
    {
        case 0:
            elba = (u32)0xa9e;    /* ZONE7 LAST TRACK */
            break;
        case 1:
            elba = (u32)0xbb0;    /* ZONE7 LAST TRACK */
            break;
        case 2:
            elba = (u32)0xcc2;    /* ZONE7 LAST TRACK */
            break;
        case 3:
            elba = (u32)0xdd4;    /* ZONE7 LAST TRACK */
            break;
        case 4:
            elba = (u32)0xee6;    /* ZONE7 LAST TRACK */
            break;
        case 5:
            elba = (u32)0xff8;    /* ZONE7 LAST TRACK */
            break;
        case 6:
            elba = (u32)0x856;    /* ZONE7 LAST TRACK */
            break;
        default:
            break;
    }

    if (test_count == 0)
        test_count = 0x01;

    Disp_Ready(0, 1, CYAN);
    sprintf( STR, "FULL TK SEEK TIME (%u[times]) ",test_count);
    Disp_Fin();

    Disp_Ready(2,  4, WHITE);
    sprintf( STR, "TESTING..           ");
    Disp_Fin();

    if (RezeroUnit() || SeekDrive(elba) || SeekDrive(slba) )
    {
        Disp_Ready(2,  4, YELLOW);
        sprintf( STR, " SEEK NG. TEST ABORT. ");
        Disp_Fin();

        Disp_Ready(2, 6, WHITE);
        sprintf( STR, "STS:%02d  SC:%02d   ",Status,SCode);
        Disp_Fin();

        Disp_Ready(1, 14, YELLOW);
        sprintf( STR, "Hit any key to return MENU");
        Disp_Fin();

        GetChr();

        return;
    }
    /* test command set. */
    pm_seek_cmd     = pm_seek_base;
    pm_seek_cmd.lba = elba;

    tcount_pre = osGetTime();
    for (cmd_count=0; cmd_count<test_count; cmd_count++)
    {
        leoCommand( (void *)&pm_seek_cmd);

        if (pm_seek_cmd.lba == slba)
            pm_seek_cmd.lba = elba;
        else
            pm_seek_cmd.lba = slba;

        /* WAIT LAST READ COMMAND FINISH. */
        do {} while (pm_seek_cmd.header.status == LEO_STATUS_BUSY);
    }
    tcount_aft = osGetTime();

    cmd_time = OS_CYCLES_TO_USEC(tcount_aft-tcount_pre) / test_count;
    Disp_Ready(2, 4, WHITE);
    sprintf( STR, "AVE = %Ld [uSEC]",cmd_time);
    Disp_Fin();

    Disp_Ready(2, 12, WHITE);
    sprintf( STR, "Include CMD Overhead.      ");
    Disp_Fin();

    Disp_Ready(1, 14, YELLOW);
    sprintf( STR, "Hit any key to return MENU");
    Disp_Fin();

    GetChr();
}


void    bench_test( void )
{
    u32 num=1;
    u32 slba, elba;
    u32 test_count;
    u16 x_l=5, y_l=2, col_l=6;
    u8  *left_d[6] = {"TRANSFER RATE TEST     ",
                      "SEEK TIME(TK-TK ) TEST ",
                      "SEEK TIME(RANDOM) TEST ",
                      "SEEK TIME(FULL  ) TEST ",
                      "                       ",
                      "EXIT                   "
                     };

    slba = 0;
    elba = MAX_L_LBA;
    test_count = 1000;

    do
    {
        ClrDisp();
        Disp_Ready(0, 0, CYAN);
        sprintf( STR, "BENCHMARK TEST MENU " );
        Disp_Fin();

        ListDisp( x_l, y_l, col_l, left_d, 0, 0, 0, (void *)0, GREEN );
        num = SelStr1( x_l, y_l, col_l, left_d, num, GREEN );

        osSetTime((OSTime)0);

        switch( num )
        {
            case 1:
                transfer_rate_test() ;
                break;

            case 2:
                slba       = GetNum32( slba      , "Input START LBA ", 2, 13, YELLOW );
                test_count = GetNum32( test_count, "Input TEST COUNT    ", 2, 13, YELLOW );
                seek_time_tk_test(test_count, slba ) ;
                break;

            case 3:
                slba       = GetNum32( slba      , "Input MIN LBA ", 2, 13, YELLOW );
                elba       = GetNum32( elba      , "Input MAX LBA ", 2, 13, YELLOW );
                test_count = GetNum32( test_count, "Input TEST COUNT  ", 2, 13, YELLOW );
                seek_time_rand_test(test_count, slba, elba ) ;
                break;

            case 4:
                test_count = GetNum32( test_count, "Input TEST COUNT ", 2, 13, YELLOW );
                seek_time_full_test(test_count ) ;
                break;

            case 5:
                break;

            case 6:
                return;

            default:
                break;
        }

    } while( num != ESC );
}