extract_mvs.c 5.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185
  1. /*
  2. * Copyright (c) 2012 Stefano Sabatini
  3. * Copyright (c) 2014 Clément Bœsch
  4. *
  5. * Permission is hereby granted, free of charge, to any person obtaining a copy
  6. * of this software and associated documentation files (the "Software"), to deal
  7. * in the Software without restriction, including without limitation the rights
  8. * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  9. * copies of the Software, and to permit persons to whom the Software is
  10. * furnished to do so, subject to the following conditions:
  11. *
  12. * The above copyright notice and this permission notice shall be included in
  13. * all copies or substantial portions of the Software.
  14. *
  15. * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  16. * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  17. * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
  18. * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  19. * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  20. * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  21. * THE SOFTWARE.
  22. */
  23. #include <libavutil/motion_vector.h>
  24. #include <libavformat/avformat.h>
  25. static AVFormatContext *fmt_ctx = NULL;
  26. static AVCodecContext *video_dec_ctx = NULL;
  27. static AVStream *video_stream = NULL;
  28. static const char *src_filename = NULL;
  29. static int video_stream_idx = -1;
  30. static AVFrame *frame = NULL;
  31. static AVPacket pkt;
  32. static int video_frame_count = 0;
  33. static int decode_packet(int *got_frame, int cached)
  34. {
  35. int decoded = pkt.size;
  36. *got_frame = 0;
  37. if (pkt.stream_index == video_stream_idx) {
  38. int ret = avcodec_decode_video2(video_dec_ctx, frame, got_frame, &pkt);
  39. if (ret < 0) {
  40. fprintf(stderr, "Error decoding video frame (%s)\n", av_err2str(ret));
  41. return ret;
  42. }
  43. if (*got_frame) {
  44. int i;
  45. AVFrameSideData *sd;
  46. video_frame_count++;
  47. sd = av_frame_get_side_data(frame, AV_FRAME_DATA_MOTION_VECTORS);
  48. if (sd) {
  49. const AVMotionVector *mvs = (const AVMotionVector *)sd->data;
  50. for (i = 0; i < sd->size / sizeof(*mvs); i++) {
  51. const AVMotionVector *mv = &mvs[i];
  52. printf("%d,%2d,%2d,%2d,%4d,%4d,%4d,%4d,0x%"PRIx64"\n",
  53. video_frame_count, mv->source,
  54. mv->w, mv->h, mv->src_x, mv->src_y,
  55. mv->dst_x, mv->dst_y, mv->flags);
  56. }
  57. }
  58. }
  59. }
  60. return decoded;
  61. }
  62. static int open_codec_context(int *stream_idx,
  63. AVFormatContext *fmt_ctx, enum AVMediaType type)
  64. {
  65. int ret;
  66. AVStream *st;
  67. AVCodecContext *dec_ctx = NULL;
  68. AVCodec *dec = NULL;
  69. AVDictionary *opts = NULL;
  70. ret = av_find_best_stream(fmt_ctx, type, -1, -1, NULL, 0);
  71. if (ret < 0) {
  72. fprintf(stderr, "Could not find %s stream in input file '%s'\n",
  73. av_get_media_type_string(type), src_filename);
  74. return ret;
  75. } else {
  76. *stream_idx = ret;
  77. st = fmt_ctx->streams[*stream_idx];
  78. /* find decoder for the stream */
  79. dec_ctx = st->codec;
  80. dec = avcodec_find_decoder(dec_ctx->codec_id);
  81. if (!dec) {
  82. fprintf(stderr, "Failed to find %s codec\n",
  83. av_get_media_type_string(type));
  84. return AVERROR(EINVAL);
  85. }
  86. /* Init the video decoder */
  87. av_dict_set(&opts, "flags2", "+export_mvs", 0);
  88. if ((ret = avcodec_open2(dec_ctx, dec, &opts)) < 0) {
  89. fprintf(stderr, "Failed to open %s codec\n",
  90. av_get_media_type_string(type));
  91. return ret;
  92. }
  93. }
  94. return 0;
  95. }
  96. int main(int argc, char **argv)
  97. {
  98. int ret = 0, got_frame;
  99. if (argc != 2) {
  100. fprintf(stderr, "Usage: %s <video>\n", argv[0]);
  101. exit(1);
  102. }
  103. src_filename = argv[1];
  104. av_register_all();
  105. if (avformat_open_input(&fmt_ctx, src_filename, NULL, NULL) < 0) {
  106. fprintf(stderr, "Could not open source file %s\n", src_filename);
  107. exit(1);
  108. }
  109. if (avformat_find_stream_info(fmt_ctx, NULL) < 0) {
  110. fprintf(stderr, "Could not find stream information\n");
  111. exit(1);
  112. }
  113. if (open_codec_context(&video_stream_idx, fmt_ctx, AVMEDIA_TYPE_VIDEO) >= 0) {
  114. video_stream = fmt_ctx->streams[video_stream_idx];
  115. video_dec_ctx = video_stream->codec;
  116. }
  117. av_dump_format(fmt_ctx, 0, src_filename, 0);
  118. if (!video_stream) {
  119. fprintf(stderr, "Could not find video stream in the input, aborting\n");
  120. ret = 1;
  121. goto end;
  122. }
  123. frame = av_frame_alloc();
  124. if (!frame) {
  125. fprintf(stderr, "Could not allocate frame\n");
  126. ret = AVERROR(ENOMEM);
  127. goto end;
  128. }
  129. printf("framenum,source,blockw,blockh,srcx,srcy,dstx,dsty,flags\n");
  130. /* initialize packet, set data to NULL, let the demuxer fill it */
  131. av_init_packet(&pkt);
  132. pkt.data = NULL;
  133. pkt.size = 0;
  134. /* read frames from the file */
  135. while (av_read_frame(fmt_ctx, &pkt) >= 0) {
  136. AVPacket orig_pkt = pkt;
  137. do {
  138. ret = decode_packet(&got_frame, 0);
  139. if (ret < 0)
  140. break;
  141. pkt.data += ret;
  142. pkt.size -= ret;
  143. } while (pkt.size > 0);
  144. av_free_packet(&orig_pkt);
  145. }
  146. /* flush cached frames */
  147. pkt.data = NULL;
  148. pkt.size = 0;
  149. do {
  150. decode_packet(&got_frame, 1);
  151. } while (got_frame);
  152. end:
  153. avcodec_close(video_dec_ctx);
  154. avformat_close_input(&fmt_ctx);
  155. av_frame_free(&frame);
  156. return ret < 0;
  157. }