bot3/input/gyro_sensor_receiver.cc

#include <inttypes.h>

#include "aos/atom_code/init.h"
#include "aos/common/logging/logging.h"
#include "aos/common/util/wrapping_counter.h"

#include "bot3/control_loops/drivetrain/drivetrain.q.h"
#include "bot3/control_loops/shooter/shooter_motor.q.h"
#include "frc971/queues/GyroAngle.q.h"
#include "frc971/input/usb_receiver.h"

#ifndef M_PI
#define M_PI 3.14159265358979323846
#endif

using ::bot3::control_loops::drivetrain;
using ::bot3::control_loops::shooter;
using ::frc971::sensors::gyro;
using ::aos::util::WrappingCounter;
using ::frc971::USBReceiver;

namespace bot3 {
namespace {

//TODO (danielp): Figure out whether the bigger gear is on the 
// encoder or not.
inline double drivetrain_translate(int32_t in) {
  return static_cast<double>(in) / (256.0 /*cpr*/ * 4.0 /*quad*/) *
      (32.0 / 44.0 /*encoder gears*/) * // the encoders are on the wheels.
      (3.5 /*wheel diameter*/ * 2.54 / 100 * M_PI);
}

// Translates values from the ADC into voltage.
inline double adc_translate(uint16_t in) {
  static const double kVRefN = 0;
  static const double kVRefP = 3.3;
  static const int kMaximumValue = 0x3FF;
  return kVRefN +
      (static_cast<double>(in) / static_cast<double>(kMaximumValue) *
       (kVRefP - kVRefN));
}

inline double gyro_translate(int64_t in) {
  return in / 16.0 / 1000.0 / (180.0 / M_PI);
}

inline double battery_translate(uint16_t in) {
  return adc_translate(in) * 80.8 / 17.8;
}

}  // namespace

class GyroSensorReceiver : public USBReceiver {
 public:
  GyroSensorReceiver() : USBReceiver(1) {}

  virtual void ProcessData(const ::aos::time::Time &/*timestamp*/) override {
    gyro.MakeWithBuilder()
        .angle(gyro_translate(data()->gyro_angle))
        .Send();

    LOG(DEBUG, "right drive: %f, left drive: %f\n", 
        drivetrain_translate(data()->main.shooter_angle),
        drivetrain_translate(data()->main.indexer));
    drivetrain.position.MakeWithBuilder()
        .right_encoder(drivetrain_translate(data()->main.wrist))
        .left_encoder(drivetrain_translate(data()->main.shooter))
        .Send();
    
    LOG(DEBUG, "battery %f %f %" PRIu16 "\n",
        battery_translate(data()->main.battery_voltage),
        adc_translate(data()->main.battery_voltage),
        data()->main.battery_voltage);
    LOG(DEBUG, "halls l=%f r=%f\n",
        adc_translate(data()->main.left_drive_hall),
        adc_translate(data()->main.right_drive_hall));
  }
};

}  // namespace bot3

int main() {
  ::aos::Init(frc971::USBReceiver::kRelativePriority);
  ::bot3::GyroSensorReceiver receiver;
  while (true) {
    receiver.RunIteration();
  }
  ::aos::Cleanup();
}