/*
 * This program 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 2 of
 * the License, or (at your option) any later version.
 *
 * This program 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.
 *
 */

#include <common.h>
#include <bootsource.h>
#include <init.h>
#include <io.h>
#include <mach/socfpga/generic.h>
#include <mach/socfpga/arria10-system-manager.h>

enum bootsource arria10_get_bootsource(void) {
	enum bootsource src = BOOTSOURCE_UNKNOWN;
	uint32_t val;
	uint32_t mask = ARRIA10_SYSMGR_BOOTINFO_BSEL_MASK;

	val = readl(ARRIA10_SYSMGR_BOOTINFO);

	switch ((val & mask) >> ARRIA10_SYSMGR_BOOTINFO_BSEL_SHIFT) {
	case 0:
		/* reserved */
		break;
	case 1:
		/* FPGA, currently not decoded */
		break;
	case 2:
	case 3:
		src = BOOTSOURCE_NAND;
		break;
	case 4:
	case 5:
		src = BOOTSOURCE_MMC;
		break;
	case 6:
	case 7:
		src = BOOTSOURCE_SPI;
		break;
	}

	return src;
}

static int arria10_boot_save_loc(void)
{
	enum bootsource src;

	src = arria10_get_bootsource();

	bootsource_set_raw(src, 0);

	return 0;
}
core_initcall(arria10_boot_save_loc);