ompl/geometric/planners/cforest/src/CForestStateSampler.cpp
00001 /********************************************************************* 00002 * Software License Agreement (BSD License) 00003 * 00004 * Copyright (c) 2014, Rice University 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 the Rice University 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: Javier V. Gómez*/ 00036 00037 #include "ompl/geometric/planners/cforest/CForestStateSampler.h" 00038 00039 void ompl::base::CForestStateSampler::sampleUniform(State *state) 00040 { 00041 if (!statesToSample_.empty()) 00042 getNextSample(state); 00043 else 00044 sampler_->sampleUniform(state); 00045 } 00046 00047 void ompl::base::CForestStateSampler::sampleUniformNear(State *state, const State *near, const double distance) 00048 { 00049 if (!statesToSample_.empty()) 00050 getNextSample(state); 00051 else 00052 sampler_->sampleUniformNear(state, near, distance); 00053 } 00054 00055 void ompl::base::CForestStateSampler::sampleGaussian(State *state, const State *mean, const double stdDev) 00056 { 00057 if (!statesToSample_.empty()) 00058 getNextSample(state); 00059 else 00060 sampler_->sampleGaussian(state, mean, stdDev); 00061 } 00062 00063 void ompl::base::CForestStateSampler::setStatesToSample(const std::vector<const State *> &states) 00064 { 00065 boost::mutex::scoped_lock slock(statesLock_); 00066 for (size_t i = 0; i < statesToSample_.size(); ++i) 00067 space_->freeState(statesToSample_[i]); 00068 statesToSample_.clear(); 00069 00070 statesToSample_.reserve(states.size()); 00071 // push in reverse order, so that the states are popped in order in getNextSample() 00072 for (std::vector<const State *>::const_iterator st = states.begin(); st != states.end(); ++st) 00073 { 00074 State *s = space_->allocState(); 00075 space_->copyState(s, *st); 00076 statesToSample_.push_back(s); 00077 } 00078 } 00079 00080 void ompl::base::CForestStateSampler::getNextSample(State *state) 00081 { 00082 boost::mutex::scoped_lock slock(statesLock_); 00083 space_->copyState(state, statesToSample_.back()); 00084 space_->freeState(statesToSample_.back()); 00085 statesToSample_.pop_back(); 00086 } 00087 00088 void ompl::base::CForestStateSampler::clear() 00089 { 00090 boost::mutex::scoped_lock slock(statesLock_); 00091 for (size_t i = 0; i < statesToSample_.size(); ++i) 00092 space_->freeState(statesToSample_[i]); 00093 statesToSample_.clear(); 00094 sampler_.reset(); 00095 }