#include "TmPacketPusA.h"

#include <cstring>

#include "../definitions.h"
#include "TmPacketBase.h"
#include "fsfw/globalfunctions/CRC.h"
#include "fsfw/globalfunctions/arrayprinter.h"
#include "fsfw/objectmanager/ObjectManagerIF.h"
#include "fsfw/serviceinterface/ServiceInterface.h"
#include "fsfw/timemanager/CCSDSTime.h"

TmPacketPusA::TmPacketPusA(uint8_t* setData) : TmPacketBase(setData) {
  tmData = reinterpret_cast<TmPacketPointerPusA*>(setData);
}

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

uint8_t TmPacketPusA::getService() { return tmData->data_field.service_type; }

uint8_t TmPacketPusA::getSubService() { return tmData->data_field.service_subtype; }

uint8_t* TmPacketPusA::getSourceData() { return &tmData->data; }

uint16_t TmPacketPusA::getSourceDataSize() {
  return SpacePacketReader::getPacketDataLen() - sizeof(tmData->data_field) - CRC_SIZE + 1;
}

ReturnValue_t TmPacketPusA::setData(uint8_t* p_Data, size_t maxSize, void* args) {
  ReturnValue_t result = SpacePacketReader::setData(p_Data, maxSize, args);
  if (result != HasReturnvaluesIF::RETURN_OK) {
    return result;
  }
  tmData = reinterpret_cast<TmPacketPointerPusA*>(const_cast<uint8_t*>(p_Data));
  return HasReturnvaluesIF::RETURN_OK;
}

size_t TmPacketPusA::getPacketMinimumSize() const { return TM_PACKET_MIN_SIZE; }

uint16_t TmPacketPusA::getDataFieldSize() { return sizeof(PUSTmDataFieldHeaderPusA); }

uint8_t* TmPacketPusA::getPacketTimeRaw() const { return tmData->data_field.time; }

void TmPacketPusA::initializeTmPacket(uint16_t apid, uint8_t service, uint8_t subservice,
                                      uint8_t packetSubcounter) {
  // Set primary header:
  initSpacePacketHeader(false, true, apid);
  // Set data Field Header:
  // First, set to zero.
  memset(&tmData->data_field, 0, sizeof(tmData->data_field));

  tmData->data_field.version_type_ack = pus::PusVersion::PUS_A_VERSION << 4;
  tmData->data_field.service_type = service;
  tmData->data_field.service_subtype = subservice;
  tmData->data_field.subcounter = packetSubcounter;
  // Timestamp packet
  if (TmPacketBase::checkAndSetStamper()) {
    timeStamper->addTimeStamp(tmData->data_field.time, sizeof(tmData->data_field.time));
  }
}

void TmPacketPusA::setSourceDataSize(uint16_t size) {
  setPacketDataLength(size + sizeof(PUSTmDataFieldHeaderPusA) + CRC_SIZE - 1);
}

size_t TmPacketPusA::getTimestampSize() const { return sizeof(tmData->data_field.time); }