ViewVC Help
View File | Revision Log | Show Annotations | View Changeset | Root Listing
root/OpenMD/trunk/src/brains/BlockSnapshotManager.cpp
Revision: 2071
Committed: Sat Mar 7 21:41:51 2015 UTC (10 years, 1 month ago) by gezelter
File size: 10892 byte(s)
Log Message:
Reducing the number of warnings when using g++ to compile.

File Contents

# User Rev Content
1 tim 318 /*
2     * Copyright (c) 2005 The University of Notre Dame. All Rights Reserved.
3     *
4     * The University of Notre Dame grants you ("Licensee") a
5     * non-exclusive, royalty free, license to use, modify and
6     * redistribute this software in source and binary code form, provided
7     * that the following conditions are met:
8     *
9 gezelter 1390 * 1. Redistributions of source code must retain the above copyright
10 tim 318 * notice, this list of conditions and the following disclaimer.
11     *
12 gezelter 1390 * 2. Redistributions in binary form must reproduce the above copyright
13 tim 318 * notice, this list of conditions and the following disclaimer in the
14     * documentation and/or other materials provided with the
15     * distribution.
16     *
17     * This software is provided "AS IS," without a warranty of any
18     * kind. All express or implied conditions, representations and
19     * warranties, including any implied warranty of merchantability,
20     * fitness for a particular purpose or non-infringement, are hereby
21     * excluded. The University of Notre Dame and its licensors shall not
22     * be liable for any damages suffered by licensee as a result of
23     * using, modifying or distributing the software or its
24     * derivatives. In no event will the University of Notre Dame or its
25     * licensors be liable for any lost revenue, profit or data, or for
26     * direct, indirect, special, consequential, incidental or punitive
27     * damages, however caused and regardless of the theory of liability,
28     * arising out of the use of or inability to use software, even if the
29     * University of Notre Dame has been advised of the possibility of
30     * such damages.
31 gezelter 1390 *
32     * SUPPORT OPEN SCIENCE! If you use OpenMD or its source code in your
33     * research, please cite the appropriate papers when you publish your
34     * work. Good starting points are:
35     *
36     * [1] Meineke, et al., J. Comp. Chem. 26, 252-271 (2005).
37     * [2] Fennell & Gezelter, J. Chem. Phys. 124, 234104 (2006).
38 gezelter 1879 * [3] Sun, Lin & Gezelter, J. Chem. Phys. 128, 234107 (2008).
39 gezelter 1782 * [4] Kuang & Gezelter, J. Chem. Phys. 133, 164101 (2010).
40     * [5] Vardeman, Stocker & Gezelter, J. Chem. Theory Comput. 7, 834 (2011).
41 tim 318 */
42     #include <algorithm>
43     #include "brains/BlockSnapshotManager.hpp"
44 gezelter 1782 //#include "utils/residentMem.h"
45     //#include "utils/physmem.h"
46 tim 324 #include "utils/Algorithm.hpp"
47 tim 318 #include "brains/SimInfo.hpp"
48     #include "io/DumpReader.hpp"
49    
50 gezelter 1390 namespace OpenMD {
51 gezelter 1596 BlockSnapshotManager::BlockSnapshotManager(SimInfo* info,
52     const std::string& filename,
53 gezelter 1782 int storageLayout,
54     long long int memSize,
55 gezelter 1596 int blockCapacity)
56 gezelter 2071 : SnapshotManager(storageLayout), info_(info),
57     blockCapacity_(blockCapacity), memSize_(memSize),
58     activeBlocks_(blockCapacity_, -1),
59 gezelter 1596 activeRefCount_(blockCapacity_, 0) {
60 tim 318
61 gezelter 507 nAtoms_ = info->getNGlobalAtoms();
62     nRigidBodies_ = info->getNGlobalRigidBodies();
63 gezelter 1782 nCutoffGroups_ = info->getNCutoffGroups();
64 tim 318
65 gezelter 1596 // eliminate suspect calls to figure out free memory:
66     // RealType physMem = physmem_total();
67     // RealType rssMem = residentMem();
68     // RealType avaliablePhysMem = physMem - rssMem;
69 gezelter 1782
70 gezelter 507 int bytesPerStuntDouble = DataStorage::getBytesPerStuntDouble(storageLayout);
71 gezelter 1965 int bytesPerCutoffGroup = DataStorage::getBytesPerStuntDouble(DataStorage::dslPosition);
72 gezelter 1966
73     int bytesPerFrameData = Snapshot::getFrameDataSize();
74 gezelter 1965 int bytesPerFrame = (nRigidBodies_ + nAtoms_) * bytesPerStuntDouble
75 gezelter 1966 + nCutoffGroups_ * bytesPerCutoffGroup
76     + bytesPerFrameData;
77 tim 318
78 xsun 1215 // total number of frames that can fit in memory
79 gezelter 1596 //RealType frameCapacity = avaliablePhysMem / bytesPerFrame;
80 gezelter 1782 RealType frameCapacity = (RealType) memSize_ / (RealType) bytesPerFrame;
81 xsun 1215
82     // number of frames in each block given the need to hold multiple blocks
83     // in memory at the same time:
84     nSnapshotPerBlock_ = int(frameCapacity) / blockCapacity_;
85 gezelter 1963 if (nSnapshotPerBlock_ <= 0) {
86     std::cerr << "not enough memory to hold two configs!" << std::endl;
87     }
88 gezelter 507 reader_ = new DumpReader(info, filename);
89     nframes_ = reader_->getNFrames();
90     int nblocks = nframes_ / nSnapshotPerBlock_;
91 xsun 1215 if (nframes_ % int(nSnapshotPerBlock_) != 0) {
92 tim 318 ++nblocks;
93 gezelter 507 }
94 tim 318
95 gezelter 507 for (int i = 0; i < nblocks; ++i) {
96 tim 337 blocks_.push_back(SnapshotBlock(i*nSnapshotPerBlock_, (i+1)*nSnapshotPerBlock_));
97 gezelter 507 }
98     //the last block may not have nSnapshotPerBlock frames, we need to consider this special situation
99     blocks_.back().second = nframes_;
100 tim 318
101 gezelter 507 snapshots_.insert(snapshots_.begin(), nframes_, static_cast<Snapshot*>(NULL));
102 tim 358
103 gezelter 1966 std::cout << "-----------------------------------------------------"
104     << std::endl;
105 xsun 1215 std::cout << "BlockSnapshotManager memory report:" << std::endl;
106     std::cout << "\n";
107 gezelter 1596 // std::cout << " Physical Memory available:\t" << (unsigned long)physMem << " bytes" <<std::endl;
108     //std::cout << " Resident Memory in use:\t" << (unsigned long)rssMem << " bytes" <<std::endl;
109     //std::cout << "Memory available for OpenMD:\t" << (unsigned long)avaliablePhysMem << " bytes" <<std::endl;
110 gezelter 1966 std::cout << "Memory requested for OpenMD:\t"
111     << (unsigned long)memSize_ << " bytes" << std::endl;
112     std::cout << " Bytes per FrameData:\t"
113     << (unsigned long)bytesPerFrameData << std::endl;
114     std::cout << " Bytes per StuntDouble:\t"
115     << (unsigned long)bytesPerStuntDouble << std::endl;
116     std::cout << " Bytes per Cutoff Group:\t"
117     << (unsigned long)bytesPerCutoffGroup << std::endl;
118     std::cout << " Bytes per Frame:\t"
119     << (unsigned long)bytesPerFrame << std::endl;
120     std::cout << " Frame Capacity:\t"
121     << (unsigned long)frameCapacity << std::endl;
122     std::cout << " Frames in trajectory:\t"
123     << (unsigned long)nframes_ << std::endl;
124     std::cout << " Snapshots per Block:\t"
125     << (unsigned long)nSnapshotPerBlock_ << std::endl;
126     std::cout << " Total number of Blocks:\t"
127     << (unsigned long)nblocks << std::endl;
128     std::cout << "-----------------------------------------------------"
129     << std::endl;
130 tim 318
131 gezelter 507 }
132 tim 318
133    
134 gezelter 507 BlockSnapshotManager::~BlockSnapshotManager() {
135 tim 318 currentSnapshot_ = NULL;
136     previousSnapshot_ = NULL;
137    
138     delete reader_;
139 tim 324
140     std::vector<int>::iterator i;
141     for (i = activeBlocks_.begin(); i != activeBlocks_.end(); ++i) {
142 gezelter 507 if (*i != -1) {
143     unloadBlock(*i);
144     }
145 tim 324 }
146 gezelter 507 }
147 tim 318
148 gezelter 1249 Snapshot* BlockSnapshotManager::getSnapshot(int id) {
149     currentSnapshot_ = snapshots_[id];
150     return snapshots_[id];
151     }
152 xsun 1215
153 gezelter 507 int BlockSnapshotManager::getNActiveBlocks() {
154 tim 347 #ifdef __RWSTD
155     int count = 0;
156     std::count_if(activeBlocks_.begin(), activeBlocks_.end(), std::bind2nd(std::not_equal_to<int>(), -1), count);
157     return count;
158     #else
159 tim 324 return std::count_if(activeBlocks_.begin(), activeBlocks_.end(), std::bind2nd(std::not_equal_to<int>(), -1));
160 tim 347 #endif
161 gezelter 507 }
162 tim 318
163    
164 tim 351
165 gezelter 507 bool BlockSnapshotManager::loadBlock(int block) {
166 tim 351 std::vector<int>::iterator i = findActiveBlock(block);
167 gezelter 2071 bool loadSuccess(false);
168 tim 351 if (i != activeBlocks_.end()) {
169 gezelter 2071 // If the block is already in memory, just increase the
170     // reference count:
171 gezelter 507 ++activeRefCount_[i - activeBlocks_.begin()];
172     loadSuccess = true;
173 tim 318 } else if (getNActiveBlocks() < blockCapacity_){
174 gezelter 2071 // If the number of active blocks is less than the block
175     // capacity, just load the block:
176 gezelter 507 internalLoad(block);
177     loadSuccess = true;
178 gezelter 1782 } else if ( hasZeroRefBlock() ) {
179 gezelter 2071 // If we have already reached the block capacity, we need to
180     // unload a block with 0 references:
181 gezelter 507 int zeroRefBlock = getFirstZeroRefBlock();
182     assert(zeroRefBlock != -1);
183     internalUnload(zeroRefBlock);
184     internalLoad(block);
185 tim 351 } else {
186 gezelter 2071 // We have reached capacity and all blocks in memory are have
187     // non-zero references:
188 gezelter 507 loadSuccess = false;
189 gezelter 2071 }
190 tim 318 return loadSuccess;
191 gezelter 507 }
192 tim 318
193 gezelter 507 bool BlockSnapshotManager::unloadBlock(int block) {
194 tim 318 bool unloadSuccess;
195 tim 351 std::vector<int>::iterator i = findActiveBlock(block);
196    
197     if (i != activeBlocks_.end()){
198 gezelter 507 --activeRefCount_[i - activeBlocks_.begin()];
199     if (activeRefCount_[i - activeBlocks_.begin()] < 0) {
200     //in case, unloadBlock called multiple times
201     activeRefCount_[i - activeBlocks_.begin()] = 0;
202     }
203 gezelter 1964
204     if (activeRefCount_[i-activeBlocks_.begin()] == 0) {
205     internalUnload(block);
206     }
207 tim 351
208 gezelter 507 unloadSuccess = true;
209 tim 351 } else {
210 gezelter 507 unloadSuccess = false;
211 tim 318 }
212 tim 324
213     return unloadSuccess;
214 gezelter 507 }
215 tim 318
216 gezelter 507 void BlockSnapshotManager::internalLoad(int block) {
217 tim 351
218     for (int i = blocks_[block].first; i < blocks_[block].second; ++i) {
219 gezelter 507 snapshots_[i] = loadFrame(i);
220 tim 351 }
221    
222     std::vector<int>::iterator j;
223     j = std::find(activeBlocks_.begin(), activeBlocks_.end(), -1);
224     assert(j != activeBlocks_.end());
225     *j = block;
226     ++activeRefCount_[j - activeBlocks_.begin()];
227 gezelter 507 }
228 tim 351
229 gezelter 507 void BlockSnapshotManager::internalUnload(int block) {
230 gezelter 1964 std::cerr << "called internal unload for block "<< block << "\n";
231 tim 351 for (int i = blocks_[block].first; i < blocks_[block].second; ++i) {
232 gezelter 507 delete snapshots_[i];
233     snapshots_[i] = NULL;
234 tim 351 }
235     std::vector<int>::iterator j;
236     j = std::find(activeBlocks_.begin(), activeBlocks_.end(), block);
237     assert(j != activeBlocks_.end());
238     *j = -1;
239 gezelter 507 }
240 tim 351
241 gezelter 507 bool BlockSnapshotManager::hasZeroRefBlock(){
242 tim 352 return std::find(activeRefCount_.begin(), activeRefCount_.end(), 0) != activeRefCount_.end() ? true : false;
243 gezelter 507 }
244 tim 351
245 gezelter 507 int BlockSnapshotManager::getFirstZeroRefBlock(){
246     std::vector<int>::iterator i = std::find(activeRefCount_.begin(), activeRefCount_.end(), 0);
247     return i != activeRefCount_.end() ? activeBlocks_[i - activeRefCount_.begin()] : -1;
248     }
249 tim 351
250 gezelter 507 std::vector<int> BlockSnapshotManager::getActiveBlocks() {
251 tim 318 std::vector<int> result;
252 gezelter 1390 OpenMD::copy_if(activeBlocks_.begin(), activeBlocks_.end(), std::back_inserter(result),
253 gezelter 507 std::bind2nd(std::not_equal_to<int>(), -1));
254 tim 324 return result;
255 gezelter 507 }
256 tim 318
257 gezelter 507 Snapshot* BlockSnapshotManager::loadFrame(int frame){
258 gezelter 1879 Snapshot* snapshot = new Snapshot(nAtoms_, nRigidBodies_, nCutoffGroups_,
259 gezelter 1793 getStorageLayout());
260 tim 318 snapshot->setID(frame);
261 gezelter 1782 snapshot->clearDerivedProperties();
262 tim 324
263     currentSnapshot_ = snapshot;
264 tim 318 reader_->readFrame(frame);
265 xsun 1215
266 tim 318 return snapshot;
267 gezelter 507 }
268 tim 318
269 gezelter 507 int BlockSnapshotManager::getNFrames() {
270 tim 322 return reader_->getNFrames();
271 gezelter 507 }
272 tim 322
273 gezelter 1106 void BlockSnapshotManager::needCOMprops(bool ncp) {
274     reader_->setNeedCOMprops(ncp);
275     }
276    
277 tim 322 }

Properties

Name Value
svn:keywords Author Id Revision Date