// --------------------------------------------------------------
#include <string.h>
#include <math.h>
#include <stdint.h>
#include <algorithm>
#include <stdio.h>

#include <crc/crc.h>
#include "DeFormatter.hpp"

// --------------------------------------------------------------
// Deformatter
// --------------------------------------------------------------
DeFormatter::DeFormatter(IDeFormatter *pDataHandler)
	: m_pDataHandler(pDataHandler)
	, m_scrambler(FRAME_SCRAMBLER_SEED, FRAME_SCRAMBLER_POLY)
	, m_dataCount(0)
	, m_state(Idle)
{
	resetStats();
}

DeFormatter::~DeFormatter()
{
}

void DeFormatter::process(uint8_t data)
{
	if (m_state == Idle)
	{
		return;
	}

	uint8_t *pData;
	if (m_state == Header)
	{
		pData = (uint8_t*)&m_Hdr;
		pData[m_dataCount++] = data;
		if (m_dataCount == (sizeof(frame_hdr_t)))
		{
			m_scrambler.stateSet(m_Hdr.prn_state);
			m_scrambler.process((uint8_t*)&m_Hdr.crc16, sizeof(m_Hdr.crc16));
			m_scrambler.process((uint8_t*)&m_Hdr.type, sizeof(m_Hdr.type));
			m_scrambler.process((uint8_t*)&m_Hdr.length, sizeof(m_Hdr.length));
			m_dataCount = 0; 
			m_state = Data;
		}
	}
	else if (m_state == Data)
	{
		pData = (uint8_t*)&m_data;
		pData[m_dataCount++] = data;
		if (m_dataCount == m_Hdr.length)
		{
			m_scrambler.process(m_data, FRAME_NUM_BYTES_PER_FRAME);
			uint16_t crc16_ist = Crc16(FRAME_CRC16_POLY, FRAME_CRC16_IV, m_data, FRAME_NUM_BYTES_PER_FRAME);
			if (crc16_ist == m_Hdr.crc16)
			{
				// Frame is valid
				if (m_Hdr.type == FRAME_TYPE_DATA)
				{
					m_frameStats.numDataBytes += m_Hdr.length;
					m_pDataHandler->onData(m_data, m_Hdr.length);
				}
			}
			m_frameStats.numDataFrame += (uint32_t)(m_Hdr.type == FRAME_TYPE_DATA);
			m_frameStats.numIdleFrame += (uint32_t)(m_Hdr.type == FRAME_TYPE_IDLE);
			m_frameStats.numInvalidFrame += (uint32_t)(crc16_ist != m_Hdr.crc16);
			m_frameStats.numTotalFrame++;
			m_dataCount = 0;
			m_state = Idle;
		}
	}
}

frame_statistics_t DeFormatter::getStats()
{
	return m_frameStats;
}

void DeFormatter::resetStats()
{
	memset(&m_frameStats, 0, sizeof(frame_statistics_t));
}

void DeFormatter::onReset()
{
	m_dataCount = 0;
	m_state = Header;
};

void DeFormatter::onData(data_t data)
{
	process((uint8_t)data);
}

// --------------------------------------------------------------