AdmittanceController.cpp

Go to the documentation of this file.

/**
 * This file is part of ArmarX.
 *
 * ArmarX is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 as
 * published by the Free Software Foundation.
 *
 * ArmarX 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/>.
 *
 * @author     Jianfeng Gao ( jianfeng dot gao at kit dot edu )
 * @date       2023
 * @copyright  http://www.gnu.org/licenses/gpl-2.0.txt
 *             GNU General Public License
 */
 
#include "AdmittanceController.h"
 
#include <armarx/control/common/utils.h>
 
namespace armarx::control::njoint_mp_controller::task_space
{
    NJointControllerRegistration<NJointTSAdmittanceMPController>
        registrationControllerNJointTSAdmittanceMPController("NJointTSAdmittanceMPController");
 
    NJointTSAdmittanceMPController::NJointTSAdmittanceMPController(
        const RobotUnitPtr& robotUnit,
        const NJointControllerConfigPtr& config,
        const VirtualRobot::RobotPtr& robot) :
        NJointTaskspaceAdmittanceController{robotUnit, config, robot}, mp::MPPool{}
    {
        ARMARX_IMPORTANT << getClassName() + " construction done";
    }
 
    std::string
    NJointTSAdmittanceMPController::getClassName(const Ice::Current&) const
    {
        return "NJointTSAdmittanceMPController";
    }
 
    void
    NJointTSAdmittanceMPController::updateMPConfig(const ::armarx::aron::data::dto::DictPtr& dto,
                                                   const Ice::Current& iceCurrent)
    {
        if (resetMPs(dto, controllableNodeSets))
        {
            for (auto& pair : limb)
            {
                pair.second->rtFirstRun.store(true);
            }
        }
    }
 
    void
    NJointTSAdmittanceMPController::additionalTask()
    {
        std::lock_guard<std::recursive_mutex> lock(mtx_mps);
        bool rtSafe = additionalTaskUpdateStatus();
 
        std::map<std::string, bool> mpRunning;
        for (auto& _mp : mps)
        {
            mpRunning[_mp.second.mp->getMPName()] = _mp.second.mp->isRunning();
            if (_mp.second.mp->isFinished())
            {
                continue;
            }
            if (_mp.second.mp->getRole() == "taskspace")
            {
                auto& arm = limb.at(_mp.second.mp->getNodeSetName());
                if (_mp.second.mp->isFirstRun())
                {
                    _mp.second.mp->validateInitialState(common::mat4ToDVec(arm->rtStatusInNonRT.currentPose));
                }
                mp::TSMPInputPtr in = std::dynamic_pointer_cast<mp::TSMPInput>(_mp.second.input);
                in->pose = arm->rtStatusInNonRT.currentPose;
                in->vel = arm->rtStatusInNonRT.currentTwist;
                in->deltaT = arm->rtStatusInNonRT.deltaT;
            }
            else if (_mp.second.mp->getRole() == "nullspace")
            {
                auto& arm = limb.at(_mp.second.mp->getNodeSetName());
                mp::JSMPInputPtr in = std::dynamic_pointer_cast<mp::JSMPInput>(_mp.second.input);
                ARMARX_CHECK_EQUAL(arm->rtStatusInNonRT.numJoints,
                                   arm->rtStatusInNonRT.jointPos.size());
                in->angleRadian = arm->rtStatusInNonRT.jointPos;
                in->angularVel = arm->rtStatusInNonRT.qvelFiltered;
                in->deltaT = arm->rtStatusInNonRT.deltaT;
            }
            else if (_mp.second.mp->getRole() == "hand")
            {
                auto& hand = hands->hands.at(_mp.second.mp->getNodeSetName());
                mp::JSMPInputPtr in = std::dynamic_pointer_cast<mp::JSMPInput>(_mp.second.input);
                in->angleRadian =
                    hand->bufferRTToNonRT.getUpToDateReadBuffer().jointPosition.value();
                in->deltaT = hand->bufferRTToNonRT.getReadBuffer().deltaT;
            }
        }
        if (not rtSafe)
        {
            handleRTNotSafeInNonRT();
            return;
            // Make sure that you do not call additionalTaskSetTarget before return;
            // since we overwrite the arm->nonRtConfig in additionalTaskUpdateStatus() with user config, which does not
            // align with the up-to-date mp status. userConfig is only updated to align with MP status everytime a MP
            // finishes execution. Only then, user can use this controller as if no mp is there.
        }
 
        runMPs(rtSafe);
 
        /// set mp target to nonRT config data structure
        for (auto& _mp : mps)
        {
            if (not mpRunning.at(_mp.second.mp->getMPName()))
            {
                continue;
            }
            for (auto& ftGuard : mpConfig.ftGuard)
            {
                if (ftGuard.mpName == _mp.second.mp->getMPName())
                {
                    bool const forceGuard =
                        (ftGuard.force.has_value() and
                         ftGuard.force.value() >= _mp.second.mp->getCanonicalValue());
                    bool const torqueGuard =
                        (ftGuard.torque.has_value() and
                         ftGuard.torque.value() >= _mp.second.mp->getCanonicalValue());
 
                    auto& arm = limb.at(_mp.second.mp->getNodeSetName());
                    bool const resetForce =
                        not arm->nonRtConfig.ftConfig.enableSafeGuardForce and forceGuard;
                    bool const resetTorque =
                        not arm->nonRtConfig.ftConfig.enableSafeGuardTorque and torqueGuard;
                    if (resetForce or resetTorque)
                    {
                        ARMARX_INFO << "Triggering force torque safety guard for "
                                    << arm->kinematicChainName << " at can value "
                                    << _mp.second.mp->getCanonicalValue();
                    }
 
                    arm->nonRtConfig.ftConfig.enableSafeGuardForce = forceGuard;
                    arm->nonRtConfig.ftConfig.enableSafeGuardTorque = torqueGuard;
                    arm->controller.ftsensor.enableSafeGuard(resetForce, resetTorque);
                }
            }
 
            if (_mp.second.mp->getRole() == "taskspace")
            {
                auto& arm = limb.at(_mp.second.mp->getNodeSetName());
                mp::TSMPOutputPtr out =
                    std::dynamic_pointer_cast<mp::TSMPOutput>(_mp.second.output);
 
                /// Note that here we only report warning but still set target to RT, we rely on MP phase stop and
                /// RT safety mechanism to stop or pause the action if needed
                common::detectAndReportPoseDeviationWarning(arm->rtStatusInNonRT.currentPose,
                                                            out->pose,
                                                            "current pose", "TSMP pose",
                                                            arm->nonRtConfig.safeDistanceMMToGoal,
                                                            arm->nonRtConfig.safeRotAngleDegreeToGoal,
                                                            _mp.second.mp->getMPName() + "mp_set_target");
                arm->nonRtConfig.desiredPose = out->pose;
                arm->nonRtConfig.desiredTwist = out->vel;
            }
            else if (_mp.second.mp->getRole() == "nullspace")
            {
                auto& arm = limb.at(_mp.second.mp->getNodeSetName());
                mp::JSMPOutputPtr out =
                    std::dynamic_pointer_cast<mp::JSMPOutput>(_mp.second.output);
                ARMARX_CHECK_EQUAL(arm->rtStatusInNonRT.numJoints, out->angleRadian.size());
                arm->nonRtConfig.desiredNullspaceJointAngles.value() = out->angleRadian;
            }
            else if (_mp.second.mp->getRole() == "hand")
            {
                mp::JSMPOutputPtr out =
                    std::dynamic_pointer_cast<mp::JSMPOutput>(_mp.second.output);
                auto& hand = hands->hands.at(_mp.second.mp->getNodeSetName());
                mp::JSMPInputPtr in = std::dynamic_pointer_cast<mp::JSMPInput>(_mp.second.input);
                ARMARX_CHECK_EQUAL(hand->targetNonRT.jointPosition.value().size(),
                                   out->angleRadian.size());
                hand->targetNonRT.jointPosition.value() = out->angleRadian;
            }
            if (mpRunning.at(_mp.second.mp->getMPName()) and _mp.second.mp->isFinished())
            {
                ARMARX_INFO << "reset buffer for nonRtConfig of " << _mp.second.mp->getMPName();
                for (auto& pair : limb)
                {
                    auto& arm = pair.second;
                    arm->bufferConfigUserToNonRt.reinitAllBuffers(arm->nonRtConfig);
                }
                for (auto& pair : limb)
                {
                    userConfig.limbs.at(pair.first) = pair.second->nonRtConfig;
                }
                if (hands)
                {
                    hands->reinitBuffer(userConfig.hands);
                }
            }
        }
        additionalTaskSetTarget();
    }
} // namespace armarx::control::njoint_mp_controller::task_space