- change max threads from 8 to 64, need to keep an eye on stack memory use here.
[blender.git] / source / blender / blenkernel / intern / writeavi.c
1 /**
2  * Functions for writing avi-format files.
3  * Added interface for generic movie support (ton)
4  *
5  * $Id$
6  *
7  * ***** BEGIN GPL LICENSE BLOCK *****
8  *
9  * This program is free software; you can redistribute it and/or
10  * modify it under the terms of the GNU General Public License
11  * as published by the Free Software Foundation; either version 2
12  * of the License, or (at your option) any later version.
13  *
14  * This program is distributed in the hope that it will be useful,
15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
17  * GNU General Public License for more details.
18  *
19  * You should have received a copy of the GNU General Public License
20  * along with this program; if not, write to the Free Software Foundation,
21  * Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
22  *
23  * The Original Code is Copyright (C) 2001-2002 by NaN Holding BV.
24  * All rights reserved.
25  *
26  * The Original Code is: all of this file.
27  *
28  * Contributor(s): Robert Wenzlaff
29  *
30  * ***** END GPL LICENSE BLOCK *****
31  * 
32  */
33
34 #include <string.h>
35
36 #include "MEM_guardedalloc.h"
37
38 #include "DNA_scene_types.h"
39
40 #include "BLI_blenlib.h"
41
42 #include "BKE_global.h"
43 #include "BKE_report.h"
44 #include "BKE_utildefines.h"
45 #include "BKE_writeavi.h"
46 #include "AVI_avi.h"
47
48 /* callbacks */
49 static int start_avi(Scene *scene, RenderData *rd, int rectx, int recty, ReportList *reports);
50 static void end_avi(void);
51 static int append_avi(RenderData *rd, int frame, int *pixels, int rectx, int recty, ReportList *reports);
52 static void filepath_avi(char *string, RenderData *rd);
53
54 /* ********************** general blender movie support ***************************** */
55
56 #ifdef WITH_QUICKTIME
57 #include "quicktime_export.h"
58 #endif
59
60 #ifdef WITH_FFMPEG
61 #include "BKE_writeffmpeg.h"
62 #endif
63
64 #include "BKE_writeframeserver.h"
65
66 bMovieHandle *BKE_get_movie_handle(int imtype)
67 {
68         static bMovieHandle mh;
69         
70         /* set the default handle, as builtin */
71         mh.start_movie= start_avi;
72         mh.append_movie= append_avi;
73         mh.end_movie= end_avi;
74         mh.get_next_frame = NULL;
75         mh.get_movie_path = filepath_avi;
76         
77         /* do the platform specific handles */
78 #ifdef __sgi
79         if (imtype == R_MOVIE) {
80                 
81         }
82 #endif
83 #if defined(_WIN32) && !defined(FREE_WINDOWS)
84         if (imtype == R_AVICODEC) {             
85                 //XXX mh.start_movie= start_avi_codec;
86                 //XXX mh.append_movie= append_avi_codec;
87                 //XXX mh.end_movie= end_avi_codec;
88         }
89 #endif
90 #ifdef WITH_QUICKTIME
91         if (imtype == R_QUICKTIME) {
92                 mh.start_movie= start_qt;
93                 mh.append_movie= append_qt;
94                 mh.end_movie= end_qt;
95                 mh.get_movie_path = filepath_qt;
96         }
97 #endif
98 #ifdef WITH_FFMPEG
99         if (ELEM4(imtype, R_FFMPEG, R_H264, R_XVID, R_THEORA)) {
100                 mh.start_movie = start_ffmpeg;
101                 mh.append_movie = append_ffmpeg;
102                 mh.end_movie = end_ffmpeg;
103                 mh.get_movie_path = filepath_ffmpeg;
104         }
105 #endif
106         if (imtype == R_FRAMESERVER) {
107                 mh.start_movie = start_frameserver;
108                 mh.append_movie = append_frameserver;
109                 mh.end_movie = end_frameserver;
110                 mh.get_next_frame = frameserver_loop;
111         }
112         
113         return &mh;
114 }
115
116 /* ****************************************************************** */
117
118
119 static AviMovie *avi=NULL;
120 static int sframe;
121
122 static void filepath_avi (char *string, RenderData *rd)
123 {
124         if (string==NULL) return;
125
126         strcpy(string, rd->pic);
127         BLI_convertstringcode(string, G.sce);
128
129         BLI_make_existing_file(string);
130
131         if (!BLI_testextensie(string, ".avi")) {
132                 /* if we dont have any #'s to insert numbers into, use 4 numbers by default */
133                 if (strchr(string, '#')==NULL)
134                         strcat(string, "####"); /* 4 numbers */
135
136                 BLI_convertstringframe_range(string, rd->sfra, rd->efra);
137                 strcat(string, ".avi");
138         }
139 }
140
141 static int start_avi(Scene *scene, RenderData *rd, int rectx, int recty, ReportList *reports)
142 {
143         int x, y;
144         char name[256];
145         AviFormat format;
146         int quality;
147         double framerate;
148         
149         filepath_avi(name, rd);
150
151         sframe = (rd->sfra);
152         x = rectx;
153         y = recty;
154
155         quality= rd->quality;
156         framerate= (double) rd->frs_sec / (double) rd->frs_sec_base;
157         
158         avi = MEM_mallocN (sizeof(AviMovie), "avimovie");
159
160         /* RPW 11-21-2002 
161          if (rd->imtype != AVI_FORMAT_MJPEG) format = AVI_FORMAT_AVI_RGB;
162         */
163         if (rd->imtype != R_AVIJPEG ) format = AVI_FORMAT_AVI_RGB;
164         else format = AVI_FORMAT_MJPEG;
165
166         if (AVI_open_compress (name, avi, 1, format) != AVI_ERROR_NONE) {
167                 BKE_report(reports, RPT_ERROR, "Cannot open or start AVI movie file.");
168                 MEM_freeN (avi);
169                 avi = NULL;
170                 return 0;
171         }
172                         
173         AVI_set_compress_option (avi, AVI_OPTION_TYPE_MAIN, 0, AVI_OPTION_WIDTH, &x);
174         AVI_set_compress_option (avi, AVI_OPTION_TYPE_MAIN, 0, AVI_OPTION_HEIGHT, &y);
175         AVI_set_compress_option (avi, AVI_OPTION_TYPE_MAIN, 0, AVI_OPTION_QUALITY, &quality);           
176         AVI_set_compress_option (avi, AVI_OPTION_TYPE_MAIN, 0, AVI_OPTION_FRAMERATE, &framerate);
177
178         avi->interlace= 0;
179         avi->odd_fields= 0;
180 /*      avi->interlace= rd->mode & R_FIELDS; */
181 /*      avi->odd_fields= (rd->mode & R_ODDFIELD)?1:0; */
182         
183         printf("Created avi: %s\n", name);
184         return 1;
185 }
186
187 static int append_avi(RenderData *rd, int frame, int *pixels, int rectx, int recty, ReportList *reports)
188 {
189         unsigned int *rt1, *rt2, *rectot;
190         int x, y;
191         char *cp, rt;
192         
193         if (avi == NULL)
194                 return 0;
195
196         /* note that libavi free's the buffer... stupid interface - zr */
197         rectot= MEM_mallocN(rectx*recty*sizeof(int), "rectot");
198         rt1= rectot;
199         rt2= (unsigned int*)pixels + (recty-1)*rectx;
200         /* flip y and convert to abgr */
201         for (y=0; y < recty; y++, rt1+= rectx, rt2-= rectx) {
202                 memcpy (rt1, rt2, rectx*sizeof(int));
203                 
204                 cp= (char *)rt1;
205                 for(x= rectx; x>0; x--) {
206                         rt= cp[0];
207                         cp[0]= cp[3];
208                         cp[3]= rt;
209                         rt= cp[1];
210                         cp[1]= cp[2];
211                         cp[2]= rt;
212                         cp+= 4;
213                 }
214         }
215         
216         AVI_write_frame (avi, (frame-sframe), AVI_FORMAT_RGB32, rectot, rectx*recty*4);
217 //      printf ("added frame %3d (frame %3d in avi): ", frame, frame-sframe);
218
219         return 1;
220 }
221
222 void end_avi(void)
223 {
224         if (avi == NULL) return;
225
226         AVI_close_compress (avi);
227         MEM_freeN (avi);
228         avi= NULL;
229 }
230
231 /* similar to BKE_makepicstring() */
232 void BKE_makeanimstring(char *string, RenderData *rd)
233 {
234         bMovieHandle *mh= BKE_get_movie_handle(rd->imtype);
235         if(mh->get_movie_path)
236                 mh->get_movie_path(string, rd);
237         else
238                 string[0]= '\0';
239 }