ompl/base/src/PrecomputedStateSampler.cpp
00001 /********************************************************************* 00002 * Software License Agreement (BSD License) 00003 * 00004 * Copyright (c) 2012, Willow Garage 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 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 #include "ompl/base/PrecomputedStateSampler.h" 00038 #include "ompl/base/StateSpace.h" 00039 #include "ompl/util/Exception.h" 00040 00041 ompl::base::PrecomputedStateSampler::PrecomputedStateSampler(const StateSpace *space, const std::vector<const State*> &states) : 00042 StateSampler(space), states_(states) 00043 { 00044 if (states_.empty()) 00045 throw Exception("Empty set of states to sample from was specified"); 00046 minStateIndex_ = 0; 00047 maxStateIndex_ = states_.size() - 1; 00048 } 00049 00050 ompl::base::PrecomputedStateSampler::PrecomputedStateSampler(const StateSpace *space, const std::vector<const State*> &states, std::size_t minStateIndex, std::size_t maxStateIndex) : 00051 StateSampler(space), states_(states), minStateIndex_(minStateIndex), maxStateIndex_(maxStateIndex) 00052 { 00053 if (states_.empty()) 00054 throw Exception("Empty set of states to sample from was specified"); 00055 if (minStateIndex > maxStateIndex) 00056 throw Exception("Minimum state index cannot be larger than maximum state index"); 00057 if (maxStateIndex >= states_.size()) 00058 throw Exception("Index range out of bounds"); 00059 } 00060 00061 void ompl::base::PrecomputedStateSampler::sampleUniform(State *state) 00062 { 00063 space_->copyState(state, states_[rng_.uniformInt(minStateIndex_, maxStateIndex_)]); 00064 } 00065 00066 void ompl::base::PrecomputedStateSampler::sampleUniformNear(State *state, const State *near, const double distance) 00067 { 00068 int index = rng_.uniformInt(minStateIndex_, maxStateIndex_); 00069 double dist = space_->distance(near, states_[index]); 00070 if (dist > distance) 00071 space_->interpolate(near, states_[index], distance / dist, state); 00072 else 00073 space_->copyState(state, states_[index]); 00074 } 00075 00076 void ompl::base::PrecomputedStateSampler::sampleGaussian(State *state, const State *mean, const double stdDev) 00077 { 00078 sampleUniformNear(state, mean, rng_.gaussian(0.0, stdDev)); 00079 }