ugv_sdk/sample/scout_demo/scout_robot_demo.cpp

/*
 * scout_robot_demo.cpp
 *
 * Created on: Jul 08, 2021 11:12
 * Description:
 *
 * Copyright (c) 2021 Weston Robot Pte. Ltd.
 */

#include <unistd.h>

#include <memory>
#include <iostream>

#include "ugv_sdk/mobile_robot/scout_robot.hpp"
#include "ugv_sdk/utilities/protocol_detector.hpp"

using namespace westonrobot;

int main(int argc, char **argv) {
  std::string device_name;
  std::string robot_subtype;

  if (argc == 2) {
    device_name = {argv[1]};
    std::cout << "Selected interface " << device_name << ", robot type: scout"
              << std::endl;
  } else if (argc == 3) {
    robot_subtype = {argv[1]};
    device_name = {argv[2]};
    std::cout << "Selected interface " << device_name
              << ", robot type: " << robot_subtype << std::endl;
  } else {
    std::cout << "Usage: demo_scout_robot [<robot-subtype>] <interface>"
              << std::endl
              << "Example 1: ./demo_scout_robot can0" << std::endl
              << "\t <robot-subtype>: mini" << std::endl;
    return -1;
  }

  bool is_scout_mini = false;
  if (robot_subtype == "mini") {
    is_scout_mini = true;
  } else if (!robot_subtype.empty() && robot_subtype != "scout") {
    std::cout
        << "Unkonwn robot subtype. Supported subtypes: \"scout\" or \"mini\""
        << std::endl;
  }

  std::unique_ptr<ScoutRobot> scout;

  ProtocolDetector detector;
  if (detector.Connect(device_name)) {
    auto proto = detector.DetectProtocolVersion(5);
    if (proto == ProtocolVersion::AGX_V1) {
      std::cout << "Detected protocol: AGX_V1" << std::endl;
      scout = std::unique_ptr<ScoutRobot>(
          new ScoutRobot(ProtocolVersion::AGX_V1, is_scout_mini));
    } else if (proto == ProtocolVersion::AGX_V2) {
      std::cout << "Detected protocol: AGX_V2" << std::endl;
      scout = std::unique_ptr<ScoutRobot>(
          new ScoutRobot(ProtocolVersion::AGX_V2, is_scout_mini));
    } else {
      std::cout << "Detected protocol: UNKONWN" << std::endl;
      return -1;
    }
  } else {
    return -1;
  }

  if (scout == nullptr)
    std::cout << "Failed to create robot object" << std::endl;

  scout->Connect(device_name);

  if (scout->GetParserProtocolVersion() == ProtocolVersion::AGX_V2) {
    scout->EnableCommandedMode();
  }

  // light control
  std::cout << "Light: const off" << std::endl;
  scout->SetLightCommand(CONST_OFF, 0, CONST_OFF, 0);
  sleep(3);
  std::cout << "Light: const on" << std::endl;
  scout->SetLightCommand(CONST_ON, 0, CONST_ON, 0);
  sleep(3);
  std::cout << "Light: breath" << std::endl;
  scout->SetLightCommand(BREATH, 0, BREATH, 0);
  sleep(3);
  std::cout << "Light: custom 30-40" << std::endl;
  scout->SetLightCommand(CUSTOM, 30, CUSTOM, 40);
  sleep(3);
  scout->SetLightCommand(CONST_OFF, 0, CONST_OFF, 0);

  int count = 0;
  while (true) {
    // motion control
    std::cout << "Motor: 1.0, 0" << std::endl;
    scout->SetMotionCommand(1.0, 0.0);

    // get robot state
    auto state = scout->GetRobotState();
    std::cout << "-------------------------------" << std::endl;
    std::cout << "count: " << count << std::endl;
    std::cout << "control mode: "
              << static_cast<int>(state.system_state.control_mode)
              << " , vehicle state: "
              << static_cast<int>(state.system_state.vehicle_state)
              << " , error code: " << std::hex << state.system_state.error_code
              << std::dec
              << ", battery voltage: " << state.system_state.battery_voltage
              << std::endl;
    std::cout << "velocity (linear, angular): "
              << state.motion_state.linear_velocity << ", "
              << state.motion_state.angular_velocity << std::endl;
    std::cout << "core state age (ms): "
              << std::chrono::duration_cast<std::chrono::milliseconds>(
                     AgxMsgRefClock::now() - state.time_stamp)
                     .count()
              << std::endl;

    auto actuator = scout->GetActuatorState();
    if (scout->GetParserProtocolVersion() == ProtocolVersion::AGX_V1) {
      for (int i = 0; i < 4; ++i) {
        printf("motor %d: current %f, rpm %d, driver temp %f, motor temp %f\n",
               actuator.actuator_state[i].motor_id,
               actuator.actuator_state[i].current,
               actuator.actuator_state[i].rpm,
               actuator.actuator_state[i].driver_temp,
               actuator.actuator_state[i].motor_temp);
      }
      std::cout << "actuator state age (ms): "
                << std::chrono::duration_cast<std::chrono::milliseconds>(
                       AgxMsgRefClock::now() - actuator.time_stamp)
                       .count()
                << std::endl;
    } else {
      for (int i = 0; i < 4; ++i) {
        printf("motor %d: current %f, rpm %d, driver temp %f, motor temp %f\n",
               actuator.actuator_hs_state[i].motor_id,
               actuator.actuator_hs_state[i].current,
               actuator.actuator_hs_state[i].rpm,
               actuator.actuator_ls_state[i].driver_temp,
               actuator.actuator_ls_state[i].motor_temp);
      }
      std::cout << "actuator state age (ms): "
                << std::chrono::duration_cast<std::chrono::milliseconds>(
                       AgxMsgRefClock::now() - actuator.time_stamp)
                       .count()
                << std::endl;
    }
    std::cout << "-------------------------------" << std::endl;

    usleep(20000);
    ++count;
  }

  return 0;
}