svn merge -r 37306:39975 https://svn.blender.org/svnroot/bf-blender/trunk/blender
[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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, 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 /** \file blender/blenkernel/intern/writeavi.c
35  *  \ingroup bke
36  */
37
38
39 #include <string.h>
40
41 #include "MEM_guardedalloc.h"
42
43 #include "DNA_scene_types.h"
44
45 #include "BLI_blenlib.h"
46 #include "BLI_utildefines.h"
47
48 #include "BKE_global.h"
49 #include "BKE_main.h"
50 #include "BKE_report.h"
51
52 #include "BKE_writeavi.h"
53 #include "AVI_avi.h"
54
55 /* callbacks */
56 static int start_avi(Scene *scene, RenderData *rd, int rectx, int recty, ReportList *reports);
57 static void end_avi(void);
58 static int append_avi(RenderData *rd, int frame, int *pixels, int rectx, int recty, ReportList *reports);
59 static void filepath_avi(char *string, RenderData *rd);
60
61 /* ********************** general blender movie support ***************************** */
62
63 #ifdef WITH_QUICKTIME
64 #include "quicktime_export.h"
65 #endif
66
67 #ifdef WITH_FFMPEG
68 #include "BKE_writeffmpeg.h"
69 #endif
70
71 #include "BKE_writeframeserver.h"
72
73 bMovieHandle *BKE_get_movie_handle(int imtype)
74 {
75         static bMovieHandle mh;
76         
77         /* set the default handle, as builtin */
78         mh.start_movie= start_avi;
79         mh.append_movie= append_avi;
80         mh.end_movie= end_avi;
81         mh.get_next_frame = NULL;
82         mh.get_movie_path = filepath_avi;
83         
84         /* do the platform specific handles */
85 #if defined(_WIN32) && !defined(FREE_WINDOWS)
86         if (imtype == R_AVICODEC) {             
87                 //XXX mh.start_movie= start_avi_codec;
88                 //XXX mh.append_movie= append_avi_codec;
89                 //XXX mh.end_movie= end_avi_codec;
90         }
91 #endif
92 #ifdef WITH_QUICKTIME
93         if (imtype == R_QUICKTIME) {
94                 mh.start_movie= start_qt;
95                 mh.append_movie= append_qt;
96                 mh.end_movie= end_qt;
97                 mh.get_movie_path = filepath_qt;
98         }
99 #endif
100 #ifdef WITH_FFMPEG
101         if (ELEM4(imtype, R_FFMPEG, R_H264, R_XVID, R_THEORA)) {
102                 mh.start_movie = start_ffmpeg;
103                 mh.append_movie = append_ffmpeg;
104                 mh.end_movie = end_ffmpeg;
105                 mh.get_movie_path = filepath_ffmpeg;
106         }
107 #endif
108 #ifdef WITH_FRAMESERVER
109         if (imtype == R_FRAMESERVER) {
110                 mh.start_movie = start_frameserver;
111                 mh.append_movie = append_frameserver;
112                 mh.end_movie = end_frameserver;
113                 mh.get_next_frame = frameserver_loop;
114         }
115 #endif
116
117         /* incase all above are disabled */
118         (void)imtype;
119
120         return &mh;
121 }
122
123 /* ****************************************************************** */
124
125
126 static AviMovie *avi=NULL;
127 static int sframe;
128
129 static void filepath_avi (char *string, RenderData *rd)
130 {
131         if (string==NULL) return;
132
133         strcpy(string, rd->pic);
134         BLI_path_abs(string, G.main->name);
135
136         BLI_make_existing_file(string);
137
138         if (!BLI_testextensie(string, ".avi")) {
139                 BLI_path_frame_range(string, rd->sfra, rd->efra, 4);
140                 strcat(string, ".avi");
141         }
142 }
143
144 static int start_avi(Scene *scene, RenderData *rd, int rectx, int recty, ReportList *reports)
145 {
146         int x, y;
147         char name[256];
148         AviFormat format;
149         int quality;
150         double framerate;
151         
152         (void)scene; /* unused */
153         
154         filepath_avi(name, rd);
155
156         sframe = (rd->sfra);
157         x = rectx;
158         y = recty;
159
160         quality= rd->quality;
161         framerate= (double) rd->frs_sec / (double) rd->frs_sec_base;
162         
163         avi = MEM_mallocN (sizeof(AviMovie), "avimovie");
164
165         /* RPW 11-21-2002 
166         if (rd->imtype != AVI_FORMAT_MJPEG) format = AVI_FORMAT_AVI_RGB;
167         */
168         if (rd->imtype != R_AVIJPEG ) format = AVI_FORMAT_AVI_RGB;
169         else format = AVI_FORMAT_MJPEG;
170
171         if (AVI_open_compress (name, avi, 1, format) != AVI_ERROR_NONE) {
172                 BKE_report(reports, RPT_ERROR, "Cannot open or start AVI movie file.");
173                 MEM_freeN (avi);
174                 avi = NULL;
175                 return 0;
176         }
177                         
178         AVI_set_compress_option (avi, AVI_OPTION_TYPE_MAIN, 0, AVI_OPTION_WIDTH, &x);
179         AVI_set_compress_option (avi, AVI_OPTION_TYPE_MAIN, 0, AVI_OPTION_HEIGHT, &y);
180         AVI_set_compress_option (avi, AVI_OPTION_TYPE_MAIN, 0, AVI_OPTION_QUALITY, &quality);           
181         AVI_set_compress_option (avi, AVI_OPTION_TYPE_MAIN, 0, AVI_OPTION_FRAMERATE, &framerate);
182
183         avi->interlace= 0;
184         avi->odd_fields= 0;
185 /*      avi->interlace= rd->mode & R_FIELDS; */
186 /*      avi->odd_fields= (rd->mode & R_ODDFIELD)?1:0; */
187         
188         printf("Created avi: %s\n", name);
189         return 1;
190 }
191
192 static int append_avi(RenderData *UNUSED(rd), int frame, int *pixels, int rectx, int recty, ReportList *UNUSED(reports))
193 {
194         unsigned int *rt1, *rt2, *rectot;
195         int x, y;
196         char *cp, rt;
197         
198         if (avi == NULL)
199                 return 0;
200
201         /* note that libavi free's the buffer... stupid interface - zr */
202         rectot= MEM_mallocN(rectx*recty*sizeof(int), "rectot");
203         rt1= rectot;
204         rt2= (unsigned int*)pixels + (recty-1)*rectx;
205         /* flip y and convert to abgr */
206         for (y=0; y < recty; y++, rt1+= rectx, rt2-= rectx) {
207                 memcpy (rt1, rt2, rectx*sizeof(int));
208                 
209                 cp= (char *)rt1;
210                 for(x= rectx; x>0; x--) {
211                         rt= cp[0];
212                         cp[0]= cp[3];
213                         cp[3]= rt;
214                         rt= cp[1];
215                         cp[1]= cp[2];
216                         cp[2]= rt;
217                         cp+= 4;
218                 }
219         }
220         
221         AVI_write_frame (avi, (frame-sframe), AVI_FORMAT_RGB32, rectot, rectx*recty*4);
222 //      printf ("added frame %3d (frame %3d in avi): ", frame, frame-sframe);
223
224         return 1;
225 }
226
227 static void end_avi(void)
228 {
229         if (avi == NULL) return;
230
231         AVI_close_compress (avi);
232         MEM_freeN (avi);
233         avi= NULL;
234 }
235
236 /* similar to BKE_makepicstring() */
237 void BKE_makeanimstring(char *string, RenderData *rd)
238 {
239         bMovieHandle *mh= BKE_get_movie_handle(rd->imtype);
240         if(mh->get_movie_path)
241                 mh->get_movie_path(string, rd);
242         else
243                 string[0]= '\0';
244 }