TmplBlockBuffer.tlh 2.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128
  1. // TmplBlockBuffer.hpp
  2. //
  3. #pragma once
  4. //-----------------------------------------------------------------------------
  5. // TmplBlockBuffer
  6. //
  7. // 定义一个连续的内存块
  8. //-----------------------------------------------------------------------------
  9. template <typename T> class TmplBlockBuffer
  10. {
  11. typedef T __BASE;
  12. private:
  13. struct MetaData
  14. {
  15. long nRefs; // reference count
  16. int nCount; // Number of item
  17. };
  18. protected:
  19. T * m_pObject;
  20. MetaData * m_pMetaData;
  21. public:
  22. TmplBlockBuffer ();
  23. explicit TmplBlockBuffer (T * object, int count);
  24. TmplBlockBuffer (const TmplBlockBuffer & h);
  25. TmplBlockBuffer (TmplBlockBuffer && h);
  26. // 从 h 构造一个新的对象, 但是我只要前面的 count 个字符, 如果 count 比 h.nCount 大, 那就重新分配然后 memcpy
  27. TmplBlockBuffer (TmplBlockBuffer && h, int count);
  28. virtual ~TmplBlockBuffer ();
  29. public:
  30. void Attach (T * object, int count);
  31. T * Detach ();
  32. // real copy from another TmplBlockBuffer
  33. void CopyFrom (const TmplBlockBuffer & h);
  34. // ref-counted copy from another TmplBlockBuffer
  35. void ShadowCopyFrom (const TmplBlockBuffer & h);
  36. TmplBlockBuffer & operator = (const TmplBlockBuffer & h);
  37. TmplBlockBuffer & operator = (TmplBlockBuffer && h);
  38. T * GetBufferSetCount (int count);
  39. void CopyFrom (const T * from, int count);
  40. void CopyTo (T * to, int count) const;
  41. operator bool () const;
  42. operator T * ();
  43. operator const T * () const;
  44. int GetCount () const
  45. {
  46. if (m_pMetaData)
  47. return m_pMetaData->nCount;
  48. return 0;
  49. }
  50. // Allow usage as a pointer
  51. T * operator -> ()
  52. {
  53. return m_pObject;
  54. }
  55. const T * operator -> () const
  56. {
  57. return m_pObject;
  58. }
  59. inline bool operator == (const TmplBlockBuffer & holder) const
  60. {
  61. return m_pObject == holder.m_pObject;
  62. }
  63. inline bool operator != (const TmplBlockBuffer & holder) const
  64. {
  65. return m_pObject != holder.m_pObject;
  66. }
  67. T * As ()
  68. {
  69. return m_pObject;
  70. }
  71. const T * As () const
  72. {
  73. return m_pObject;
  74. }
  75. template <typename C> C * AS ()
  76. {
  77. return (C *) m_pObject;
  78. }
  79. template <typename C> const C * AS () const
  80. {
  81. return (const C *) m_pObject;
  82. }
  83. bool IsEmpty () const;
  84. void Release ();
  85. protected:
  86. inline MetaData * GetData () const
  87. {
  88. return m_pMetaData;
  89. }
  90. protected:
  91. void Init ();
  92. void MoveAssign (TmplBlockBuffer & Src);
  93. };
  94. // 为像素数据定义一个专用的类型
  95. typedef TmplBlockBuffer <UINT8> BlockBuffer;
  96. typedef BlockBuffer IBlockBuffer;
  97. typedef BlockBuffer OBlockBuffer;