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

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

多媒体编程

开发平台：

Visual C++

MP4Splitter.cpp：源码内容
  1. /* 
  2.  * Copyright (C) 2003-2005 Gabest
  3.  * http://www.gabest.org
  4.  *
  5.  *  This Program is free software; you can redistribute it and/or modify
  6.  *  it under the terms of the GNU General Public License as published by
  7.  *  the Free Software Foundation; either version 2, or (at your option)
  8.  *  any later version.
  9.  *   
  10.  *  This Program is distributed in the hope that it will be useful,
  11.  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  12.  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  13.  *  GNU General Public License for more details.
  14.  *   
  15.  *  You should have received a copy of the GNU General Public License
  16.  *  along with GNU Make; see the file COPYING.  If not, write to
  17.  *  the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA. 
  18.  *  http://www.gnu.org/copyleft/gpl.html
  19.  *
  20.  */
  22. #include "StdAfx.h"
  23. #include "MP4Splitter.h"
  24. #include "......DSUtilDSUtil.h"
  26. #include <initguid.h>
  27. #include "........includemoreuuids.h"
  29. #include "Ap4.h"
  30. #include "Ap4File.h"
  31. #include "Ap4StssAtom.h"
  32. #include "Ap4IsmaCryp.h"
  33. #include "Ap4AvcCAtom.h"
  34. #include "Ap4ChplAtom.h"
  35. #include "Ap4DataAtom.h"
  37. #ifdef REGISTER_FILTER
  39. const AMOVIESETUP_MEDIATYPE sudPinTypesIn[] =
  40. {
  41. {&MEDIATYPE_Stream, &MEDIASUBTYPE_MP4},
  42. {&MEDIATYPE_Stream, &MEDIASUBTYPE_NULL},
  43. };
  45. const AMOVIESETUP_PIN sudpPins[] =
  46. {
  47. {L"Input", FALSE, FALSE, FALSE, FALSE, &CLSID_NULL, NULL, countof(sudPinTypesIn), sudPinTypesIn},
  48. {L"Output", FALSE, TRUE, FALSE, FALSE, &CLSID_NULL, NULL, 0, NULL}
  49. };
  51. const AMOVIESETUP_FILTER sudFilter[] =
  52. {
  53. {&__uuidof(CMP4SplitterFilter), L"MP4 Splitter", MERIT_NORMAL, countof(sudpPins), sudpPins},
  54. {&__uuidof(CMP4SourceFilter), L"MP4 Source", MERIT_NORMAL, 0, NULL},
  55. };
  57. CFactoryTemplate g_Templates[] =
  58. {
  59. {sudFilter[0].strName, sudFilter[0].clsID, CreateInstance<CMP4SplitterFilter>, NULL, &sudFilter[0]},
  60. {sudFilter[1].strName, sudFilter[1].clsID, CreateInstance<CMP4SourceFilter>, NULL, &sudFilter[1]},
  61. };
  63. int g_cTemplates = countof(g_Templates);
  65. STDAPI DllRegisterServer()
  66. {
  67. DeleteRegKey(_T("Media Type\Extensions\"), _T(".mp4"));
  69. CList<CString> chkbytes;
  71. chkbytes.AddTail(_T("4,4,,66747970")); // ftyp
  72. chkbytes.AddTail(_T("4,4,,6d6f6f76")); // moov
  73. chkbytes.AddTail(_T("4,4,,6d646174")); // mdat
  74. chkbytes.AddTail(_T("4,4,,736b6970")); // skip
  76. RegisterSourceFilter(CLSID_AsyncReader, MEDIASUBTYPE_MP4, chkbytes, NULL);
  78. return AMovieDllRegisterServer2(TRUE);
  79. }
  81. STDAPI DllUnregisterServer()
  82. {
  83. UnRegisterSourceFilter(MEDIASUBTYPE_MP4);
  85. return AMovieDllRegisterServer2(FALSE);
  86. }
  88. extern "C" BOOL WINAPI DllEntryPoint(HINSTANCE, ULONG, LPVOID);
  90. BOOL APIENTRY DllMain(HANDLE hModule, DWORD dwReason, LPVOID lpReserved)
  91. {
  92.     return DllEntryPoint((HINSTANCE)hModule, dwReason, 0); // "DllMain" of the dshow baseclasses;
  93. }
  95. #endif
  97. //
  98. // CMP4SplitterFilter
  99. //
  101. CMP4SplitterFilter::CMP4SplitterFilter(LPUNKNOWN pUnk, HRESULT* phr)
  102. : CBaseSplitterFilter(NAME("CMP4SplitterFilter"), pUnk, phr, __uuidof(this))
  103. {
  104. }
  106. CMP4SplitterFilter::~CMP4SplitterFilter()
  107. {
  108. }
  110. HRESULT CMP4SplitterFilter::CreateOutputs(IAsyncReader* pAsyncReader)
  111. {
  112. CheckPointer(pAsyncReader, E_POINTER);
  114. HRESULT hr = E_FAIL;
  116. m_trackpos.RemoveAll();
  118. m_pFile.Free();
  119. m_pFile.Attach(new CMP4SplitterFile(pAsyncReader, hr));
  120. if(!m_pFile) return E_OUTOFMEMORY;
  121. if(FAILED(hr)) {m_pFile.Free(); return hr;}
  123. m_rtNewStart = m_rtCurrent = 0;
  124. m_rtNewStop = m_rtStop = m_rtDuration = 0;
  126. if(AP4_Movie* movie = (AP4_Movie*)m_pFile->GetMovie())
  127. {
  128. for(AP4_List<AP4_Track>::Item* item = movie->GetTracks().FirstItem();
  129. item;
  130. item = item->GetNext())
  131. {
  132. AP4_Track* track = item->GetData();
  134. if(track->GetType() != AP4_Track::TYPE_VIDEO 
  135. && track->GetType() != AP4_Track::TYPE_AUDIO
  136. && track->GetType() != AP4_Track::TYPE_TEXT)
  137. continue;
  139. AP4_Sample sample;
  141. if(!AP4_SUCCEEDED(track->GetSample(0, sample)) || sample.GetDescriptionIndex() == 0xFFFFFFFF)
  142. continue;
  144. CArray<CMediaType> mts;
  146. CMediaType mt;
  147. mt.SetSampleSize(1);
  149. VIDEOINFOHEADER* vih = NULL;
  150. WAVEFORMATEX* wfe = NULL;
  152. AP4_DataBuffer empty;
  154. if(AP4_SampleDescription* desc = track->GetSampleDescription(sample.GetDescriptionIndex()))
  155. {
  156. AP4_MpegSampleDescription* mpeg_desc = NULL;
  158. if(desc->GetType() == AP4_SampleDescription::TYPE_MPEG)
  159. {
  160. mpeg_desc = dynamic_cast<AP4_MpegSampleDescription*>(desc);
  161. }
  162. else if(desc->GetType() == AP4_SampleDescription::TYPE_ISMACRYP)
  163. {
  164. AP4_IsmaCrypSampleDescription* isma_desc = dynamic_cast<AP4_IsmaCrypSampleDescription*>(desc);
  165. mpeg_desc = isma_desc->GetOriginalSampleDescription();
  166. }
  168. if(AP4_MpegVideoSampleDescription* video_desc = 
  169. dynamic_cast<AP4_MpegVideoSampleDescription*>(mpeg_desc))
  170. {
  171. const AP4_DataBuffer* di = video_desc->GetDecoderInfo();
  172. if(!di) di = &empty;
  174. mt.majortype = MEDIATYPE_Video;
  175. mt.formattype = FORMAT_VideoInfo;
  176. vih = (VIDEOINFOHEADER*)mt.AllocFormatBuffer(sizeof(VIDEOINFOHEADER) + di->GetDataSize());
  177. memset(vih, 0, mt.FormatLength());
  178. vih->dwBitRate = video_desc->GetAvgBitrate()/8;
  179. vih->bmiHeader.biSize = sizeof(vih->bmiHeader);
  180. vih->bmiHeader.biWidth = (LONG)video_desc->GetWidth();
  181. vih->bmiHeader.biHeight = (LONG)video_desc->GetHeight();
  182. memcpy(vih + 1, di->GetData(), di->GetDataSize());
  184. switch(video_desc->GetObjectTypeId())
  185. {
  186. case AP4_MPEG4_VISUAL_OTI:
  187. mt.subtype = FOURCCMap('v4pm');
  188. mt.formattype = FORMAT_MPEG2Video;
  189. {
  190. MPEG2VIDEOINFO* vih = (MPEG2VIDEOINFO*)mt.AllocFormatBuffer(FIELD_OFFSET(MPEG2VIDEOINFO, dwSequenceHeader) + di->GetDataSize());
  191. memset(vih, 0, mt.FormatLength());
  192. vih->hdr.bmiHeader.biSize = sizeof(vih->hdr.bmiHeader);
  193. vih->hdr.bmiHeader.biWidth = (LONG)video_desc->GetWidth();
  194. vih->hdr.bmiHeader.biHeight = (LONG)video_desc->GetHeight();
  195. vih->hdr.bmiHeader.biCompression = 'v4pm';
  196. vih->hdr.bmiHeader.biPlanes = 1;
  197. vih->hdr.bmiHeader.biBitCount = 24;
  198. vih->hdr.dwPictAspectRatioX = vih->hdr.bmiHeader.biWidth;
  199. vih->hdr.dwPictAspectRatioY = vih->hdr.bmiHeader.biHeight;
  200. vih->cbSequenceHeader = di->GetDataSize();
  201. memcpy(vih->dwSequenceHeader, di->GetData(), di->GetDataSize());
  202. mts.Add(mt);
  203. mt.subtype = FOURCCMap(vih->hdr.bmiHeader.biCompression = 'V4PM');
  204. mts.Add(mt);
  205. }
  206. break;
  207. case AP4_MPEG2_VISUAL_SIMPLE_OTI:
  208. case AP4_MPEG2_VISUAL_MAIN_OTI:
  209. case AP4_MPEG2_VISUAL_SNR_OTI:
  210. case AP4_MPEG2_VISUAL_SPATIAL_OTI:
  211. case AP4_MPEG2_VISUAL_HIGH_OTI:
  212. case AP4_MPEG2_VISUAL_422_OTI:
  213. mt.subtype = MEDIASUBTYPE_MPEG2_VIDEO;
  214. {
  215. m_pFile->Seek(sample.GetOffset());
  216. CBaseSplitterFileEx::seqhdr h;
  217. CMediaType mt2;
  218. if(m_pFile->Read(h, sample.GetSize(), &mt2))
  219. mt = mt2;
  220. }
  221. mts.Add(mt);
  222. break;
  223. case AP4_MPEG1_VISUAL_OTI: // ???
  224. mt.subtype = MEDIASUBTYPE_MPEG1Payload;
  225. {
  226. m_pFile->Seek(sample.GetOffset());
  227. CBaseSplitterFileEx::seqhdr h;
  228. CMediaType mt2;
  229. if(m_pFile->Read(h, sample.GetSize(), &mt2))
  230. mt = mt2;
  231. }
  232. mts.Add(mt);
  233. break;
  234. }
  236. if(mt.subtype == GUID_NULL)
  237. {
  238. TRACE(_T("Unknown video OBI: %02xn"), video_desc->GetObjectTypeId());
  239. }
  240. }
  241. else if(AP4_MpegAudioSampleDescription* audio_desc = 
  242. dynamic_cast<AP4_MpegAudioSampleDescription*>(mpeg_desc))
  243. {
  244. const AP4_DataBuffer* di = audio_desc->GetDecoderInfo();
  245. if(!di) di = &empty;
  247. mt.majortype = MEDIATYPE_Audio;
  248. mt.formattype = FORMAT_WaveFormatEx;
  250. wfe = (WAVEFORMATEX*)mt.AllocFormatBuffer(sizeof(WAVEFORMATEX) + di->GetDataSize());
  251. memset(wfe, 0, mt.FormatLength());
  252. wfe->nSamplesPerSec = audio_desc->GetSampleRate();
  253. wfe->nAvgBytesPerSec = audio_desc->GetAvgBitrate()/8;
  254. wfe->nChannels = audio_desc->GetChannelCount();
  255. wfe->wBitsPerSample = audio_desc->GetSampleSize();
  256. wfe->cbSize = (WORD)di->GetDataSize();
  257. memcpy(wfe + 1, di->GetData(), di->GetDataSize());
  259. switch(audio_desc->GetObjectTypeId())
  260. {
  261. case AP4_MPEG4_AUDIO_OTI:
  262. case AP4_MPEG2_AAC_AUDIO_MAIN_OTI: // ???
  263. case AP4_MPEG2_AAC_AUDIO_LC_OTI: // ???
  264. case AP4_MPEG2_AAC_AUDIO_SSRP_OTI: // ???
  265. mt.subtype = FOURCCMap(wfe->wFormatTag = WAVE_FORMAT_AAC);
  266. mts.Add(mt);
  267. break;
  268. case AP4_MPEG2_PART3_AUDIO_OTI: // ???
  269. break;
  270. case AP4_MPEG1_AUDIO_OTI:
  271. mt.subtype = FOURCCMap(wfe->wFormatTag = WAVE_FORMAT_MP3);
  272. {
  273. m_pFile->Seek(sample.GetOffset());
  274. CBaseSplitterFileEx::mpahdr h;
  275. CMediaType mt2;
  276. if(m_pFile->Read(h, sample.GetSize(), false, &mt2))
  277. mt = mt2;
  278. }
  279. mts.Add(mt);
  280. break;
  281. }
  283. if(mt.subtype == GUID_NULL)
  284. {
  285. TRACE(_T("Unknown audio OBI: %02xn"), audio_desc->GetObjectTypeId());
  286. }
  287. }
  288. else if(AP4_UnknownSampleDescription* unknown_desc = 
  289. dynamic_cast<AP4_UnknownSampleDescription*>(desc)) // TEMP
  290. {
  291. AP4_SampleEntry* sample_entry = unknown_desc->GetSampleEntry();
  293. if(dynamic_cast<AP4_TextSampleEntry*>(sample_entry)
  294. || dynamic_cast<AP4_Tx3gSampleEntry*>(sample_entry))
  295. {
  296. mt.majortype = MEDIATYPE_Subtitle;
  297. mt.subtype = MEDIASUBTYPE_ASS;
  298. mt.formattype = FORMAT_SubtitleInfo;
  299. CStringA hdr = "[Script Info]nScriptType: v4.00+n";
  300. SUBTITLEINFO* si = (SUBTITLEINFO*)mt.AllocFormatBuffer(sizeof(SUBTITLEINFO) + hdr.GetLength());
  301. memset(si, 0, mt.FormatLength());
  302. si->dwOffset = sizeof(SUBTITLEINFO);
  303. /* dong
  304. strcpy_s(si->IsoLang, countof(si->IsoLang), track->GetTrackLanguage().c_str());
  305. wcscpy_s(si->TrackName, countof(si->TrackName), CStringW(track->GetTrackName().c_str()));
  306. */
  307. strcpy(si->IsoLang, track->GetTrackLanguage().c_str());
  308. wcscpy(si->TrackName,CStringW(track->GetTrackName().c_str()));
  310. memcpy(si + 1, (LPCSTR)hdr, hdr.GetLength());
  311. // mts.Add(mt); // TODO
  313. mt.subtype = MEDIASUBTYPE_UTF8;
  314. si = (SUBTITLEINFO*)mt.ReallocFormatBuffer(sizeof(SUBTITLEINFO));
  315. mts.Add(mt);
  316. }
  317. }
  318. }
  319. else if(AP4_Avc1SampleEntry* avc1 = dynamic_cast<AP4_Avc1SampleEntry*>(
  320. track->GetTrakAtom()->FindChild("mdia/minf/stbl/stsd/avc1")))
  321. {
  322. if(AP4_AvcCAtom* avcC = dynamic_cast<AP4_AvcCAtom*>(avc1->GetChild(AP4_ATOM_TYPE_AVCC)))
  323. {
  324. const AP4_DataBuffer* di = avcC->GetDecoderInfo();
  325. if(!di) di = &empty;
  327. const AP4_Byte* data = di->GetData();
  328. AP4_Size size = di->GetDataSize();
  330. mt.majortype = MEDIATYPE_Video;
  331. mt.subtype = FOURCCMap('1cva');
  332. mt.formattype = FORMAT_MPEG2Video;
  334. MPEG2VIDEOINFO* vih = (MPEG2VIDEOINFO*)mt.AllocFormatBuffer(FIELD_OFFSET(MPEG2VIDEOINFO, dwSequenceHeader) + size - 7);
  335. memset(vih, 0, mt.FormatLength());
  336. vih->hdr.bmiHeader.biSize = sizeof(vih->hdr.bmiHeader);
  337. vih->hdr.bmiHeader.biWidth = (LONG)avc1->GetWidth();
  338. vih->hdr.bmiHeader.biHeight = (LONG)avc1->GetHeight();
  339. vih->hdr.bmiHeader.biCompression = '1cva';
  340. vih->hdr.bmiHeader.biPlanes = 1;
  341. vih->hdr.bmiHeader.biBitCount = 24;
  342. vih->hdr.dwPictAspectRatioX = vih->hdr.bmiHeader.biWidth;
  343. vih->hdr.dwPictAspectRatioY = vih->hdr.bmiHeader.biHeight;
  344. vih->dwProfile = data[1];
  345. vih->dwLevel = data[3];
  346. vih->dwFlags = (data[4] & 3) + 1;
  348. vih->cbSequenceHeader = 0;
  350. BYTE* src = (BYTE*)data + 5;
  351. BYTE* dst = (BYTE*)vih->dwSequenceHeader;
  353. BYTE* src_end = (BYTE*)data + size;
  354. BYTE* dst_end = (BYTE*)vih->dwSequenceHeader + size;
  356. for(int i = 0; i < 2; i++)
  357. {
  358. for(int n = *src++ & 0x1f; n > 0; n--)
  359. {
  360. int len = ((src[0] << 8) | src[1]) + 2;
  361. if(src + len > src_end || dst + len > dst_end) {ASSERT(0); break;}
  362. memcpy(dst, src, len);
  363. src += len; 
  364. dst += len;
  365. vih->cbSequenceHeader += len;
  366. }
  367. }
  369. mts.Add(mt);
  371. mt.subtype = FOURCCMap(vih->hdr.bmiHeader.biCompression = '1CVA');
  372. mts.Add(mt);
  373. }
  374. }
  375. else if(AP4_VisualSampleEntry* s263 = dynamic_cast<AP4_VisualSampleEntry*>(
  376. track->GetTrakAtom()->FindChild("mdia/minf/stbl/stsd/s263")))
  377. {
  378. mt.majortype = MEDIATYPE_Video;
  379. mt.subtype = FOURCCMap('362s');
  380. mt.formattype = FORMAT_VideoInfo;
  381. vih = (VIDEOINFOHEADER*)mt.AllocFormatBuffer(sizeof(VIDEOINFOHEADER));
  382. memset(vih, 0, mt.FormatLength());
  383. vih->bmiHeader.biSize = sizeof(vih->bmiHeader);
  384. vih->bmiHeader.biWidth = (LONG)s263->GetWidth();
  385. vih->bmiHeader.biHeight = (LONG)s263->GetHeight();
  386. vih->bmiHeader.biCompression = '362s';
  387. mts.Add(mt);
  389. mt.subtype = FOURCCMap(vih->bmiHeader.biCompression = '362S');
  390. mts.Add(mt);
  391. }
  392. else if(AP4_AudioSampleEntry* samr = dynamic_cast<AP4_AudioSampleEntry*>(
  393. track->GetTrakAtom()->FindChild("mdia/minf/stbl/stsd/samr")))
  394. {
  395. mt.majortype = MEDIATYPE_Audio;
  396. mt.subtype = FOURCCMap('rmas');
  397. mt.formattype = FORMAT_WaveFormatEx;
  398. wfe = (WAVEFORMATEX*)mt.AllocFormatBuffer(sizeof(WAVEFORMATEX));
  399. memset(wfe, 0, mt.FormatLength());
  400. wfe->nSamplesPerSec = samr->GetSampleRate();
  401. wfe->nChannels = samr->GetChannelCount();
  402. mts.Add(mt);
  404. mt.subtype = FOURCCMap('RMAS');
  405. mts.Add(mt);
  406. }
  408. REFERENCE_TIME rtDuration = 10000i64 * track->GetDurationMs();
  409. if(m_rtDuration < rtDuration) m_rtDuration = rtDuration;
  411. DWORD id = track->GetId();
  413. CStringW name, lang;
  414. name.Format(L"Output %d", id);
  416. AP4_String TrackName = track->GetTrackName();
  417. AP4_String TrackLanguage = track->GetTrackLanguage();
  419. if(!TrackName.empty())
  420. {
  421. name.Format(L"%s", CStringW(TrackName.c_str()));
  422. SetProperty(L"NAME", CStringW(TrackName.c_str()));
  423. }
  425. if(!TrackLanguage.empty())
  426. {
  427. if(TrackLanguage != "und") name += L" (" + CStringW(TrackLanguage.c_str()) + L")";
  428. SetProperty(L"LANG", CStringW(TrackLanguage.c_str()));
  429. }
  431. CAutoPtr<CBaseSplitterOutputPin> pPinOut(new CBaseSplitterOutputPin(mts, name, this, this, &hr));
  433. EXECUTE_ASSERT(SUCCEEDED(AddOutputPin(id, pPinOut)));
  435. m_trackpos[id] = trackpos();
  436. }
  438. if(AP4_ChplAtom* chpl = dynamic_cast<AP4_ChplAtom*>(movie->GetMoovAtom()->FindChild("udta/chpl")))
  439. {
  440. AP4_Array<AP4_ChplAtom::AP4_Chapter>& chapters = chpl->GetChapters();
  442. for(AP4_Cardinal i = 0; i < chapters.ItemCount(); i++)
  443. {
  444. AP4_ChplAtom::AP4_Chapter& chapter = chapters[i];
  445. ChapAppend(chapter.Time, UTF8To16(ConvertMBCS(chapter.Name.c_str(), ANSI_CHARSET, CP_UTF8)));
  446. }
  448. ChapSort();
  449. }
  451. if(AP4_ContainerAtom* ilst = dynamic_cast<AP4_ContainerAtom*>(movie->GetMoovAtom()->FindChild("udta/meta/ilst")))
  452. {
  453. CStringW title, artist, writer, album, year, appl, desc, track;
  455. for(AP4_List<AP4_Atom>::Item* item = ilst->GetChildren().FirstItem();
  456. item;
  457. item = item->GetNext())
  458. {
  459. if(AP4_ContainerAtom* atom = dynamic_cast<AP4_ContainerAtom*>(item->GetData()))
  460. {
  461. if(AP4_DataAtom* data = dynamic_cast<AP4_DataAtom*>(atom->GetChild(AP4_ATOM_TYPE_DATA)))
  462. {
  463. const AP4_DataBuffer* db = data->GetData();
  465. if(atom->GetType() == AP4_ATOM_TYPE_TRKN)
  466. {
  467. if(db->GetDataSize() >= 4)
  468. {
  469. unsigned short n = (db->GetData()[2] << 8) | db->GetData()[3];
  470. if(n > 0 && n < 100) track.Format(L"%02d", n);
  471. else if(n >= 100) track.Format(L"%d", n);
  472. }
  473. }
  474. else
  475. {
  476. CStringW str = UTF8To16(CStringA((LPCSTR)db->GetData(), db->GetDataSize()));
  478. switch(atom->GetType())
  479. {
  480. case AP4_ATOM_TYPE_NAM: title = str; break;
  481. case AP4_ATOM_TYPE_ART: artist = str; break;
  482. case AP4_ATOM_TYPE_WRT: writer = str; break;
  483. case AP4_ATOM_TYPE_ALB: album = str; break;
  484. case AP4_ATOM_TYPE_DAY: year = str; break;
  485. case AP4_ATOM_TYPE_TOO: appl = str; break;
  486. case AP4_ATOM_TYPE_CMT: desc = str; break;
  487. }
  488. }
  489. }
  490. }
  491. }
  493. if(!title.IsEmpty())
  494. {
  495. if(!track.IsEmpty()) title = track + L" - " + title;
  496. if(!album.IsEmpty()) title = album + L" - " + title;
  497. if(!year.IsEmpty()) title += L" - " +  year;
  498. SetProperty(L"TITL", title);
  499. }
  501. if(!artist.IsEmpty()) SetProperty(L"AUTH", artist);
  502. else if(!writer.IsEmpty()) SetProperty(L"AUTH", writer);
  504. if(!appl.IsEmpty()) SetProperty(L"APPL", appl);
  506. if(!desc.IsEmpty()) SetProperty(L"DESC", desc);
  507. }
  508. }
  510. m_rtNewStop = m_rtStop = m_rtDuration;
  512. return m_pOutputs.GetCount() > 0 ? S_OK : E_FAIL;
  513. }
  515. bool CMP4SplitterFilter::DemuxInit()
  516. {
  517. AP4_Movie* movie = (AP4_Movie*)m_pFile->GetMovie();
  519. POSITION pos = m_trackpos.GetStartPosition();
  520. while(pos)
  521. {
  522. CAtlMap<DWORD, trackpos>::CPair* pPair = m_trackpos.GetNext(pos);
  524. pPair->m_value.index = 0;
  525. pPair->m_value.ts = 0;
  527. AP4_Track* track = movie->GetTrack(pPair->m_key);
  529. AP4_Sample sample;
  530. if(AP4_SUCCEEDED(track->GetSample(0, sample)))
  531. pPair->m_value.ts = sample.GetCts();
  532. }
  534. return true;
  535. }
  537. void CMP4SplitterFilter::DemuxSeek(REFERENCE_TIME rt)
  538. {
  539. AP4_TimeStamp ts = (AP4_TimeStamp)(rt / 10000);
  541. AP4_Movie* movie = (AP4_Movie*)m_pFile->GetMovie();
  543. POSITION pos = m_trackpos.GetStartPosition();
  544. while(pos)
  545. {
  546. CAtlMap<DWORD, trackpos>::CPair* pPair = m_trackpos.GetNext(pos);
  548. AP4_Track* track = movie->GetTrack(pPair->m_key);
  550. if(AP4_FAILED(track->GetSampleIndexForTimeStampMs(ts, pPair->m_value.index)))
  551. pPair->m_value.index = 0;
  553. AP4_Sample sample;
  554. if(AP4_SUCCEEDED(track->GetSample(pPair->m_value.index, sample)))
  555. pPair->m_value.ts = sample.GetCts();
  557. // FIXME: slow search & stss->m_Entries is private
  559. if(AP4_StssAtom* stss = dynamic_cast<AP4_StssAtom*>(track->GetTrakAtom()->FindChild("mdia/minf/stbl/stss")))
  560. {
  561. if(stss->m_Entries.ItemCount() > 0)
  562. {
  563. AP4_Cardinal i = -1;
  564. while(++i < stss->m_Entries.ItemCount() && stss->m_Entries[i]-1 <= pPair->m_value.index);
  565. if(i > 0) i--;
  566. pPair->m_value.index = stss->m_Entries[i]-1;
  567. }
  568. }
  569. }
  570. }
  572. bool CMP4SplitterFilter::DemuxLoop()
  573. {
  574. HRESULT hr = S_OK;
  576. AP4_Movie* movie = (AP4_Movie*)m_pFile->GetMovie();
  578. while(SUCCEEDED(hr) && !CheckRequest(NULL))
  579. {
  580. CAtlMap<DWORD, trackpos>::CPair* pPairNext = NULL;
  581. REFERENCE_TIME rtNext = 0;
  583. POSITION pos = m_trackpos.GetStartPosition();
  584. while(pos)
  585. {
  586. CAtlMap<DWORD, trackpos>::CPair* pPair = m_trackpos.GetNext(pos);
  588. AP4_Track* track = movie->GetTrack(pPair->m_key);
  590. REFERENCE_TIME rt = (REFERENCE_TIME)(10000000.0 / track->GetMediaTimeScale() * pPair->m_value.ts);
  592. if(pPair->m_value.index < track->GetSampleCount() && (!pPairNext || rt < rtNext))
  593. {
  594. pPairNext = pPair;
  595. rtNext = rt;
  596. }
  597. }
  599. if(!pPairNext) break;
  601. AP4_Track* track = movie->GetTrack(pPairNext->m_key);
  603. AP4_Sample sample;
  604. AP4_DataBuffer data;
  606. if(AP4_SUCCEEDED(track->ReadSample(pPairNext->m_value.index, sample, data)))
  607. {
  608. CAutoPtr<Packet> p(new Packet());
  610. p->TrackNumber = (DWORD)track->GetId();
  611. p->rtStart = (REFERENCE_TIME)(10000000.0 / track->GetMediaTimeScale() * sample.GetCts());
  612. p->rtStop = p->rtStart + 1;
  613. p->bSyncPoint = TRUE;
  615. if(track->GetType() == AP4_Track::TYPE_TEXT)
  616. {
  617. const AP4_Byte* ptr = data.GetData();
  618. AP4_Size avail = data.GetDataSize();
  620. if(avail > 2)
  621. {
  622. AP4_UI16 size = (ptr[0] << 8) | ptr[1];
  624. if(size <= avail-2)
  625. {
  626. CStringA str;
  628. if(size >= 2 && ptr[2] == 0xfe && ptr[3] == 0xff)
  629. {
  630. CStringW wstr = CStringW((LPCWSTR)&ptr[2], size/2);
  631. for(int i = 0; i < wstr.GetLength(); i++) wstr.SetAt(i, ((WORD)wstr[i] >> 8) | ((WORD)wstr[i] << 8));
  632. str = UTF16To8(wstr);
  633. }
  634. else
  635. {
  636. str = CStringA((LPCSTR)&ptr[2], size);
  637. }
  639. // TODO: still very incomplete
  641. if(CBaseSplitterOutputPin* pPin = GetOutputPin(p->TrackNumber))
  642. if(pPin->CurrentMediaType().subtype == MEDIASUBTYPE_ASS)
  643. {
  644. CStringW wstr = UTF8To16(str);
  646. AP4_SampleDescription* desc = track->GetSampleDescription(sample.GetDescriptionIndex());
  648. if(AP4_UnknownSampleDescription* unknown_desc = dynamic_cast<AP4_UnknownSampleDescription*>(desc)) // TEMP
  649. {
  650. AP4_SampleEntry* sample_entry = unknown_desc->GetSampleEntry();
  652. if(AP4_TextSampleEntry* text = dynamic_cast<AP4_TextSampleEntry*>(sample_entry))
  653. {
  654. const AP4_TextSampleEntry::AP4_TextDescription& d = text->GetDescription();
  655. }
  656. else if(AP4_Tx3gSampleEntry* tx3g = dynamic_cast<AP4_Tx3gSampleEntry*>(sample_entry))
  657. {
  658. const AP4_Tx3gSampleEntry::AP4_Tx3gDescription& d = tx3g->GetDescription();
  660. int align = 2;
  661. signed char h = (signed char)d.HorizontalJustification;
  662. signed char v = (signed char)d.VerticalJustification;
  663. if(h == 0 && v < 0) align = 1;
  664. else if(h > 0 && v < 0) align = 2;
  665. else if(h < 0 && v < 0) align = 3;
  666. else if(h == 0 && v > 0) align = 4;
  667. else if(h > 0 && v > 0) align = 5;
  668. else if(h < 0 && v > 0) align = 6;
  669. else if(h == 0 && v == 0) align = 7;
  670. else if(h > 0 && v == 0) align = 8;
  671. else if(h < 0 && v == 0) align = 9;
  672. wstr.Format(L"{\an%d}%s", align, CStringW(wstr));
  674. if(d.BackgroundColor)
  675. {
  676. DWORD c = d.BackgroundColor;
  677. wstr.Format(L"{\3c%02x%02x%02x}%s", (c>>8)&0xff, (c>>16)&0xff, (c>>24)&0xff, CStringW(wstr));
  678. if(c&0xff) wstr.Format(L"{\3a%02x}%s", 255 - (c&0xff), CStringW(wstr));
  679. }
  681. if(d.Style.Font.Color)
  682. {
  683. DWORD c = d.Style.Font.Color;
  684. wstr.Format(L"{\1c%02x%02x%02x}%s", (c>>8)&0xff, (c>>16)&0xff, (c>>24)&0xff, CStringW(wstr));
  685. if(c&0xff) wstr.Format(L"{\1a%02x}%s", 255 - (c&0xff), CStringW(wstr));
  686. }
  688. if(d.Style.Font.Size) wstr.Format(L"{\fs%d}%s", d.Style.Font.Size, CStringW(wstr));
  690. if(d.Style.Font.Face&1) wstr = L"{\b1}" + wstr;
  691. if(d.Style.Font.Face&2) wstr = L"{\i1}" + wstr;
  692. if(d.Style.Font.Face&4) wstr = L"{\u1}" + wstr;
  693. }
  694. }
  696. str = "0,0,Default,,0000,0000,0000,," + UTF16To8(wstr);
  697. }
  699. p->pData.SetSize(str.GetLength());
  700. memcpy(p->pData.GetData(), (LPCSTR)str, str.GetLength());
  702. AP4_Sample sample;
  703. if(AP4_SUCCEEDED(track->GetSample(pPairNext->m_value.index+1, sample)))
  704. p->rtStop = (REFERENCE_TIME)(10000000.0 / track->GetMediaTimeScale() * sample.GetCts());
  705. }
  706. }
  707. }
  708. else
  709. {
  710. p->pData.SetSize(data.GetDataSize());
  711. memcpy(p->pData.GetData(), data.GetData(), data.GetDataSize());
  712. }
  714. // FIXME: slow search & stss->m_Entries is private
  716. if(AP4_StssAtom* stss = dynamic_cast<AP4_StssAtom*>(track->GetTrakAtom()->FindChild("mdia/minf/stbl/stss")))
  717. {
  718. if(stss->m_Entries.ItemCount() > 0)
  719. {
  720. p->bSyncPoint = FALSE;
  722. AP4_Cardinal i = -1;
  723. while(++i < stss->m_Entries.ItemCount())
  724. if(stss->m_Entries[i]-1 == pPairNext->m_value.index)
  725. p->bSyncPoint = TRUE;
  726. }
  727. }
  729. hr = DeliverPacket(p);
  730. }
  732. {
  733. AP4_Sample sample;
  734. if(AP4_SUCCEEDED(track->GetSample(++pPairNext->m_value.index, sample)))
  735. pPairNext->m_value.ts = sample.GetCts();
  736. }
  738. }
  740. return true;
  741. }
  743. // IKeyFrameInfo
  745. STDMETHODIMP CMP4SplitterFilter::GetKeyFrameCount(UINT& nKFs)
  746. {
  747. CheckPointer(m_pFile, E_UNEXPECTED);
  749. if(!m_pFile) return E_UNEXPECTED;
  751. AP4_Movie* movie = (AP4_Movie*)m_pFile->GetMovie();
  753. POSITION pos = m_trackpos.GetStartPosition();
  754. while(pos)
  755. {
  756. CAtlMap<DWORD, trackpos>::CPair* pPair = m_trackpos.GetNext(pos);
  758. AP4_Track* track = movie->GetTrack(pPair->m_key);
  760. if(track->GetType() != AP4_Track::TYPE_VIDEO)
  761. continue;
  763. if(AP4_StssAtom* stss = dynamic_cast<AP4_StssAtom*>(track->GetTrakAtom()->FindChild("mdia/minf/stbl/stss")))
  764. {
  765. nKFs = stss->m_Entries.ItemCount();
  766. return S_OK;
  767. }
  768. }
  770. return E_FAIL;
  771. }
  773. STDMETHODIMP CMP4SplitterFilter::GetKeyFrames(const GUID* pFormat, REFERENCE_TIME* pKFs, UINT& nKFs)
  774. {
  775. CheckPointer(pFormat, E_POINTER);
  776. CheckPointer(pKFs, E_POINTER);
  777. CheckPointer(m_pFile, E_UNEXPECTED);
  779. if(*pFormat != TIME_FORMAT_MEDIA_TIME) return E_INVALIDARG;
  781. if(!m_pFile) return E_UNEXPECTED;
  783. AP4_Movie* movie = (AP4_Movie*)m_pFile->GetMovie();
  785. POSITION pos = m_trackpos.GetStartPosition();
  786. while(pos)
  787. {
  788. CAtlMap<DWORD, trackpos>::CPair* pPair = m_trackpos.GetNext(pos);
  790. AP4_Track* track = movie->GetTrack(pPair->m_key);
  792. if(track->GetType() != AP4_Track::TYPE_VIDEO)
  793. continue;
  795. if(AP4_StssAtom* stss = dynamic_cast<AP4_StssAtom*>(track->GetTrakAtom()->FindChild("mdia/minf/stbl/stss")))
  796. {
  797. nKFs = 0;
  799. for(AP4_Cardinal i = 0; i < stss->m_Entries.ItemCount(); i++)
  800. {
  801. AP4_Sample sample;
  802. if(AP4_SUCCEEDED(track->GetSample(stss->m_Entries[i]-1, sample)))
  803. pKFs[nKFs++] = 10000000i64 * sample.GetCts() / track->GetMediaTimeScale();
  804. }
  806. return S_OK;
  807. }
  808. }
  810. return E_FAIL;
  811. }
  813. //
  814. // CMP4SourceFilter
  815. //
  817. CMP4SourceFilter::CMP4SourceFilter(LPUNKNOWN pUnk, HRESULT* phr)
  818. : CMP4SplitterFilter(pUnk, phr)
  819. {
  820. m_clsid = __uuidof(this);
  821. m_pInput.Free();
  822. }