/*
 *  VAS_EBOOT  --  GRand Unified Bootloader
 *  Copyright (C) 2016  Free Software Foundation, Inc.
 *
 *  VAS_EBOOT is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 *
 *  VAS_EBOOT is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with VAS_EBOOT.  If not, see <http://www.gnu.org/licenses/>.
 */

#include <VasEBoot/mm.h>
#include <VasEBoot/machine/kernel.h>
#include <VasEBoot/types.h>
#include <VasEBoot/err.h>
#include <VasEBoot/time.h>
#include <VasEBoot/fdtbus.h>
#include <VasEBoot/misc.h>

VasEBoot_uint64_t
VasEBoot_armv7_get_timer_value(void);

VasEBoot_uint32_t
VasEBoot_armv7_get_timer_frequency(void);

VasEBoot_uint32_t
VasEBoot_arm_pfr1(void);

static int have_timer = 0;
static volatile VasEBoot_uint32_t *sp804_regs;

static VasEBoot_uint64_t
sp804_get_time_ms (void)
{
  static VasEBoot_uint32_t high, last_low;
  VasEBoot_uint32_t low = ~sp804_regs[1];
  if (last_low > low)
    high++;
  last_low = low;
  return VasEBoot_divmod64 ((((VasEBoot_uint64_t) high) << 32) | low,
			1000, 0);
}

static VasEBoot_err_t
sp804_attach(const struct VasEBoot_fdtbus_dev *dev)
{
  if (have_timer)
    return VAS_EBOOT_ERR_NONE;
  sp804_regs = VasEBoot_fdtbus_map_reg (dev, 0, 0);
  if (!VasEBoot_fdtbus_is_mapping_valid (sp804_regs))
    return VasEBoot_error (VAS_EBOOT_ERR_IO, "could not map sp804: %p", sp804_regs);
  VasEBoot_install_get_time_ms (sp804_get_time_ms);
  have_timer = 1;
  return VAS_EBOOT_ERR_NONE;
}

struct VasEBoot_fdtbus_driver sp804 =
{
  .compatible = "arm,sp804",
  .attach = sp804_attach
};

static VasEBoot_uint32_t timer_frequency_in_khz;

static VasEBoot_uint64_t
generic_get_time_ms (void)
{
  return VasEBoot_divmod64 (VasEBoot_armv7_get_timer_value(), timer_frequency_in_khz, 0);
}

static int
try_generic_timer (void)
{
  if (((VasEBoot_arm_pfr1 () >> 16) & 0xf) != 1)
    return 0;
  VasEBoot_printf ("freq = %x\n", VasEBoot_armv7_get_timer_frequency());
  timer_frequency_in_khz = 0x016e3600 / 1000; //VasEBoot_armv7_get_timer_frequency() / 1000;
  if (timer_frequency_in_khz == 0)
    return 0;
  VasEBoot_install_get_time_ms (generic_get_time_ms);
  have_timer = 1;
  return 1;
}

void
VasEBoot_machine_timer_init (void)
{
  VasEBoot_fdtbus_register (&sp804);

  if (!have_timer)
    try_generic_timer ();
  if (!have_timer)
    VasEBoot_fatal ("No timer found");
}