PixelBuffer.h
上传用户:ghyvgy
上传日期:2009-05-26
资源大小:547k
文件大小:7k
源码类别:

其他游戏

开发平台:

Python

  1. /*
  2. s_p_oneil@hotmail.com
  3. Copyright (c) 2000, Sean O'Neil
  4. All rights reserved.
  5. Redistribution and use in source and binary forms, with or without
  6. modification, are permitted provided that the following conditions are met:
  7. * Redistributions of source code must retain the above copyright notice,
  8.   this list of conditions and the following disclaimer.
  9. * Redistributions in binary form must reproduce the above copyright notice,
  10.   this list of conditions and the following disclaimer in the documentation
  11.   and/or other materials provided with the distribution.
  12. * Neither the name of this project nor the names of its contributors
  13.   may be used to endorse or promote products derived from this software
  14.   without specific prior written permission.
  15. THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
  16. AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  17. IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  18. ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
  19. LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  20. CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  21. SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  22. INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  23. CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  24. ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  25. POSSIBILITY OF SUCH DAMAGE.
  26. */
  27. #ifndef __PixelBuffer_h__
  28. #define __PixelBuffer_h__
  29. #include "Matrix.h"
  30. #define ALIGN_SIZE 64
  31. #define ALIGN_MASK (ALIGN_SIZE-1)
  32. #define ALIGN(x) (((unsigned int)x+ALIGN_MASK) & ~ALIGN_MASK)
  33. typedef enum
  34. {
  35. UnsignedByteType = GL_UNSIGNED_BYTE,
  36. SignedByteType = GL_BYTE,
  37. UnsignedShortType = GL_UNSIGNED_SHORT,
  38. SignedShortType = GL_SHORT,
  39. UnsignedIntType = GL_UNSIGNED_INT,
  40. SignedIntType = GL_INT,
  41. FloatType = GL_FLOAT,
  42. DoubleType = GL_DOUBLE
  43. } BufferDataType;
  44. inline const int GetDataTypeSize(const int nDataType)
  45. {
  46. int nSize;
  47. switch(nDataType)
  48. {
  49. case UnsignedByteType:
  50. case SignedByteType:
  51. nSize = 1;
  52. break;
  53. case UnsignedShortType:
  54. case SignedShortType:
  55. nSize = 2;
  56. break;
  57. case UnsignedIntType:
  58. case SignedIntType:
  59. case FloatType:
  60. nSize = 4;
  61. break;
  62. case DoubleType:
  63. nSize = 8;
  64. break;
  65. default:
  66. nSize = 0;
  67. break;
  68. }
  69. return nSize;
  70. }
  71. class C2DBuffer
  72. {
  73. protected:
  74. int m_nWidth; // The width of the buffer
  75. int m_nHeight; // The height of the buffer
  76. int m_nDataType; // The data type stored in the buffer (i.e. GL_UNSIGNED_BYTE, GL_FLOAT)
  77. int m_nChannels; // The number of channels of data stored in the buffer
  78. int m_nElementSize; // The size of one element in the buffer
  79. int m_nRowSize; // The size of one row in the buffer
  80. void *m_pAlloc; // The pointer to the pixel buffer
  81. void *m_pBuffer; // A byte-aligned pointer (for faster memory access)
  82. public:
  83. C2DBuffer() { m_pAlloc = m_pBuffer = NULL; }
  84. C2DBuffer(const C2DBuffer &buf) { *this = buf; }
  85. C2DBuffer(const int nWidth, const int nHeight, const int nDataType, const int nChannels=1, void *pBuffer=NULL)
  86. {
  87. m_pAlloc = m_pBuffer = NULL;
  88. Init(nWidth, nHeight, nDataType, nChannels, pBuffer);
  89. }
  90. ~C2DBuffer() { Cleanup(); }
  91. void operator=(const C2DBuffer &buf)
  92. {
  93. Init(buf.m_nWidth, buf.m_nHeight, buf.m_nDataType, buf.m_nChannels);
  94. memcpy(m_pBuffer, buf.m_pBuffer, GetBufferSize());
  95. }
  96. bool operator==(const C2DBuffer &buf)
  97. {
  98. return (m_nWidth == buf.m_nWidth && m_nHeight == buf.m_nHeight && m_nDataType == buf.m_nDataType && m_nChannels == buf.m_nChannels);
  99. }
  100. void *operator[](const int n)
  101. {
  102. return (void *)((unsigned int)m_pBuffer + n * m_nElementSize);
  103. }
  104. void *operator()(const int x, const int y)
  105. {
  106. return (void *)((unsigned int)m_pBuffer + y * m_nRowSize + x * m_nElementSize);
  107. }
  108. void Init(const int nWidth, const int nHeight, const int nDataType, const int nChannels=1, void *pBuffer=NULL)
  109. {
  110. // If the buffer is already initialized to the specified settings, then nothing needs to be done
  111. if(m_pAlloc && m_nWidth == nWidth && m_nHeight == nHeight && m_nDataType == nDataType && m_nChannels == nChannels)
  112. return;
  113. Cleanup();
  114. m_nWidth = nWidth;
  115. m_nHeight = nHeight;
  116. m_nDataType = nDataType;
  117. m_nChannels = nChannels;
  118. m_nElementSize = m_nChannels * GetDataTypeSize(m_nDataType);
  119. m_nRowSize = m_nWidth * m_nElementSize;
  120. if(pBuffer)
  121. m_pBuffer = pBuffer;
  122. else
  123. {
  124. m_pAlloc = new unsigned char[GetBufferSize() + ALIGN_MASK];
  125. m_pBuffer = (void *)ALIGN(m_pAlloc);
  126. }
  127. }
  128. void Cleanup()
  129. {
  130. if(m_pAlloc)
  131. {
  132. delete m_pAlloc;
  133. m_pAlloc = m_pBuffer = NULL;
  134. }
  135. }
  136. int GetWidth() const  { return m_nWidth; }
  137. int GetHeight() const { return m_nHeight; }
  138. int GetDataType() const { return m_nDataType; }
  139. int GetChannels() const { return m_nChannels; }
  140. int GetBufferSize() const { return m_nHeight * m_nRowSize; }
  141. void *GetBuffer() const { return m_pBuffer; }
  142. void ClearBuffer() { memset(m_pBuffer, 0, GetBufferSize()); }
  143. void SwapBuffers(C2DBuffer &buf)
  144. {
  145. void *pTemp;
  146. ASSERT(*this == buf);
  147. SWAP(m_pAlloc, buf.m_pAlloc, pTemp);
  148. SWAP(m_pBuffer, buf.m_pBuffer, pTemp);
  149. }
  150. };
  151. /*******************************************************************************
  152. * Class: CPixelBuffer
  153. ********************************************************************************
  154. * This class implements a general-purpose pixel buffer to be used for anything.
  155. * It is often used by CTexture to set up OpenGL textures, so many of the
  156. * parameters you use to initialize it look like the parameters you would pass
  157. * to glTexImage1D or glTexImage2D. Some of the standard pixel buffer routines
  158. * call fast MMX functions implemented in PixelBuffer.asm.
  159. *******************************************************************************/
  160. class CPixelBuffer : public C2DBuffer
  161. {
  162. protected:
  163. int m_nFormat; // The format of the pixel data (i.e. GL_LUMINANCE, GL_RGBA)
  164. public:
  165. CPixelBuffer() : C2DBuffer() {}
  166. CPixelBuffer(int nWidth, int nHeight, int nChannels=3, int nFormat=GL_RGB, int nDataType=UnsignedByteType) : C2DBuffer(nWidth, nHeight, nDataType, nChannels)
  167. {
  168. m_nFormat = nFormat;
  169. }
  170. int GetFormat() { return m_nFormat; }
  171. void Init(int nWidth, int nHeight, int nChannels=3, int nFormat=GL_RGB, int nDataType=GL_UNSIGNED_BYTE, void *pBuffer=NULL)
  172. {
  173. C2DBuffer::Init(nWidth, nHeight, nDataType, nChannels, pBuffer);
  174. m_nFormat = nFormat;
  175. }
  176. // Miscellaneous initalization routines
  177. void MakeGlow(float fExpose=5.75f, float fSizeDisc=0);
  178. void MakeGlow2(float fExpose=5.75f, float fSizeDisc=0);
  179. void Invert();
  180. };
  181. #endif // __PixelBuffer_h__