Clean up headers a bit more.
[blender-staging.git] / source / blender / blenkernel / intern / writeffmpeg.c
1 /*
2  * $Id$
3  *
4  * ffmpeg-write support
5  *
6  * Partial Copyright (c) 2006 Peter Schlaile
7  *
8  * This program is free software; you can redistribute it and/or modify
9  * it under the terms of the GNU General Public License as published by
10  * the Free Software Foundation; either version 2 of the License, or
11  * (at your option) any later version.
12  *
13  * This program is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16  * GNU General Public License for more details.
17  *
18  */
19
20 #ifdef WITH_FFMPEG
21 #include <string.h>
22 #include <stdio.h>
23
24 #if defined(_WIN32) && defined(DEBUG) && !defined(__MINGW32__) && !defined(__CYGWIN__)
25 /* This does not seem necessary or present on MSVC 8, but may be needed in earlier versions? */
26 #if _MSC_VER < 1400
27 #include <stdint.h>
28 #endif
29 #endif
30
31 #include <stdlib.h>
32
33 #include <libavformat/avformat.h>
34 #include <libavcodec/avcodec.h>
35 #include <libavutil/rational.h>
36 #include <libswscale/swscale.h>
37 #include <libavcodec/opt.h>
38
39 #if LIBAVFORMAT_VERSION_INT < (49 << 16)
40 #define FFMPEG_OLD_FRAME_RATE 1
41 #else
42 #define FFMPEG_CODEC_IS_POINTER 1
43 #define FFMPEG_CODEC_TIME_BASE  1
44 #endif
45
46 #if LIBAVFORMAT_VERSION_INT >= (52 << 16)
47 #define OUTFILE_PB (outfile->pb)
48 #else
49 #define OUTFILE_PB (&outfile->pb)
50 #endif
51
52 #if defined(WIN32) && (!(defined snprintf))
53 #define snprintf _snprintf
54 #endif
55
56 #include "MEM_guardedalloc.h"
57
58 #include "DNA_scene_types.h"
59
60 #include "BLI_blenlib.h"
61
62 #include "AUD_C-API.h" /* must be before BKE_sound.h for define */
63
64 #include "BKE_global.h"
65 #include "BKE_idprop.h"
66 #include "BKE_main.h"
67 #include "BKE_report.h"
68 #include "BKE_sound.h"
69 #include "BKE_writeffmpeg.h"
70
71 #include "IMB_imbuf_types.h"
72 #include "IMB_imbuf.h"
73
74 extern void do_init_ffmpeg(void);
75
76 static int ffmpeg_type = 0;
77 static int ffmpeg_codec = CODEC_ID_MPEG4;
78 static int ffmpeg_audio_codec = CODEC_ID_NONE;
79 static int ffmpeg_video_bitrate = 1150;
80 static int ffmpeg_audio_bitrate = 128;
81 static int ffmpeg_gop_size = 12;
82 static int ffmpeg_autosplit = 0;
83 static int ffmpeg_autosplit_count = 0;
84
85 static AVFormatContext* outfile = 0;
86 static AVStream* video_stream = 0;
87 static AVStream* audio_stream = 0;
88 static AVFrame* current_frame = 0;
89 static struct SwsContext *img_convert_ctx = 0;
90
91 static uint8_t* video_buffer = 0;
92 static int video_buffersize = 0;
93
94 static uint8_t* audio_input_buffer = 0;
95 static int audio_input_samples = 0;
96 static uint8_t* audio_output_buffer = 0;
97 static int audio_outbuf_size = 0;
98 static double audio_time = 0.0f;
99
100 static AUD_Device* audio_mixdown_device = 0;
101
102 #define FFMPEG_AUTOSPLIT_SIZE 2000000000
103
104 /* Delete a picture buffer */
105
106 static void delete_picture(AVFrame* f)
107 {
108         if (f) {
109                 if (f->data[0]) MEM_freeN(f->data[0]);
110                 av_free(f);
111         }
112 }
113
114 #ifdef FFMPEG_CODEC_IS_POINTER
115 static AVCodecContext* get_codec_from_stream(AVStream* stream)
116 {
117         return stream->codec;
118 }
119 #else
120 static AVCodecContext* get_codec_from_stream(AVStream* stream)
121 {
122         return &stream->codec;
123 }
124 #endif
125
126 static int write_audio_frame(void) 
127 {
128         AVCodecContext* c = NULL;
129         AVPacket pkt;
130
131         c = get_codec_from_stream(audio_stream);
132
133         av_init_packet(&pkt);
134         pkt.size = 0;
135
136         AUD_readDevice(audio_mixdown_device, audio_input_buffer, audio_input_samples);
137         audio_time += (double) audio_input_samples / (double) c->sample_rate;
138
139         pkt.size = avcodec_encode_audio(c, audio_output_buffer,
140                                         audio_outbuf_size,
141                                         (short*) audio_input_buffer);
142
143         if(pkt.size < 0)
144         {
145                 // XXX error("Error writing audio packet");
146                 return -1;
147         }
148
149         pkt.data = audio_output_buffer;
150
151         if(c->coded_frame && c->coded_frame->pts != AV_NOPTS_VALUE)
152         {
153 #ifdef FFMPEG_CODEC_TIME_BASE
154                 pkt.pts = av_rescale_q(c->coded_frame->pts,
155                                            c->time_base, audio_stream->time_base);
156 #else
157                 pkt.pts = c->coded_frame->pts;
158 #endif
159                 fprintf(stderr, "Audio Frame PTS: %d\n", (int)pkt.pts);
160         }
161
162         pkt.stream_index = audio_stream->index;
163         pkt.flags |= PKT_FLAG_KEY;
164         if (av_interleaved_write_frame(outfile, &pkt) != 0) {
165                 fprintf(stderr, "Error writing audio packet!\n");
166                 return -1;
167         }
168         return 0;
169 }
170
171 /* Allocate a temporary frame */
172 static AVFrame* alloc_picture(int pix_fmt, int width, int height) 
173 {
174         AVFrame* f;
175         uint8_t* buf;
176         int size;
177         
178         /* allocate space for the struct */
179         f = avcodec_alloc_frame();
180         if (!f) return NULL;
181         size = avpicture_get_size(pix_fmt, width, height);
182         /* allocate the actual picture buffer */
183         buf = MEM_mallocN(size, "AVFrame buffer");
184         if (!buf) {
185                 free(f);
186                 return NULL;
187         }
188         avpicture_fill((AVPicture*)f, buf, pix_fmt, width, height);
189         return f;
190 }
191
192 /* Get the correct file extensions for the requested format,
193    first is always desired guess_format parameter */
194 static const char** get_file_extensions(int format) 
195 {
196         switch(format) {
197         case FFMPEG_DV: {
198                 static const char * rv[] = { ".dv", NULL };
199                 return rv;
200         }
201         case FFMPEG_MPEG1: {
202                 static const char * rv[] = { ".mpg", ".mpeg", NULL };
203                 return rv;
204         }
205         case FFMPEG_MPEG2: {
206                 static const char * rv[] = { ".dvd", ".vob", ".mpg", ".mpeg",
207                                                  NULL };
208                 return rv;
209         }
210         case FFMPEG_MPEG4: {
211                 static const char * rv[] = { ".mp4", ".mpg", ".mpeg", NULL };
212                 return rv;
213         }
214         case FFMPEG_AVI: {
215                 static const char * rv[] = { ".avi", NULL };
216                 return rv;
217         }
218         case FFMPEG_MOV: {
219                 static const char * rv[] = { ".mov", NULL };
220                 return rv;
221         }
222         case FFMPEG_H264: {
223                 /* FIXME: avi for now... */
224                 static const char * rv[] = { ".avi", NULL };
225                 return rv;
226         }
227
228         case FFMPEG_XVID: {
229                 /* FIXME: avi for now... */
230                 static const char * rv[] = { ".avi", NULL };
231                 return rv;
232         }
233         case FFMPEG_FLV: {
234                 static const char * rv[] = { ".flv", NULL };
235                 return rv;
236         }
237         case FFMPEG_MKV: {
238                 static const char * rv[] = { ".mkv", NULL };
239                 return rv;
240         }
241         case FFMPEG_OGG: {
242                 static const char * rv[] = { ".ogg", ".ogv", NULL };
243                 return rv;
244         }
245         case FFMPEG_MP3: {
246                 static const char * rv[] = { ".mp3", NULL };
247                 return rv;
248         }
249         case FFMPEG_WAV: {
250                 static const char * rv[] = { ".wav", NULL };
251                 return rv;
252         }
253         default:
254                 return NULL;
255         }
256 }
257
258 /* Write a frame to the output file */
259 static int write_video_frame(RenderData *rd, AVFrame* frame, ReportList *reports)
260 {
261         int outsize = 0;
262         int ret, success= 1;
263         AVCodecContext* c = get_codec_from_stream(video_stream);
264 #ifdef FFMPEG_CODEC_TIME_BASE
265         frame->pts = rd->cfra - rd->sfra;
266 #endif
267         if (rd->mode & R_FIELDS) {
268                 frame->top_field_first = ((rd->mode & R_ODDFIELD) != 0);
269         }
270
271         outsize = avcodec_encode_video(c, video_buffer, video_buffersize, 
272                                            frame);
273         if (outsize != 0) {
274                 AVPacket packet;
275                 av_init_packet(&packet);
276
277                 if (c->coded_frame->pts != AV_NOPTS_VALUE) {
278 #ifdef FFMPEG_CODEC_TIME_BASE
279                         packet.pts = av_rescale_q(c->coded_frame->pts,
280                                                   c->time_base,
281                                                   video_stream->time_base);
282 #else
283                         packet.pts = c->coded_frame->pts;
284 #endif
285                         fprintf(stderr, "Video Frame PTS: %d\n", (int)packet.pts);
286                 } else {
287                         fprintf(stderr, "Video Frame PTS: not set\n");
288                 }
289                 if (c->coded_frame->key_frame)
290                         packet.flags |= PKT_FLAG_KEY;
291                 packet.stream_index = video_stream->index;
292                 packet.data = video_buffer;
293                 packet.size = outsize;
294                 ret = av_interleaved_write_frame(outfile, &packet);
295         } else {
296                 ret = 0;
297         }
298
299         if (ret != 0) {
300                 success= 0;
301                 BKE_report(reports, RPT_ERROR, "Error writing frame.");
302         }
303
304         return success;
305 }
306
307 /* read and encode a frame of audio from the buffer */
308 static AVFrame* generate_video_frame(uint8_t* pixels, ReportList *reports) 
309 {
310         uint8_t* rendered_frame;
311
312         AVCodecContext* c = get_codec_from_stream(video_stream);
313         int width = c->width;
314         int height = c->height;
315         AVFrame* rgb_frame;
316
317         if (c->pix_fmt != PIX_FMT_BGR32) {
318                 rgb_frame = alloc_picture(PIX_FMT_BGR32, width, height);
319                 if (!rgb_frame) {
320                         BKE_report(reports, RPT_ERROR, "Couldn't allocate temporary frame.");
321                         return NULL;
322                 }
323         } else {
324                 rgb_frame = current_frame;
325         }
326
327         rendered_frame = pixels;
328
329         /* Do RGBA-conversion and flipping in one step depending
330            on CPU-Endianess */
331
332         if (ENDIAN_ORDER == L_ENDIAN) {
333                 int y;
334                 for (y = 0; y < height; y++) {
335                         uint8_t* target = rgb_frame->data[0]
336                                 + width * 4 * (height - y - 1);
337                         uint8_t* src = rendered_frame + width * 4 * y;
338                         uint8_t* end = src + width * 4;
339                         while (src != end) {
340                                 target[3] = src[3];
341                                 target[2] = src[2];
342                                 target[1] = src[1];
343                                 target[0] = src[0];
344
345                                 target += 4;
346                                 src += 4;
347                         }
348                 }
349         } else {
350                 int y;
351                 for (y = 0; y < height; y++) {
352                         uint8_t* target = rgb_frame->data[0]
353                                 + width * 4 * (height - y - 1);
354                         uint8_t* src = rendered_frame + width * 4 * y;
355                         uint8_t* end = src + width * 4;
356                         while (src != end) {
357                                 target[3] = src[0];
358                                 target[2] = src[1];
359                                 target[1] = src[2];
360                                 target[0] = src[3];
361
362                                 target += 4;
363                                 src += 4;
364                         }
365                 }
366         }
367
368         if (c->pix_fmt != PIX_FMT_BGR32) {
369                 sws_scale(img_convert_ctx, (const uint8_t * const*) rgb_frame->data,
370                           rgb_frame->linesize, 0, c->height, 
371                           current_frame->data, current_frame->linesize);
372                 delete_picture(rgb_frame);
373         }
374         return current_frame;
375 }
376
377 static void set_ffmpeg_property_option(AVCodecContext* c, IDProperty * prop)
378 {
379         char name[128];
380         char * param;
381         const AVOption * rv = NULL;
382
383         fprintf(stderr, "FFMPEG expert option: %s: ", prop->name);
384
385         BLI_strncpy(name, prop->name, sizeof(name));
386
387         param = strchr(name, ':');
388
389         if (param) {
390                 *param++ = 0;
391         }
392
393         switch(prop->type) {
394         case IDP_STRING:
395                 fprintf(stderr, "%s.\n", IDP_String(prop));
396                 rv = av_set_string(c, prop->name, IDP_String(prop));
397                 break;
398         case IDP_FLOAT:
399                 fprintf(stderr, "%g.\n", IDP_Float(prop));
400                 rv = av_set_double(c, prop->name, IDP_Float(prop));
401                 break;
402         case IDP_INT:
403                 fprintf(stderr, "%d.\n", IDP_Int(prop));
404                 
405                 if (param) {
406                         if (IDP_Int(prop)) {
407                                 rv = av_set_string(c, name, param);
408                         } else {
409                                 return;
410                         }
411                 } else {
412                         rv = av_set_int(c, prop->name, IDP_Int(prop));
413                 }
414                 break;
415         }
416
417         if (!rv) {
418                 fprintf(stderr, "ffmpeg-option not supported: %s! Skipping.\n",
419                         prop->name);
420         }
421 }
422
423 static void set_ffmpeg_properties(RenderData *rd, AVCodecContext *c, const char * prop_name)
424 {
425         IDProperty * prop;
426         void * iter;
427         IDProperty * curr;
428
429         if (!rd->ffcodecdata.properties) {
430                 return;
431         }
432         
433         prop = IDP_GetPropertyFromGroup(
434                 rd->ffcodecdata.properties, (char*) prop_name);
435         if (!prop) {
436                 return;
437         }
438
439         iter = IDP_GetGroupIterator(prop);
440
441         while ((curr = IDP_GroupIterNext(iter)) != NULL) {
442                 set_ffmpeg_property_option(c, curr);
443         }
444 }
445
446 /* prepare a video stream for the output file */
447
448 static AVStream* alloc_video_stream(RenderData *rd, int codec_id, AVFormatContext* of,
449                                         int rectx, int recty) 
450 {
451         AVStream* st;
452         AVCodecContext* c;
453         AVCodec* codec;
454         st = av_new_stream(of, 0);
455         if (!st) return NULL;
456
457         /* Set up the codec context */
458         
459         c = get_codec_from_stream(st);
460         c->codec_id = codec_id;
461         c->codec_type = CODEC_TYPE_VIDEO;
462
463
464         /* Get some values from the current render settings */
465         
466         c->width = rectx;
467         c->height = recty;
468
469 #ifdef FFMPEG_CODEC_TIME_BASE
470         /* FIXME: Really bad hack (tm) for NTSC support */
471         if (ffmpeg_type == FFMPEG_DV && rd->frs_sec != 25) {
472                 c->time_base.den = 2997;
473                 c->time_base.num = 100;
474         } else if ((double) ((int) rd->frs_sec_base) == 
475                    rd->frs_sec_base) {
476                 c->time_base.den = rd->frs_sec;
477                 c->time_base.num = (int) rd->frs_sec_base;
478         } else {
479                 c->time_base.den = rd->frs_sec * 100000;
480                 c->time_base.num = ((double) rd->frs_sec_base) * 100000;
481         }
482 #else
483         /* FIXME: Really bad hack (tm) for NTSC support */
484         if (ffmpeg_type == FFMPEG_DV && rd->frs_sec != 25) {
485                 c->frame_rate = 2997;
486                 c->frame_rate_base = 100;
487         } else if ((double) ((int) rd->frs_sec_base) == 
488                    rd->frs_sec_base) {
489                 c->frame_rate = rd->frs_sec;
490                 c->frame_rate_base = rd->frs_sec_base;
491         } else {
492                 c->frame_rate = rd->frs_sec * 100000;
493                 c->frame_rate_base = ((double) rd->frs_sec_base)*100000;
494         }
495 #endif
496         
497         c->gop_size = ffmpeg_gop_size;
498         c->bit_rate = ffmpeg_video_bitrate*1000;
499         c->rc_max_rate = rd->ffcodecdata.rc_max_rate*1000;
500         c->rc_min_rate = rd->ffcodecdata.rc_min_rate*1000;
501         c->rc_buffer_size = rd->ffcodecdata.rc_buffer_size * 1024;
502         c->rc_initial_buffer_occupancy 
503                 = rd->ffcodecdata.rc_buffer_size*3/4;
504         c->rc_buffer_aggressivity = 1.0;
505         c->me_method = ME_EPZS;
506         
507         codec = avcodec_find_encoder(c->codec_id);
508         if (!codec) return NULL;
509         
510         /* Be sure to use the correct pixel format(e.g. RGB, YUV) */
511         
512         if (codec->pix_fmts) {
513                 c->pix_fmt = codec->pix_fmts[0];
514         } else {
515                 /* makes HuffYUV happy ... */
516                 c->pix_fmt = PIX_FMT_YUV422P;
517         }
518
519         if (codec_id == CODEC_ID_XVID) {
520                 /* arghhhh ... */
521                 c->pix_fmt = PIX_FMT_YUV420P;
522                 c->codec_tag = (('D'<<24) + ('I'<<16) + ('V'<<8) + 'X');
523         }
524
525         if (codec_id == CODEC_ID_H264) {
526                 /* correct wrong default ffmpeg param which crash x264 */
527                 c->qmin=10;
528                 c->qmax=51;
529         }
530         
531         if ((of->oformat->flags & AVFMT_GLOBALHEADER)
532 //              || !strcmp(of->oformat->name, "mp4")
533 //          || !strcmp(of->oformat->name, "mov")
534 //          || !strcmp(of->oformat->name, "3gp")
535                 ) {
536                 fprintf(stderr, "Using global header\n");
537                 c->flags |= CODEC_FLAG_GLOBAL_HEADER;
538         }
539         
540         /* Determine whether we are encoding interlaced material or not */
541         if (rd->mode & R_FIELDS) {
542                 fprintf(stderr, "Encoding interlaced video\n");
543                 c->flags |= CODEC_FLAG_INTERLACED_DCT;
544                 c->flags |= CODEC_FLAG_INTERLACED_ME;
545         }
546
547         /* xasp & yasp got float lately... */
548
549         st->sample_aspect_ratio = c->sample_aspect_ratio = av_d2q(
550                 ((double) rd->xasp / (double) rd->yasp), 255);
551
552         set_ffmpeg_properties(rd, c, "video");
553         
554         if (avcodec_open(c, codec) < 0) {
555                 //
556                 //XXX error("Couldn't initialize codec");
557                 return NULL;
558         }
559
560         video_buffersize = 2000000;
561         video_buffer = (uint8_t*)MEM_mallocN(video_buffersize, 
562                                                  "FFMPEG video buffer");
563         
564         current_frame = alloc_picture(c->pix_fmt, c->width, c->height);
565
566         img_convert_ctx = sws_getContext(c->width, c->height,
567                                          PIX_FMT_BGR32,
568                                          c->width, c->height,
569                                          c->pix_fmt,
570                                          SWS_BICUBIC,
571                                          NULL, NULL, NULL);
572         return st;
573 }
574
575 /* Prepare an audio stream for the output file */
576
577 static AVStream* alloc_audio_stream(RenderData *rd, int codec_id, AVFormatContext* of) 
578 {
579         AVStream* st;
580         AVCodecContext* c;
581         AVCodec* codec;
582
583         st = av_new_stream(of, 1);
584         if (!st) return NULL;
585
586         c = get_codec_from_stream(st);
587         c->codec_id = codec_id;
588         c->codec_type = CODEC_TYPE_AUDIO;
589
590         c->sample_rate = rd->ffcodecdata.audio_mixrate;
591         c->bit_rate = ffmpeg_audio_bitrate*1000;
592         c->sample_fmt = SAMPLE_FMT_S16;
593         c->channels = 2;
594         codec = avcodec_find_encoder(c->codec_id);
595         if (!codec) {
596                 //XXX error("Couldn't find a valid audio codec");
597                 return NULL;
598         }
599
600         set_ffmpeg_properties(rd, c, "audio");
601
602         if (avcodec_open(c, codec) < 0) {
603                 //XXX error("Couldn't initialize audio codec");
604                 return NULL;
605         }
606
607         audio_outbuf_size = FF_MIN_BUFFER_SIZE;
608
609         if((c->codec_id >= CODEC_ID_PCM_S16LE) && (c->codec_id <= CODEC_ID_PCM_DVD))
610                 audio_input_samples = audio_outbuf_size * 8 / c->bits_per_coded_sample / c->channels;
611         else
612         {
613                 audio_input_samples = c->frame_size;
614                 if(c->frame_size * c->channels * sizeof(int16_t) * 4 > audio_outbuf_size)
615                         audio_outbuf_size = c->frame_size * c->channels * sizeof(int16_t) * 4;
616         }
617
618         audio_output_buffer = (uint8_t*)MEM_mallocN(
619                 audio_outbuf_size, "FFMPEG audio encoder input buffer");
620
621         audio_input_buffer = (uint8_t*)MEM_mallocN(
622                 audio_input_samples * c->channels * sizeof(int16_t),
623                 "FFMPEG audio encoder output buffer");
624
625         audio_time = 0.0f;
626
627         return st;
628 }
629 /* essential functions -- start, append, end */
630
631 static int start_ffmpeg_impl(struct RenderData *rd, int rectx, int recty, ReportList *reports)
632 {
633         /* Handle to the output file */
634         AVFormatContext* of;
635         AVOutputFormat* fmt;
636         char name[256];
637         const char ** exts;
638
639         ffmpeg_type = rd->ffcodecdata.type;
640         ffmpeg_codec = rd->ffcodecdata.codec;
641         ffmpeg_audio_codec = rd->ffcodecdata.audio_codec;
642         ffmpeg_video_bitrate = rd->ffcodecdata.video_bitrate;
643         ffmpeg_audio_bitrate = rd->ffcodecdata.audio_bitrate;
644         ffmpeg_gop_size = rd->ffcodecdata.gop_size;
645         ffmpeg_autosplit = rd->ffcodecdata.flags
646                 & FFMPEG_AUTOSPLIT_OUTPUT;
647         
648         do_init_ffmpeg();
649
650         /* Determine the correct filename */
651         filepath_ffmpeg(name, rd);
652         fprintf(stderr, "Starting output to %s(ffmpeg)...\n"
653                 "  Using type=%d, codec=%d, audio_codec=%d,\n"
654                 "  video_bitrate=%d, audio_bitrate=%d,\n"
655                 "  gop_size=%d, autosplit=%d\n"
656                 "  render width=%d, render height=%d\n", 
657                 name, ffmpeg_type, ffmpeg_codec, ffmpeg_audio_codec,
658                 ffmpeg_video_bitrate, ffmpeg_audio_bitrate,
659                 ffmpeg_gop_size, ffmpeg_autosplit, rectx, recty);
660         
661         exts = get_file_extensions(ffmpeg_type);
662         if (!exts) {
663                 BKE_report(reports, RPT_ERROR, "No valid formats found.");
664                 return 0;
665         }
666         fmt = guess_format(NULL, exts[0], NULL);
667         if (!fmt) {
668                 BKE_report(reports, RPT_ERROR, "No valid formats found.");
669                 return 0;
670         }
671
672         of = av_alloc_format_context();
673         if (!of) {
674                 BKE_report(reports, RPT_ERROR, "Error opening output file");
675                 return 0;
676         }
677         
678         of->oformat = fmt;
679         of->packet_size= rd->ffcodecdata.mux_packet_size;
680         if (ffmpeg_audio_codec != CODEC_ID_NONE) {
681                 of->mux_rate = rd->ffcodecdata.mux_rate;
682         } else {
683                 of->mux_rate = 0;
684         }
685
686         of->preload = (int)(0.5*AV_TIME_BASE);
687         of->max_delay = (int)(0.7*AV_TIME_BASE);
688
689         fmt->audio_codec = ffmpeg_audio_codec;
690
691         snprintf(of->filename, sizeof(of->filename), "%s", name);
692         /* set the codec to the user's selection */
693         switch(ffmpeg_type) {
694         case FFMPEG_AVI:
695         case FFMPEG_MOV:
696         case FFMPEG_OGG:
697         case FFMPEG_MKV:
698                 fmt->video_codec = ffmpeg_codec;
699                 break;
700         case FFMPEG_DV:
701                 fmt->video_codec = CODEC_ID_DVVIDEO;
702                 break;
703         case FFMPEG_MPEG1:
704                 fmt->video_codec = CODEC_ID_MPEG1VIDEO;
705                 break;
706         case FFMPEG_MPEG2:
707                 fmt->video_codec = CODEC_ID_MPEG2VIDEO;
708                 break;
709         case FFMPEG_H264:
710                 fmt->video_codec = CODEC_ID_H264;
711                 break;
712         case FFMPEG_XVID:
713                 fmt->video_codec = CODEC_ID_XVID;
714                 break;
715         case FFMPEG_FLV:
716                 fmt->video_codec = CODEC_ID_FLV1;
717                 break;
718         case FFMPEG_MP3:
719                 fmt->audio_codec = CODEC_ID_MP3;
720         case FFMPEG_WAV:
721                 fmt->video_codec = CODEC_ID_NONE;
722                 break;
723         case FFMPEG_MPEG4:
724         default:
725                 fmt->video_codec = CODEC_ID_MPEG4;
726                 break;
727         }
728         if (fmt->video_codec == CODEC_ID_DVVIDEO) {
729                 if (rectx != 720) {
730                         BKE_report(reports, RPT_ERROR, "Render width has to be 720 pixels for DV!");
731                         return 0;
732                 }
733                 if (rd->frs_sec != 25 && recty != 480) {
734                         BKE_report(reports, RPT_ERROR, "Render height has to be 480 pixels for DV-NTSC!");
735                         return 0;
736                 }
737                 if (rd->frs_sec == 25 && recty != 576) {
738                         BKE_report(reports, RPT_ERROR, "Render height has to be 576 pixels for DV-PAL!");
739                         return 0;
740                 }
741         }
742         
743         if (ffmpeg_type == FFMPEG_DV) {
744                 fmt->audio_codec = CODEC_ID_PCM_S16LE;
745                 if (ffmpeg_audio_codec != CODEC_ID_NONE && rd->ffcodecdata.audio_mixrate != 48000) {
746                         BKE_report(reports, RPT_ERROR, "FFMPEG only supports 48khz / stereo audio for DV!");
747                         return 0;
748                 }
749         }
750         
751         if (fmt->video_codec != CODEC_ID_NONE) {
752                 video_stream = alloc_video_stream(rd, fmt->video_codec, of, rectx, recty);
753                 printf("alloc video stream %p\n", video_stream);
754                 if (!video_stream) {
755                         BKE_report(reports, RPT_ERROR, "Error initializing video stream.");
756                         return 0;
757                 }
758         }
759
760         if (ffmpeg_audio_codec != CODEC_ID_NONE) {
761                 audio_stream = alloc_audio_stream(rd, fmt->audio_codec, of);
762                 if (!audio_stream) {
763                         BKE_report(reports, RPT_ERROR, "Error initializing audio stream.");
764                         return 0;
765                 }
766         }
767         if (av_set_parameters(of, NULL) < 0) {
768                 BKE_report(reports, RPT_ERROR, "Error setting output parameters.");
769                 return 0;
770         }
771         if (!(fmt->flags & AVFMT_NOFILE)) {
772                 if (url_fopen(&of->pb, name, URL_WRONLY) < 0) {
773                         BKE_report(reports, RPT_ERROR, "Could not open file for writing.");
774                         return 0;
775                 }
776         }
777
778         av_write_header(of);
779         outfile = of;
780         dump_format(of, 0, name, 1);
781
782         return 1;
783 }
784
785 /**
786  * Writes any delayed frames in the encoder. This function is called before 
787  * closing the encoder.
788  *
789  * <p>
790  * Since an encoder may use both past and future frames to predict 
791  * inter-frames (H.264 B-frames, for example), it can output the frames 
792  * in a different order from the one it was given.
793  * For example, when sending frames 1, 2, 3, 4 to the encoder, it may write
794  * them in the order 1, 4, 2, 3 - first the two frames used for predition, 
795  * and then the bidirectionally-predicted frames. What this means in practice 
796  * is that the encoder may not immediately produce one output frame for each 
797  * input frame. These delayed frames must be flushed before we close the 
798  * stream. We do this by calling avcodec_encode_video with NULL for the last 
799  * parameter.
800  * </p>
801  */
802 void flush_ffmpeg(void)
803 {
804         int outsize = 0;
805         int ret = 0;
806         
807         AVCodecContext* c = get_codec_from_stream(video_stream);
808         /* get the delayed frames */
809         while (1) {
810                 AVPacket packet;
811                 av_init_packet(&packet);
812                 
813                 outsize = avcodec_encode_video(c, video_buffer, video_buffersize, NULL);
814                 if (outsize < 0) {
815                         fprintf(stderr, "Error encoding delayed frame %d\n", outsize);
816                         break;
817                 }
818                 if (outsize == 0) {
819                         break;
820                 }
821                 if (c->coded_frame->pts != AV_NOPTS_VALUE) {
822 #ifdef FFMPEG_CODEC_TIME_BASE
823                         packet.pts = av_rescale_q(c->coded_frame->pts,
824                                                   c->time_base,
825                                                   video_stream->time_base);
826 #else
827                         packet.pts = c->coded_frame->pts;
828 #endif
829                         fprintf(stderr, "Video Frame PTS: %d\n", (int)packet.pts);
830                 } else {
831                         fprintf(stderr, "Video Frame PTS: not set\n");
832                 }
833                 if (c->coded_frame->key_frame) {
834                         packet.flags |= PKT_FLAG_KEY;
835                 }
836                 packet.stream_index = video_stream->index;
837                 packet.data = video_buffer;
838                 packet.size = outsize;
839                 ret = av_interleaved_write_frame(outfile, &packet);
840                 if (ret != 0) {
841                         fprintf(stderr, "Error writing delayed frame %d\n", ret);
842                         break;
843                 }
844         }
845         avcodec_flush_buffers(get_codec_from_stream(video_stream));
846 }
847
848 /* **********************************************************************
849    * public interface
850    ********************************************************************** */
851
852 /* Get the output filename-- similar to the other output formats */
853 void filepath_ffmpeg(char* string, RenderData* rd) {
854         char autosplit[20];
855
856         const char ** exts = get_file_extensions(rd->ffcodecdata.type);
857         const char ** fe = exts;
858
859         if (!string || !exts) return;
860
861         strcpy(string, rd->pic);
862         BLI_path_abs(string, G.main->name);
863
864         BLI_make_existing_file(string);
865
866         autosplit[0] = 0;
867
868         if ((rd->ffcodecdata.flags & FFMPEG_AUTOSPLIT_OUTPUT) != 0) {
869                 sprintf(autosplit, "_%03d", ffmpeg_autosplit_count);
870         }
871
872         while (*fe) {
873                 if (BLI_strcasecmp(string + strlen(string) - strlen(*fe), 
874                                    *fe) == 0) {
875                         break;
876                 }
877                 fe++;
878         }
879
880         if (!*fe) {
881                 strcat(string, autosplit);
882
883                 BLI_path_frame_range(string, rd->sfra, rd->efra, 4);
884                 strcat(string, *exts);
885         } else {
886                 *(string + strlen(string) - strlen(*fe)) = 0;
887                 strcat(string, autosplit);
888                 strcat(string, *fe);
889         }
890 }
891
892 int start_ffmpeg(struct Scene *scene, RenderData *rd, int rectx, int recty, ReportList *reports)
893 {
894         int success;
895
896         ffmpeg_autosplit_count = 0;
897
898         success = start_ffmpeg_impl(rd, rectx, recty, reports);
899
900         if(audio_stream)
901         {
902                 AVCodecContext* c = get_codec_from_stream(audio_stream);
903                 AUD_DeviceSpecs specs;
904                 specs.channels = c->channels;
905                 specs.format = AUD_FORMAT_S16;
906                 specs.rate = rd->ffcodecdata.audio_mixrate;
907                 audio_mixdown_device = sound_mixdown(scene, specs, rd->sfra, rd->ffcodecdata.audio_volume);
908         }
909
910         return success;
911 }
912
913 void end_ffmpeg(void);
914
915 static void write_audio_frames(double to_pts)
916 {
917         int finished = 0;
918
919         while (audio_stream && !finished) {
920                 if((audio_time >= to_pts) ||
921                    (write_audio_frame())) {
922                         finished = 1;
923                 }
924         }
925 }
926
927 int append_ffmpeg(RenderData *rd, int frame, int *pixels, int rectx, int recty, ReportList *reports) 
928 {
929         AVFrame* avframe;
930         int success = 1;
931
932         fprintf(stderr, "Writing frame %i, "
933                 "render width=%d, render height=%d\n", frame,
934                 rectx, recty);
935
936 // why is this done before writing the video frame and again at end_ffmpeg?
937 //      write_audio_frames(frame / (((double)rd->frs_sec) / rd->frs_sec_base));
938
939         if(video_stream)
940         {
941                 avframe= generate_video_frame((unsigned char*) pixels, reports);
942                 success= (avframe && write_video_frame(rd, avframe, reports));
943
944                 if (ffmpeg_autosplit) {
945                         if (url_ftell(OUTFILE_PB) > FFMPEG_AUTOSPLIT_SIZE) {
946                                 end_ffmpeg();
947                                 ffmpeg_autosplit_count++;
948                                 success &= start_ffmpeg_impl(rd, rectx, recty, reports);
949                         }
950                 }
951         }
952
953         write_audio_frames((frame - rd->sfra) / (((double)rd->frs_sec) / rd->frs_sec_base));
954
955         return success;
956 }
957
958 void end_ffmpeg(void)
959 {
960         unsigned int i;
961         
962         fprintf(stderr, "Closing ffmpeg...\n");
963
964 /*      if (audio_stream) { SEE UPPER
965                 write_audio_frames();
966         }*/
967
968         if(audio_mixdown_device)
969         {
970                 AUD_closeReadDevice(audio_mixdown_device);
971                 audio_mixdown_device = 0;
972         }
973         
974         if (video_stream && get_codec_from_stream(video_stream)) {
975                 fprintf(stderr, "Flushing delayed frames...\n");
976                 flush_ffmpeg ();                
977         }
978         
979         if (outfile) {
980                 av_write_trailer(outfile);
981         }
982         
983         /* Close the video codec */
984
985         if (video_stream && get_codec_from_stream(video_stream)) {
986                 avcodec_close(get_codec_from_stream(video_stream));
987                 printf("zero video stream %p\n", video_stream);
988                 video_stream = 0;
989         }
990
991         
992         /* Close the output file */
993         if (outfile) {
994                 for (i = 0; i < outfile->nb_streams; i++) {
995                         if (&outfile->streams[i]) {
996                                 av_freep(&outfile->streams[i]);
997                         }
998                 }
999         }
1000         /* free the temp buffer */
1001         if (current_frame) {
1002                 delete_picture(current_frame);
1003                 current_frame = 0;
1004         }
1005         if (outfile && outfile->oformat) {
1006                 if (!(outfile->oformat->flags & AVFMT_NOFILE)) {
1007                         url_fclose(OUTFILE_PB);
1008                 }
1009         }
1010         if (outfile) {
1011                 av_free(outfile);
1012                 outfile = 0;
1013         }
1014         if (video_buffer) {
1015                 MEM_freeN(video_buffer);
1016                 video_buffer = 0;
1017         }
1018         if (audio_output_buffer) {
1019                 MEM_freeN(audio_output_buffer);
1020                 audio_output_buffer = 0;
1021         }
1022         if (audio_input_buffer) {
1023                 MEM_freeN(audio_input_buffer);
1024                 audio_input_buffer = 0;
1025         }
1026
1027         if (img_convert_ctx) {
1028                 sws_freeContext(img_convert_ctx);
1029                 img_convert_ctx = 0;
1030         }
1031 }
1032
1033 /* properties */
1034
1035 void ffmpeg_property_del(RenderData *rd, void *type, void *prop_)
1036 {
1037         struct IDProperty *prop = (struct IDProperty *) prop_;
1038         IDProperty * group;
1039         
1040         if (!rd->ffcodecdata.properties) {
1041                 return;
1042         }
1043
1044         group = IDP_GetPropertyFromGroup(
1045                 rd->ffcodecdata.properties, (char*) type);
1046         if (group && prop) {
1047                 IDP_RemFromGroup(group, prop);
1048                 IDP_FreeProperty(prop);
1049                 MEM_freeN(prop);
1050         }
1051 }
1052
1053 IDProperty *ffmpeg_property_add(RenderData *rd, char * type, int opt_index, int parent_index)
1054 {
1055         AVCodecContext c;
1056         const AVOption * o;
1057         const AVOption * parent;
1058         IDProperty * group;
1059         IDProperty * prop;
1060         IDPropertyTemplate val;
1061         int idp_type;
1062         char name[256];
1063         
1064         val.i = 0;
1065
1066         avcodec_get_context_defaults(&c);
1067
1068         o = c.av_class->option + opt_index;
1069         parent = c.av_class->option + parent_index;
1070
1071         if (!rd->ffcodecdata.properties) {
1072                 rd->ffcodecdata.properties 
1073                         = IDP_New(IDP_GROUP, val, "ffmpeg"); 
1074         }
1075
1076         group = IDP_GetPropertyFromGroup(
1077                 rd->ffcodecdata.properties, (char*) type);
1078         
1079         if (!group) {
1080                 group = IDP_New(IDP_GROUP, val, (char*) type); 
1081                 IDP_AddToGroup(rd->ffcodecdata.properties, group);
1082         }
1083
1084         if (parent_index) {
1085                 sprintf(name, "%s:%s", parent->name, o->name);
1086         } else {
1087                 strcpy(name, o->name);
1088         }
1089
1090         fprintf(stderr, "ffmpeg_property_add: %s %d %d %s\n",
1091                 type, parent_index, opt_index, name);
1092
1093         prop = IDP_GetPropertyFromGroup(group, name);
1094         if (prop) {
1095                 return prop;
1096         }
1097
1098         switch (o->type) {
1099         case FF_OPT_TYPE_INT:
1100         case FF_OPT_TYPE_INT64:
1101                 val.i = o->default_val;
1102                 idp_type = IDP_INT;
1103                 break;
1104         case FF_OPT_TYPE_DOUBLE:
1105         case FF_OPT_TYPE_FLOAT:
1106                 val.f = o->default_val;
1107                 idp_type = IDP_FLOAT;
1108                 break;
1109         case FF_OPT_TYPE_STRING:
1110                 val.str = "                                                                               ";
1111                 idp_type = IDP_STRING;
1112                 break;
1113         case FF_OPT_TYPE_CONST:
1114                 val.i = 1;
1115                 idp_type = IDP_INT;
1116                 break;
1117         default:
1118                 return NULL;
1119         }
1120         prop = IDP_New(idp_type, val, name);
1121         IDP_AddToGroup(group, prop);
1122         return prop;
1123 }
1124
1125 /* not all versions of ffmpeg include that, so here we go ... */
1126
1127 static const AVOption *my_av_find_opt(void *v, const char *name, 
1128                                           const char *unit, int mask, int flags){
1129         AVClass *c= *(AVClass**)v; 
1130         const AVOption *o= c->option;
1131
1132         for(;o && o->name; o++){
1133                 if(!strcmp(o->name, name) && 
1134                    (!unit || (o->unit && !strcmp(o->unit, unit))) && 
1135                    (o->flags & mask) == flags )
1136                         return o;
1137         }
1138         return NULL;
1139 }
1140
1141 int ffmpeg_property_add_string(RenderData *rd, const char * type, const char * str)
1142 {
1143         AVCodecContext c;
1144         const AVOption * o = 0;
1145         const AVOption * p = 0;
1146         char name_[128];
1147         char * name;
1148         char * param;
1149         IDProperty * prop;
1150         
1151         avcodec_get_context_defaults(&c);
1152
1153         strncpy(name_, str, sizeof(name_));
1154
1155         name = name_;
1156         while (*name == ' ') name++;
1157
1158         param = strchr(name, ':');
1159
1160         if (!param) {
1161                 param = strchr(name, ' ');
1162         }
1163         if (param) {
1164                 *param++ = 0;
1165                 while (*param == ' ') param++;
1166         }
1167         
1168         o = my_av_find_opt(&c, name, NULL, 0, 0);       
1169         if (!o) {
1170                 return 0;
1171         }
1172         if (param && o->type == FF_OPT_TYPE_CONST) {
1173                 return 0;
1174         }
1175         if (param && o->type != FF_OPT_TYPE_CONST && o->unit) {
1176                 p = my_av_find_opt(&c, param, o->unit, 0, 0);   
1177                 prop = ffmpeg_property_add(rd,
1178                         (char*) type, p - c.av_class->option, 
1179                         o - c.av_class->option);
1180         } else {
1181                 prop = ffmpeg_property_add(rd,
1182                         (char*) type, o - c.av_class->option, 0);
1183         }
1184                 
1185
1186         if (!prop) {
1187                 return 0;
1188         }
1189
1190         if (param && !p) {
1191                 switch (prop->type) {
1192                 case IDP_INT:
1193                         IDP_Int(prop) = atoi(param);
1194                         break;
1195                 case IDP_FLOAT:
1196                         IDP_Float(prop) = atof(param);
1197                         break;
1198                 case IDP_STRING:
1199                         strncpy(IDP_String(prop), param, prop->len);
1200                         break;
1201                 }
1202         }
1203         return 1;
1204 }
1205
1206 void ffmpeg_set_preset(RenderData *rd, int preset)
1207 {
1208         int isntsc = (rd->frs_sec != 25);
1209
1210         switch (preset) {
1211         case FFMPEG_PRESET_VCD:
1212                 rd->ffcodecdata.type = FFMPEG_MPEG1;
1213                 rd->ffcodecdata.video_bitrate = 1150;
1214                 rd->xsch = 352;
1215                 rd->ysch = isntsc ? 240 : 288;
1216                 rd->ffcodecdata.gop_size = isntsc ? 18 : 15;
1217                 rd->ffcodecdata.rc_max_rate = 1150;
1218                 rd->ffcodecdata.rc_min_rate = 1150;
1219                 rd->ffcodecdata.rc_buffer_size = 40*8;
1220                 rd->ffcodecdata.mux_packet_size = 2324;
1221                 rd->ffcodecdata.mux_rate = 2352 * 75 * 8;
1222                 break;
1223
1224         case FFMPEG_PRESET_SVCD:
1225                 rd->ffcodecdata.type = FFMPEG_MPEG2;
1226                 rd->ffcodecdata.video_bitrate = 2040;
1227                 rd->xsch = 480;
1228                 rd->ysch = isntsc ? 480 : 576;
1229                 rd->ffcodecdata.gop_size = isntsc ? 18 : 15;
1230                 rd->ffcodecdata.rc_max_rate = 2516;
1231                 rd->ffcodecdata.rc_min_rate = 0;
1232                 rd->ffcodecdata.rc_buffer_size = 224*8;
1233                 rd->ffcodecdata.mux_packet_size = 2324;
1234                 rd->ffcodecdata.mux_rate = 0;
1235                 break;
1236
1237         case FFMPEG_PRESET_DVD:
1238                 rd->ffcodecdata.type = FFMPEG_MPEG2;
1239                 rd->ffcodecdata.video_bitrate = 6000;
1240                 rd->xsch = 720;
1241                 rd->ysch = isntsc ? 480 : 576;
1242                 rd->ffcodecdata.gop_size = isntsc ? 18 : 15;
1243                 rd->ffcodecdata.rc_max_rate = 9000;
1244                 rd->ffcodecdata.rc_min_rate = 0;
1245                 rd->ffcodecdata.rc_buffer_size = 224*8;
1246                 rd->ffcodecdata.mux_packet_size = 2048;
1247                 rd->ffcodecdata.mux_rate = 10080000;
1248                 break;
1249
1250         case FFMPEG_PRESET_DV:
1251                 rd->ffcodecdata.type = FFMPEG_DV;
1252                 rd->xsch = 720;
1253                 rd->ysch = isntsc ? 480 : 576;
1254                 break;
1255
1256         case FFMPEG_PRESET_H264:
1257                 rd->ffcodecdata.type = FFMPEG_AVI;
1258                 rd->ffcodecdata.codec = CODEC_ID_H264;
1259                 rd->ffcodecdata.video_bitrate = 6000;
1260                 rd->ffcodecdata.gop_size = isntsc ? 18 : 15;
1261                 rd->ffcodecdata.rc_max_rate = 9000;
1262                 rd->ffcodecdata.rc_min_rate = 0;
1263                 rd->ffcodecdata.rc_buffer_size = 224*8;
1264                 rd->ffcodecdata.mux_packet_size = 2048;
1265                 rd->ffcodecdata.mux_rate = 10080000;
1266
1267                 /*
1268                  * All options here are for x264, but must be set via ffmpeg.
1269                  * The names are therefore different - Search for "x264 to FFmpeg option mapping"
1270                  * to get a list.
1271                  */
1272                 
1273                 /*
1274                  * Use CABAC coder. Using "coder:1", which should be equivalent,
1275                  * crashes Blender for some reason. Either way - this is no big deal.
1276                  */
1277                 ffmpeg_property_add_string(rd, "video", "coder:vlc");
1278                 
1279                 /* 
1280                  * The other options were taken from the libx264-default.preset
1281                  * included in the ffmpeg distribution.
1282                  */
1283                 ffmpeg_property_add_string(rd, "video", "flags:loop");
1284                 ffmpeg_property_add_string(rd, "video", "cmp:chroma");
1285                 ffmpeg_property_add_string(rd, "video", "partitions:parti4x4");
1286                 ffmpeg_property_add_string(rd, "video", "partitions:partp8x8");
1287                 ffmpeg_property_add_string(rd, "video", "partitions:partb8x8");
1288                 ffmpeg_property_add_string(rd, "video", "me:hex");
1289                 ffmpeg_property_add_string(rd, "video", "subq:6");
1290                 ffmpeg_property_add_string(rd, "video", "me_range:16");
1291                 ffmpeg_property_add_string(rd, "video", "qdiff:4");
1292                 ffmpeg_property_add_string(rd, "video", "keyint_min:25");
1293                 ffmpeg_property_add_string(rd, "video", "sc_threshold:40");
1294                 ffmpeg_property_add_string(rd, "video", "i_qfactor:0.71");
1295                 ffmpeg_property_add_string(rd, "video", "b_strategy:1");
1296                 ffmpeg_property_add_string(rd, "video", "bf:3");
1297                 ffmpeg_property_add_string(rd, "video", "refs:2");
1298                 ffmpeg_property_add_string(rd, "video", "qcomp:0.6");
1299                 ffmpeg_property_add_string(rd, "video", "directpred:3");
1300                 ffmpeg_property_add_string(rd, "video", "trellis:0");
1301                 ffmpeg_property_add_string(rd, "video", "flags2:wpred");
1302                 ffmpeg_property_add_string(rd, "video", "flags2:dct8x8");
1303                 ffmpeg_property_add_string(rd, "video", "flags2:fastpskip");
1304                 ffmpeg_property_add_string(rd, "video", "wpredp:2");
1305                 
1306                 // This makes x264 output lossless. Will be a separate option later.
1307                 //ffmpeg_property_add_string(rd, "video", "cqp:0");
1308                 break;
1309
1310         case FFMPEG_PRESET_THEORA:
1311         case FFMPEG_PRESET_XVID:
1312                 if(preset == FFMPEG_PRESET_XVID) {
1313                         rd->ffcodecdata.type = FFMPEG_AVI;
1314                         rd->ffcodecdata.codec = CODEC_ID_XVID;
1315                 }
1316                 else if(preset == FFMPEG_PRESET_THEORA) {
1317                         rd->ffcodecdata.type = FFMPEG_OGG; // XXX broken
1318                         rd->ffcodecdata.codec = CODEC_ID_THEORA;
1319                 }
1320
1321                 rd->ffcodecdata.video_bitrate = 6000;
1322                 rd->ffcodecdata.gop_size = isntsc ? 18 : 15;
1323                 rd->ffcodecdata.rc_max_rate = 9000;
1324                 rd->ffcodecdata.rc_min_rate = 0;
1325                 rd->ffcodecdata.rc_buffer_size = 224*8;
1326                 rd->ffcodecdata.mux_packet_size = 2048;
1327                 rd->ffcodecdata.mux_rate = 10080000;
1328                 break;
1329
1330         }
1331 }
1332
1333 void ffmpeg_verify_image_type(RenderData *rd)
1334 {
1335         int audio= 0;
1336
1337         if(rd->imtype == R_FFMPEG) {
1338                 if(rd->ffcodecdata.type <= 0 ||
1339                    rd->ffcodecdata.codec <= 0 ||
1340                    rd->ffcodecdata.audio_codec <= 0 ||
1341                    rd->ffcodecdata.video_bitrate <= 1) {
1342
1343                         rd->ffcodecdata.codec = CODEC_ID_MPEG2VIDEO;
1344                         /* Don't set preset, disturbs render resolution.
1345                          * ffmpeg_set_preset(rd, FFMPEG_PRESET_DVD); */
1346                 }
1347
1348                 audio= 1;
1349         }
1350         else if(rd->imtype == R_H264) {
1351                 if(rd->ffcodecdata.codec != CODEC_ID_H264) {
1352                         ffmpeg_set_preset(rd, FFMPEG_PRESET_H264);
1353                         audio= 1;
1354                 }
1355         }
1356         else if(rd->imtype == R_XVID) {
1357                 if(rd->ffcodecdata.codec != CODEC_ID_XVID) {
1358                         ffmpeg_set_preset(rd, FFMPEG_PRESET_XVID);
1359                         audio= 1;
1360                 }
1361         }
1362         else if(rd->imtype == R_THEORA) {
1363                 if(rd->ffcodecdata.codec != CODEC_ID_THEORA) {
1364                         ffmpeg_set_preset(rd, FFMPEG_PRESET_THEORA);
1365                         audio= 1;
1366                 }
1367         }
1368
1369         if(audio && rd->ffcodecdata.audio_codec < 0) {
1370                 rd->ffcodecdata.audio_codec = CODEC_ID_NONE;
1371                 rd->ffcodecdata.audio_bitrate = 128;
1372         }
1373 }
1374
1375 #endif
1376