ompl/control/src/SimpleDirectedControlSampler.cpp
00001 /********************************************************************* 00002 * Software License Agreement (BSD License) 00003 * 00004 * Copyright (c) 2011, 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 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/control/SimpleDirectedControlSampler.h" 00038 #include "ompl/control/SpaceInformation.h" 00039 00040 ompl::control::SimpleDirectedControlSampler::SimpleDirectedControlSampler(const SpaceInformation *si, unsigned int k) : 00041 DirectedControlSampler(si), cs_(si->allocControlSampler()), numControlSamples_(k) 00042 { 00043 } 00044 00045 ompl::control::SimpleDirectedControlSampler::~SimpleDirectedControlSampler() 00046 { 00047 } 00048 00049 unsigned int ompl::control::SimpleDirectedControlSampler::sampleTo(Control *control, const base::State *source, base::State *dest) 00050 { 00051 return getBestControl(control, source, dest, NULL); 00052 } 00053 00054 unsigned int ompl::control::SimpleDirectedControlSampler::sampleTo(Control *control, const Control *previous, const base::State *source, base::State *dest) 00055 { 00056 return getBestControl(control, source, dest, previous); 00057 } 00058 00059 unsigned int ompl::control::SimpleDirectedControlSampler::getBestControl (Control *control, const base::State *source, base::State *dest, const Control *previous) 00060 { 00061 // Sample the first control 00062 if (previous) 00063 cs_->sampleNext(control, previous, source); 00064 else 00065 cs_->sample(control, source); 00066 00067 const unsigned int minDuration = si_->getMinControlDuration(); 00068 const unsigned int maxDuration = si_->getMaxControlDuration(); 00069 00070 unsigned int steps = cs_->sampleStepCount(minDuration, maxDuration); 00071 // Propagate the first control, and find how far it is from the target state 00072 base::State *bestState = si_->allocState(); 00073 steps = si_->propagateWhileValid(source, control, steps, bestState); 00074 00075 if (numControlSamples_ > 1) 00076 { 00077 Control *tempControl = si_->allocControl(); 00078 base::State *tempState = si_->allocState(); 00079 double bestDistance = si_->distance(bestState, dest); 00080 00081 // Sample k-1 more controls, and save the control that gets closest to target 00082 for (unsigned int i = 1; i < numControlSamples_; ++i) 00083 { 00084 unsigned int sampleSteps = cs_->sampleStepCount(minDuration, maxDuration); 00085 if (previous) 00086 cs_->sampleNext(tempControl, previous, source); 00087 else 00088 cs_->sample(tempControl, source); 00089 00090 sampleSteps = si_->propagateWhileValid(source, tempControl, sampleSteps, tempState); 00091 double tempDistance = si_->distance(tempState, dest); 00092 if (tempDistance < bestDistance) 00093 { 00094 si_->copyState(bestState, tempState); 00095 si_->copyControl(control, tempControl); 00096 bestDistance = tempDistance; 00097 steps = sampleSteps; 00098 } 00099 } 00100 00101 si_->freeState(tempState); 00102 si_->freeControl(tempControl); 00103 } 00104 00105 si_->copyState(dest, bestState); 00106 si_->freeState(bestState); 00107 00108 return steps; 00109 }