// Copyright 2016 ETH Zurich and University of Bologna.
// Copyright and related rights are licensed under the Solderpad Hardware
// License, Version 0.51 (the “License”); you may not use this file except in
// compliance with the License. You may obtain a copy of the License at
// http://solderpad.org/licenses/SHL-0.51. Unless required by applicable law
// or agreed to in writing, software, hardware and materials distributed under
// this License is distributed on an “AS IS” BASIS, WITHOUT WARRANTIES OR
// CONDITIONS OF ANY KIND, either express or implied. See the License for the
// specific language governing permissions and limitations under the License.

#include "bench.h"
#include "uart.h"
#include "gpio.h"
#include "spr-defs.h"

void bench_timer_start(void) {
  set_gpio_pin_value(0, 1);
  start_timer();
}

void bench_timer_stop(void) {
  stop_timer();
  set_gpio_pin_value(0, 0);
}

void bench_timer_reset(void) {
  reset_timer();
}

void print_result(testcase_t *test, testresult_t *result)
{
  printf("== test: %s -> ", test->name);

  if (result->errors == 0)
    printf("success, ");
  else
    printf("fail, ");

  printf("nr. of errors: %d", result->errors);

  if(result->time == 0)
    printf("\n");
  else
    printf(", execution time: %d\n", result->time);
}

void print_summary(unsigned int errors)
{
  printf("==== SUMMARY: ");

  if (errors == 0) {
    printf("SUCCESS\n");
  } else {
    printf("FAIL\n");
  }
}

void run_benchmark(testcase_t *test, testresult_t *result)
{
  set_gpio_pin_value(0, 0);
  set_gpio_pin_direction(0, DIR_OUT);
  set_pin_function(0, FUNC_GPIO);
  result->errors = 0;

  bench_timer_reset();

  test->test(result, bench_timer_start, bench_timer_stop);

  result->time = get_time();
}

unsigned int run_suite(testcase_t *tests)
{
  // figure out how many tests should be run
  size_t num = 0;
  while(tests[num].name != 0) num++;

  unsigned int errors = 0;
  size_t i;

  // perform the tests
  for (i = 0; i < num; i++) {
    testresult_t result;
    run_benchmark(&tests[i], &result);
    print_result(&tests[i], &result);

    errors += result.errors;
  }

  print_summary(errors);

  return errors;
}

void check_uint32(testresult_t* result, const char* fail_msg, uint32_t actual, uint32_t expected)
{
  if(actual != expected) {
    result->errors += 1;
    printf("%s: Actual %X, expected %X\n", fail_msg, (unsigned int) actual, (unsigned int) expected);
  }
}

void perf_print_all(void) {
#ifdef __riscv__
  printf("Perf CYCLES:   %d\n", cpu_perf_get(0));
  printf("Perf INSN:     %d\n", cpu_perf_get(1));
  printf("Perf CINSN:    %d\n", cpu_perf_get(10));
  printf("Perf LD_STALL: %d\n", cpu_perf_get(2));
  printf("Perf JR_STALL: %d\n", cpu_perf_get(3));
  printf("Perf IMISS:    %d\n", cpu_perf_get(4));
  printf("Perf #LD:      %d\n", cpu_perf_get(5));
  printf("Perf #ST:      %d\n", cpu_perf_get(6));
  printf("Perf #JUMP:    %d\n", cpu_perf_get(7));
  printf("Perf #BRANCH:  %d\n", cpu_perf_get(8));
  printf("Perf #TAKEN:   %d\n", cpu_perf_get(9));
  printf("Perf #RVC:     %d\n", cpu_perf_get(10));
#else
  printf("Perf CYCLES: %d\n",      cpu_perf_get(SPR_PCER_CYCLES));
  printf("Perf INSTR: %d\n",       cpu_perf_get(SPR_PCER_INSTR));
  printf("Perf LD_STALL: %d\n",    cpu_perf_get(SPR_PCER_LD_STALL));
  printf("Perf JMP_STALL: %d\n",   cpu_perf_get(SPR_PCER_JMP_STALL));
  printf("Perf IMISS: %d\n",       cpu_perf_get(SPR_PCER_IMISS));
  printf("Perf WBRANCH: %d\n",     cpu_perf_get(SPR_PCER_WBRANCH));
  printf("Perf WBRANCH_CYC: %d\n", cpu_perf_get(SPR_PCER_WBRANCH_CYC));
  printf("Perf LD: %d\n",          cpu_perf_get(SPR_PCER_LD));
  printf("Perf ST: %d\n",          cpu_perf_get(SPR_PCER_ST));
  printf("Perf JUMP: %d\n",        cpu_perf_get(SPR_PCER_JUMP));
  printf("Perf BRANCH: %d\n",      cpu_perf_get(SPR_PCER_BRANCH));
  printf("Perf DELAY NOP: %d\n",   cpu_perf_get(SPR_PCER_DELAY_NOP));
#endif
}

// use weak attribute here, so we can overwrite this function to provide custom exception handlers, e.g. for tests
//__attribute__((interrupt))
__attribute__((weak))
void illegal_insn_handler_c(void)
{
  unsigned int exception_address, insn;
#ifdef __riscv__
  asm("csrr %0, 0x341" : "=r" (exception_address) : );
#else
  exception_address = mfspr(SPR_EPCR_BASE);
#endif
  insn = *((unsigned int*)(exception_address));
  printf("Illegal instruction encountered at address 0x%08X: %X\n", exception_address, insn);
  exit(1);
}