97062d5dfca304ddc091a133f03780ff5a741e5e
[libav.git] / libavformat / seek-test.c
1 /*
2 * Copyright (c) 2003 Fabrice Bellard
3 * Copyright (c) 2007 Michael Niedermayer
4 *
5 * This file is part of Libav.
6 *
7 * Libav is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
11 *
12 * Libav is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
16 *
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with Libav; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20 */
21
22 #include <stdint.h>
23 #include <stdlib.h>
24 #include <stdio.h>
25 #include <string.h>
26
27 #include "libavutil/common.h"
28 #include "libavutil/mathematics.h"
29 #include "libavformat/avformat.h"
30
31 static char buffer[20];
32
33 static const char *ret_str(int v)
34 {
35 switch (v) {
36 case AVERROR_EOF: return "-EOF";
37 case AVERROR(EIO): return "-EIO";
38 case AVERROR(ENOMEM): return "-ENOMEM";
39 case AVERROR(EINVAL): return "-EINVAL";
40 default:
41 snprintf(buffer, sizeof(buffer), "%2d", v);
42 return buffer;
43 }
44 }
45
46 static void ts_str(char buffer[60], int64_t ts, AVRational base)
47 {
48 double tsval;
49 if (ts == AV_NOPTS_VALUE) {
50 strcpy(buffer, " NOPTS ");
51 return;
52 }
53 tsval = ts * av_q2d(base);
54 snprintf(buffer, 60, "%9f", tsval);
55 }
56
57 int main(int argc, char **argv)
58 {
59 const char *filename;
60 AVFormatContext *ic = NULL;
61 int i, ret, stream_id;
62 int64_t timestamp;
63 AVDictionary *format_opts = NULL;
64
65 av_dict_set(&format_opts, "channels", "1", 0);
66 av_dict_set(&format_opts, "sample_rate", "22050", 0);
67
68 /* initialize libavcodec, and register all codecs and formats */
69 av_register_all();
70
71 if (argc != 2) {
72 printf("usage: %s input_file\n"
73 "\n", argv[0]);
74 return 1;
75 }
76
77 filename = argv[1];
78
79 ret = avformat_open_input(&ic, filename, NULL, &format_opts);
80 av_dict_free(&format_opts);
81 if (ret < 0) {
82 fprintf(stderr, "cannot open %s\n", filename);
83 return 1;
84 }
85
86 ret = avformat_find_stream_info(ic, NULL);
87 if (ret < 0) {
88 fprintf(stderr, "%s: could not find codec parameters\n", filename);
89 return 1;
90 }
91
92 for(i=0; ; i++){
93 AVPacket pkt = { 0 };
94 AVStream *av_uninit(st);
95 char ts_buf[60];
96
97 if(ret>=0){
98 ret= av_read_frame(ic, &pkt);
99 if(ret>=0){
100 char dts_buf[60];
101 st= ic->streams[pkt.stream_index];
102 ts_str(dts_buf, pkt.dts, st->time_base);
103 ts_str(ts_buf, pkt.pts, st->time_base);
104 printf("ret:%-10s st:%2d flags:%d dts:%s pts:%s pos:%7" PRId64 " size:%6d", ret_str(ret), pkt.stream_index, pkt.flags, dts_buf, ts_buf, pkt.pos, pkt.size);
105 av_packet_unref(&pkt);
106 } else
107 printf("ret:%s", ret_str(ret)); // necessary to avoid trailing whitespace
108 printf("\n");
109 }
110
111 if(i>25) break;
112
113 stream_id= (i>>1)%(ic->nb_streams+1) - 1;
114 timestamp= (i*19362894167LL) % (4*AV_TIME_BASE) - AV_TIME_BASE;
115 if(stream_id>=0){
116 st= ic->streams[stream_id];
117 timestamp= av_rescale_q(timestamp, AV_TIME_BASE_Q, st->time_base);
118 }
119 //FIXME fully test the new seek API
120 if(i&1) ret = avformat_seek_file(ic, stream_id, INT64_MIN, timestamp, timestamp, 0);
121 else ret = avformat_seek_file(ic, stream_id, timestamp, timestamp, INT64_MAX, 0);
122 ts_str(ts_buf, timestamp, stream_id < 0 ? AV_TIME_BASE_Q : st->time_base);
123 printf("ret:%-10s st:%2d flags:%d ts:%s\n", ret_str(ret), stream_id, i&1, ts_buf);
124 }
125
126 avformat_close_input(&ic);
127
128 return 0;
129 }