| [28c025] | 1 | /* | 
|---|
|  | 2 | * Project: MoleCuilder | 
|---|
|  | 3 | * Description: creates and alters molecular systems | 
|---|
|  | 4 | * Copyright (C)  2012 University of Bonn. All rights reserved. | 
|---|
| [5aaa43] | 5 | * Copyright (C)  2013 Frederik Heber. All rights reserved. | 
|---|
| [28c025] | 6 | * | 
|---|
|  | 7 | * | 
|---|
|  | 8 | *   This file is part of MoleCuilder. | 
|---|
|  | 9 | * | 
|---|
|  | 10 | *    MoleCuilder is free software: you can redistribute it and/or modify | 
|---|
|  | 11 | *    it under the terms of the GNU General Public License as published by | 
|---|
|  | 12 | *    the Free Software Foundation, either version 2 of the License, or | 
|---|
|  | 13 | *    (at your option) any later version. | 
|---|
|  | 14 | * | 
|---|
|  | 15 | *    MoleCuilder is distributed in the hope that it will be useful, | 
|---|
|  | 16 | *    but WITHOUT ANY WARRANTY; without even the implied warranty of | 
|---|
|  | 17 | *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the | 
|---|
|  | 18 | *    GNU General Public License for more details. | 
|---|
|  | 19 | * | 
|---|
|  | 20 | *    You should have received a copy of the GNU General Public License | 
|---|
|  | 21 | *    along with MoleCuilder.  If not, see <http://www.gnu.org/licenses/>. | 
|---|
|  | 22 | */ | 
|---|
|  | 23 |  | 
|---|
|  | 24 | /* | 
|---|
|  | 25 | * SamplingGrid.cpp | 
|---|
|  | 26 | * | 
|---|
|  | 27 | *  Created on: 25.07.2012 | 
|---|
|  | 28 | *      Author: heber | 
|---|
|  | 29 | */ | 
|---|
|  | 30 |  | 
|---|
|  | 31 | // include config.h | 
|---|
|  | 32 | #ifdef HAVE_CONFIG_H | 
|---|
|  | 33 | #include <config.h> | 
|---|
|  | 34 | #endif | 
|---|
|  | 35 |  | 
|---|
|  | 36 | // include headers that implement a archive in simple text format | 
|---|
|  | 37 | // otherwise BOOST_CLASS_EXPORT_IMPLEMENT has no effect | 
|---|
|  | 38 | #include <boost/archive/text_oarchive.hpp> | 
|---|
|  | 39 | #include <boost/archive/text_iarchive.hpp> | 
|---|
|  | 40 |  | 
|---|
| [9eb71b3] | 41 | //#include "CodePatterns/MemDebug.hpp" | 
|---|
| [28c025] | 42 |  | 
|---|
| [fbf143] | 43 | #include "Fragmentation/Summation/SetValues/SamplingGrid.hpp" | 
|---|
| [28c025] | 44 |  | 
|---|
| [06653a] | 45 | #include <boost/assign.hpp> | 
|---|
| [fb3485] | 46 | #include <boost/bind.hpp> | 
|---|
| [57fda0] | 47 | #include <boost/shared_ptr.hpp> | 
|---|
|  | 48 |  | 
|---|
| [8eafd6] | 49 | #include <algorithm> | 
|---|
| [98f8fe] | 50 | #include <limits> | 
|---|
|  | 51 |  | 
|---|
| [c889b7] | 52 | #include "CodePatterns/Assert.hpp" | 
|---|
| [cb3363] | 53 | #include "CodePatterns/Log.hpp" | 
|---|
| [28c025] | 54 |  | 
|---|
| [6a5eb2] | 55 | #include "LinearAlgebra/Vector.hpp" | 
|---|
|  | 56 |  | 
|---|
| [1a00bb] | 57 | // static instances | 
|---|
| [5b1e5e] | 58 | const double SamplingGrid::zeroOffset[NDIM] = { 0., 0., 0. }; | 
|---|
| [1a00bb] | 59 |  | 
|---|
|  | 60 | SamplingGrid::SamplingGrid() : | 
|---|
|  | 61 | SamplingGridProperties() | 
|---|
|  | 62 | { | 
|---|
|  | 63 | setWindowSize(zeroOffset, zeroOffset); | 
|---|
|  | 64 | ASSERT( getWindowGridPoints() == (size_t)0, | 
|---|
|  | 65 | "SamplingGrid::SamplingGrid() - incorrect number of samples given for the window."); | 
|---|
|  | 66 | } | 
|---|
|  | 67 |  | 
|---|
| [5b1e5e] | 68 | SamplingGrid::SamplingGrid(const double _begin[NDIM], | 
|---|
|  | 69 | const double _end[NDIM], | 
|---|
| [c6355f] | 70 | const int _level) : | 
|---|
|  | 71 | SamplingGridProperties(_begin, _end, _level) | 
|---|
|  | 72 | { | 
|---|
|  | 73 | setWindowSize(zeroOffset, zeroOffset); | 
|---|
|  | 74 | ASSERT( getWindowGridPoints() == (size_t)0, | 
|---|
|  | 75 | "SamplingGrid::SamplingGrid() - incorrect number of samples given for the window."); | 
|---|
|  | 76 | } | 
|---|
|  | 77 |  | 
|---|
| [5b1e5e] | 78 | SamplingGrid::SamplingGrid(const double _begin[NDIM], | 
|---|
|  | 79 | const double _end[NDIM], | 
|---|
| [28c025] | 80 | const int _level, | 
|---|
| [c889b7] | 81 | const sampledvalues_t &_sampled_grid) : | 
|---|
| [3d9a8d] | 82 | SamplingGridProperties(_begin, _end, _level), | 
|---|
| [28c025] | 83 | sampled_grid(_sampled_grid) | 
|---|
| [1a00bb] | 84 | { | 
|---|
|  | 85 | setWindowSize(_begin, _end); | 
|---|
|  | 86 | ASSERT( getWindowGridPoints() == (size_t)_sampled_grid.size(), | 
|---|
|  | 87 | "SamplingGrid::SamplingGrid() - incorrect number of samples given for the window."); | 
|---|
|  | 88 | } | 
|---|
| [28c025] | 89 |  | 
|---|
|  | 90 | SamplingGrid::SamplingGrid(const SamplingGrid &_grid) : | 
|---|
|  | 91 | SamplingGridProperties(_grid), | 
|---|
|  | 92 | sampled_grid(_grid.sampled_grid) | 
|---|
| [1a00bb] | 93 | { | 
|---|
| [c6355f] | 94 | setWindowSize(_grid.begin_window, _grid.end_window); | 
|---|
|  | 95 | ASSERT( getWindowGridPoints() == _grid.getWindowGridPoints(), | 
|---|
|  | 96 | "SamplingGrid::SamplingGrid() - incorrect number of samples given for the window."); | 
|---|
| [1a00bb] | 97 | } | 
|---|
| [28c025] | 98 |  | 
|---|
|  | 99 | SamplingGrid::SamplingGrid(const SamplingGridProperties &_props) : | 
|---|
|  | 100 | SamplingGridProperties(_props) | 
|---|
| [1a00bb] | 101 | { | 
|---|
| [c6355f] | 102 | setWindowSize(zeroOffset, zeroOffset); | 
|---|
|  | 103 | ASSERT( getWindowGridPoints() == (size_t)0, | 
|---|
|  | 104 | "SamplingGrid::SamplingGrid() - incorrect number of samples given for the window."); | 
|---|
| [1a00bb] | 105 | } | 
|---|
| [28c025] | 106 |  | 
|---|
| [c889b7] | 107 | SamplingGrid::SamplingGrid( | 
|---|
|  | 108 | const SamplingGridProperties &_props, | 
|---|
|  | 109 | const sampledvalues_t &_sampled_grid) : | 
|---|
|  | 110 | SamplingGridProperties(_props), | 
|---|
|  | 111 | sampled_grid(_sampled_grid) | 
|---|
| [1a00bb] | 112 | { | 
|---|
|  | 113 | setWindowSize(_props.begin, _props.end); | 
|---|
|  | 114 | ASSERT( getWindowGridPoints() == (size_t)_sampled_grid.size(), | 
|---|
|  | 115 | "SamplingGrid::SamplingGrid() - incorrect number of samples given for the window."); | 
|---|
|  | 116 | } | 
|---|
| [c889b7] | 117 |  | 
|---|
| [28c025] | 118 | SamplingGrid::~SamplingGrid() | 
|---|
|  | 119 | {} | 
|---|
|  | 120 |  | 
|---|
| [c0e8fb] | 121 | bool SamplingGrid::isCongruent(const SamplingGrid &_props) const | 
|---|
|  | 122 | { | 
|---|
|  | 123 | bool status = true; | 
|---|
|  | 124 | status &= (static_cast<const SamplingGridProperties &>(*this) == | 
|---|
|  | 125 | static_cast<const SamplingGridProperties &>(_props)); | 
|---|
| [5b1e5e] | 126 | for(size_t i = 0; i<NDIM; ++i) { | 
|---|
| [c0e8fb] | 127 | status &= begin_window[i] == _props.begin_window[i]; | 
|---|
|  | 128 | status &= end_window[i] == _props.end_window[i]; | 
|---|
|  | 129 | } | 
|---|
|  | 130 | return status; | 
|---|
|  | 131 | } | 
|---|
|  | 132 |  | 
|---|
| [c889b7] | 133 | SamplingGrid& SamplingGrid::operator=(const SamplingGrid& other) | 
|---|
|  | 134 | { | 
|---|
|  | 135 | // check for self-assignment | 
|---|
|  | 136 | if (this != &other) { | 
|---|
|  | 137 | static_cast<SamplingGridProperties &>(*this) = other; | 
|---|
| [e2404f] | 138 | setWindowSize(other.begin_window, other.end_window); | 
|---|
| [c889b7] | 139 | sampled_grid = other.sampled_grid; | 
|---|
|  | 140 | } | 
|---|
|  | 141 | return *this; | 
|---|
|  | 142 | } | 
|---|
|  | 143 |  | 
|---|
| [313f83] | 144 | static void multiplyElements( | 
|---|
|  | 145 | double &dest, | 
|---|
|  | 146 | const double &source, | 
|---|
|  | 147 | const double prefactor) | 
|---|
|  | 148 | { | 
|---|
|  | 149 | dest *= prefactor*(source); | 
|---|
|  | 150 | } | 
|---|
|  | 151 |  | 
|---|
| [8eafd6] | 152 | SamplingGrid& SamplingGrid::operator*=(const double _value) | 
|---|
|  | 153 | { | 
|---|
|  | 154 | std::transform( | 
|---|
|  | 155 | sampled_grid.begin(), sampled_grid.end(), | 
|---|
|  | 156 | sampled_grid.begin(), | 
|---|
|  | 157 | boost::bind(std::multiplies<double>(), _1, _value)); | 
|---|
|  | 158 |  | 
|---|
|  | 159 | return *this; | 
|---|
|  | 160 | } | 
|---|
| [313f83] | 161 |  | 
|---|
| [57fda0] | 162 | static void getDownsampledGrid( | 
|---|
|  | 163 | const SamplingGrid &_reference_grid, | 
|---|
|  | 164 | const SamplingGrid &_grid, | 
|---|
|  | 165 | boost::shared_ptr<SamplingGrid> &_weight_downsampled) | 
|---|
|  | 166 | { | 
|---|
|  | 167 | static const double round_offset( | 
|---|
|  | 168 | (std::numeric_limits<size_t>::round_style == std::round_toward_zero) ? | 
|---|
|  | 169 | 0.5 : 0.); // need offset to get to round_toward_nearest behavior | 
|---|
|  | 170 | const int surplus_level = _reference_grid.getSurplusLevel(_grid)+round_offset; | 
|---|
|  | 171 | // need to downsample first | 
|---|
|  | 172 | _weight_downsampled.reset(new SamplingGrid); // may use default cstor as downsample does all settings | 
|---|
|  | 173 | SamplingGrid::downsample(*_weight_downsampled, _grid, _reference_grid.level-surplus_level); | 
|---|
|  | 174 | } | 
|---|
|  | 175 |  | 
|---|
| [a1fcc6] | 176 | SamplingGrid& SamplingGrid::operator*=(const SamplingGrid& other) | 
|---|
|  | 177 | { | 
|---|
|  | 178 | // check that grids are compatible | 
|---|
| [57fda0] | 179 | ASSERT(isCompatible(other), | 
|---|
|  | 180 | "SamplingGrid::operator*=() - multiplying incomatible grids is so far not in the cards."); | 
|---|
|  | 181 | const SamplingGrid *other_grid = &other; | 
|---|
|  | 182 | boost::shared_ptr<SamplingGrid> other_downsampled; | 
|---|
|  | 183 | if (!isEquivalent(other)) { | 
|---|
|  | 184 | getDownsampledGrid(*this, other, other_downsampled); | 
|---|
|  | 185 | other_grid = other_downsampled.get(); | 
|---|
|  | 186 | } | 
|---|
|  | 187 | /// get minimum of window | 
|---|
|  | 188 | double min_begin_window[NDIM]; | 
|---|
|  | 189 | double min_end_window[NDIM]; | 
|---|
|  | 190 | bool doShrink = false; | 
|---|
|  | 191 | for (size_t index=0; index<NDIM;++index) { | 
|---|
|  | 192 | if (begin_window[index] <= other.begin_window[index]) { | 
|---|
|  | 193 | min_begin_window[index] = other.begin_window[index]; | 
|---|
|  | 194 | doShrink = true; | 
|---|
|  | 195 | } else { | 
|---|
|  | 196 | min_begin_window[index] = begin_window[index]; | 
|---|
|  | 197 | } | 
|---|
|  | 198 | if (end_window[index] <= other.end_window[index]) { | 
|---|
|  | 199 | min_end_window[index] = end_window[index]; | 
|---|
|  | 200 | } else { | 
|---|
|  | 201 | min_end_window[index] = other.end_window[index]; | 
|---|
|  | 202 | doShrink = true; | 
|---|
| [313f83] | 203 | } | 
|---|
| [a1fcc6] | 204 | } | 
|---|
| [57fda0] | 205 | LOG(4, "DEBUG: min begin is " << min_begin_window[0] << "," << min_begin_window[1] << "," << min_begin_window[2] << "."); | 
|---|
|  | 206 | LOG(4, "DEBUG: min end is " << min_end_window[0] << "," << min_end_window[1] << "," << min_end_window[2] << "."); | 
|---|
|  | 207 | if (doShrink) | 
|---|
|  | 208 | shrinkWindow(min_begin_window, min_end_window); | 
|---|
|  | 209 | addWindowOntoWindow( | 
|---|
|  | 210 | other_grid->begin_window, | 
|---|
|  | 211 | other_grid->end_window, | 
|---|
|  | 212 | begin_window, | 
|---|
|  | 213 | end_window, | 
|---|
|  | 214 | sampled_grid, | 
|---|
|  | 215 | other_grid->sampled_grid, | 
|---|
|  | 216 | boost::bind(multiplyElements, _1, _2, 1.), | 
|---|
|  | 217 | sourcewindow); | 
|---|
|  | 218 |  | 
|---|
| [a1fcc6] | 219 | return *this; | 
|---|
|  | 220 | } | 
|---|
|  | 221 |  | 
|---|
| [c889b7] | 222 | void SamplingGrid::superposeOtherGrids(const SamplingGrid &other, const double prefactor) | 
|---|
|  | 223 | { | 
|---|
| [57fda0] | 224 | // check that grids are compatible | 
|---|
|  | 225 | ASSERT(isCompatible(other), | 
|---|
|  | 226 | "SamplingGrid::superposeOtherGrids() - superposing incompatible grids is so far not in the cards."); | 
|---|
|  | 227 | const SamplingGrid *other_grid = &other; | 
|---|
|  | 228 | boost::shared_ptr<SamplingGrid> other_downsampled; | 
|---|
|  | 229 | if (!isEquivalent(other)) { | 
|---|
|  | 230 | getDownsampledGrid(*this, other, other_downsampled); | 
|---|
|  | 231 | other_grid = other_downsampled.get(); | 
|---|
|  | 232 | } | 
|---|
|  | 233 | /// get maximum of window | 
|---|
|  | 234 | double max_begin_window[NDIM]; | 
|---|
|  | 235 | double max_end_window[NDIM]; | 
|---|
|  | 236 | bool doExtend = false; | 
|---|
|  | 237 | for (size_t index=0; index<NDIM;++index) { | 
|---|
|  | 238 | if (begin_window[index] >= other.begin_window[index]) { | 
|---|
|  | 239 | max_begin_window[index] = other.begin_window[index]; | 
|---|
|  | 240 | doExtend = true; | 
|---|
|  | 241 | } else { | 
|---|
|  | 242 | max_begin_window[index] = begin_window[index]; | 
|---|
|  | 243 | } | 
|---|
|  | 244 | if (end_window[index] >= other.end_window[index]) { | 
|---|
|  | 245 | max_end_window[index] = end_window[index]; | 
|---|
|  | 246 | } else { | 
|---|
|  | 247 | max_end_window[index] = other.end_window[index]; | 
|---|
|  | 248 | doExtend = true; | 
|---|
| [1a00bb] | 249 | } | 
|---|
| [c889b7] | 250 | } | 
|---|
| [57fda0] | 251 | LOG(4, "DEBUG: max begin is " << max_begin_window[0] << "," << max_begin_window[1] << "," << max_begin_window[2] << "."); | 
|---|
|  | 252 | LOG(4, "DEBUG: max end is " << max_end_window[0] << "," << max_end_window[1] << "," << max_end_window[2] << "."); | 
|---|
|  | 253 | if (doExtend) | 
|---|
|  | 254 | extendWindow(max_begin_window, max_end_window); | 
|---|
|  | 255 | /// and copy other into larger window, too | 
|---|
|  | 256 | addOntoWindow(other_grid->begin_window, other_grid->end_window, other_grid->sampled_grid, prefactor); | 
|---|
| [c889b7] | 257 | } | 
|---|
|  | 258 |  | 
|---|
| [620517] | 259 | const size_t SamplingGrid::getWindowGridPointsPerAxis(const size_t axis) const | 
|---|
| [3d9a8d] | 260 | { | 
|---|
| [620517] | 261 | static const double round_offset( | 
|---|
|  | 262 | (std::numeric_limits<size_t>::round_style == std::round_toward_zero) ? | 
|---|
|  | 263 | 0.5 : 0.); // need offset to get to round_toward_nearest behavior | 
|---|
|  | 264 | //  const double total = getTotalLengthPerAxis(axis); | 
|---|
|  | 265 | const double delta = getDeltaPerAxis(axis); | 
|---|
|  | 266 | if (delta == 0) | 
|---|
|  | 267 | return 0; | 
|---|
|  | 268 | const double length = getWindowLengthPerAxis(axis); | 
|---|
|  | 269 | if (length == 0) | 
|---|
| [1a00bb] | 270 | return 0; | 
|---|
| [620517] | 271 | return (size_t)(length/delta+round_offset); | 
|---|
| [1a00bb] | 272 | } | 
|---|
|  | 273 |  | 
|---|
| [cb3363] | 274 | double SamplingGrid::integral() const | 
|---|
|  | 275 | { | 
|---|
| [98f8fe] | 276 | const double volume_element = getVolume()/(double)getTotalGridPoints(); | 
|---|
| [cb3363] | 277 | double int_value = 0.; | 
|---|
|  | 278 | for (sampledvalues_t::const_iterator iter = sampled_grid.begin(); | 
|---|
|  | 279 | iter != sampled_grid.end(); | 
|---|
|  | 280 | ++iter) | 
|---|
|  | 281 | int_value += *iter; | 
|---|
|  | 282 | int_value *= volume_element; | 
|---|
|  | 283 | LOG(2, "DEBUG: SamplingGrid::integral() is " << scientific << setprecision(13) << int_value << "."); | 
|---|
|  | 284 | return int_value; | 
|---|
|  | 285 | } | 
|---|
|  | 286 |  | 
|---|
| [e72c61] | 287 | double SamplingGrid::integral(const SamplingGrid &weight) const | 
|---|
|  | 288 | { | 
|---|
| [57fda0] | 289 | // check that grids are compatible | 
|---|
|  | 290 | ASSERT(isCompatible(weight), | 
|---|
|  | 291 | "SamplingGrid::integral() - integrating with weights from incompatible grids is so far not in the cards."); | 
|---|
|  | 292 | const SamplingGrid *weight_grid = &weight; | 
|---|
|  | 293 | boost::shared_ptr<SamplingGrid> weight_downsampled; | 
|---|
|  | 294 | if (!isEquivalent(weight)) { | 
|---|
|  | 295 | getDownsampledGrid(*this, weight, weight_downsampled); | 
|---|
|  | 296 | weight_grid = weight_downsampled.get(); | 
|---|
|  | 297 | } | 
|---|
|  | 298 | const double volume_element = getVolume()/(double)getTotalGridPoints(); | 
|---|
|  | 299 | double int_value = 0.; | 
|---|
|  | 300 | sampledvalues_t::const_iterator iter = sampled_grid.begin(); | 
|---|
|  | 301 | sampledvalues_t::const_iterator weightiter = weight_grid->sampled_grid.begin(); | 
|---|
|  | 302 | for (;iter != sampled_grid.end();++iter,++weightiter) | 
|---|
|  | 303 | int_value += (*weightiter) * (*iter); | 
|---|
|  | 304 | int_value *= volume_element; | 
|---|
|  | 305 | //LOG(2, "DEBUG: SamplingGrid::integral() is " << scientific << setprecision(13) << int_value << "."); | 
|---|
|  | 306 | return int_value; | 
|---|
| [e72c61] | 307 | } | 
|---|
| [57fda0] | 308 |  | 
|---|
| [1a00bb] | 309 | void SamplingGrid::setWindowSize( | 
|---|
| [5b1e5e] | 310 | const double _begin_window[NDIM], | 
|---|
|  | 311 | const double _end_window[NDIM]) | 
|---|
| [1a00bb] | 312 | { | 
|---|
| [5b1e5e] | 313 | for (size_t index=0;index<NDIM;++index) { | 
|---|
| [64bafe0] | 314 | begin_window[index] = getNearestLowerGridPoint(_begin_window[index], index); | 
|---|
|  | 315 | ASSERT( begin_window[index] >= begin[index], | 
|---|
| [e2404f] | 316 | "SamplingGrid::setWindowSize() - window starts earlier than domain on " | 
|---|
|  | 317 | +toString(index)+"th component."); | 
|---|
| [64bafe0] | 318 | end_window[index] = getNearestHigherGridPoint(_end_window[index], index); | 
|---|
|  | 319 | ASSERT( end_window[index] <= end[index], | 
|---|
| [e2404f] | 320 | "SamplingGrid::setWindowSize() - window ends later than domain on " | 
|---|
|  | 321 | +toString(index)+"th component."); | 
|---|
| [1a00bb] | 322 | } | 
|---|
|  | 323 | } | 
|---|
|  | 324 |  | 
|---|
|  | 325 | void SamplingGrid::setWindow( | 
|---|
| [5b1e5e] | 326 | const double _begin_window[NDIM], | 
|---|
|  | 327 | const double _end_window[NDIM]) | 
|---|
| [1a00bb] | 328 | { | 
|---|
|  | 329 | setWindowSize(_begin_window, _end_window); | 
|---|
|  | 330 | const size_t gridpoints_window = getWindowGridPoints(); | 
|---|
| [c6355f] | 331 | sampled_grid.clear(); | 
|---|
| [1a00bb] | 332 | sampled_grid.resize(gridpoints_window, 0.); | 
|---|
|  | 333 | } | 
|---|
|  | 334 |  | 
|---|
| [e2404f] | 335 | void SamplingGrid::setDomain( | 
|---|
| [5b1e5e] | 336 | const double _begin[NDIM], | 
|---|
|  | 337 | const double _end[NDIM]) | 
|---|
| [e2404f] | 338 | { | 
|---|
|  | 339 | setDomainSize(_begin, _end); | 
|---|
|  | 340 | setWindowSize(_begin, _end); | 
|---|
|  | 341 | const size_t gridpoints = getTotalGridPoints(); | 
|---|
|  | 342 | sampled_grid.resize(gridpoints, 0.); | 
|---|
|  | 343 | } | 
|---|
|  | 344 |  | 
|---|
| [1a00bb] | 345 | void SamplingGrid::extendWindow( | 
|---|
| [5b1e5e] | 346 | const double _begin_window[NDIM], | 
|---|
|  | 347 | const double _end_window[NDIM]) | 
|---|
| [1a00bb] | 348 | { | 
|---|
|  | 349 | #ifndef NDEBUG | 
|---|
| [5b1e5e] | 350 | for(size_t index=0;index < NDIM; ++index) { | 
|---|
| [c6355f] | 351 | // check that we truly have to extend the window | 
|---|
| [1a00bb] | 352 | ASSERT ( begin_window[index] >= _begin_window[index], | 
|---|
|  | 353 | "SamplingGrid::extendWindow() - component "+toString(index)+ | 
|---|
|  | 354 | " of window start is greater than old value."); | 
|---|
|  | 355 | ASSERT ( end_window[index] <= _end_window[index], | 
|---|
|  | 356 | "SamplingGrid::extendWindow() - component "+toString(index)+ | 
|---|
|  | 357 | " of window end is less than old value."); | 
|---|
| [c6355f] | 358 |  | 
|---|
|  | 359 | // check that we are still less than domain | 
|---|
|  | 360 | ASSERT ( _begin_window[index] >= begin[index], | 
|---|
|  | 361 | "SamplingGrid::extendWindow() - component "+toString(index)+ | 
|---|
|  | 362 | " of window start is less than domain start."); | 
|---|
|  | 363 | ASSERT ( _end_window[index] <= end[index], | 
|---|
|  | 364 | "SamplingGrid::extendWindow() - component "+toString(index)+ | 
|---|
|  | 365 | " of window end is greater than domain end."); | 
|---|
| [1a00bb] | 366 | } | 
|---|
|  | 367 | #endif | 
|---|
|  | 368 | // copy old window size and values | 
|---|
| [5b1e5e] | 369 | double old_begin_window[NDIM]; | 
|---|
|  | 370 | double old_end_window[NDIM]; | 
|---|
|  | 371 | for(size_t index=0;index<NDIM;++index) { | 
|---|
| [1a00bb] | 372 | old_begin_window[index] = begin_window[index]; | 
|---|
|  | 373 | old_end_window[index] = end_window[index]; | 
|---|
|  | 374 | } | 
|---|
|  | 375 | sampledvalues_t old_values(sampled_grid); | 
|---|
| [de6dfb] | 376 | // set new window | 
|---|
|  | 377 | setWindow(_begin_window,_end_window); | 
|---|
| [1a00bb] | 378 | // now extend it ... | 
|---|
| [de6dfb] | 379 | addOntoWindow(old_begin_window, old_end_window, old_values, +1.); | 
|---|
| [10f0cb] | 380 | LOG(6, "DEBUG: Grid after extension is " << sampled_grid << "."); | 
|---|
| [1a00bb] | 381 | } | 
|---|
|  | 382 |  | 
|---|
| [313f83] | 383 | void SamplingGrid::shrinkWindow( | 
|---|
| [5b1e5e] | 384 | const double _begin_window[NDIM], | 
|---|
|  | 385 | const double _end_window[NDIM]) | 
|---|
| [313f83] | 386 | { | 
|---|
|  | 387 | #ifndef NDEBUG | 
|---|
| [5b1e5e] | 388 | for(size_t index=0;index < NDIM; ++index) { | 
|---|
| [313f83] | 389 | // check that we truly have to shrink the window | 
|---|
|  | 390 | ASSERT ( begin_window[index] <= _begin_window[index], | 
|---|
|  | 391 | "SamplingGrid::shrinkWindow() - component "+toString(index)+ | 
|---|
|  | 392 | " of window start is less than old value."); | 
|---|
|  | 393 | ASSERT ( end_window[index] >= _end_window[index], | 
|---|
|  | 394 | "SamplingGrid::shrinkWindow() - component "+toString(index)+ | 
|---|
|  | 395 | " of window end is greater than old value."); | 
|---|
|  | 396 |  | 
|---|
|  | 397 | // check that we are still less than domain | 
|---|
|  | 398 | ASSERT ( _begin_window[index] >= begin[index], | 
|---|
|  | 399 | "SamplingGrid::shrinkWindow() - component "+toString(index)+ | 
|---|
|  | 400 | " of window start is less than domain start."); | 
|---|
|  | 401 | ASSERT ( _end_window[index] <= end[index], | 
|---|
|  | 402 | "SamplingGrid::shrinkWindow() - component "+toString(index)+ | 
|---|
|  | 403 | " of window end is greater than domain end."); | 
|---|
|  | 404 | } | 
|---|
|  | 405 | #endif | 
|---|
|  | 406 | // copy old window size and values | 
|---|
| [5b1e5e] | 407 | double old_begin_window[NDIM]; | 
|---|
|  | 408 | double old_end_window[NDIM]; | 
|---|
|  | 409 | for(size_t index=0;index<NDIM;++index) { | 
|---|
| [313f83] | 410 | old_begin_window[index] = begin_window[index]; | 
|---|
|  | 411 | old_end_window[index] = end_window[index]; | 
|---|
|  | 412 | } | 
|---|
|  | 413 | sampledvalues_t old_values(sampled_grid); | 
|---|
|  | 414 | // set new window | 
|---|
|  | 415 | setWindow(_begin_window,_end_window); | 
|---|
|  | 416 | // now extend it ... | 
|---|
|  | 417 | addIntoWindow(old_begin_window, old_end_window, old_values, +1.); | 
|---|
| [10f0cb] | 418 | LOG(6, "DEBUG: Grid after extension is " << sampled_grid << "."); | 
|---|
| [313f83] | 419 | } | 
|---|
|  | 420 |  | 
|---|
| [fb3485] | 421 | static void addElements( | 
|---|
|  | 422 | double &dest, | 
|---|
|  | 423 | const double &source, | 
|---|
|  | 424 | const double prefactor) | 
|---|
|  | 425 | { | 
|---|
|  | 426 | dest += prefactor*(source); | 
|---|
|  | 427 | } | 
|---|
|  | 428 |  | 
|---|
| [1a00bb] | 429 | void SamplingGrid::addOntoWindow( | 
|---|
| [5b1e5e] | 430 | const double _begin_window[NDIM], | 
|---|
|  | 431 | const double _end_window[NDIM], | 
|---|
| [de6dfb] | 432 | const sampledvalues_t &_sampled_grid, | 
|---|
|  | 433 | const double prefactor) | 
|---|
| [fb3485] | 434 | { | 
|---|
|  | 435 | addWindowOntoWindow( | 
|---|
|  | 436 | begin_window, | 
|---|
|  | 437 | end_window, | 
|---|
|  | 438 | _begin_window, | 
|---|
|  | 439 | _end_window, | 
|---|
|  | 440 | sampled_grid, | 
|---|
|  | 441 | _sampled_grid, | 
|---|
|  | 442 | boost::bind(addElements, _1, _2, boost::cref(prefactor)), | 
|---|
| [313f83] | 443 | destwindow); | 
|---|
|  | 444 | } | 
|---|
|  | 445 |  | 
|---|
|  | 446 | void SamplingGrid::addIntoWindow( | 
|---|
| [5b1e5e] | 447 | const double _begin_window[NDIM], | 
|---|
|  | 448 | const double _end_window[NDIM], | 
|---|
| [313f83] | 449 | const sampledvalues_t &_sampled_grid, | 
|---|
|  | 450 | const double prefactor) | 
|---|
|  | 451 | { | 
|---|
|  | 452 | addWindowOntoWindow( | 
|---|
|  | 453 | _begin_window, | 
|---|
|  | 454 | _end_window, | 
|---|
|  | 455 | begin_window, | 
|---|
|  | 456 | end_window, | 
|---|
|  | 457 | sampled_grid, | 
|---|
|  | 458 | _sampled_grid, | 
|---|
|  | 459 | boost::bind(addElements, _1, _2, boost::cref(prefactor)), | 
|---|
|  | 460 | sourcewindow); | 
|---|
| [fb3485] | 461 | } | 
|---|
|  | 462 |  | 
|---|
| [c1948c] | 463 | void SamplingGrid::getDiscreteWindowIndices( | 
|---|
|  | 464 | size_t _wbegin[NDIM], | 
|---|
|  | 465 | size_t _wlength[NDIM], | 
|---|
|  | 466 | size_t _wend[NDIM]) const | 
|---|
|  | 467 | { | 
|---|
|  | 468 | const double round_offset = | 
|---|
|  | 469 | (std::numeric_limits<size_t>::round_style == std::round_toward_zero) ? | 
|---|
|  | 470 | 0.5 : 0.; // need offset to get to round_toward_nearest behavior | 
|---|
|  | 471 | for(size_t index=0;index<NDIM;++index) { | 
|---|
|  | 472 | if (fabs(end[index] - begin[index]) > std::numeric_limits<double>::epsilon()*1e4) { | 
|---|
|  | 473 | // we refrain from using floor/ceil as the window's starts and ends, | 
|---|
|  | 474 | // the grids have to be compatible (equal level), should always be on | 
|---|
|  | 475 | // discrete grid point locations. | 
|---|
|  | 476 | const double delta = getDeltaPerAxis(index); | 
|---|
|  | 477 | // delta is conversion factor from box length to discrete length, i.e. number of points | 
|---|
|  | 478 | _wbegin[index] = (begin_window[index] - begin[index])/delta+round_offset; | 
|---|
|  | 479 | _wlength[index] = (end_window[index] - begin_window[index])/delta+round_offset; | 
|---|
|  | 480 | _wend[index] = (end_window[index] - begin[index])/delta+round_offset; | 
|---|
|  | 481 | } else { | 
|---|
|  | 482 | _wbegin[index] = 0; | 
|---|
|  | 483 | _wlength[index] = 0; | 
|---|
|  | 484 | _wend[index] = 0; | 
|---|
|  | 485 | } | 
|---|
|  | 486 | // total is used as safe-guard against loss due to discrete conversion | 
|---|
|  | 487 | ASSERT( (_wend[index] - _wbegin[index]) == _wlength[index], | 
|---|
|  | 488 | "SamplingGrid::getDiscreteWindowCopyIndices() - end - begin is not equal to length for " | 
|---|
|  | 489 | +toString(index)+"th component."); | 
|---|
|  | 490 | } | 
|---|
|  | 491 | } | 
|---|
|  | 492 |  | 
|---|
| [336da8] | 493 | void SamplingGrid::getDiscreteWindowOffsets( | 
|---|
|  | 494 | size_t _pre_offset[NDIM], | 
|---|
|  | 495 | size_t _post_offset[NDIM], | 
|---|
|  | 496 | size_t _length[NDIM], | 
|---|
|  | 497 | size_t _total[NDIM]) const | 
|---|
|  | 498 | { | 
|---|
|  | 499 | const double round_offset = | 
|---|
|  | 500 | (std::numeric_limits<size_t>::round_style == std::round_toward_zero) ? | 
|---|
|  | 501 | 0.5 : 0.; // need offset to get to round_toward_nearest behavior | 
|---|
|  | 502 | for(size_t index=0;index<NDIM;++index) { | 
|---|
|  | 503 | if (fabs(end[index] - begin[index]) > std::numeric_limits<double>::epsilon()*1e4) { | 
|---|
|  | 504 | // we refrain from using floor/ceil as the window's starts and ends, | 
|---|
|  | 505 | // the grids have to be compatible (equal level), should always be on | 
|---|
|  | 506 | // discrete grid point locations. | 
|---|
|  | 507 | const double delta = getDeltaPerAxis(index); | 
|---|
|  | 508 | // delta is conversion factor from box length to discrete length, i.e. number of points | 
|---|
|  | 509 | _pre_offset[index] = (begin_window[index] - begin[index])/delta+round_offset; | 
|---|
|  | 510 | _post_offset[index] = (end[index] - end_window[index])/delta+round_offset; | 
|---|
|  | 511 | _length[index] = (end_window[index] - begin_window[index])/delta+round_offset; | 
|---|
|  | 512 | _total[index] = (end[index] - begin[index])/delta+round_offset; | 
|---|
|  | 513 | } else { | 
|---|
|  | 514 | _pre_offset[index] = 0; | 
|---|
|  | 515 | _post_offset[index] = 0; | 
|---|
|  | 516 | _length[index] = 0; | 
|---|
|  | 517 | _total[index] = 0; | 
|---|
|  | 518 | } | 
|---|
|  | 519 | // total is used as safe-guard against loss due to discrete conversion | 
|---|
|  | 520 | ASSERT( (_pre_offset[index] + _post_offset[index]) + _length[index] == _total[index], | 
|---|
|  | 521 | "SamplingGrid::getDiscreteWindowCopyIndices() - pre, length, post are not equal to total for " | 
|---|
|  | 522 | +toString(index)+"th component."); | 
|---|
|  | 523 | } | 
|---|
|  | 524 | } | 
|---|
|  | 525 |  | 
|---|
| [e51f2c] | 526 | void SamplingGrid::getDiscreteWindowCopyIndices( | 
|---|
| [3f64ee] | 527 | const double *larger_wbegin, | 
|---|
|  | 528 | const double *larger_wend, | 
|---|
|  | 529 | const double *smaller_wbegin, | 
|---|
|  | 530 | const double *smaller_wend, | 
|---|
|  | 531 | size_t *pre_offset, | 
|---|
|  | 532 | size_t *post_offset, | 
|---|
|  | 533 | size_t *length, | 
|---|
|  | 534 | size_t *total) const | 
|---|
|  | 535 | { | 
|---|
|  | 536 | const double round_offset = | 
|---|
|  | 537 | (std::numeric_limits<size_t>::round_style == std::round_toward_zero) ? | 
|---|
|  | 538 | 0.5 : 0.; // need offset to get to round_toward_nearest behavior | 
|---|
| [5b1e5e] | 539 | for(size_t index=0;index<NDIM;++index) { | 
|---|
| [3f64ee] | 540 | if (fabs(end[index] - begin[index]) > std::numeric_limits<double>::epsilon()*1e4) { | 
|---|
|  | 541 | // we refrain from using floor/ceil as the window's starts and ends, | 
|---|
|  | 542 | // the grids have to be compatible (equal level), should always be on | 
|---|
|  | 543 | // discrete grid point locations. | 
|---|
| [9782343] | 544 | const double delta = getDeltaPerAxis(index); | 
|---|
| [3f64ee] | 545 | // delta is conversion factor from box length to discrete length, i.e. number of points | 
|---|
| [9782343] | 546 | pre_offset[index] = (smaller_wbegin[index] - larger_wbegin[index])/delta+round_offset; | 
|---|
|  | 547 | length[index] = (smaller_wend[index] - smaller_wbegin[index])/delta+round_offset; | 
|---|
|  | 548 | post_offset[index] = (larger_wend[index] - smaller_wend[index])/delta+round_offset; | 
|---|
|  | 549 | total[index] = (larger_wend[index] - larger_wbegin[index])/delta+round_offset; | 
|---|
| [3f64ee] | 550 | } else { | 
|---|
|  | 551 | pre_offset[index] = 0; | 
|---|
|  | 552 | length[index] = 0; | 
|---|
|  | 553 | post_offset[index] = 0; | 
|---|
|  | 554 | total[index] = 0; | 
|---|
|  | 555 | } | 
|---|
|  | 556 | // total is used as safe-guard against loss due to discrete conversion | 
|---|
|  | 557 | ASSERT( pre_offset[index]+post_offset[index]+length[index] == total[index], | 
|---|
| [e51f2c] | 558 | "SamplingGrid::getDiscreteWindowCopyIndices() - pre, post, and length don't sum up to total for " | 
|---|
| [3f64ee] | 559 | +toString(index)+"th component."); | 
|---|
|  | 560 | } | 
|---|
|  | 561 | } | 
|---|
|  | 562 |  | 
|---|
| [fb3485] | 563 | void SamplingGrid::addWindowOntoWindow( | 
|---|
| [5b1e5e] | 564 | const double larger_wbegin[NDIM], | 
|---|
|  | 565 | const double larger_wend[NDIM], | 
|---|
|  | 566 | const double smaller_wbegin[NDIM], | 
|---|
|  | 567 | const double smaller_wend[NDIM], | 
|---|
| [313f83] | 568 | sampledvalues_t &dest_sampled_grid, | 
|---|
|  | 569 | const sampledvalues_t &source_sampled_grid, | 
|---|
| [fb3485] | 570 | boost::function<void (double &, const double &)> op, | 
|---|
| [313f83] | 571 | enum eLargerWindow larger_window) | 
|---|
| [1a00bb] | 572 | { | 
|---|
|  | 573 | #ifndef NDEBUG | 
|---|
| [5b1e5e] | 574 | for(size_t index=0;index<NDIM;++index) { | 
|---|
| [313f83] | 575 | ASSERT( smaller_wbegin[index] >= larger_wbegin[index], | 
|---|
| [fb3485] | 576 | "SamplingGrid::addWindowOntoWindow() - given smaller window starts earlier than larger window in component " | 
|---|
| [1a00bb] | 577 | +toString(index)+"."); | 
|---|
| [313f83] | 578 | ASSERT( smaller_wend[index] <= larger_wend[index], | 
|---|
| [fb3485] | 579 | "SamplingGrid::addWindowOntoWindow() - given smaller window ends later than larger window in component " | 
|---|
| [1a00bb] | 580 | +toString(index)+"."); | 
|---|
|  | 581 | } | 
|---|
|  | 582 | #endif | 
|---|
|  | 583 | // the only issue are indices | 
|---|
| [5b1e5e] | 584 | size_t pre_offset[NDIM]; | 
|---|
|  | 585 | size_t post_offset[NDIM]; | 
|---|
|  | 586 | size_t length[NDIM]; | 
|---|
|  | 587 | size_t total[NDIM]; | 
|---|
| [e51f2c] | 588 | getDiscreteWindowCopyIndices( | 
|---|
| [3f64ee] | 589 | larger_wbegin, larger_wend, | 
|---|
|  | 590 | smaller_wbegin, smaller_wend, | 
|---|
|  | 591 | pre_offset, | 
|---|
|  | 592 | post_offset, | 
|---|
|  | 593 | length, | 
|---|
|  | 594 | total | 
|---|
|  | 595 | ); | 
|---|
| [64bafe0] | 596 | // assert that calculated lengths match with given vector sizes | 
|---|
| [c6355f] | 597 | #ifndef NDEBUG | 
|---|
|  | 598 | const size_t calculated_size = length[0]*length[1]*length[2]; | 
|---|
| [313f83] | 599 | if (larger_window == destwindow) { | 
|---|
|  | 600 | ASSERT( calculated_size == source_sampled_grid.size(), | 
|---|
|  | 601 | "SamplingGrid::addWindowOntoWindow() - not enough source sampled values given: " | 
|---|
|  | 602 | +toString(calculated_size)+" != "+toString(source_sampled_grid.size())+"."); | 
|---|
|  | 603 | ASSERT( calculated_size <= dest_sampled_grid.size(), | 
|---|
|  | 604 | "SamplingGrid::addWindowOntoWindow() - not enough sampled values available: " | 
|---|
|  | 605 | +toString(calculated_size)+" <= "+toString(dest_sampled_grid.size())+"."); | 
|---|
|  | 606 | } else { | 
|---|
|  | 607 | ASSERT( calculated_size == dest_sampled_grid.size(), | 
|---|
|  | 608 | "SamplingGrid::addWindowOntoWindow() - not enough dest sampled values given: " | 
|---|
|  | 609 | +toString(calculated_size)+" != "+toString(dest_sampled_grid.size())+"."); | 
|---|
|  | 610 | ASSERT( calculated_size <= source_sampled_grid.size(), | 
|---|
|  | 611 | "SamplingGrid::addWindowOntoWindow() - not enough source sampled values available: " | 
|---|
|  | 612 | +toString(calculated_size)+" <= "+toString(source_sampled_grid.size())+"."); | 
|---|
|  | 613 | } | 
|---|
| [c6355f] | 614 | const size_t total_size = total[0]*total[1]*total[2]; | 
|---|
| [313f83] | 615 | if (larger_window == destwindow) { | 
|---|
|  | 616 | ASSERT( total_size == dest_sampled_grid.size(), | 
|---|
|  | 617 | "SamplingGrid::addWindowOntoWindow() - total size is not equal to number of present dest points: " | 
|---|
|  | 618 | +toString(total_size)+" != "+toString(dest_sampled_grid.size())+"."); | 
|---|
|  | 619 | } else { | 
|---|
|  | 620 | ASSERT( total_size == source_sampled_grid.size(), | 
|---|
|  | 621 | "SamplingGrid::addWindowOntoWindow() - total size is not equal to number of present source points: " | 
|---|
|  | 622 | +toString(total_size)+" != "+toString(source_sampled_grid.size())+"."); | 
|---|
|  | 623 | } | 
|---|
| [c6355f] | 624 | #endif | 
|---|
| [5b1e5e] | 625 | size_t N[NDIM]; | 
|---|
| [64bafe0] | 626 | //  size_t counter = 0; | 
|---|
| [313f83] | 627 | sampledvalues_t::iterator destiter = dest_sampled_grid.begin(); | 
|---|
|  | 628 | sampledvalues_t::const_iterator sourceiter = source_sampled_grid.begin(); | 
|---|
|  | 629 | if (larger_window == destwindow) | 
|---|
|  | 630 | std::advance(destiter, pre_offset[0]*total[1]*total[2]); | 
|---|
| [fb3485] | 631 | else | 
|---|
| [313f83] | 632 | std::advance(sourceiter, pre_offset[0]*total[1]*total[2]); | 
|---|
| [de6dfb] | 633 | for(N[0]=0; N[0] < length[0]; ++N[0]) { | 
|---|
| [313f83] | 634 | if (larger_window == destwindow) | 
|---|
|  | 635 | std::advance(destiter, pre_offset[1]*total[2]); | 
|---|
| [fb3485] | 636 | else | 
|---|
| [313f83] | 637 | std::advance(sourceiter, pre_offset[1]*total[2]); | 
|---|
| [de6dfb] | 638 | for(N[1]=0; N[1] < length[1]; ++N[1]) { | 
|---|
| [313f83] | 639 | if (larger_window == destwindow) | 
|---|
|  | 640 | std::advance(destiter, pre_offset[2]); | 
|---|
| [fb3485] | 641 | else | 
|---|
| [313f83] | 642 | std::advance(sourceiter, pre_offset[2]); | 
|---|
| [de6dfb] | 643 | for(N[2]=0; N[2] < length[2]; ++N[2]) { | 
|---|
| [313f83] | 644 | ASSERT( destiter != dest_sampled_grid.end(), | 
|---|
|  | 645 | "SamplingGrid::addWindowOntoWindow() - destiter is already at end of window."); | 
|---|
|  | 646 | ASSERT( sourceiter != source_sampled_grid.end(), | 
|---|
| [06653a] | 647 | "SamplingGrid::addWindowOntoWindow() - sourceiter is already at end of window."); | 
|---|
| [313f83] | 648 | op(*destiter, *sourceiter); | 
|---|
|  | 649 | ++destiter; | 
|---|
|  | 650 | ++sourceiter; | 
|---|
| [1a00bb] | 651 | } | 
|---|
| [313f83] | 652 | if (larger_window == destwindow) | 
|---|
|  | 653 | std::advance(destiter, post_offset[2]); | 
|---|
| [fb3485] | 654 | else | 
|---|
| [313f83] | 655 | std::advance(sourceiter, post_offset[2]); | 
|---|
| [1a00bb] | 656 | } | 
|---|
| [313f83] | 657 | if (larger_window == destwindow) | 
|---|
|  | 658 | std::advance(destiter, post_offset[1]*total[2]); | 
|---|
| [fb3485] | 659 | else | 
|---|
| [313f83] | 660 | std::advance(sourceiter, post_offset[1]*total[2]); | 
|---|
| [1a00bb] | 661 | } | 
|---|
| [c6355f] | 662 | #ifndef NDEBUG | 
|---|
| [313f83] | 663 | if (larger_window == destwindow) | 
|---|
|  | 664 | std::advance(destiter, post_offset[0]*total[1]*total[2]); | 
|---|
| [fb3485] | 665 | else | 
|---|
| [313f83] | 666 | std::advance(sourceiter, post_offset[0]*total[1]*total[2]); | 
|---|
|  | 667 | ASSERT( destiter == dest_sampled_grid.end(), | 
|---|
|  | 668 | "SamplingGrid::addWindowOntoWindow() - destiter is not at end of window."); | 
|---|
|  | 669 | ASSERT( sourceiter == source_sampled_grid.end(), | 
|---|
|  | 670 | "SamplingGrid::addWindowOntoWindow() - sourceiter is not at end of window."); | 
|---|
| [c6355f] | 671 | #endif | 
|---|
| [10f0cb] | 672 | LOG(8, "DEBUG: Grid after adding other is " << dest_sampled_grid << "."); | 
|---|
| [1a00bb] | 673 | } | 
|---|
|  | 674 |  | 
|---|
| [955051] | 675 |  | 
|---|
|  | 676 | bool SamplingGrid::operator==(const SamplingGrid &other) const | 
|---|
|  | 677 | { | 
|---|
|  | 678 | bool status = | 
|---|
|  | 679 | static_cast<const SamplingGridProperties &>(*this) | 
|---|
|  | 680 | == static_cast<const SamplingGridProperties &>(other); | 
|---|
|  | 681 | // compare general properties | 
|---|
|  | 682 | if (status) { | 
|---|
|  | 683 | // compare windows | 
|---|
| [5b1e5e] | 684 | for (size_t i=0; i<NDIM; ++i) { | 
|---|
| [955051] | 685 | status &= begin_window[i] == other.begin_window[i]; | 
|---|
|  | 686 | status &= end_window[i] == other.end_window[i]; | 
|---|
|  | 687 | } | 
|---|
|  | 688 | // compare grids | 
|---|
|  | 689 | if (status) | 
|---|
|  | 690 | status &= sampled_grid == other.sampled_grid; | 
|---|
|  | 691 | } | 
|---|
|  | 692 | return status; | 
|---|
|  | 693 | } | 
|---|
|  | 694 |  | 
|---|
| [06653a] | 695 | /** Struct contains a single point with displacements from the | 
|---|
|  | 696 | * central point and the weight in the restriction. | 
|---|
|  | 697 | */ | 
|---|
|  | 698 | struct PointWeight_t { | 
|---|
|  | 699 | PointWeight_t(const int &d1, const int &d2, const int &d3, const double &_weight) : | 
|---|
|  | 700 | displacement(NDIM), | 
|---|
|  | 701 | weight(_weight) | 
|---|
|  | 702 | { | 
|---|
|  | 703 | displacement[0] = d1; displacement[1] = d2; displacement[2] = d3; | 
|---|
|  | 704 | } | 
|---|
|  | 705 | typedef std::vector<int>  displacement_t; | 
|---|
|  | 706 | displacement_t displacement; | 
|---|
|  | 707 | double weight; | 
|---|
|  | 708 | }; | 
|---|
|  | 709 |  | 
|---|
| [91e7658] | 710 | static void getLengthsOfWindow( | 
|---|
| [06653a] | 711 | int _total[NDIM], | 
|---|
|  | 712 | const SamplingGrid &_grid) | 
|---|
|  | 713 | { | 
|---|
|  | 714 | const size_t gridpoints_axis = _grid.getGridPointsPerAxis(); | 
|---|
|  | 715 | static const double round_offset = | 
|---|
|  | 716 | (std::numeric_limits<size_t>::round_style == std::round_toward_zero) ? | 
|---|
|  | 717 | 0.5 : 0.; // need offset to get to round_toward_nearest behavior | 
|---|
|  | 718 | for (size_t index=0; index<NDIM; ++index) { | 
|---|
|  | 719 | if (fabs(_grid.end[index] - _grid.begin[index]) > std::numeric_limits<double>::epsilon()*1e4) { | 
|---|
|  | 720 | const double delta = (double)gridpoints_axis/(_grid.end[index] - _grid.begin[index]); | 
|---|
|  | 721 | _total[index] = delta*(_grid.end_window[index] - _grid.begin_window[index])+round_offset; | 
|---|
|  | 722 | } else | 
|---|
|  | 723 | _total[index] = 0; | 
|---|
| [91e7658] | 724 | // we can only assert that its atmost the maximum number of grid points | 
|---|
|  | 725 | ASSERT (_total[index] <= ::pow(2, _grid.level), | 
|---|
| [06653a] | 726 | "SamplingGrid::downsample() - total "+toString(_total[index]) | 
|---|
| [91e7658] | 727 | +" is not equal or less than 2^level: "+toString(_grid.level)); | 
|---|
| [06653a] | 728 | } | 
|---|
|  | 729 | } | 
|---|
|  | 730 |  | 
|---|
|  | 731 | //!> stencil for full weight restriction, see vmg's stencils.hpp | 
|---|
|  | 732 | static const std::vector< PointWeight_t > FullWeightNearestNeighbor = | 
|---|
|  | 733 | boost::assign::list_of | 
|---|
|  | 734 | ( PointWeight_t( 0,  0,  0, 0.125) ) | 
|---|
|  | 735 | ( PointWeight_t( 1,  0,  0, 0.0625) ) | 
|---|
|  | 736 | ( PointWeight_t(-1,  0,  0, 0.0625) ) | 
|---|
|  | 737 | ( PointWeight_t( 0,  1,  0, 0.0625) ) | 
|---|
|  | 738 | ( PointWeight_t( 0, -1,  0, 0.0625) ) | 
|---|
|  | 739 | ( PointWeight_t( 0,  0,  1, 0.0625) ) | 
|---|
|  | 740 | ( PointWeight_t( 0,  0, -1, 0.0625) ) | 
|---|
|  | 741 | ( PointWeight_t( 1,  1,  0, 0.03125) ) | 
|---|
|  | 742 | ( PointWeight_t( 1, -1,  0, 0.03125) ) | 
|---|
|  | 743 | ( PointWeight_t(-1,  1,  0, 0.03125) ) | 
|---|
|  | 744 | ( PointWeight_t(-1, -1,  0, 0.03125) ) | 
|---|
|  | 745 | ( PointWeight_t( 0,  1,  1, 0.03125) ) | 
|---|
|  | 746 | ( PointWeight_t( 0,  1, -1, 0.03125) ) | 
|---|
|  | 747 | ( PointWeight_t( 0, -1,  1, 0.03125) ) | 
|---|
|  | 748 | ( PointWeight_t( 0, -1, -1, 0.03125) ) | 
|---|
|  | 749 | ( PointWeight_t( 1,  0,  1, 0.03125) ) | 
|---|
|  | 750 | ( PointWeight_t( 1,  0, -1, 0.03125) ) | 
|---|
|  | 751 | ( PointWeight_t(-1,  0,  1, 0.03125) ) | 
|---|
|  | 752 | ( PointWeight_t(-1,  0, -1, 0.03125) ) | 
|---|
|  | 753 | ( PointWeight_t( 1,  1,  1, 0.015625) ) | 
|---|
|  | 754 | ( PointWeight_t( 1,  1, -1, 0.015625) ) | 
|---|
|  | 755 | ( PointWeight_t( 1, -1,  1, 0.015625) ) | 
|---|
|  | 756 | ( PointWeight_t(-1,  1,  1, 0.015625) ) | 
|---|
|  | 757 | ( PointWeight_t( 1, -1, -1, 0.015625) ) | 
|---|
|  | 758 | ( PointWeight_t(-1,  1, -1, 0.015625) ) | 
|---|
|  | 759 | ( PointWeight_t(-1, -1,  1, 0.015625) ) | 
|---|
|  | 760 | ( PointWeight_t(-1, -1, -1, 0.015625) ) | 
|---|
|  | 761 | ; | 
|---|
|  | 762 |  | 
|---|
|  | 763 | int getValidIndex( | 
|---|
|  | 764 | const PointWeight_t::displacement_t &_disp, | 
|---|
|  | 765 | const int N[NDIM], | 
|---|
|  | 766 | const int length[NDIM]) | 
|---|
|  | 767 | { | 
|---|
|  | 768 | int index = 0; | 
|---|
|  | 769 | // we simply truncate in case of out of bounds access | 
|---|
|  | 770 | if ((N[2]+_disp[2] >= 0) && (N[2]+_disp[2] < length[2])) | 
|---|
|  | 771 | index += _disp[2]; | 
|---|
|  | 772 | if ((N[1]+_disp[1] >= 0) && (N[1]+_disp[1] < length[1])) | 
|---|
|  | 773 | index += _disp[1]*length[2]; | 
|---|
|  | 774 | if ((N[0]+_disp[0] >= 0) && (N[0]+_disp[0] < length[0])) | 
|---|
|  | 775 | index += _disp[0]*length[1]*length[2]; | 
|---|
|  | 776 | return index; | 
|---|
|  | 777 | } | 
|---|
|  | 778 |  | 
|---|
|  | 779 | void restrictFullWeight( | 
|---|
|  | 780 | SamplingGrid::sampledvalues_t &_coarse_level, | 
|---|
|  | 781 | const int length_c[NDIM], | 
|---|
|  | 782 | const SamplingGrid::sampledvalues_t &_fine_level, | 
|---|
|  | 783 | const int length_f[NDIM]) | 
|---|
|  | 784 | { | 
|---|
|  | 785 | int N_c[NDIM]; | 
|---|
|  | 786 | int N_f[NDIM]; | 
|---|
|  | 787 | SamplingGrid::sampledvalues_t::iterator coarseiter = _coarse_level.begin(); | 
|---|
|  | 788 | for(N_c[0]=0, N_f[0]=0; (N_c[0] < length_c[0]) && (N_f[0] < length_f[0]); ++N_c[0], N_f[0] +=2) { | 
|---|
|  | 789 | for(N_c[1]=0, N_f[1]=0; (N_c[1] < length_c[1]) && (N_f[1] < length_f[1]); ++N_c[1], N_f[1] +=2) { | 
|---|
|  | 790 | for(N_c[2]=0, N_f[2]=0; (N_c[2] < length_c[2]) && (N_f[2] < length_f[2]); ++N_c[2], N_f[2] +=2) { | 
|---|
|  | 791 | const int index_base = N_f[2] + (N_f[1] + N_f[0]*length_f[1])*length_f[2]; | 
|---|
|  | 792 | // go through stencil and add each point relative to displacement with weight | 
|---|
|  | 793 | for (std::vector< PointWeight_t >::const_iterator weightiter = FullWeightNearestNeighbor.begin(); | 
|---|
|  | 794 | weightiter != FullWeightNearestNeighbor.end(); ++weightiter) { | 
|---|
|  | 795 | const PointWeight_t::displacement_t disp = weightiter->displacement; | 
|---|
|  | 796 | const int index_disp = getValidIndex(disp, N_f, length_f); | 
|---|
|  | 797 | *coarseiter += _fine_level[index_base+index_disp]*weightiter->weight; | 
|---|
|  | 798 | } | 
|---|
|  | 799 | ++coarseiter; | 
|---|
|  | 800 | } | 
|---|
|  | 801 | ASSERT ( (N_c[2] == length_c[2]) && (N_f[2] == length_f[2]), | 
|---|
|  | 802 | "restrictFullWeight() - N_c "+toString(N_c[2])+" != length_c "+toString(length_c[2]) | 
|---|
|  | 803 | +" or N_f "+toString(N_f[2])+" != length_f "+toString(length_f[2])); | 
|---|
|  | 804 | } | 
|---|
|  | 805 | ASSERT ( (N_c[1] == length_c[1]) && (N_f[1] == length_f[1]), | 
|---|
|  | 806 | "restrictFullWeight() - N_c "+toString(N_c[1])+" != length_c "+toString(length_c[1]) | 
|---|
|  | 807 | +" or N_f "+toString(N_f[1])+" != length_f "+toString(length_f[1])); | 
|---|
|  | 808 | } | 
|---|
|  | 809 | ASSERT ( (N_c[0] == length_c[0]) && (N_f[0] == length_f[0]), | 
|---|
|  | 810 | "restrictFullWeight() - N_c "+toString(N_c[0])+" != length_c "+toString(length_c[0]) | 
|---|
|  | 811 | +" or N_f "+toString(N_f[0])+" != length_f "+toString(length_f[0])); | 
|---|
|  | 812 | ASSERT( coarseiter ==  _coarse_level.end(), | 
|---|
|  | 813 | "restrictFullWeight() - coarseiter is not at end of values."); | 
|---|
|  | 814 | } | 
|---|
|  | 815 |  | 
|---|
| [d56e21] | 816 | void SamplingGrid::padWithZerosForEvenNumberedSamples() | 
|---|
|  | 817 | { | 
|---|
|  | 818 | size_t wbegin_index[NDIM]; | 
|---|
|  | 819 | size_t wend_index[NDIM]; | 
|---|
|  | 820 | size_t wlength_index[NDIM]; | 
|---|
|  | 821 | getDiscreteWindowIndices(wbegin_index, wlength_index, wend_index); | 
|---|
|  | 822 |  | 
|---|
|  | 823 | // calculate new window (always extend it such that both indices are even) | 
|---|
|  | 824 | bool changed = false; | 
|---|
|  | 825 | size_t wnewbegin_index[NDIM]; | 
|---|
|  | 826 | size_t wnewend_index[NDIM]; | 
|---|
|  | 827 | for(size_t i=0;i<NDIM;++i) { | 
|---|
| [6a5eb2] | 828 | LOG(2, "INFO: window[" << i << "] starts at " << wbegin_index[i] << " and ends at " | 
|---|
|  | 829 | << wend_index[i] << " with length " << wlength_index[i]); | 
|---|
| [d56e21] | 830 | // We require begin and end of window on even indices (and inside grid). | 
|---|
|  | 831 | wnewbegin_index[i] = wbegin_index[i]; | 
|---|
|  | 832 | if ((wnewbegin_index[i] % (size_t)2) != 0) { | 
|---|
|  | 833 | if (wnewbegin_index[i] > 0) | 
|---|
|  | 834 | --wnewbegin_index[i]; | 
|---|
|  | 835 | else | 
|---|
|  | 836 | wnewbegin_index[i] = 0; | 
|---|
|  | 837 | changed = true; | 
|---|
|  | 838 | } | 
|---|
|  | 839 | wnewend_index[i] = wend_index[i]; | 
|---|
|  | 840 | if ((wnewend_index[i] % (size_t)2) != 0) { | 
|---|
|  | 841 | if (wnewend_index[i] < getGridPointsPerAxis()) | 
|---|
|  | 842 | ++wnewend_index[i]; | 
|---|
|  | 843 | else | 
|---|
|  | 844 | wnewend_index[i] = getGridPointsPerAxis(); | 
|---|
|  | 845 | changed = true; | 
|---|
|  | 846 | } | 
|---|
|  | 847 | ASSERT( (wbegin_index[i] >= 0) && (wend_index[i] <= getGridPointsPerAxis()), | 
|---|
|  | 848 | "SamplingGrid::padWithZerosForEvenNumberedSamples() - indices " | 
|---|
|  | 849 | +toString(wbegin_index[i])+" and "+toString(wend_index[i])+" larger than grid " | 
|---|
|  | 850 | +toString(getGridPointsPerAxis())+"."); | 
|---|
|  | 851 | } | 
|---|
|  | 852 | if (changed) { | 
|---|
|  | 853 | double begin_newwindow[NDIM]; | 
|---|
|  | 854 | double end_newwindow[NDIM]; | 
|---|
|  | 855 | for(size_t i=0;i<NDIM;++i) { | 
|---|
|  | 856 | const double delta = getDeltaPerAxis(i); | 
|---|
|  | 857 | begin_newwindow[i] = begin[i]+delta*wnewbegin_index[i]; | 
|---|
|  | 858 | end_newwindow[i] = begin[i]+delta*wnewend_index[i]; | 
|---|
|  | 859 | } | 
|---|
| [6a5eb2] | 860 | LOG(2, "INFO: Original window is " << Vector(begin_window) << " <-> "  << Vector(end_window)); | 
|---|
|  | 861 | LOG(2, "INFO: Padded window is " << Vector(begin_newwindow) << " <-> "  << Vector(end_newwindow)); | 
|---|
| [d56e21] | 862 | // extend window | 
|---|
|  | 863 | extendWindow(begin_newwindow, end_newwindow); | 
|---|
|  | 864 | } | 
|---|
|  | 865 | ASSERT( getWindowGridPoints() % (size_t)8 == 0, | 
|---|
|  | 866 | "SamplingGrid::padWithZerosForEvenNumberedSamples() - new size " | 
|---|
|  | 867 | +toString(getWindowGridPoints())+" is still not divisible by 8."); | 
|---|
|  | 868 | } | 
|---|
|  | 869 |  | 
|---|
| [06653a] | 870 | void SamplingGrid::downsample( | 
|---|
|  | 871 | SamplingGrid& instance, | 
|---|
|  | 872 | const SamplingGrid& other, | 
|---|
|  | 873 | const int _level) | 
|---|
|  | 874 | { | 
|---|
|  | 875 | if (&instance != &other) { | 
|---|
| [6a5eb2] | 876 | LOG(2, "INFO: other's window is " << Vector(other.begin_window) | 
|---|
|  | 877 | << " <-> " << Vector(other.end_window)); | 
|---|
| [06653a] | 878 | // take over properties | 
|---|
|  | 879 | static_cast<SamplingGridProperties &>(instance) = other; | 
|---|
|  | 880 | instance.setWindowSize(other.begin_window, other.end_window); | 
|---|
| [6a5eb2] | 881 | LOG(2, "INFO: Set instance's window to " << Vector(instance.begin_window) | 
|---|
|  | 882 | << " <-> " << Vector(instance.end_window)); | 
|---|
| [91e7658] | 883 | ASSERT( _level <= other.level, | 
|---|
|  | 884 | "SamplingGrid::downsample() - desired level "+toString(_level) | 
|---|
|  | 885 | +" larger than level "+toString(other.level)+" of the given values."); | 
|---|
|  | 886 | if (_level == other.level) { | 
|---|
| [06653a] | 887 | instance.sampled_grid = other.sampled_grid; | 
|---|
|  | 888 | } else { | 
|---|
|  | 889 | // if desired level is smaller we need to downsample | 
|---|
|  | 890 | // we do this similarly to vmg::RestrictionFullWeight (i.e. a full nearest | 
|---|
|  | 891 | // neighbor interpolation) and always one grid level at a time till we | 
|---|
|  | 892 | // have reached the desired one | 
|---|
|  | 893 |  | 
|---|
| [6a5eb2] | 894 | // we need to copy the other grid because we might need to pad it with zeros anyway | 
|---|
|  | 895 | SamplingGrid FinerGrid(other); | 
|---|
|  | 896 | int length_d[NDIM]; | 
|---|
|  | 897 | int length_s[NDIM]; | 
|---|
| [06653a] | 898 | for (instance.level = other.level-1; instance.level >= _level; --instance.level) { | 
|---|
| [6a5eb2] | 899 | // pad with zeros for even indices and get length of fine grid window | 
|---|
|  | 900 | FinerGrid.padWithZerosForEvenNumberedSamples(); | 
|---|
|  | 901 | getLengthsOfWindow(length_s, FinerGrid); | 
|---|
|  | 902 | ASSERT( FinerGrid.getWindowGridPoints() % 8 == 0, | 
|---|
|  | 903 | "SamplingGrid::downsample() - at level "+toString( instance.level) | 
|---|
|  | 904 | +" given grid points "+toString(FinerGrid.getWindowGridPoints())+" are not even numbered per axis anymore."); | 
|---|
|  | 905 | // re-adjust the window (length), get the corresponding window length and downsample | 
|---|
|  | 906 | instance.setWindow(FinerGrid.begin_window, FinerGrid.end_window); | 
|---|
| [91e7658] | 907 | getLengthsOfWindow(length_d, instance); | 
|---|
| [6a5eb2] | 908 | ASSERT( instance.sampled_grid.size() == FinerGrid.getWindowGridPoints()/(size_t)8, | 
|---|
| [91e7658] | 909 | "SamplingGrid::downsample() - at level "+toString( instance.level) | 
|---|
| [6a5eb2] | 910 | +" points on coarser grid "+toString(instance.sampled_grid.size()) | 
|---|
|  | 911 | +" and downsampled number on finer grid " | 
|---|
|  | 912 | +toString(FinerGrid.getWindowGridPoints()/(size_t)8)+" do not match."); | 
|---|
|  | 913 | restrictFullWeight(instance.sampled_grid, length_d, FinerGrid.sampled_grid, length_s); | 
|---|
|  | 914 | // then use as new finer grid for next downsampling (if it's not the last level) | 
|---|
|  | 915 | if (instance.level > _level) | 
|---|
|  | 916 | FinerGrid = instance; | 
|---|
| [06653a] | 917 | } | 
|---|
| [6a5eb2] | 918 | // loop stops at _level-1 | 
|---|
| [06653a] | 919 | instance.level = _level; | 
|---|
|  | 920 |  | 
|---|
|  | 921 | // and finally, renormalize downsampled grid to old value | 
|---|
|  | 922 | //      instance *= other.integral()/instance.integral(); | 
|---|
|  | 923 | } | 
|---|
|  | 924 | } | 
|---|
|  | 925 | } | 
|---|
|  | 926 |  | 
|---|
| [c889b7] | 927 | std::ostream & operator<<(std::ostream &ost, const SamplingGrid& other) | 
|---|
|  | 928 | { | 
|---|
| [1a00bb] | 929 | ost << "SamplingGrid"; | 
|---|
|  | 930 | ost << " starting at " << other.begin[0] << "," << other.begin[1] << "," << other.begin[2]; | 
|---|
|  | 931 | ost << " ending at " << other.end[0] << "," << other.end[1] << "," << other.end[2]; | 
|---|
|  | 932 | ost << ", window starting at " << other.begin_window[0] << "," << other.begin_window[1] << "," << other.begin_window[2]; | 
|---|
|  | 933 | ost << ", window ending at " << other.end_window[0] << "," << other.end_window[1] << "," << other.end_window[2]; | 
|---|
| [80959a1] | 934 | ost << ", level of " << other.level; | 
|---|
|  | 935 | ost << " and integrated value of " << other.integral(); | 
|---|
| [c889b7] | 936 | return ost; | 
|---|
|  | 937 | } | 
|---|
|  | 938 |  | 
|---|
| [beb16e] | 939 | template<> SamplingGrid ZeroInstance<SamplingGrid>() | 
|---|
|  | 940 | { | 
|---|
|  | 941 | SamplingGrid returnvalue; | 
|---|
|  | 942 | return returnvalue; | 
|---|
|  | 943 | } | 
|---|
|  | 944 |  | 
|---|
| [28c025] | 945 | // we need to explicitly instantiate the serialization functions as | 
|---|
|  | 946 | // its is only serialized through its base class FragmentJob | 
|---|
|  | 947 | BOOST_CLASS_EXPORT_IMPLEMENT(SamplingGrid) | 
|---|