Line | Branch | Exec | Source |
---|---|---|---|
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 "libavutil/mem.h" | ||
29 | #include "libavcodec/avcodec.h" | ||
30 | #include "libavformat/avformat.h" | ||
31 | #include "libavutil/imgutils.h" | ||
32 | #include "libavutil/timestamp.h" | ||
33 | |||
34 | 1 | static int video_decode_example(const char *input_filename) | |
35 | { | ||
36 | 1 | const AVCodec *codec = NULL; | |
37 | 1 | AVCodecContext *ctx= NULL; | |
38 | 1 | AVCodecParameters *origin_par = NULL; | |
39 | 1 | AVFrame *fr = NULL; | |
40 | 1 | uint8_t *byte_buffer = NULL; | |
41 | AVPacket *pkt; | ||
42 | 1 | AVFormatContext *fmt_ctx = NULL; | |
43 | int number_of_written_bytes; | ||
44 | int video_stream; | ||
45 | int byte_buffer_size; | ||
46 | 1 | int i = 0; | |
47 | int result; | ||
48 | |||
49 | 1 | result = avformat_open_input(&fmt_ctx, input_filename, NULL, NULL); | |
50 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
|
1 | if (result < 0) { |
51 | ✗ | av_log(NULL, AV_LOG_ERROR, "Can't open file\n"); | |
52 | ✗ | return result; | |
53 | } | ||
54 | |||
55 | 1 | result = avformat_find_stream_info(fmt_ctx, NULL); | |
56 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
|
1 | if (result < 0) { |
57 | ✗ | av_log(NULL, AV_LOG_ERROR, "Can't get stream info\n"); | |
58 | ✗ | return result; | |
59 | } | ||
60 | |||
61 | 1 | video_stream = av_find_best_stream(fmt_ctx, AVMEDIA_TYPE_VIDEO, -1, -1, NULL, 0); | |
62 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
|
1 | 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 | 1 | origin_par = fmt_ctx->streams[video_stream]->codecpar; | |
68 | |||
69 | 1 | codec = avcodec_find_decoder(origin_par->codec_id); | |
70 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
|
1 | if (!codec) { |
71 | ✗ | av_log(NULL, AV_LOG_ERROR, "Can't find decoder\n"); | |
72 | ✗ | return -1; | |
73 | } | ||
74 | |||
75 | 1 | ctx = avcodec_alloc_context3(codec); | |
76 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
|
1 | if (!ctx) { |
77 | ✗ | av_log(NULL, AV_LOG_ERROR, "Can't allocate decoder context\n"); | |
78 | ✗ | return AVERROR(ENOMEM); | |
79 | } | ||
80 | |||
81 | 1 | result = avcodec_parameters_to_context(ctx, origin_par); | |
82 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
|
1 | if (result) { |
83 | ✗ | av_log(NULL, AV_LOG_ERROR, "Can't copy decoder context\n"); | |
84 | ✗ | return result; | |
85 | } | ||
86 | |||
87 | 1 | result = avcodec_open2(ctx, codec, NULL); | |
88 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
|
1 | if (result < 0) { |
89 | ✗ | av_log(ctx, AV_LOG_ERROR, "Can't open decoder\n"); | |
90 | ✗ | return result; | |
91 | } | ||
92 | |||
93 | 1 | fr = av_frame_alloc(); | |
94 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
|
1 | if (!fr) { |
95 | ✗ | av_log(NULL, AV_LOG_ERROR, "Can't allocate frame\n"); | |
96 | ✗ | return AVERROR(ENOMEM); | |
97 | } | ||
98 | |||
99 | 1 | pkt = av_packet_alloc(); | |
100 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
|
1 | if (!pkt) { |
101 | ✗ | av_log(NULL, AV_LOG_ERROR, "Cannot allocate packet\n"); | |
102 | ✗ | return AVERROR(ENOMEM); | |
103 | } | ||
104 | |||
105 | 1 | byte_buffer_size = av_image_get_buffer_size(ctx->pix_fmt, ctx->width, ctx->height, 16); | |
106 | 1 | byte_buffer = av_malloc(byte_buffer_size); | |
107 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
|
1 | if (!byte_buffer) { |
108 | ✗ | av_log(NULL, AV_LOG_ERROR, "Can't allocate buffer\n"); | |
109 | ✗ | return AVERROR(ENOMEM); | |
110 | } | ||
111 | |||
112 | 1 | printf("#tb %d: %d/%d\n", video_stream, fmt_ctx->streams[video_stream]->time_base.num, fmt_ctx->streams[video_stream]->time_base.den); | |
113 | 1 | i = 0; | |
114 | |||
115 | 1 | result = 0; | |
116 |
1/2✓ Branch 0 taken 18 times.
✗ Branch 1 not taken.
|
18 | while (result >= 0) { |
117 | 18 | result = av_read_frame(fmt_ctx, pkt); | |
118 |
3/4✓ Branch 0 taken 17 times.
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 17 times.
|
18 | if (result >= 0 && pkt->stream_index != video_stream) { |
119 | ✗ | av_packet_unref(pkt); | |
120 | ✗ | continue; | |
121 | } | ||
122 | |||
123 |
2/2✓ Branch 0 taken 1 times.
✓ Branch 1 taken 17 times.
|
18 | if (result < 0) |
124 | 1 | result = avcodec_send_packet(ctx, NULL); | |
125 | else { | ||
126 |
1/2✓ Branch 0 taken 17 times.
✗ Branch 1 not taken.
|
17 | if (pkt->pts == AV_NOPTS_VALUE) |
127 | 17 | pkt->pts = pkt->dts = i; | |
128 | 17 | result = avcodec_send_packet(ctx, pkt); | |
129 | } | ||
130 | 18 | av_packet_unref(pkt); | |
131 | |||
132 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
|
18 | if (result < 0) { |
133 | ✗ | av_log(NULL, AV_LOG_ERROR, "Error submitting a packet for decoding\n"); | |
134 | ✗ | return result; | |
135 | } | ||
136 | |||
137 |
1/2✓ Branch 0 taken 35 times.
✗ Branch 1 not taken.
|
35 | while (result >= 0) { |
138 | 35 | result = avcodec_receive_frame(ctx, fr); | |
139 |
2/2✓ Branch 0 taken 1 times.
✓ Branch 1 taken 34 times.
|
35 | if (result == AVERROR_EOF) |
140 | 1 | goto finish; | |
141 |
2/2✓ Branch 0 taken 17 times.
✓ Branch 1 taken 17 times.
|
34 | else if (result == AVERROR(EAGAIN)) { |
142 | 17 | result = 0; | |
143 | 17 | break; | |
144 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 17 times.
|
17 | } else if (result < 0) { |
145 | ✗ | av_log(NULL, AV_LOG_ERROR, "Error decoding frame\n"); | |
146 | ✗ | return result; | |
147 | } | ||
148 | |||
149 | 17 | number_of_written_bytes = av_image_copy_to_buffer(byte_buffer, byte_buffer_size, | |
150 | 17 | (const uint8_t* const *)fr->data, (const int*) fr->linesize, | |
151 | 17 | ctx->pix_fmt, ctx->width, ctx->height, 1); | |
152 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 17 times.
|
17 | if (number_of_written_bytes < 0) { |
153 | ✗ | av_log(NULL, AV_LOG_ERROR, "Can't copy image to buffer\n"); | |
154 | ✗ | av_frame_unref(fr); | |
155 | ✗ | return number_of_written_bytes; | |
156 | } | ||
157 | 17 | printf("%d, %s, %s, %8"PRId64", %8d, 0x%08"PRIx32"\n", video_stream, | |
158 | 17 | av_ts2str(fr->pts), av_ts2str(fr->pkt_dts), fr->duration, | |
159 | number_of_written_bytes, av_adler32_update(0, (const uint8_t*)byte_buffer, number_of_written_bytes)); | ||
160 | |||
161 | 17 | av_frame_unref(fr); | |
162 | } | ||
163 | 17 | i++; | |
164 | } | ||
165 | |||
166 | ✗ | finish: | |
167 | 1 | av_packet_free(&pkt); | |
168 | 1 | av_frame_free(&fr); | |
169 | 1 | avformat_close_input(&fmt_ctx); | |
170 | 1 | avcodec_free_context(&ctx); | |
171 | 1 | av_freep(&byte_buffer); | |
172 | 1 | return 0; | |
173 | } | ||
174 | |||
175 | 1 | int main(int argc, char **argv) | |
176 | { | ||
177 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
|
1 | if (argc < 2) |
178 | { | ||
179 | ✗ | av_log(NULL, AV_LOG_ERROR, "Incorrect input\n"); | |
180 | ✗ | return 1; | |
181 | } | ||
182 | |||
183 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 1 times.
|
1 | if (video_decode_example(argv[1]) != 0) |
184 | ✗ | return 1; | |
185 | |||
186 | 1 | return 0; | |
187 | } | ||
188 |