55 |
|
|
56 |
|
} |
57 |
|
|
58 |
< |
DataStorage::DataStorage(int size, int storageLayout) : size_(size){ |
58 |
> |
DataStorage::DataStorage(std::size_t size, int storageLayout) : size_(size){ |
59 |
|
setStorageLayout(storageLayout); |
60 |
|
resize(size); |
61 |
|
} |
62 |
|
|
63 |
< |
int DataStorage::getSize() { |
63 |
> |
std::size_t DataStorage::getSize() { |
64 |
|
|
65 |
|
if (storageLayout_ & dslPosition && position.size() != size_) { |
66 |
|
//error |
146 |
|
//error |
147 |
|
cerr << "size does not match"<< endl; |
148 |
|
} |
149 |
+ |
|
150 |
+ |
if (storageLayout_ & dslSitePotential && sitePotential.size() != size_) { |
151 |
+ |
//error |
152 |
+ |
cerr << "size does not match"<< endl; |
153 |
+ |
} |
154 |
|
|
155 |
|
return size_; |
156 |
|
|
157 |
|
} |
158 |
|
|
159 |
< |
void DataStorage::resize(int newSize) { |
159 |
> |
void DataStorage::resize(std::size_t newSize) { |
160 |
|
|
161 |
|
if (storageLayout_ & dslPosition) { |
162 |
|
internalResize(position, newSize); |
226 |
|
internalResize(flucQFrc, newSize); |
227 |
|
} |
228 |
|
|
229 |
+ |
if (storageLayout_ & dslSitePotential) { |
230 |
+ |
internalResize(sitePotential, newSize); |
231 |
+ |
} |
232 |
+ |
|
233 |
|
size_ = newSize; |
234 |
|
} |
235 |
|
|
236 |
< |
void DataStorage::reserve(int size) { |
236 |
> |
void DataStorage::reserve(std::size_t size) { |
237 |
|
if (storageLayout_ & dslPosition) { |
238 |
|
position.reserve(size); |
239 |
|
} |
301 |
|
if (storageLayout_ & dslFlucQForce) { |
302 |
|
flucQFrc.reserve(size); |
303 |
|
} |
304 |
+ |
|
305 |
+ |
if (storageLayout_ & dslSitePotential) { |
306 |
+ |
sitePotential.reserve(size); |
307 |
+ |
} |
308 |
|
} |
309 |
|
|
310 |
< |
void DataStorage::copy(int source, int num, int target) { |
310 |
> |
void DataStorage::copy(int source, std::size_t num, std::size_t target) { |
311 |
|
if (num + target > size_ ) { |
312 |
|
//error |
313 |
|
} |
378 |
|
if (storageLayout_ & dslFlucQForce) { |
379 |
|
internalCopy(flucQFrc, source, num, target); |
380 |
|
} |
381 |
+ |
|
382 |
+ |
if (storageLayout_ & dslSitePotential) { |
383 |
+ |
internalCopy(sitePotential, source, num, target); |
384 |
+ |
} |
385 |
|
} |
386 |
|
|
387 |
|
int DataStorage::getStorageLayout() { |
447 |
|
case dslFlucQForce: |
448 |
|
return internalGetArrayPointer(flucQFrc); |
449 |
|
|
450 |
+ |
case dslSitePotential: |
451 |
+ |
return internalGetArrayPointer(sitePotential); |
452 |
+ |
|
453 |
|
default: |
454 |
|
//error message |
455 |
|
return NULL; |
483 |
|
} |
484 |
|
|
485 |
|
template<typename T> |
486 |
< |
void DataStorage::internalResize(std::vector<T>& v, int newSize){ |
487 |
< |
int oldSize = v.size(); |
486 |
> |
void DataStorage::internalResize(std::vector<T>& v, std::size_t newSize){ |
487 |
> |
std::size_t oldSize = v.size(); |
488 |
|
|
489 |
|
if (oldSize == newSize) { |
490 |
|
return; |
499 |
|
} |
500 |
|
|
501 |
|
template<typename T> |
502 |
< |
void DataStorage::internalCopy(std::vector<T>& v, int source, int num, int target) { |
502 |
> |
void DataStorage::internalCopy(std::vector<T>& v, int source, |
503 |
> |
std::size_t num, std::size_t target) { |
504 |
|
typename std::vector<T>::iterator first; |
505 |
|
typename std::vector<T>::iterator last; |
506 |
|
typename std::vector<T>::iterator result; |
517 |
|
std::copy(first, last, result); |
518 |
|
} |
519 |
|
|
520 |
< |
int DataStorage::getBytesPerStuntDouble(int layout) { |
521 |
< |
int bytes = 0; |
520 |
> |
std::size_t DataStorage::getBytesPerStuntDouble(int layout) { |
521 |
> |
std::size_t bytes = 0; |
522 |
|
if (layout & dslPosition) { |
523 |
|
bytes += sizeof(Vector3d); |
524 |
|
} |
570 |
|
if (layout & dslFlucQForce) { |
571 |
|
bytes += sizeof(RealType); |
572 |
|
} |
573 |
+ |
if (layout & dslSitePotential) { |
574 |
+ |
bytes += sizeof(RealType); |
575 |
+ |
} |
576 |
|
|
577 |
|
return bytes; |
578 |
|
} |