FFmpeg  4.0
api-seek-test.c
Go to the documentation of this file.
1 /*
2  * Copyright (c) 2015 Ludmila Glinskih
3  *
4  * Permission is hereby granted, free of charge, to any person obtaining a copy
5  * of this software and associated documentation files (the "Software"), to deal
6  * in the Software without restriction, including without limitation the rights
7  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
8  * copies of the Software, and to permit persons to whom the Software is
9  * furnished to do so, subject to the following conditions:
10  *
11  * The above copyright notice and this permission notice shall be included in
12  * all copies or substantial portions of the Software.
13  *
14  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
15  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
16  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
17  * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
18  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
19  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
20  * THE SOFTWARE.
21  */
22 
23 /**
24  * Seek test.
25  */
26 
27 #include "libavutil/adler32.h"
28 #include "libavcodec/avcodec.h"
29 #include "libavformat/avformat.h"
30 #include "libavutil/imgutils.h"
31 
32 int64_t *pts_array;
33 uint32_t *crc_array;
36 
37 static int add_crc_to_array(uint32_t crc, int64_t pts)
38 {
40  if (size_of_array == 0)
41  size_of_array = 10;
42  size_of_array *= 2;
43  crc_array = av_realloc_f(crc_array, size_of_array, sizeof(uint32_t));
44  pts_array = av_realloc_f(pts_array, size_of_array, sizeof(int64_t));
45  if ((crc_array == NULL) || (pts_array == NULL)) {
46  av_log(NULL, AV_LOG_ERROR, "Can't allocate array to store crcs\n");
47  return AVERROR(ENOMEM);
48  }
49  }
53  return 0;
54 }
55 
56 static int compare_crc_in_array(uint32_t crc, int64_t pts)
57 {
58  int i;
59  for (i = 0; i < number_of_elements; i++) {
60  if (pts_array[i] == pts) {
61  if (crc_array[i] == crc) {
62  printf("Comparing 0x%08"PRIx32" %"PRId64" %d is OK\n", crc, pts, i);
63  return 0;
64  }
65  else {
66  av_log(NULL, AV_LOG_ERROR, "Incorrect crc of a frame after seeking\n");
67  return -1;
68  }
69  }
70  }
71  av_log(NULL, AV_LOG_ERROR, "Incorrect pts of a frame after seeking\n");
72  return -1;
73 }
74 
76  AVCodecContext *ctx, AVFrame *fr, uint64_t ts_start, uint64_t ts_end, int no_seeking)
77 {
78  int number_of_written_bytes;
79  int got_frame = 0;
80  int result;
81  int end_of_stream = 0;
82  int byte_buffer_size;
83  uint8_t *byte_buffer;
84  uint32_t crc;
85  AVPacket pkt;
86 
87  byte_buffer_size = av_image_get_buffer_size(ctx->pix_fmt, ctx->width, ctx->height, 16);
88  byte_buffer = av_malloc(byte_buffer_size);
89  if (!byte_buffer) {
90  av_log(NULL, AV_LOG_ERROR, "Can't allocate buffer\n");
91  return AVERROR(ENOMEM);
92  }
93 
94  if (!no_seeking) {
95  result = av_seek_frame(fmt_ctx, video_stream, ts_start, AVSEEK_FLAG_ANY);
96  printf("Seeking to %"PRId64", computing crc for frames with pts < %"PRId64"\n", ts_start, ts_end);
97  if (result < 0) {
98  av_log(NULL, AV_LOG_ERROR, "Error in seeking\n");
99  return result;
100  }
102  }
103 
104  av_init_packet(&pkt);
105  do {
106  if (!end_of_stream)
107  if (av_read_frame(fmt_ctx, &pkt) < 0)
108  end_of_stream = 1;
109  if (end_of_stream) {
110  pkt.data = NULL;
111  pkt.size = 0;
112  }
113  if (pkt.stream_index == video_stream || end_of_stream) {
114  got_frame = 0;
115  if ((pkt.pts == AV_NOPTS_VALUE) && (!end_of_stream)) {
116  av_log(NULL, AV_LOG_ERROR, "Error: frames doesn't have pts values\n");
117  return -1;
118  }
119  result = avcodec_decode_video2(ctx, fr, &got_frame, &pkt);
120  if (result < 0) {
121  av_log(NULL, AV_LOG_ERROR, "Error decoding frame\n");
122  return result;
123  }
124  if (got_frame) {
125  number_of_written_bytes = av_image_copy_to_buffer(byte_buffer, byte_buffer_size,
126  (const uint8_t* const *)fr->data, (const int*) fr->linesize,
127  ctx->pix_fmt, ctx->width, ctx->height, 1);
128  if (number_of_written_bytes < 0) {
129  av_log(NULL, AV_LOG_ERROR, "Can't copy image to buffer\n");
130  return number_of_written_bytes;
131  }
132  if ((!no_seeking) && (fr->pts > ts_end))
133  break;
134  crc = av_adler32_update(0, (const uint8_t*)byte_buffer, number_of_written_bytes);
135  printf("%10"PRId64", 0x%08"PRIx32"\n", fr->pts, crc);
136  if (no_seeking) {
137  if (add_crc_to_array(crc, fr->pts) < 0)
138  return -1;
139  }
140  else {
141  if (compare_crc_in_array(crc, fr->pts) < 0)
142  return -1;
143  }
144  }
145  }
146  av_packet_unref(&pkt);
147  av_init_packet(&pkt);
148  } while ((!end_of_stream || got_frame) && (no_seeking || (fr->pts + fr->pkt_duration <= ts_end)));
149 
150  av_packet_unref(&pkt);
151  av_freep(&byte_buffer);
152 
153  return 0;
154 }
155 
156 static long int read_seek_range(const char *string_with_number)
157 {
158  long int number;
159  char *end_of_string = NULL;
160  number = strtol(string_with_number, &end_of_string, 10);
161  if ((strlen(string_with_number) != end_of_string - string_with_number) || (number < 0)) {
162  av_log(NULL, AV_LOG_ERROR, "Incorrect input ranges of seeking\n");
163  return -1;
164  }
165  else if ((number == LONG_MAX) || (number == LONG_MIN)) {
166  if (errno == ERANGE) {
167  av_log(NULL, AV_LOG_ERROR, "Incorrect input ranges of seeking\n");
168  return -1;
169  }
170  }
171  return number;
172 }
173 
174 static int seek_test(const char *input_filename, const char *start, const char *end)
175 {
176  AVCodec *codec = NULL;
178  AVCodecParameters *origin_par = NULL;
179  AVFrame *fr = NULL;
181  int video_stream;
182  int result;
183  int i, j;
184  long int start_ts, end_ts;
185 
186  size_of_array = 0;
187  number_of_elements = 0;
188  crc_array = NULL;
189  pts_array = NULL;
190 
191  result = avformat_open_input(&fmt_ctx, input_filename, NULL, NULL);
192  if (result < 0) {
193  av_log(NULL, AV_LOG_ERROR, "Can't open file\n");
194  return result;
195  }
196 
197  result = avformat_find_stream_info(fmt_ctx, NULL);
198  if (result < 0) {
199  av_log(NULL, AV_LOG_ERROR, "Can't get stream info\n");
200  goto end;
201  }
202 
203  start_ts = read_seek_range(start);
204  end_ts = read_seek_range(end);
205  if ((start_ts < 0) || (end_ts < 0)) {
206  result = -1;
207  goto end;
208  }
209 
210  //TODO: add ability to work with audio format
211  video_stream = av_find_best_stream(fmt_ctx, AVMEDIA_TYPE_VIDEO, -1, -1, NULL, 0);
212  if (video_stream < 0) {
213  av_log(NULL, AV_LOG_ERROR, "Can't find video stream in input file\n");
214  result = video_stream;
215  goto end;
216  }
217 
218  origin_par = fmt_ctx->streams[video_stream]->codecpar;
219 
220  codec = avcodec_find_decoder(origin_par->codec_id);
221  if (!codec) {
222  av_log(NULL, AV_LOG_ERROR, "Can't find decoder\n");
223  result = AVERROR_DECODER_NOT_FOUND;
224  goto end;
225  }
226 
227  ctx = avcodec_alloc_context3(codec);
228  if (!ctx) {
229  av_log(NULL, AV_LOG_ERROR, "Can't allocate decoder context\n");
230  result = AVERROR(ENOMEM);
231  goto end;
232  }
233 
234  result = avcodec_parameters_to_context(ctx, origin_par);
235  if (result) {
236  av_log(NULL, AV_LOG_ERROR, "Can't copy decoder context\n");
237  goto end;
238  }
239 
240  result = avcodec_open2(ctx, codec, NULL);
241  if (result < 0) {
242  av_log(ctx, AV_LOG_ERROR, "Can't open decoder\n");
243  goto end;
244  }
245 
246  fr = av_frame_alloc();
247  if (!fr) {
248  av_log(NULL, AV_LOG_ERROR, "Can't allocate frame\n");
249  result = AVERROR(ENOMEM);
250  goto end;
251  }
252 
253  result = compute_crc_of_packets(fmt_ctx, video_stream, ctx, fr, 0, 0, 1);
254  if (result != 0)
255  goto end;
256 
257  for (i = start_ts; i < end_ts; i += 100) {
258  for (j = i + 100; j < end_ts; j += 100) {
259  result = compute_crc_of_packets(fmt_ctx, video_stream, ctx, fr, i, j, 0);
260  if (result != 0)
261  break;
262  }
263  }
264 
265 end:
268  av_frame_free(&fr);
269  avcodec_close(ctx);
270  avformat_close_input(&fmt_ctx);
271  avcodec_free_context(&ctx);
272  return result;
273 }
274 
275 int main(int argc, char **argv)
276 {
277  if (argc < 4) {
278  av_log(NULL, AV_LOG_ERROR, "Incorrect input\n");
279  return 1;
280  }
281 
282  if (seek_test(argv[1], argv[2], argv[3]) != 0)
283  return 1;
284 
285  return 0;
286 }
#define NULL
Definition: coverity.c:32
This structure describes decoded (raw) audio or video data.
Definition: frame.h:218
#define av_realloc_f(p, o, n)
int av_image_copy_to_buffer(uint8_t *dst, int dst_size, const uint8_t *const src_data[4], const int src_linesize[4], enum AVPixelFormat pix_fmt, int width, int height, int align)
Copy image data from an image into a buffer.
Definition: imgutils.c:453
static int compare_crc_in_array(uint32_t crc, int64_t pts)
Definition: api-seek-test.c:56
misc image utilities
#define AVSEEK_FLAG_ANY
seek to any frame, even non-keyframes
Definition: avformat.h:2488
static AVFormatContext * fmt_ctx
enum AVCodecID codec_id
Specific type of the encoded data (the codec used).
Definition: avcodec.h:3884
int size
Definition: avcodec.h:1431
enum AVPixelFormat pix_fmt
Pixel format, see AV_PIX_FMT_xxx.
Definition: avcodec.h:1727
static AVStream * video_stream
int number_of_elements
Definition: api-seek-test.c:35
static AVPacket pkt
AVCodec.
Definition: avcodec.h:3408
static int add_crc_to_array(uint32_t crc, int64_t pts)
Definition: api-seek-test.c:37
This struct describes the properties of an encoded stream.
Definition: avcodec.h:3876
Format I/O context.
Definition: avformat.h:1342
uint8_t
#define av_malloc(s)
AVFrame * av_frame_alloc(void)
Allocate an AVFrame and set its fields to default values.
Definition: frame.c:189
static av_cold int end(AVCodecContext *avctx)
Definition: avrndec.c:90
unsigned long av_adler32_update(unsigned long adler, const uint8_t *buf, unsigned int len)
Calculate the Adler32 checksum of a buffer.
Definition: adler32.c:44
int64_t pts
Presentation timestamp in time_base units (time when frame should be shown to user).
Definition: frame.h:311
AVStream ** streams
A list of all streams in the file.
Definition: avformat.h:1410
int avcodec_parameters_to_context(AVCodecContext *codec, const AVCodecParameters *par)
Fill the codec context based on the values from the supplied codec parameters.
Definition: utils.c:2078
uint8_t * data
Definition: avcodec.h:1430
static int seek_test(const char *input_filename, const char *start, const char *end)
int size_of_array
Definition: api-seek-test.c:34
#define av_log(a,...)
int av_find_best_stream(AVFormatContext *ic, enum AVMediaType type, int wanted_stream_nb, int related_stream, AVCodec **decoder_ret, int flags)
Find the "best" stream in the file.
Definition: utils.c:4169
int avcodec_close(AVCodecContext *avctx)
Close a given AVCodecContext and free all the data associated with it (but not the AVCodecContext its...
Definition: utils.c:1061
int av_image_get_buffer_size(enum AVPixelFormat pix_fmt, int width, int height, int align)
Return the size in bytes of the amount of data required to store an image with the given parameters...
Definition: imgutils.c:431
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:176
#define AVERROR(e)
Definition: error.h:43
void av_frame_free(AVFrame **frame)
Free the frame and any dynamically allocated objects in it, e.g.
Definition: frame.c:202
static long int read_seek_range(const char *string_with_number)
uint32_t * crc_array
Definition: api-seek-test.c:33
static int compute_crc_of_packets(AVFormatContext *fmt_ctx, int video_stream, AVCodecContext *ctx, AVFrame *fr, uint64_t ts_start, uint64_t ts_end, int no_seeking)
Definition: api-seek-test.c:75
AVCodecContext * avcodec_alloc_context3(const AVCodec *codec)
Allocate an AVCodecContext and set its fields to default values.
Definition: options.c:156
int width
picture width / height.
Definition: avcodec.h:1690
AVFormatContext * ctx
Definition: movenc.c:48
static const char * input_filename
Definition: ffplay.c:311
void avcodec_flush_buffers(AVCodecContext *avctx)
Reset the internal decoder state / flush internal buffers.
Definition: decode.c:1946
Public header for Adler-32 hash function implementation.
Libavcodec external API header.
int64_t pkt_duration
duration of the corresponding packet, expressed in AVStream->time_base units, 0 if unknown...
Definition: frame.h:498
attribute_deprecated int avcodec_decode_video2(AVCodecContext *avctx, AVFrame *picture, int *got_picture_ptr, const AVPacket *avpkt)
Decode the video frame of size avpkt->size from avpkt->data into picture.
Definition: decode.c:826
void avcodec_free_context(AVCodecContext **avctx)
Free the codec context and everything associated with it and write NULL to the provided pointer...
Definition: options.c:171
int linesize[AV_NUM_DATA_POINTERS]
For video, size in bytes of each picture line.
Definition: frame.h:249
main external API structure.
Definition: avcodec.h:1518
AVCodec * avcodec_find_decoder(enum AVCodecID id)
Find a registered decoder with a matching codec ID.
Definition: allcodecs.c:862
void av_packet_unref(AVPacket *pkt)
Wipe the packet.
Definition: avpacket.c:592
int avcodec_open2(AVCodecContext *avctx, const AVCodec *codec, AVDictionary **options)
Initialize the AVCodecContext to use the given AVCodec.
Definition: utils.c:538
int av_read_frame(AVFormatContext *s, AVPacket *pkt)
Return the next frame of a stream.
Definition: utils.c:1767
static int64_t pts
uint8_t * data[AV_NUM_DATA_POINTERS]
pointer to the picture/channel planes.
Definition: frame.h:232
int av_seek_frame(AVFormatContext *s, int stream_index, int64_t timestamp, int flags)
Seek to the keyframe at timestamp.
Definition: utils.c:2505
Main libavformat public API header.
int64_t * pts_array
Seek test.
Definition: api-seek-test.c:32
int avformat_find_stream_info(AVFormatContext *ic, AVDictionary **options)
Read packets of a media file to get stream information.
Definition: utils.c:3550
void av_init_packet(AVPacket *pkt)
Initialize optional fields of a packet with default values.
Definition: avpacket.c:33
void avformat_close_input(AVFormatContext **s)
Close an opened input AVFormatContext.
Definition: utils.c:4413
#define AVERROR_DECODER_NOT_FOUND
Decoder not found.
Definition: error.h:52
int main(int argc, char **argv)
int avformat_open_input(AVFormatContext **ps, const char *url, AVInputFormat *fmt, AVDictionary **options)
Open an input stream and read the header.
Definition: utils.c:537
#define av_freep(p)
void INT64 start
Definition: avisynth_c.h:690
AVCodecParameters * codecpar
Codec parameters associated with this stream.
Definition: avformat.h:1020
int stream_index
Definition: avcodec.h:1432
This structure stores compressed data.
Definition: avcodec.h:1407
int64_t pts
Presentation timestamp in AVStream->time_base units; the time at which the decompressed packet will b...
Definition: avcodec.h:1423
#define AV_NOPTS_VALUE
Undefined timestamp value.
Definition: avutil.h:248