| 45 |  | #include "brains/SnapshotManager.hpp" | 
| 46 |  | namespace oopse { | 
| 47 |  |  | 
| 48 | < | class SimInfo; | 
| 49 | < | class DumpReader; | 
| 48 | > | class SimInfo; | 
| 49 | > | class DumpReader; | 
| 50 |  |  | 
| 51 | < | typedef std::pair<int, int> SnapshotBlock; | 
| 51 | > | typedef std::pair<int, int> SnapshotBlock; | 
| 52 |  |  | 
| 53 | < | /** | 
| 54 | < | * @class BlockSnapshotManager | 
| 55 | < | * @todo document | 
| 56 | < | */ | 
| 57 | < | class BlockSnapshotManager : public SnapshotManager{ | 
| 53 | > | /** | 
| 54 | > | * @class BlockSnapshotManager | 
| 55 | > | * @todo document | 
| 56 | > | */ | 
| 57 | > | class BlockSnapshotManager : public SnapshotManager{ | 
| 58 |  |  | 
| 59 | < | public: | 
| 60 | < | BlockSnapshotManager(SimInfo* info, const std::string& filename, int storageLayout, int blockCapacity = 2); | 
| 61 | < | ~BlockSnapshotManager(); | 
| 59 | > | public: | 
| 60 | > | BlockSnapshotManager(SimInfo* info, const std::string& filename, int storageLayout, int blockCapacity = 2); | 
| 61 | > | ~BlockSnapshotManager(); | 
| 62 |  |  | 
| 63 | < | virtual Snapshot* getSnapshot(int id) { return snapshots_[id]; } | 
| 63 | > | virtual Snapshot* getSnapshot(int id) { return snapshots_[id]; } | 
| 64 |  |  | 
| 65 | – | /** Returns number of snapshot blocks in this BlockSnapshotManager*/ | 
| 66 | – | int getNBlocks() { | 
| 67 | – | return blocks_.size(); | 
| 68 | – | } | 
| 65 |  |  | 
| 66 | < | SnapshotBlock getSnapshotBlock(int block) { | 
| 67 | < | return blocks_.at(block); | 
| 68 | < | } | 
| 66 | > | /** Returns number of snapshot blocks in this BlockSnapshotManager*/ | 
| 67 | > | int getNBlocks() { | 
| 68 | > | return blocks_.size(); | 
| 69 | > | } | 
| 70 | > |  | 
| 71 | > | SnapshotBlock getSnapshotBlock(int block) { | 
| 72 | > | return blocks_.at(block); | 
| 73 | > | } | 
| 74 |  |  | 
| 75 | < | int getNActiveBlocks(); | 
| 75 | > | int getNActiveBlocks(); | 
| 76 | > |  | 
| 77 | > | void needCOMprops(bool ncp); | 
| 78 | > |  | 
| 79 | > |  | 
| 80 | > | bool isBlockActive(int block) { | 
| 81 | > | return  findActiveBlock(block) != activeBlocks_.end() ? true : false; | 
| 82 | > | } | 
| 83 | > |  | 
| 84 | > | bool loadBlock(int block); | 
| 85 |  |  | 
| 86 | < | bool isBlockActive(int block); | 
| 77 | < |  | 
| 78 | < | bool loadBlock(int block); | 
| 79 | < |  | 
| 80 | < | bool unloadBlock(int block); | 
| 86 | > | bool unloadBlock(int block); | 
| 87 |  |  | 
| 88 | < | std::vector<int> getActiveBlocks(); | 
| 88 | > | std::vector<int> getActiveBlocks(); | 
| 89 |  |  | 
| 90 | < | int getBlockCapacity() { | 
| 91 | < | return blockCapacity_; | 
| 92 | < | } | 
| 90 | > | int getBlockCapacity() { | 
| 91 | > | return blockCapacity_; | 
| 92 | > | } | 
| 93 |  |  | 
| 94 | < | int getNFrames(); | 
| 94 | > | int getNFrames(); | 
| 95 |  |  | 
| 96 | < | private: | 
| 96 | > | private: | 
| 97 |  |  | 
| 98 | < | Snapshot* loadFrame(int frame); | 
| 98 | > | std::vector<int>::iterator findActiveBlock(int block) { | 
| 99 | > | return std::find(activeBlocks_.begin(), activeBlocks_.end(), block); | 
| 100 | > | } | 
| 101 | > |  | 
| 102 | > | bool hasZeroRefBlock(); | 
| 103 | > |  | 
| 104 | > | int getFirstZeroRefBlock(); | 
| 105 | > |  | 
| 106 | > | void internalLoad(int block); | 
| 107 | > | void internalUnload(int block); | 
| 108 | > | Snapshot* loadFrame(int frame); | 
| 109 |  |  | 
| 110 | < | SimInfo* info_; | 
| 111 | < | int blockCapacity_; | 
| 110 | > | SimInfo* info_; | 
| 111 | > | int blockCapacity_; | 
| 112 |  |  | 
| 113 | < | std::vector<Snapshot*> snapshots_; | 
| 114 | < | std::vector<SnapshotBlock> blocks_; | 
| 115 | < | std::vector<int> activeBlocks_; | 
| 113 | > | std::vector<Snapshot*> snapshots_; | 
| 114 | > | std::vector<SnapshotBlock> blocks_; | 
| 115 | > | std::vector<int> activeBlocks_; | 
| 116 | > | std::vector<int> activeRefCount_; | 
| 117 |  |  | 
| 118 | < | int nAtoms_; | 
| 119 | < | int nRigidBodies_; | 
| 118 | > | int nAtoms_; | 
| 119 | > | int nRigidBodies_; | 
| 120 |  |  | 
| 121 | < | DumpReader* reader_; | 
| 122 | < | int nframes_; | 
| 123 | < | int nSnapshotPerBlock_; | 
| 121 | > | DumpReader* reader_; | 
| 122 | > | int nframes_; | 
| 123 | > | int nSnapshotPerBlock_; | 
| 124 |  |  | 
| 125 | < | }; | 
| 125 | > | }; | 
| 126 |  |  | 
| 127 |  | } | 
| 128 |  |  |