Patch #34204: [Render Animation] Fails with "Error: Specified sample_fmt is not suppo...
[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 #ifdef WITH_AVI
54 #  include "AVI_avi.h"
55
56 /* callbacks */
57 static int start_avi(Scene *scene, RenderData *rd, int rectx, int recty, ReportList *reports);
58 static void end_avi(void);
59 static int append_avi(RenderData *rd, int start_frame, int frame, int *pixels,
60                       int rectx, int recty, ReportList *reports);
61 static void filepath_avi(char *string, RenderData *rd);
62 #endif  /* WITH_AVI */
63
64 #ifdef WITH_QUICKTIME
65 #  include "quicktime_export.h"
66 #endif
67
68 #ifdef WITH_FFMPEG
69 #  include "BKE_writeffmpeg.h"
70 #endif
71
72 #include "BKE_writeframeserver.h"
73
74 bMovieHandle *BKE_movie_handle_get(const char imtype)
75 {
76         static bMovieHandle mh = {0};
77         
78         /* set the default handle, as builtin */
79 #ifdef WITH_AVI
80         mh.start_movie = start_avi;
81         mh.append_movie = append_avi;
82         mh.end_movie = end_avi;
83         mh.get_next_frame = NULL;
84         mh.get_movie_path = filepath_avi;
85 #endif
86
87         /* do the platform specific handles */
88 #ifdef WITH_QUICKTIME
89         if (imtype == R_IMF_IMTYPE_QUICKTIME) {
90                 mh.start_movie = start_qt;
91                 mh.append_movie = append_qt;
92                 mh.end_movie = end_qt;
93                 mh.get_movie_path = filepath_qt;
94         }
95 #endif
96 #ifdef WITH_FFMPEG
97         if (ELEM4(imtype, R_IMF_IMTYPE_FFMPEG, R_IMF_IMTYPE_H264, R_IMF_IMTYPE_XVID, R_IMF_IMTYPE_THEORA)) {
98                 mh.start_movie = BKE_ffmpeg_start;
99                 mh.append_movie = BKE_ffmpeg_append;
100                 mh.end_movie = BKE_ffmpeg_end;
101                 mh.get_movie_path = BKE_ffmpeg_filepath_get;
102         }
103 #endif
104 #ifdef WITH_FRAMESERVER
105         if (imtype == R_IMF_IMTYPE_FRAMESERVER) {
106                 mh.start_movie = BKE_frameserver_start;
107                 mh.append_movie = BKE_frameserver_append;
108                 mh.end_movie = BKE_frameserver_end;
109                 mh.get_next_frame = BKE_frameserver_loop;
110         }
111 #endif
112
113         /* in case all above are disabled */
114         (void)imtype;
115
116         return &mh;
117 }
118
119 /* ****************************************************************** */
120
121
122 #ifdef WITH_AVI
123
124 static AviMovie *avi = NULL;
125
126 static void filepath_avi(char *string, RenderData *rd)
127 {
128         if (string == NULL) return;
129
130         strcpy(string, rd->pic);
131         BLI_path_abs(string, G.main->name);
132
133         BLI_make_existing_file(string);
134
135         if (!BLI_testextensie(string, ".avi")) {
136                 BLI_path_frame_range(string, rd->sfra, rd->efra, 4);
137                 strcat(string, ".avi");
138         }
139 }
140
141 static int start_avi(Scene *scene, RenderData *rd, int rectx, int recty, ReportList *reports)
142 {
143         int x, y;
144         char name[256];
145         AviFormat format;
146         int quality;
147         double framerate;
148         
149         (void)scene; /* unused */
150         
151         filepath_avi(name, rd);
152
153         x = rectx;
154         y = recty;
155
156         quality = rd->im_format.quality;
157         framerate = (double) rd->frs_sec / (double) rd->frs_sec_base;
158         
159         avi = MEM_mallocN(sizeof(AviMovie), "avimovie");
160
161         if (rd->im_format.imtype != R_IMF_IMTYPE_AVIJPEG) format = AVI_FORMAT_AVI_RGB;
162         else format = AVI_FORMAT_MJPEG;
163
164         if (AVI_open_compress(name, avi, 1, format) != AVI_ERROR_NONE) {
165                 BKE_report(reports, RPT_ERROR, "Cannot open or start AVI movie file");
166                 MEM_freeN(avi);
167                 avi = NULL;
168                 return 0;
169         }
170                         
171         AVI_set_compress_option(avi, AVI_OPTION_TYPE_MAIN, 0, AVI_OPTION_WIDTH, &x);
172         AVI_set_compress_option(avi, AVI_OPTION_TYPE_MAIN, 0, AVI_OPTION_HEIGHT, &y);
173         AVI_set_compress_option(avi, AVI_OPTION_TYPE_MAIN, 0, AVI_OPTION_QUALITY, &quality);
174         AVI_set_compress_option(avi, AVI_OPTION_TYPE_MAIN, 0, AVI_OPTION_FRAMERATE, &framerate);
175
176         avi->interlace = 0;
177         avi->odd_fields = 0;
178 /*  avi->interlace = rd->mode & R_FIELDS; */
179 /*  avi->odd_fields = (rd->mode & R_ODDFIELD) ? 1 : 0; */
180         
181         printf("Created avi: %s\n", name);
182         return 1;
183 }
184
185 static int append_avi(RenderData *UNUSED(rd), int start_frame, int frame, int *pixels,
186                       int rectx, int recty, ReportList *UNUSED(reports))
187 {
188         unsigned int *rt1, *rt2, *rectot;
189         int x, y;
190         char *cp, rt;
191         
192         if (avi == NULL)
193                 return 0;
194
195         /* note that libavi free's the buffer... stupid interface - zr */
196         rectot = MEM_mallocN(rectx * recty * sizeof(int), "rectot");
197         rt1 = rectot;
198         rt2 = (unsigned int *)pixels + (recty - 1) * rectx;
199         /* flip y and convert to abgr */
200         for (y = 0; y < recty; y++, rt1 += rectx, rt2 -= rectx) {
201                 memcpy(rt1, rt2, rectx * sizeof(int));
202                 
203                 cp = (char *)rt1;
204                 for (x = rectx; x > 0; x--) {
205                         rt = cp[0];
206                         cp[0] = cp[3];
207                         cp[3] = rt;
208                         rt = cp[1];
209                         cp[1] = cp[2];
210                         cp[2] = rt;
211                         cp += 4;
212                 }
213         }
214         
215         AVI_write_frame(avi, (frame - start_frame), AVI_FORMAT_RGB32, rectot, rectx * recty * 4);
216 //      printf ("added frame %3d (frame %3d in avi): ", frame, frame-start_frame);
217
218         return 1;
219 }
220
221 static void end_avi(void)
222 {
223         if (avi == NULL) return;
224
225         AVI_close_compress(avi);
226         MEM_freeN(avi);
227         avi = NULL;
228 }
229 #endif  /* WITH_AVI */
230
231 /* similar to BKE_makepicstring() */
232 void BKE_movie_filepath_get(char *string, RenderData *rd)
233 {
234         bMovieHandle *mh = BKE_movie_handle_get(rd->im_format.imtype);
235         if (mh->get_movie_path)
236                 mh->get_movie_path(string, rd);
237         else
238                 string[0] = '\0';
239 }