Merge branch 'blender2.7'
[blender.git] / source / blender / blenkernel / intern / writeavi.c
1 /*
2  * This program is free software; you can redistribute it and/or
3  * modify it under the terms of the GNU General Public License
4  * as published by the Free Software Foundation; either version 2
5  * of the License, or (at your option) any later version.
6  *
7  * This program is distributed in the hope that it will be useful,
8  * but WITHOUT ANY WARRANTY; without even the implied warranty of
9  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
10  * GNU General Public License for more details.
11  *
12  * You should have received a copy of the GNU General Public License
13  * along with this program; if not, write to the Free Software Foundation,
14  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
15  *
16  * The Original Code is Copyright (C) 2001-2002 by NaN Holding BV.
17  * All rights reserved.
18  * Functions for writing avi-format files.
19  * Added interface for generic movie support (ton)
20  */
21
22 /** \file \ingroup bke
23  */
24
25
26 #include <string.h>
27
28 #include "MEM_guardedalloc.h"
29
30 #include "DNA_scene_types.h"
31
32 #include "BLI_utildefines.h"
33
34 #include "BKE_report.h"
35 #ifdef WITH_AVI
36 #  include "BLI_blenlib.h"
37
38 #  include "BKE_main.h"
39 #endif
40
41 #include "BKE_writeavi.h"
42
43 /* ********************** general blender movie support ***************************** */
44
45 static int start_stub(void *UNUSED(context_v), Scene *UNUSED(scene), RenderData *UNUSED(rd), int UNUSED(rectx), int UNUSED(recty),
46                       ReportList *UNUSED(reports), bool UNUSED(preview), const char *UNUSED(suffix))
47 { return 0; }
48
49 static void end_stub(void *UNUSED(context_v))
50 {}
51
52 static int append_stub(void *UNUSED(context_v), RenderData *UNUSED(rd), int UNUSED(start_frame), int UNUSED(frame), int *UNUSED(pixels),
53                        int UNUSED(rectx), int UNUSED(recty), const char *UNUSED(suffix), ReportList *UNUSED(reports))
54 { return 0; }
55
56 static void *context_create_stub(void)
57 { return NULL; }
58
59 static void context_free_stub(void *UNUSED(context_v))
60 {}
61
62 #ifdef WITH_AVI
63 #  include "AVI_avi.h"
64
65 /* callbacks */
66 static int start_avi(void *context_v, Scene *scene, RenderData *rd, int rectx, int recty, ReportList *reports, bool preview, const char *suffix);
67 static void end_avi(void *context_v);
68 static int append_avi(void *context_v, RenderData *rd, int start_frame, int frame, int *pixels,
69                       int rectx, int recty, const char *suffix, ReportList *reports);
70 static void filepath_avi(char *string, RenderData *rd, bool preview, const char *suffix);
71 static void *context_create_avi(void);
72 static void context_free_avi(void *context_v);
73 #endif  /* WITH_AVI */
74
75 #ifdef WITH_FFMPEG
76 #  include "BKE_writeffmpeg.h"
77 #endif
78
79 bMovieHandle *BKE_movie_handle_get(const char imtype)
80 {
81         static bMovieHandle mh = {NULL};
82         /* stub callbacks in case none of the movie formats is supported */
83         mh.start_movie = start_stub;
84         mh.append_movie = append_stub;
85         mh.end_movie = end_stub;
86         mh.get_next_frame = NULL;
87         mh.get_movie_path = NULL;
88         mh.context_create = context_create_stub;
89         mh.context_free = context_free_stub;
90
91         /* set the default handle, as builtin */
92 #ifdef WITH_AVI
93         mh.start_movie = start_avi;
94         mh.append_movie = append_avi;
95         mh.end_movie = end_avi;
96         mh.get_movie_path = filepath_avi;
97         mh.context_create = context_create_avi;
98         mh.context_free = context_free_avi;
99 #endif
100
101         /* do the platform specific handles */
102 #ifdef WITH_FFMPEG
103         if (ELEM(imtype, R_IMF_IMTYPE_FFMPEG, R_IMF_IMTYPE_H264, R_IMF_IMTYPE_XVID, R_IMF_IMTYPE_THEORA)) {
104                 mh.start_movie = BKE_ffmpeg_start;
105                 mh.append_movie = BKE_ffmpeg_append;
106                 mh.end_movie = BKE_ffmpeg_end;
107                 mh.get_movie_path = BKE_ffmpeg_filepath_get;
108                 mh.context_create = BKE_ffmpeg_context_create;
109                 mh.context_free = BKE_ffmpeg_context_free;
110         }
111 #endif
112
113         /* in case all above are disabled */
114         (void)imtype;
115
116         return (mh.append_movie != append_stub) ? &mh : NULL;
117 }
118
119 /* ****************************************************************** */
120
121
122 #ifdef WITH_AVI
123
124 static void filepath_avi(char *string, RenderData *rd, bool preview, const char *suffix)
125 {
126         int sfra, efra;
127
128         if (string == NULL) return;
129
130         if (preview) {
131                 sfra = rd->psfra;
132                 efra = rd->pefra;
133         }
134         else {
135                 sfra = rd->sfra;
136                 efra = rd->efra;
137         }
138
139         strcpy(string, rd->pic);
140         BLI_path_abs(string, BKE_main_blendfile_path_from_global());
141
142         BLI_make_existing_file(string);
143
144         if (rd->scemode & R_EXTENSION) {
145                 if (!BLI_path_extension_check(string, ".avi")) {
146                         BLI_path_frame_range(string, sfra, efra, 4);
147                         strcat(string, ".avi");
148                 }
149         }
150         else {
151                 if (BLI_path_frame_check_chars(string)) {
152                         BLI_path_frame_range(string, sfra, efra, 4);
153                 }
154         }
155
156         BLI_path_suffix(string, FILE_MAX, suffix, "");
157 }
158
159 static int start_avi(void *context_v, Scene *UNUSED(scene), RenderData *rd, int rectx, int recty,
160                      ReportList *reports, bool preview, const char *suffix)
161 {
162         int x, y;
163         char name[256];
164         AviFormat format;
165         int quality;
166         double framerate;
167         AviMovie *avi = context_v;
168
169         filepath_avi(name, rd, preview, suffix);
170
171         x = rectx;
172         y = recty;
173
174         quality = rd->im_format.quality;
175         framerate = (double) rd->frs_sec / (double) rd->frs_sec_base;
176
177         if (rd->im_format.imtype != R_IMF_IMTYPE_AVIJPEG) format = AVI_FORMAT_AVI_RGB;
178         else format = AVI_FORMAT_MJPEG;
179
180         if (AVI_open_compress(name, avi, 1, format) != AVI_ERROR_NONE) {
181                 BKE_report(reports, RPT_ERROR, "Cannot open or start AVI movie file");
182                 return 0;
183         }
184
185         AVI_set_compress_option(avi, AVI_OPTION_TYPE_MAIN, 0, AVI_OPTION_WIDTH, &x);
186         AVI_set_compress_option(avi, AVI_OPTION_TYPE_MAIN, 0, AVI_OPTION_HEIGHT, &y);
187         AVI_set_compress_option(avi, AVI_OPTION_TYPE_MAIN, 0, AVI_OPTION_QUALITY, &quality);
188         AVI_set_compress_option(avi, AVI_OPTION_TYPE_MAIN, 0, AVI_OPTION_FRAMERATE, &framerate);
189
190         avi->interlace = 0;
191         avi->odd_fields = 0;
192
193         printf("Created avi: %s\n", name);
194         return 1;
195 }
196
197 static int append_avi(void *context_v, RenderData *UNUSED(rd), int start_frame, int frame, int *pixels,
198                       int rectx, int recty, const char *UNUSED(suffix), ReportList *UNUSED(reports))
199 {
200         unsigned int *rt1, *rt2, *rectot;
201         int x, y;
202         char *cp, rt;
203         AviMovie *avi = context_v;
204
205         if (avi == NULL)
206                 return 0;
207
208         /* note that libavi free's the buffer... stupid interface - zr */
209         rectot = MEM_mallocN(rectx * recty * sizeof(int), "rectot");
210         rt1 = rectot;
211         rt2 = (unsigned int *)pixels + (recty - 1) * rectx;
212         /* flip y and convert to abgr */
213         for (y = 0; y < recty; y++, rt1 += rectx, rt2 -= rectx) {
214                 memcpy(rt1, rt2, rectx * sizeof(int));
215
216                 cp = (char *)rt1;
217                 for (x = rectx; x > 0; x--) {
218                         rt = cp[0];
219                         cp[0] = cp[3];
220                         cp[3] = rt;
221                         rt = cp[1];
222                         cp[1] = cp[2];
223                         cp[2] = rt;
224                         cp += 4;
225                 }
226         }
227
228         AVI_write_frame(avi, (frame - start_frame), AVI_FORMAT_RGB32, rectot, rectx * recty * 4);
229 //      printf("added frame %3d (frame %3d in avi): ", frame, frame-start_frame);
230
231         return 1;
232 }
233
234 static void end_avi(void *context_v)
235 {
236         AviMovie *avi = context_v;
237
238         if (avi == NULL) return;
239
240         AVI_close_compress(avi);
241 }
242
243 static void *context_create_avi(void)
244 {
245         AviMovie *avi = MEM_mallocN(sizeof(AviMovie), "avimovie");
246         return avi;
247 }
248
249 static void context_free_avi(void *context_v)
250 {
251         AviMovie *avi = context_v;
252         if (avi) {
253                 MEM_freeN(avi);
254         }
255 }
256
257 #endif  /* WITH_AVI */
258
259 /* similar to BKE_image_path_from_imformat() */
260 void BKE_movie_filepath_get(char *string, RenderData *rd, bool preview, const char *suffix)
261 {
262         bMovieHandle *mh = BKE_movie_handle_get(rd->im_format.imtype);
263         if (mh && mh->get_movie_path) {
264                 mh->get_movie_path(string, rd, preview, suffix);
265         }
266         else {
267                 string[0] = '\0';
268         }
269 }