device_procedure_machine.h

Go to the documentation of this file.

#pragma once
 
#include <chrono>
#include <future>
#include <optional>
 
#include "core/containers/thread_safe_queue.h"
#include "device_procedure.h"
#include "machine.h"
#include "motion_master_error.h"
 
class Cia402Drive;
 
class DeviceProcedureMachine : public Machine {
 public:
  DeviceProcedureMachine(
      VirtualDeviceManager& vdm, ctpl::thread_pool& thread_pool,
      Notifier& notifier,
      mm::core::containers::ThreadSafeQueue<OutgoingMachineMessage>&
          outgoing_machine_message_queue);
 
  void execute(uint64_t pending_signals) override;
  void abort_client_requests(const std::string& connection_id) override;
  bool done() override;
 
 private:
  class Axis {
   public:
    Axis(VirtualDeviceManager& vdm,
         IncomingMachineMessage& incoming_machine_message);
 
    enum class Stage { kInit, kStart, kWait, kDeinit, kDone };
 
    // TODO(msankovic): Find another way of dealing with other device types
    std::unique_ptr<Cia402Drive> cia402_drive_;
 
    IncomingMachineMessage incoming_machine_message_;
 
    std::unique_ptr<DeviceProcedure> device_procedure_;
    std::optional<std::future<MotionMasterError>> future_;
    Stage stage_ = Stage::kInit;
  };
 
  std::map<uint32_t, Axis> axis_map_;
 
  void send_status(
      const IncomingMachineMessage& incoming_machine_message,
      motionmaster::MotionMasterMessage::MotionMasterMessage::Status* status);
 
  void send_initial_confirmation_status(
      uint32_t device_address,
      IncomingMachineMessage& incoming_machine_message);
 
  void send_error_status(uint32_t device_address,
                         IncomingMachineMessage& incoming_machine_message,
                         MotionMasterError error);
 
  void send_success_status(uint32_t device_address,
                           IncomingMachineMessage& incoming_machine_message);
};