synfig-core/trunk/src/modules/mod_ffmpeg/mptr_ffmpeg.cpp

   1 /* === S Y N F I G ========================================================= */
   2 /*!     \file mptr_ffmpeg.cpp
   3 **      \brief ppm Target Module
   4 **
   5 **      $Id$
   6 **
   7 **      \legal
   8 **      Copyright (c) 2002-2005 Robert B. Quattlebaum Jr., Adrian Bentley
   9 **      Copyright (c) 2007 Chris Moore
  10 **
  11 **      This package is free software; you can redistribute it and/or
  12 **      modify it under the terms of the GNU General Public License as
  13 **      published by the Free Software Foundation; either version 2 of
  14 **      the License, or (at your option) any later version.
  15 **
  16 **      This package is distributed in the hope that it will be useful,
  17 **      but WITHOUT ANY WARRANTY; without even the implied warranty of
  18 **      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  19 **      General Public License for more details.
  20 **      \endlegal
  21 **
  22 ** === N O T E S ===========================================================
  23 **
  24 ** ========================================================================= */
  25
  26 /* === H E A D E R S ======================================================= */
  27
  28 #ifdef USING_PCH
  29 #       include "pch.h"
  30 #else
  31 #ifdef HAVE_CONFIG_H
  32 #       include <config.h>
  33 #endif
  34
  35 #include <ETL/stringf>
  36 #include "mptr_ffmpeg.h"
  37 #include <stdio.h>
  38 #include <sys/types.h>
  39 #if HAVE_SYS_WAIT_H
  40  #include <sys/wait.h>
  41 #endif
  42 #if HAVE_IO_H
  43  #include <io.h>
  44 #endif
  45 #if HAVE_PROCESS_H
  46  #include <process.h>
  47 #endif
  48 #if HAVE_FCNTL_H
  49  #include <fcntl.h>
  50 #endif
  51 #include <unistd.h>
  52 #include <iostream>
  53 #include <algorithm>
  54 #include <functional>
  55 #include <ETL/stringf>
  56 #endif
  57
  58 /* === M A C R O S ========================================================= */
  59
  60 using namespace synfig;
  61 using namespace std;
  62 using namespace etl;
  63
  64 #if defined(HAVE_FORK) && defined(HAVE_PIPE) && defined(HAVE_WAITPID)
  65  #define UNIX_PIPE_TO_PROCESSES
  66 #else
  67  #define WIN32_PIPE_TO_PROCESSES
  68 #endif
  69
  70 /* === G L O B A L S ======================================================= */
  71
  72 SYNFIG_IMPORTER_INIT(ffmpeg_mptr);
  73 SYNFIG_IMPORTER_SET_NAME(ffmpeg_mptr,"ffmpeg");
  74 SYNFIG_IMPORTER_SET_EXT(ffmpeg_mptr,"avi");
  75 SYNFIG_IMPORTER_SET_VERSION(ffmpeg_mptr,"0.1");
  76 SYNFIG_IMPORTER_SET_CVS_ID(ffmpeg_mptr,"$Id$");
  77
  78 /* === M E T H O D S ======================================================= */
  79
  80 bool ffmpeg_mptr::is_animated()
  81 {
  82         return true;
  83 }
  84
  85 bool
  86 ffmpeg_mptr::seek_to(int frame)
  87 {
  88         if(frame<cur_frame || !file)
  89         {
  90                 if(file)
  91                 {
  92 #if defined(WIN32_PIPE_TO_PROCESSES)
  93                         pclose(file);
  94 #elif defined(UNIX_PIPE_TO_PROCESSES)
  95                         fclose(file);
  96                         int status;
  97                         waitpid(pid,&status,0);
  98 #endif
  99                 }
 100
 101 #if defined(WIN32_PIPE_TO_PROCESSES)
 102
 103                 string command;
 104
 105                 command=strprintf("ffmpeg -ss 00:00:00.%d -i \"%s\" -an -f image2pipe -vcodec ppm -\n",frame,filename.c_str());
 106
 107                 file=popen(command.c_str(),POPEN_BINARY_READ_TYPE);
 108
 109 #elif defined(UNIX_PIPE_TO_PROCESSES)
 110
 111                 int p[2];
 112
 113                 if (pipe(p)) {
 114                         cerr<<"Unable to open pipe to ffmpeg"<<endl;
 115                         return false;
 116                 };
 117
 118                 pid = fork();
 119
 120                 if (pid == -1) {
 121                         cerr<<"Unable to open pipe to ffmpeg"<<endl;
 122                         return false;
 123                 }
 124
 125                 if (pid == 0){
 126                         // Child process
 127                         // Close pipein, not needed
 128                         close(p[0]);
 129                         // Dup pipein to stdout
 130                         if( dup2( p[1], STDOUT_FILENO ) == -1 ){
 131                                 cerr<<"Unable to open pipe to ffmpeg"<<endl;
 132                                 return false;
 133                         }
 134                         // Close the unneeded pipein
 135                         close(p[1]);
 136                         string time = strprintf("00:00:00.%d",frame);
 137                         execlp("ffmpeg", "ffmpeg", "-ss", time.c_str(), "-i", filename.c_str(), "-an", "-f", "image2pipe", "-vcodec", "ppm", "-", (const char *)NULL);
 138                         // We should never reach here unless the exec failed
 139                         cerr<<"Unable to open pipe to ffmpeg"<<endl;
 140                         _exit(1);
 141                 } else {
 142                         // Parent process
 143                         // Close pipeout, not needed
 144                         close(p[1]);
 145                         // Save pipein to file handle, will read from it later
 146                         file = fdopen(p[0], "rb");
 147                 }
 148
 149 #else
 150         #error There are no known APIs for creating child processes
 151 #endif
 152
 153                 if(!file)
 154                 {
 155                         cerr<<"Unable to open pipe to ffmpeg"<<endl;
 156                         return false;
 157                 }
 158                 cur_frame=-1;
 159         }
 160
 161         while(cur_frame<frame-1)
 162         {
 163                 cerr<<"Seeking to..."<<frame<<'('<<cur_frame<<')'<<endl;
 164                 if(!grab_frame())
 165                         return false;
 166         }
 167         return true;
 168 }
 169
 170 bool
 171 ffmpeg_mptr::grab_frame(void)
 172 {
 173         if(!file)
 174         {
 175                 cerr<<"unable to open "<<filename<<endl;
 176                 return false;
 177         }
 178         int w,h;
 179         float divisor;
 180         char cookie[2];
 181         cookie[0]=fgetc(file);
 182
 183         if(feof(file))
 184                 return false;
 185
 186         cookie[1]=fgetc(file);
 187
 188         if(cookie[0]!='P' || cookie[1]!='6')
 189         {
 190                 cerr<<"stream not in PPM format \""<<cookie[0]<<cookie[1]<<'"'<<endl;
 191                 return false;
 192         }
 193
 194         fgetc(file);
 195         fscanf(file,"%d %d\n",&w,&h);
 196         fscanf(file,"%f",&divisor);
 197         fgetc(file);
 198
 199         if(feof(file))
 200                 return false;
 201
 202         int x;
 203         int y;
 204         frame.set_wh(w,h);
 205         for(y=0;y<frame.get_h();y++)
 206                 for(x=0;x<frame.get_w();x++)
 207                 {
 208                         if(feof(file))
 209                                 return false;
 210 /*
 211                         frame[y][x]=Color(
 212                                 (float)(unsigned char)fgetc(file)/divisor,
 213                                 (float)(unsigned char)fgetc(file)/divisor,
 214                                 (float)(unsigned char)fgetc(file)/divisor,
 215                                 1.0
 216 */
 217                         float r=gamma().r_U8_to_F32((unsigned char)fgetc(file));
 218                         float g=gamma().g_U8_to_F32((unsigned char)fgetc(file));
 219                         float b=gamma().b_U8_to_F32((unsigned char)fgetc(file));
 220                         frame[y][x]=Color(
 221                                 r,
 222                                 g,
 223                                 b,
 224                                 1.0
 225                         );
 226                 }
 227         cur_frame++;
 228         return true;
 229 }
 230
 231 ffmpeg_mptr::ffmpeg_mptr(const char *f)
 232 {
 233         pid=-1;
 234 #ifdef HAVE_TERMIOS_H
 235         tcgetattr (0, &oldtty);
 236 #endif
 237         filename=f;
 238         file=NULL;
 239         fps=23.98;
 240         cur_frame=-1;
 241 }
 242
 243 ffmpeg_mptr::~ffmpeg_mptr()
 244 {
 245         if(file)
 246         {
 247 #if defined(WIN32_PIPE_TO_PROCESSES)
 248                 pclose(file);
 249 #elif defined(UNIX_PIPE_TO_PROCESSES)
 250                 fclose(file);
 251                 int status;
 252                 waitpid(pid,&status,0);
 253 #endif
 254         }
 255 #ifdef HAVE_TERMIOS_H
 256         tcsetattr(0,TCSANOW,&oldtty);
 257 #endif
 258 }
 259
 260 bool
 261 ffmpeg_mptr::get_frame(synfig::Surface &surface,Time time, synfig::ProgressCallback *)
 262 {
 263         int i=(int)(time*fps);
 264         if(i!=cur_frame)
 265         {
 266                 if(!seek_to(i))
 267                         return false;
 268                 if(!grab_frame())
 269                         return false;
 270         }
 271
 272         surface=frame;
 273         return true;
 274 }