Merge branch 'master' into blender2.8
[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
51 /* ********************** general blender movie support ***************************** */
52
53 static int start_stub(void *UNUSED(context_v), Scene *UNUSED(scene), RenderData *UNUSED(rd), int UNUSED(rectx), int UNUSED(recty),
54                       ReportList *UNUSED(reports), bool UNUSED(preview), const char *UNUSED(suffix))
55 { return 0; }
56
57 static void end_stub(void *UNUSED(context_v))
58 {}
59
60 static int append_stub(void *UNUSED(context_v), RenderData *UNUSED(rd), int UNUSED(start_frame), int UNUSED(frame), int *UNUSED(pixels),
61                        int UNUSED(rectx), int UNUSED(recty), const char *UNUSED(suffix), ReportList *UNUSED(reports))
62 { return 0; }
63
64 static void *context_create_stub(void)
65 { return NULL; }
66
67 static void context_free_stub(void *UNUSED(context_v))
68 {}
69
70 #ifdef WITH_AVI
71 #  include "AVI_avi.h"
72
73 /* callbacks */
74 static int start_avi(void *context_v, Scene *scene, RenderData *rd, int rectx, int recty, ReportList *reports, bool preview, const char *suffix);
75 static void end_avi(void *context_v);
76 static int append_avi(void *context_v, RenderData *rd, int start_frame, int frame, int *pixels,
77                       int rectx, int recty, const char *suffix, ReportList *reports);
78 static void filepath_avi(char *string, RenderData *rd, bool preview, const char *suffix);
79 static void *context_create_avi(void);
80 static void context_free_avi(void *context_v);
81 #endif  /* WITH_AVI */
82
83 #ifdef WITH_QUICKTIME
84 #  include "quicktime_export.h"
85 #endif
86
87 #ifdef WITH_FFMPEG
88 #  include "BKE_writeffmpeg.h"
89 #endif
90
91 #ifdef WITH_FRAMESERVER
92 #  include "BKE_writeframeserver.h"
93 #endif
94
95 bMovieHandle *BKE_movie_handle_get(const char imtype)
96 {
97         static bMovieHandle mh = {NULL};
98         /* stub callbacks in case none of the movie formats is supported */
99         mh.start_movie = start_stub;
100         mh.append_movie = append_stub;
101         mh.end_movie = end_stub;
102         mh.get_next_frame = NULL;
103         mh.get_movie_path = NULL;
104         mh.context_create = context_create_stub;
105         mh.context_free = context_free_stub;
106
107         /* set the default handle, as builtin */
108 #ifdef WITH_AVI
109         mh.start_movie = start_avi;
110         mh.append_movie = append_avi;
111         mh.end_movie = end_avi;
112         mh.get_movie_path = filepath_avi;
113         mh.context_create = context_create_avi;
114         mh.context_free = context_free_avi;
115 #endif
116
117         /* do the platform specific handles */
118 #ifdef WITH_QUICKTIME
119         if (imtype == R_IMF_IMTYPE_QUICKTIME) {
120                 mh.start_movie = start_qt;
121                 mh.append_movie = append_qt;
122                 mh.end_movie = end_qt;
123                 mh.get_movie_path = filepath_qt;
124                 mh.context_create = context_create_qt;
125                 mh.context_free = context_free_qt;
126         }
127 #endif
128 #ifdef WITH_FFMPEG
129         if (ELEM(imtype, R_IMF_IMTYPE_FFMPEG, R_IMF_IMTYPE_H264, R_IMF_IMTYPE_XVID, R_IMF_IMTYPE_THEORA)) {
130                 mh.start_movie = BKE_ffmpeg_start;
131                 mh.append_movie = BKE_ffmpeg_append;
132                 mh.end_movie = BKE_ffmpeg_end;
133                 mh.get_movie_path = BKE_ffmpeg_filepath_get;
134                 mh.context_create = BKE_ffmpeg_context_create;
135                 mh.context_free = BKE_ffmpeg_context_free;
136         }
137 #endif
138 #ifdef WITH_FRAMESERVER
139         if (imtype == R_IMF_IMTYPE_FRAMESERVER) {
140                 mh.start_movie = BKE_frameserver_start;
141                 mh.append_movie = BKE_frameserver_append;
142                 mh.end_movie = BKE_frameserver_end;
143                 mh.get_next_frame = BKE_frameserver_loop;
144                 mh.context_create = BKE_frameserver_context_create;
145                 mh.context_free = BKE_frameserver_context_free;
146         }
147 #endif
148
149         /* in case all above are disabled */
150         (void)imtype;
151
152         return (mh.append_movie != append_stub) ? &mh : NULL;
153 }
154
155 /* ****************************************************************** */
156
157
158 #ifdef WITH_AVI
159
160 static void filepath_avi(char *string, RenderData *rd, bool preview, const char *suffix)
161 {
162         int sfra, efra;
163
164         if (string == NULL) return;
165
166         if (preview) {
167                 sfra = rd->psfra;
168                 efra = rd->pefra;
169         }
170         else {
171                 sfra = rd->sfra;
172                 efra = rd->efra;
173         }
174
175         strcpy(string, rd->pic);
176         BLI_path_abs(string, G.main->name);
177
178         BLI_make_existing_file(string);
179
180         if (rd->scemode & R_EXTENSION) {
181                 if (!BLI_testextensie(string, ".avi")) {
182                         BLI_path_frame_range(string, sfra, efra, 4);
183                         strcat(string, ".avi");
184                 }
185         }
186         else {
187                 if (BLI_path_frame_check_chars(string)) {
188                         BLI_path_frame_range(string, sfra, efra, 4);
189                 }
190         }
191
192         BLI_path_suffix(string, FILE_MAX, suffix, "");
193 }
194
195 static int start_avi(void *context_v, Scene *UNUSED(scene), RenderData *rd, int rectx, int recty,
196                      ReportList *reports, bool preview, const char *suffix)
197 {
198         int x, y;
199         char name[256];
200         AviFormat format;
201         int quality;
202         double framerate;
203         AviMovie *avi = context_v;
204
205         filepath_avi(name, rd, preview, suffix);
206
207         x = rectx;
208         y = recty;
209
210         quality = rd->im_format.quality;
211         framerate = (double) rd->frs_sec / (double) rd->frs_sec_base;
212
213         if (rd->im_format.imtype != R_IMF_IMTYPE_AVIJPEG) format = AVI_FORMAT_AVI_RGB;
214         else format = AVI_FORMAT_MJPEG;
215
216         if (AVI_open_compress(name, avi, 1, format) != AVI_ERROR_NONE) {
217                 BKE_report(reports, RPT_ERROR, "Cannot open or start AVI movie file");
218                 return 0;
219         }
220                         
221         AVI_set_compress_option(avi, AVI_OPTION_TYPE_MAIN, 0, AVI_OPTION_WIDTH, &x);
222         AVI_set_compress_option(avi, AVI_OPTION_TYPE_MAIN, 0, AVI_OPTION_HEIGHT, &y);
223         AVI_set_compress_option(avi, AVI_OPTION_TYPE_MAIN, 0, AVI_OPTION_QUALITY, &quality);
224         AVI_set_compress_option(avi, AVI_OPTION_TYPE_MAIN, 0, AVI_OPTION_FRAMERATE, &framerate);
225
226         avi->interlace = 0;
227         avi->odd_fields = 0;
228 /*  avi->interlace = rd->mode & R_FIELDS; */
229 /*  avi->odd_fields = (rd->mode & R_ODDFIELD) ? 1 : 0; */
230         
231         printf("Created avi: %s\n", name);
232         return 1;
233 }
234
235 static int append_avi(void *context_v, RenderData *UNUSED(rd), int start_frame, int frame, int *pixels,
236                       int rectx, int recty, const char *UNUSED(suffix), ReportList *UNUSED(reports))
237 {
238         unsigned int *rt1, *rt2, *rectot;
239         int x, y;
240         char *cp, rt;
241         AviMovie *avi = context_v;
242         
243         if (avi == NULL)
244                 return 0;
245
246         /* note that libavi free's the buffer... stupid interface - zr */
247         rectot = MEM_mallocN(rectx * recty * sizeof(int), "rectot");
248         rt1 = rectot;
249         rt2 = (unsigned int *)pixels + (recty - 1) * rectx;
250         /* flip y and convert to abgr */
251         for (y = 0; y < recty; y++, rt1 += rectx, rt2 -= rectx) {
252                 memcpy(rt1, rt2, rectx * sizeof(int));
253                 
254                 cp = (char *)rt1;
255                 for (x = rectx; x > 0; x--) {
256                         rt = cp[0];
257                         cp[0] = cp[3];
258                         cp[3] = rt;
259                         rt = cp[1];
260                         cp[1] = cp[2];
261                         cp[2] = rt;
262                         cp += 4;
263                 }
264         }
265         
266         AVI_write_frame(avi, (frame - start_frame), AVI_FORMAT_RGB32, rectot, rectx * recty * 4);
267 //      printf("added frame %3d (frame %3d in avi): ", frame, frame-start_frame);
268
269         return 1;
270 }
271
272 static void end_avi(void *context_v)
273 {
274         AviMovie *avi = context_v;
275
276         if (avi == NULL) return;
277
278         AVI_close_compress(avi);
279 }
280
281 static void *context_create_avi(void)
282 {
283         AviMovie *avi = MEM_mallocN(sizeof(AviMovie), "avimovie");
284         return avi;
285 }
286
287 static void context_free_avi(void *context_v)
288 {
289         AviMovie *avi = context_v;
290         if (avi) {
291                 MEM_freeN(avi);
292         }
293 }
294
295 #endif  /* WITH_AVI */
296
297 /* similar to BKE_image_path_from_imformat() */
298 void BKE_movie_filepath_get(char *string, RenderData *rd, bool preview, const char *suffix)
299 {
300         bMovieHandle *mh = BKE_movie_handle_get(rd->im_format.imtype);
301         if (mh && mh->get_movie_path) {
302                 mh->get_movie_path(string, rd, preview, suffix);
303         }
304         else {
305                 string[0] = '\0';
306         }
307 }