tests: do not use AVFrame accessor
[ffmpeg.git] / tests / api / api-h264-test.c
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  * H264 codec 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 static int video_decode_example(const char *input_filename)
33 {
34     AVCodec *codec = NULL;
35     AVCodecContext *ctx= NULL;
36     AVCodecParameters *origin_par = NULL;
37     AVFrame *fr = NULL;
38     uint8_t *byte_buffer = NULL;
39     AVPacket pkt;
40     AVFormatContext *fmt_ctx = NULL;
41     int number_of_written_bytes;
42     int video_stream;
43     int got_frame = 0;
44     int byte_buffer_size;
45     int i = 0;
46     int result;
47     int end_of_stream = 0;
48
49     result = avformat_open_input(&fmt_ctx, input_filename, NULL, NULL);
50     if (result < 0) {
51         av_log(NULL, AV_LOG_ERROR, "Can't open file\n");
52         return result;
53     }
54
55     result = avformat_find_stream_info(fmt_ctx, NULL);
56     if (result < 0) {
57         av_log(NULL, AV_LOG_ERROR, "Can't get stream info\n");
58         return result;
59     }
60
61     video_stream = av_find_best_stream(fmt_ctx, AVMEDIA_TYPE_VIDEO, -1, -1, NULL, 0);
62     if (video_stream < 0) {
63       av_log(NULL, AV_LOG_ERROR, "Can't find video stream in input file\n");
64       return -1;
65     }
66
67     origin_par = fmt_ctx->streams[video_stream]->codecpar;
68
69     codec = avcodec_find_decoder(origin_par->codec_id);
70     if (!codec) {
71         av_log(NULL, AV_LOG_ERROR, "Can't find decoder\n");
72         return -1;
73     }
74
75     ctx = avcodec_alloc_context3(codec);
76     if (!ctx) {
77         av_log(NULL, AV_LOG_ERROR, "Can't allocate decoder context\n");
78         return AVERROR(ENOMEM);
79     }
80
81     result = avcodec_parameters_to_context(ctx, origin_par);
82     if (result) {
83         av_log(NULL, AV_LOG_ERROR, "Can't copy decoder context\n");
84         return result;
85     }
86
87     result = avcodec_open2(ctx, codec, NULL);
88     if (result < 0) {
89         av_log(ctx, AV_LOG_ERROR, "Can't open decoder\n");
90         return result;
91     }
92
93     fr = av_frame_alloc();
94     if (!fr) {
95         av_log(NULL, AV_LOG_ERROR, "Can't allocate frame\n");
96         return AVERROR(ENOMEM);
97     }
98
99     byte_buffer_size = av_image_get_buffer_size(ctx->pix_fmt, ctx->width, ctx->height, 16);
100     byte_buffer = av_malloc(byte_buffer_size);
101     if (!byte_buffer) {
102         av_log(NULL, AV_LOG_ERROR, "Can't allocate buffer\n");
103         return AVERROR(ENOMEM);
104     }
105
106     printf("#tb %d: %d/%d\n", video_stream, fmt_ctx->streams[video_stream]->time_base.num, fmt_ctx->streams[video_stream]->time_base.den);
107     i = 0;
108     av_init_packet(&pkt);
109     do {
110         if (!end_of_stream)
111             if (av_read_frame(fmt_ctx, &pkt) < 0)
112                 end_of_stream = 1;
113         if (end_of_stream) {
114             pkt.data = NULL;
115             pkt.size = 0;
116         }
117         if (pkt.stream_index == video_stream || end_of_stream) {
118             got_frame = 0;
119             if (pkt.pts == AV_NOPTS_VALUE)
120                 pkt.pts = pkt.dts = i;
121             result = avcodec_decode_video2(ctx, fr, &got_frame, &pkt);
122             if (result < 0) {
123                 av_log(NULL, AV_LOG_ERROR, "Error decoding frame\n");
124                 return result;
125             }
126             if (got_frame) {
127                 number_of_written_bytes = av_image_copy_to_buffer(byte_buffer, byte_buffer_size,
128                                         (const uint8_t* const *)fr->data, (const int*) fr->linesize,
129                                         ctx->pix_fmt, ctx->width, ctx->height, 1);
130                 if (number_of_written_bytes < 0) {
131                     av_log(NULL, AV_LOG_ERROR, "Can't copy image to buffer\n");
132                     return number_of_written_bytes;
133                 }
134                 printf("%d, %10"PRId64", %10"PRId64", %8"PRId64", %8d, 0x%08lx\n", video_stream,
135                         fr->pts, fr->pkt_dts, fr->pkt_duration,
136                         number_of_written_bytes, av_adler32_update(0, (const uint8_t*)byte_buffer, number_of_written_bytes));
137             }
138             av_packet_unref(&pkt);
139             av_init_packet(&pkt);
140         }
141         i++;
142     } while (!end_of_stream || got_frame);
143
144     av_packet_unref(&pkt);
145     av_frame_free(&fr);
146     avcodec_close(ctx);
147     avformat_close_input(&fmt_ctx);
148     avcodec_free_context(&ctx);
149     av_freep(&byte_buffer);
150     return 0;
151 }
152
153 int main(int argc, char **argv)
154 {
155     if (argc < 2)
156     {
157         av_log(NULL, AV_LOG_ERROR, "Incorrect input\n");
158         return 1;
159     }
160
161     av_register_all();
162
163     if (video_decode_example(argv[1]) != 0)
164         return 1;
165
166     return 0;
167 }