1.1 --- /dev/null	Thu Jan 01 00:00:00 1970 +0000
     1.2 +++ b/src/win32/7zip/7z/CPP/7zip/Common/MemBlocks.h	Sat Mar 03 10:31:27 2012 -0600
     1.3 @@ -0,0 +1,77 @@
     1.4 +// MemBlocks.h
     1.5 +
     1.6 +#ifndef __MEMBLOCKS_H
     1.7 +#define __MEMBLOCKS_H
     1.8 +
     1.9 +extern "C"
    1.10 +{
    1.11 +#include "../../../C/Alloc.h"
    1.12 +}
    1.13 +
    1.14 +#include "Common/Types.h"
    1.15 +#include "Common/MyVector.h"
    1.16 +
    1.17 +#include "Windows/Synchronization.h"
    1.18 +
    1.19 +#include "../IStream.h"
    1.20 +
    1.21 +class CMemBlockManager
    1.22 +{
    1.23 +  void *_data;
    1.24 +  size_t _blockSize;
    1.25 +  void *_headFree;
    1.26 +public:
    1.27 +  CMemBlockManager(size_t blockSize = (1 << 20)): _data(0), _blockSize(blockSize), _headFree(0) {}
    1.28 +  ~CMemBlockManager() { FreeSpace(); }
    1.29 +
    1.30 +  bool AllocateSpace(size_t numBlocks);
    1.31 +  void FreeSpace();
    1.32 +  size_t GetBlockSize() const { return _blockSize; }
    1.33 +  void *AllocateBlock();
    1.34 +  void FreeBlock(void *p);
    1.35 +};
    1.36 +
    1.37 +
    1.38 +class CMemBlockManagerMt: public CMemBlockManager
    1.39 +{
    1.40 +  NWindows::NSynchronization::CCriticalSection _criticalSection;
    1.41 +public:
    1.42 +  NWindows::NSynchronization::CSemaphore Semaphore;
    1.43 +
    1.44 +  CMemBlockManagerMt(size_t blockSize = (1 << 20)): CMemBlockManager(blockSize) {}
    1.45 +  ~CMemBlockManagerMt() { FreeSpace(); }
    1.46 +
    1.47 +  HRes AllocateSpace(size_t numBlocks, size_t numNoLockBlocks = 0);
    1.48 +  HRes AllocateSpaceAlways(size_t desiredNumberOfBlocks, size_t numNoLockBlocks = 0);
    1.49 +  void FreeSpace();
    1.50 +  void *AllocateBlock();
    1.51 +  void FreeBlock(void *p, bool lockMode = true);
    1.52 +  HRes ReleaseLockedBlocks(int number) { return Semaphore.Release(number); }
    1.53 +};
    1.54 +
    1.55 +
    1.56 +class CMemBlocks
    1.57 +{
    1.58 +  void Free(CMemBlockManagerMt *manager);
    1.59 +public:
    1.60 +  CRecordVector<void *> Blocks;
    1.61 +  UInt64 TotalSize;
    1.62 +  
    1.63 +  CMemBlocks(): TotalSize(0) {}
    1.64 +
    1.65 +  void FreeOpt(CMemBlockManagerMt *manager);
    1.66 +  HRESULT WriteToStream(size_t blockSize, ISequentialOutStream *outStream) const;
    1.67 +};
    1.68 +
    1.69 +struct CMemLockBlocks: public CMemBlocks
    1.70 +{
    1.71 +  bool LockMode;
    1.72 +
    1.73 +  CMemLockBlocks(): LockMode(true) {};
    1.74 +  void Free(CMemBlockManagerMt *memManager);
    1.75 +  void FreeBlock(int index, CMemBlockManagerMt *memManager);
    1.76 +  HRes SwitchToNoLockMode(CMemBlockManagerMt *memManager);
    1.77 +  void Detach(CMemLockBlocks &blocks, CMemBlockManagerMt *memManager);
    1.78 +};
    1.79 +
    1.80 +#endif