FFmpeg  4.4.5
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 
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));
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 
77  uint64_t ts_start, uint64_t ts_end, int no_seeking)
78 {
79  int number_of_written_bytes;
80  int result;
81  int byte_buffer_size;
82  uint8_t *byte_buffer;
83  uint32_t crc;
84 
85  byte_buffer_size = av_image_get_buffer_size(ctx->pix_fmt, ctx->width, ctx->height, 16);
86  byte_buffer = av_malloc(byte_buffer_size);
87  if (!byte_buffer) {
88  av_log(NULL, AV_LOG_ERROR, "Can't allocate buffer\n");
89  return AVERROR(ENOMEM);
90  }
91 
92  if (!no_seeking) {
93  result = av_seek_frame(fmt_ctx, video_stream, ts_start, AVSEEK_FLAG_ANY);
94  printf("Seeking to %"PRId64", computing crc for frames with pts < %"PRId64"\n", ts_start, ts_end);
95  if (result < 0) {
96  av_log(NULL, AV_LOG_ERROR, "Error in seeking\n");
97  return result;
98  }
100  }
101 
102  do {
103  result = av_read_frame(fmt_ctx, pkt);
104  if (result >= 0 && pkt->stream_index != video_stream) {
106  continue;
107  }
108 
109  if (result < 0)
110  result = avcodec_send_packet(ctx, NULL);
111  else {
112  if (pkt->pts == AV_NOPTS_VALUE) {
113  av_log(NULL, AV_LOG_ERROR, "Error: frames doesn't have pts values\n");
114  return -1;
115  }
116  result = avcodec_send_packet(ctx, pkt);
117  }
118 
120 
121  if (result < 0) {
122  av_log(NULL, AV_LOG_ERROR, "Error submitting a packet for decoding\n");
123  return result;
124  }
125 
126  while (result >= 0) {
127  result = avcodec_receive_frame(ctx, fr);
128  if (result == AVERROR_EOF)
129  goto finish;
130  else if (result == AVERROR(EAGAIN)) {
131  result = 0;
132  break;
133  } else if (result < 0) {
134  av_log(NULL, AV_LOG_ERROR, "Error decoding frame\n");
135  return result;
136  }
137 
138  number_of_written_bytes = av_image_copy_to_buffer(byte_buffer, byte_buffer_size,
139  (const uint8_t* const *)fr->data, (const int*) fr->linesize,
140  ctx->pix_fmt, ctx->width, ctx->height, 1);
141  if (number_of_written_bytes < 0) {
142  av_log(NULL, AV_LOG_ERROR, "Can't copy image to buffer\n");
143  return number_of_written_bytes;
144  }
145  if ((!no_seeking) && (fr->pts > ts_end))
146  break;
147  crc = av_adler32_update(0, (const uint8_t*)byte_buffer, number_of_written_bytes);
148  printf("%10"PRId64", 0x%08"PRIx32"\n", fr->pts, crc);
149  if (no_seeking) {
150  if (add_crc_to_array(crc, fr->pts) < 0)
151  return -1;
152  }
153  else {
154  if (compare_crc_in_array(crc, fr->pts) < 0)
155  return -1;
156  }
157  av_frame_unref(fr);
158  }
159  } while (result >= 0 && (no_seeking || (fr->pts + fr->pkt_duration <= ts_end)));
160 
161 finish:
162  av_freep(&byte_buffer);
163 
164  return 0;
165 }
166 
167 static long int read_seek_range(const char *string_with_number)
168 {
169  long int number;
170  char *end_of_string = NULL;
171  number = strtol(string_with_number, &end_of_string, 10);
172  if ((strlen(string_with_number) != end_of_string - string_with_number) || (number < 0)) {
173  av_log(NULL, AV_LOG_ERROR, "Incorrect input ranges of seeking\n");
174  return -1;
175  }
176  else if ((number == LONG_MAX) || (number == LONG_MIN)) {
177  if (errno == ERANGE) {
178  av_log(NULL, AV_LOG_ERROR, "Incorrect input ranges of seeking\n");
179  return -1;
180  }
181  }
182  return number;
183 }
184 
185 static int seek_test(const char *input_filename, const char *start, const char *end)
186 {
187  AVCodec *codec = NULL;
189  AVCodecParameters *origin_par = NULL;
190  AVPacket *pkt = NULL;
191  AVFrame *fr = NULL;
193  int video_stream;
194  int result;
195  int i, j;
196  long int start_ts, end_ts;
197 
198  size_of_array = 0;
199  number_of_elements = 0;
200  crc_array = NULL;
201  pts_array = NULL;
202 
204  if (result < 0) {
205  av_log(NULL, AV_LOG_ERROR, "Can't open file\n");
206  return result;
207  }
208 
210  if (result < 0) {
211  av_log(NULL, AV_LOG_ERROR, "Can't get stream info\n");
212  goto end;
213  }
214 
215  start_ts = read_seek_range(start);
216  end_ts = read_seek_range(end);
217  if ((start_ts < 0) || (end_ts < 0)) {
218  result = -1;
219  goto end;
220  }
221 
222  //TODO: add ability to work with audio format
224  if (video_stream < 0) {
225  av_log(NULL, AV_LOG_ERROR, "Can't find video stream in input file\n");
226  result = video_stream;
227  goto end;
228  }
229 
230  origin_par = fmt_ctx->streams[video_stream]->codecpar;
231 
232  codec = avcodec_find_decoder(origin_par->codec_id);
233  if (!codec) {
234  av_log(NULL, AV_LOG_ERROR, "Can't find decoder\n");
235  result = AVERROR_DECODER_NOT_FOUND;
236  goto end;
237  }
238 
239  ctx = avcodec_alloc_context3(codec);
240  if (!ctx) {
241  av_log(NULL, AV_LOG_ERROR, "Can't allocate decoder context\n");
242  result = AVERROR(ENOMEM);
243  goto end;
244  }
245 
246  result = avcodec_parameters_to_context(ctx, origin_par);
247  if (result) {
248  av_log(NULL, AV_LOG_ERROR, "Can't copy decoder context\n");
249  goto end;
250  }
251 
252  result = avcodec_open2(ctx, codec, NULL);
253  if (result < 0) {
254  av_log(ctx, AV_LOG_ERROR, "Can't open decoder\n");
255  goto end;
256  }
257 
258  fr = av_frame_alloc();
259  if (!fr) {
260  av_log(NULL, AV_LOG_ERROR, "Can't allocate frame\n");
261  result = AVERROR(ENOMEM);
262  goto end;
263  }
264 
265  pkt = av_packet_alloc();
266  if (!pkt) {
267  av_log(NULL, AV_LOG_ERROR, "Cannot allocate packet\n");
268  result = AVERROR(ENOMEM);
269  goto end;
270  }
271 
272  result = compute_crc_of_packets(fmt_ctx, video_stream, ctx, pkt, fr, 0, 0, 1);
273  if (result != 0)
274  goto end;
275 
276  for (i = start_ts; i < end_ts; i += 100) {
277  for (j = i + 100; j < end_ts; j += 100) {
278  result = compute_crc_of_packets(fmt_ctx, video_stream, ctx, pkt, fr, i, j, 0);
279  if (result != 0)
280  break;
281  }
282  }
283 
284 end:
288  av_frame_free(&fr);
291  return result;
292 }
293 
294 int main(int argc, char **argv)
295 {
296  if (argc < 4) {
297  av_log(NULL, AV_LOG_ERROR, "Incorrect input\n");
298  return 1;
299  }
300 
301  if (seek_test(argv[1], argv[2], argv[3]) != 0)
302  return 1;
303 
304  return 0;
305 }
Public header for Adler-32 hash function implementation.
static int add_crc_to_array(uint32_t crc, int64_t pts)
Definition: api-seek-test.c:37
static int compute_crc_of_packets(AVFormatContext *fmt_ctx, int video_stream, AVCodecContext *ctx, AVPacket *pkt, AVFrame *fr, uint64_t ts_start, uint64_t ts_end, int no_seeking)
Definition: api-seek-test.c:75
int main(int argc, char **argv)
static long int read_seek_range(const char *string_with_number)
int size_of_array
Definition: api-seek-test.c:34
static int seek_test(const char *input_filename, const char *start, const char *end)
uint32_t * crc_array
Definition: api-seek-test.c:33
int number_of_elements
Definition: api-seek-test.c:35
int64_t * pts_array
Seek test.
Definition: api-seek-test.c:32
static int compare_crc_in_array(uint32_t crc, int64_t pts)
Definition: api-seek-test.c:56
uint8_t
Libavcodec external API header.
Main libavformat public API header.
#define AVSEEK_FLAG_ANY
seek to any frame, even non-keyframes
Definition: avformat.h:2417
#define NULL
Definition: coverity.c:32
long long int64_t
Definition: coverity.c:34
static AVFormatContext * fmt_ctx
static AVStream * video_stream
static const char * input_filename
Definition: ffplay.c:312
int attribute_align_arg avcodec_open2(AVCodecContext *avctx, const AVCodec *codec, AVDictionary **options)
Initialize the AVCodecContext to use the given AVCodec.
Definition: avcodec.c:144
AVCodec * avcodec_find_decoder(enum AVCodecID id)
Find a registered decoder with a matching codec ID.
Definition: allcodecs.c:946
int avcodec_parameters_to_context(AVCodecContext *codec, const AVCodecParameters *par)
Fill the codec context based on the values from the supplied codec parameters.
Definition: codec_par.c:147
AVCodecContext * avcodec_alloc_context3(const AVCodec *codec)
Allocate an AVCodecContext and set its fields to default values.
Definition: options.c:173
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:188
int avcodec_receive_frame(AVCodecContext *avctx, AVFrame *frame)
Return decoded output data from a decoder.
Definition: decode.c:652
int avcodec_send_packet(AVCodecContext *avctx, const AVPacket *avpkt)
Supply raw packet data as input to a decoder.
Definition: decode.c:589
void avcodec_flush_buffers(AVCodecContext *avctx)
Reset the internal codec state / flush internal buffers.
Definition: avcodec.c:491
void av_packet_free(AVPacket **pkt)
Free the packet, if the packet is reference counted, it will be unreferenced first.
Definition: avpacket.c:75
void av_packet_unref(AVPacket *pkt)
Wipe the packet.
Definition: avpacket.c:634
AVPacket * av_packet_alloc(void)
Allocate an AVPacket and set its fields to default values.
Definition: avpacket.c:64
int av_read_frame(AVFormatContext *s, AVPacket *pkt)
Return the next frame of a stream.
Definition: utils.c:1741
int av_seek_frame(AVFormatContext *s, int stream_index, int64_t timestamp, int flags)
Seek to the keyframe at timestamp.
Definition: utils.c:2489
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:4230
int avformat_find_stream_info(AVFormatContext *ic, AVDictionary **options)
Read packets of a media file to get stream information.
Definition: utils.c:3602
void avformat_close_input(AVFormatContext **s)
Close an opened input AVFormatContext.
Definition: utils.c:4481
int avformat_open_input(AVFormatContext **ps, const char *url, ff_const59 AVInputFormat *fmt, AVDictionary **options)
Open an input stream and read the header.
Definition: utils.c:512
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:45
#define AVERROR_DECODER_NOT_FOUND
Decoder not found.
Definition: error.h:52
#define AVERROR_EOF
End of file.
Definition: error.h:55
#define AVERROR(e)
Definition: error.h:43
void av_frame_unref(AVFrame *frame)
Unreference all the buffers referenced by frame and reset the frame fields.
Definition: frame.c:553
void av_frame_free(AVFrame **frame)
Free the frame and any dynamically allocated objects in it, e.g.
Definition: frame.c:203
AVFrame * av_frame_alloc(void)
Allocate an AVFrame and set its fields to default values.
Definition: frame.c:190
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:194
@ AVMEDIA_TYPE_VIDEO
Definition: avutil.h:201
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:466
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:505
#define AV_NOPTS_VALUE
Undefined timestamp value.
Definition: avutil.h:248
misc image utilities
int i
Definition: input.c:407
main external API structure.
Definition: avcodec.h:536
This struct describes the properties of an encoded stream.
Definition: codec_par.h:52
enum AVCodecID codec_id
Specific type of the encoded data (the codec used).
Definition: codec_par.h:60
AVCodec.
Definition: codec.h:197
Format I/O context.
Definition: avformat.h:1232
AVStream ** streams
A list of all streams in the file.
Definition: avformat.h:1300
This structure describes decoded (raw) audio or video data.
Definition: frame.h:318
int64_t pts
Presentation timestamp in time_base units (time when frame should be shown to user).
Definition: frame.h:411
uint8_t * data[AV_NUM_DATA_POINTERS]
pointer to the picture/channel planes.
Definition: frame.h:332
int64_t pkt_duration
duration of the corresponding packet, expressed in AVStream->time_base units, 0 if unknown.
Definition: frame.h:597
int linesize[AV_NUM_DATA_POINTERS]
For video, size in bytes of each picture line.
Definition: frame.h:349
This structure stores compressed data.
Definition: packet.h:346
int stream_index
Definition: packet.h:371
int64_t pts
Presentation timestamp in AVStream->time_base units; the time at which the decompressed packet will b...
Definition: packet.h:362
AVCodecParameters * codecpar
Codec parameters associated with this stream.
Definition: avformat.h:1038
#define av_realloc_f(p, o, n)
#define av_freep(p)
#define av_malloc(s)
#define av_log(a,...)
AVPacket * pkt
Definition: movenc.c:59
AVFormatContext * ctx
Definition: movenc.c:48
static void finish(void)
Definition: movenc.c:342
static int64_t pts