PointCloudToArViz.h

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/>.
 *
 * @package    VisionX::ArmarXObjects::PointCloudToArViz
 * @author     Rainer Kartmann ( rainer dot kartmann at kit dot edu )
 * @date       2020
 * @copyright  http://www.gnu.org/licenses/gpl-2.0.txt
 *             GNU General Public License
 */
 
#pragma once
 
#include <map>
#include <mutex>
#include <optional>
#include <VirtualRobot/VirtualRobot.h>
 
#include <ArmarXCore/core/Component.h>
 
#include <ArmarXGui/libraries/ArmarXGuiComponentPlugins/LightweightRemoteGuiComponentPlugin.h>
 
#include "RobotAPI/libraries/armem/client/plugins/ReaderWriterPlugin.h"
#include <RobotAPI/libraries/RobotAPIComponentPlugins/ArVizComponentPlugin.h>
#include <RobotAPI/libraries/armem_robot_state/client/common/VirtualRobotReader.h>
#include <RobotAPI/libraries/armem/client/plugins.h>
 
// The has_member macro from
// > simox/SimoxUtility/meta/has_member_macros/has_member.hpp
// breaks compilation here if boost thread is not already included
#ifdef has_member
#undef has_member
#endif
 
#include <VisionX/interface/components/PointCloudToArViz.h>
#include <VisionX/components/pointcloud_core/PointCloudProcessor.h>
 
namespace visionx
{
 
    class PointCloudToArVizPropertyDefinitions :
        public visionx::PointCloudProcessorPropertyDefinitions
    {
    public:
        PointCloudToArVizPropertyDefinitions(std::string prefix);
    };
 
 
    class PointCloudToArViz :
          virtual public visionx::PointCloudToArVizInterface
        , virtual public visionx::PointCloudProcessor
        , virtual public armarx::ArVizComponentPluginUser
        , virtual public armarx::LightweightRemoteGuiComponentPluginUser
    {
    public:
        PointCloudToArViz();
 
        std::string getDefaultName() const override;
 
        virtual void setPointSizeInPixels(float pointSizeInPixels, const Ice::Current& = Ice::emptyCurrent) override;
        virtual void setCustomTransform(const armarx::PoseBasePtr& transform, const Ice::Current& = Ice::emptyCurrent) override;
 
 
    protected:
 
        void onInitPointCloudProcessor() override;
 
        void onConnectPointCloudProcessor() override;
 
        void onDisconnectPointCloudProcessor() override;
 
        void onExitPointCloudProcessor() override;
 
 
        void process() override;
 
 
        armarx::PropertyDefinitionsPtr createPropertyDefinitions() override;
 
 
        // LightweightRemoteGuiComponentPluginUser interface
    public:
 
        void createRemoteGuiTab();
        void RemoteGui_update() override;
 
 
    private:
 
        struct Parameters
        {
            IceUtil::Time providerWaitTime = IceUtil::Time::milliSeconds(100);
 
            bool checkFinite = false;
 
            float pointSizeInPixels = 1;
 
            bool labeled = false;
 
            // std::optional<Eigen::Matrix4f> customTransform;
 
 
            std::string robotName = "Armar6";
 
            // If the point cloud header does not provide a frame id, this info is used.
            std::string pointCloudNodeName;
 
        };
        std::mutex paramMutex;
        Parameters params;
 
        template <class PointT>
            void applyCustomTransform(pcl::PointCloud<PointT>& pointCloud);
 
 
        std::map<std::string, armarx::viz::PointCloud> cache;
 
 
        struct RemoteGuiTab : armarx::RemoteGui::Client::Tab
        {
            armarx::RemoteGui::Client::CheckBox checkFinite;
            armarx::RemoteGui::Client::FloatSpinBox pointSizeInPixels;
            armarx::RemoteGui::Client::CheckBox labeled;
        };
        RemoteGuiTab tab;
 
 
        armarx::armem::client::plugins::ReaderWriterPlugin<
            armarx::armem::robot_state::VirtualRobotReader>* virtualRobotReaderPlugin = nullptr;
 
        VirtualRobot::RobotPtr robot;
 
    };
}