You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

307 lines
14KB

  1. /*
  2. * nut
  3. * Copyright (c) 2004-2007 Michael Niedermayer
  4. *
  5. * This file is part of FFmpeg.
  6. *
  7. * FFmpeg is free software; you can redistribute it and/or
  8. * modify it under the terms of the GNU Lesser General Public
  9. * License as published by the Free Software Foundation; either
  10. * version 2.1 of the License, or (at your option) any later version.
  11. *
  12. * FFmpeg is distributed in the hope that it will be useful,
  13. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  15. * Lesser General Public License for more details.
  16. *
  17. * You should have received a copy of the GNU Lesser General Public
  18. * License along with FFmpeg; if not, write to the Free Software
  19. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  20. */
  21. #include "libavutil/mathematics.h"
  22. #include "libavutil/tree.h"
  23. #include "nut.h"
  24. #include "riff.h"
  25. #include "internal.h"
  26. const AVCodecTag ff_nut_subtitle_tags[] = {
  27. { AV_CODEC_ID_TEXT, MKTAG('U', 'T', 'F', '8') },
  28. { AV_CODEC_ID_SSA, MKTAG('S', 'S', 'A', 0 ) },
  29. { AV_CODEC_ID_DVD_SUBTITLE, MKTAG('D', 'V', 'D', 'S') },
  30. { AV_CODEC_ID_DVB_SUBTITLE, MKTAG('D', 'V', 'B', 'S') },
  31. { AV_CODEC_ID_DVB_TELETEXT, MKTAG('D', 'V', 'B', 'T') },
  32. { AV_CODEC_ID_NONE, 0 }
  33. };
  34. const AVCodecTag ff_nut_data_tags[] = {
  35. { AV_CODEC_ID_TEXT, MKTAG('U', 'T', 'F', '8') },
  36. { AV_CODEC_ID_NONE, 0 }
  37. };
  38. const AVCodecTag ff_nut_video_tags[] = {
  39. { AV_CODEC_ID_VP9, MKTAG('V', 'P', '9', '0') },
  40. { AV_CODEC_ID_RAWVIDEO, MKTAG('R', 'G', 'B', 15 ) },
  41. { AV_CODEC_ID_RAWVIDEO, MKTAG('B', 'G', 'R', 15 ) },
  42. { AV_CODEC_ID_RAWVIDEO, MKTAG('R', 'G', 'B', 16 ) },
  43. { AV_CODEC_ID_RAWVIDEO, MKTAG('B', 'G', 'R', 16 ) },
  44. { AV_CODEC_ID_RAWVIDEO, MKTAG(15 , 'B', 'G', 'R') },
  45. { AV_CODEC_ID_RAWVIDEO, MKTAG(15 , 'R', 'G', 'B') },
  46. { AV_CODEC_ID_RAWVIDEO, MKTAG(16 , 'B', 'G', 'R') },
  47. { AV_CODEC_ID_RAWVIDEO, MKTAG(16 , 'R', 'G', 'B') },
  48. { AV_CODEC_ID_RAWVIDEO, MKTAG('R', 'G', 'B', 12 ) },
  49. { AV_CODEC_ID_RAWVIDEO, MKTAG('B', 'G', 'R', 12 ) },
  50. { AV_CODEC_ID_RAWVIDEO, MKTAG(12 , 'B', 'G', 'R') },
  51. { AV_CODEC_ID_RAWVIDEO, MKTAG(12 , 'R', 'G', 'B') },
  52. { AV_CODEC_ID_RAWVIDEO, MKTAG('R', 'G', 'B', 'A') },
  53. { AV_CODEC_ID_RAWVIDEO, MKTAG('R', 'G', 'B', 0 ) },
  54. { AV_CODEC_ID_RAWVIDEO, MKTAG('B', 'G', 'R', 'A') },
  55. { AV_CODEC_ID_RAWVIDEO, MKTAG('B', 'G', 'R', 0 ) },
  56. { AV_CODEC_ID_RAWVIDEO, MKTAG('A', 'B', 'G', 'R') },
  57. { AV_CODEC_ID_RAWVIDEO, MKTAG( 0 , 'B', 'G', 'R') },
  58. { AV_CODEC_ID_RAWVIDEO, MKTAG('A', 'R', 'G', 'B') },
  59. { AV_CODEC_ID_RAWVIDEO, MKTAG( 0 , 'R', 'G', 'B') },
  60. { AV_CODEC_ID_RAWVIDEO, MKTAG('R', 'G', 'B', 24 ) },
  61. { AV_CODEC_ID_RAWVIDEO, MKTAG('B', 'G', 'R', 24 ) },
  62. { AV_CODEC_ID_RAWVIDEO, MKTAG('4', '1', '1', 'P') },
  63. { AV_CODEC_ID_RAWVIDEO, MKTAG('4', '2', '2', 'P') },
  64. { AV_CODEC_ID_RAWVIDEO, MKTAG('4', '2', '2', 'P') },
  65. { AV_CODEC_ID_RAWVIDEO, MKTAG('4', '4', '0', 'P') },
  66. { AV_CODEC_ID_RAWVIDEO, MKTAG('4', '4', '0', 'P') },
  67. { AV_CODEC_ID_RAWVIDEO, MKTAG('4', '4', '4', 'P') },
  68. { AV_CODEC_ID_RAWVIDEO, MKTAG('4', '4', '4', 'P') },
  69. { AV_CODEC_ID_RAWVIDEO, MKTAG('B', '1', 'W', '0') },
  70. { AV_CODEC_ID_RAWVIDEO, MKTAG('B', '0', 'W', '1') },
  71. { AV_CODEC_ID_RAWVIDEO, MKTAG('B', 'G', 'R', 8 ) },
  72. { AV_CODEC_ID_RAWVIDEO, MKTAG('R', 'G', 'B', 8 ) },
  73. { AV_CODEC_ID_RAWVIDEO, MKTAG('B', 'G', 'R', 4 ) },
  74. { AV_CODEC_ID_RAWVIDEO, MKTAG('R', 'G', 'B', 4 ) },
  75. { AV_CODEC_ID_RAWVIDEO, MKTAG('B', '4', 'B', 'Y') },
  76. { AV_CODEC_ID_RAWVIDEO, MKTAG('R', '4', 'B', 'Y') },
  77. { AV_CODEC_ID_RAWVIDEO, MKTAG('B', 'G', 'R', 48 ) },
  78. { AV_CODEC_ID_RAWVIDEO, MKTAG('R', 'G', 'B', 48 ) },
  79. { AV_CODEC_ID_RAWVIDEO, MKTAG(48 , 'B', 'G', 'R') },
  80. { AV_CODEC_ID_RAWVIDEO, MKTAG(48 , 'R', 'G', 'B') },
  81. { AV_CODEC_ID_RAWVIDEO, MKTAG('R', 'B', 'A', 64 ) },
  82. { AV_CODEC_ID_RAWVIDEO, MKTAG('B', 'R', 'A', 64 ) },
  83. { AV_CODEC_ID_RAWVIDEO, MKTAG(64 , 'R', 'B', 'A') },
  84. { AV_CODEC_ID_RAWVIDEO, MKTAG(64 , 'B', 'R', 'A') },
  85. { AV_CODEC_ID_RAWVIDEO, MKTAG('Y', '3', 11 , 10 ) },
  86. { AV_CODEC_ID_RAWVIDEO, MKTAG(10 , 11 , '3', 'Y') },
  87. { AV_CODEC_ID_RAWVIDEO, MKTAG('Y', '3', 10 , 10 ) },
  88. { AV_CODEC_ID_RAWVIDEO, MKTAG(10 , 10 , '3', 'Y') },
  89. { AV_CODEC_ID_RAWVIDEO, MKTAG('Y', '3', 0 , 10 ) },
  90. { AV_CODEC_ID_RAWVIDEO, MKTAG(10 , 0 , '3', 'Y') },
  91. { AV_CODEC_ID_RAWVIDEO, MKTAG('Y', '3', 11 , 12 ) },
  92. { AV_CODEC_ID_RAWVIDEO, MKTAG(12 , 11 , '3', 'Y') },
  93. { AV_CODEC_ID_RAWVIDEO, MKTAG('Y', '3', 10 , 12 ) },
  94. { AV_CODEC_ID_RAWVIDEO, MKTAG(12 , 10 , '3', 'Y') },
  95. { AV_CODEC_ID_RAWVIDEO, MKTAG('Y', '3', 0 , 12 ) },
  96. { AV_CODEC_ID_RAWVIDEO, MKTAG(12 , 0 , '3', 'Y') },
  97. { AV_CODEC_ID_RAWVIDEO, MKTAG('Y', '3', 11 , 14 ) },
  98. { AV_CODEC_ID_RAWVIDEO, MKTAG(14 , 11 , '3', 'Y') },
  99. { AV_CODEC_ID_RAWVIDEO, MKTAG('Y', '3', 10 , 14 ) },
  100. { AV_CODEC_ID_RAWVIDEO, MKTAG(14 , 10 , '3', 'Y') },
  101. { AV_CODEC_ID_RAWVIDEO, MKTAG('Y', '3', 0 , 14 ) },
  102. { AV_CODEC_ID_RAWVIDEO, MKTAG(14 , 0 , '3', 'Y') },
  103. { AV_CODEC_ID_RAWVIDEO, MKTAG('Y', '1', 0 , 16 ) },
  104. { AV_CODEC_ID_RAWVIDEO, MKTAG(16 , 0 , '1', 'Y') },
  105. { AV_CODEC_ID_RAWVIDEO, MKTAG('Y', '3', 11 , 16 ) },
  106. { AV_CODEC_ID_RAWVIDEO, MKTAG(16 , 11 , '3', 'Y') },
  107. { AV_CODEC_ID_RAWVIDEO, MKTAG('Y', '3', 10 , 16 ) },
  108. { AV_CODEC_ID_RAWVIDEO, MKTAG(16 , 10 , '3', 'Y') },
  109. { AV_CODEC_ID_RAWVIDEO, MKTAG('Y', '3', 0 , 16 ) },
  110. { AV_CODEC_ID_RAWVIDEO, MKTAG(16 , 0 , '3', 'Y') },
  111. { AV_CODEC_ID_RAWVIDEO, MKTAG('Y', '4', 11 , 8 ) },
  112. { AV_CODEC_ID_RAWVIDEO, MKTAG('Y', '4', 10 , 8 ) },
  113. { AV_CODEC_ID_RAWVIDEO, MKTAG('Y', '4', 0 , 8 ) },
  114. { AV_CODEC_ID_RAWVIDEO, MKTAG('Y', '2', 0 , 8 ) },
  115. { AV_CODEC_ID_RAWVIDEO, MKTAG('Y', '1', 0, 9) },
  116. { AV_CODEC_ID_RAWVIDEO, MKTAG(9, 0, '1', 'Y') },
  117. { AV_CODEC_ID_RAWVIDEO, MKTAG('Y', '4', 11, 9) },
  118. { AV_CODEC_ID_RAWVIDEO, MKTAG(9, 11, '4', 'Y') },
  119. { AV_CODEC_ID_RAWVIDEO, MKTAG('Y', '4', 10, 9) },
  120. { AV_CODEC_ID_RAWVIDEO, MKTAG(9, 10, '4', 'Y') },
  121. { AV_CODEC_ID_RAWVIDEO, MKTAG('Y', '4', 0, 9) },
  122. { AV_CODEC_ID_RAWVIDEO, MKTAG(9, 0, '4', 'Y') },
  123. { AV_CODEC_ID_RAWVIDEO, MKTAG('Y', '1', 0, 10) },
  124. { AV_CODEC_ID_RAWVIDEO, MKTAG(10, 0, '1', 'Y') },
  125. { AV_CODEC_ID_RAWVIDEO, MKTAG('Y', '4', 11, 10) },
  126. { AV_CODEC_ID_RAWVIDEO, MKTAG(10, 11, '4', 'Y') },
  127. { AV_CODEC_ID_RAWVIDEO, MKTAG('Y', '4', 10, 10) },
  128. { AV_CODEC_ID_RAWVIDEO, MKTAG(10, 10, '4', 'Y') },
  129. { AV_CODEC_ID_RAWVIDEO, MKTAG('Y', '4', 0, 10) },
  130. { AV_CODEC_ID_RAWVIDEO, MKTAG(10, 0, '4', 'Y') },
  131. { AV_CODEC_ID_RAWVIDEO, MKTAG('Y', '1', 0, 16) },
  132. { AV_CODEC_ID_RAWVIDEO, MKTAG(16, 0, '1', 'Y') },
  133. { AV_CODEC_ID_RAWVIDEO, MKTAG('Y', '4', 11, 16) },
  134. { AV_CODEC_ID_RAWVIDEO, MKTAG(16, 11, '4', 'Y') },
  135. { AV_CODEC_ID_RAWVIDEO, MKTAG('Y', '4', 10, 16) },
  136. { AV_CODEC_ID_RAWVIDEO, MKTAG(16, 10, '4', 'Y') },
  137. { AV_CODEC_ID_RAWVIDEO, MKTAG('Y', '4', 0, 16) },
  138. { AV_CODEC_ID_RAWVIDEO, MKTAG(16, 0, '4', 'Y') },
  139. { AV_CODEC_ID_RAWVIDEO, MKTAG('G', '3', 0, 8) },
  140. { AV_CODEC_ID_RAWVIDEO, MKTAG('G', '3', 0, 9) },
  141. { AV_CODEC_ID_RAWVIDEO, MKTAG( 9, 0, '3', 'G') },
  142. { AV_CODEC_ID_RAWVIDEO, MKTAG('G', '3', 0, 10) },
  143. { AV_CODEC_ID_RAWVIDEO, MKTAG(10, 0, '3', 'G') },
  144. { AV_CODEC_ID_RAWVIDEO, MKTAG('G', '3', 0, 12) },
  145. { AV_CODEC_ID_RAWVIDEO, MKTAG(12, 0, '3', 'G') },
  146. { AV_CODEC_ID_RAWVIDEO, MKTAG('G', '3', 0, 14) },
  147. { AV_CODEC_ID_RAWVIDEO, MKTAG(14, 0, '3', 'G') },
  148. { AV_CODEC_ID_RAWVIDEO, MKTAG('G', '3', 0, 16) },
  149. { AV_CODEC_ID_RAWVIDEO, MKTAG(16, 0, '3', 'G') },
  150. { AV_CODEC_ID_RAWVIDEO, MKTAG('X', 'Y', 'Z' , 36 ) },
  151. { AV_CODEC_ID_RAWVIDEO, MKTAG(36 , 'Z' , 'Y', 'X') },
  152. { AV_CODEC_ID_RAWVIDEO, MKTAG(0xBA, 'B', 'G', 8 ) },
  153. { AV_CODEC_ID_RAWVIDEO, MKTAG(0xBA, 'B', 'G', 16 ) },
  154. { AV_CODEC_ID_RAWVIDEO, MKTAG(16 , 'G', 'B', 0xBA) },
  155. { AV_CODEC_ID_RAWVIDEO, MKTAG(0xBA, 'R', 'G', 8 ) },
  156. { AV_CODEC_ID_RAWVIDEO, MKTAG(0xBA, 'R', 'G', 16 ) },
  157. { AV_CODEC_ID_RAWVIDEO, MKTAG(16 , 'G', 'R', 0xBA) },
  158. { AV_CODEC_ID_RAWVIDEO, MKTAG(0xBA, 'G', 'B', 8 ) },
  159. { AV_CODEC_ID_RAWVIDEO, MKTAG(0xBA, 'G', 'B', 16 ) },
  160. { AV_CODEC_ID_RAWVIDEO, MKTAG(16, 'B', 'G', 0xBA) },
  161. { AV_CODEC_ID_RAWVIDEO, MKTAG(0xBA, 'G', 'R', 8 ) },
  162. { AV_CODEC_ID_RAWVIDEO, MKTAG(0xBA, 'G', 'R', 16 ) },
  163. { AV_CODEC_ID_RAWVIDEO, MKTAG(16, 'R', 'G', 0xBA) },
  164. { AV_CODEC_ID_NONE, 0 }
  165. };
  166. const AVCodecTag ff_nut_audio_extra_tags[] = {
  167. { AV_CODEC_ID_COMFORT_NOISE, MKTAG('3', '3', '8', '9') },
  168. { AV_CODEC_ID_PCM_ALAW, MKTAG('A', 'L', 'A', 'W') },
  169. { AV_CODEC_ID_PCM_MULAW, MKTAG('U', 'L', 'A', 'W') },
  170. { AV_CODEC_ID_MP3, MKTAG('M', 'P', '3', ' ') },
  171. { AV_CODEC_ID_NONE, 0 }
  172. };
  173. const AVCodecTag ff_nut_audio_tags[] = {
  174. { AV_CODEC_ID_PCM_F32BE, MKTAG(32 , 'D', 'F', 'P') },
  175. { AV_CODEC_ID_PCM_F32LE, MKTAG('P', 'F', 'D', 32 ) },
  176. { AV_CODEC_ID_PCM_F64BE, MKTAG(64 , 'D', 'F', 'P') },
  177. { AV_CODEC_ID_PCM_F64LE, MKTAG('P', 'F', 'D', 64 ) },
  178. { AV_CODEC_ID_PCM_S16BE, MKTAG(16 , 'D', 'S', 'P') },
  179. { AV_CODEC_ID_PCM_S16LE, MKTAG('P', 'S', 'D', 16 ) },
  180. { AV_CODEC_ID_PCM_S24BE, MKTAG(24 , 'D', 'S', 'P') },
  181. { AV_CODEC_ID_PCM_S24LE, MKTAG('P', 'S', 'D', 24 ) },
  182. { AV_CODEC_ID_PCM_S32BE, MKTAG(32 , 'D', 'S', 'P') },
  183. { AV_CODEC_ID_PCM_S32LE, MKTAG('P', 'S', 'D', 32 ) },
  184. { AV_CODEC_ID_PCM_S8, MKTAG('P', 'S', 'D', 8 ) },
  185. { AV_CODEC_ID_PCM_U16BE, MKTAG(16 , 'D', 'U', 'P') },
  186. { AV_CODEC_ID_PCM_U16LE, MKTAG('P', 'U', 'D', 16 ) },
  187. { AV_CODEC_ID_PCM_U24BE, MKTAG(24 , 'D', 'U', 'P') },
  188. { AV_CODEC_ID_PCM_U24LE, MKTAG('P', 'U', 'D', 24 ) },
  189. { AV_CODEC_ID_PCM_U32BE, MKTAG(32 , 'D', 'U', 'P') },
  190. { AV_CODEC_ID_PCM_U32LE, MKTAG('P', 'U', 'D', 32 ) },
  191. { AV_CODEC_ID_PCM_U8, MKTAG('P', 'U', 'D', 8 ) },
  192. { AV_CODEC_ID_PCM_S8_PLANAR, MKTAG('P', 'S', 'P', 8 ) },
  193. { AV_CODEC_ID_PCM_S16BE_PLANAR, MKTAG(16 , 'P', 'S', 'P') },
  194. { AV_CODEC_ID_PCM_S16LE_PLANAR, MKTAG('P', 'S', 'P', 16 ) },
  195. { AV_CODEC_ID_PCM_S24LE_PLANAR, MKTAG('P', 'S', 'P', 24 ) },
  196. { AV_CODEC_ID_PCM_S32LE_PLANAR, MKTAG('P', 'S', 'P', 32 ) },
  197. { AV_CODEC_ID_NONE, 0 }
  198. };
  199. const AVCodecTag * const ff_nut_codec_tags[] = {
  200. ff_nut_video_tags, ff_nut_audio_tags, ff_nut_subtitle_tags,
  201. ff_codec_bmp_tags, ff_codec_wav_tags, ff_nut_audio_extra_tags, ff_nut_data_tags, 0
  202. };
  203. void ff_nut_reset_ts(NUTContext *nut, AVRational time_base, int64_t val)
  204. {
  205. int i;
  206. for (i = 0; i < nut->avf->nb_streams; i++)
  207. nut->stream[i].last_pts =
  208. av_rescale_rnd(val,
  209. time_base.num * (int64_t)nut->stream[i].time_base->den,
  210. time_base.den * (int64_t)nut->stream[i].time_base->num,
  211. AV_ROUND_DOWN);
  212. }
  213. int64_t ff_lsb2full(StreamContext *stream, int64_t lsb)
  214. {
  215. int64_t mask = (1ULL << stream->msb_pts_shift) - 1;
  216. int64_t delta = stream->last_pts - mask / 2;
  217. return ((lsb - delta) & mask) + delta;
  218. }
  219. int ff_nut_sp_pos_cmp(const Syncpoint *a, const Syncpoint *b)
  220. {
  221. return ((a->pos - b->pos) >> 32) - ((b->pos - a->pos) >> 32);
  222. }
  223. int ff_nut_sp_pts_cmp(const Syncpoint *a, const Syncpoint *b)
  224. {
  225. return ((a->ts - b->ts) >> 32) - ((b->ts - a->ts) >> 32);
  226. }
  227. int ff_nut_add_sp(NUTContext *nut, int64_t pos, int64_t back_ptr, int64_t ts)
  228. {
  229. Syncpoint *sp = av_mallocz(sizeof(Syncpoint));
  230. struct AVTreeNode *node = av_tree_node_alloc();
  231. if (!sp || !node) {
  232. av_freep(&sp);
  233. av_freep(&node);
  234. return AVERROR(ENOMEM);
  235. }
  236. nut->sp_count++;
  237. sp->pos = pos;
  238. sp->back_ptr = back_ptr;
  239. sp->ts = ts;
  240. av_tree_insert(&nut->syncpoints, sp, (void *) ff_nut_sp_pos_cmp, &node);
  241. if (node) {
  242. av_free(sp);
  243. av_free(node);
  244. }
  245. return 0;
  246. }
  247. static int enu_free(void *opaque, void *elem)
  248. {
  249. av_free(elem);
  250. return 0;
  251. }
  252. void ff_nut_free_sp(NUTContext *nut)
  253. {
  254. av_tree_enumerate(nut->syncpoints, NULL, NULL, enu_free);
  255. av_tree_destroy(nut->syncpoints);
  256. }
  257. const Dispositions ff_nut_dispositions[] = {
  258. { "default", AV_DISPOSITION_DEFAULT },
  259. { "dub", AV_DISPOSITION_DUB },
  260. { "original", AV_DISPOSITION_ORIGINAL },
  261. { "comment", AV_DISPOSITION_COMMENT },
  262. { "lyrics", AV_DISPOSITION_LYRICS },
  263. { "karaoke", AV_DISPOSITION_KARAOKE },
  264. { "", 0 }
  265. };
  266. const AVMetadataConv ff_nut_metadata_conv[] = {
  267. { "Author", "artist" },
  268. { "X-CreationTime", "date" },
  269. { "CreationTime", "date" },
  270. { "SourceFilename", "filename" },
  271. { "X-Language", "language" },
  272. { "X-Disposition", "disposition" },
  273. { "X-Replaces", "replaces" },
  274. { "X-Depends", "depends" },
  275. { "X-Uses", "uses" },
  276. { "X-UsesFont", "usesfont" },
  277. { 0 },
  278. };