00001 // Copyright (C) 2010-2011 Institut de Robotica i Informatica Industrial, CSIC-UPC. 00002 // Author 00003 // All rights reserved. 00004 // 00005 // This file is part of iri-ros-pkg 00006 // iri-ros-pkg is free software: you can redistribute it and/or modify 00007 // it under the terms of the GNU Lesser General Public License as published by 00008 // the Free Software Foundation, either version 3 of the License, or 00009 // at your option) any later version. 00010 // 00011 // This program is distributed in the hope that it will be useful, 00012 // but WITHOUT ANY WARRANTY; without even the implied warranty of 00013 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 00014 // GNU Lesser General Public License for more details. 00015 // 00016 // You should have received a copy of the GNU Lesser General Public License 00017 // along with this program. If not, see <http://www.gnu.org/licenses/>. 00018 // 00019 // IMPORTANT NOTE: This code has been generated through a script from the 00020 // iri_ros_scripts. Please do NOT delete any comments to guarantee the correctness 00021 // of the scripts. ROS topics can be easly add by using those scripts. Please 00022 // refer to the IRI wiki page for more information: 00023 // http://wikiri.upc.es/index.php/Robotics_Lab 00024 00025 #ifndef _leaf_fitting_alg_node_h_ 00026 #define _leaf_fitting_alg_node_h_ 00027 00028 #include <iri_base_algorithm/iri_base_algorithm.h> 00029 #include "leaf_fitting_alg.h" 00030 00031 // [publisher subscriber headers] 00032 #include <sensor_msgs/Image.h> 00033 #include <sensor_msgs/image_encodings.h> 00034 #include <iri_leaf_segmentation/ClusterSet.h> 00035 #include <std_msgs/Float32MultiArray.h> 00036 00037 // [service client headers] 00038 00039 // [action server client headers] 00040 00041 // [opencv2] 00042 #include <opencv2/highgui/highgui.hpp> 00043 #include <opencv2/imgproc/imgproc.hpp> 00044 00045 #include <cv_bridge/CvBridge.h> 00046 #include <cv_bridge/cv_bridge.h> 00047 00048 #include <iri_leaf_fitting/GetConfidenceLeafFitting.h> 00049 00050 #include "leaf-fitting.h" 00051 #include "pnmfile.h" 00056 class LeafFittingAlgNode : public algorithm_base::IriBaseAlgorithm<LeafFittingAlgorithm> 00057 { 00058 private: 00059 int num_ccs_fin_; 00060 int **cluster_labels_; 00061 cv_bridge::CvImagePtr cv_ptr_; 00062 std_msgs::Float32MultiArray fit_score_list_msg_; 00063 00064 // [publisher attributes] 00065 ros::Publisher output_image_publisher_; 00066 sensor_msgs::Image Image_msg_; 00067 00068 // [subscriber attributes] 00069 00070 // [service attributes] 00071 ros::ServiceServer get_confidence_leaf_fitting_srv_; 00072 bool get_confidence_leaf_fitting_srv_callback(iri_leaf_fitting::GetConfidenceLeafFitting::Request &req, iri_leaf_fitting::GetConfidenceLeafFitting::Response &res); 00073 00074 // [client attributes] 00075 00076 // [action server attributes] 00077 00078 // [action client attributes] 00079 00080 public: 00087 LeafFittingAlgNode(void); 00088 00095 ~LeafFittingAlgNode(void); 00096 00097 protected: 00110 void mainNodeThread(void); 00111 00124 void node_config_update(Config &config, uint32_t level); 00125 00132 void addNodeDiagnostics(void); 00133 00134 // [diagnostic functions] 00135 00136 // [test functions] 00137 }; 00138 00139 #endif