tahoma2d/toonz/sources/include/tbigmemorymanager.h

#pragma once

#ifndef _TBIGMEMORYMANAGER_
#define _TBIGMEMORYMANAGER_

class Chunkinfo;

#undef DVAPI
#undef DVVAR
#ifdef TSYSTEM_EXPORTS
#define DVAPI DV_EXPORT_API
#define DVVAR DV_EXPORT_VAR
#else
#define DVAPI DV_IMPORT_API
#define DVVAR DV_IMPORT_VAR
#endif

#include "tcommon.h"
#include "tthreadmessage.h"
class TRaster;

class DVAPI TBigMemoryManager {
  TThread::Mutex m_mutex;
  UCHAR *m_theMemory;
  std::map<UCHAR *, Chunkinfo> m_chunks;
  UCHAR *allocate(UINT &size);

  TUINT32 m_availableMemory, m_allocatedMemory;
  std::map<UCHAR *, Chunkinfo>::iterator shiftBlock(
      const std::map<UCHAR *, Chunkinfo>::iterator &it, TUINT32 offset);
  TRaster *findRaster(TRaster *ras);
  void checkConsistency();
  UCHAR *remap(TUINT32 RequestedSize);
  void printLog(TUINT32 size);

public:
  TBigMemoryManager();
  ~TBigMemoryManager();
  bool init(TUINT32 sizeinKb);
  bool putRaster(TRaster *ras, bool canPutOnDisk = true);
  bool releaseRaster(TRaster *ras);
  void lock(UCHAR *buffer);
  void unlock(UCHAR *buffer);
  UCHAR *getBuffer(UINT size);
  static TBigMemoryManager *instance();
  bool isActive() const { return m_theMemory != 0; }
  // void releaseSubraster(UCHAR *chunk, TRaster*subRas);
  TUINT32 getAvailableMemoryinKb() const {
    assert(m_theMemory != 0);
    return m_availableMemory >> 10;
  }
  void getRasterInfo(int &rasterCount, TUINT32 &totRasterMemInKb,
                     int &notCachedRasterCount,
                     TUINT32 &notCachedRasterMemInKb);
#ifdef _DEBUG
  TUINT32 m_totRasterMemInKb;
  void printMap();
#endif
  int getAllocationPeak();
  int getAllocationMean();

  void setRunOutOfContiguousMemoryHandler(void (*callback)(unsigned long size));

private:
  friend class TRaster;
  void (*m_runOutCallback)(unsigned long);
};

#endif