converted more mixed tab/space indentations to tabs. only whitespace changes.
[blender.git] / source / blender / blenkernel / intern / writeavi.c
index 0d94215..ba7f9bd 100644 (file)
@@ -1,17 +1,15 @@
-/**
+/*
  * Functions for writing avi-format files.
+ * Added interface for generic movie support (ton)
  *
  * $Id$
  *
- * ***** BEGIN GPL/BL DUAL LICENSE BLOCK *****
+ * ***** BEGIN GPL LICENSE BLOCK *****
  *
  * This program is free software; you can redistribute it and/or
  * modify it under the terms of the GNU General Public License
  * as published by the Free Software Foundation; either version 2
- * of the License, or (at your option) any later version. The Blender
- * Foundation also sells licenses for use in proprietary software under
- * the Blender License.  See http://www.blender.org/BL/ for information
- * about this.
+ * of the License, or (at your option) any later version.
  *
  * This program is distributed in the hope that it will be useful,
  * but WITHOUT ANY WARRANTY; without even the implied warranty of
@@ -20,7 +18,7 @@
  *
  * You should have received a copy of the GNU General Public License
  * along with this program; if not, write to the Free Software Foundation,
- * Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
+ * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  *
  * The Original Code is Copyright (C) 2001-2002 by NaN Holding BV.
  * All rights reserved.
  *
  * Contributor(s): Robert Wenzlaff
  *
- * ***** END GPL/BL DUAL LICENSE BLOCK *****
+ * ***** END GPL LICENSE BLOCK *****
  * 
  */
 
+/** \file blender/blenkernel/intern/writeavi.c
+ *  \ingroup bke
+ */
+
+
 #include <string.h>
 
 #include "MEM_guardedalloc.h"
+
+#include "DNA_scene_types.h"
+
 #include "BLI_blenlib.h"
+#include "BLI_utildefines.h"
+
+#include "BKE_global.h"
+#include "BKE_main.h"
+#include "BKE_report.h"
 
+#include "BKE_writeavi.h"
 #include "AVI_avi.h"
 
-#include "BKE_bad_level_calls.h"
-#include "BKE_global.h"
+/* callbacks */
+static int start_avi(Scene *scene, RenderData *rd, int rectx, int recty, ReportList *reports);
+static void end_avi(void);
+static int append_avi(RenderData *rd, int frame, int *pixels, int rectx, int recty, ReportList *reports);
+static void filepath_avi(char *string, RenderData *rd);
 
-/* RPW 11-21-2002 */
-#include "DNA_scene_types.h"
-/* RPW - End */
+/* ********************** general blender movie support ***************************** */
+
+#ifdef WITH_QUICKTIME
+#include "quicktime_export.h"
+#endif
+
+#ifdef WITH_FFMPEG
+#include "BKE_writeffmpeg.h"
+#endif
+
+#include "BKE_writeframeserver.h"
+
+bMovieHandle *BKE_get_movie_handle(int imtype)
+{
+       static bMovieHandle mh;
+       
+       /* set the default handle, as builtin */
+       mh.start_movie= start_avi;
+       mh.append_movie= append_avi;
+       mh.end_movie= end_avi;
+       mh.get_next_frame = NULL;
+       mh.get_movie_path = filepath_avi;
+       
+       /* do the platform specific handles */
+#if defined(_WIN32) && !defined(FREE_WINDOWS)
+       if (imtype == R_AVICODEC) {             
+               //XXX mh.start_movie= start_avi_codec;
+               //XXX mh.append_movie= append_avi_codec;
+               //XXX mh.end_movie= end_avi_codec;
+       }
+#endif
+#ifdef WITH_QUICKTIME
+       if (imtype == R_QUICKTIME) {
+               mh.start_movie= start_qt;
+               mh.append_movie= append_qt;
+               mh.end_movie= end_qt;
+               mh.get_movie_path = filepath_qt;
+       }
+#endif
+#ifdef WITH_FFMPEG
+       if (ELEM4(imtype, R_FFMPEG, R_H264, R_XVID, R_THEORA)) {
+               mh.start_movie = start_ffmpeg;
+               mh.append_movie = append_ffmpeg;
+               mh.end_movie = end_ffmpeg;
+               mh.get_movie_path = filepath_ffmpeg;
+       }
+#endif
+       if (imtype == R_FRAMESERVER) {
+               mh.start_movie = start_frameserver;
+               mh.append_movie = append_frameserver;
+               mh.end_movie = end_frameserver;
+               mh.get_next_frame = frameserver_loop;
+       }
+       
+       return &mh;
+}
+
+/* ****************************************************************** */
 
-#include "BKE_writeavi.h"
 
 static AviMovie *avi=NULL;
 static int sframe;
 
-void makeavistring (char *string) 
+static void filepath_avi (char *string, RenderData *rd)
 {
-       char txt[64];
-
-       if (string==0) return;
+       if (string==NULL) return;
 
-       strcpy(string, G.scene->r.pic);
-       BLI_convertstringcode(string, G.sce, G.scene->r.cfra);
+       strcpy(string, rd->pic);
+       BLI_path_abs(string, G.main->name);
 
-       RE_make_existing_file(string);
+       BLI_make_existing_file(string);
 
-       if (BLI_strcasecmp(string + strlen(string) - 4, ".avi")) {
-               sprintf(txt, "%04d_%04d.avi", (G.scene->r.sfra) , (G.scene->r.efra) );
-               strcat(string, txt);
+       if (!BLI_testextensie(string, ".avi")) {
+               BLI_path_frame_range(string, rd->sfra, rd->efra, 4);
+               strcat(string, ".avi");
        }
 }
 
-void start_avi(void)
+static int start_avi(Scene *scene, RenderData *rd, int rectx, int recty, ReportList *reports)
 {
        int x, y;
        char name[256];
        AviFormat format;
-       int quality, framerate;
+       int quality;
+       double framerate;
        
-       makeavistring(name);
+       (void)scene; /* unused */
+       
+       filepath_avi(name, rd);
 
-       sframe = (G.scene->r.sfra);
-       x = R.rectx;
-       y = R.recty;
+       sframe = (rd->sfra);
+       x = rectx;
+       y = recty;
 
-       quality= R.r.quality;
-       framerate= R.r.frs_sec;
+       quality= rd->quality;
+       framerate= (double) rd->frs_sec / (double) rd->frs_sec_base;
        
        avi = MEM_mallocN (sizeof(AviMovie), "avimovie");
 
        /* RPW 11-21-2002 
-        if (R.r.imtype != AVI_FORMAT_MJPEG) format = AVI_FORMAT_AVI_RGB;
+       if (rd->imtype != AVI_FORMAT_MJPEG) format = AVI_FORMAT_AVI_RGB;
        */
-       if (R.r.imtype != R_AVIJPEG ) format = AVI_FORMAT_AVI_RGB;
+       if (rd->imtype != R_AVIJPEG ) format = AVI_FORMAT_AVI_RGB;
        else format = AVI_FORMAT_MJPEG;
 
        if (AVI_open_compress (name, avi, 1, format) != AVI_ERROR_NONE) {
-               error("open movie");
+               BKE_report(reports, RPT_ERROR, "Cannot open or start AVI movie file.");
                MEM_freeN (avi);
                avi = NULL;
-               return;
+               return 0;
        }
                        
        AVI_set_compress_option (avi, AVI_OPTION_TYPE_MAIN, 0, AVI_OPTION_WIDTH, &x);
@@ -107,33 +177,32 @@ void start_avi(void)
 
        avi->interlace= 0;
        avi->odd_fields= 0;
-/*     avi->interlace= R.r.mode & R_FIELDS; */
-/*     avi->odd_fields= (R.r.mode & R_ODDFIELD)?1:0; */
+/*     avi->interlace= rd->mode & R_FIELDS; */
+/*     avi->odd_fields= (rd->mode & R_ODDFIELD)?1:0; */
        
        printf("Created avi: %s\n", name);
+       return 1;
 }
 
-void append_avi(int frame)
+static int append_avi(RenderData *UNUSED(rd), int frame, int *pixels, int rectx, int recty, ReportList *UNUSED(reports))
 {
        unsigned int *rt1, *rt2, *rectot;
        int x, y;
        char *cp, rt;
        
-       if (avi == NULL) {
-               G.afbreek = 1;
-               return;
-       }
+       if (avi == NULL)
+               return 0;
 
        /* note that libavi free's the buffer... stupid interface - zr */
-       rectot= MEM_mallocN(R.rectx*R.recty*sizeof(int), "rectot");
+       rectot= MEM_mallocN(rectx*recty*sizeof(int), "rectot");
        rt1= rectot;
-       rt2= R.rectot + (R.recty-1)*R.rectx;
+       rt2= (unsigned int*)pixels + (recty-1)*rectx;
        /* flip y and convert to abgr */
-       for (y=0; y < R.recty; y++, rt1+= R.rectx, rt2-= R.rectx) {
-               memcpy (rt1, rt2, R.rectx*sizeof(int));
+       for (y=0; y < recty; y++, rt1+= rectx, rt2-= rectx) {
+               memcpy (rt1, rt2, rectx*sizeof(int));
                
                cp= (char *)rt1;
-               for(x= R.rectx; x>0; x--) {
+               for(x= rectx; x>0; x--) {
                        rt= cp[0];
                        cp[0]= cp[3];
                        cp[3]= rt;
@@ -144,11 +213,13 @@ void append_avi(int frame)
                }
        }
        
-       AVI_write_frame (avi, (frame-sframe), AVI_FORMAT_RGB32, rectot, R.rectx*R.recty*4);
-       printf ("added frame %3d (frame %3d in avi): ", frame, frame-sframe);
+       AVI_write_frame (avi, (frame-sframe), AVI_FORMAT_RGB32, rectot, rectx*recty*4);
+//     printf ("added frame %3d (frame %3d in avi): ", frame, frame-sframe);
+
+       return 1;
 }
 
-void end_avi(void)
+static void end_avi(void)
 {
        if (avi == NULL) return;
 
@@ -156,3 +227,13 @@ void end_avi(void)
        MEM_freeN (avi);
        avi= NULL;
 }
+
+/* similar to BKE_makepicstring() */
+void BKE_makeanimstring(char *string, RenderData *rd)
+{
+       bMovieHandle *mh= BKE_get_movie_handle(rd->imtype);
+       if(mh->get_movie_path)
+               mh->get_movie_path(string, rd);
+       else
+               string[0]= '\0';
+}