frc971/actors/pickup_actor.cc

#include "frc971/actors/pickup_actor.h"

#include <cmath>

#include "aos/common/logging/logging.h"
#include "aos/common/controls/control_loop.h"
#include "aos/common/util/phased_loop.h"
#include "aos/common/time.h"
#include "frc971/control_loops/claw/claw.q.h"

namespace frc971 {
namespace actors {
namespace {
constexpr double kClawPickupVelocity = 3.00;
constexpr double kClawPickupAcceleration = 4.0;
constexpr double kClawMoveVelocity = 3.00;
constexpr double kClawMoveAcceleration = 8.0;
}  // namespace

PickupActor::PickupActor(PickupActionQueueGroup* queues)
    : aos::common::actions::ActorBase<PickupActionQueueGroup>(queues) {}

bool PickupActor::RunAction(const PickupParams& params) {
  constexpr double kAngleEpsilon = 0.10;
  {
    auto message = control_loops::claw_queue.goal.MakeMessage();
    message->angle = params.pickup_angle;
    message->max_velocity = kClawPickupVelocity;
    message->max_acceleration = kClawPickupAcceleration;
    message->angular_velocity = 0.0;
    message->intake = 0.0;
    message->rollers_closed = true;

    LOG_STRUCT(DEBUG, "Sending claw goal", *message);
    message.Send();
  }
  while (true) {
    control_loops::claw_queue.status.FetchAnother();
    if (ShouldCancel()) return true;
    const double current_angle = control_loops::claw_queue.status->angle;
    LOG_STRUCT(DEBUG, "Got claw status", *control_loops::claw_queue.status);

    if (current_angle > params.suck_angle ||
        ::std::abs(current_angle - params.pickup_angle) < kAngleEpsilon) {
      break;
    }
  }

  {
    auto message = control_loops::claw_queue.goal.MakeMessage();
    message->angle = params.pickup_angle;
    message->max_velocity = kClawPickupVelocity;
    message->max_acceleration = kClawPickupAcceleration;
    message->angular_velocity = 0.0;
    message->intake = params.intake_voltage;
    message->rollers_closed = true;

    LOG_STRUCT(DEBUG, "Sending claw goal", *message);
    message.Send();
  }

  while (true) {
    control_loops::claw_queue.status.FetchAnother();
    if (ShouldCancel()) return true;
    const double current_angle = control_loops::claw_queue.status->angle;
    LOG_STRUCT(DEBUG, "Got claw status", *control_loops::claw_queue.status);

    if (::std::abs(current_angle - params.pickup_angle) < kAngleEpsilon) {
      break;
    }
  }

  {
    auto message = control_loops::claw_queue.goal.MakeMessage();
    message->angle = params.suck_angle_finish;
    message->max_velocity = kClawMoveVelocity;
    message->max_acceleration = kClawMoveAcceleration;
    message->angular_velocity = 0.0;
    message->intake = params.intake_voltage;
    message->rollers_closed = true;

    LOG_STRUCT(DEBUG, "Sending claw goal", *message);
    message.Send();
  }

  ::aos::time::Time end_time =
      ::aos::time::Time::Now() + aos::time::Time::InSeconds(params.intake_time);
  while ( ::aos::time::Time::Now() <= end_time) {
    ::aos::time::PhasedLoopXMS(::aos::controls::kLoopFrequency.ToMSec(), 2500);
    if (ShouldCancel()) return true;
  }

  {
    auto message = control_loops::claw_queue.goal.MakeMessage();
    message->angle = params.pickup_finish_angle;
    message->max_velocity = kClawMoveVelocity;
    message->max_acceleration = kClawMoveAcceleration;
    message->angular_velocity = 0.0;
    message->intake = 0.0;
    message->rollers_closed = true;

    LOG_STRUCT(DEBUG, "Sending claw goal", *message);
    message.Send();
  }

  while (true) {
    control_loops::claw_queue.status.FetchAnother();
    if (ShouldCancel()) return true;
    const double current_angle = control_loops::claw_queue.status->angle;
    LOG_STRUCT(DEBUG, "Got claw status", *control_loops::claw_queue.status);

    if (::std::abs(current_angle - params.pickup_finish_angle) <
        kAngleEpsilon) {
      break;
    }
  }

  return true;
}

::std::unique_ptr<PickupAction> MakePickupAction(const PickupParams& params) {
  return ::std::unique_ptr<PickupAction>(
      new PickupAction(&::frc971::actors::pickup_action, params));
}

}  // namespace actors
}  // namespace frc971