FFmpeg
 All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Macros Groups Pages
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 int64_t *crc_array;
36 
37 static int add_crc_to_array(int64_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(crc_array, size_of_array * sizeof(int64_t));
44  pts_array = av_realloc(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(int64_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%08lx %"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  int64_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 ((fr->pkt_pts > ts_end) && (!no_seeking))
133  break;
134  crc = av_adler32_update(0, (const uint8_t*)byte_buffer, number_of_written_bytes);
135  printf("%10"PRId64", 0x%08lx\n", fr->pkt_pts, crc);
136  if (no_seeking) {
137  if (add_crc_to_array(crc, fr->pkt_pts) < 0)
138  return -1;
139  }
140  else {
141  if (compare_crc_in_array(crc, fr->pkt_pts) < 0)
142  return -1;
143  }
144  }
145  }
146  av_free_packet(&pkt);
147  av_init_packet(&pkt);
148  } while ((!end_of_stream || got_frame) && (no_seeking || (fr->pkt_pts + av_frame_get_pkt_duration(fr) <= ts_end)));
149 
150  av_free_packet(&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;
177  AVCodecContext *origin_ctx = NULL, *ctx= NULL;
178  AVFrame *fr = NULL;
180  int video_stream;
181  int result;
182  int i, j;
183  long int start_ts, end_ts;
184 
185  size_of_array = 0;
186  number_of_elements = 0;
188 
189  result = avformat_open_input(&fmt_ctx, input_filename, NULL, NULL);
190  if (result < 0) {
191  av_log(NULL, AV_LOG_ERROR, "Can't open file\n");
192  return result;
193  }
194 
195  result = avformat_find_stream_info(fmt_ctx, NULL);
196  if (result < 0) {
197  av_log(NULL, AV_LOG_ERROR, "Can't get stream info\n");
198  return result;
199  }
200 
201  start_ts = read_seek_range(start);
202  end_ts = read_seek_range(end);
203  if ((start_ts < 0) || (end_ts < 0))
204  return -1;
205 
206  //TODO: add ability to work with audio format
207  video_stream = av_find_best_stream(fmt_ctx, AVMEDIA_TYPE_VIDEO, -1, -1, NULL, 0);
208  if (video_stream < 0) {
209  av_log(NULL, AV_LOG_ERROR, "Can't find video stream in input file\n");
210  return -1;
211  }
212 
213  origin_ctx = fmt_ctx->streams[video_stream]->codec;
214 
215  codec = avcodec_find_decoder(origin_ctx->codec_id);
216  if (!codec) {
217  av_log(NULL, AV_LOG_ERROR, "Can't find decoder\n");
218  return -1;
219  }
220 
221  ctx = avcodec_alloc_context3(codec);
222  if (!ctx) {
223  av_log(NULL, AV_LOG_ERROR, "Can't allocate decoder context\n");
224  return AVERROR(ENOMEM);
225  }
226 
227  result = avcodec_copy_context(ctx, origin_ctx);
228  if (result) {
229  av_log(NULL, AV_LOG_ERROR, "Can't copy decoder context\n");
230  return result;
231  }
232 
233  result = avcodec_open2(ctx, codec, NULL);
234  if (result < 0) {
235  av_log(ctx, AV_LOG_ERROR, "Can't open decoder\n");
236  return result;
237  }
238 
239  fr = av_frame_alloc();
240  if (!fr) {
241  av_log(NULL, AV_LOG_ERROR, "Can't allocate frame\n");
242  return AVERROR(ENOMEM);
243  }
244 
245  result = compute_crc_of_packets(fmt_ctx, video_stream, ctx, fr, i, j, 1);
246  if (result != 0)
247  return -1;
248 
249  for (i = start_ts; i < end_ts; i += 100) {
250  for (j = i + 100; j < end_ts; j += 100)
251  result = compute_crc_of_packets(fmt_ctx, video_stream, ctx, fr, i, j, 0);
252  if (result != 0)
253  return -1;
254  }
255 
258  av_frame_free(&fr);
259  avcodec_close(ctx);
260  avformat_close_input(&fmt_ctx);
261  avcodec_free_context(&ctx);
262  return 0;
263 }
264 
265 int main(int argc, char **argv)
266 {
267  if (argc < 4) {
268  av_log(NULL, AV_LOG_ERROR, "Incorrect input\n");
269  return 1;
270  }
271 
272  av_register_all();
273 
274  if (seek_test(argv[1], argv[2], argv[3]) != 0)
275  return 1;
276 
277  return 0;
278 }
#define NULL
Definition: coverity.c:32
int64_t av_frame_get_pkt_duration(const AVFrame *frame)
void av_free_packet(AVPacket *pkt)
Free a packet.
Definition: avpacket.c:280
This structure describes decoded (raw) audio or video data.
Definition: frame.h:171
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:369
misc image utilities
int avformat_open_input(AVFormatContext **ps, const char *filename, AVInputFormat *fmt, AVDictionary **options)
Open an input stream and read the header.
Definition: utils.c:405
#define AVSEEK_FLAG_ANY
seek to any frame, even non-keyframes
Definition: avformat.h:2279
static AVFormatContext * fmt_ctx
int size
Definition: avcodec.h:1424
enum AVPixelFormat pix_fmt
Pixel format, see AV_PIX_FMT_xxx.
Definition: avcodec.h:1722
static AVStream * video_stream
int64_t * crc_array
Definition: api-seek-test.c:33
int number_of_elements
Definition: api-seek-test.c:35
static AVPacket pkt
AVCodec.
Definition: avcodec.h:3472
int avcodec_copy_context(AVCodecContext *dest, const AVCodecContext *src)
Copy the settings of the source AVCodecContext into the destination AVCodecContext.
Definition: options.c:182
Format I/O context.
Definition: avformat.h:1273
uint8_t
#define av_malloc(s)
AVFrame * av_frame_alloc(void)
Allocate an AVFrame and set its fields to default values.
Definition: frame.c:135
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
AVStream ** streams
A list of all streams in the file.
Definition: avformat.h:1341
uint8_t * data
Definition: avcodec.h:1423
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:3572
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:2902
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:347
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:176
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: utils.c:2408
#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:148
static long int read_seek_range(const char *string_with_number)
Libavcodec external API header.
AVCodecContext * codec
Codec context associated with this stream.
Definition: avformat.h:861
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:149
int width
picture width / height.
Definition: avcodec.h:1681
static const char * input_filename
Definition: ffplay.c:309
void avcodec_flush_buffers(AVCodecContext *avctx)
Reset the internal decoder state / flush internal buffers.
Definition: utils.c:3303
Public header for libavutil Adler32 hasher.
static int add_crc_to_array(int64_t crc, int64_t pts)
Definition: api-seek-test.c:37
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:164
enum AVCodecID codec_id
Definition: avcodec.h:1519
int linesize[AV_NUM_DATA_POINTERS]
For video, size in bytes of each picture line.
Definition: frame.h:199
main external API structure.
Definition: avcodec.h:1502
AVCodec * avcodec_find_decoder(enum AVCodecID id)
Find a registered decoder with a matching codec ID.
Definition: utils.c:3016
int avcodec_open2(AVCodecContext *avctx, const AVCodec *codec, AVDictionary **options)
Initialize the AVCodecContext to use the given AVCodec.
Definition: utils.c:1349
int av_read_frame(AVFormatContext *s, AVPacket *pkt)
Return the next frame of a stream.
Definition: utils.c:1478
int64_t pkt_pts
PTS copied from the AVPacket that was decoded to produce this frame.
Definition: frame.h:262
static int64_t pts
Global timestamp for the audio frames.
static int compare_crc_in_array(int64_t crc, int64_t pts)
Definition: api-seek-test.c:56
uint8_t * data[AV_NUM_DATA_POINTERS]
pointer to the picture/channel planes.
Definition: frame.h:182
int av_seek_frame(AVFormatContext *s, int stream_index, int64_t timestamp, int flags)
Seek to the keyframe at timestamp.
Definition: utils.c:2199
Main libavformat public API header.
int64_t * pts_array
Seek test.
Definition: api-seek-test.c:32
void * av_realloc(void *ptr, size_t size)
Allocate or reallocate a block of memory.
Definition: mem.c:145
int avformat_find_stream_info(AVFormatContext *ic, AVDictionary **options)
Read packets of a media file to get stream information.
Definition: utils.c:3084
void av_init_packet(AVPacket *pkt)
Initialize optional fields of a packet with default values.
Definition: avpacket.c:49
void avformat_close_input(AVFormatContext **s)
Close an opened input AVFormatContext.
Definition: utils.c:3721
int main(int argc, char **argv)
#define av_freep(p)
void INT64 start
Definition: avisynth_c.h:553
int stream_index
Definition: avcodec.h:1425
This structure stores compressed data.
Definition: avcodec.h:1400
void av_register_all(void)
Initialize libavformat and register all the muxers, demuxers and protocols.
Definition: allformats.c:51
int64_t pts
Presentation timestamp in AVStream->time_base units; the time at which the decompressed packet will b...
Definition: avcodec.h:1416
#define AV_NOPTS_VALUE
Undefined timestamp value.
Definition: avutil.h:240