frc971/wpilib/encoder_and_potentiometer.h

#ifndef FRC971_ENCODER_AND_POTENTIOMETER_H_
#define FRC971_ENCODER_AND_POTENTIOMETER_H_

#include <atomic>
#include <thread>

#include "aos/common/macros.h"
#include "aos/common/mutex.h"

#include "Encoder.h"
#include "DigitalSource.h"
#include "AnalogInput.h"
#include "dma.h"

#include "frc971/wpilib/dma_edge_counting.h"

namespace frc971 {
namespace wpilib {

// Latches values from an encoder and potentiometer on positive edges from
// another input using an interrupt.
class InterruptEncoderAndPotentiometer {
 public:
  // priority is the priority the thread will run at.
  InterruptEncoderAndPotentiometer(int priority) : priority_(priority) {}

  // Starts the thread running so it can receive interrupts.
  void Start();

  // Tells the thread to stop running and then waits for it to finish.
  void Stop() {
    run_ = false;
    thread_.join();
  }

  // Loops until Stop() is called, reading interrupts.
  // Designed to be called by ::std::thread internally.
  void operator()();

  // Returns the mutex which must be held while calling index_posedge_count(),
  // last_encoder_value(), and last_potentiometer_voltage().
  // Holding this mutex will increase the handling latency.
  ::aos::Mutex *mutex() { return &mutex_; }

  void set_encoder(::std::unique_ptr<Encoder> encoder) {
    encoder_ = ::std::move(encoder);
  }
  Encoder *encoder() const { return encoder_.get(); }

  void set_index(::std::unique_ptr<DigitalSource> index) {
    index_ = ::std::move(index);
  }
  DigitalSource *index() const { return index_.get(); }

  void set_potentiometer(::std::unique_ptr<AnalogInput> potentiometer) {
    potentiometer_ = ::std::move(potentiometer);
  }
  AnalogInput *potentiometer() const { return potentiometer_.get(); }

  // Returns the number of poseges that have happened on the index input.
  // mutex() must be held while calling this.
  uint32_t index_posedge_count() const { return index_posedge_count_; }
  // Returns the value of the encoder at the last index posedge.
  // mutex() must be held while calling this.
  int32_t last_encoder_value() const { return last_encoder_value_; }
  // Returns the voltage of the potentiometer at the last index posedge.
  // mutex() must be held while calling this.
  float last_potentiometer_voltage() const {
    return last_potentiometer_voltage_;
  }

 private:
  ::std::unique_ptr<Encoder> encoder_;
  ::std::unique_ptr<DigitalSource> index_;
  ::std::unique_ptr<AnalogInput> potentiometer_;

  int32_t last_encoder_value_{0};
  float last_potentiometer_voltage_{0.0f};
  uint32_t index_posedge_count_{0};

  ::aos::Mutex mutex_;

  const int priority_;

  ::std::atomic<bool> run_{true};
  ::std::thread thread_;

  DISALLOW_COPY_AND_ASSIGN(InterruptEncoderAndPotentiometer);
};

// Latches values from an encoder and potentiometer on positive edges from
// another input using DMA.
class DMAEncoderAndPotentiometer : public DMASampleHandlerInterface {
 public:
  DMAEncoderAndPotentiometer() {}

  void set_encoder(::std::unique_ptr<Encoder> encoder) {
    encoder_ = ::std::move(encoder);
  }
  Encoder *encoder() const { return encoder_.get(); }

  void set_index(::std::unique_ptr<DigitalSource> index) {
    index_ = ::std::move(index);
  }
  DigitalSource *index() const { return index_.get(); }

  void set_potentiometer(::std::unique_ptr<AnalogInput> potentiometer) {
    potentiometer_ = ::std::move(potentiometer);
  }
  AnalogInput *potentiometer() const { return potentiometer_.get(); }

  // Returns the most recent polled value of the encoder.
  uint32_t polled_encoder_value() const { return polled_encoder_value_; }
  // Returns the most recent polled voltage of the potentiometer.
  float polled_potentiometer_voltage() const {
    return polled_potentiometer_voltage_;
  }

  // Returns the number of poseges that have happened on the index input.
  uint32_t index_posedge_count() const { return index_posedge_count_; }
  // Returns the value of the encoder at the last index posedge.
  int32_t last_encoder_value() const { return last_encoder_value_; }
  // Returns the voltage of the potentiometer at the last index posedge.
  float last_potentiometer_voltage() const {
    return last_potentiometer_voltage_;
  }

  virtual void UpdateFromSample(const DMASample &sample) override;

  virtual void PollFromSample(const DMASample &sample) override {
    polled_encoder_value_ = sample.GetRaw(encoder_.get());
    polled_potentiometer_voltage_ = sample.GetVoltage(potentiometer_.get());
  }

  virtual void UpdatePolledValue() override {
    polled_encoder_value_ = encoder_->GetRaw();
    polled_potentiometer_voltage_ = potentiometer_->GetVoltage();
  }

  virtual void AddToDMA(DMA *dma) override {
    dma->Add(encoder_.get());
    dma->Add(index_.get());
    dma->Add(potentiometer_.get());
    dma->SetExternalTrigger(index_.get(), true, true);
  }

 private:
  ::std::unique_ptr<Encoder> encoder_;
  ::std::unique_ptr<DigitalSource> index_;
  ::std::unique_ptr<AnalogInput> potentiometer_;

  int32_t polled_encoder_value_ = 0;
  float polled_potentiometer_voltage_ = 0.0f;

  int32_t last_encoder_value_ = 0;
  float last_potentiometer_voltage_ = 0.0f;

  uint32_t index_posedge_count_ = 0;

  // Whether or not it was triggered in the last sample.
  bool index_last_value_ = false;

  DISALLOW_COPY_AND_ASSIGN(DMAEncoderAndPotentiometer);
};

}  // namespace wpilib
}  // namespace frc971

#endif  // FRC971_ENCODER_AND_POTENTIOMETER_H_