nut.c 7.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190
  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 "internal.h"
  25. const AVCodecTag ff_nut_subtitle_tags[] = {
  26. { AV_CODEC_ID_TEXT , MKTAG('U', 'T', 'F', '8') },
  27. { AV_CODEC_ID_SSA , MKTAG('S', 'S', 'A', 0 ) },
  28. { AV_CODEC_ID_DVD_SUBTITLE, MKTAG('D', 'V', 'D', 'S') },
  29. { AV_CODEC_ID_DVB_SUBTITLE, MKTAG('D', 'V', 'B', 'S') },
  30. { AV_CODEC_ID_DVB_TELETEXT, MKTAG('D', 'V', 'B', 'T') },
  31. { AV_CODEC_ID_NONE , 0 }
  32. };
  33. const AVCodecTag ff_nut_video_tags[] = {
  34. { AV_CODEC_ID_RAWVIDEO, MKTAG('R', 'G', 'B', 15 ) },
  35. { AV_CODEC_ID_RAWVIDEO, MKTAG('B', 'G', 'R', 15 ) },
  36. { AV_CODEC_ID_RAWVIDEO, MKTAG('R', 'G', 'B', 16 ) },
  37. { AV_CODEC_ID_RAWVIDEO, MKTAG('B', 'G', 'R', 16 ) },
  38. { AV_CODEC_ID_RAWVIDEO, MKTAG(15 , 'B', 'G', 'R') },
  39. { AV_CODEC_ID_RAWVIDEO, MKTAG(15 , 'R', 'G', 'B') },
  40. { AV_CODEC_ID_RAWVIDEO, MKTAG(16 , 'B', 'G', 'R') },
  41. { AV_CODEC_ID_RAWVIDEO, MKTAG(16 , 'R', 'G', 'B') },
  42. { AV_CODEC_ID_RAWVIDEO, MKTAG('R', 'G', 'B', 12 ) },
  43. { AV_CODEC_ID_RAWVIDEO, MKTAG('B', 'G', 'R', 12 ) },
  44. { AV_CODEC_ID_RAWVIDEO, MKTAG(12 , 'B', 'G', 'R') },
  45. { AV_CODEC_ID_RAWVIDEO, MKTAG(12 , 'R', 'G', 'B') },
  46. { AV_CODEC_ID_RAWVIDEO, MKTAG('R', 'G', 'B', 'A') },
  47. { AV_CODEC_ID_RAWVIDEO, MKTAG('R', 'G', 'B', 0 ) },
  48. { AV_CODEC_ID_RAWVIDEO, MKTAG('B', 'G', 'R', 'A') },
  49. { AV_CODEC_ID_RAWVIDEO, MKTAG('B', 'G', 'R', 0 ) },
  50. { AV_CODEC_ID_RAWVIDEO, MKTAG('A', 'B', 'G', 'R') },
  51. { AV_CODEC_ID_RAWVIDEO, MKTAG( 0 , 'B', 'G', 'R') },
  52. { AV_CODEC_ID_RAWVIDEO, MKTAG('A', 'R', 'G', 'B') },
  53. { AV_CODEC_ID_RAWVIDEO, MKTAG( 0 , 'R', 'G', 'B') },
  54. { AV_CODEC_ID_RAWVIDEO, MKTAG('R', 'G', 'B', 24 ) },
  55. { AV_CODEC_ID_RAWVIDEO, MKTAG('B', 'G', 'R', 24 ) },
  56. { AV_CODEC_ID_RAWVIDEO, MKTAG('4', '1', '1', 'P') },
  57. { AV_CODEC_ID_RAWVIDEO, MKTAG('4', '2', '2', 'P') },
  58. { AV_CODEC_ID_RAWVIDEO, MKTAG('4', '2', '2', 'P') },
  59. { AV_CODEC_ID_RAWVIDEO, MKTAG('4', '4', '0', 'P') },
  60. { AV_CODEC_ID_RAWVIDEO, MKTAG('4', '4', '0', 'P') },
  61. { AV_CODEC_ID_RAWVIDEO, MKTAG('4', '4', '4', 'P') },
  62. { AV_CODEC_ID_RAWVIDEO, MKTAG('4', '4', '4', 'P') },
  63. { AV_CODEC_ID_RAWVIDEO, MKTAG('B', '1', 'W', '0') },
  64. { AV_CODEC_ID_RAWVIDEO, MKTAG('B', '0', 'W', '1') },
  65. { AV_CODEC_ID_RAWVIDEO, MKTAG('B', 'G', 'R', 8 ) },
  66. { AV_CODEC_ID_RAWVIDEO, MKTAG('R', 'G', 'B', 8 ) },
  67. { AV_CODEC_ID_RAWVIDEO, MKTAG('B', 'G', 'R', 4 ) },
  68. { AV_CODEC_ID_RAWVIDEO, MKTAG('R', 'G', 'B', 4 ) },
  69. { AV_CODEC_ID_RAWVIDEO, MKTAG('B', '4', 'B', 'Y') },
  70. { AV_CODEC_ID_RAWVIDEO, MKTAG('R', '4', 'B', 'Y') },
  71. { AV_CODEC_ID_RAWVIDEO, MKTAG('B', 'G', 'R', 48 ) },
  72. { AV_CODEC_ID_RAWVIDEO, MKTAG('R', 'G', 'B', 48 ) },
  73. { AV_CODEC_ID_RAWVIDEO, MKTAG(48 , 'B', 'G', 'R') },
  74. { AV_CODEC_ID_RAWVIDEO, MKTAG(48 , 'R', 'G', 'B') },
  75. { AV_CODEC_ID_RAWVIDEO, MKTAG('B', 'R', 'A', 64 ) },
  76. { AV_CODEC_ID_RAWVIDEO, MKTAG('R', 'B', 'A', 64 ) },
  77. { AV_CODEC_ID_RAWVIDEO, MKTAG(64 , 'B', 'R', 'A') },
  78. { AV_CODEC_ID_RAWVIDEO, MKTAG(64 , 'R', 'B', 'A') },
  79. { AV_CODEC_ID_RAWVIDEO, MKTAG('Y', '3', 11 , 10 ) },
  80. { AV_CODEC_ID_RAWVIDEO, MKTAG(10 , 11 , '3', 'Y') },
  81. { AV_CODEC_ID_RAWVIDEO, MKTAG('Y', '3', 10 , 10 ) },
  82. { AV_CODEC_ID_RAWVIDEO, MKTAG(10 , 10 , '3', 'Y') },
  83. { AV_CODEC_ID_RAWVIDEO, MKTAG('Y', '3', 0 , 10 ) },
  84. { AV_CODEC_ID_RAWVIDEO, MKTAG(10 , 0 , '3', 'Y') },
  85. { AV_CODEC_ID_RAWVIDEO, MKTAG('Y', '3', 11 , 12 ) },
  86. { AV_CODEC_ID_RAWVIDEO, MKTAG(12 , 11 , '3', 'Y') },
  87. { AV_CODEC_ID_RAWVIDEO, MKTAG('Y', '3', 10 , 12 ) },
  88. { AV_CODEC_ID_RAWVIDEO, MKTAG(12 , 10 , '3', 'Y') },
  89. { AV_CODEC_ID_RAWVIDEO, MKTAG('Y', '3', 0 , 12 ) },
  90. { AV_CODEC_ID_RAWVIDEO, MKTAG(12 , 0 , '3', 'Y') },
  91. { AV_CODEC_ID_RAWVIDEO, MKTAG('Y', '3', 11 , 14 ) },
  92. { AV_CODEC_ID_RAWVIDEO, MKTAG(14 , 11 , '3', 'Y') },
  93. { AV_CODEC_ID_RAWVIDEO, MKTAG('Y', '3', 10 , 14 ) },
  94. { AV_CODEC_ID_RAWVIDEO, MKTAG(14 , 10 , '3', 'Y') },
  95. { AV_CODEC_ID_RAWVIDEO, MKTAG('Y', '3', 0 , 14 ) },
  96. { AV_CODEC_ID_RAWVIDEO, MKTAG(14 , 0 , '3', 'Y') },
  97. { AV_CODEC_ID_RAWVIDEO, MKTAG('Y', '1', 0 , 16 ) },
  98. { AV_CODEC_ID_RAWVIDEO, MKTAG(16 , 0 , '1', 'Y') },
  99. { AV_CODEC_ID_RAWVIDEO, MKTAG('Y', '3', 11 , 16 ) },
  100. { AV_CODEC_ID_RAWVIDEO, MKTAG(16 , 11 , '3', 'Y') },
  101. { AV_CODEC_ID_RAWVIDEO, MKTAG('Y', '3', 10 , 16 ) },
  102. { AV_CODEC_ID_RAWVIDEO, MKTAG(16 , 10 , '3', 'Y') },
  103. { AV_CODEC_ID_RAWVIDEO, MKTAG('Y', '3', 0 , 16 ) },
  104. { AV_CODEC_ID_RAWVIDEO, MKTAG(16 , 0 , '3', 'Y') },
  105. { AV_CODEC_ID_RAWVIDEO, MKTAG('Y', '4', 11 , 8 ) },
  106. { AV_CODEC_ID_RAWVIDEO, MKTAG('Y', '4', 10 , 8 ) },
  107. { AV_CODEC_ID_RAWVIDEO, MKTAG('Y', '4', 0 , 8 ) },
  108. { AV_CODEC_ID_RAWVIDEO, MKTAG('Y', '2', 0 , 8 ) },
  109. { AV_CODEC_ID_NONE , 0 }
  110. };
  111. void ff_nut_reset_ts(NUTContext *nut, AVRational time_base, int64_t val){
  112. int i;
  113. for(i=0; i<nut->avf->nb_streams; i++){
  114. nut->stream[i].last_pts= av_rescale_rnd(
  115. val,
  116. time_base.num * (int64_t)nut->stream[i].time_base->den,
  117. time_base.den * (int64_t)nut->stream[i].time_base->num,
  118. AV_ROUND_DOWN);
  119. }
  120. }
  121. int64_t ff_lsb2full(StreamContext *stream, int64_t lsb){
  122. int64_t mask = (1<<stream->msb_pts_shift)-1;
  123. int64_t delta= stream->last_pts - mask/2;
  124. return ((lsb - delta)&mask) + delta;
  125. }
  126. int ff_nut_sp_pos_cmp(const Syncpoint *a, const Syncpoint *b){
  127. return ((a->pos - b->pos) >> 32) - ((b->pos - a->pos) >> 32);
  128. }
  129. int ff_nut_sp_pts_cmp(const Syncpoint *a, const Syncpoint *b){
  130. return ((a->ts - b->ts) >> 32) - ((b->ts - a->ts) >> 32);
  131. }
  132. void ff_nut_add_sp(NUTContext *nut, int64_t pos, int64_t back_ptr, int64_t ts){
  133. Syncpoint *sp= av_mallocz(sizeof(Syncpoint));
  134. struct AVTreeNode *node= av_mallocz(av_tree_node_size);
  135. nut->sp_count++;
  136. sp->pos= pos;
  137. sp->back_ptr= back_ptr;
  138. sp->ts= ts;
  139. av_tree_insert(&nut->syncpoints, sp, (void *) ff_nut_sp_pos_cmp, &node);
  140. if(node){
  141. av_free(sp);
  142. av_free(node);
  143. }
  144. }
  145. static int enu_free(void *opaque, void *elem)
  146. {
  147. av_free(elem);
  148. return 0;
  149. }
  150. void ff_nut_free_sp(NUTContext *nut)
  151. {
  152. av_tree_enumerate(nut->syncpoints, NULL, NULL, enu_free);
  153. av_tree_destroy(nut->syncpoints);
  154. }
  155. const Dispositions ff_nut_dispositions[] = {
  156. {"default" , AV_DISPOSITION_DEFAULT},
  157. {"dub" , AV_DISPOSITION_DUB},
  158. {"original" , AV_DISPOSITION_ORIGINAL},
  159. {"comment" , AV_DISPOSITION_COMMENT},
  160. {"lyrics" , AV_DISPOSITION_LYRICS},
  161. {"karaoke" , AV_DISPOSITION_KARAOKE},
  162. {"" , 0}
  163. };
  164. const AVMetadataConv ff_nut_metadata_conv[] = {
  165. { "Author", "artist" },
  166. { "X-CreationTime", "date" },
  167. { "CreationTime", "date" },
  168. { "SourceFilename", "filename" },
  169. { "X-Language", "language" },
  170. { "X-Disposition", "disposition" },
  171. { "X-Replaces", "replaces" },
  172. { "X-Depends", "depends" },
  173. { "X-Uses", "uses" },
  174. { "X-UsesFont", "usesfont" },
  175. { 0 },
  176. };