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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Windows编程 > DirextX编程 > 查看源码
CFilterTitleOverlay.cpp
资源名称：DirectShowcode.rar　[点击查看]
上传用户：hhs829
上传日期：2022-06-17
资源大小：586k
文件大小：13k
源码类别：

DirextX编程

开发平台：

Visual C++

CFilterTitleOverlay.cpp：源码内容
  1. //
  2. // CFilterTitleOverlay.cpp
  3. //
  5. #include <streams.h>          // quartz, includes windows
  6. // Eliminate two expected level 4 warnings from the Microsoft compiler.
  7. // The class does not have an assignment or copy operator, and so cannot
  8. // be passed by value.  This is normal.  This file compiles clean at the
  9. // highest (most picky) warning level (-W4).
  10. #pragma warning(disable: 4511 4512)
  12. #include <measure.h>          // performance measurement (MSR_)
  13. #include <initguid.h>
  15. #if (1100 > _MSC_VER)
  16. #include <olectlid.h>
  17. #else
  18. #include <olectl.h>
  19. #endif
  21. #include "CFilterTitleOverlay.h"
  22. #include "CTitleOverlayProp.h"
  23. #include "CSysTimeOverlayController.h"
  24. #include "CScrollController.h"
  26. // {E3FB4BFE-8E5C-4aec-8162-7DA55BE486A1}
  27. DEFINE_GUID(CLSID_HQTitleOverlay, 
  28. 0xe3fb4bfe, 0x8e5c, 0x4aec, 0x81, 0x62, 0x7d, 0xa5, 0x5b, 0xe4, 0x86, 0xa1);
  30. // {E70FE57A-19AA-4a4c-B39A-408D49D73851}
  31. DEFINE_GUID(CLSID_HQTitleOverlayProp, 
  32. 0xe70fe57a, 0x19aa, 0x4a4c, 0xb3, 0x9a, 0x40, 0x8d, 0x49, 0xd7, 0x38, 0x51);
  35. //
  36. // setup data
  37. //
  38. const AMOVIESETUP_MEDIATYPE sudPinTypes =
  39. {
  40.     &MEDIATYPE_NULL,            // Major type
  41.     &MEDIASUBTYPE_NULL          // Minor type
  42. };
  44. const AMOVIESETUP_PIN psudPins[] =
  45. {
  46.     {
  47.         L"Input",           // String pin name
  48.         FALSE,              // Is it rendered
  49.         FALSE,              // Is it an output
  50.         FALSE,              // Allowed none
  51.         FALSE,              // Allowed many
  52.         &CLSID_NULL,        // Connects to filter
  53.         L"Output",          // Connects to pin
  54.         1,                  // Number of types
  55.         &sudPinTypes },     // The pin details
  56.       { L"Output",          // String pin name
  57.         FALSE,              // Is it rendered
  58.         TRUE,               // Is it an output
  59.         FALSE,              // Allowed none
  60.         FALSE,              // Allowed many
  61.         &CLSID_NULL,        // Connects to filter
  62.         L"Input",           // Connects to pin
  63.         1,                  // Number of types
  64.         &sudPinTypes        // The pin details
  65.     }
  66. };
  69. const AMOVIESETUP_FILTER sudFilter =
  70. {
  71.     &CLSID_HQTitleOverlay,       // Filter CLSID
  72.     L"HQ Title Overlay Std.",    // Filter name
  73.     MERIT_DO_NOT_USE,        // Its merit
  74.     2,                       // Number of pins
  75.     psudPins                 // Pin details
  76. };
  79. // List of class IDs and creator functions for the class factory. This
  80. // provides the link between the OLE entry point in the DLL and an object
  81. // being created. The class factory will call the static CreateInstance
  82. CFactoryTemplate g_Templates[] = 
  83. {
  84.     { 
  85. L"HQ Title Overlay Std.",
  86. &CLSID_HQTitleOverlay,
  87. CFilterTitleOverlay::CreateInstance,
  88. NULL,
  89. &sudFilter 
  90. },
  91. { 
  92. L"HQ Title Overlay Property Page",
  93. &CLSID_HQTitleOverlayProp,
  94. CTitleOverlayProp::CreateInstance 
  95. }
  96. };
  97. int g_cTemplates = sizeof(g_Templates) / sizeof(g_Templates[0]);
  101. // ----------------------------------------------------------------------------
  102. //            Filter implementation
  103. // ----------------------------------------------------------------------------
  104. CFilterTitleOverlay::CFilterTitleOverlay(TCHAR *tszName, LPUNKNOWN punk, HRESULT *phr) : 
  105. CTransInPlaceFilter(tszName, punk, CLSID_HQTitleOverlay, phr)
  106. {
  107. mOverlayType = OT_STATIC;
  108. mOverlayController     = new COverlayController();
  109. mNeedEstimateFrameRate = FALSE;
  111. char szTitle[] = "Hello, DirectShow!"; 
  112. put_Title(szTitle, sizeof(szTitle));
  113. }
  115. CFilterTitleOverlay::~CFilterTitleOverlay()
  116. {
  117. ReleaseOverlayController();
  118. }
  120. void CFilterTitleOverlay::ReleaseOverlayController(void)
  121. {
  122. if (mOverlayController)
  123. {
  124. delete mOverlayController;
  125. mOverlayController = NULL;
  126. }
  127. }
  129. //
  130. // CreateInstance
  131. //
  132. // Override CClassFactory method.
  133. // Provide the way for COM to create a CNullInPlace object
  134. //
  135. CUnknown * WINAPI CFilterTitleOverlay::CreateInstance(LPUNKNOWN punk, HRESULT *phr) 
  136. {
  137. #if 1
  138. char    szCreatorPath[256], szCreatorName[256];
  139. ::strcpy(szCreatorPath, "");
  140. ::strcpy(szCreatorName, "");
  141. HMODULE hModule = ::GetModuleHandle(NULL);
  142. ::GetModuleFileName(hModule, szCreatorPath, 256);
  143. char * backSlash = ::strrchr(szCreatorPath, '\');
  144. if (backSlash)
  145. {
  146. strcpy(szCreatorName, backSlash);
  147. }
  148. ::_strlwr(szCreatorName);
  149. // Please specify your app name with lowercase
  150. if (::strstr(szCreatorName, "graphedt") == NULL &&
  151. ::strstr(szCreatorName, "ourapp") == NULL)
  152. {
  153. *phr = E_FAIL;
  154. return NULL;
  155. }
  156. #endif
  158. CFilterTitleOverlay *pNewObject = new CFilterTitleOverlay(NAME("TitleOverlay"), punk, phr);
  159. if (pNewObject == NULL) 
  160. {
  161. *phr = E_OUTOFMEMORY;
  162. }
  163. return pNewObject;
  164. } 
  166. //
  167. // Basic COM - used here to reveal our own interfaces
  168. STDMETHODIMP CFilterTitleOverlay::NonDelegatingQueryInterface(REFIID riid, void ** ppv)
  169. {
  170. CheckPointer(ppv, E_POINTER);
  172. if (riid == IID_ISpecifyPropertyPages) 
  173. {
  174. return GetInterface((ISpecifyPropertyPages *) this, ppv);
  175. }
  176. else if (riid == IID_ITitleOverlay)
  177. {
  178. return GetInterface((ITitleOverlay *) this, ppv);
  179. }
  180. else 
  181. {
  182. return CTransInPlaceFilter::NonDelegatingQueryInterface(riid, ppv);
  183. }
  184. } // NonDelegatingQueryInterface
  186. // Only RGB 32/24/565/555 supported 
  187. HRESULT CFilterTitleOverlay::CheckInputType(const CMediaType* mtIn)
  188. {
  189. // Dynamic format change will never be allowed!
  190. if (IsStopped() && *mtIn->Type() == MEDIATYPE_Video)
  191. {
  192. if (*mtIn->Subtype() == MEDIASUBTYPE_RGB32 ||
  193. *mtIn->Subtype() == MEDIASUBTYPE_RGB24 ||
  194. *mtIn->Subtype() == MEDIASUBTYPE_RGB555 ||
  195. *mtIn->Subtype() == MEDIASUBTYPE_RGB565)
  196. {
  197. return NOERROR;
  198. }
  199. }
  200. return E_INVALIDARG;
  201. }
  203. HRESULT CFilterTitleOverlay::Transform(IMediaSample *pSample)
  204. {
  205. // If we cann't read frame rate info from input pin's connection media type,
  206. // We estimate it from the first sample's time stamp!
  207. if (mNeedEstimateFrameRate)
  208. {
  209. mNeedEstimateFrameRate = FALSE;
  210. REFERENCE_TIME  startTime = 0;
  211. REFERENCE_TIME  endTime   = 0;
  212. double          estimated = 25;
  213. if (SUCCEEDED(pSample->GetTime(&startTime, &endTime)))
  214. {
  215. estimated = 1.0 * UNITS / (endTime - startTime);
  216. }
  217. mOverlayController->SetEstimatedFrameRate(estimated);
  218. }
  220. if (mOverlayType != OT_NONE)
  221. {
  222. PBYTE  pData = NULL;
  223. pSample->GetPointer(&pData);
  224. mOverlayController->DoTitleOverlay(pData);
  225. }
  227. return NOERROR;
  228. }
  230. HRESULT CFilterTitleOverlay::SetInputVideoInfoToController(void)
  231. {
  232. if (mOverlayController && m_pInput && m_pInput->IsConnected())
  233. {
  234. CMediaType  mt = m_pInput->CurrentMediaType();
  235. if (mt.formattype != FORMAT_VideoInfo)
  236. {
  237. return E_FAIL;
  238. }
  240. RGB_FORMAT   colorSpace = FT_NONE;
  241. if (mt.subtype == MEDIASUBTYPE_RGB32)  // Determine RGB format
  242. {
  243. colorSpace = FT_RGB32;
  244. }
  245. else if (mt.subtype == MEDIASUBTYPE_RGB24)
  246. {
  247. colorSpace = FT_RGB24;
  248. }
  249. else if (mt.subtype == MEDIASUBTYPE_RGB555)
  250. {
  251. colorSpace = FT_RGB555;
  252. }
  253. else if (mt.subtype == MEDIASUBTYPE_RGB565)
  254. {
  255. colorSpace = FT_RGB565;
  256. }
  257. else if (mt.subtype == MEDIASUBTYPE_RGB8)
  258. {
  259. colorSpace = FT_RGB8;
  260. }
  261. mOverlayController->SetInputColorSpace(colorSpace);
  262. VIDEOINFOHEADER * pHeader = (VIDEOINFOHEADER *) mt.pbFormat;
  263. mNeedEstimateFrameRate    = pHeader->AvgTimePerFrame > 0 ? FALSE : TRUE;
  264. mOverlayController->SetInputVideoInfo(pHeader);
  265. return NOERROR;
  266. }
  267. return E_FAIL;
  268. }
  270. HRESULT CFilterTitleOverlay::CompleteConnect(PIN_DIRECTION direction, IPin *pReceivePin)
  271. {
  272. HRESULT hr = CTransInPlaceFilter::CompleteConnect(direction, pReceivePin);
  274. if (SUCCEEDED(hr) && direction == PINDIR_INPUT)
  275. {
  276. hr = SetInputVideoInfoToController();
  277. }
  278. return hr;
  279. }
  281. HRESULT CFilterTitleOverlay::StartStreaming()
  282. {
  283. BOOL pass = mOverlayController->StartTitleOverlay();
  284. return pass ? S_OK : E_FAIL;
  285. }
  287. HRESULT CFilterTitleOverlay::StopStreaming()
  288. {
  289. mOverlayController->StopTitleOverlay();
  290. return NOERROR;
  291. }
  294. // --- ISpecifyPropertyPages ---
  295. STDMETHODIMP CFilterTitleOverlay::GetPages(CAUUID *pPages)
  296. {
  297. pPages->cElems = 1;
  298. pPages->pElems = (GUID *) CoTaskMemAlloc(sizeof(GUID));
  299. if (pPages->pElems == NULL) 
  300. {
  301. return E_OUTOFMEMORY;
  302. }
  303. *(pPages->pElems) = CLSID_HQTitleOverlayProp;
  304. return NOERROR;
  305. }
  307. // 当字符叠加的类型改变时，需要生成新的应用逻辑控制类对象实例，
  308. // 并且在新生成的控制对象上设置参数信息。
  309. void CFilterTitleOverlay::SideEffectOverlayTypeChanged(void)
  310. {
  311. ReleaseOverlayController();
  312. switch (mOverlayType)
  313. {
  314. case OT_SYSTIME:
  315. mOverlayController = new CSysTimeOverlayController();
  316. break;
  318. case OT_SCROLL_TOP:
  319. mOverlayController = new CScrollController();
  320. ((CScrollController*) mOverlayController)->SetScrollBottomOrTop(FALSE);
  321. break;
  322. case OT_SCROLL_BOTTOM:
  323. mOverlayController = new CScrollController();
  324. ((CScrollController*) mOverlayController)->SetScrollBottomOrTop(TRUE);
  325. break;
  327. case OT_NONE:
  328. case OT_STATIC:
  329. default:
  330. mOverlayController = new COverlayController();
  331. }
  332. // Make sure to set input video info to the new controller 
  333. SetInputVideoInfoToController();
  334. }
  336. // --- ITitleOverlay methods ---
  337. STDMETHODIMP CFilterTitleOverlay::put_TitleOverlayType(long inOverlayType)
  338. {
  339. CAutoLock   lockit(&mITitleOverlaySync);
  340. // When filter graph is active, change overlay type is not allowed!
  341. if (m_State != State_Stopped)
  342. {
  343. return E_UNEXPECTED;
  344. }
  346. if (mOverlayType != (OVERLAY_TYPE)inOverlayType)
  347. {
  348. mOverlayType = (OVERLAY_TYPE)inOverlayType;
  349. SideEffectOverlayTypeChanged();
  350. }
  351. return NOERROR;
  352. }
  354. STDMETHODIMP CFilterTitleOverlay::get_TitleOverlayType(long * outOverlayType)
  355. {
  356. CAutoLock   lockit(&mITitleOverlaySync);
  357. *outOverlayType = mOverlayType;
  358. return NOERROR;
  359. }
  361. STDMETHODIMP CFilterTitleOverlay::put_TitleOverlayStyle(int inUsingCover)
  362. {
  363. CAutoLock   lockit(&mITitleOverlaySync);
  364. mOverlayController->SetOverlayStyle(inUsingCover);
  365. return NOERROR;
  366. }
  368. STDMETHODIMP CFilterTitleOverlay::get_TitleOverlayStyle(int * outUsingCover)
  369. {
  370. CAutoLock   lockit(&mITitleOverlaySync);
  371. mOverlayController->GetOverlayStyle(outUsingCover);
  372. return NOERROR;
  373. }
  375. STDMETHODIMP CFilterTitleOverlay::put_Title(const char * inTitle, int inLength)
  376. {
  377. CAutoLock   lockit(&mITitleOverlaySync);
  378. // When filter graph is active, change title text is not allowed!
  379. if (m_State != State_Stopped)
  380. {
  381. return E_UNEXPECTED;
  382. }
  383. mOverlayController->SetTitle(inTitle, inLength);
  384. return NOERROR;
  385. }
  387. STDMETHODIMP CFilterTitleOverlay::get_Title(char * outBuffer, int * outLength)
  388. {
  389. CAutoLock   lockit(&mITitleOverlaySync);
  390. *outLength = mOverlayController->GetTitle(outBuffer);
  391. return NOERROR;
  392. }
  394. STDMETHODIMP CFilterTitleOverlay::put_TitleColor(BYTE inR, BYTE inG, BYTE inB)
  395. {
  396. CAutoLock   lockit(&mITitleOverlaySync);
  397. mOverlayController->SetTitleColor(inR, inG, inB);
  398. return NOERROR;
  399. }
  401. STDMETHODIMP CFilterTitleOverlay::get_TitleColor(BYTE * outR, BYTE * outG, BYTE * outB)
  402. {
  403. CAutoLock   lockit(&mITitleOverlaySync);
  404. mOverlayController->GetTitleColor(outR, outG, outB);
  405. return NOERROR;
  406. }
  408. STDMETHODIMP CFilterTitleOverlay::put_TitleStartPosition(POINT inStartPos)
  409. {
  410. CAutoLock   lockit(&mITitleOverlaySync);
  411. mOverlayController->SetTitleStartPosition(inStartPos);
  412. return NOERROR;
  413. }
  415. STDMETHODIMP CFilterTitleOverlay::get_TitleStartPosition(POINT * outStartPos)
  416. {
  417. CAutoLock   lockit(&mITitleOverlaySync);
  418. mOverlayController->GetTitleStartPosition(outStartPos);
  419. return NOERROR;
  420. }
  422. STDMETHODIMP CFilterTitleOverlay::put_TitleFont(LOGFONT inFont)
  423. {
  424. CAutoLock   lockit(&mITitleOverlaySync);
  425. mOverlayController->SetTitleFont(inFont);
  426. return NOERROR;
  427. }
  429. STDMETHODIMP CFilterTitleOverlay::get_TitleFont(LOGFONT * outFont)
  430. {
  431. CAutoLock   lockit(&mITitleOverlaySync);
  432. mOverlayController->GetTitleFont(outFont);
  433. return NOERROR;
  434. }
  436. STDMETHODIMP CFilterTitleOverlay::put_TitleDuration(double inStart, double inEnd)
  437. {
  438. CAutoLock   lockit(&mITitleOverlaySync);
  439. mOverlayController->SetTitleDuration(inStart, inEnd);
  440. return NOERROR;
  441. }
  443. STDMETHODIMP CFilterTitleOverlay::get_TitleDuration(double * outStart, double * outEnd)
  444. {
  445. CAutoLock   lockit(&mITitleOverlaySync);
  446. mOverlayController->GetTitleDuration(outStart, outEnd);
  447. return NOERROR;
  448. }
  458. /******************************Public Routine******************************
  459. * exported entry points for registration and
  460. * unregistration (in this case they only call
  461. * through to default implmentations).
  462. *
  463. *
  464. *
  465. * History:
  466. *
  467. **************************************************************************/
  468. STDAPI DllRegisterServer()
  469. {
  470. return AMovieDllRegisterServer2( TRUE );
  471. }
  473. STDAPI DllUnregisterServer()
  474. {
  475. return AMovieDllRegisterServer2( FALSE );
  476. }
  478. // Microsoft C Compiler will give hundreds of warnings about
  479. // unused inline functions in header files.  Try to disable them.
  480. #pragma warning( disable:4514)