aos/common/control_loop/ControlLoop-tmpl.h

#include <stddef.h>

#include "aos/common/logging/logging.h"
#include "aos/common/control_loop/Timing.h"
#include "aos/common/messages/RobotState.q.h"
#include "aos/common/logging/queue_logging.h"

#include "bbb/sensor_generation.q.h"

namespace aos {
namespace control_loops {

// TODO(aschuh): Tests.

template <class T, bool has_position, bool fail_no_position>
constexpr ::aos::time::Time
    ControlLoop<T, has_position, fail_no_position>::kStaleLogInterval;

template <class T, bool has_position, bool fail_no_position>
void ControlLoop<T, has_position, fail_no_position>::ZeroOutputs() {
  aos::ScopedMessagePtr<OutputType> output =
      control_loop_->output.MakeMessage();
  Zero(output.get());
  output.Send();
}

template <class T, bool has_position, bool fail_no_position>
void ControlLoop<T, has_position, fail_no_position>::Iterate() {
  LOG_INTERVAL(no_prior_goal_);
  LOG_INTERVAL(no_sensor_generation_);
  LOG_INTERVAL(very_stale_position_);
  LOG_INTERVAL(no_prior_position_);
  LOG_INTERVAL(driver_station_old_);
  LOG_INTERVAL(no_driver_station_);

  // Fetch the latest control loop goal and position.  If there is no new
  // goal, we will just reuse the old one.
  // If there is no goal, we haven't started up fully.  It isn't worth
  // the added complexity for each loop implementation to handle that case.
  control_loop_->goal.FetchLatest();
  // TODO(aschuh): Check the age here if we want the loop to stop on old
  // goals.
  const GoalType *goal = control_loop_->goal.get();
  if (goal == NULL) {
    no_prior_goal_.WantToLog();
    ZeroOutputs();
    return;
  }

  ::bbb::sensor_generation.FetchLatest();
  if (::bbb::sensor_generation.get() == nullptr) {
    no_sensor_generation_.WantToLog();
    ZeroOutputs();
    return;
  }
  if (!has_sensor_reset_counters_ ||
      ::bbb::sensor_generation->reader_pid != reader_pid_ ||
      ::bbb::sensor_generation->cape_resets != cape_resets_) {
    char buffer[128];
    size_t characters = ::bbb::sensor_generation->Print(buffer, sizeof(buffer));
    LOG(INFO, "new sensor_generation message %.*s\n",
        static_cast<int>(characters), buffer);

    reader_pid_ = ::bbb::sensor_generation->reader_pid;
    cape_resets_ = ::bbb::sensor_generation->cape_resets;
    has_sensor_reset_counters_ = true;
    reset_ = true;
  }
    
  LOG_STRUCT(DEBUG, "goal", *goal);

  // Only pass in a position if we got one this cycle.
  const PositionType *position = NULL;

  // Only fetch the latest position if we have one.
  if (has_position) {
    // If the position is stale, this is really bad.  Try fetching a position
    // and check how fresh it is, and then take the appropriate action.
    if (control_loop_->position.FetchLatest()) {
      position = control_loop_->position.get();
    } else {
      if (control_loop_->position.get() && !reset_) {
        int msec_age = control_loop_->position.Age().ToMSec();
        if (!control_loop_->position.IsNewerThanMS(kPositionTimeoutMs)) {
          very_stale_position_.WantToLog();
          ZeroOutputs();
          return;
        } else {
          LOG(ERROR, "Stale position. %d ms (< %d ms)\n", msec_age,
              kPositionTimeoutMs);
        }
      } else {
        no_prior_position_.WantToLog();
        if (fail_no_position) {
          ZeroOutputs();
          return;
        }
      }
    }
    if (position) {
      LOG_STRUCT(DEBUG, "position", *position);
    }
  }

  bool outputs_enabled = false;

  // Check to see if we got a driver station packet recently.
  if (::aos::robot_state.FetchLatest()) {
    outputs_enabled = true;
  } else if (::aos::robot_state.IsNewerThanMS(kDSPacketTimeoutMs)) {
    outputs_enabled = true;
  } else {
    if (::aos::robot_state.get()) {
      driver_station_old_.WantToLog();
    } else {
      no_driver_station_.WantToLog();
    }
  }

  // Run the iteration.
  aos::ScopedMessagePtr<StatusType> status =
      control_loop_->status.MakeMessage();
  if (status.get() == NULL) {
    return;
  }

  if (outputs_enabled) {
    aos::ScopedMessagePtr<OutputType> output =
        control_loop_->output.MakeMessage();
    RunIteration(goal, position, output.get(), status.get());

    LOG_STRUCT(DEBUG, "output", *output);
    output.Send();
  } else {
    // The outputs are disabled, so pass NULL in for the output.
    RunIteration(goal, position, nullptr, status.get());
    ZeroOutputs();
  }

  LOG_STRUCT(DEBUG, "status", *status);
  status.Send();
}

template <class T, bool has_position, bool fail_no_position>
void ControlLoop<T, has_position, fail_no_position>::Run() {
  while (true) {
    time::SleepUntil(NextLoopTime());
    Iterate();
  }
}

}  // namespace control_loops
}  // namespace aos