/*

 * Master Firmware: Slave Sensor Data Aquisition

 *

 * Wireless Observational Modular Aerial Network

 * 

 * Ethan Zonca

 * Matthew Kanning

 * Kyle Ripperger

 * Matthew Kroening

 *

 */

#include "../config.h"

#include <avr/io.h>

#include <stdbool.h>

#include <stdlib.h>

#include <stdio.h>

#include <string.h>

#include <util/delay.h>

#include <avr/wdt.h>

#include "serial.h"

#include "serparser.h"

#include "slavesensors.h"

#include "sensordata.h"

#include "led.h"

#include "looptime.h"

uint8_t currentSlave = 0;

uint8_t currentSlaveSensor = 0;

bool requesting = false;

void slavesensors_setup() 

{

}

//#define DEBUG_NETWORKSCAN

char* bufPtr = 0x00;

static char slaveAddressLow[MAX_NUM_SLAVES][9];

static char slaveAddressHigh[MAX_NUM_SLAVES][9];

static char slaveNames[MAX_NUM_SLAVES][20];

uint8_t loggerIndex = 255;

uint8_t nodeCount = 0;

bool dataReady = false;

char* slavesensors_slavename(uint8_t id) {

	return slaveNames[id];

}

void slavesensors_network_scan() {

	serial0_ioff();

	int atOK;

	#ifdef DEBUG_OUTPUT

	serial0_sendString("Beginning network scan...\r\n\r\n");

	#endif

	_delay_ms(500); // xbee warmup

	wdt_reset();

	led_on(LED_ACTIVITY);

	atOK = slavesensors_enterAT();

	// wait for OK

	if(atOK == 0)

	{

		led_on(LED_CYCLE);

		serial0_sendString("ATND");

		serial0_sendChar(0x0D);

		// wait for scan to complete

		uint16_t scanStart = time_millis(); 		

		while(!serial0_hasChar()) {

			if(time_millis() - scanStart > 7000) {

				led_errorcode(ERROR_XBEETIMEOUT);

				return;

			}

			wdt_reset();

		}

		// Scan data end when newline by itself ("")	

		int lineCount = 0;	

	serial0_sendChar('+');

	_delay_ms(2);

	// Wait 1ms until we get "OK"

	//int atStart = time_millis();

	//while(!serial0_hasChar()) {

		//if(time_millis() - atStart > 500) {

			//led_errorcode(ERROR_ATFAIL);

			//wdt_reset();

			//return 1;

		//}

	//};	

	return xbeeIsOk();

}

int xbeeIsOk() 

{

	char* tmppntr = serial0_readLine();

	if(strcmp(tmppntr, "OK") == 0)

	{

		return 0;

	}

	else

	{

		led_errorcode(ERROR_NOXBEE);

		return 1;

	}

}

bool slavesensors_dataReady() {

	return dataReady;

}

bool slavesensors_isrequesting() 

{

	return requesting;	

}

void slavesensors_startprocess() 

{

	requesting = true;

	slavesensors_request();		

}

// TODO: inline. static.

void slavesensors_request() 

{

	slavesensors_selectnode(currentSlave);

	serial_sendCommand("@"); // Request data!

	slavesensors_selectlogger();

}

uint8_t numReadingsToExpect = 0; // number of values that the slave is about to send (testing)

// TODO: needs to skip logger!

void slavesensors_process(uint8_t parseResult) 

{

	if(!requesting) {

		// we got a command when we didn't request anything. probably skip it.

		return;

	}

	// TODO: timeout. If we're at NODATA for a long time and we are requesting, that's an issue.

	else if(parseResult == PARSERESULT_NODATA) {

		// Wait for data

	}

	// Finished reception of a message (one sensor data value). If not finished, send out command to get the next one

	else if(parseResult == PARSERESULT_PARSEOK)

	{

		#ifdef DEBUG_GETSLAVEDATA

		char debug[50];

		snprintf(debug, 50, "Slave %u sensor %u of total nodes %u\r\n", currentSlave, currentSlaveSensor,nodeCount);

		serial0_sendString(debug);

		#endif

		// We got some data, let's handle it

		// ASCII payload

		uint8_t len = getPayloadLength();

		uint8_t* load = getPayload();

		uint8_t type = getPayloadType();

		uint16_t parsedVal = atoi(load);

		// Special case for slave telling us how many things we're about to get		

		if(type + 0x30 == '@') {

			#ifdef DEBUG_GETSLAVEDATA

			serial0_sendString("Got an awesome count!\r\n");

			serial0_sendChar(parsedVal + 0x30);

			serial0_sendString("\r\n");

			#endif

			numReadingsToExpect = parsedVal;

			currentSlaveSensor = 0;

			requesting = true;

		}

		else {

			// Store data in structure

			sensordata_set(currentSlave,type,parsedVal);

			#ifdef DEBUG_GETSLAVEDATA

			serial0_sendString("Stored some sexy data!\r\n");

			#endif 

			// If we finished all sensors for all slaves

			if(currentSlave >= (nodeCount-1) && currentSlaveSensor >= (numReadingsToExpect-1))

			{

				#ifdef DEBUG_GETSLAVEDATA