#include <framework/globalfunctions/crc_ccitt.h>
#include <framework/serviceinterface/ServiceInterfaceStream.h>
#include <framework/tmtcpacket/pus/TcPacketBase.h>
#include <string.h>

TcPacketBase::TcPacketBase(const uint8_t* set_data) :
		SpacePacketBase(set_data) {
	tcData = (TcPacketPointer*) set_data;
}

TcPacketBase::~TcPacketBase() {
	//Nothing to do.
}

uint8_t TcPacketBase::getService() {
	return tcData->data_field.service_type;
}

uint8_t TcPacketBase::getSubService() {
	return tcData->data_field.service_subtype;
}

uint8_t TcPacketBase::getAcknowledgeFlags() {
	return tcData->data_field.version_type_ack & 0b00001111;
}

const uint8_t* TcPacketBase::getApplicationData() const {
	return &tcData->data;
}

size_t TcPacketBase::getApplicationDataSize() {
	return getPacketDataLength() - sizeof(tcData->data_field) - CRC_SIZE + 1;
}

uint16_t TcPacketBase::getErrorControl() {
	uint16_t size = getApplicationDataSize() + CRC_SIZE;
	uint8_t* p_to_buffer = &tcData->data;
	return (p_to_buffer[size - 2] << 8) + p_to_buffer[size - 1];
}

void TcPacketBase::setErrorControl() {
	uint32_t full_size = getFullSize();
	uint16_t crc = ::Calculate_CRC(getWholeData(), full_size - CRC_SIZE);
	uint32_t size = getApplicationDataSize();
	(&tcData->data)[size] = (crc & 0XFF00) >> 8;	// CRCH
	(&tcData->data)[size + 1] = (crc) & 0X00FF; 		// CRCL
}

void TcPacketBase::setData(const uint8_t* p_Data) {
	SpacePacketBase::setData(p_Data);
	tcData = (TcPacketPointer*) p_Data;
}

uint8_t TcPacketBase::getSecondaryHeaderFlag() {
	return (tcData->data_field.version_type_ack & 0b10000000) >> 7;
}

uint8_t TcPacketBase::getPusVersionNumber() {
	return (tcData->data_field.version_type_ack & 0b01110000) >> 4;
}

void TcPacketBase::print() {
	uint8_t * wholeData = getWholeData();
	debug << "TcPacket contains: " << std::endl;
	for (uint8_t count = 0; count < getFullSize(); ++count) {
		debug << std::hex << (uint16_t) wholeData[count] << " ";
	}
	debug << std::dec << std::endl;
}

void TcPacketBase::initializeTcPacket(uint16_t apid, uint16_t sequenceCount,
		uint8_t ack, uint8_t service, uint8_t subservice) {
	initSpacePacketHeader(true, true, apid, sequenceCount);
	memset(&tcData->data_field, 0, sizeof(tcData->data_field));
	setPacketDataLength(sizeof(tcData->data_field) + CRC_SIZE);
	//Data Field Header:
	//Set CCSDS_secondary_header_flag to 0, version number to 001 and ack to 0000
	tcData->data_field.version_type_ack = 0b00010000;
	tcData->data_field.version_type_ack |= (ack & 0x0F);
	tcData->data_field.service_type = service;
	tcData->data_field.service_subtype = subservice;
}