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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > P2P编程 > 查看源码
TVPinStream.cpp
资源名称：p2p_vod.rar　[点击查看]
上传用户：liguizhu
上传日期：2015-11-01
资源大小：2422k
文件大小：22k
源码类别：

P2P编程

开发平台：

Visual C++

TVPinStream.cpp：源码内容
  1. #include "stdafx.h"
  2. #include "Dvdmedia.h"
  4. CTVStreamSourcePin::CTVStreamSourcePin(
  5.    TCHAR *pObjectName, HRESULT *phr, CSource *pFilter, bool isAudio)
  6. : CSourceStream(pObjectName, phr, pFilter, L"Out"), 
  7. m_pFilter((CTVStreamSource*)pFilter), 
  8. m_iVideoSampleSize(0), 
  9. m_iAudioSampleSize(0),
  10. m_pAnotherPin(NULL) ,
  11. m_bPinAvaible(TRUE) ,
  12.         m_bFlushforSync(0),
  13. CSourceSeeking(NAME("Chaos Filter Seeking Pin"), NULL, phr, pFilter->pStateLock())
  14. {
  15.     m_Data = new BYTE[65536];
  16. m_rtStart    = 0;
  17. m_rtStop     = 1 * UNITS;
  18. m_rtDuration = 1 * UNITS; 
  19. //
  20. m_isAudioPin = isAudio;
  21. //
  22. m_SeekData.rtSeekTo = _I64_MAX;
  23. m_SeekData.bSycn    = FALSE;
  24. //
  25. //
  26. SetSeekData(0);
  28. m_bSync = FALSE;
  30. m_bTryGetMediaType = true;
  31. m_MediaType.cbFormat = 0;
  32. // 必须是winxp 及其以上版本的OS
  33. if(m_pFilter->CanDynamicReconnect()) {
  34. // Winxp之前不能支持切换编码
  35. SetDefaultMediaType();
  36. }
  38. #ifdef USE_LOG
  39.     if(m_isAudioPin)
  40.         log = fopen("c:\chaosaudiopinlog.log", "w+");
  41.     else
  42.         log = fopen("c:\chaosvideopinlog.log", "w+");
  43.     
  44. fprintf(log, "start pinrn");
  45. fflush(log);
  46. #endif
  48. }
  49. //
  50. CTVStreamSourcePin::~CTVStreamSourcePin() 
  51. {
  52.     delete[] m_Data;
  53. #ifdef USE_LOG
  54. if(log)
  55. fclose(log);
  56. #endif
  57. }
  58. //
  59. void CTVStreamSourcePin::SetDefaultMediaType()
  60. {
  61. // raw video type
  62. static const BYTE videoData[] = {
  63. 0x76, 0x69, 0x64, 0x73, 0x00, 0x00, 0x10, 0x00, 0x80, 0x00, 0x00, 0xAA, 0x00, 0x38, 0x9B, 0x71, 
  64. 0x7D, 0xEB, 0x36, 0xE4, 0x4F, 0x52, 0xCE, 0x11, 0x9F, 0x53, 0x00, 0x20, 0xAF, 0x0B, 0xA7, 0x70, 
  65. 0x80, 0x9F, 0x58, 0x05, 0x56, 0xC3, 0xCE, 0x11, 0xBF, 0x01, 0x00, 0xAA, 0x00, 0x55, 0x59, 0x5A, 
  66. 0x01, 0x00, 0x00, 0x00, 0xC2, 0x02, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 
  67. 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 
  68. 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 
  69. 0x62, 0x17, 0x05, 0x00, 0x00, 0x00, 0x00, 0x00, 0x28, 0x00, 0x00, 0x00, 0x40, 0x01, 0x00, 0x00, 
  70. 0xF0, 0x00, 0x00, 0x00, 0x01, 0x00, 0x18, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x84, 0x03, 0x00, 
  71. 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00
  72. };
  74. /*
  75. // raw audio type
  76. static const BYTE audioData[] = {
  77. 0x61, 0x75, 0x64, 0x73, 0x00, 0x00, 0x10, 0x00, 0x80, 0x00, 0x00, 0xAA, 0x00, 0x38, 0x9B, 0x71, 
  78. 0x01, 0x00, 0x00, 0x00, 0x00, 0x00, 0x10, 0x00, 0x80, 0x00, 0x00, 0xAA, 0x00, 0x38, 0x9B, 0x71, 
  79. 0x81, 0x9F, 0x58, 0x05, 0x56, 0xC3, 0xCE, 0x11, 0xBF, 0x01, 0x00, 0xAA, 0x00, 0x55, 0x59, 0x5A, 
  80. 0x04, 0x00, 0x00, 0x00, 0x91, 0x00, 0x00, 0x00, 0x01, 0x00, 0x02, 0x00, 0x44, 0xAC, 0x00, 0x00, 
  81. 0x10, 0xB1, 0x02, 0x00, 0x04, 0x00, 0x10, 0x00, 0x00, 0x00
  82. };
  83. */
  85. // 直接无法创建，只能从现有的filter中寻找（所以默认的音频必须是GSM，这样才能找到）
  86. // gsm6.10
  87. static const BYTE audioData[] = {
  88. 0x61, 0x75, 0x64, 0x73, 0x00, 0x00, 0x10, 0x00, 0x80, 0x00, 0x00, 0xAA, 0x00, 0x38, 0x9B, 0x71, 
  89. 0x31, 0x00, 0x00, 0x00, 0x00, 0x00, 0x10, 0x00, 0x80, 0x00, 0x00, 0xAA, 0x00, 0x38, 0x9B, 0x71, 
  90. 0x81, 0x9F, 0x58, 0x05, 0x56, 0xC3, 0xCE, 0x11, 0xBF, 0x01, 0x00, 0xAA, 0x00, 0x55, 0x59, 0x5A, 
  91. 0x41, 0x00, 0x00, 0x00, 0xA1, 0x00, 0x00, 0x00, 0x31, 0x00, 0x01, 0x00, 0x44, 0xAC, 0x00, 0x00, 
  92. 0xFD, 0x22, 0x00, 0x00, 0x41, 0x00, 0x00, 0x00, 0x02, 0x00, 0x40, 0x01
  93. };
  95. TVMEDIATYPESECTION tv;
  96. BYTE data[256];
  98. if(m_isAudioPin) {
  99. memcpy(&tv, audioData, sizeof(tv));
  100. if(tv.cbFormat)
  101. memcpy(data, audioData+sizeof(tv), tv.cbFormat);
  102. }
  103. else {
  104. memcpy(&tv, videoData, sizeof(tv));
  105. if(tv.cbFormat)
  106. memcpy(data, videoData+sizeof(tv), tv.cbFormat);
  107. }
  109. CMediaType MediaType;
  110. //memset(&MediaType, 0, sizeof(CMediaType));
  111. LPVOID *pvi = (LPVOID*)MediaType.AllocFormatBuffer(tv.cbFormat);
  112. //
  113. memcpy(pvi, data, tv.cbFormat);
  114. MediaType.SetType(&tv.majortype);
  115. MediaType.SetSubtype(&tv.subtype);
  116. MediaType.SetFormatType(&tv.formattype);
  117. MediaType.SetTemporalCompression(tv.bTemporalCompression);
  118. MediaType.SetSampleSize(tv.lSampleSize);
  119. MediaType.bFixedSizeSamples = tv.bFixedSizeSamples;
  120. MediaType.cbFormat = tv.cbFormat;
  121. //
  122. m_MediaType.Set(MediaType);
  124. }
  126. void CTVStreamSourcePin::GetFirstMediaType()
  127. {
  128.     // 不尝试多次
  129. if(!m_bTryGetMediaType)
  130.     return;
  132.     m_bTryGetMediaType = false;
  133.     PBYTE pData = m_Data;
  134. memset(pData, 0 , 65536);
  135.     int iRet = 0;
  136.     m_header.bAudioSample = m_isAudioPin;
  138.     if(m_pFilter->CanDynamicReconnect())
  139.     {
  140.        iRet = m_pFilter->m_pComm->GetData(m_header, pData, 65536, m_isAudioPin, false);
  141.     }
  142.     else
  143.     {
  144.     // 10秒钟内必须读取到mediatype，否则，不予等待
  145.         int i = 0;
  146.     for(; i < 100; ++i)
  147.     {
  148. //     
  149.     iRet = m_pFilter->m_pComm->GetData(m_header, pData, 65536, m_isAudioPin, false);
  150. //
  151.     if (iRet > 0)//取到数据
  152.                 break;
  154.     Sleep(100);
  155.     }
  157.         if(i >= 100 && iRet <= 0)
  158.             return;
  159.     }
  160.     if(iRet <= 0)
  161.     {
  162.  #ifdef USE_LOG
  163.      fprintf(log, "first get media data fail! rn");
  164. #endif
  165.         return;
  166.     }
  168. #ifdef USE_LOG
  169.      fprintf(log, "first get media data OK rn");
  170. #endif
  172. if (m_header.start == 0 && m_header.length == 0)//mediaType
  173. {
  174. //
  175. TVMEDIATYPESECTION tv;
  176. memset(&tv, 0, sizeof(TVMEDIATYPESECTION));
  177. memcpy(&tv, pData, sizeof(TVMEDIATYPESECTION));
  178. //
  179. LPVOID *pvi = (LPVOID*)m_MediaType.AllocFormatBuffer(tv.cbFormat);
  180. if (pvi == 0) 
  181. {
  182. return;
  183. }
  184. //
  185. memcpy(pvi, pData + sizeof(TVMEDIATYPESECTION), tv.cbFormat);
  186. m_MediaType.SetType(&tv.majortype);
  187. m_MediaType.SetSubtype(&tv.subtype);
  188. m_MediaType.SetFormatType(&tv.formattype);
  189. m_MediaType.SetTemporalCompression(tv.bTemporalCompression);
  190. m_MediaType.SetSampleSize(tv.lSampleSize);
  191. m_MediaType.bFixedSizeSamples = tv.bFixedSizeSamples;
  192. m_MediaType.cbFormat = tv.cbFormat;
  193. //
  194. SetInfor(tv);
  195. //
  196. }//header
  198. #ifdef USE_LOG
  199.      fflush(log);
  200. #endif
  202. //
  203. return;
  204. }
  205. //
  206. HRESULT CTVStreamSourcePin::GetMediaType(CMediaType *pMediaType)
  207. {
  209.     CheckPointer(pMediaType, E_POINTER);
  210. //
  211. if (NULL == m_pFilter)
  212. {
  213. ASSERT(FALSE);
  214. return S_FALSE;
  215. }
  216. //
  217. if (m_pFilter->m_pComm       == NULL && 
  218. m_pFilter->m_pFileReader == NULL)
  219. {
  220. ASSERT(FALSE);
  221. return S_FALSE;
  222. }
  223. //
  224. CAutoLock cAutoLock(m_pFilter->pStateLock());
  225. // 取得媒体类型的数据
  226. TVMEDIATYPESECTION tv;
  227. BOOL  nRet     = FALSE;
  228. PBYTE lpviTemp = NULL;
  229. //
  230. if(m_pFilter->m_pComm )  
  231. {
  232. // 必须是winxp 及其以上版本的OS
  233. //      if(!m_pFilter->CanDynamicReconnect()) {
  234. // 由于没有默认媒体类型，需要等待网络返回
  235. GetFirstMediaType();
  237. // 没有获取到媒体类型数据，返回错误
  238. if(m_MediaType.cbFormat == 0)
  239. return S_FALSE;
  240. // }
  241. return pMediaType->Set(m_MediaType);
  242. }
  243. else if(m_pFilter->m_pFileReader)
  244. {
  245. nRet = m_pFilter->m_pFileReader->GetMediaType(tv, m_isAudioPin);
  246. if(nRet != TRUE)
  247. {
  248. ASSERT(FALSE);
  249. return S_FALSE;
  250. }
  251. //
  252. lpviTemp = new BYTE[tv.cbFormat];
  253. nRet = m_pFilter->m_pFileReader->GetMediaData(lpviTemp, m_isAudioPin);
  254. //
  255. if(nRet != TRUE)
  256. {
  257. ASSERT(FALSE);
  258. return S_FALSE;
  259. }
  260. }
  261. else 
  262. {
  263. ASSERT(FALSE);
  264. return E_POINTER;
  265. }
  266. //
  267. if(tv.bThisPinOnly && m_isAudioPin)
  268. {
  269. m_pFilter->m_isAudioOnly = TRUE;
  270. }
  271. //
  272. if(tv.cbFormat == 0 && tv.lSampleSize == 0) 
  273. {
  274. return S_FALSE;
  275. }
  276. // Allocate enough room for the MPEG1VIDEOINFO and the color tables
  277.     LPVOID *pvi = (LPVOID*)pMediaType->AllocFormatBuffer(tv.cbFormat);
  278.     if (pvi == 0) 
  279.     {
  280. return(E_OUTOFMEMORY);
  281. }
  282. //
  283. memcpy(pvi,lpviTemp,tv.cbFormat);
  284.     pMediaType->SetType(&tv.majortype);
  285. pMediaType->SetSubtype(&tv.subtype);
  286.     pMediaType->SetFormatType(&tv.formattype);
  287.     pMediaType->SetTemporalCompression(tv.bTemporalCompression);
  288.     pMediaType->SetSampleSize(tv.lSampleSize);
  289. pMediaType->bFixedSizeSamples = tv.bFixedSizeSamples;
  290. pMediaType->cbFormat = tv.cbFormat;
  291. //
  292. SetInfor(tv);
  293. // 删除临时变量
  294. if (NULL != lpviTemp)
  295. {
  296. delete [] lpviTemp;
  297. lpviTemp = NULL;
  298. }
  299. //
  300. return S_OK;
  301. }
  302. //
  303. void CTVStreamSourcePin::SetInfor(TVMEDIATYPESECTION tv)
  304. {
  305. if(tv.bFixedSizeSamples)
  306. {
  307. (m_isAudioPin?m_iAudioSampleSize:m_iVideoSampleSize) = tv.lSampleSize;
  308. }
  309. // 因为媒体类型中所制定的SampleSize可能有误（过小），所以要保证大于