7456
|
1 #include <stdio.h>
|
|
2 #include <stdlib.h>
|
|
3 #include <string.h>
|
8078
|
4 #include <errno.h>
|
7456
|
5
|
|
6 #include "../config.h"
|
|
7 #include "../mp_msg.h"
|
|
8
|
|
9 #ifdef HAVE_XVID
|
|
10
|
|
11 #include "codec-cfg.h"
|
|
12 #include "stream.h"
|
|
13 #include "demuxer.h"
|
|
14 #include "stheader.h"
|
|
15
|
|
16 #include "aviwrite.h"
|
|
17
|
|
18 #include "img_format.h"
|
|
19 #include "mp_image.h"
|
|
20 #include "vf.h"
|
|
21
|
|
22 #include <xvid.h>
|
|
23 #include "xvid_vbr.h"
|
|
24
|
|
25 #include "cfgparser.h"
|
|
26
|
|
27 /**********************************************************************/
|
|
28 /* Divx4 quality to XviD encoder motion flag presets */
|
|
29 static int const divx4_motion_presets[7] = {
|
8078
|
30 0,
|
|
31 PMV_QUICKSTOP16,
|
7456
|
32 PMV_EARLYSTOP16,
|
8078
|
33 PMV_EARLYSTOP16 | PMV_EARLYSTOP8,
|
|
34 PMV_EARLYSTOP16 | PMV_HALFPELREFINE16 | PMV_EARLYSTOP8 | PMV_HALFPELDIAMOND8,
|
|
35 PMV_EARLYSTOP16 | PMV_HALFPELREFINE16 | PMV_EARLYSTOP8 | PMV_HALFPELDIAMOND8 | PMV_ADVANCEDDIAMOND16,
|
|
36 PMV_EARLYSTOP16 | PMV_HALFPELREFINE16 | PMV_EXTSEARCH16 | PMV_EARLYSTOP8 | PMV_HALFPELREFINE8 |
|
|
37 PMV_HALFPELDIAMOND8 | PMV_USESQUARES16
|
|
38
|
7456
|
39 };
|
|
40
|
|
41 /* Divx4 quality to general encoder flag presets */
|
|
42 static int const divx4_general_presets[7] = {
|
|
43 0,
|
|
44 XVID_H263QUANT,
|
|
45 XVID_H263QUANT,
|
|
46 XVID_H263QUANT | XVID_HALFPEL,
|
|
47 XVID_H263QUANT | XVID_INTER4V | XVID_HALFPEL,
|
|
48 XVID_H263QUANT | XVID_INTER4V | XVID_HALFPEL,
|
|
49 XVID_H263QUANT | XVID_INTER4V | XVID_HALFPEL
|
|
50 };
|
|
51
|
8078
|
52 extern char* passtmpfile;
|
7456
|
53
|
8078
|
54 static int xvidenc_pass = 0;
|
|
55 static int xvidenc_quality = sizeof(divx4_motion_presets) / sizeof(divx4_motion_presets[0]) - 1; /* best quality */
|
|
56 static int xvidenc_bitrate = -1;
|
|
57 static int xvidenc_rc_reaction_delay_factor = -1;
|
|
58 static int xvidenc_rc_averaging_period = -1;
|
|
59 static int xvidenc_rc_buffer = -1;
|
|
60 static int xvidenc_min_quantizer = 2;
|
|
61 static int xvidenc_max_quantizer = -1;
|
|
62 static int xvidenc_min_key_interval = 0;
|
|
63 static int xvidenc_max_key_interval = -1;
|
|
64 static int xvidenc_mpeg_quant = 0;
|
|
65 static int xvidenc_lumi_mask = 0;
|
|
66 static int xvidenc_keyframe_boost = 0;
|
|
67 static int xvidenc_kfthreshold = 0;
|
|
68 static int xvidenc_kfreduction = 0;
|
|
69 static int xvidenc_fixed_quant = 0;
|
|
70 static int xvidenc_debug = 0;
|
|
71 static int xvidenc_hintedme = 0;
|
|
72 static char* xvidenc_hintfile = "xvid_hint_me.dat";
|
7458
|
73
|
7456
|
74 struct config xvidencopts_conf[] = {
|
8078
|
75 { "pass", &xvidenc_pass, CONF_TYPE_INT, CONF_RANGE, 0, 2, NULL},
|
|
76 { "quality", &xvidenc_quality, CONF_TYPE_INT, CONF_RANGE, 0,
|
7456
|
77 sizeof(divx4_motion_presets) / sizeof(divx4_motion_presets[0]) - 1, NULL},
|
8078
|
78 { "br", &xvidenc_bitrate, CONF_TYPE_INT, CONF_RANGE, 4, 24000000, NULL},
|
|
79 { "rc_reaction_delay_factor", &xvidenc_rc_reaction_delay_factor, CONF_TYPE_INT, 0, 0, 0, NULL},
|
|
80 { "rc_averaging_period", &xvidenc_rc_averaging_period, CONF_TYPE_INT, 0, 0, 0, NULL},
|
|
81 { "rc_buffer", &xvidenc_rc_buffer, CONF_TYPE_INT, 0, 0, 0, NULL},
|
|
82 { "min_quantizer", &xvidenc_min_quantizer, CONF_TYPE_INT, CONF_RANGE, 1, 31, NULL},
|
|
83 { "max_quantizer", &xvidenc_max_quantizer, CONF_TYPE_INT, CONF_RANGE, 1, 31, NULL},
|
|
84 { "min_key_interval", &xvidenc_min_key_interval, CONF_TYPE_INT, 0, 0, 0, NULL}, /* for XVID_MODE_2PASS_2 */
|
|
85 { "max_key_interval", &xvidenc_max_key_interval, CONF_TYPE_INT, 0, 0, 0, NULL},
|
|
86 { "mpeg_quant", &xvidenc_mpeg_quant, CONF_TYPE_FLAG, 0, 0, 1, NULL},
|
|
87 { "lumi_mask", &xvidenc_lumi_mask, CONF_TYPE_FLAG, 0, 0, 1, NULL},
|
|
88 { "keyframe_boost", &xvidenc_keyframe_boost, CONF_TYPE_INT, 0, 0, 0, NULL}, /* for XVID_MODE_2PASS_2 */
|
|
89 { "kfthreshold", &xvidenc_kfthreshold, CONF_TYPE_INT, 0, 0, 0, NULL}, /* for XVID_MODE_2PASS_2 */
|
|
90 { "kfreduction", &xvidenc_kfreduction, CONF_TYPE_INT, 0, 0, 0, NULL}, /* for XVID_MODE_2PASS_2 */
|
|
91 { "fixed_quant", &xvidenc_fixed_quant, CONF_TYPE_INT, CONF_RANGE, 1, 31, NULL}, /* for XVID_MODE_FIXED_QUANT */
|
|
92 { "debug", &xvidenc_debug, CONF_TYPE_FLAG, 0, 0, 1, NULL},
|
|
93 { "hintedme", &xvidenc_hintedme, CONF_TYPE_FLAG, 0, 0, 1, NULL},
|
|
94 { "hintfile", &xvidenc_hintfile, CONF_TYPE_STRING, 0, 0, 0, NULL},
|
7456
|
95 { NULL, NULL, 0, 0, 0, 0, NULL}
|
|
96 };
|
|
97
|
|
98 struct vf_priv_s {
|
|
99 aviwrite_stream_t* mux;
|
|
100 XVID_ENC_FRAME enc_frame;
|
|
101 void* enc_handle;
|
|
102 vbr_control_t vbr_state;
|
|
103 FILE *hintfile;
|
8078
|
104 void *hintstream;
|
7456
|
105 };
|
|
106
|
|
107 static int
|
|
108 config(struct vf_instance_s* vf,
|
|
109 int width, int height, int d_width, int d_height,
|
|
110 unsigned int flags, unsigned int outfmt)
|
|
111 {
|
|
112 XVID_ENC_PARAM enc_param;
|
8078
|
113 struct vf_priv_s *fp = vf->priv;
|
7456
|
114
|
8078
|
115 fp->mux->bih->biWidth = width;
|
|
116 fp->mux->bih->biHeight = height;
|
|
117 fp->mux->bih->biSizeImage = fp->mux->bih->biWidth * fp->mux->bih->biHeight * 3;
|
|
118 mp_msg(MSGT_MENCODER,MSGL_INFO,"videocodec: XViD (%dx%d fourcc=%x [%.4s])\n",
|
|
119 width, height, fp->mux->bih->biCompression, (char *)&fp->mux->bih->biCompression);
|
7456
|
120
|
8078
|
121 // initialize XViD core parameters
|
|
122 // ===============================
|
7456
|
123 memset(&enc_param, 0, sizeof(enc_param));
|
|
124 enc_param.width = width;
|
|
125 enc_param.height = height;
|
8078
|
126 enc_param.fincr = fp->mux->h.dwScale;
|
|
127 enc_param.fbase = fp->mux->h.dwRate;
|
|
128 if (xvidenc_bitrate > 16000)
|
|
129 enc_param.rc_bitrate = xvidenc_bitrate;
|
|
130 else if (xvidenc_bitrate > 0)
|
|
131 enc_param.rc_bitrate = xvidenc_bitrate * 1000;
|
|
132 else
|
|
133 enc_param.rc_bitrate = -1;
|
|
134 enc_param.rc_reaction_delay_factor = xvidenc_rc_reaction_delay_factor;
|
|
135 enc_param.rc_averaging_period = xvidenc_rc_averaging_period;
|
|
136 enc_param.rc_buffer = xvidenc_rc_buffer;
|
|
137 enc_param.min_quantizer = xvidenc_min_quantizer;
|
|
138 enc_param.max_quantizer = xvidenc_max_quantizer;
|
|
139 enc_param.max_key_interval = xvidenc_max_key_interval;
|
7456
|
140 switch (xvid_encore(NULL, XVID_ENC_CREATE, &enc_param, NULL)) {
|
|
141 case XVID_ERR_FAIL:
|
8078
|
142 mp_msg(MSGT_MENCODER,MSGL_ERR, "xvid: encoder creation failed\n");
|
7456
|
143 return 0;
|
|
144 case XVID_ERR_MEMORY:
|
8078
|
145 mp_msg(MSGT_MENCODER,MSGL_ERR, "xvid: encoder creation failed, out of memory\n");
|
7456
|
146 return 0;
|
|
147 case XVID_ERR_FORMAT:
|
8078
|
148 mp_msg(MSGT_MENCODER,MSGL_ERR, "xvid: encoder creation failed, bad format\n");
|
7456
|
149 return 0;
|
|
150 }
|
8078
|
151 fp->enc_handle = enc_param.handle;
|
7456
|
152
|
8078
|
153 // initialize XViD per-frame static parameters
|
|
154 // ===========================================
|
|
155 fp->enc_frame.general = divx4_general_presets[xvidenc_quality];
|
|
156 fp->enc_frame.motion = divx4_motion_presets[xvidenc_quality];
|
|
157 if (xvidenc_mpeg_quant) {
|
|
158 fp->enc_frame.general &= ~XVID_H263QUANT;
|
|
159 fp->enc_frame.general |= XVID_MPEGQUANT;
|
|
160 }
|
|
161 if (xvidenc_lumi_mask)
|
|
162 fp->enc_frame.general |= XVID_LUMIMASKING;
|
|
163
|
7456
|
164 switch (outfmt) {
|
|
165 case IMGFMT_YV12:
|
8078
|
166 fp->enc_frame.colorspace = XVID_CSP_YV12;
|
7456
|
167 break;
|
|
168 case IMGFMT_IYUV: case IMGFMT_I420:
|
8078
|
169 fp->enc_frame.colorspace = XVID_CSP_I420;
|
7456
|
170 break;
|
|
171 case IMGFMT_YUY2:
|
8078
|
172 fp->enc_frame.colorspace = XVID_CSP_YUY2;
|
7456
|
173 break;
|
|
174 case IMGFMT_UYVY:
|
8078
|
175 fp->enc_frame.colorspace = XVID_CSP_UYVY;
|
7456
|
176 break;
|
|
177 case IMGFMT_RGB24: case IMGFMT_BGR24:
|
8078
|
178 fp->enc_frame.colorspace = XVID_CSP_RGB24;
|
7456
|
179 break;
|
|
180 default:
|
|
181 mp_msg(MSGT_MENCODER,MSGL_ERR,"xvid: unsupported picture format (%s)!\n",
|
|
182 vo_format_name(outfmt));
|
|
183 return 0;
|
|
184 }
|
8078
|
185 fp->enc_frame.quant_intra_matrix = 0;
|
|
186 fp->enc_frame.quant_inter_matrix = 0;
|
|
187
|
|
188 // hinted ME
|
|
189 fp->hintstream = NULL;
|
|
190 fp->hintfile = NULL;
|
|
191 if (xvidenc_hintedme && (xvidenc_pass == 1 || xvidenc_pass == 2)) {
|
|
192 fp->hintstream = malloc( 100000 ); // this is what the vfw code in XViD CVS allocates
|
|
193 if (fp->hintstream == NULL)
|
|
194 mp_msg(MSGT_MENCODER,MSGL_ERR, "xvid: cannot allocate memory for hinted ME\n");
|
|
195 else {
|
|
196 fp->hintfile = fopen(xvidenc_hintfile, xvidenc_pass == 1 ? "w" : "r");
|
|
197 if (fp->hintfile == NULL) {
|
|
198 mp_msg(MSGT_MENCODER,MSGL_ERR, "xvid: %s: %s\n", strerror(errno), xvidenc_hintfile);
|
|
199 free(fp->hintstream);
|
|
200 }
|
|
201 }
|
|
202 if (fp->hintstream == NULL || fp->hintfile == NULL)
|
|
203 xvidenc_hintedme = 0;
|
|
204 }
|
7456
|
205
|
8078
|
206 // initialize VBR engine
|
|
207 // =====================
|
|
208 vbrSetDefaults(&fp->vbr_state);
|
|
209 if (xvidenc_pass == 0) {
|
|
210 if (xvidenc_fixed_quant >= 1) {
|
|
211 fp->vbr_state.mode = VBR_MODE_FIXED_QUANT;
|
|
212 fp->vbr_state.fixed_quant = xvidenc_fixed_quant;
|
|
213 } else
|
|
214 fp->vbr_state.mode = VBR_MODE_1PASS;
|
7456
|
215 }
|
8078
|
216 else if (xvidenc_pass == 1)
|
|
217 fp->vbr_state.mode = VBR_MODE_2PASS_1;
|
|
218 else if (xvidenc_pass == 2)
|
|
219 fp->vbr_state.mode = VBR_MODE_2PASS_2;
|
|
220 else
|
|
221 return -1;
|
|
222 fp->vbr_state.fps = (double)enc_param.fbase / enc_param.fincr;
|
|
223 fp->vbr_state.filename = passtmpfile;
|
|
224 fp->vbr_state.desired_bitrate = enc_param.rc_bitrate;
|
|
225 fp->vbr_state.min_iquant = fp->vbr_state.min_pquant = enc_param.min_quantizer;
|
|
226 fp->vbr_state.max_iquant = fp->vbr_state.max_pquant = enc_param.max_quantizer;
|
|
227 if (xvidenc_keyframe_boost)
|
|
228 fp->vbr_state.keyframe_boost = xvidenc_keyframe_boost;
|
|
229 if (xvidenc_kfthreshold)
|
|
230 fp->vbr_state.kftreshold = xvidenc_kfthreshold;
|
|
231 if (xvidenc_kfreduction)
|
|
232 fp->vbr_state.kfreduction = xvidenc_kfreduction;
|
|
233 if (xvidenc_min_key_interval)
|
|
234 fp->vbr_state.min_key_interval = xvidenc_min_key_interval;
|
|
235 fp->vbr_state.max_key_interval = enc_param.max_key_interval;
|
|
236 fp->vbr_state.debug = xvidenc_debug;
|
|
237 vbrInit(&fp->vbr_state);
|
|
238
|
7456
|
239 return 1;
|
|
240 }
|
|
241
|
|
242 static void
|
|
243 uninit(struct vf_instance_s* vf)
|
|
244 {
|
8078
|
245 struct vf_priv_s *fp = vf->priv;
|
|
246
|
|
247 if (fp->hintfile)
|
|
248 fclose(fp->hintfile);
|
|
249 if (fp->hintstream)
|
|
250 free(fp->hintstream);
|
|
251 vbrFinish(&fp->vbr_state);
|
7456
|
252 }
|
|
253
|
|
254 static int
|
|
255 control(struct vf_instance_s* vf, int request, void* data)
|
|
256 {
|
|
257 return CONTROL_UNKNOWN;
|
|
258 }
|
|
259
|
|
260 static int
|
|
261 query_format(struct vf_instance_s* vf, unsigned int fmt)
|
|
262 {
|
|
263 switch(fmt){
|
|
264 case IMGFMT_YV12: case IMGFMT_IYUV: case IMGFMT_I420:
|
|
265 return VFCAP_CSP_SUPPORTED | VFCAP_CSP_SUPPORTED_BY_HW;
|
|
266 case IMGFMT_YUY2: case IMGFMT_UYVY:
|
|
267 return VFCAP_CSP_SUPPORTED;
|
|
268 case IMGFMT_RGB24: case IMGFMT_BGR24:
|
|
269 return VFCAP_CSP_SUPPORTED | VFCAP_FLIPPED;
|
|
270 }
|
|
271 return 0;
|
|
272 }
|
|
273
|
|
274 static int
|
|
275 put_image(struct vf_instance_s* vf, mp_image_t *mpi)
|
|
276 {
|
|
277 XVID_ENC_STATS enc_stats;
|
8078
|
278 struct vf_priv_s *fp = vf->priv;
|
7456
|
279
|
8078
|
280 fp->enc_frame.bitstream = fp->mux->buffer;
|
|
281 fp->enc_frame.length = -1 /* fp->mux->buffer_size */;
|
|
282 fp->enc_frame.image = mpi->planes[0];
|
|
283 fp->enc_frame.quant = vbrGetQuant(&fp->vbr_state);
|
|
284 fp->enc_frame.intra = vbrGetIntra(&fp->vbr_state);
|
|
285 if (xvidenc_hintedme && xvidenc_pass == 1) {
|
|
286 fp->enc_frame.hint.hintstream = fp->hintstream;
|
|
287 fp->enc_frame.hint.rawhints = 0;
|
|
288 fp->enc_frame.general |= XVID_HINTEDME_GET;
|
7456
|
289 }
|
8078
|
290 else if (xvidenc_hintedme && xvidenc_pass == 2) {
|
|
291 size_t read;
|
|
292 int blocksize;
|
|
293 fp->enc_frame.general &= ~XVID_HINTEDME_SET;
|
|
294 read = fread(&blocksize, sizeof(blocksize), 1, fp->hintfile);
|
|
295 if (read == 1) {
|
|
296 read = fread(fp->hintstream, (size_t)blocksize, 1, fp->hintfile);
|
7456
|
297 if (read == 1) {
|
8078
|
298 fp->enc_frame.hint.hintstream = fp->hintstream;
|
|
299 fp->enc_frame.hint.hintlength = 0;
|
|
300 fp->enc_frame.hint.rawhints = 0;
|
|
301 fp->enc_frame.general |= XVID_HINTEDME_SET;
|
|
302 }
|
7456
|
303 else
|
8078
|
304 perror("xvid: hint file read block failure");
|
|
305 }
|
|
306 else
|
|
307 perror("xvid: hint file read failure");
|
7456
|
308 }
|
8078
|
309 switch (xvid_encore(fp->enc_handle, XVID_ENC_ENCODE, &fp->enc_frame, &enc_stats)) {
|
7456
|
310 case XVID_ERR_OK:
|
|
311 break;
|
|
312 case XVID_ERR_MEMORY:
|
|
313 mp_msg(MSGT_MENCODER, MSGL_ERR, "xvid: out of memory\n");
|
|
314 break;
|
|
315 case XVID_ERR_FORMAT:
|
|
316 mp_msg(MSGT_MENCODER, MSGL_ERR, "xvid: bad format\n");
|
|
317 break;
|
|
318 default:
|
|
319 mp_msg(MSGT_MENCODER, MSGL_ERR, "xvid: failure\n");
|
|
320 break;
|
|
321 }
|
8078
|
322 mencoder_write_chunk(fp->mux, fp->enc_frame.length, fp->enc_frame.intra ? 0x10 : 0);
|
|
323 vbrUpdate(&fp->vbr_state, enc_stats.quant, fp->enc_frame.intra,
|
|
324 enc_stats.hlength, fp->enc_frame.length, enc_stats.kblks, enc_stats.mblks, enc_stats.ublks);
|
|
325 if (fp->enc_frame.general & XVID_HINTEDME_GET) {
|
|
326 size_t wrote = fwrite(&fp->enc_frame.hint.hintlength, sizeof(fp->enc_frame.hint.hintlength), 1, fp->hintfile);
|
|
327 if (wrote == 1) {
|
|
328 wrote = fwrite(fp->enc_frame.hint.hintstream, fp->enc_frame.hint.hintlength, 1, fp->hintfile);
|
|
329 if (wrote != 1)
|
|
330 perror("xvid: hint write block failure");
|
7456
|
331 }
|
8078
|
332 else
|
|
333 perror("xvid: hint write failure");
|
7456
|
334 }
|
|
335 return 1;
|
|
336 }
|
|
337
|
|
338 //===========================================================================//
|
|
339
|
|
340 static int
|
|
341 vf_open(vf_instance_t *vf, char* args)
|
|
342 {
|
|
343 XVID_INIT_PARAM params = { 0, 0, 0};
|
|
344 vf->config = config;
|
|
345 vf->control = control;
|
|
346 vf->uninit = uninit;
|
|
347 vf->query_format = query_format;
|
|
348 vf->put_image = put_image;
|
|
349 vf->priv = malloc(sizeof(struct vf_priv_s));
|
|
350 memset(vf->priv, 0, sizeof(struct vf_priv_s));
|
|
351 vf->priv->mux = (aviwrite_stream_t*)args;
|
|
352
|
|
353 vf->priv->mux->bih = malloc(sizeof(BITMAPINFOHEADER));
|
|
354 vf->priv->mux->bih->biSize = sizeof(BITMAPINFOHEADER);
|
|
355 vf->priv->mux->bih->biWidth = 0;
|
|
356 vf->priv->mux->bih->biHeight = 0;
|
|
357 vf->priv->mux->bih->biPlanes = 1;
|
|
358 vf->priv->mux->bih->biBitCount = 24;
|
|
359 vf->priv->mux->bih->biCompression = mmioFOURCC('X','V','I','D');
|
|
360
|
|
361 if (xvid_init(NULL, 0, ¶ms, NULL) != XVID_ERR_OK) {
|
8078
|
362 mp_msg(MSGT_MENCODER,MSGL_ERR, "xvid: initialisation failure\n");
|
7456
|
363 abort();
|
|
364 }
|
|
365 if (params.api_version != API_VERSION) {
|
8078
|
366 mp_msg(MSGT_MENCODER,MSGL_ERR, "xvid: XviD library API version mismatch\n"
|
7456
|
367 "\texpected %d.%d, got %d.%d, you should recompile MPlayer.\n",
|
|
368 API_VERSION >> 16, API_VERSION & 0xff,
|
|
369 params.api_version >> 16, params.api_version & 0xff);
|
|
370 abort();
|
|
371 }
|
|
372
|
|
373 return 1;
|
|
374 }
|
|
375
|
|
376 vf_info_t ve_info_xvid = {
|
|
377 "XviD encoder",
|
|
378 "xvid",
|
8078
|
379 "Kim Minh Kaplan & Rémi Guyomarch",
|
7456
|
380 "for internal use by mencoder",
|
|
381 vf_open
|
|
382 };
|
|
383
|
|
384 //===========================================================================//
|
|
385 #endif
|