/*
 * (C) Copyright 2002
 * Gerald Van Baren, Custom IDEAS, vanbaren@cideas.com
 *
 * See file CREDITS for list of people who contributed to this
 * project.
 *
 * 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.
 *
 * 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
 */

/*
 * SPI Read/Write Utilities
 */

#include <common.h>
#include <command.h>
#include <spi.h>

/*-----------------------------------------------------------------------
 * Definitions
 */

#ifndef MAX_SPI_BYTES
#   define MAX_SPI_BYTES 32	/* Maximum number of bytes we can handle */
#endif

#ifndef CONFIG_DEFAULT_SPI_BUS
#   define CONFIG_DEFAULT_SPI_BUS	0
#endif
#ifndef CONFIG_DEFAULT_SPI_MODE
#   define CONFIG_DEFAULT_SPI_MODE	SPI_MODE_0
#endif

/*
 * Values from last command.
 */
static unsigned int	device;
static int		bitlen;
static uchar		dout[MAX_SPI_BYTES];
static uchar		din[MAX_SPI_BYTES];
static uchar		*din_p		= (uchar *) &din;
static unsigned int	bus		= CONFIG_DEFAULT_SPI_BUS;
static unsigned int	mode		= CONFIG_DEFAULT_SPI_MODE;
static unsigned int	max_hz		= 1000000;

/*
 * SPI read/write
 *
 * Syntax:
 *   spi {dev} {num_bits} {dout} {use_din} {din} {bus} {max_hz}
 *     {dev} is the device number for controlling chip select (see TBD)
 *     {num_bits} is the number of bits to send & receive (base 10)
 *     {dout} is a hexadecimal string of data to send
 *     {use_din} is wether we should ignore the received data or not
 *     {bus} is which on the SPI controller to use
 *     {mode} is the mode flag used for initializing the SPI slave
 *     {max_hz} is the maximum allowed frequenzy
 * The command prints the hexadecimal string sent and received via SPI.
 */

int do_spi (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])
{
	struct spi_slave *slave;
	char  *cp = 0;
	uchar tmp;
	int   j;
	int   rcode = 0;

	/*
	 * We use the last specified parameters, unless new ones are
	 * entered.
	 */
	if ((flag & CMD_FLAG_REPEAT) == 0)
	{
		if (argc >= 2)
			device = simple_strtoul(argv[1], NULL, 10);
		if (argc >= 3)
			bitlen = simple_strtoul(argv[2], NULL, 10);
		if (argc >= 4) {
			cp = argv[3];
			for(j = 0; *cp; j++, cp++) {
				tmp = *cp - '0';
				if(tmp > 9)
					tmp -= ('A' - '0') - 10;
				if(tmp > 15)
					tmp -= ('a' - 'A');
				if(tmp > 15) {
					printf("Hex conversion error on %c, giving up.\n", *cp);
					return 1;
				}
				if((j % 2) == 0)
					dout[j / 2] = (tmp << 4);
				else
					dout[j / 2] |= tmp;
			}
		}
		if (argc >= 5) {
			if (0 == simple_strtoul(argv[4], NULL, 10))
				din_p = NULL;
			else
				din_p = (uchar *) &din;
		}
		if (argc >= 6)
			bus = simple_strtoul(argv[5], NULL, 10);
		if (argc >= 7)
			mode = simple_strtoul(argv[6], NULL, 10);
		if (argc >= 8)
			max_hz = simple_strtoul(argv[7], NULL, 10);
	}

	if ((bitlen < 0) || (bitlen >  (MAX_SPI_BYTES * 8))) {
		printf("Invalid bitlen %d, giving up.\n", bitlen);
		return 1;
	}

	slave = spi_setup_slave(bus, device, max_hz, mode);
	if (!slave) {
		printf("Invalid device %d, giving up.\n", device);
		return 1;
	}

	debug ("spi chipsel = %08X\n", device);

	printf("dout: 0x");
	for (j = 0; j < ((bitlen + 7) / 8); j++)
		printf("%02X", dout[j]);
	printf("\n");

	spi_claim_bus(slave);
	if (spi_xfer(slave, bitlen, dout, din_p,
				SPI_XFER_BEGIN | SPI_XFER_END) != 0) {
		printf("Error with the SPI transaction.\n");
		rcode = 1;
	} else {
		if (din_p != NULL) {
			printf("din:  0x");
			for (j = 0; j < ((bitlen + 7) / 8); j++)
				printf("%02X", din[j]);
			printf("\n");
		} else
			printf("SPI message sent successfully.\n");
	}
	spi_release_bus(slave);
	spi_free_slave(slave);

	return rcode;
}

/***************************************************/

U_BOOT_CMD(
	sspi,	8,	1,	do_spi,
	"SPI utility commands",
	"Send <bit_len> bits from <dout> out the SPI\n"
	"<device> <bit_len> <dout> <use_din> <bus> <mode> <max_hz>\n"
	"  <device>  - Identifies the chip select of the device\n"
	"  <bit_len> - Number of bits to send (base 10)\n"
	"  <dout>    - Hexadecimal string that gets sent\n"
	"  <use_din> - 0 if no data should be read, != 0 otherwise\n"
	"  <bus>     - The bus of the SPI controller\n"
	"  <mode>    - Mode flag\n"
	"  <max_hz>  - The maximum allowed frequency for the transfers\n"
);