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