/*
 * Common fdt based SPI driver front end
 *
 * Copyright (c) 2013 NVIDIA Corporation
 *
 * See file CREDITS for list of people who contributed to this
 * project.
 *
 * This software is licensed under the terms of the GNU General Public
 * License version 2, as published by the Free Software Foundation, and
 * may be copied, distributed, and modified under those terms.
 *
 * 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.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston,
 * MA 02111-1307 USA
 */

#include <common.h>
#include <malloc.h>
#include <asm/io.h>
#include <asm/gpio.h>
#include <asm/arch/clock.h>
#include <asm/arch-tegra/clk_rst.h>
#include <asm/arch-tegra20/tegra20_sflash.h>
#include <asm/arch-tegra20/tegra20_slink.h>
#include <asm/arch-tegra114/tegra114_spi.h>
#include <spi.h>
#include <fdtdec.h>

DECLARE_GLOBAL_DATA_PTR;

struct fdt_spi_driver {
        int compat;
        int max_ctrls;
        int (*init)(int *node_list, int count);
        int (*claim_bus)(struct spi_slave *slave);
        int (*release_bus)(struct spi_slave *slave);
        int (*cs_is_valid)(unsigned int bus, unsigned int cs);
        struct spi_slave *(*setup_slave)(unsigned int bus, unsigned int cs,
                                        unsigned int max_hz, unsigned int mode);
        void (*free_slave)(struct spi_slave *slave);
        void (*cs_activate)(struct spi_slave *slave);
        void (*cs_deactivate)(struct spi_slave *slave);
        int (*xfer)(struct spi_slave *slave, unsigned int bitlen,
                    const void *data_out, void *data_in, unsigned long flags);
};

static struct fdt_spi_driver fdt_spi_drivers[] = {
#ifdef CONFIG_TEGRA20_SFLASH
        {
                .compat         = COMPAT_NVIDIA_TEGRA20_SFLASH,
                .max_ctrls      = 1,
                .init           = tegra20_spi_init,
                .claim_bus      = tegra20_spi_claim_bus,
                .cs_is_valid    = tegra20_spi_cs_is_valid,
                .setup_slave    = tegra20_spi_setup_slave,
                .free_slave     = tegra20_spi_free_slave,
                .cs_activate    = tegra20_spi_cs_activate,
                .cs_deactivate  = tegra20_spi_cs_deactivate,
                .xfer           = tegra20_spi_xfer,
        },
#endif
#ifdef CONFIG_TEGRA20_SLINK
        {
                .compat         = COMPAT_NVIDIA_TEGRA20_SLINK,
                .max_ctrls      = CONFIG_TEGRA_SLINK_CTRLS,
                .init           = tegra30_spi_init,
                .claim_bus      = tegra30_spi_claim_bus,
                .cs_is_valid    = tegra30_spi_cs_is_valid,
                .setup_slave    = tegra30_spi_setup_slave,
                .free_slave     = tegra30_spi_free_slave,
                .cs_activate    = tegra30_spi_cs_activate,
                .cs_deactivate  = tegra30_spi_cs_deactivate,
                .xfer           = tegra30_spi_xfer,
        },
#endif
#ifdef CONFIG_TEGRA114_SPI
        {
                .compat         = COMPAT_NVIDIA_TEGRA114_SPI,
                .max_ctrls      = CONFIG_TEGRA114_SPI_CTRLS,
                .init           = tegra114_spi_init,
                .claim_bus      = tegra114_spi_claim_bus,
                .cs_is_valid    = tegra114_spi_cs_is_valid,
                .setup_slave    = tegra114_spi_setup_slave,
                .free_slave     = tegra114_spi_free_slave,
                .cs_activate    = tegra114_spi_cs_activate,
                .cs_deactivate  = tegra114_spi_cs_deactivate,
                .xfer           = tegra114_spi_xfer,
        },
#endif
};

static struct fdt_spi_driver *driver;

int spi_cs_is_valid(unsigned int bus, unsigned int cs)
{
        if (!driver)
                return 0;
        else if (!driver->cs_is_valid)
                return 1;
        else
                return driver->cs_is_valid(bus, cs);
}

struct spi_slave *spi_setup_slave(unsigned int bus, unsigned int cs,
                unsigned int max_hz, unsigned int mode)
{
        if (!driver || !driver->setup_slave)
                return NULL;

        return driver->setup_slave(bus, cs, max_hz, mode);
}

void spi_free_slave(struct spi_slave *slave)
{
        if (driver && driver->free_slave)
                return driver->free_slave(slave);
}

static int spi_init_driver(struct fdt_spi_driver *driver)
{
        int count;
        int node_list[driver->max_ctrls];

        count = fdtdec_find_aliases_for_id(gd->fdt_blob, "spi",
                                           driver->compat,
                                           node_list,
                                           driver->max_ctrls);
        return driver->init(node_list, count);
}

void spi_init(void)
{
        int i;

        for (i = 0; i < ARRAY_SIZE(fdt_spi_drivers); i++) {
                driver = &fdt_spi_drivers[i];
                if (!spi_init_driver(driver))
                        break;
        }
        if (i == ARRAY_SIZE(fdt_spi_drivers))
                driver = NULL;
}

int spi_claim_bus(struct spi_slave *slave)
{
        if (!driver)
                return 1;
        if (!driver->claim_bus)
                return 0;

        return driver->claim_bus(slave);
}

void spi_release_bus(struct spi_slave *slave)
{
        if (driver && driver->release_bus)
                driver->release_bus(slave);
}

void spi_cs_activate(struct spi_slave *slave)
{
        if (driver && driver->cs_activate)
                driver->cs_activate(slave);
}

void spi_cs_deactivate(struct spi_slave *slave)
{
        if (driver && driver->cs_deactivate)
                driver->cs_deactivate(slave);
}

int spi_xfer(struct spi_slave *slave, unsigned int bitlen,
             const void *data_out, void *data_in, unsigned long flags)
{
        if (!driver || !driver->xfer)
                return -1;

        return driver->xfer(slave, bitlen, data_out, data_in, flags);
}