00001 /********************************************************************* 00002 * Software License Agreement (BSD License) 00003 * 00004 * Copyright (c) 2012, Willow Garage, Inc. 00005 * All rights reserved. 00006 * 00007 * Redistribution and use in source and binary forms, with or without 00008 * modification, are permitted provided that the following conditions 00009 * are met: 00010 * 00011 * * Redistributions of source code must retain the above copyright 00012 * notice, this list of conditions and the following disclaimer. 00013 * * Redistributions in binary form must reproduce the above 00014 * copyright notice, this list of conditions and the following 00015 * disclaimer in the documentation and/or other materials provided 00016 * with the distribution. 00017 * * Neither the name of Willow Garage nor the names of its 00018 * contributors may be used to endorse or promote products derived 00019 * from this software without specific prior written permission. 00020 * 00021 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 00022 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 00023 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS 00024 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE 00025 * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, 00026 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, 00027 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; 00028 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER 00029 * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 00030 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN 00031 * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 00032 * POSSIBILITY OF SUCH DAMAGE. 00033 *********************************************************************/ 00034 00035 /* Author: Ioan Sucan */ 00036 00037 #ifndef MOVEIT_PICK_PLACE_MANIPULATION_PIPELINE_ 00038 #define MOVEIT_PICK_PLACE_MANIPULATION_PIPELINE_ 00039 00040 #include <moveit/pick_place/manipulation_stage.h> 00041 #include <boost/thread.hpp> 00042 #include <boost/shared_ptr.hpp> 00043 #include <boost/function.hpp> 00044 #include <vector> 00045 #include <deque> 00046 00047 namespace pick_place 00048 { 00049 00051 class ManipulationPipeline 00052 { 00053 public: 00054 00055 ManipulationPipeline(const std::string &name, unsigned int nthreads); 00056 virtual ~ManipulationPipeline(); 00057 00058 const std::string& getName() const 00059 { 00060 return name_; 00061 } 00062 00063 void setSolutionCallback(const boost::function<void()> &callback) 00064 { 00065 solution_callback_ = callback; 00066 } 00067 00068 void setEmptyQueueCallback(const boost::function<void()> &callback) 00069 { 00070 empty_queue_callback_ = callback; 00071 } 00072 00073 ManipulationPipeline& addStage(const ManipulationStagePtr &next); 00074 const ManipulationStagePtr& getFirstStage() const; 00075 const ManipulationStagePtr& getLastStage() const; 00076 void reset(); 00077 00078 void setVerbose(bool flag); 00079 00080 void signalStop(); 00081 void start(); 00082 void stop(); 00083 00084 void push(const ManipulationPlanPtr &grasp); 00085 void clear(); 00086 00087 const std::vector<ManipulationPlanPtr>& getSuccessfulManipulationPlans() const 00088 { 00089 return success_; 00090 } 00091 00092 const std::vector<ManipulationPlanPtr>& getFailedManipulationPlans() const 00093 { 00094 return failed_; 00095 } 00096 00097 void reprocessLastFailure(); 00098 00099 protected: 00100 00101 void processingThread(unsigned int index); 00102 00103 std::string name_; 00104 unsigned int nthreads_; 00105 bool verbose_; 00106 std::vector<ManipulationStagePtr> stages_; 00107 00108 std::deque<ManipulationPlanPtr> queue_; 00109 std::vector<ManipulationPlanPtr> success_; 00110 std::vector<ManipulationPlanPtr> failed_; 00111 00112 std::vector<boost::thread*> processing_threads_; 00113 boost::condition_variable queue_access_cond_; 00114 boost::mutex queue_access_lock_; 00115 boost::mutex result_lock_; 00116 00117 boost::function<void()> solution_callback_; 00118 boost::function<void()> empty_queue_callback_; 00119 unsigned int empty_queue_threads_; 00120 00121 bool stop_processing_; 00122 00123 }; 00124 00125 } 00126 00127 #endif