eive-obsw/mission/devices/PayloadPcduHandler.cpp

#include "PayloadPcduHandler.h"

#include <fsfw/src/fsfw/datapool/PoolReadGuard.h>

#include "devices/gpioIds.h"

PayloadPcduHandler::PayloadPcduHandler(object_id_t objectId, object_id_t comIF, CookieIF* cookie,
                                       GpioIF* gpioIF)
    : DeviceHandlerBase(objectId, comIF, cookie), adcSet(this), gpioIF(gpioIF) {}

void PayloadPcduHandler::doStartUp() {
  if ((state != States::PCDU_OFF) and (state != States::ON_TRANS_SSR)) {
    // Config error
    sif::error << "PayloadPcduHandler::doStartUp: Invalid state" << std::endl;
  }
  if (state == States::PCDU_OFF) {
    // Switch on relays here
    gpioIF->pullHigh(gpioIds::PLPCDU_ENB_VBAT0);
    gpioIF->pullHigh(gpioIds::PLPCDU_ENB_VBAT1);
    state = States::ON_TRANS_SSR;
    transitionOk = true;
  }
  if (state == States::ON_TRANS_SSR) {
    // If necessary, check whether a certain amount of time has elapsed
    if (transitionOk) {
      transitionOk = false;
      // We are now in ON mode
      setMode(MODE_ON);
      adcState = AdcStates::BOOT_DELAY;
      // The ADC can now be read. If the values are not close to zero, we should not allow
      // transition
      monMode = MonitoringMode::CLOSE_TO_ZERO;
    }
  }
}

void PayloadPcduHandler::doTransition(Mode_t modeFrom, Submode_t subModeFrom) {
  if (mode == _MODE_TO_NORMAL) {
    if (adcState == AdcStates::BOOT_DELAY) {
      if (adcCountdown.hasTimedOut()) {
        adcState = AdcStates::SEND_SETUP;
        adcCmdExecuted = false;
      }
    }
    if (adcState == AdcStates::SEND_SETUP) {
      if (adcCmdExecuted) {
        adcState = AdcStates::NORMAL;
        adcCmdExecuted = false;
      }
    }
    if (state == States::ON_TRANS_ADC_CLOSE_ZERO) {
      if (not commandExecuted) {
        countdown.resetTimer();
        commandExecuted = true;
      }
      // ADC values are ok, 5 seconds have elapsed
      if (transitionOk and countdown.hasTimedOut()) {
        state = States::ON_TRANS_DRO;
        // Now start monitoring for negative voltages instead
        monMode = MonitoringMode::NEGATIVE;
        countdown.resetTimer();
        commandExecuted = false;
        transitionOk = false;
      }
    }
    if (state == States::ON_TRANS_DRO) {
      if (not commandExecuted) {
        // Switch on DRO and start monitoring for negative voltagea
        gpioIF->pullHigh(gpioIds::PLPCDU_ENB_DRO);
        commandExecuted = true;
      }
      // ADC values are ok, 5 seconds have elapsed
      if (transitionOk and countdown.hasTimedOut()) {
        state = States::ON_TRANS_X8;
        countdown.resetTimer();
        commandExecuted = false;
        transitionOk = false;
      }
    }
    if (state == States::ON_TRANS_X8) {
      if (not commandExecuted) {
        // Switch on X8
        gpioIF->pullHigh(gpioIds::PLPCDU_ENB_X8);
        commandExecuted = true;
      }
      // ADC values are ok, 5 seconds have elapsed
      if (transitionOk and countdown.hasTimedOut()) {
        state = States::ON_TRANS_TX;
        countdown.resetTimer();
        commandExecuted = false;
        transitionOk = false;
      }
    }
    if (state == States::ON_TRANS_TX) {
      if (not commandExecuted) {
        // Switch on TX
        gpioIF->pullHigh(gpioIds::PLPCDU_ENB_TX);
        commandExecuted = true;
      }
      // ADC values are ok, 5 seconds have elapsed
      if (transitionOk and countdown.hasTimedOut()) {
        state = States::ON_TRANS_MPA;
        countdown.resetTimer();
        commandExecuted = false;
        transitionOk = false;
      }
    }
    if (state == States::ON_TRANS_MPA) {
      if (not commandExecuted) {
        // Switch on MPA
        gpioIF->pullHigh(gpioIds::PLPCDU_ENB_MPA);
        commandExecuted = true;
      }
      // ADC values are ok, 5 seconds have elapsed
      if (transitionOk and countdown.hasTimedOut()) {
        state = States::ON_TRANS_HPA;
        countdown.resetTimer();
        commandExecuted = false;
        transitionOk = false;
      }
    }
    if (state == States::ON_TRANS_HPA) {
      if (not commandExecuted) {
        // Switch on HPA
        gpioIF->pullHigh(gpioIds::PLPCDU_ENB_HPA);
        commandExecuted = true;
      }
      // ADC values are ok, 5 seconds have elapsed
      if (transitionOk and countdown.hasTimedOut()) {
        state = States::PCDU_ON;
        setMode(MODE_NORMAL);
        countdown.resetTimer();
        commandExecuted = false;
        transitionOk = false;
      }
    }
  }
}

void PayloadPcduHandler::doShutDown() {}

ReturnValue_t PayloadPcduHandler::buildNormalDeviceCommand(DeviceCommandId_t* id) {
  switch (adcState) {
    case (AdcStates::SEND_SETUP): {
      *id = plpcdu::SETUP_CMD;
      buildCommandFromCommand(*id, nullptr, 0);
      break;
    }
    case (AdcStates::NORMAL): {
      *id = plpcdu::READ_WITH_TEMP;
      buildCommandFromCommand(*id, nullptr, 0);
      break;
    }
    default: {
      break;
    }
  }
  return NOTHING_TO_SEND;
}

ReturnValue_t PayloadPcduHandler::buildTransitionDeviceCommand(DeviceCommandId_t* id) {
  if (adcState == AdcStates::SEND_SETUP) {
    *id = plpcdu::SETUP_CMD;
    buildCommandFromCommand(*id, nullptr, 0);
  }
  return HasReturnvaluesIF::RETURN_OK;
}

void PayloadPcduHandler::fillCommandAndReplyMap() {
  insertInCommandAndReplyMap(plpcdu::READ_CMD, 2, &adcSet);
  insertInCommandAndReplyMap(plpcdu::READ_TEMP, 1, &adcSet);
  insertInCommandAndReplyMap(plpcdu::READ_WITH_TEMP, 1, &adcSet);
  insertInCommandAndReplyMap(plpcdu::SETUP_CMD, 1);
}

ReturnValue_t PayloadPcduHandler::buildCommandFromCommand(DeviceCommandId_t deviceCommand,
                                                          const uint8_t* commandData,
                                                          size_t commandDataLen) {
  switch (deviceCommand) {
    case (plpcdu::SETUP_CMD): {
      cmdBuf[0] = plpcdu::SETUP_BYTE;
      rawPacket = cmdBuf.data();
      rawPacketLen = 1;
      break;
    }
    case (plpcdu::READ_CMD): {
      max1227::prepareExternallyClockedRead0ToN(cmdBuf.data(), plpcdu::CHANNEL_N, rawPacketLen);
      rawPacket = cmdBuf.data();
      break;
    }
    case (plpcdu::READ_TEMP): {
      max1227::prepareExternallyClockedTemperatureRead(cmdBuf.data(), rawPacketLen);
      rawPacket = cmdBuf.data();
      break;
    }
    case (plpcdu::READ_WITH_TEMP): {
      size_t sz = 0;
      max1227::prepareExternallyClockedRead0ToN(cmdBuf.data(), plpcdu::CHANNEL_N, sz);
      max1227::prepareExternallyClockedTemperatureRead(cmdBuf.data() + sz, sz);
      rawPacketLen = sz;
      rawPacket = cmdBuf.data();
    }
  }
  return DeviceHandlerIF::COMMAND_NOT_IMPLEMENTED;
}

ReturnValue_t PayloadPcduHandler::scanForReply(const uint8_t* start, size_t remainingSize,
                                               DeviceCommandId_t* foundId, size_t* foundLen) {
  // SPI is full duplex
  *foundId = getPendingCommand();
  *foundLen = remainingSize;
  return HasReturnvaluesIF::RETURN_OK;
}

ReturnValue_t PayloadPcduHandler::interpretDeviceReply(DeviceCommandId_t id,
                                                       const uint8_t* packet) {
  using namespace plpcdu;
  switch (id) {
    case (SETUP_CMD): {
      break;
    }
    case (READ_TEMP): {
      uint8_t tempStartIdx = TEMP_REPLY_SIZE - 2;
      adcSet.tempC.value =
          max1227::getTemperature(packet[tempStartIdx] << 8 | packet[tempStartIdx + 2]);
      break;
    }
    case (READ_CMD): {
      PoolReadGuard pg(&adcSet);
      if (pg.getReadResult() != HasReturnvaluesIF::RETURN_OK) {
        return pg.getReadResult();
      }
      for (uint8_t idx = 0; idx < 12; idx++) {
        adcSet.channels[idx] = packet[idx * 2 + 1] << 8 | packet[idx * 2 + 2];
      }
      break;
    }
    case (READ_WITH_TEMP): {
      PoolReadGuard pg(&adcSet);
      if (pg.getReadResult() != HasReturnvaluesIF::RETURN_OK) {
        return pg.getReadResult();
      }
      for (uint8_t idx = 0; idx < 12; idx++) {
        adcSet.channels[idx] = packet[idx * 2 + 1] << 8 | packet[idx * 2 + 2];
      }
      uint8_t tempStartIdx = ADC_REPLY_SIZE + TEMP_REPLY_SIZE - 2;
      adcSet.tempC.value =
          max1227::getTemperature(packet[tempStartIdx] << 8 | packet[tempStartIdx + 2]);
      break;
    }
    default: {
      break;
    }
  }
  return HasReturnvaluesIF::RETURN_OK;
}

uint32_t PayloadPcduHandler::getTransitionDelayMs(Mode_t modeFrom, Mode_t modeTo) {
  // 20 minutes transition delay is allowed
  return 20 * 60 * 60;
}

ReturnValue_t PayloadPcduHandler::initializeLocalDataPool(localpool::DataPool& localDataPoolMap,
                                                          LocalDataPoolManager& poolManager) {
  localDataPoolMap.emplace(plpcdu::PlPcduPoolIds::CHANNEL_VEC, &channelValues);
  localDataPoolMap.emplace(plpcdu::PlPcduPoolIds::TEMP, &tempC);
  return HasReturnvaluesIF::RETURN_OK;
}

void PayloadPcduHandler::setToGoToNormalModeImmediately(bool enable) {
  this->goToNormalMode = enable;
}