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

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

多媒体编程

开发平台：

Visual C++

DSMSplitterFile.cpp：源码内容
  1. #include "StdAfx.h"
  2. #include "DSMSplitterFile.h"
  3. #include "......DSUtilDSUtil.h"
  4. #include "........includematroskamatroska.h"
  6. CDSMSplitterFile::CDSMSplitterFile(IAsyncReader* pReader, HRESULT& hr, IDSMResourceBagImpl& res, IDSMChapterBagImpl& chap) 
  7. : CBaseSplitterFile(pReader, hr)
  8. , m_rtFirst(0)
  9. , m_rtDuration(0)
  10. {
  11. if(FAILED(hr)) return;
  13. hr = Init(res, chap);
  14. }
  16. HRESULT CDSMSplitterFile::Init(IDSMResourceBagImpl& res, IDSMChapterBagImpl& chap)
  17. {
  18. Seek(0);
  20. if(BitRead(DSMSW_SIZE<<3) != DSMSW || BitRead(5) != DSMP_FILEINFO)
  21. return E_FAIL;
  23. Seek(0);
  25. m_mts.RemoveAll();
  26. m_rtFirst = m_rtDuration = 0;
  27. m_fim.RemoveAll();
  28. m_sim.RemoveAll();
  29. res.ResRemoveAll();
  30. chap.ChapRemoveAll();
  32. dsmp_t type;
  33. UINT64 len;
  34. int limit = 65536;
  36. // examine the beginning of the file ...
  38. while(Sync(type, len, 0))
  39. {
  40. __int64 pos = GetPos();
  42. if(type == DSMP_MEDIATYPE)
  43. {
  44. BYTE id;
  45. CMediaType mt;
  46. if(Read(len, id, mt)) m_mts[id] = mt;
  47. }
  48. else if(type == DSMP_SAMPLE)
  49. {
  50. Packet p;
  51. if(Read(len, &p, false) && p.rtStart != Packet::INVALID_TIME)
  52. {
  53. m_rtFirst = p.rtStart;
  54. break;
  55. }
  56. }
  57. else if(type == DSMP_FILEINFO) {if((BYTE)BitRead(8) > DSMF_VERSION) return E_FAIL; Read(len-1, m_fim);}
  58. else if(type == DSMP_STREAMINFO) {Read(len-1, m_sim[(BYTE)BitRead(8)]);}
  59. else if(type == DSMP_SYNCPOINTS) {Read(len, m_sps);}
  60. else if(type == DSMP_RESOURCE) {Read(len, res);}
  61. else if(type == DSMP_CHAPTERS) {Read(len, chap);}
  63. Seek(pos + len);
  64. }
  66. if(type != DSMP_SAMPLE)
  67. return E_FAIL;
  69. // ... and the end 
  71. for(int i = 1, j = (int)((GetLength()+limit/2)/limit); i <= j; i++)
  72. {
  73. __int64 seekpos = max(0, (__int64)GetLength()-i*limit);
  74. Seek(seekpos);
  76. while(Sync(type, len, limit) && GetPos() < seekpos+limit)
  77. {
  78. __int64 pos = GetPos();
  80. if(type == DSMP_SAMPLE)
  81. {
  82. Packet p;
  83. if(Read(len, &p, false) && p.rtStart != Packet::INVALID_TIME)
  84. {
  85. m_rtDuration = max(m_rtDuration, p.rtStop - m_rtFirst); // max isn't really needed, only for safety
  86. i = j;
  87. }
  88. }
  89. else if(type == DSMP_FILEINFO) {if((BYTE)BitRead(8) > DSMF_VERSION) return E_FAIL; Read(len-1, m_fim);}
  90. else if(type == DSMP_STREAMINFO) {Read(len-1, m_sim[(BYTE)BitRead(8)]);}
  91. else if(type == DSMP_SYNCPOINTS) {Read(len, m_sps);}
  92. else if(type == DSMP_RESOURCE) {Read(len, res);}
  93. else if(type == DSMP_CHAPTERS) {Read(len, chap);}
  95. Seek(pos + len);
  96. }
  97. }
  99. if(m_rtFirst < 0)
  100. {
  101. m_rtDuration += m_rtFirst;
  102. m_rtFirst = 0;
  103. }
  105. return m_mts.GetCount() > 0 ? S_OK : E_FAIL;
  106. }
  108. bool CDSMSplitterFile::Sync(dsmp_t& type, UINT64& len, __int64 limit)
  109. {
  110. UINT64 pos;
  111. return Sync(pos, type, len, limit);
  112. }
  114. bool CDSMSplitterFile::Sync(UINT64& syncpos, dsmp_t& type, UINT64& len, __int64 limit)
  115. {
  116. BitByteAlign();
  118. limit += DSMSW_SIZE;
  120. for(UINT64 id = 0; (id&((1ui64<<(DSMSW_SIZE<<3))-1)) != DSMSW; id = (id << 8) | (BYTE)BitRead(8))
  121. {
  122. if(limit-- <= 0 || GetPos() >= GetLength()-2)
  123. return(false);
  124. }
  126. syncpos = GetPos() - (DSMSW_SIZE<<3);
  127. type = (dsmp_t)BitRead(5);
  128. len = BitRead(((int)BitRead(3)+1)<<3);
  130. return(true);
  131. }
  133. bool CDSMSplitterFile::Read(__int64 len, BYTE& id, CMediaType& mt)
  134. {
  135. id = (BYTE)BitRead(8);
  136. ByteRead((BYTE*)&mt.majortype, sizeof(mt.majortype));
  137. ByteRead((BYTE*)&mt.subtype, sizeof(mt.subtype));
  138. mt.bFixedSizeSamples = (BOOL)BitRead(1);
  139. mt.bTemporalCompression = (BOOL)BitRead(1);
  140. mt.lSampleSize = (ULONG)BitRead(30);
  141. ByteRead((BYTE*)&mt.formattype, sizeof(mt.formattype));
  142. len -= 5 + sizeof(GUID)*3;
  143. ASSERT(len >= 0);
  144. if(len > 0) {mt.AllocFormatBuffer((LONG)len); ByteRead(mt.Format(), mt.FormatLength());}
  145. else mt.ResetFormatBuffer();
  146. return true;
  147. }
  149. bool CDSMSplitterFile::Read(__int64 len, Packet* p, bool fData)
  150. {
  151. if(!p) return false;
  153. p->TrackNumber = (DWORD)BitRead(8);
  154. p->bSyncPoint = (BOOL)BitRead(1);
  155. bool fSign = !!BitRead(1);
  156. int iTimeStamp = (int)BitRead(3);
  157. int iDuration = (int)BitRead(3);
  159. if(fSign && !iTimeStamp)
  160. {
  161. ASSERT(!iDuration);
  162. p->rtStart = Packet::INVALID_TIME;
  163. p->rtStop = Packet::INVALID_TIME + 1;
  164. }
  165. else
  166. {
  167. p->rtStart = (REFERENCE_TIME)BitRead(iTimeStamp<<3) * (fSign ? -1 : 1);
  168. p->rtStop = p->rtStart + BitRead(iDuration<<3);
  169. }
  171. if(fData)
  172. {
  173.         p->pData.SetSize((INT_PTR)len - (2 + iTimeStamp + iDuration));
  174. ByteRead(p->pData.GetData(), p->pData.GetSize());
  175. }
  177. return true;
  178. }
  180. bool CDSMSplitterFile::Read(__int64 len, CArray<SyncPoint>& sps)
  181. {
  182. SyncPoint sp = {0, 0};
  183. sps.RemoveAll();
  185. while(len > 0)
  186. {
  187. bool fSign = !!BitRead(1);
  188. int iTimeStamp = (int)BitRead(3);
  189. int iFilePos = (int)BitRead(3);
  190. BitRead(1); // reserved
  192. sp.rt += (REFERENCE_TIME)BitRead(iTimeStamp<<3) * (fSign ? -1 : 1);
  193. sp.fp += BitRead(iFilePos<<3);
  194. sps.Add(sp);
  196. len -= 1 + iTimeStamp + iFilePos;
  197. }
  199. if(len != 0)
  200. {
  201. sps.RemoveAll();
  202. return false;
  203. }
  205. // TODO: sort sps
  207. return true;
  208. }
  210. bool CDSMSplitterFile::Read(__int64 len, CStreamInfoMap& im)
  211. {
  212. while(len >= 5)
  213. {
  214. CStringA key;
  215. ByteRead((BYTE*)key.GetBufferSetLength(4), 4);
  216. len -= 4;
  217. len -= Read(len, im[key]);
  218. }
  220. return len == 0;
  221. }
  223. bool CDSMSplitterFile::Read(__int64 len, IDSMResourceBagImpl& res)
  224. {
  225. BYTE compression = (BYTE)BitRead(2);
  226. BYTE reserved = (BYTE)BitRead(6);
  227. len--;
  229. CDSMResource r;
  230. len -= Read(len, r.name);
  231. len -= Read(len, r.desc);
  232. len -= Read(len, r.mime);
  234. if(compression != 0) return false; // TODO
  236. r.data.SetSize(len);
  237. ByteRead(r.data.GetData(), r.data.GetSize());
  239. res += r;
  241. return true;
  242. }
  244. bool CDSMSplitterFile::Read(__int64 len, IDSMChapterBagImpl& chap)
  245. {
  246. CDSMChapter c(0, L"");
  248. while(len > 0)
  249. {
  250. bool fSign = !!BitRead(1);
  251. int iTimeStamp = (int)BitRead(3);
  252. BitRead(4); // reserved
  253. len--;
  255. c.rt += (REFERENCE_TIME)BitRead(iTimeStamp<<3) * (fSign ? -1 : 1);
  256. len -= iTimeStamp;
  257. len -= Read(len, c.name);
  259. chap += c;
  260. }
  262. chap.ChapSort();
  264. return len == 0;
  265. }
  267. __int64 CDSMSplitterFile::Read(__int64 len, CStringW& str)
  268. {
  269. char c;
  270. CStringA s;
  271. __int64 i = 0;
  272. while(i++ < len && (c = (char)BitRead(8)) != 0) s += c;
  273. str = UTF8To16(s);
  274. return i;
  275. }
  277. __int64 CDSMSplitterFile::FindSyncPoint(REFERENCE_TIME rt)
  278. {
  279. if(/*!m_sps.IsEmpty()*/ m_sps.GetCount() > 1)
  280. {
  281. int i = range_bsearch(m_sps, m_rtFirst + rt);
  282. return i >= 0 ? m_sps[i].fp : 0;
  283. }
  285. if(m_rtDuration <= 0 || rt <= m_rtFirst)
  286. return 0;
  288. // ok, do the hard way then
  290. dsmp_t type;
  291. UINT64 syncpos, len;
  293. // 1. find some boundaries close to rt's position (minpos, maxpos)
  295. __int64 minpos = 0, maxpos = GetLength();
  297. for(int i = 0; i < 10 && (maxpos - minpos) >= 1024*1024; i++)
  298. {
  299. Seek((minpos + maxpos) / 2);
  301. while(GetPos() < maxpos)
  302. {
  303. if(!Sync(syncpos, type, len))
  304. continue;
  306. __int64 pos = GetPos();
  308. if(type == DSMP_SAMPLE)
  309. {
  310. Packet p;
  311. if(Read(len, &p, false) && p.rtStart != Packet::INVALID_TIME)
  312. {
  313. REFERENCE_TIME dt = (p.rtStart -= m_rtFirst) - rt;
  314. if(dt >= 0) maxpos = max((__int64)syncpos - 65536, minpos);
  315. else minpos = syncpos;
  316. break;
  317. }
  318. }
  320. Seek(pos + len);
  321. }
  322. }
  324. // 2. find the first packet just after rt (maxpos)
  326. Seek(minpos);
  328. while(GetPos() < GetLength())
  329. {
  330. if(!Sync(syncpos, type, len))
  331. continue;
  333. __int64 pos = GetPos();
  335. if(type == DSMP_SAMPLE)
  336. {
  337. Packet p;
  338. if(Read(len, &p, false) && p.rtStart != Packet::INVALID_TIME)
  339. {
  340. REFERENCE_TIME dt = (p.rtStart -= m_rtFirst) - rt;
  341. if(dt >= 0) {maxpos = (__int64)syncpos; break;}
  342. }
  343. }
  345. Seek(pos + len);
  346. }
  348. // 3. iterate backwards from maxpos and find at least one syncpoint for every stream, except for subtitle streams
  350. CAtlMap<BYTE,BYTE> ids;
  352. {
  353. POSITION pos = m_mts.GetStartPosition();
  354. while(pos)
  355. {
  356. BYTE id;
  357. CMediaType mt;
  358. m_mts.GetNextAssoc(pos, id, mt);
  359. if(mt.majortype != MEDIATYPE_Text && mt.majortype != MEDIATYPE_Subtitle)
  360. ids[id] = 0;
  361. }
  362. }
  364. __int64 ret = maxpos;
  366. while(maxpos > 0 && !ids.IsEmpty())
  367. {
  368. minpos = max(0, maxpos - 65536);
  370. Seek(minpos);
  372. while(Sync(syncpos, type, len) && GetPos() < maxpos)
  373. {
  374. UINT64 pos = GetPos();
  376. if(type == DSMP_SAMPLE)
  377. {
  378. Packet p;
  379. if(Read(len, &p, false) && p.rtStart != Packet::INVALID_TIME && p.bSyncPoint)
  380. {
  381. BYTE id = (BYTE)p.TrackNumber, tmp;
  382. if(ids.Lookup(id, tmp))
  383. {
  384. ids.RemoveKey((BYTE)p.TrackNumber);
  385. ret = min(ret, (__int64)syncpos);
  386. }
  387. }
  388. }
  390. Seek(pos + len);
  391. }
  393. maxpos = minpos;
  394. }
  396. return ret;
  397. }