OLD | NEW |
1 /* | 1 /* |
2 * Copyright (c) 2010 The WebM project authors. All Rights Reserved. | 2 * Copyright (c) 2010 The WebM project authors. All Rights Reserved. |
3 * | 3 * |
4 * Use of this source code is governed by a BSD-style license | 4 * Use of this source code is governed by a BSD-style license |
5 * that can be found in the LICENSE file in the root of the source | 5 * that can be found in the LICENSE file in the root of the source |
6 * tree. An additional intellectual property rights grant can be found | 6 * tree. An additional intellectual property rights grant can be found |
7 * in the file PATENTS. All contributing project authors may | 7 * in the file PATENTS. All contributing project authors may |
8 * be found in the AUTHORS file in the root of the source tree. | 8 * be found in the AUTHORS file in the root of the source tree. |
9 */ | 9 */ |
10 | 10 |
(...skipping 46 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
57 // Encoding A Frame | 57 // Encoding A Frame |
58 // ---------------- | 58 // ---------------- |
59 // The frame is read as a continuous block (size width * height * 3 / 2) | 59 // The frame is read as a continuous block (size width * height * 3 / 2) |
60 // from the input file. If a frame was read (the input file has not hit | 60 // from the input file. If a frame was read (the input file has not hit |
61 // EOF) then the frame is passed to the encoder. Otherwise, a NULL | 61 // EOF) then the frame is passed to the encoder. Otherwise, a NULL |
62 // is passed, indicating the End-Of-Stream condition to the encoder. The | 62 // is passed, indicating the End-Of-Stream condition to the encoder. The |
63 // `frame_cnt` is reused as the presentation time stamp (PTS) and each | 63 // `frame_cnt` is reused as the presentation time stamp (PTS) and each |
64 // frame is shown for one frame-time in duration. The flags parameter is | 64 // frame is shown for one frame-time in duration. The flags parameter is |
65 // unused in this example. The deadline is set to VPX_DL_REALTIME to | 65 // unused in this example. The deadline is set to VPX_DL_REALTIME to |
66 // make the example run as quickly as possible. | 66 // make the example run as quickly as possible. |
| 67 |
| 68 // Forced Keyframes |
| 69 // ---------------- |
| 70 // Keyframes can be forced by setting the VPX_EFLAG_FORCE_KF bit of the |
| 71 // flags passed to `vpx_codec_control()`. In this example, we force a |
| 72 // keyframe every <keyframe-interval> frames. Note, the output stream can |
| 73 // contain additional keyframes beyond those that have been forced using the |
| 74 // VPX_EFLAG_FORCE_KF flag because of automatic keyframe placement by the |
| 75 // encoder. |
67 // | 76 // |
68 // Processing The Encoded Data | 77 // Processing The Encoded Data |
69 // --------------------------- | 78 // --------------------------- |
70 // Each packet of type `VPX_CODEC_CX_FRAME_PKT` contains the encoded data | 79 // Each packet of type `VPX_CODEC_CX_FRAME_PKT` contains the encoded data |
71 // for this frame. We write a IVF frame header, followed by the raw data. | 80 // for this frame. We write a IVF frame header, followed by the raw data. |
72 // | 81 // |
73 // Cleanup | 82 // Cleanup |
74 // ------- | 83 // ------- |
75 // The `vpx_codec_destroy` call frees any memory allocated by the codec. | 84 // The `vpx_codec_destroy` call frees any memory allocated by the codec. |
76 // | 85 // |
(...skipping 19 matching lines...) Expand all Loading... |
96 #include "vpx/vpx_encoder.h" | 105 #include "vpx/vpx_encoder.h" |
97 | 106 |
98 #include "./tools_common.h" | 107 #include "./tools_common.h" |
99 #include "./video_writer.h" | 108 #include "./video_writer.h" |
100 | 109 |
101 static const char *exec_name; | 110 static const char *exec_name; |
102 | 111 |
103 void usage_exit() { | 112 void usage_exit() { |
104 fprintf(stderr, | 113 fprintf(stderr, |
105 "Usage: %s <codec> <width> <height> <infile> <outfile> " | 114 "Usage: %s <codec> <width> <height> <infile> <outfile> " |
106 "[<error-resilient>]\nSee comments in simple_encoder.c for more " | 115 "<keyframe-interval> [<error-resilient>]\nSee comments in " |
107 "information.\n", | 116 "simple_encoder.c for more information.\n", |
108 exec_name); | 117 exec_name); |
109 exit(EXIT_FAILURE); | 118 exit(EXIT_FAILURE); |
110 } | 119 } |
111 | 120 |
112 static void encode_frame(vpx_codec_ctx_t *codec, | 121 static void encode_frame(vpx_codec_ctx_t *codec, |
113 vpx_image_t *img, | 122 vpx_image_t *img, |
114 int frame_index, | 123 int frame_index, |
| 124 int flags, |
115 VpxVideoWriter *writer) { | 125 VpxVideoWriter *writer) { |
116 vpx_codec_iter_t iter = NULL; | 126 vpx_codec_iter_t iter = NULL; |
117 const vpx_codec_cx_pkt_t *pkt = NULL; | 127 const vpx_codec_cx_pkt_t *pkt = NULL; |
118 const vpx_codec_err_t res = vpx_codec_encode(codec, img, frame_index, 1, 0, | 128 const vpx_codec_err_t res = vpx_codec_encode(codec, img, frame_index, 1, |
119 VPX_DL_GOOD_QUALITY); | 129 flags, VPX_DL_GOOD_QUALITY); |
120 if (res != VPX_CODEC_OK) | 130 if (res != VPX_CODEC_OK) |
121 die_codec(codec, "Failed to encode frame"); | 131 die_codec(codec, "Failed to encode frame"); |
122 | 132 |
123 while ((pkt = vpx_codec_get_cx_data(codec, &iter)) != NULL) { | 133 while ((pkt = vpx_codec_get_cx_data(codec, &iter)) != NULL) { |
124 if (pkt->kind == VPX_CODEC_CX_FRAME_PKT) { | 134 if (pkt->kind == VPX_CODEC_CX_FRAME_PKT) { |
125 const int keyframe = (pkt->data.frame.flags & VPX_FRAME_IS_KEY) != 0; | 135 const int keyframe = (pkt->data.frame.flags & VPX_FRAME_IS_KEY) != 0; |
126 if (!vpx_video_writer_write_frame(writer, | 136 if (!vpx_video_writer_write_frame(writer, |
127 pkt->data.frame.buf, | 137 pkt->data.frame.buf, |
128 pkt->data.frame.sz, | 138 pkt->data.frame.sz, |
129 pkt->data.frame.pts)) { | 139 pkt->data.frame.pts)) { |
(...skipping 11 matching lines...) Expand all Loading... |
141 vpx_codec_ctx_t codec; | 151 vpx_codec_ctx_t codec; |
142 vpx_codec_enc_cfg_t cfg; | 152 vpx_codec_enc_cfg_t cfg; |
143 int frame_count = 0; | 153 int frame_count = 0; |
144 vpx_image_t raw; | 154 vpx_image_t raw; |
145 vpx_codec_err_t res; | 155 vpx_codec_err_t res; |
146 VpxVideoInfo info = {0}; | 156 VpxVideoInfo info = {0}; |
147 VpxVideoWriter *writer = NULL; | 157 VpxVideoWriter *writer = NULL; |
148 const VpxInterface *encoder = NULL; | 158 const VpxInterface *encoder = NULL; |
149 const int fps = 30; // TODO(dkovalev) add command line argument | 159 const int fps = 30; // TODO(dkovalev) add command line argument |
150 const int bitrate = 200; // kbit/s TODO(dkovalev) add command line argument | 160 const int bitrate = 200; // kbit/s TODO(dkovalev) add command line argument |
| 161 int keyframe_interval = 0; |
| 162 |
| 163 // TODO(dkovalev): Add some simple command line parsing code to make the |
| 164 // command line more flexible. |
151 const char *codec_arg = NULL; | 165 const char *codec_arg = NULL; |
152 const char *width_arg = NULL; | 166 const char *width_arg = NULL; |
153 const char *height_arg = NULL; | 167 const char *height_arg = NULL; |
154 const char *infile_arg = NULL; | 168 const char *infile_arg = NULL; |
155 const char *outfile_arg = NULL; | 169 const char *outfile_arg = NULL; |
| 170 const char *keyframe_interval_arg = NULL; |
156 | 171 |
157 exec_name = argv[0]; | 172 exec_name = argv[0]; |
158 | 173 |
159 if (argc < 6) | 174 if (argc < 7) |
160 die("Invalid number of arguments"); | 175 die("Invalid number of arguments"); |
161 | 176 |
162 codec_arg = argv[1]; | 177 codec_arg = argv[1]; |
163 width_arg = argv[2]; | 178 width_arg = argv[2]; |
164 height_arg = argv[3]; | 179 height_arg = argv[3]; |
165 infile_arg = argv[4]; | 180 infile_arg = argv[4]; |
166 outfile_arg = argv[5]; | 181 outfile_arg = argv[5]; |
| 182 keyframe_interval_arg = argv[6]; |
167 | 183 |
168 encoder = get_vpx_encoder_by_name(codec_arg); | 184 encoder = get_vpx_encoder_by_name(codec_arg); |
169 if (!encoder) | 185 if (!encoder) |
170 die("Unsupported codec."); | 186 die("Unsupported codec."); |
171 | 187 |
172 info.codec_fourcc = encoder->fourcc; | 188 info.codec_fourcc = encoder->fourcc; |
173 info.frame_width = strtol(width_arg, NULL, 0); | 189 info.frame_width = strtol(width_arg, NULL, 0); |
174 info.frame_height = strtol(height_arg, NULL, 0); | 190 info.frame_height = strtol(height_arg, NULL, 0); |
175 info.time_base.numerator = 1; | 191 info.time_base.numerator = 1; |
176 info.time_base.denominator = fps; | 192 info.time_base.denominator = fps; |
177 | 193 |
178 if (info.frame_width <= 0 || | 194 if (info.frame_width <= 0 || |
179 info.frame_height <= 0 || | 195 info.frame_height <= 0 || |
180 (info.frame_width % 2) != 0 || | 196 (info.frame_width % 2) != 0 || |
181 (info.frame_height % 2) != 0) { | 197 (info.frame_height % 2) != 0) { |
182 die("Invalid frame size: %dx%d", info.frame_width, info.frame_height); | 198 die("Invalid frame size: %dx%d", info.frame_width, info.frame_height); |
183 } | 199 } |
184 | 200 |
185 if (!vpx_img_alloc(&raw, VPX_IMG_FMT_I420, info.frame_width, | 201 if (!vpx_img_alloc(&raw, VPX_IMG_FMT_I420, info.frame_width, |
186 info.frame_height, 1)) { | 202 info.frame_height, 1)) { |
187 die("Failed to allocate image."); | 203 die("Failed to allocate image."); |
188 } | 204 } |
189 | 205 |
| 206 keyframe_interval = strtol(keyframe_interval_arg, NULL, 0); |
| 207 if (keyframe_interval < 0) |
| 208 die("Invalid keyframe interval value."); |
| 209 |
190 printf("Using %s\n", vpx_codec_iface_name(encoder->interface())); | 210 printf("Using %s\n", vpx_codec_iface_name(encoder->interface())); |
191 | 211 |
192 res = vpx_codec_enc_config_default(encoder->interface(), &cfg, 0); | 212 res = vpx_codec_enc_config_default(encoder->interface(), &cfg, 0); |
193 if (res) | 213 if (res) |
194 die_codec(&codec, "Failed to get default codec config."); | 214 die_codec(&codec, "Failed to get default codec config."); |
195 | 215 |
196 cfg.g_w = info.frame_width; | 216 cfg.g_w = info.frame_width; |
197 cfg.g_h = info.frame_height; | 217 cfg.g_h = info.frame_height; |
198 cfg.g_timebase.num = info.time_base.numerator; | 218 cfg.g_timebase.num = info.time_base.numerator; |
199 cfg.g_timebase.den = info.time_base.denominator; | 219 cfg.g_timebase.den = info.time_base.denominator; |
200 cfg.rc_target_bitrate = bitrate; | 220 cfg.rc_target_bitrate = bitrate; |
201 cfg.g_error_resilient = argc > 6 ? strtol(argv[6], NULL, 0) : 0; | 221 cfg.g_error_resilient = argc > 7 ? strtol(argv[7], NULL, 0) : 0; |
202 | 222 |
203 writer = vpx_video_writer_open(outfile_arg, kContainerIVF, &info); | 223 writer = vpx_video_writer_open(outfile_arg, kContainerIVF, &info); |
204 if (!writer) | 224 if (!writer) |
205 die("Failed to open %s for writing.", outfile_arg); | 225 die("Failed to open %s for writing.", outfile_arg); |
206 | 226 |
207 if (!(infile = fopen(infile_arg, "rb"))) | 227 if (!(infile = fopen(infile_arg, "rb"))) |
208 die("Failed to open %s for reading.", infile_arg); | 228 die("Failed to open %s for reading.", infile_arg); |
209 | 229 |
210 if (vpx_codec_enc_init(&codec, encoder->interface(), &cfg, 0)) | 230 if (vpx_codec_enc_init(&codec, encoder->interface(), &cfg, 0)) |
211 die_codec(&codec, "Failed to initialize encoder"); | 231 die_codec(&codec, "Failed to initialize encoder"); |
212 | 232 |
213 while (vpx_img_read(&raw, infile)) | 233 while (vpx_img_read(&raw, infile)) { |
214 encode_frame(&codec, &raw, frame_count++, writer); | 234 int flags = 0; |
215 encode_frame(&codec, NULL, -1, writer); // flush the encoder | 235 if (keyframe_interval > 0 && frame_count % keyframe_interval == 0) |
| 236 flags |= VPX_EFLAG_FORCE_KF; |
| 237 encode_frame(&codec, &raw, frame_count++, flags, writer); |
| 238 } |
| 239 encode_frame(&codec, NULL, -1, 0, writer); // flush the encoder |
216 | 240 |
217 printf("\n"); | 241 printf("\n"); |
218 fclose(infile); | 242 fclose(infile); |
219 printf("Processed %d frames.\n", frame_count); | 243 printf("Processed %d frames.\n", frame_count); |
220 | 244 |
221 vpx_img_free(&raw); | 245 vpx_img_free(&raw); |
222 if (vpx_codec_destroy(&codec)) | 246 if (vpx_codec_destroy(&codec)) |
223 die_codec(&codec, "Failed to destroy codec."); | 247 die_codec(&codec, "Failed to destroy codec."); |
224 | 248 |
225 vpx_video_writer_close(writer); | 249 vpx_video_writer_close(writer); |
226 | 250 |
227 return EXIT_SUCCESS; | 251 return EXIT_SUCCESS; |
228 } | 252 } |
OLD | NEW |