amvideo.cpp
上传用户:liguizhu
上传日期:2015-11-01
资源大小:2422k
文件大小:9k
源码类别:

P2P编程

开发平台:

Visual C++

  1. //------------------------------------------------------------------------------
  2. // File: AMVideo.cpp
  3. //
  4. // Desc: DirectShow base classes - implements helper functions for
  5. //       bitmap formats.
  6. //
  7. // Copyright (c) Microsoft Corporation.  All rights reserved.
  8. //------------------------------------------------------------------------------
  9. #include <streams.h>
  10. #include <limits.h>
  11. // These are bit field masks for true colour devices
  12. const DWORD bits555[] = {0x007C00,0x0003E0,0x00001F};
  13. const DWORD bits565[] = {0x00F800,0x0007E0,0x00001F};
  14. const DWORD bits888[] = {0xFF0000,0x00FF00,0x0000FF};
  15. // This maps bitmap subtypes into a bits per pixel value and also a
  16. // name. unicode and ansi versions are stored because we have to
  17. // return a pointer to a static string.
  18. const struct {
  19.     const GUID *pSubtype;
  20.     WORD BitCount;
  21.     CHAR *pName;
  22.     WCHAR *wszName;
  23. } BitCountMap[] =  { &MEDIASUBTYPE_RGB1,        1,   "RGB Monochrome",     L"RGB Monochrome",   
  24.                      &MEDIASUBTYPE_RGB4,        4,   "RGB VGA",            L"RGB VGA",          
  25.                      &MEDIASUBTYPE_RGB8,        8,   "RGB 8",              L"RGB 8",            
  26.                      &MEDIASUBTYPE_RGB565,      16,  "RGB 565 (16 bit)",   L"RGB 565 (16 bit)", 
  27.                      &MEDIASUBTYPE_RGB555,      16,  "RGB 555 (16 bit)",   L"RGB 555 (16 bit)", 
  28.                      &MEDIASUBTYPE_RGB24,       24,  "RGB 24",             L"RGB 24",           
  29.                      &MEDIASUBTYPE_RGB32,       32,  "RGB 32",             L"RGB 32",
  30.                      &MEDIASUBTYPE_ARGB32,    32,  "ARGB 32",             L"ARGB 32",
  31.                      &MEDIASUBTYPE_Overlay,     0,   "Overlay",            L"Overlay",          
  32.                      &GUID_NULL,                0,   "UNKNOWN",            L"UNKNOWN"           
  33. };
  34. // Return the size of the bitmap as defined by this header
  35. STDAPI_(DWORD) GetBitmapSize(const BITMAPINFOHEADER *pHeader)
  36. {
  37.     return DIBSIZE(*pHeader);
  38. }
  39. // This is called if the header has a 16 bit colour depth and needs to work
  40. // out the detailed type from the bit fields (either RGB 565 or RGB 555)
  41. STDAPI_(const GUID) GetTrueColorType(const BITMAPINFOHEADER *pbmiHeader)
  42. {
  43.     BITMAPINFO *pbmInfo = (BITMAPINFO *) pbmiHeader;
  44.     ASSERT(pbmiHeader->biBitCount == 16);
  45.     // If its BI_RGB then it's RGB 555 by default
  46.     if (pbmiHeader->biCompression == BI_RGB) {
  47.         return MEDIASUBTYPE_RGB555;
  48.     }
  49.     // Compare the bit fields with RGB 555
  50.     DWORD *pMask = (DWORD *) pbmInfo->bmiColors;
  51.     if (pMask[0] == bits555[0]) {
  52.         if (pMask[1] == bits555[1]) {
  53.             if (pMask[2] == bits555[2]) {
  54.                 return MEDIASUBTYPE_RGB555;
  55.             }
  56.         }
  57.     }
  58.     // Compare the bit fields with RGB 565
  59.     pMask = (DWORD *) pbmInfo->bmiColors;
  60.     if (pMask[0] == bits565[0]) {
  61.         if (pMask[1] == bits565[1]) {
  62.             if (pMask[2] == bits565[2]) {
  63.                 return MEDIASUBTYPE_RGB565;
  64.             }
  65.         }
  66.     }
  67.     return GUID_NULL;
  68. }
  69. // Given a BITMAPINFOHEADER structure this returns the GUID sub type that is
  70. // used to describe it in format negotiations. For example a video codec fills
  71. // in the format block with a VIDEOINFO structure, it also fills in the major
  72. // type with MEDIATYPE_VIDEO and the subtype with a GUID that matches the bit
  73. // count, for example if it is an eight bit image then MEDIASUBTYPE_RGB8
  74. STDAPI_(const GUID) GetBitmapSubtype(const BITMAPINFOHEADER *pbmiHeader)
  75. {
  76.     ASSERT(pbmiHeader);
  77.     // If it's not RGB then create a GUID from the compression type
  78.     if (pbmiHeader->biCompression != BI_RGB) {
  79.         if (pbmiHeader->biCompression != BI_BITFIELDS) {
  80.             FOURCCMap FourCCMap(pbmiHeader->biCompression);
  81.             return (const GUID) FourCCMap;
  82.         }
  83.     }
  84.     // Map the RGB DIB bit depth to a image GUID
  85.     switch(pbmiHeader->biBitCount) {
  86.         case 1    :   return MEDIASUBTYPE_RGB1;
  87.         case 4    :   return MEDIASUBTYPE_RGB4;
  88.         case 8    :   return MEDIASUBTYPE_RGB8;
  89.         case 16   :   return GetTrueColorType(pbmiHeader);
  90.         case 24   :   return MEDIASUBTYPE_RGB24;
  91.         case 32   :   return MEDIASUBTYPE_RGB32;
  92.     }
  93.     return GUID_NULL;
  94. }
  95. // Given a video bitmap subtype we return the number of bits per pixel it uses
  96. // We return a WORD bit count as thats what the BITMAPINFOHEADER uses. If the
  97. // GUID subtype is not found in the table we return an invalid USHRT_MAX
  98. STDAPI_(WORD) GetBitCount(const GUID *pSubtype)
  99. {
  100.     ASSERT(pSubtype);
  101.     const GUID *pMediaSubtype;
  102.     INT iPosition = 0;
  103.     // Scan the mapping list seeing if the source GUID matches any known
  104.     // bitmap subtypes, the list is terminated by a GUID_NULL entry
  105.     while (TRUE) {
  106.         pMediaSubtype = BitCountMap[iPosition].pSubtype;
  107.         if (IsEqualGUID(*pMediaSubtype,GUID_NULL)) {
  108.             return USHRT_MAX;
  109.         }
  110.         if (IsEqualGUID(*pMediaSubtype,*pSubtype)) {
  111.             return BitCountMap[iPosition].BitCount;
  112.         }
  113.         iPosition++;
  114.     }
  115. }
  116. // Given a bitmap subtype we return a description name that can be used for
  117. // debug purposes. In a retail build this function still returns the names
  118. // If the subtype isn't found in the lookup table we return string UNKNOWN
  119. int LocateSubtype(const GUID *pSubtype)
  120. {
  121.     ASSERT(pSubtype);
  122.     const GUID *pMediaSubtype;
  123.     INT iPosition = 0;
  124.     // Scan the mapping list seeing if the source GUID matches any known
  125.     // bitmap subtypes, the list is terminated by a GUID_NULL entry
  126.     while (TRUE) {
  127.         pMediaSubtype = BitCountMap[iPosition].pSubtype;
  128.         if (IsEqualGUID(*pMediaSubtype,*pSubtype) ||
  129.             IsEqualGUID(*pMediaSubtype,GUID_NULL)
  130.             )
  131.         {
  132.             break;
  133.         }
  134.         
  135.         iPosition++;
  136.     }
  137.     return iPosition;
  138. }
  139. STDAPI_(WCHAR *) GetSubtypeNameW(const GUID *pSubtype)
  140. {
  141.     return BitCountMap[LocateSubtype(pSubtype)].wszName;
  142. }
  143. STDAPI_(CHAR *) GetSubtypeNameA(const GUID *pSubtype)
  144. {
  145.     return BitCountMap[LocateSubtype(pSubtype)].pName;
  146. }
  147. #ifndef GetSubtypeName
  148. #error wxutil.h should have defined GetSubtypeName
  149. #endif
  150. #undef GetSubtypeName
  151. // this is here for people that linked to it directly; most people
  152. // would use the header file that picks the A or W version.
  153. STDAPI_(CHAR *) GetSubtypeName(const GUID *pSubtype)
  154. {
  155.     return GetSubtypeNameA(pSubtype);
  156. }
  157. // The mechanism for describing a bitmap format is with the BITMAPINFOHEADER
  158. // This is really messy to deal with because it invariably has fields that
  159. // follow it holding bit fields, palettes and the rest. This function gives
  160. // the number of bytes required to hold a VIDEOINFO that represents it. This
  161. // count includes the prefix information (like the rcSource rectangle) the
  162. // BITMAPINFOHEADER field, and any other colour information on the end.
  163. //
  164. // WARNING If you want to copy a BITMAPINFOHEADER into a VIDEOINFO always make
  165. // sure that you use the HEADER macro because the BITMAPINFOHEADER field isn't
  166. // right at the start of the VIDEOINFO (there are a number of other fields),
  167. //
  168. //     CopyMemory(HEADER(pVideoInfo),pbmi,sizeof(BITMAPINFOHEADER));
  169. //
  170. STDAPI_(LONG) GetBitmapFormatSize(const BITMAPINFOHEADER *pHeader)
  171. {
  172.     // Everyone has this to start with this  
  173.     LONG Size = SIZE_PREHEADER + pHeader->biSize;
  174.     ASSERT(pHeader->biSize >= sizeof(BITMAPINFOHEADER));
  175.     
  176.     // Does this format use a palette, if the number of colours actually used
  177.     // is zero then it is set to the maximum that are allowed for that colour
  178.     // depth (an example is 256 for eight bits). Truecolour formats may also
  179.     // pass a palette with them in which case the used count is non zero
  180.     // This would scare me.
  181.     ASSERT(pHeader->biBitCount <= iPALETTE || pHeader->biClrUsed == 0);
  182.     if (pHeader->biBitCount <= iPALETTE || pHeader->biClrUsed) {
  183.         LONG Entries = (DWORD) 1 << pHeader->biBitCount;
  184.         if (pHeader->biClrUsed) {
  185.             Entries = pHeader->biClrUsed;
  186.         }
  187.         Size += Entries * sizeof(RGBQUAD);
  188.     }
  189.     // Truecolour formats may have a BI_BITFIELDS specifier for compression
  190.     // type which means that room for three DWORDs should be allocated that
  191.     // specify where in each pixel the RGB colour components may be found
  192.     if (pHeader->biCompression == BI_BITFIELDS) {
  193.         Size += SIZE_MASKS;
  194.     }
  195.     // A BITMAPINFO for a palettised image may also contain a palette map that
  196.     // provides the information to map from a source palette to a destination
  197.     // palette during a BitBlt for example, because this information is only
  198.     // ever processed during drawing you don't normally store the palette map
  199.     // nor have any way of knowing if it is present in the data structure
  200.     return Size;
  201. }
  202. // Returns TRUE if the VIDEOINFO contains a palette
  203. STDAPI_(BOOL) ContainsPalette(const VIDEOINFOHEADER *pVideoInfo)
  204. {
  205.     if (PALETTISED(pVideoInfo) == FALSE) {
  206.         if (pVideoInfo->bmiHeader.biClrUsed == 0) {
  207.             return FALSE;
  208.         }
  209.     }
  210.     return TRUE;
  211. }
  212. // Return a pointer to the first entry in a palette
  213. STDAPI_(const RGBQUAD *) GetBitmapPalette(const VIDEOINFOHEADER *pVideoInfo)
  214. {
  215.     if (pVideoInfo->bmiHeader.biCompression == BI_BITFIELDS) {
  216.         return TRUECOLOR(pVideoInfo)->bmiColors;
  217.     }
  218.     return COLORS(pVideoInfo);
  219. }