2.5: merge with trunk, previous merge was only up to yesterday.
[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_writeavi.h"
44 #include "AVI_avi.h"
45
46
47 /* ********************** general blender movie support ***************************** */
48
49 #ifdef WITH_QUICKTIME
50 #include "quicktime_export.h"
51 #endif
52
53 #ifdef WITH_FFMPEG
54 #include "BKE_writeffmpeg.h"
55 #endif
56
57 #include "BKE_writeframeserver.h"
58
59 bMovieHandle *BKE_get_movie_handle(int imtype)
60 {
61         static bMovieHandle mh;
62         
63         /* set the default handle, as builtin */
64         mh.start_movie= start_avi;
65         mh.append_movie= append_avi;
66         mh.end_movie= end_avi;
67         mh.get_next_frame = 0;
68         
69         /* do the platform specific handles */
70 #ifdef __sgi
71         if (imtype == R_MOVIE) {
72                 
73         }
74 #endif
75 #if defined(_WIN32) && !defined(FREE_WINDOWS)
76         if (imtype == R_AVICODEC) {             
77                 //XXX mh.start_movie= start_avi_codec;
78                 //XXX mh.append_movie= append_avi_codec;
79                 //XXX mh.end_movie= end_avi_codec;
80         }
81 #endif
82 #ifdef WITH_QUICKTIME
83         if (imtype == R_QUICKTIME) {
84                 mh.start_movie= start_qt;
85                 mh.append_movie= append_qt;
86                 mh.end_movie= end_qt;
87         }
88 #endif
89 #ifdef WITH_FFMPEG
90         if (imtype == R_FFMPEG) {
91                 mh.start_movie = start_ffmpeg;
92                 mh.append_movie = append_ffmpeg;
93                 mh.end_movie = end_ffmpeg;
94         }
95 #endif
96         if (imtype == R_FRAMESERVER) {
97                 mh.start_movie = start_frameserver;
98                 mh.append_movie = append_frameserver;
99                 mh.end_movie = end_frameserver;
100                 mh.get_next_frame = frameserver_loop;
101         }
102         
103         return &mh;
104 }
105
106 /* ****************************************************************** */
107
108
109 static AviMovie *avi=NULL;
110 static int sframe;
111
112 void makeavistring (RenderData *rd, char *string) 
113 {
114         char txt[64];
115
116         if (string==0) return;
117
118         strcpy(string, rd->pic);
119         BLI_convertstringcode(string, G.sce);
120
121         BLI_make_existing_file(string);
122
123         if (BLI_strcasecmp(string + strlen(string) - 4, ".avi")) {
124                 sprintf(txt, "%04d_%04d.avi", (rd->sfra) , (rd->efra) );
125                 strcat(string, txt);
126         }
127 }
128
129 void start_avi(RenderData *rd, int rectx, int recty)
130 {
131         int x, y;
132         char name[256];
133         AviFormat format;
134         int quality;
135         double framerate;
136         
137         makeavistring(rd, name);
138
139         sframe = (rd->sfra);
140         x = rectx;
141         y = recty;
142
143         quality= rd->quality;
144         framerate= (double) rd->frs_sec / (double) rd->frs_sec_base;
145         
146         avi = MEM_mallocN (sizeof(AviMovie), "avimovie");
147
148         /* RPW 11-21-2002 
149          if (rd->imtype != AVI_FORMAT_MJPEG) format = AVI_FORMAT_AVI_RGB;
150         */
151         if (rd->imtype != R_AVIJPEG ) format = AVI_FORMAT_AVI_RGB;
152         else format = AVI_FORMAT_MJPEG;
153
154         if (AVI_open_compress (name, avi, 1, format) != AVI_ERROR_NONE) {
155                 printf("cannot open or start AVI movie file");
156                 MEM_freeN (avi);
157                 avi = NULL;
158                 return;
159         }
160                         
161         AVI_set_compress_option (avi, AVI_OPTION_TYPE_MAIN, 0, AVI_OPTION_WIDTH, &x);
162         AVI_set_compress_option (avi, AVI_OPTION_TYPE_MAIN, 0, AVI_OPTION_HEIGHT, &y);
163         AVI_set_compress_option (avi, AVI_OPTION_TYPE_MAIN, 0, AVI_OPTION_QUALITY, &quality);           
164         AVI_set_compress_option (avi, AVI_OPTION_TYPE_MAIN, 0, AVI_OPTION_FRAMERATE, &framerate);
165
166         avi->interlace= 0;
167         avi->odd_fields= 0;
168 /*      avi->interlace= rd->mode & R_FIELDS; */
169 /*      avi->odd_fields= (rd->mode & R_ODDFIELD)?1:0; */
170         
171         printf("Created avi: %s\n", name);
172 }
173
174 void append_avi(int frame, int *pixels, int rectx, int recty)
175 {
176         unsigned int *rt1, *rt2, *rectot;
177         int x, y;
178         char *cp, rt;
179         
180         if (avi == NULL) {
181                 G.afbreek = 1;
182                 return;
183         }
184
185         /* note that libavi free's the buffer... stupid interface - zr */
186         rectot= MEM_mallocN(rectx*recty*sizeof(int), "rectot");
187         rt1= rectot;
188         rt2= (unsigned int*)pixels + (recty-1)*rectx;
189         /* flip y and convert to abgr */
190         for (y=0; y < recty; y++, rt1+= rectx, rt2-= rectx) {
191                 memcpy (rt1, rt2, rectx*sizeof(int));
192                 
193                 cp= (char *)rt1;
194                 for(x= rectx; x>0; x--) {
195                         rt= cp[0];
196                         cp[0]= cp[3];
197                         cp[3]= rt;
198                         rt= cp[1];
199                         cp[1]= cp[2];
200                         cp[2]= rt;
201                         cp+= 4;
202                 }
203         }
204         
205         AVI_write_frame (avi, (frame-sframe), AVI_FORMAT_RGB32, rectot, rectx*recty*4);
206 //      printf ("added frame %3d (frame %3d in avi): ", frame, frame-sframe);
207 }
208
209 void end_avi(void)
210 {
211         if (avi == NULL) return;
212
213         AVI_close_compress (avi);
214         MEM_freeN (avi);
215         avi= NULL;
216 }