/*
 * tps65912-spi.c  --  SPI access for TI TPS65912x PMIC
 *
 * Copyright 2011 Texas Instruments Inc.
 *
 * Author: Margarita Olaya Cabrera <magi@slimlogic.co.uk>
 *
 *  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 driver is based on wm8350 implementation.
 */

#include <linux/module.h>
#include <linux/moduleparam.h>
#include <linux/init.h>
#include <linux/slab.h>
#include <linux/gpio.h>
#include <linux/spi/spi.h>
#include <linux/mfd/core.h>
#include <linux/mfd/tps65912.h>
#include <linux/of_device.h>

static int tps65912_spi_write(struct tps65912 *tps65912, u8 addr,
							int bytes, void *src)
{
	struct spi_device *spi = tps65912->control_data;
	/* bit 23 is the read/write bit */
	u32 tx = 1 << 23 | addr << 15 | *(u8 *)src;
	struct spi_transfer xfer;
	struct spi_message msg;

	if ((spi == NULL) || (bytes == 0))
		return -ENODATA;

	if (bytes > 1)
		dev_err(tps65912->dev, "too many bytes to write %d\n", bytes);

	memset(&xfer, 0, sizeof(xfer));
	/* Do not setup the bits per transfer or the speed, as this causes the
	   driver to re-initialize, which wastes time. */
	xfer.tx_buf = &tx;
	xfer.len = sizeof(tx);
	xfer.bits_per_word = 24;
	spi_message_init(&msg);
	spi_message_add_tail(&xfer, &msg);
	return spi_sync(spi, &msg);
}

static int tps65912_spi_read(struct tps65912 *tps65912, u8 addr,
							int bytes, void *dest)
{
	struct spi_device *spi = tps65912->control_data;
	/* bit 23 is the read/write bit */
	u32 tx = 0 << 23 | addr << 15;
	struct spi_transfer xfer;
	struct spi_message msg;
	u32 rx;
	int ret;

	if ((spi == NULL) || (bytes == 0))
		return -ENODATA;

	if (bytes > 1)
		dev_err(tps65912->dev, "too many bytes to read %d\n", bytes);

	memset(&xfer, 0, sizeof(xfer));
	/* Do not setup the bits per transfer or the speed, as this causes the
	   driver to re-initialize, which wastes time. */
	xfer.tx_buf = &tx;
	xfer.rx_buf = &rx;
	xfer.len = sizeof(tx);
	xfer.bits_per_word = 24;
	spi_message_init(&msg);
	spi_message_add_tail(&xfer, &msg);
	ret = spi_sync(spi, &msg);
	if (!ret)
		*(u8 *)dest = (u8)rx & 0xff;
	return ret;
}

#ifdef CONFIG_OF
static struct of_device_id tps65912_of_match[] = {
	{ .compatible = "ti,tps65912" },
	{ },
};
MODULE_DEVICE_TABLE(of, tps65912_of_match);
#endif

static int tps65912_spi_probe(struct spi_device *spi)
{
	struct tps65912 *tps65912;

	tps65912 = kzalloc(sizeof(struct tps65912), GFP_KERNEL);
	if (tps65912 == NULL)
		return -ENOMEM;

	tps65912->dev = &spi->dev;
	tps65912->control_data = spi;
	tps65912->read = tps65912_spi_read;
	tps65912->write = tps65912_spi_write;

	mutex_init(&tps65912->pm_lock);

	spi_set_drvdata(spi, tps65912);

	return tps65912_device_init(tps65912);
}

static int tps65912_spi_remove(struct spi_device *spi)
{
	struct tps65912 *tps65912 = spi_get_drvdata(spi);

	tps65912_device_exit(tps65912);

	return 0;
}

int tps65912_spi_suspend_late(struct device *dev)
{
	struct tps65912 *tps65912 = dev_get_drvdata(dev);
	if (tps65912)
		mutex_lock(&tps65912->pm_lock);
	return 0;
}
int tps65912_spi_resume_early(struct device *dev)
{
	struct tps65912 *tps65912 = dev_get_drvdata(dev);
	if (tps65912)
		mutex_unlock(&tps65912->pm_lock);
	return 0;
}

static struct dev_pm_ops tps65912_pm_ops = {
	.suspend_late = tps65912_spi_suspend_late,
	.resume_early = tps65912_spi_resume_early,
};

static struct spi_driver tps65912_spi_driver = {
	.driver = {
		.name = "tps65912",
		.pm = &tps65912_pm_ops,
		.bus = &spi_bus_type,
		.owner = THIS_MODULE,
		.of_match_table = of_match_ptr(tps65912_of_match),
	},
	.probe	= tps65912_spi_probe,
	.remove = tps65912_spi_remove,
};

static int __init tps65912_spi_init(void)
{
	int ret;

	ret = spi_register_driver(&tps65912_spi_driver);
	if (ret != 0)
		pr_err("Failed to register TPS65912 SPI driver: %d\n", ret);

	return 0;
}
/* init early so consumer devices can complete system boot */
subsys_initcall(tps65912_spi_init);

static void __exit tps65912_spi_exit(void)
{
	spi_unregister_driver(&tps65912_spi_driver);
}
module_exit(tps65912_spi_exit);

MODULE_AUTHOR("Margarita Olaya	<magi@slimlogic.co.uk>");
MODULE_DESCRIPTION("SPI support for TPS65912 chip family mfd");
MODULE_LICENSE("GPL");