ompl/control/spaces/DiscreteControlSpace.h
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 #ifndef OMPL_CONTROL_SPACES_DISCRETE_CONTROL_SPACE_ 00038 #define OMPL_CONTROL_SPACES_DISCRETE_CONTROL_SPACE_ 00039 00040 #include "ompl/control/ControlSpace.h" 00041 00042 namespace ompl 00043 { 00044 namespace control 00045 { 00046 00048 class DiscreteControlSampler : public ControlSampler 00049 { 00050 public: 00051 00053 DiscreteControlSampler(const ControlSpace *space) : ControlSampler(space) 00054 { 00055 } 00056 00057 virtual void sample(Control *control); 00058 }; 00059 00064 class DiscreteControlSpace : public ControlSpace 00065 { 00066 public: 00067 00069 class ControlType : public Control 00070 { 00071 public: 00072 00074 int value; 00075 }; 00076 00078 DiscreteControlSpace(const base::StateSpacePtr &stateSpace, int lowerBound, int upperBound) : 00079 ControlSpace(stateSpace), lowerBound_(lowerBound), upperBound_(upperBound) 00080 { 00081 setName("Discrete" + getName()); 00082 type_ = CONTROL_SPACE_DISCRETE; 00083 } 00084 00085 virtual ~DiscreteControlSpace() 00086 { 00087 } 00088 00089 virtual unsigned int getDimension() const; 00090 00091 virtual void copyControl(Control *destination, const Control *source) const; 00092 00093 virtual bool equalControls(const Control *control1, const Control *control2) const; 00094 00095 virtual ControlSamplerPtr allocDefaultControlSampler() const; 00096 00097 virtual Control* allocControl() const; 00098 00099 virtual void freeControl(Control *control) const; 00100 00102 virtual void nullControl(Control *control) const; 00103 00104 virtual void printControl(const Control *control, std::ostream &out) const; 00105 00106 virtual void printSettings(std::ostream &out) const; 00107 00109 unsigned int getControlCount() const 00110 { 00111 return upperBound_ - lowerBound_ + 1; 00112 } 00113 00115 int getLowerBound() const 00116 { 00117 return lowerBound_; 00118 } 00119 00121 int getUpperBound() const 00122 { 00123 return upperBound_; 00124 } 00125 00127 void setBounds(int lowerBound, int upperBound) 00128 { 00129 lowerBound_ = lowerBound; 00130 upperBound_ = upperBound; 00131 } 00132 00133 virtual void setup(); 00134 00136 virtual unsigned int getSerializationLength() const; 00137 00139 virtual void serialize(void *serialization, const Control *ctrl) const; 00140 00142 virtual void deserialize(Control *ctrl, const void *serialization) const; 00143 00144 protected: 00145 00147 int lowerBound_; 00148 00150 int upperBound_; 00151 }; 00152 00153 } 00154 } 00155 00156 #endif