frc971/joystick_reader.cc

#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <math.h>

#include "aos/linux_code/init.h"
#include "aos/prime/input/joystick_input.h"
#include "aos/common/input/driver_station_data.h"
#include "aos/common/logging/logging.h"
#include "aos/common/util/log_interval.h"
#include "aos/common/time.h"
#include "aos/common/actions/actions.h"

#include "frc971/control_loops/claw/claw.q.h"
#include "frc971/control_loops/drivetrain/drivetrain.q.h"
#include "frc971/control_loops/fridge/fridge.q.h"
#include "frc971/constants.h"
#include "frc971/queues/gyro.q.h"
#include "frc971/autonomous/auto.q.h"
#include "frc971/actors/fridge_profile_actor.h"
#include "frc971/actors/pickup_actor.h"
#include "frc971/actors/stack_actor.h"
#include "frc971/actors/lift_actor.h"

using ::frc971::control_loops::claw_queue;
using ::frc971::control_loops::drivetrain_queue;
using ::frc971::control_loops::fridge_queue;
using ::frc971::sensors::gyro_reading;

using ::aos::input::driver_station::ButtonLocation;
using ::aos::input::driver_station::JoystickAxis;
using ::aos::input::driver_station::ControlBit;

namespace frc971 {
namespace input {
namespace joysticks {

// preset motion limits
static const double kArmDebugVelocity = 0.40;
static const double kArmDebugAcceleration = 1.0;
static const double kElevatorDebugVelocity = 0.5;
static const double kElevatorDebugAcceleration = 2.2;

const JoystickAxis kSteeringWheel(1, 1), kDriveThrottle(2, 2);
const ButtonLocation kShiftHigh(2, 1), kShiftLow(2, 3);
const ButtonLocation kQuickTurn(1, 5);

// TODO(danielp): Real buttons for all of these.
const ButtonLocation kElevatorUp(3, 10);
const ButtonLocation kElevatorDown(3, 3);
const ButtonLocation kArmUp(3, 8);
const ButtonLocation kArmDown(2, 6);
const ButtonLocation kClawUp(3, 7);
const ButtonLocation kClawDown(3, 6);
const ButtonLocation kClawOpen(3, 11);
const ButtonLocation kClawClosed(3, 5);
const ButtonLocation kFridgeOpen(3, 1);
const ButtonLocation kFridgeClosed(2, 11);
const ButtonLocation kRollersIn(3, 4);
const ButtonLocation kClawMiddle(3, 2);
const ButtonLocation kPickup(2, 10);
const ButtonLocation kZero(2, 7);

const ButtonLocation kStack(3, 9);

// TODO(ben): Real buttons for all of these.
const ButtonLocation kArmPresetOne(99, 99);
const ButtonLocation kArmPresetTwo(99, 99);
const ButtonLocation kElevatorPresetOne(99, 99);
const ButtonLocation kElevatorPresetTwo(99, 99);

// Testing mode.
const double kElevatorVelocity = 0.5;
const double kArmVelocity = 0.5;
const double kClawVelocity = 2.0;
// TODO(danielp): Verify.
const double kJoystickDt = 0.01;

class Reader : public ::aos::input::JoystickInput {
 public:
  Reader() : was_running_(false) {}

  virtual void RunIteration(const ::aos::input::driver_station::Data &data) {
    bool last_auto_running = auto_running_;
    auto_running_ = data.GetControlBit(ControlBit::kAutonomous) &&
                    data.GetControlBit(ControlBit::kEnabled);
    if (auto_running_ != last_auto_running) {
      if (auto_running_) {
        StartAuto();
      } else {
        StopAuto();
      }
    }

    if (!data.GetControlBit(ControlBit::kAutonomous)) {
      HandleDrivetrain(data);
      if (data.GetControlBit(ControlBit::kTestMode)) {
        HandleTest(data);
      } else {
        HandleTeleop(data);
      }
    }
  }

  void HandleDrivetrain(const ::aos::input::driver_station::Data &data) {
    const double wheel = -data.GetAxis(kSteeringWheel);
    const double throttle = -data.GetAxis(kDriveThrottle);

    if (!drivetrain_queue.goal.MakeWithBuilder()
             .steering(wheel)
             .throttle(throttle)
             .quickturn(data.IsPressed(kQuickTurn))
             .control_loop_driving(false)
             .Send()) {
      LOG(WARNING, "sending stick values failed\n");
    }
  }

  void HandleTeleop(const ::aos::input::driver_station::Data &data) {
    if (!data.GetControlBit(ControlBit::kEnabled)) {
      action_queue_.CancelAllActions();
      LOG(DEBUG, "Canceling\n");
    }

    if (data.PosEdge(kElevatorUp)) {
      actors::LiftParams params;
      params.lift_height = 0.45;
      params.lift_arm = 0.2;
      action_queue_.EnqueueAction(actors::MakeLiftAction(params));

      claw_goal_ = 0.0;
      if (!claw_queue.goal.MakeWithBuilder()
               .angle(claw_goal_)
               .rollers_closed(claw_rollers_closed_)
               .intake(0.0)
               .Send()) {
        LOG(ERROR, "Sending claw goal failed.\n");
      }
    }
    if (data.PosEdge(kStack)) {
      actors::StackParams params;
      params.claw_out_angle = 0.6;
      action_queue_.EnqueueAction(actors::MakeStackAction(params));
    }
    if (data.PosEdge(kPickup)) {
      actors::PickupParams params;
      params.pickup_angle = 0.7;
      params.suck_angle = 0.5;
      params.suck_angle_finish = 0.4;
      params.pickup_finish_angle = 0.87;
      params.intake_time = 0.5;
      params.intake_voltage = 12.0;
      action_queue_.EnqueueAction(actors::MakePickupAction(params));
    }
    if (data.PosEdge(kElevatorDown)) {
      claw_goal_ = 0.0;

      actors::FridgeProfileParams fridge_params;
      fridge_params.arm_max_velocity = kArmDebugVelocity;
      fridge_params.arm_max_acceleration = kArmDebugAcceleration;
      fridge_params.elevator_max_velocity = kElevatorDebugVelocity;
      fridge_params.elevator_max_acceleration = kElevatorDebugAcceleration;

      fridge_params.arm_angle = 0.0;
      fridge_params.elevator_height = 0.035;

      fridge_params.top_front_grabber = fridge_closed_;
      fridge_params.top_back_grabber = fridge_closed_;
      fridge_params.bottom_front_grabber = fridge_closed_;
      fridge_params.bottom_back_grabber = fridge_closed_;
      action_queue_.EnqueueAction(MakeFridgeProfileAction(fridge_params));
    }

    if (data.PosEdge(kClawMiddle)) {
      claw_goal_ = 0.8;
    }

    if (data.PosEdge(kZero)) {
      elevator_goal_ = 0.0;
      arm_goal_ = 0.0;
      claw_goal_ = 0.0;
    }

    if (data.PosEdge(kClawClosed)) {
      claw_rollers_closed_ = true;
    }
    if (data.PosEdge(kClawOpen)) {
      claw_rollers_closed_ = false;
    }

    if (data.PosEdge(kFridgeClosed)) {
      fridge_closed_ = true;
    }
    if (data.PosEdge(kFridgeOpen)) {
      fridge_closed_ = false;
    }

    if (data.PosEdge(ControlBit::kEnabled)) {
      // If we got enabled, wait for everything to zero.
      LOG(INFO, "Waiting for zero.\n");
      waiting_for_zero_ = true;
    }

    if (waiting_for_zero_) {
      claw_queue.status.FetchLatest();
      fridge_queue.status.FetchLatest();
      if (!claw_queue.status.get()) {
        LOG(ERROR, "Got no claw status packet.\n");
        // Not safe to continue.
        return;
      }
      if (!fridge_queue.status.get()) {
        LOG(ERROR, "Got no fridge status packet.\n");
        return;
      }

      if (claw_queue.status->zeroed && fridge_queue.status->zeroed) {
        LOG(INFO, "Zeroed! Starting teleop mode.\n");
        waiting_for_zero_ = false;

        // Set the initial goals to where we are now.
        elevator_goal_ = fridge_queue.status->goal_height;
        arm_goal_ = fridge_queue.status->goal_angle;
        claw_goal_ = claw_queue.status->angle;
      } else {
        return;
      }
    } else {
      if (!action_queue_.Running()) {
        auto new_fridge_goal = fridge_queue.goal.MakeMessage();
        new_fridge_goal->height = elevator_goal_;
        new_fridge_goal->angle = arm_goal_;
        new_fridge_goal->angular_velocity = 0.0;
        new_fridge_goal->velocity = 0.0;
        new_fridge_goal->grabbers.top_front = fridge_closed_;
        new_fridge_goal->grabbers.top_back = fridge_closed_;
        new_fridge_goal->grabbers.bottom_front = fridge_closed_;
        new_fridge_goal->grabbers.bottom_back = fridge_closed_;

        if (!new_fridge_goal.Send()) {
          LOG(ERROR, "Sending fridge goal failed.\n");
        } else {
          LOG(DEBUG, "sending goals: elevator: %f, arm: %f\n", elevator_goal_,
              arm_goal_);
        }
        if (!claw_queue.goal.MakeWithBuilder()
                 .angle(claw_goal_)
                 .rollers_closed(claw_rollers_closed_)
                 .intake(data.IsPressed(kRollersIn) ? 12.0 : 0.0)
                 .Send()) {
          LOG(ERROR, "Sending claw goal failed.\n");
        }
      }
    }

    if (action_queue_.Running()) {
      // If we are running an action, update our goals to the current goals.
      control_loops::fridge_queue.status.FetchLatest();
      if (control_loops::fridge_queue.status.get()) {
        arm_goal_ = control_loops::fridge_queue.status->goal_angle;
        elevator_goal_ = control_loops::fridge_queue.status->goal_height;
      } else {
        LOG(ERROR, "No fridge status!\n");
      }

      // If we are running an action, update our goals to the current goals.
      control_loops::claw_queue.status.FetchLatest();
      if (control_loops::claw_queue.status.get()) {
        claw_goal_ = control_loops::claw_queue.status->goal_angle;
      } else {
        LOG(ERROR, "No fridge status!\n");
      }
    }
    action_queue_.Tick();
    was_running_ = action_queue_.Running();
  }

  void HandleTest(const ::aos::input::driver_station::Data &data) {
    if (action_queue_.Running()) {
      // We don't really want any actions running.
      LOG(DEBUG, "Cancelling actions for test mode.\n");
      action_queue_.CancelAllActions();
    }

    if (data.GetControlBit(ControlBit::kEnabled)) {
      if (data.PosEdge(ControlBit::kEnabled)) {
        // If we got enabled, wait for everything to zero.
        LOG(INFO, "Waiting for zero.\n");
        waiting_for_zero_ = true;
      }
      if (waiting_for_zero_) {
        claw_queue.status.FetchLatest();
        fridge_queue.status.FetchLatest();
        if (!claw_queue.status.get()) {
          LOG(ERROR, "Got no claw status packet.\n");
          // Not safe to continue.
          return;
        }
        if (!fridge_queue.status.get()) {
          LOG(ERROR, "Got no fridge status packet.\n");
          return;
        }

        if (claw_queue.status->zeroed && fridge_queue.status->zeroed) {
          LOG(INFO, "Zeroed! Starting test mode.\n");
          waiting_for_zero_ = false;

          // Set the initial goals to where we are now.
          elevator_goal_ = fridge_queue.status->height;
          arm_goal_ = fridge_queue.status->angle;
          claw_goal_ = claw_queue.status->angle;
        } else {
          return;
        }
      }

      // These buttons move a subsystem up or down for as long as they are
      // pressed, at low velocity.
      if (data.IsPressed(kElevatorUp)) {
        elevator_goal_ += kElevatorVelocity * kJoystickDt;
      }
      if (data.IsPressed(kElevatorDown)) {
        elevator_goal_ -= kElevatorVelocity * kJoystickDt;
      }
      if (data.IsPressed(kArmUp)) {
        arm_goal_ += kArmVelocity * kJoystickDt;
      }
      if (data.IsPressed(kArmDown)) {
        arm_goal_ -= kArmVelocity * kJoystickDt;
      }
      if (data.IsPressed(kClawUp)) {
        claw_goal_ += kClawVelocity * kJoystickDt;
      }
      if (data.IsPressed(kClawDown)) {
        claw_goal_ -= kClawVelocity * kJoystickDt;
      }

      if (!action_queue_.Running()) {
        if (!fridge_queue.goal.MakeWithBuilder()
                 .height(elevator_goal_)
                 .angle(arm_goal_)
                 .Send()) {
          LOG(ERROR, "Sending fridge goal failed.\n");
        } else {
          LOG(DEBUG, "sending goals: elevator: %f, arm: %f\n", elevator_goal_,
              arm_goal_);
        }
        if (!claw_queue.goal.MakeWithBuilder().angle(claw_goal_).Send()) {
          LOG(ERROR, "Sending claw goal failed.\n");
        }
      }
      /*
      if (data.IsPressed(kArmPresetOne) || data.IsPressed(kArmPresetTwo)) {
        actors::FridgeProfileParams fridge_params;
        fridge_params.arm_max_velocity = kArmDebugVelocity;
        fridge_params.arm_max_acceleration = kArmDebugAcceleration;
        if (data.IsPressed(kArmPresetOne)) {
          LOG(INFO, "Preset asked for test arm position one position.\n");
          fridge_params.arm_angle = M_PI / 4.0;
          fridge_params.top_front_grabber = false;
          fridge_params.top_back_grabber = false;
          fridge_params.bottom_front_grabber = false;
          fridge_params.bottom_back_grabber = false;
          action_queue_.EnqueueAction(MakeFridgeProfileAction(fridge_params));
        } else if (data.IsPressed(kArmPresetTwo)) {
          LOG(INFO, "Preset asked for test arm position two position.\n");
          fridge_params.arm_angle = -M_PI / 4.0;
          fridge_params.top_front_grabber = true;
          fridge_params.top_back_grabber = true;
          fridge_params.bottom_front_grabber = true;
          fridge_params.bottom_back_grabber = true;
          action_queue_.EnqueueAction(MakeFridgeProfileAction(fridge_params));
        }
      } else if (data.IsPressed(kElevatorPresetOne) ||
                 data.IsPressed(kElevatorPresetOne)) {
        actors::FridgeProfileParams fridge_params;
        fridge_params.elevator_max_velocity = kElevatorDebugVelocity;
        fridge_params.elevator_max_acceleration = kElevatorDebugAcceleration;
        if (data.IsPressed(kElevatorPresetOne)) {
          LOG(INFO, "Preset asked for test elevator position one position.\n");
          fridge_params.elevator_height = 0.5;
          fridge_params.top_front_grabber = false;
          fridge_params.top_back_grabber = false;
          fridge_params.bottom_front_grabber = false;
          fridge_params.bottom_back_grabber = false;
          action_queue_.EnqueueAction(MakeFridgeProfileAction(fridge_params));
        } else if (data.IsPressed(kElevatorPresetTwo)) {
          LOG(INFO, "Preset asked for test elevator position two position.\n");
          fridge_params.elevator_height = 1.2;
          fridge_params.top_front_grabber = true;
          fridge_params.top_back_grabber = true;
          fridge_params.bottom_front_grabber = true;
          fridge_params.bottom_back_grabber = true;
          action_queue_.EnqueueAction(MakeFridgeProfileAction(fridge_params));
        }
      }
      */
    }
  }

 private:
  void StartAuto() {
    LOG(INFO, "Starting auto mode\n");
    ::frc971::autonomous::autonomous.MakeWithBuilder().run_auto(true).Send();
  }

  void StopAuto() {
    LOG(INFO, "Stopping auto mode\n");
    ::frc971::autonomous::autonomous.MakeWithBuilder().run_auto(false).Send();
  }

  bool was_running_;

  // Previous goals for systems.
  double elevator_goal_ = 0.2;
  double arm_goal_ = 0.0;
  double claw_goal_ = 0.0;
  bool claw_rollers_closed_ = false;
  bool fridge_closed_ = false;

  // If we're waiting for the subsystems to zero.
  bool waiting_for_zero_ = true;

  bool auto_running_ = false;

  ::aos::common::actions::ActionQueue action_queue_;

  ::aos::util::SimpleLogInterval no_drivetrain_status_ =
      ::aos::util::SimpleLogInterval(::aos::time::Time::InSeconds(0.2), WARNING,
                                     "no drivetrain status");
};

}  // namespace joysticks
}  // namespace input
}  // namespace frc971

int main() {
  ::aos::Init();
  ::frc971::input::joysticks::Reader reader;
  reader.Run();
  ::aos::Cleanup();
}