mock_end_effector_driver.cpp

Go to the documentation of this file.

/*
mock_end_effector_driver.cpp
Copyright (C) 2020 Niryo
All rights reserved.
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program.  If not, see <http:// www.gnu.org/licenses/>.
*/
 
#include "ttl_driver/mock_end_effector_driver.hpp"
 
#include "ttl_driver/end_effector_reg.hpp"
#include <cstddef>
#include <memory>
#include <string>
#include <utility>
#include <vector>
 
namespace ttl_driver
{
// definition of methods
 
MockEndEffectorDriver::MockEndEffectorDriver(std::shared_ptr<FakeTtlData> data) : _fake_data(std::move(data)) {}
 
//*****************************
// AbstractTtlDriver interface
//*****************************
 
std::string MockEndEffectorDriver::str() const
{
    return common::model::HardwareTypeEnum(EndEffectorReg::motor_type).toString() + " : " + ttl_driver::AbstractEndEffectorDriver::str();
}
 
int MockEndEffectorDriver::ping(uint8_t id)
{
    if (std::find(_fake_data->full_id_list.begin(), _fake_data->full_id_list.end(), id) != _fake_data->full_id_list.end())
        return COMM_SUCCESS;
    return COMM_TX_FAIL;
}
 
int MockEndEffectorDriver::getModelNumber(uint8_t id, uint16_t &model_number)
{
    if (_fake_data->end_effector.id == id)
        model_number = _fake_data->end_effector.model_number;
    else
        return COMM_RX_FAIL;
    return COMM_SUCCESS;
}
 
int MockEndEffectorDriver::checkModelNumber(uint8_t id, uint16_t& model_number)
{
    int ping_result = getModelNumber(id, model_number);
 
    if (ping_result == COMM_SUCCESS)
    {
        if (model_number)
        {
            return PING_WRONG_MODEL_NUMBER;
        }
    }
 
    return ping_result;
}
 
int MockEndEffectorDriver::scan(std::vector<uint8_t> &id_list)
{
    id_list = _fake_data->full_id_list;
    return COMM_SUCCESS;
}
 
int MockEndEffectorDriver::reboot(uint8_t id) { return ping(id); }
 
int MockEndEffectorDriver::readFirmwareVersion(uint8_t id, std::string &version)
{
    if (COMM_SUCCESS != ping(id))
        return COMM_RX_FAIL;
 
    version = _fake_data->end_effector.firmware;
    return COMM_SUCCESS;
}
 
// ram read
 
int MockEndEffectorDriver::readTemperature(uint8_t id, uint8_t &temperature)
{
    if (COMM_SUCCESS != ping(id))
        return COMM_RX_FAIL;
 
    temperature = _fake_data->end_effector.temperature;
    return COMM_SUCCESS;
}
 
int MockEndEffectorDriver::readVoltage(uint8_t id, double &voltage)
{
    if (COMM_SUCCESS != ping(id))
        return COMM_RX_FAIL;
 
    voltage = static_cast<double>(voltage) / EndEffectorReg::VOLTAGE_CONVERSION;
    return COMM_SUCCESS;
}
 
int MockEndEffectorDriver::readHwErrorStatus(uint8_t id, uint8_t &hardware_error_status)
{
    if (COMM_SUCCESS != ping(id))
        return COMM_RX_FAIL;
 
    hardware_error_status = 0;
    return COMM_SUCCESS;
}
 
int MockEndEffectorDriver::syncReadFirmwareVersion(const std::vector<uint8_t> &id_list, std::vector<std::string> &firmware_list)
{
    int res = 0;
    firmware_list.clear();
    for (size_t i = 0; i < id_list.size(); i++)
        firmware_list.emplace_back(_fake_data->end_effector.firmware);
    return res;
}
 
int MockEndEffectorDriver::syncReadTemperature(const std::vector<uint8_t> &id_list, std::vector<uint8_t> &temperature_list)
{
    temperature_list.clear();
    for (size_t i = 0; i < id_list.size(); i++)
        temperature_list.emplace_back(_fake_data->end_effector.temperature);
    return COMM_SUCCESS;
}
 
int MockEndEffectorDriver::syncReadVoltage(const std::vector<uint8_t> &id_list, std::vector<double> &voltage_list)
{
    voltage_list.clear();
    for (size_t i = 0; i < id_list.size(); i++)
        voltage_list.emplace_back(static_cast<double>(_fake_data->end_effector.voltage) / EndEffectorReg::VOLTAGE_CONVERSION);
    return COMM_SUCCESS;
}
 
int MockEndEffectorDriver::syncReadRawVoltage(const std::vector<uint8_t> &id_list, std::vector<double> &voltage_list)
{
    voltage_list.clear();
    for (size_t i = 0; i < id_list.size(); i++)
        voltage_list.emplace_back(static_cast<double>(_fake_data->end_effector.voltage));
    return COMM_SUCCESS;
}
 
int MockEndEffectorDriver::syncReadHwStatus(const std::vector<uint8_t> &id_list, std::vector<std::pair<double, uint8_t>> &data_list)
{
    data_list.clear();
 
    for (size_t i = 0; i < id_list.size(); i++)
    {
        double voltage = _fake_data->end_effector.voltage;
        uint8_t temperature = _fake_data->end_effector.temperature;
        data_list.emplace_back(std::make_pair(voltage, temperature));
    }
    return COMM_SUCCESS;
}
 
int MockEndEffectorDriver::syncReadHwErrorStatus(const std::vector<uint8_t> & /*id_list*/, std::vector<uint8_t> &hw_error_list)
{
    hw_error_list.clear();
    hw_error_list.emplace_back(0);
    return COMM_SUCCESS;
}
 
// buttons status
 
int MockEndEffectorDriver::readButton0Status(uint8_t id, common::model::EActionType &action)
{
    if (COMM_SUCCESS != ping(id))
        return COMM_RX_FAIL;
 
    action = interpretActionValue(_fake_data->end_effector.button0_action);
    return COMM_SUCCESS;
}
 
int MockEndEffectorDriver::readButton1Status(uint8_t id, common::model::EActionType &action)
{
    if (COMM_SUCCESS != ping(id))
        return COMM_RX_FAIL;
 
    action = interpretActionValue(_fake_data->end_effector.button1_action);
    return COMM_SUCCESS;
}
 
int MockEndEffectorDriver::readButton2Status(uint8_t id, common::model::EActionType &action)
{
    if (COMM_SUCCESS != ping(id))
        return COMM_RX_FAIL;
 
    action = interpretActionValue(_fake_data->end_effector.button2_action);
    return COMM_SUCCESS;
}
 
int MockEndEffectorDriver::syncReadButtonsStatus(const uint8_t &id, std::vector<common::model::EActionType> &action_list)
{
    action_list.clear();
 
    if (COMM_SUCCESS != ping(id))
        return COMM_RX_FAIL;
 
    action_list.emplace_back(interpretActionValue(_fake_data->end_effector.button0_action));
    action_list.emplace_back(interpretActionValue(_fake_data->end_effector.button1_action));
    action_list.emplace_back(interpretActionValue(_fake_data->end_effector.button2_action));
 
    return COMM_SUCCESS;
}
 
// accelerometers and collision
 
int MockEndEffectorDriver::readAccelerometerXValue(uint8_t id, uint32_t &x_value)
{
    if (COMM_SUCCESS != ping(id))
        return COMM_RX_FAIL;
 
    x_value = _fake_data->end_effector.x_value;
    return COMM_SUCCESS;
}
 
int MockEndEffectorDriver::readAccelerometerYValue(uint8_t id, uint32_t &y_value)
{
    if (COMM_SUCCESS != ping(id))
        return COMM_RX_FAIL;
 
    y_value = _fake_data->end_effector.y_value;
    return COMM_SUCCESS;
}
 
int MockEndEffectorDriver::readAccelerometerZValue(uint8_t id, uint32_t &z_value)
{
    if (COMM_SUCCESS != ping(id))
        return COMM_RX_FAIL;
 
    z_value = _fake_data->end_effector.z_value;
    return COMM_SUCCESS;
}
 
int MockEndEffectorDriver::readCollisionStatus(uint8_t id, bool &status)
{
    if (COMM_SUCCESS != ping(id))
        return COMM_RX_FAIL;
 
    status = false;
    return COMM_SUCCESS;
}
 
int MockEndEffectorDriver::writeCollisionThresh(uint8_t id, int thresh)
{
    if (COMM_SUCCESS != ping(id))
        return COMM_RX_FAIL;
    return COMM_SUCCESS;
}
 
int MockEndEffectorDriver::writeCollisionThreshAlgo2(uint8_t id, int thresh)
{
    if (COMM_SUCCESS != ping(id))
        return COMM_RX_FAIL;
    return COMM_SUCCESS;
}
 
int MockEndEffectorDriver::readDigitalInput(uint8_t id, bool &in)
{
    if (COMM_SUCCESS != ping(id))
        return COMM_RX_FAIL;
 
    in = _fake_data->end_effector.digital_input;
    return COMM_SUCCESS;
}
 
int MockEndEffectorDriver::writeDigitalInput(uint8_t id, bool in)
{
    if (COMM_SUCCESS != ping(id))
        return COMM_RX_FAIL;
 
    _fake_data->end_effector.digital_input = in;
    return COMM_SUCCESS;
}
 
int MockEndEffectorDriver::writeDigitalOutput(uint8_t id, bool out)
{
    if (COMM_SUCCESS != ping(id))
        return COMM_RX_FAIL;
 
    _fake_data->end_effector.digital_output = out;
    return COMM_SUCCESS;
}
 
}  // namespace ttl_driver