| 40 |
|
*/ |
| 41 |
|
#ifndef BRAINS_BLOCKSNAPSHOTMANAGER_HPP |
| 42 |
|
#define BRAINS_BLOCKSNAPSHOTMANAGER_HPP |
| 43 |
+ |
#include <vector> |
| 44 |
|
|
| 45 |
|
#include "brains/SnapshotManager.hpp" |
| 46 |
|
namespace oopse { |
| 54 |
|
* @class BlockSnapshotManager |
| 55 |
|
* @todo document |
| 56 |
|
*/ |
| 57 |
< |
class BlockSnapshotMananger : public SnapshotManager{ |
| 57 |
> |
class BlockSnapshotManager : public SnapshotManager{ |
| 58 |
|
|
| 59 |
|
public: |
| 60 |
< |
BlockSnapshotMananger(SimInfo* info, const std::string& filename, int storageLayout, int blockCapacity = 2); |
| 61 |
< |
~BlockSnapshotMananger(); |
| 60 |
> |
BlockSnapshotManager(SimInfo* info, const std::string& filename, int storageLayout, int blockCapacity = 2); |
| 61 |
> |
~BlockSnapshotManager(); |
| 62 |
|
|
| 63 |
< |
virtual Snapshot* getSnapshot(int id); |
| 63 |
> |
virtual Snapshot* getSnapshot(int id) { return snapshots_[id]; } |
| 64 |
|
|
| 65 |
|
/** Returns number of snapshot blocks in this BlockSnapshotManager*/ |
| 66 |
|
int getNBlocks() { |
| 73 |
|
|
| 74 |
|
int getNActiveBlocks(); |
| 75 |
|
|
| 76 |
< |
bool isBlockActive(int block); |
| 77 |
< |
|
| 76 |
> |
|
| 77 |
> |
bool isBlockActive(int block) { |
| 78 |
> |
return findActiveBlock(block) != activeBlocks_.end() ? true : false; |
| 79 |
> |
} |
| 80 |
> |
|
| 81 |
|
bool loadBlock(int block); |
| 82 |
|
|
| 83 |
|
bool unloadBlock(int block); |
| 92 |
|
|
| 93 |
|
private: |
| 94 |
|
|
| 95 |
+ |
std::vector<int>::iterator findActiveBlock(int block) { |
| 96 |
+ |
return std::find(activeBlocks_.begin(), activeBlocks_.end(), block); |
| 97 |
+ |
} |
| 98 |
+ |
|
| 99 |
+ |
bool hasZeroRefBlock(); |
| 100 |
+ |
|
| 101 |
+ |
int getFirstZeroRefBlock(); |
| 102 |
+ |
|
| 103 |
+ |
void internalLoad(int block); |
| 104 |
+ |
void internalUnload(int block); |
| 105 |
|
Snapshot* loadFrame(int frame); |
| 106 |
|
|
| 107 |
|
SimInfo* info_; |
| 94 |
– |
int storageLayout_; |
| 108 |
|
int blockCapacity_; |
| 109 |
|
|
| 110 |
|
std::vector<Snapshot*> snapshots_; |
| 111 |
|
std::vector<SnapshotBlock> blocks_; |
| 112 |
|
std::vector<int> activeBlocks_; |
| 113 |
+ |
std::vector<int> activeRefCount_; |
| 114 |
|
|
| 115 |
|
int nAtoms_; |
| 116 |
|
int nRigidBodies_; |