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->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->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 + av_frame_get_pkt_duration(fr) <= 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;
189 
190  result = avformat_open_input(&fmt_ctx, input_filename, NULL, NULL);
191  if (result < 0) {
192  av_log(NULL, AV_LOG_ERROR, "Can't open file\n");
193  return result;
194  }
195 
196  result = avformat_find_stream_info(fmt_ctx, NULL);
197  if (result < 0) {
198  av_log(NULL, AV_LOG_ERROR, "Can't get stream info\n");
199  return result;
200  }
201 
202  start_ts = read_seek_range(start);
203  end_ts = read_seek_range(end);
204  if ((start_ts < 0) || (end_ts < 0))
205  return -1;
206 
207  //TODO: add ability to work with audio format
208  video_stream = av_find_best_stream(fmt_ctx, AVMEDIA_TYPE_VIDEO, -1, -1, NULL, 0);
209  if (video_stream < 0) {
210  av_log(NULL, AV_LOG_ERROR, "Can't find video stream in input file\n");
211  return -1;
212  }
213 
214  origin_par = fmt_ctx->streams[video_stream]->codecpar;
215 
216  codec = avcodec_find_decoder(origin_par->codec_id);
217  if (!codec) {
218  av_log(NULL, AV_LOG_ERROR, "Can't find decoder\n");
219  return -1;
220  }
221 
222  ctx = avcodec_alloc_context3(codec);
223  if (!ctx) {
224  av_log(NULL, AV_LOG_ERROR, "Can't allocate decoder context\n");
225  return AVERROR(ENOMEM);
226  }
227 
228  result = avcodec_parameters_to_context(ctx, origin_par);
229  if (result) {
230  av_log(NULL, AV_LOG_ERROR, "Can't copy decoder context\n");
231  return result;
232  }
233 
234  result = avcodec_open2(ctx, codec, NULL);
235  if (result < 0) {
236  av_log(ctx, AV_LOG_ERROR, "Can't open decoder\n");
237  return result;
238  }
239 
240  fr = av_frame_alloc();
241  if (!fr) {
242  av_log(NULL, AV_LOG_ERROR, "Can't allocate frame\n");
243  return AVERROR(ENOMEM);
244  }
245 
246  result = compute_crc_of_packets(fmt_ctx, video_stream, ctx, fr, i, j, 1);
247  if (result != 0)
248  return -1;
249 
250  for (i = start_ts; i < end_ts; i += 100) {
251  for (j = i + 100; j < end_ts; j += 100)
252  result = compute_crc_of_packets(fmt_ctx, video_stream, ctx, fr, i, j, 0);
253  if (result != 0)
254  return -1;
255  }
256 
259  av_frame_free(&fr);
260  avcodec_close(ctx);
261  avformat_close_input(&fmt_ctx);
262  avcodec_free_context(&ctx);
263  return 0;
264 }
265 
266 int main(int argc, char **argv)
267 {
268  if (argc < 4) {
269  av_log(NULL, AV_LOG_ERROR, "Incorrect input\n");
270  return 1;
271  }
272 
273  av_register_all();
274 
275  if (seek_test(argv[1], argv[2], argv[3]) != 0)
276  return 1;
277 
278  return 0;
279 }
#define NULL
Definition: coverity.c:32
int64_t av_frame_get_pkt_duration(const AVFrame *frame)
This structure describes decoded (raw) audio or video data.
Definition: frame.h:184
void * av_realloc(void *ptr, size_t size)
Allocate, reallocate, or free a block of memory.
Definition: mem.c:145
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:383
misc image utilities
#define AVSEEK_FLAG_ANY
seek to any frame, even non-keyframes
Definition: avformat.h:2388
static AVFormatContext * fmt_ctx
enum AVCodecID codec_id
Specific type of the encoded data (the codec used).
Definition: avcodec.h:3980
int size
Definition: avcodec.h:1602
enum AVPixelFormat pix_fmt
Pixel format, see AV_PIX_FMT_xxx.
Definition: avcodec.h:1904
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:3600
This struct describes the properties of an encoded stream.
Definition: avcodec.h:3972
Format I/O context.
Definition: avformat.h:1338
uint8_t
#define av_malloc(s)
AVFrame * av_frame_alloc(void)
Allocate an AVFrame and set its fields to default values.
Definition: frame.c:145
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:268
AVStream ** streams
A list of all streams in the file.
Definition: avformat.h:1406
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:4223
uint8_t * data
Definition: avcodec.h:1601
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:3934
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:3023
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:361
#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:158
static long int read_seek_range(const char *string_with_number)
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:1863
AVFormatContext * ctx
Definition: movenc.c:48
static const char * input_filename
Definition: ffplay.c:312
void avcodec_flush_buffers(AVCodecContext *avctx)
Reset the internal decoder state / flush internal buffers.
Definition: utils.c:3435
Public header for Adler-32 hash function implementation.
static int add_crc_to_array(int64_t crc, int64_t pts)
Definition: api-seek-test.c:37
Libavcodec external API header.
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: utils.c:2213
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:215
main external API structure.
Definition: avcodec.h:1676
AVCodec * avcodec_find_decoder(enum AVCodecID id)
Find a registered decoder with a matching codec ID.
Definition: utils.c:3127
void av_packet_unref(AVPacket *pkt)
Wipe the packet.
Definition: avpacket.c:567
int avcodec_open2(AVCodecContext *avctx, const AVCodec *codec, AVDictionary **options)
Initialize the AVCodecContext to use the given AVCodec.
Definition: utils.c:1241
int av_read_frame(AVFormatContext *s, AVPacket *pkt)
Return the next frame of a stream.
Definition: utils.c:1676
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:198
int av_seek_frame(AVFormatContext *s, int stream_index, int64_t timestamp, int flags)
Seek to the keyframe at timestamp.
Definition: utils.c:2412
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:3336
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:4165
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:503
#define av_freep(p)
void INT64 start
Definition: avisynth_c.h:690
AVCodecParameters * codecpar
Definition: avformat.h:1241
int stream_index
Definition: avcodec.h:1603
This structure stores compressed data.
Definition: avcodec.h:1578
void av_register_all(void)
Initialize libavformat and register all the muxers, demuxers and protocols.
Definition: allformats.c:44
int64_t pts
Presentation timestamp in AVStream->time_base units; the time at which the decompressed packet will b...
Definition: avcodec.h:1594
#define AV_NOPTS_VALUE
Undefined timestamp value.
Definition: avutil.h:242