00001 // -*- mode: c++; -*- 00002 /********************************************************************* 00003 * Software License Agreement (BSD License) 00004 * 00005 * Copyright (c) 2014, JSK Lab 00006 * All rights reserved. 00007 * 00008 * Redistribution and use in source and binary forms, with or without 00009 * modification, are permitted provided that the following conditions 00010 * are met: 00011 * 00012 * * Redistributions of source code must retain the above copyright 00013 * notice, this list of conditions and the following disclaimer. 00014 * * Redistributions in binary form must reproduce the above 00015 * copyright notice, this list of conditions and the following 00016 * disclaimer in the documentation and/o2r other materials provided 00017 * with the distribution. 00018 * * Neither the name of the Willow Garage nor the names of its 00019 * contributors may be used to endorse or promote products derived 00020 * from this software without specific prior written permission. 00021 * 00022 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 00023 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 00024 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS 00025 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE 00026 * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, 00027 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, 00028 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; 00029 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER 00030 * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 00031 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN 00032 * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 00033 * POSSIBILITY OF SUCH DAMAGE. 00034 *********************************************************************/ 00035 00036 #ifndef JSK_RVIZ_PLUGINS_BOUDNING_BOX_ARRAY_DISPLAY_H_ 00037 #define JSK_RVIZ_PLUGINS_BOUDNING_BOX_ARRAY_DISPLAY_H_ 00038 00039 #include <jsk_pcl_ros/BoundingBoxArray.h> 00040 #include <rviz/properties/color_property.h> 00041 #include <rviz/properties/bool_property.h> 00042 #include <rviz/properties/float_property.h> 00043 #include <rviz/message_filter_display.h> 00044 #include <rviz/ogre_helpers/shape.h> 00045 #include <rviz/ogre_helpers/billboard_line.h> 00046 #include <OGRE/OgreSceneNode.h> 00047 00048 namespace jsk_rviz_plugin 00049 { 00050 class BoundingBoxArrayDisplay: public rviz::MessageFilterDisplay<jsk_pcl_ros::BoundingBoxArray> 00051 { 00052 Q_OBJECT 00053 public: 00054 typedef boost::shared_ptr<rviz::Shape> ShapePtr; 00055 typedef boost::shared_ptr<rviz::BillboardLine> BillboardLinePtr; 00056 BoundingBoxArrayDisplay(); 00057 virtual ~BoundingBoxArrayDisplay(); 00058 protected: 00059 virtual void onInitialize(); 00060 virtual void reset(); 00061 void allocateShapes(int num); 00062 void allocateBillboardLines(int num); 00063 QColor getColor(size_t index); 00064 rviz::ColorProperty* color_property_; 00065 rviz::FloatProperty* alpha_property_; 00066 rviz::BoolProperty* only_edge_property_; 00067 rviz::FloatProperty* line_width_property_; 00068 rviz::BoolProperty* auto_color_property_; 00069 QColor color_; 00070 double alpha_; 00071 bool only_edge_; 00072 bool auto_color_; 00073 double line_width_; 00074 std::vector<ShapePtr> shapes_; 00075 std::vector<BillboardLinePtr> edges_; 00076 private Q_SLOTS: 00077 void updateColor(); 00078 void updateAlpha(); 00079 void updateOnlyEdge(); 00080 void updateAutoColor(); 00081 void updateLineWidth(); 00082 private: 00083 void processMessage(const jsk_pcl_ros::BoundingBoxArray::ConstPtr& msg); 00084 }; 00085 00086 } 00087 #endif