merging game_engine branch changes into trunk, 2d-filters and opengl speedup
[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/BL DUAL 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. The Blender
13  * Foundation also sells licenses for use in proprietary software under
14  * the Blender License.  See http://www.blender.org/BL/ for information
15  * about this.
16  *
17  * This program is distributed in the hope that it will be useful,
18  * but WITHOUT ANY WARRANTY; without even the implied warranty of
19  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
20  * GNU General Public License for more details.
21  *
22  * You should have received a copy of the GNU General Public License
23  * along with this program; if not, write to the Free Software Foundation,
24  * Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
25  *
26  * The Original Code is Copyright (C) 2001-2002 by NaN Holding BV.
27  * All rights reserved.
28  *
29  * The Original Code is: all of this file.
30  *
31  * Contributor(s): Robert Wenzlaff
32  *
33  * ***** END GPL/BL DUAL LICENSE BLOCK *****
34  * 
35  */
36
37 #include <string.h>
38
39 #include "MEM_guardedalloc.h"
40
41 #include "DNA_scene_types.h"
42
43 #include "BLI_blenlib.h"
44
45 #include "BKE_global.h"
46 #include "BKE_writeavi.h"
47 #include "AVI_avi.h"
48
49
50 /* ********************** general blender movie support ***************************** */
51
52 #ifdef WITH_QUICKTIME
53 #include "quicktime_export.h"
54 #endif
55
56 #if defined(_WIN32) && !defined(FREE_WINDOWS)
57 #include "BIF_writeavicodec.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 = 0;
75         
76         /* do the platform specific handles */
77 #ifdef __sgi
78         if (imtype == R_MOVIE) {
79                 
80         }
81 #endif
82 #if defined(_WIN32) && !defined(FREE_WINDOWS)
83         if (imtype == R_AVICODEC) {             
84                 mh.start_movie= start_avi_codec;
85                 mh.append_movie= append_avi_codec;
86                 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         }
95 #endif
96 #ifdef WITH_FFMPEG
97         if (imtype == R_FFMPEG) {
98                 mh.start_movie = start_ffmpeg;
99                 mh.append_movie = append_ffmpeg;
100                 mh.end_movie = end_ffmpeg;
101         }
102 #endif
103         if (imtype == R_FRAMESERVER) {
104                 mh.start_movie = start_frameserver;
105                 mh.append_movie = append_frameserver;
106                 mh.end_movie = end_frameserver;
107                 mh.get_next_frame = frameserver_loop;
108         }
109         
110         return &mh;
111 }
112
113 /* ****************************************************************** */
114
115
116 static AviMovie *avi=NULL;
117 static int sframe;
118
119 void makeavistring (RenderData *rd, char *string) 
120 {
121         char txt[64];
122
123         if (string==0) return;
124
125         strcpy(string, rd->pic);
126         BLI_convertstringcode(string, G.sce, rd->cfra);
127
128         BLI_make_existing_file(string);
129
130         if (BLI_strcasecmp(string + strlen(string) - 4, ".avi")) {
131                 sprintf(txt, "%04d_%04d.avi", (rd->sfra) , (rd->efra) );
132                 strcat(string, txt);
133         }
134 }
135
136 void start_avi(RenderData *rd, int rectx, int recty)
137 {
138         int x, y;
139         char name[256];
140         AviFormat format;
141         int quality;
142         double framerate;
143         
144         makeavistring(rd, name);
145
146         sframe = (rd->sfra);
147         x = rectx;
148         y = recty;
149
150         quality= rd->quality;
151         framerate= (double) rd->frs_sec / (double) rd->frs_sec_base;
152         
153         avi = MEM_mallocN (sizeof(AviMovie), "avimovie");
154
155         /* RPW 11-21-2002 
156          if (rd->imtype != AVI_FORMAT_MJPEG) format = AVI_FORMAT_AVI_RGB;
157         */
158         if (rd->imtype != R_AVIJPEG ) format = AVI_FORMAT_AVI_RGB;
159         else format = AVI_FORMAT_MJPEG;
160
161         if (AVI_open_compress (name, avi, 1, format) != AVI_ERROR_NONE) {
162                 printf("cannot open or start AVI movie file");
163                 MEM_freeN (avi);
164                 avi = NULL;
165                 return;
166         }
167                         
168         AVI_set_compress_option (avi, AVI_OPTION_TYPE_MAIN, 0, AVI_OPTION_WIDTH, &x);
169         AVI_set_compress_option (avi, AVI_OPTION_TYPE_MAIN, 0, AVI_OPTION_HEIGHT, &y);
170         AVI_set_compress_option (avi, AVI_OPTION_TYPE_MAIN, 0, AVI_OPTION_QUALITY, &quality);           
171         AVI_set_compress_option (avi, AVI_OPTION_TYPE_MAIN, 0, AVI_OPTION_FRAMERATE, &framerate);
172
173         avi->interlace= 0;
174         avi->odd_fields= 0;
175 /*      avi->interlace= rd->mode & R_FIELDS; */
176 /*      avi->odd_fields= (rd->mode & R_ODDFIELD)?1:0; */
177         
178         printf("Created avi: %s\n", name);
179 }
180
181 void append_avi(int frame, int *pixels, int rectx, int recty)
182 {
183         unsigned int *rt1, *rt2, *rectot;
184         int x, y;
185         char *cp, rt;
186         
187         if (avi == NULL) {
188                 G.afbreek = 1;
189                 return;
190         }
191
192         /* note that libavi free's the buffer... stupid interface - zr */
193         rectot= MEM_mallocN(rectx*recty*sizeof(int), "rectot");
194         rt1= rectot;
195         rt2= (unsigned int*)pixels + (recty-1)*rectx;
196         /* flip y and convert to abgr */
197         for (y=0; y < recty; y++, rt1+= rectx, rt2-= rectx) {
198                 memcpy (rt1, rt2, rectx*sizeof(int));
199                 
200                 cp= (char *)rt1;
201                 for(x= rectx; x>0; x--) {
202                         rt= cp[0];
203                         cp[0]= cp[3];
204                         cp[3]= rt;
205                         rt= cp[1];
206                         cp[1]= cp[2];
207                         cp[2]= rt;
208                         cp+= 4;
209                 }
210         }
211         
212         AVI_write_frame (avi, (frame-sframe), AVI_FORMAT_RGB32, rectot, rectx*recty*4);
213 //      printf ("added frame %3d (frame %3d in avi): ", frame, frame-sframe);
214 }
215
216 void end_avi(void)
217 {
218         if (avi == NULL) return;
219
220         AVI_close_compress (avi);
221         MEM_freeN (avi);
222         avi= NULL;
223 }