/**
 ******************************************************************************
 * @addtogroup PIOS PIOS Core hardware abstraction layer
 * @{
 * @addtogroup PIOS_Spektrum Spektrum/JR DSMx satellite receiver functions
 * @brief Code to bind and read Spektrum/JR DSMx satellite receiver serial stream
 * @{
 *
 * @file       pios_spektrum.c
 * @author     The OpenPilot Team, http://www.openpilot.org Copyright (C) 2011.
 * @brief      Code bind and read Spektrum/JR DSMx satellite receiver serial stream
 * @see        The GNU Public License (GPL) Version 3
 *
 *****************************************************************************/
/* 
 * 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 3 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
 */

/* Project Includes */
#include "pios.h"
#include "pios_spektrum_priv.h"

#if defined(PIOS_INCLUDE_SPEKTRUM)

/* Forward Declarations */
static int32_t PIOS_Spektrum_Get(uint32_t rcvr_id, uint8_t channel);
static uint16_t PIOS_Spektrum_RxInCallback(uint32_t context,
					   uint8_t *buf,
					   uint16_t buf_len,
					   uint16_t *headroom,
					   bool *need_yield);
static void PIOS_Spektrum_Supervisor(uint32_t spektrum_id);

/* Local Variables */
const struct pios_rcvr_driver pios_spektrum_rcvr_driver = {
	.read = PIOS_Spektrum_Get,
};

enum pios_spektrum_dev_magic {
	PIOS_SPEKTRUM_DEV_MAGIC = 0x44534d78,
};

struct pios_spektrum_state {
	uint16_t channel_data[PIOS_SPEKTRUM_NUM_INPUTS];
	uint8_t received_data[SPEKTRUM_FRAME_LENGTH];
	uint8_t receive_timer;
	uint8_t failsafe_timer;
	uint8_t frame_found;
	uint8_t byte_count;
	uint8_t resolution;
	uint8_t frame_mode;
#ifdef SPEKTRUM_LOST_FRAME_COUNTER
	uint8_t	frames_lost_last;
	uint16_t frames_lost;
#endif
};

struct pios_spektrum_dev {
	enum pios_spektrum_dev_magic magic;
	const struct pios_spektrum_cfg *cfg;
	struct pios_spektrum_state state;
};

/* Allocate Spektrum device descriptor */
#if defined(PIOS_INCLUDE_FREERTOS)
static struct pios_spektrum_dev *PIOS_Spektrum_Alloc(void)
{
	struct pios_spektrum_dev *spektrum_dev;

	spektrum_dev = (struct pios_spektrum_dev *)pvPortMalloc(sizeof(*spektrum_dev));
	if (!spektrum_dev)
		return NULL;

	spektrum_dev->magic = PIOS_SPEKTRUM_DEV_MAGIC;
	return spektrum_dev;
}
#else
static struct pios_spektrum_dev pios_spektrum_devs[PIOS_SPEKTRUM_MAX_DEVS];
static uint8_t pios_spektrum_num_devs;
static struct pios_spektrum_dev *PIOS_Spektrum_Alloc(void)
{
	struct pios_spektrum_dev *spektrum_dev;

	if (pios_spektrum_num_devs >= PIOS_SPEKTRUM_MAX_DEVS)
		return NULL;

	spektrum_dev = &pios_spektrum_devs[pios_spektrum_num_devs++];
	spektrum_dev->magic = PIOS_SPEKTRUM_DEV_MAGIC;

	return spektrum_dev;
}
#endif

/* Validate Spektrum device descriptor */
static bool PIOS_Spektrum_Validate(struct pios_spektrum_dev *spektrum_dev)
{
	return (spektrum_dev->magic == PIOS_SPEKTRUM_DEV_MAGIC);
}

/* Try to bind DSMx satellite using specified number of pulses */
static void PIOS_Spektrum_Bind(const struct pios_spektrum_cfg *cfg, uint8_t bind)
{
	GPIO_InitTypeDef GPIO_InitStructure;
	GPIO_InitStructure.GPIO_Pin = cfg->bind.init.GPIO_Pin;
	GPIO_InitStructure.GPIO_Speed = cfg->bind.init.GPIO_Speed;
	GPIO_InitStructure.GPIO_Mode = GPIO_Mode_IPU;

	/* just to limit bind pulses */
	if (bind > 10)
		bind = 10;

	GPIO_Init(cfg->bind.gpio, &cfg->bind.init);

	/* RX line, set high */
	GPIO_SetBits(cfg->bind.gpio, cfg->bind.init.GPIO_Pin);

	/* on CC works up to 140ms, guess bind window is around 20-140ms after power up */
	PIOS_DELAY_WaitmS(60);

	for (int i = 0; i < bind ; i++) {
		/* RX line, drive low for 120us */
		GPIO_ResetBits(cfg->bind.gpio, cfg->bind.init.GPIO_Pin);
		PIOS_DELAY_WaituS(120);
		/* RX line, drive high for 120us */
		GPIO_SetBits(cfg->bind.gpio, cfg->bind.init.GPIO_Pin);
		PIOS_DELAY_WaituS(120);
	}
	/* RX line, set input and wait for data */
	GPIO_Init(cfg->bind.gpio, &GPIO_InitStructure);
}

/* Reset channels in case of lost signal or explicit failsafe receiver flag */
static void PIOS_Spektrum_ResetChannels(struct pios_spektrum_state *state)
{
	for (int i = 0; i < PIOS_SPEKTRUM_NUM_INPUTS; i++) {
		state->channel_data[i] = PIOS_RCVR_TIMEOUT;
	}
}

/**
 * Get the value of an input channel
 * \param[in] channel Number of the channel desired (zero based)
 * \output PIOS_RCVR_INVALID channel not available
 * \output PIOS_RCVR_TIMEOUT failsafe condition or missing receiver
 * \output >0 channel value
 */
static int32_t PIOS_Spektrum_Get(uint32_t rcvr_id, uint8_t channel)
{
	struct pios_spektrum_dev *spektrum_dev = (struct pios_spektrum_dev *)rcvr_id;

	if (!PIOS_Spektrum_Validate(spektrum_dev))
		return PIOS_RCVR_INVALID;

	/* return error if channel is not available */
	if (channel >= PIOS_SPEKTRUM_NUM_INPUTS)
		return PIOS_RCVR_INVALID;

	/* may also be PIOS_RCVR_TIMEOUT set by other function */
	return spektrum_dev->state.channel_data[channel];
}

/* Reset Spektrum receiver state */
static void PIOS_Spektrum_ResetState(struct pios_spektrum_state *state)
{
	state->receive_timer = 0;
	state->failsafe_timer = 0;
	state->frame_found = 0;
#ifdef SPEKTRUM_LOST_FRAME_COUNTER
	state->frames_lost_last = 0;
	state->frames_lost = 0;
#endif
	PIOS_Spektrum_ResetChannels(state);
}

/**
 * Check and unroll complete frame data.
 * \output 0 frame data accepted
 * \output -1 frame error found
 */
static int PIOS_Spektrum_UnrollChannels(struct pios_spektrum_state *state)
{
#ifdef SPEKTRUM_LOST_FRAME_COUNTER
	/* increment the lost frame counter */
	uint8_t frames_lost = state->received_data[0];
	state->frames_lost += (frames_lost - state->frames_lost_last);
	state->frames_lost_last = frames_lost;
#endif

	/* check the frame type assuming master satellite stream */
	uint8_t type = state->received_data[1];
	switch (type) {
	case 0x01:
	case 0x02:
	case 0x12:
		state->resolution = (type & SPEKTRUM_DSM2_RES_MASK) ? 11 : 10;
		state->frame_mode = (type & SPEKTRUM_DSM2_FNUM_MASK);
		break;
	case 0xB2:
		/* FIXME: we should guess the resolution using frame mode found */
		goto fail;
	default:
		goto fail;
	}

	/* unroll channels */
	uint8_t *s = &(state->received_data[2]);
	uint16_t mask = (state->resolution == 10) ? 0x03ff : 0x07ff;

	for (int i = 0; i < SPEKTRUM_CHANNELS_PER_FRAME; i++) {
		uint16_t word = ((uint16_t)s[0] << 8) | s[1];
		s += 2;

		/* check for frame mode or missing channel data */
		if (word & SPEKTRUM_2ND_FRAME_MASK) {
			/* this is ok for the 1st word only */
			if (i == 0) {
				/* means the 2-frame mode */
				state->frame_mode = 2;
			} else {
				if (word == 0xffff)
					/* skip the missing channel */
					continue;
				else
					/* wrong frame received */
					goto fail;
			}
		}

		/* extract and save the channel value */
		uint8_t channel_num = (word >> state->resolution) & 0x0f;
		if (channel_num < PIOS_SPEKTRUM_NUM_INPUTS)
			state->channel_data[channel_num] = (word & mask);
	}

#ifdef SPEKTRUM_LOST_FRAME_COUNTER
	/* put lost frames counter into the last channel for debugging */
	state->channel_data[PIOS_SPEKTRUM_NUM_INPUTS-1] = state->frames_lost;
#endif

	/* all channels processed, missing channels skipped */
	return 0;

fail:
	return -1;
}

/* Update decoder state processing input byte from the DSMx stream */
static void PIOS_Spektrum_UpdateState(struct pios_spektrum_state *state, uint8_t b)
{
	if (state->frame_found) {
		/* receiving the data frame */
		if (state->byte_count < SPEKTRUM_FRAME_LENGTH) {
			/* store next byte */
			state->received_data[state->byte_count++] = b;
			if (state->byte_count == SPEKTRUM_FRAME_LENGTH) {
				/* full frame received - process and wait for new one */
				if (!PIOS_Spektrum_UnrollChannels(state))
					/* data looking good */
					state->failsafe_timer = 0;

				/* prepare for the next frame */
				state->frame_found = 0;
			}
		}
	}
}

/* Initialise Spektrum receiver interface */
int32_t PIOS_Spektrum_Init(uint32_t *spektrum_id,
			   const struct pios_spektrum_cfg *cfg,
			   const struct pios_com_driver *driver,
			   uint32_t lower_id,
			   uint8_t bind)
{
	PIOS_DEBUG_Assert(spektrum_id);
	PIOS_DEBUG_Assert(cfg);
	PIOS_DEBUG_Assert(driver);

	struct pios_spektrum_dev *spektrum_dev;

	spektrum_dev = (struct pios_spektrum_dev *)PIOS_Spektrum_Alloc();
	if (!spektrum_dev)
		return -1;

	/* Bind the configuration to the device instance */
	spektrum_dev->cfg = cfg;

	/* Bind the receiver if requested */
	if (bind)
		PIOS_Spektrum_Bind(cfg, bind);

	PIOS_Spektrum_ResetState(&(spektrum_dev->state));

	*spektrum_id = (uint32_t)spektrum_dev;

	/* Set comm driver callback */
	(driver->bind_rx_cb)(lower_id, PIOS_Spektrum_RxInCallback, *spektrum_id);

	if (!PIOS_RTC_RegisterTickCallback(PIOS_Spektrum_Supervisor, *spektrum_id)) {
		PIOS_DEBUG_Assert(0);
	}

	return 0;
}

/* Comm byte received callback */
static uint16_t PIOS_Spektrum_RxInCallback(uint32_t context,
					   uint8_t *buf,
					   uint16_t buf_len,
					   uint16_t *headroom,
					   bool *need_yield)
{
	struct pios_spektrum_dev *spektrum_dev = (struct pios_spektrum_dev *)context;

	bool valid = PIOS_Spektrum_Validate(spektrum_dev);
	PIOS_Assert(valid);

	struct pios_spektrum_state *state = &(spektrum_dev->state);

	/* process byte(s) and clear receive timer */
	for (uint8_t i = 0; i < buf_len; i++) {
		PIOS_Spektrum_UpdateState(state, buf[i]);
		state->receive_timer = 0;
	}

	/* Always signal that we can accept another byte */
	if (headroom)
		*headroom = SPEKTRUM_FRAME_LENGTH;

	/* We never need a yield */
	*need_yield = false;

	/* Always indicate that all bytes were consumed */
	return buf_len;
}

/**
 * Input data supervisor is called periodically and provides
 * two functions: frame syncing and failsafe triggering.
 *
 * Spektrum frames come at 11ms or 22ms rate at 115200bps.
 * RTC timer is running at 625Hz (1.6ms). So with divider 5 it gives
 * 8ms pause between frames which is good for both Spektrum frame rates.
 *
 * Data receive function must clear the receive_timer to confirm new
 * data reception. If no new data received in 100ms, we must call the
 * failsafe function which clears all channels.
 */
static void PIOS_Spektrum_Supervisor(uint32_t spektrum_id)
{
	struct pios_spektrum_dev *spektrum_dev = (struct pios_spektrum_dev *)spektrum_id;

	bool valid = PIOS_Spektrum_Validate(spektrum_dev);
	PIOS_Assert(valid);

	struct pios_spektrum_state *state = &(spektrum_dev->state);

	/* waiting for new frame if no bytes were received in 8ms */
	if (++state->receive_timer > 4) {
		state->frame_found = 1;
		state->byte_count = 0;
		state->receive_timer = 0;
	}

	/* activate failsafe if no frames have arrived in 102.4ms */
	if (++state->failsafe_timer > 64) {
		PIOS_Spektrum_ResetChannels(state);
		state->failsafe_timer = 0;
	}
}

#endif	/* PIOS_INCLUDE_SPEKTRUM */

/** 
 * @}
 * @}
 */