| 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 | 
  | 
 |