/*
 * lib/atrf.c - ATRF access functions library
 *
 * Written 2010-2011 by Werner Almesberger
 * Copyright 2010-2011 Werner Almesberger
 *
 * 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.
 */


#include <stdlib.h>
#include <stdio.h>

#include "at86rf230.h"

#include "driver.h"
#include "atrf.h"


extern struct atrf_driver atusb_driver;
extern struct atrf_driver atben_driver;


struct atrf_dsc {
	struct atrf_driver *driver;
	void *handle;
	enum atrf_chip_id chip_id;
};


void *atrf_usb_handle(struct atrf_dsc *dsc)
{
#ifdef HAVE_USB
	return dsc->handle;
#else
	return NULL;
#endif
}


int atrf_error(struct atrf_dsc *dsc)
{
	return dsc->driver->error ? dsc->driver->error(dsc->handle) : 0;
}


int atrf_clear_error(struct atrf_dsc *dsc)
{
	return dsc->driver->clear_error ?
	    dsc->driver->clear_error(dsc->handle) : 0;
}


static enum atrf_chip_id identify(struct atrf_dsc *dsc)
{
	uint8_t part, version;

	part = atrf_reg_read(dsc, REG_PART_NUM);
	version = atrf_reg_read(dsc, REG_VERSION_NUM);
	switch (part) {
	case 2:	/* AT86RF230 */
		switch (version) {
		case 1: /* rev A */
		case 2: /* rev B */
			return artf_at86rf230;
		default:
			return atrf_unknown_chip;
		}
		break;
	case 3:	/* AT86RF231 */
		switch (version) {
		case 2: /* rev A */
			return artf_at86rf231;
		default:
			return atrf_unknown_chip;
		}
		break;
	default:
		return atrf_unknown_chip;
	}
	return atrf_unknown_chip;
}


struct atrf_dsc *atrf_open(void)
{
	struct atrf_dsc *dsc;
	struct atrf_driver *driver;
	void *handle;

#ifdef HAVE_USB
	driver = &atusb_driver;
#elif HAVE_BEN
	driver = &atben_driver;
#else
#error Need either HAVE_USB or HAVE_BEN
#endif
	handle = driver->open();
	if (!handle)
		return NULL;
	dsc = malloc(sizeof(*dsc));
	if (!dsc) {
		perror("malloc");
		exit(1);
	}
	dsc->driver = driver;
	dsc->handle = handle;
	dsc->chip_id = identify(dsc);
	return dsc;
}


void atrf_close(struct atrf_dsc *dsc)
{
	if (dsc->driver->close)
		dsc->driver->close(dsc->handle);
	free(dsc);
}


void atrf_reset(struct atrf_dsc *dsc)
{
	if (dsc->driver->reset)
		dsc->driver->reset(dsc->handle);
}


void atrf_reset_rf(struct atrf_dsc *dsc)
{
	dsc->driver->reset_rf(dsc->handle);
}


enum atrf_chip_id atrf_identify(struct atrf_dsc *dsc)
{
	return dsc->chip_id;
}


int atrf_test_mode(struct atrf_dsc *dsc)
{
	if (!dsc->driver->test_mode)
		return 0;
	dsc->driver->test_mode(dsc->handle);
	return 1;
}


int atrf_slp_tr(struct atrf_dsc *dsc, int on)
{
	if (!dsc->driver->slp_tr)
		return 0;
	dsc->driver->slp_tr(dsc->handle, on);
	return 1;
}


int atrf_set_clkm_generic(
    void (*reg_write)(void *dsc, uint8_t reg, uint8_t value),
    void *handle, int mhz)
{
	uint8_t clkm;

	if (!mhz) {
		reg_write(handle, REG_TRX_CTRL_0, 0); /* disable CLKM */
		return 1;
	}
	switch (mhz) {
	case 1:
		clkm = CLKM_CTRL_1MHz;
		break;
	case 2:
		clkm = CLKM_CTRL_2MHz;
		break;
	case 4:
		clkm = CLKM_CTRL_4MHz;
		break;
	case 8:
		clkm = CLKM_CTRL_8MHz;
		break;
	case 16:
		clkm = CLKM_CTRL_16MHz;
		break;
	default:
		fprintf(stderr, "unsupported CLKM frequency %d MHz\n", mhz);
		return 0;
	}
	reg_write(handle, REG_TRX_CTRL_0,
	    (PAD_IO_8mA << PAD_IO_CLKM_SHIFT) | clkm);
	return 1;
}


int atrf_set_clkm(struct atrf_dsc *dsc, int mhz)
{
	if (dsc->driver->set_clkm)
		return dsc->driver->set_clkm(dsc->handle, mhz);
	else
		return atrf_set_clkm_generic(dsc->driver->reg_write,
		    dsc->handle, mhz);
}


void atrf_reg_write(struct atrf_dsc *dsc, uint8_t reg, uint8_t value)
{
	dsc->driver->reg_write(dsc->handle, reg, value);
}


uint8_t atrf_reg_read(struct atrf_dsc *dsc, uint8_t reg)
{
	return dsc->driver->reg_read(dsc->handle, reg);
}


void atrf_buf_write(struct atrf_dsc *dsc, const void *buf, int size)
{
	dsc->driver->buf_write(dsc->handle, buf, size);
}


int atrf_buf_read(struct atrf_dsc *dsc, void *buf, int size)
{
	return dsc->driver->buf_read(dsc->handle, buf, size);
}


int atrf_interrupt(struct atrf_dsc *dsc)
{
	return
	    dsc->driver->interrupt ? dsc->driver->interrupt(dsc->handle) : 1;
}