开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > Windows编程 > 多媒体编程 > 查看源码
Copy (2) of QTDec.cpp
资源名称：fullMPCcode.rar　[点击查看]
上传用户：xjjlds
上传日期：2015-12-05
资源大小：22823k
文件大小：13k
源码类别：

多媒体编程

开发平台：

Visual C++

Copy (2) of QTDec.cpp：源码内容
  1. #include "StdAfx.h"
  2. #include <atlbase.h>
  3. #include "......DSUtilMediaTypes.h"
  4. #include <initguid.h>
  5. #include "qtdec.h"
  7. using namespace QT;
  9. static DWORD DWSWAP(DWORD dw) {return(((dw&0xff)<<24)|((dw&0xff00)<<8)|((dw&0xff0000)>>8)|((dw&0xff000000)>>24));}
  11. #ifdef REGISTER_FILTER
  13. const AMOVIESETUP_MEDIATYPE sudPinTypesIn[] =
  14. {
  15. {&MEDIATYPE_Video, &MEDIASUBTYPE_NULL},
  16. };
  18. const AMOVIESETUP_MEDIATYPE sudPinTypesOut[] =
  19. {
  20. {&MEDIATYPE_Video, &MEDIASUBTYPE_NULL},
  21. };
  23. const AMOVIESETUP_PIN sudpPins[] =
  24. {
  25.     { L"Input",             // Pins string name
  26.       FALSE,                // Is it rendered
  27.       FALSE,                // Is it an output
  28.       FALSE,                // Are we allowed none
  29.       FALSE,                // And allowed many
  30.       &CLSID_NULL,          // Connects to filter
  31.       NULL,                 // Connects to pin
  32.       countof(sudPinTypesIn), // Number of types
  33.       sudPinTypesIn // Pin information
  34.     },
  35.     { L"Output",            // Pins string name
  36.       FALSE,                // Is it rendered
  37.       TRUE,                 // Is it an output
  38.       FALSE,                // Are we allowed none
  39.       FALSE,                // And allowed many
  40.       &CLSID_NULL,          // Connects to filter
  41.       NULL,                 // Connects to pin
  42.       countof(sudPinTypesOut), // Number of types
  43.       sudPinTypesOut // Pin information
  44.     }
  45. };
  47. const AMOVIESETUP_FILTER sudFilter =
  48. {
  49.     &CLSID_QTDec, // Filter CLSID
  50.     L"QuickTime Decoder", // String name
  51.     MERIT_DO_NOT_USE/*MERIT_NORMAL*/, // Filter merit
  52.     countof(sudpPins), // Number of pins
  53.     sudpPins                // Pin information
  54. };
  56. CFactoryTemplate g_Templates[] =
  57. {
  58.     { L"QuickTime Decoder"
  59.     , &CLSID_QTDec
  60.     , CQTDec::CreateInstance
  61.     , NULL
  62.     , &sudFilter }
  63. };
  65. int g_cTemplates = countof(g_Templates);
  67. STDAPI DllRegisterServer()
  68. {
  69. return AMovieDllRegisterServer2(TRUE);
  70. }
  72. STDAPI DllUnregisterServer()
  73. {
  74. return AMovieDllRegisterServer2(FALSE);
  75. }
  77. extern "C" BOOL WINAPI DllEntryPoint(HINSTANCE, ULONG, LPVOID);
  79. BOOL APIENTRY DllMain(HANDLE hModule, DWORD dwReason, LPVOID lpReserved)
  80. {
  81.     return DllEntryPoint((HINSTANCE)hModule, dwReason, 0); // "DllMain" of the dshow baseclasses;
  82. }
  84. //
  85. // CQTDec
  86. //
  88. CUnknown* WINAPI CQTDec::CreateInstance(LPUNKNOWN lpunk, HRESULT* phr)
  89. {
  90.     CUnknown* punk = new CQTDec(lpunk, phr);
  91.     if(punk == NULL) *phr = E_OUTOFMEMORY;
  92. return punk;
  93. }
  95. #endif
  97. CQTDec::CQTDec(LPUNKNOWN lpunk, HRESULT* phr)
  98. : CTransformFilter(NAME("CQTDec"), lpunk, CLSID_QTDec)
  99. , m_fQtInitialized(false)
  100. , m_pImageGWorld(NULL)
  101. , m_hImageDesc(NULL)
  102. , m_cinst(NULL)
  103. {
  104. if(phr) *phr = S_OK;
  106. m_fQtInitialized = false;
  107. if(InitializeQTML(0) != 0) {if(phr) *phr = E_FAIL; return;}
  108. // if(EnterMovies() != 0) {TerminateQTML(); if(phr) *phr = E_FAIL; return;}
  109. m_fQtInitialized = true;
  110. }
  112. CQTDec::~CQTDec()
  113. {
  114. if(m_cinst)
  115. CloseComponent(m_cinst),
  116. m_cinst = NULL;
  118. if(m_hImageDesc)
  119. DisposeHandle((Handle)m_hImageDesc), 
  120. m_hImageDesc = NULL;
  122. FreeGWorld(m_pImageGWorld);
  124. if(m_fQtInitialized)
  125. {
  126. // ExitMovies();
  127. TerminateQTML();
  128. }
  129. }
  131. //
  133. GWorldPtr CQTDec::MakeGWorld()
  134. {
  135. if(!m_pOutput->IsConnected())
  136. return NULL;
  138. const CMediaType& mt = m_pOutput->CurrentMediaType();
  139. VIDEOINFOHEADER* vih = (VIDEOINFOHEADER*)mt.pbFormat;
  140. BITMAPINFOHEADER& bih = vih->bmiHeader;
  142. GWorldPtr pImageWorld = NULL;
  143. Rect rect = {0, 0, (short)abs(bih.biHeight), (short)bih.biWidth};
  145. OSType PixelFormat = 
  146. mt.subtype == MEDIASUBTYPE_YUY2 ? kYUVSPixelFormat : // 'yuvs'
  147. mt.subtype == MEDIASUBTYPE_UYVY ? k2vuyPixelFormat : // '2vuy'
  148. mt.subtype == MEDIASUBTYPE_RGB32 ? k32BGRAPixelFormat : // 'BGRA'
  149. mt.subtype == MEDIASUBTYPE_RGB565 ? k16LE565PixelFormat : // 'L565'
  150. mt.subtype == MEDIASUBTYPE_RGB555 ? k16LE555PixelFormat : // 'L555'
  151. 0;
  153. if(!PixelFormat || noErr != QTNewGWorld(&pImageWorld, PixelFormat, &rect, NULL, NULL, 0))
  154. pImageWorld = NULL;
  156. if(pImageWorld)
  157.         LockPixels(GetGWorldPixMap(pImageWorld));
  159. return pImageWorld;
  160. }
  162. void CQTDec::FreeGWorld(GWorldPtr& pImageGWorld)
  163. {
  164. if(pImageGWorld)
  165. {
  166.         UnlockPixels(GetGWorldPixMap(pImageGWorld));
  167. DisposeGWorld(pImageGWorld);
  168. pImageGWorld = NULL;
  169. }
  170. }
  172. bool CQTDec::InitComponent()
  173. {
  174. if(m_cinst)
  175. CloseComponent(m_cinst),
  176. m_cinst = NULL;
  178. if(m_hImageDesc)
  179. DisposeHandle((Handle)m_hImageDesc), 
  180. m_hImageDesc = NULL;
  182. if(!m_pInput->IsConnected())
  183. return NULL;
  185. BITMAPINFOHEADER& bih = ((VIDEOINFOHEADER*)m_pInput->CurrentMediaType().pbFormat)->bmiHeader;
  187. ComponentDescription cd = {decompressorComponentType, DWSWAP(bih.biCompression), 0, 0, 0};
  188. Component c = FindNextComponent(0, &cd);
  189. if(!c) return(false);
  191. m_cinst = OpenComponent(c);
  192. if(!m_cinst) return(false);
  194. ComponentResult cres;
  196.     ImageSubCodecDecompressCapabilities icap;
  197.     memset(&icap, 0, sizeof(icap));
  198.     cres = ImageCodecInitialize(m_cinst, &icap);
  200.     CodecInfo cinfo;
  201.     memset(&cinfo, 0, sizeof(cinfo));
  202.     cres = ImageCodecGetCodecInfo(m_cinst, &cinfo);
  204. m_hImageDesc = (ImageDescriptionHandle)NewHandleClear(sizeof(ImageDescription));
  205. if(!m_hImageDesc)
  206. return(false);
  208.     (**m_hImageDesc).idSize = sizeof(ImageDescription);
  209. (**m_hImageDesc).cType = DWSWAP(bih.biCompression);
  210.     (**m_hImageDesc).temporalQuality = 0;
  211.     (**m_hImageDesc).spatialQuality = codecNormalQuality;
  212. (**m_hImageDesc).width = (short)bih.biWidth;
  213.     (**m_hImageDesc).height = (short)abs(bih.biHeight);
  214.     (**m_hImageDesc).hRes = 72 << 16;
  215.     (**m_hImageDesc).vRes = 72 << 16;
  216.     (**m_hImageDesc).frameCount = 1; // ?
  217. (**m_hImageDesc).depth = bih.biBitCount; // should be 24 for unknown/compressed types
  218.     (**m_hImageDesc).clutID = -1;
  220. memset(&m_cdpar, 0, sizeof(m_cdpar));
  221.     m_cdpar.imageDescription = m_hImageDesc;
  222.     m_cdpar.startLine = 0;
  223.     m_cdpar.stopLine = (**m_hImageDesc).height;
  224.     m_cdpar.frameNumber = 1;
  225.     m_cdpar.matrixFlags = 0;
  226.     m_cdpar.matrixType = 0;
  227.     m_cdpar.matrix = 0;
  228.     m_cdpar.capabilities = &m_ccap;
  229. memset(&m_ccap, 0, sizeof(m_ccap));
  230.     m_cdpar.accuracy = codecNormalQuality;
  231.     m_cdpar.port = m_pImageGWorld;
  232. Rect rect = {0, 0, (**m_hImageDesc).height, (**m_hImageDesc).width};
  233. m_cdpar.srcRect = rect;
  234.     m_cdpar.transferMode = srcCopy;
  235.     m_cdpar.dstPixMap = **GetGWorldPixMap(m_pImageGWorld);
  237.     cres = ImageCodecPreDecompress(m_cinst, &m_cdpar);
  239. return(true);
  240. }
  242. //
  244. HRESULT CQTDec::BreakConnect(PIN_DIRECTION dir)
  245. {
  246. if(dir == PINDIR_INPUT)
  247. {
  248. m_mts.RemoveAll();
  249. }
  250. else if(dir == PINDIR_OUTPUT)
  251. {
  252. FreeGWorld(m_pImageGWorld);
  253. }
  255. return __super::BreakConnect(dir);
  256. }
  258. HRESULT CQTDec::CompleteConnect(PIN_DIRECTION dir, IPin* pReceivePin)
  259. {
  260. if(dir == PINDIR_INPUT)
  261. {
  262. m_mts.RemoveAll();
  264. VIDEOINFOHEADER* vihin = (VIDEOINFOHEADER*)m_pInput->CurrentMediaType().pbFormat;
  265. BITMAPINFOHEADER& bihin = vihin->bmiHeader;
  267. CMediaType mt;
  268. mt.majortype = MEDIATYPE_Video;
  269. mt.subtype = MEDIASUBTYPE_None;
  270. mt.formattype = FORMAT_VideoInfo;
  271. mt.bFixedSizeSamples = TRUE;
  272. mt.bTemporalCompression = FALSE;
  273. mt.lSampleSize = 0;
  274. mt.pUnk = NULL;
  276. VIDEOINFOHEADER vih;
  277. memset(&vih, 0, sizeof(vih));
  278. vih.AvgTimePerFrame = vihin->AvgTimePerFrame;
  279. vih.rcSource = vihin->rcSource;
  280. vih.rcTarget = vihin->rcTarget;
  281. vih.dwBitRate = vihin->dwBitRate;
  282. vih.dwBitErrorRate = vihin->dwBitErrorRate;
  284. BITMAPINFOHEADER& bih = vih.bmiHeader;
  285. bih.biSize = sizeof(bih);
  286. bih.biWidth = bihin.biWidth;
  287. bih.biHeight = abs(bihin.biHeight);
  288. bih.biPlanes = 1;
  289. bih.biXPelsPerMeter = bih.biYPelsPerMeter = 0;
  290. bih.biClrUsed = bih.biClrImportant = 0;
  292. // if(fRGB32) // always can decompress to (?)
  293. {
  294. VIDEOINFOHEADER* vihout = (VIDEOINFOHEADER*)mt.AllocFormatBuffer(sizeof(VIDEOINFOHEADER));
  295. memcpy(vihout, &vih, sizeof(vih));
  297. BITMAPINFOHEADER& bihout = vihout->bmiHeader;
  298. bihout.biBitCount = 32;
  299. bihout.biSizeImage = bihout.biWidth*abs(bihout.biHeight)*bihout.biBitCount>>3;
  301. mt.subtype = MEDIASUBTYPE_RGB32;
  303. ((VIDEOINFOHEADER*)mt.pbFormat)->bmiHeader.biCompression = BI_BITFIELDS;
  304. ((VIDEOINFOHEADER*)mt.pbFormat)->bmiHeader.biHeight = -bih.biHeight;
  305. CorrectMediaType(&mt);
  306. m_mts.Add(mt);
  308. ((VIDEOINFOHEADER*)mt.pbFormat)->bmiHeader.biCompression = BI_RGB;
  309. ((VIDEOINFOHEADER*)mt.pbFormat)->bmiHeader.biHeight = bih.biHeight;
  310. CorrectMediaType(&mt);
  311. m_mts.Add(mt);
  312. }
  314. // if(fRGB16) // always can decompress to (?)
  315. {
  316. VIDEOINFOHEADER* vihout = (VIDEOINFOHEADER*)mt.AllocFormatBuffer(sizeof(VIDEOINFOHEADER));
  317. memcpy(vihout, &vih, sizeof(vih));
  319. BITMAPINFOHEADER& bihout = vihout->bmiHeader;
  320. bihout.biBitCount = 16;
  321. bihout.biSizeImage = bihout.biWidth*abs(bihout.biHeight)*bihout.biBitCount>>3;
  323. mt.subtype = MEDIASUBTYPE_RGB565;
  325. ((VIDEOINFOHEADER*)mt.pbFormat)->bmiHeader.biCompression = BI_BITFIELDS;
  326. ((VIDEOINFOHEADER*)mt.pbFormat)->bmiHeader.biHeight = -bih.biHeight;
  327. CorrectMediaType(&mt);
  328. m_mts.Add(mt);
  330. ((VIDEOINFOHEADER*)mt.pbFormat)->bmiHeader.biCompression = BI_RGB;
  331. ((VIDEOINFOHEADER*)mt.pbFormat)->bmiHeader.biHeight = bih.biHeight;
  332. CorrectMediaType(&mt);
  333. m_mts.Add(mt);
  335. mt.subtype = MEDIASUBTYPE_RGB555;
  337. ((VIDEOINFOHEADER*)mt.pbFormat)->bmiHeader.biCompression = BI_BITFIELDS;
  338. ((VIDEOINFOHEADER*)mt.pbFormat)->bmiHeader.biHeight = -bih.biHeight;
  339. CorrectMediaType(&mt);
  340. m_mts.Add(mt);
  342. ((VIDEOINFOHEADER*)mt.pbFormat)->bmiHeader.biCompression = BI_RGB;
  343. ((VIDEOINFOHEADER*)mt.pbFormat)->bmiHeader.biHeight = bih.biHeight;
  344. CorrectMediaType(&mt);
  345. m_mts.Add(mt);
  346. }
  347. }
  348. else if(dir == PINDIR_OUTPUT)
  349. {
  350. FreeGWorld(m_pImageGWorld);
  351. m_pImageGWorld = MakeGWorld();
  352. }
  354. return __super::CompleteConnect(dir, pReceivePin);
  355. }
  357. HRESULT CQTDec::Transform(IMediaSample* pSample, IMediaSample* pOutSample)
  358. {
  359. HRESULT hr;
  361. BYTE* pIn = NULL;
  362. if(FAILED(hr = pSample->GetPointer(&pIn))) return hr;
  363. long len = pSample->GetActualDataLength();
  364. if(len <= 0) return S_FALSE;
  366. BYTE* pOut = NULL;
  367. if(FAILED(hr = pOutSample->GetPointer(&pOut))) return hr;
  368. int size = pOutSample->GetSize();
  370. VIDEOINFOHEADER* vih = (VIDEOINFOHEADER*)m_pOutput->CurrentMediaType().pbFormat;
  371. BITMAPINFOHEADER& bih = vih->bmiHeader;
  373. bool fInitialized = true;
  375. if(m_cinst)
  376. {
  377. AM_MEDIA_TYPE* pmt = NULL;
  378. if(S_OK == pSample->GetMediaType((AM_MEDIA_TYPE**)&pmt) && pmt)
  379. {
  380. CMediaType mt(*pmt);
  381. DeleteMediaType(pmt), pmt = NULL;
  383. if(mt != m_pInput->CurrentMediaType())
  384. {
  385. m_pInput->SetMediaType(&mt);
  386. fInitialized = InitComponent();
  387. }
  388. }
  389. }
  390. else
  391. {
  392. fInitialized = InitComponent();
  393. }
  395. if(!fInitialized)
  396. return E_FAIL;
  398. m_cdpar.data = (Ptr)pIn;
  399. m_cdpar.bufferSize = len;
  400. (**m_cdpar.imageDescription).dataSize = len;
  401.     ComponentResult cres = ImageCodecBandDecompress(m_cinst, &m_cdpar);
  402.     m_cdpar.frameNumber++;
  404. if(cres == noErr)
  405. {
  406. PixMapHandle hPixMap = GetGWorldPixMap(m_pImageGWorld);
  407. Ptr pPixels = GetPixBaseAddr(hPixMap);
  408. long theRowBytes = QTGetPixMapHandleRowBytes(hPixMap);
  410. DWORD pitch = bih.biWidth*bih.biBitCount>>3;
  412. for(int i = 0, h = abs(bih.biHeight); i < h; i++)
  413. {
  414. memcpy(pOut + i*pitch, (BYTE*)pPixels + i*theRowBytes, min(pitch, theRowBytes));
  415. }
  416. }
  418. return S_OK;
  419. }
  421. HRESULT CQTDec::CheckInputType(const CMediaType* mtIn)
  422. {
  423. if(mtIn->majortype != MEDIATYPE_Video || mtIn->formattype != FORMAT_VideoInfo)
  424. return VFW_E_TYPE_NOT_ACCEPTED;
  426. VIDEOINFOHEADER* vih = (VIDEOINFOHEADER*)mtIn->pbFormat;
  427. BITMAPINFOHEADER& bih = vih->bmiHeader;
  429. OSErr err;
  430. ComponentInstance ci;
  431. if(noErr == (err = OpenADefaultComponent('imdc', DWSWAP(bih.biCompression), &ci)))
  432. {
  433. err = CloseComponent(ci);
  434. return S_OK;
  435. }
  437. return VFW_E_TYPE_NOT_ACCEPTED;
  438. }
  440. HRESULT CQTDec::CheckTransform(const CMediaType* mtIn, const CMediaType* mtOut)
  441. {
  442. return S_OK;
  443. // TODO
  444. return VFW_E_TYPE_NOT_ACCEPTED;
  445. }
  447. HRESULT CQTDec::DecideBufferSize(IMemAllocator* pAllocator, ALLOCATOR_PROPERTIES* pProperties)
  448. {
  449. if(m_pInput->IsConnected() == FALSE) return E_UNEXPECTED;
  451. CComPtr<IMemAllocator> pAllocatorIn;
  452. m_pInput->GetAllocator(&pAllocatorIn);
  453. if(!pAllocatorIn) return E_UNEXPECTED;
  455. pAllocatorIn->GetProperties(pProperties);
  457. VIDEOINFOHEADER* vih = (VIDEOINFOHEADER*)m_pInput->CurrentMediaType().pbFormat;
  458. BITMAPINFOHEADER& bih = vih->bmiHeader;
  460. pProperties->cBuffers = 1;
  461. pProperties->cbBuffer = bih.biWidth*abs(bih.biHeight)*4; // TODO
  462. pProperties->cbAlign = 1;
  463. pProperties->cbPrefix = 0;
  465. HRESULT hr;
  466. ALLOCATOR_PROPERTIES Actual;
  467.     if(FAILED(hr = pAllocator->SetProperties(pProperties, &Actual))) 
  468. return hr;
  470.     return(pProperties->cBuffers > Actual.cBuffers || pProperties->cbBuffer > Actual.cbBuffer
  471. ? E_FAIL
  472. : NOERROR);
  473. }
  475. HRESULT CQTDec::GetMediaType(int iPosition, CMediaType* pmt)
  476. {
  477.     if(m_pInput->IsConnected() == FALSE) return E_UNEXPECTED;
  479. if(iPosition < 0) return E_INVALIDARG;
  480. if(iPosition >= m_mts.GetCount()) return VFW_S_NO_MORE_ITEMS;
  482. *pmt = m_mts[iPosition];
  484. return S_OK;
  485. }