JointMaker.hh

Go to the documentation of this file.

/*
 * Copyright (C) 2014-2015 Open Source Robotics Foundation
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
*/

#ifndef _JOINTMAKER_HH_
#define _JOINTMAKER_HH_

#include <string>
#include <vector>
#include <boost/unordered/unordered_map.hpp>

#include <sdf/sdf.hh>

#include "gazebo/common/MouseEvent.hh"
#include "gazebo/common/KeyEvent.hh"
#include "gazebo/common/CommonTypes.hh"
#include "gazebo/math/Pose.hh"
#include "gazebo/rendering/RenderTypes.hh"
#include "gazebo/gui/qt.h"
#include "gazebo/util/system.hh"

namespace Ogre
{
  class BillboardSet;
}

namespace boost
{
  class recursive_mutex;
}

namespace gazebo
{
  namespace gui
  {
    class JointData;
    class JointInspector;


    class GAZEBO_VISIBLE JointMaker : public QObject
    {
      Q_OBJECT

      public: enum JointType
      {
        JOINT_NONE,
        JOINT_FIXED,
        JOINT_SLIDER,
        JOINT_HINGE,
        JOINT_HINGE2,
        JOINT_SCREW,
        JOINT_UNIVERSAL,
        JOINT_BALL
      };

      public: JointMaker();

      public: virtual ~JointMaker();

      public: void Reset();

      public: void EnableEventHandlers();

      public: void DisableEventHandlers();

      public: void AddJoint(const std::string &_type);

      public: void AddJoint(JointType _type);

      public: JointData *CreateJoint(rendering::VisualPtr _parent,
          rendering::VisualPtr _child);

      public: void CreateHotSpot(JointData *_joint);

      public: void Update();

      public: void RemoveJoint(const std::string &_jointName);

      public: void RemoveJointsByPart(const std::string &_partName);

      public: void GenerateSDF();

      public: sdf::ElementPtr GetSDF() const;

      public: static int GetJointAxisCount(JointMaker::JointType _type);

      public: static std::string GetTypeAsString(JointMaker::JointType _type);

      public: JointMaker::JointType GetState() const;

      public: void Stop();

      public: unsigned int GetJointCount();

      private: bool OnMousePress(const common::MouseEvent &_event);

      private: bool OnMouseRelease(const common::MouseEvent &_event);

      private: bool OnMouseMove(const common::MouseEvent &_event);

      private: bool OnMouseDoubleClick(const common::MouseEvent &_event);

      private: bool OnKeyPress(const common::KeyEvent &_event);

      private: math::Vector3 GetPartWorldCentroid(
          const rendering::VisualPtr _visual);

      private: void OpenInspector(const std::string &_name);

      Q_SIGNALS: void JointAdded();

      private slots: void OnOpenInspector();

      private: std::vector<math::Vector3> UnitVectors;

      private: JointMaker::JointType jointType;

      private: rendering::VisualPtr hoverVis;

      private: rendering::VisualPtr prevHoverVis;

      private: rendering::VisualPtr selectedVis;

      private: rendering::VisualPtr inspectVis;

      private: boost::unordered_map<std::string, JointData *> joints;

      private: JointData *mouseJoint;

      private: std::vector<event::ConnectionPtr> connections;

      private: bool newJointCreated;

      private: boost::unordered_map<JointMaker::JointType, std::string>
          jointMaterials;

      private: sdf::ElementPtr modelSDF;

      private: int jointCounter;

      private: QAction *inspectAct;

      private: boost::recursive_mutex *updateMutex;

      private: rendering::VisualPtr selectedJoint;
    };


    class GAZEBO_VISIBLE JointData : public QObject
    {
      Q_OBJECT

      public: std::string name;

      public: rendering::VisualPtr visual;

      public: rendering::VisualPtr hotspot;

      public: rendering::VisualPtr parent;

      public: rendering::VisualPtr child;

      public: rendering::DynamicLines *line;

      public: Ogre::BillboardSet *handles;

      public: JointMaker::JointType type;

      public: math::Vector3 axis[2];

      public: double lowerLimit[2];

      public: double upperLimit[2];

      public: math::Pose pose;

      public: bool dirty;

      public: JointInspector *inspector;

      private slots: void OnApply();
    };
  }
}
#endif