arm.hpp

Go to the documentation of this file.

 
#pragma once
 
#include <string>
 
#include <boost/variant/variant.hpp>
 
#include <viam/api/common/v1/common.pb.h>
#include <viam/api/component/arm/v1/arm.pb.h>
 
#include <viam/sdk/common/pose.hpp>
#include <viam/sdk/resource/stoppable.hpp>
#include <viam/sdk/spatialmath/geometry.hpp>
 
namespace viam {
namespace sdk {
 
 
class Arm : public Component, public Stoppable {
    // Base class for use below in defining kinematics data strong typedefs
    template <class Tag>
    struct raw_bytes {
        // Pre c++17 our derived classes aren't aggregate initializable so we need to define
        // and using declare some ctors
 
        raw_bytes() = default;
        raw_bytes(std::vector<unsigned char> b) : bytes(std::move(b)) {}
 
        std::vector<unsigned char> bytes{};
    };
 
    // Comparison operator helper for the data types below
    template <class DataType>
    struct EqCompare {
        inline friend bool operator==(const DataType& lhs, const DataType& rhs) {
            return lhs.bytes == rhs.bytes;
        }
    };
 
   public:
    struct KinematicsDataUnspecified : raw_bytes<KinematicsDataUnspecified>,
                                       EqCompare<KinematicsDataUnspecified> {};
    struct KinematicsDataSVA : raw_bytes<KinematicsDataSVA>, EqCompare<KinematicsDataSVA> {
        using raw_bytes<KinematicsDataSVA>::raw_bytes;
    };
    struct KinematicsDataURDF : raw_bytes<KinematicsDataURDF>, EqCompare<KinematicsDataURDF> {
        using raw_bytes<KinematicsDataURDF>::raw_bytes;
    };
 
    using KinematicsData =
        boost::variant<KinematicsDataUnspecified, KinematicsDataSVA, KinematicsDataURDF>;
 
    static KinematicsData from_proto(const viam::common::v1::GetKinematicsResponse& proto);
 
    inline pose get_end_position() {
        return get_end_position({});
    }
 
    virtual pose get_end_position(const AttributeMap& extra) = 0;
 
    inline void move_to_position(const pose& pose) {
        move_to_position(pose, {});
    }
 
    virtual void move_to_position(const pose& pose, const AttributeMap& extra) = 0;
 
    inline std::vector<double> get_joint_positions() {
        return get_joint_positions({});
    }
 
    virtual std::vector<double> get_joint_positions(const AttributeMap& extra) = 0;
 
    virtual void move_to_joint_positions(const std::vector<double>& positions,
                                         const AttributeMap& extra) = 0;
 
    virtual bool is_moving() = 0;
 
    virtual AttributeMap do_command(const AttributeMap& command) = 0;
 
    virtual KinematicsData get_kinematics(const AttributeMap& extra) = 0;
 
    inline KinematicsData get_kinematics() {
        return get_kinematics({});
    }
 
    inline std::vector<GeometryConfig> get_geometries() {
        return get_geometries({});
    }
 
    virtual std::vector<GeometryConfig> get_geometries(const AttributeMap& extra) = 0;
 
    API api() const override;
 
   protected:
    explicit Arm(std::string name);
};
 
template <>
struct API::traits<Arm> {
    static API api();
};
 
}  // namespace sdk
}  // namespace viam