/*
*   Copyright (C) 2016,2023,2025 by Jonathan Naylor G4KLX
*
*   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.
*
*   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., 675 Mass Ave, Cambridge, MA 02139, USA.
*/

#include "DStarSlowData.h"
#include "DStarDefines.h"
#include "Utils.h"
#include "CRC.h"
#include "Log.h"

#include <cstdio>
#include <cassert>
#include <cstring>

CDStarSlowData::CDStarSlowData() :
m_header(nullptr),
m_ptr(0U),
m_buffer(nullptr),
m_text(nullptr),
m_textPtr(0U),
m_textBits(0x00U),
m_state(SDD_STATE::FIRST),
m_complete(false)
{
	m_header = new unsigned char[50U];		// DSTAR_HEADER_LENGTH_BYTES
	m_buffer = new unsigned char[DSTAR_DATA_FRAME_LENGTH_BYTES * 2U];
	m_text   = new unsigned char[24U];
}

CDStarSlowData::~CDStarSlowData()
{
	delete[] m_header;
	delete[] m_buffer;
	delete[] m_text;
}

void CDStarSlowData::add(const unsigned char* data)
{
	assert(data != nullptr);

	switch (m_state) {
	case SDD_STATE::FIRST:
		m_buffer[0U] = data[9U]  ^ DSTAR_SCRAMBLER_BYTES[0U];
		m_buffer[1U] = data[10U] ^ DSTAR_SCRAMBLER_BYTES[1U];
		m_buffer[2U] = data[11U] ^ DSTAR_SCRAMBLER_BYTES[2U];
		m_state = SDD_STATE::SECOND;
		m_complete = false;
		return;

	case SDD_STATE::SECOND:
		m_buffer[3U] = data[9U]  ^ DSTAR_SCRAMBLER_BYTES[0U];
		m_buffer[4U] = data[10U] ^ DSTAR_SCRAMBLER_BYTES[1U];
		m_buffer[5U] = data[11U] ^ DSTAR_SCRAMBLER_BYTES[2U];
		m_state = SDD_STATE::FIRST;
		m_complete = true;
		CUtils::dump(1U, "D-Star slow data element", m_buffer, 6U);
		loadHeader();
		loadText();
		break;
	}
}

void CDStarSlowData::loadHeader()
{
	if ((m_buffer[0U] & DSTAR_SLOW_DATA_TYPE_MASK) != DSTAR_SLOW_DATA_TYPE_HEADER)
		return;

	if (m_ptr >= 45U)
		return;

	::memcpy(m_header + m_ptr, m_buffer + 1U, 5U);
	m_ptr += 5U;
}

CDStarHeader* CDStarSlowData::getHeader()
{
	// Clean up the data
	m_header[0U] &= (DSTAR_INTERRUPTED_MASK | DSTAR_URGENT_MASK | DSTAR_REPEATER_MASK);
	m_header[1U] = 0x00U;
	m_header[2U] = 0x00U;

	for (unsigned int i = 3U; i < 39U; i++)
		m_header[i] &= 0x7FU;

	// Check the CRC
	bool ret = CCRC::checkCCITT161(m_header, DSTAR_HEADER_LENGTH_BYTES);
	if (!ret) {
		if (m_ptr == 45U)
			LogMessage("D-Star, invalid slow data header");
		return nullptr;
	}

	return new CDStarHeader(m_header);
}

void CDStarSlowData::loadText()
{
	switch (m_buffer[0U]) {
	case DSTAR_SLOW_DATA_TYPE_TEXT | 0U:
		CUtils::dump(1U, "D-Star slow data text fragment 0", m_buffer, 6U);
		m_text[0U] = m_buffer[1U] & 0x7FU;
		m_text[1U] = m_buffer[2U] & 0x7FU;
		m_text[2U] = m_buffer[3U] & 0x7FU;
		m_text[3U] = m_buffer[4U] & 0x7FU;
		m_text[4U] = m_buffer[5U] & 0x7FU;
		m_textBits |= 0x01U;
		break;

	case DSTAR_SLOW_DATA_TYPE_TEXT | 1U:
		CUtils::dump(1U, "D-Star slow data text fragment 1", m_buffer, 6U);
		m_text[5U] = m_buffer[1U] & 0x7FU;
		m_text[6U] = m_buffer[2U] & 0x7FU;
		m_text[7U] = m_buffer[3U] & 0x7FU;
		m_text[8U] = m_buffer[4U] & 0x7FU;
		m_text[9U] = m_buffer[5U] & 0x7FU;
		m_textBits |= 0x02U;
		break;

	case DSTAR_SLOW_DATA_TYPE_TEXT | 2U:
		CUtils::dump(1U, "D-Star slow data text fragment 2", m_buffer, 6U);
		m_text[10U] = m_buffer[1U] & 0x7FU;
		m_text[11U] = m_buffer[2U] & 0x7FU;
		m_text[12U] = m_buffer[3U] & 0x7FU;
		m_text[13U] = m_buffer[4U] & 0x7FU;
		m_text[14U] = m_buffer[5U] & 0x7FU;
		m_textBits |= 0x04U;
		break;

	case DSTAR_SLOW_DATA_TYPE_TEXT | 3U:
		CUtils::dump(1U, "D-Star slow data text fragment 3", m_buffer, 6U);
		m_text[15U] = m_buffer[1U] & 0x7FU;
		m_text[16U] = m_buffer[2U] & 0x7FU;
		m_text[17U] = m_buffer[3U] & 0x7FU;
		m_text[18U] = m_buffer[4U] & 0x7FU;
		m_text[19U] = m_buffer[5U] & 0x7FU;
		m_text[20U] = 0x00U;
		m_textBits |= 0x08U;
		break;

	default:
		break;
	}
}

const unsigned char* CDStarSlowData::getText()
{
	if (m_textBits != 0x0FU)
		return nullptr;

	CUtils::dump(1U, "D-Star slow data text", m_text, 20U);

	m_textBits = 0x00U;

	return m_text;
}

void CDStarSlowData::start()
{
	::memset(m_header, 0x00U, DSTAR_HEADER_LENGTH_BYTES);

	m_ptr      = 0U;
	m_state    = SDD_STATE::FIRST;
	m_textBits = 0x00U;
	m_complete = false;
}

void CDStarSlowData::reset()
{
	m_ptr      = 0U;
	m_state    = SDD_STATE::FIRST;
	m_textBits = 0x00U;
	m_complete = false;
}

void CDStarSlowData::setText(const char* text)
{
	assert(text != nullptr);

	m_text[0U] = DSTAR_SLOW_DATA_TYPE_TEXT | 0U;
	m_text[1U] = text[0U];
	m_text[2U] = text[1U];
	m_text[3U] = text[2U];
	m_text[4U] = text[3U];
	m_text[5U] = text[4U];

	m_text[6U] = DSTAR_SLOW_DATA_TYPE_TEXT | 1U;
	m_text[7U] = text[5U];
	m_text[8U] = text[6U];
	m_text[9U] = text[7U];
	m_text[10U] = text[8U];
	m_text[11U] = text[9U];

	m_text[12U] = DSTAR_SLOW_DATA_TYPE_TEXT | 2U;
	m_text[13U] = text[10U];
	m_text[14U] = text[11U];
	m_text[15U] = text[12U];
	m_text[16U] = text[13U];
	m_text[17U] = text[14U];

	m_text[18U] = DSTAR_SLOW_DATA_TYPE_TEXT | 3U;
	m_text[19U] = text[15U];
	m_text[20U] = text[16U];
	m_text[21U] = text[17U];
	m_text[22U] = text[18U];
	m_text[23U] = text[19U];

	m_textPtr  = 0U;
	m_textBits = 0x00U;
}

void CDStarSlowData::getSlowData(unsigned char* data)
{
	assert(data != nullptr);

	if (m_textPtr < 24U) {
		data[0U] = m_text[m_textPtr++] ^ DSTAR_SCRAMBLER_BYTES[0U];
		data[1U] = m_text[m_textPtr++] ^ DSTAR_SCRAMBLER_BYTES[1U];
		data[2U] = m_text[m_textPtr++] ^ DSTAR_SCRAMBLER_BYTES[2U];
	} else {
		data[0U] = 'f' ^ DSTAR_SCRAMBLER_BYTES[0U];
		data[1U] = 'f' ^ DSTAR_SCRAMBLER_BYTES[1U];
		data[2U] = 'f' ^ DSTAR_SCRAMBLER_BYTES[2U];
	}
}

unsigned char CDStarSlowData::getType() const
{
	return m_buffer[0U] & DSTAR_SLOW_DATA_TYPE_MASK;
}

bool CDStarSlowData::isComplete() const
{
	return m_complete;
}