$search
00001 /* 00002 * Copyright (c) 2011, Willow Garage, Inc. 00003 * All rights reserved. 00004 * 00005 * Redistribution and use in source and binary forms, with or without 00006 * modification, are permitted provided that the following conditions are met: 00007 * 00008 * * Redistributions of source code must retain the above copyright 00009 * notice, this list of conditions and the following disclaimer. 00010 * * Redistributions in binary form must reproduce the above copyright 00011 * notice, this list of conditions and the following disclaimer in the 00012 * documentation and/or other materials provided with the distribution. 00013 * * Neither the name of the Willow Garage, Inc. nor the names of its 00014 * contributors may be used to endorse or promote products derived from 00015 * this software without specific prior written permission. 00016 * 00017 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" 00018 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 00019 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 00020 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE 00021 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 00022 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 00023 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 00024 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 00025 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 00026 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 00027 * POSSIBILITY OF SUCH DAMAGE. 00028 */ 00029 00030 #ifndef RVIZ_INTERACTIVE_MARKER_H_ 00031 #define RVIZ_INTERACTIVE_MARKER_H_ 00032 00033 #include "rviz/selection/forwards.h" 00034 00035 #include "interactive_marker_control.h" 00036 00037 #include <ogre_tools/axes.h> 00038 00039 #include <visualization_msgs/InteractiveMarker.h> 00040 #include <visualization_msgs/InteractiveMarkerPose.h> 00041 #include <visualization_msgs/InteractiveMarkerFeedback.h> 00042 00043 #include <geometry_msgs/Pose.h> 00044 00045 #include <OGRE/OgreVector3.h> 00046 #include <OGRE/OgreQuaternion.h> 00047 00048 #include <boost/shared_ptr.hpp> 00049 00050 #include <ros/publisher.h> 00051 00052 namespace Ogre { 00053 class SceneNode; 00054 } 00055 00056 class wxMenu; 00057 00058 namespace rviz 00059 { 00060 class VisualizationManager; 00061 class InteractiveMarkerDisplay; 00062 00063 class InteractiveMarker : public wxEvtHandler 00064 { 00065 public: 00066 InteractiveMarker( InteractiveMarkerDisplay *owner, VisualizationManager *vis_manager, std::string topic_ns, std::string client_id ); 00067 virtual ~InteractiveMarker(); 00068 00069 // reset contents to reflect the data from a new message 00070 // @return success 00071 bool processMessage( visualization_msgs::InteractiveMarkerConstPtr message ); 00072 00073 // reset contents to reflect the data from a new message 00074 // @return success 00075 void processMessage( visualization_msgs::InteractiveMarkerPoseConstPtr message ); 00076 00077 // called every frame update 00078 void update(float wall_dt); 00079 00080 // set the pose of the parent frame, relative to the fixed frame 00081 void setReferencePose( Ogre::Vector3 position, Ogre::Quaternion orientation ); 00082 00083 // directly set the pose, relative to parent frame 00084 // if publish is set to true, publish the change 00085 void setPose( Ogre::Vector3 position, Ogre::Quaternion orientation, const std::string &control_name ); 00086 00087 void translate( Ogre::Vector3 delta_position, const std::string &control_name ); 00088 void rotate( Ogre::Quaternion delta_orientation, const std::string &control_name ); 00089 00090 // schedule a pose reset once dragging is finished 00091 void requestPoseUpdate( Ogre::Vector3 position, Ogre::Quaternion orientation ); 00092 00093 void startDragging(); 00094 void stopDragging(); 00095 00096 const Ogre::Vector3& getPosition() { return position_; } 00097 const Ogre::Quaternion& getOrientation() { return orientation_; } 00098 00099 float getSize() { return scale_; } 00100 const std::string &getReferenceFrame() { return reference_frame_; } 00101 const std::string& getName() { return name_; } 00102 00103 // show name above marker 00104 void setShowDescription( bool show ); 00105 00106 // show axes in origin 00107 void setShowAxes( bool show ); 00108 00109 // @return true if the mouse event was intercepted, false if it was ignored 00110 bool handleMouseEvent(ViewportMouseEvent& event, const std::string &control_name ); 00111 00112 void handleMenuSelect(wxCommandEvent &evt); 00113 00114 // fill in current marker pose & name, publish 00115 void publishFeedback(visualization_msgs::InteractiveMarkerFeedback &feedback, 00116 bool mouse_point_valid = false, 00117 const Ogre::Vector3& mouse_point_rel_world = Ogre::Vector3(0,0,0) ); 00118 00119 protected: 00120 00121 void publishPose(); 00122 00123 void reset(); 00124 00125 void updateReferencePose(); 00126 00127 wxString makeMenuString( const std::string &entry ); 00128 00129 // Recursively append menu and submenu entries to menu, based on a 00130 // vector of menu entry id numbers describing the menu entries at the 00131 // current level. 00132 void populateMenu( wxMenu* menu, std::vector<uint32_t>& ids ); 00133 00134 InteractiveMarkerDisplay *owner_; 00135 VisualizationManager *vis_manager_; 00136 00137 // pose of parent coordinate frame 00138 std::string reference_frame_; 00139 ros::Time reference_time_; 00140 bool frame_locked_; 00141 00142 // node representing reference frame in tf, like /map, /base_link, /head, etc. 00143 Ogre::SceneNode *reference_node_; 00144 00145 // pose being controlled, relative to reference frame 00146 Ogre::Vector3 position_; 00147 Ogre::Quaternion orientation_; 00148 00149 // has the pose changed since the last feedback was sent? 00150 bool pose_changed_; 00151 double time_since_last_feedback_; 00152 00153 typedef boost::shared_ptr<InteractiveMarkerControl> InteractiveMarkerControlPtr; 00154 std::list<InteractiveMarkerControlPtr> controls_; 00155 00156 std::string name_; 00157 std::string description_; 00158 00159 bool dragging_; 00160 std::string old_target_frame_; 00161 00162 // pose being controlled 00163 bool pose_update_requested_; 00164 Ogre::Vector3 requested_position_; 00165 Ogre::Quaternion requested_orientation_; 00166 00167 float scale_; 00168 00169 boost::shared_ptr<wxMenu> menu_; 00170 00171 // Helper to more simply represent the menu tree. 00172 struct MenuNode 00173 { 00174 visualization_msgs::MenuEntry entry; 00175 std::vector<uint32_t> child_ids; 00176 }; 00177 00178 // maps menu index to menu entry and item 00179 std::map< uint32_t, MenuNode > menu_entries_; 00180 00181 // Helper to store the top level of the menu tree. 00182 std::vector<uint32_t> top_level_menu_ids_; 00183 00184 // which control has popped up the menu 00185 std::string last_control_name_; 00186 00187 double heart_beat_t_; 00188 00189 // visual aids 00190 00191 ogre_tools::Axes *axes_; 00192 Ogre::SceneNode *axes_node_; 00193 00194 InteractiveMarkerControlPtr description_control_; 00195 00196 ros::Publisher feedback_pub_; 00197 std::string topic_ns_; 00198 std::string client_id_; 00199 00200 boost::recursive_mutex mutex_; 00201 00202 boost::shared_ptr< boost::thread > sys_thread_; 00203 00204 bool got_3d_point_for_menu_; 00205 Ogre::Vector3 three_d_point_for_menu_; 00206 }; 00207 00208 00209 } 00210 00211 #endif /* INTERACTIVE_MARKER_H_ */