00001 /*!@file Media/FfmpegEncoder.H Low-level class for using ffmpeg to decode movie files */ 00002 00003 // //////////////////////////////////////////////////////////////////// // 00004 // The iLab Neuromorphic Vision C++ Toolkit - Copyright (C) 2000-2005 // 00005 // by the University of Southern California (USC) and the iLab at USC. // 00006 // See http://iLab.usc.edu for information about this project. // 00007 // //////////////////////////////////////////////////////////////////// // 00008 // Major portions of the iLab Neuromorphic Vision Toolkit are protected // 00009 // under the U.S. patent ``Computation of Intrinsic Perceptual Saliency // 00010 // in Visual Environments, and Applications'' by Christof Koch and // 00011 // Laurent Itti, California Institute of Technology, 2001 (patent // 00012 // pending; application number 09/912,225 filed July 23, 2001; see // 00013 // http://pair.uspto.gov/cgi-bin/final/home.pl for current status). // 00014 // //////////////////////////////////////////////////////////////////// // 00015 // This file is part of the iLab Neuromorphic Vision C++ Toolkit. // 00016 // // 00017 // The iLab Neuromorphic Vision C++ Toolkit is free software; you can // 00018 // redistribute it and/or modify it under the terms of the GNU General // 00019 // Public License as published by the Free Software Foundation; either // 00020 // version 2 of the License, or (at your option) any later version. // 00021 // // 00022 // The iLab Neuromorphic Vision C++ Toolkit is distributed in the hope // 00023 // that it will be useful, but WITHOUT ANY WARRANTY; without even the // 00024 // implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR // 00025 // PURPOSE. See the GNU General Public License for more details. // 00026 // // 00027 // You should have received a copy of the GNU General Public License // 00028 // along with the iLab Neuromorphic Vision C++ Toolkit; if not, write // 00029 // to the Free Software Foundation, Inc., 59 Temple Place, Suite 330, // 00030 // Boston, MA 02111-1307 USA. // 00031 // //////////////////////////////////////////////////////////////////// // 00032 // 00033 // Primary maintainer for this file:Rob Peters <rjpeters at usc dot edu> 00034 // $HeadURL: svn://isvn.usc.edu/software/invt/trunk/saliency/src/Media/FfmpegEncoder.H $ 00035 // $Id: FfmpegEncoder.H 12962 2010-03-06 02:13:53Z irock $ 00036 // 00037 00038 #ifndef MEDIA_FFMPEGENCODER_H_DEFINED 00039 #define MEDIA_FFMPEGENCODER_H_DEFINED 00040 00041 #ifdef INVT_HAVE_AVCODEC 00042 00043 #include "Image/Range.H" 00044 #include "Transport/LowLevelEncoder.H" 00045 #include "Util/Types.H" // for byte 00046 00047 #include <cstdio> 00048 #define attribute_deprecated 00049 extern "C" 00050 { 00051 #ifdef HAVE_LIBAVCODEC_AVCODEC_H 00052 #include <libavcodec/avcodec.h> 00053 #include <libavformat/avformat.h> 00054 #else 00055 #include <ffmpeg/avcodec.h> 00056 #include <ffmpeg/avformat.h> 00057 #endif 00058 00059 } 00060 #include <string> 00061 00062 class Dims; 00063 class VideoFrame; 00064 template <class T> class Image; 00065 template <class T> class PixRGB; 00066 00067 /// Low-level class for using ffmpeg to decode movie files 00068 /** You probably want to use MpegOutputStream instead, which relies on 00069 FfmpegEncoder internally for its implementation but offers a 00070 richer, higher-level interface. */ 00071 class FfmpegEncoder : public LowLevelEncoder 00072 { 00073 public: 00074 FfmpegEncoder(const std::string& fname, 00075 const std::string& codecname, 00076 const int bitrate, 00077 const int framerate, 00078 const int frameratebase, 00079 const Dims& dims, 00080 const int bufsz, 00081 const bool useFormatContext = false); 00082 00083 virtual ~FfmpegEncoder(); 00084 00085 virtual int close(); 00086 00087 void writeRawFrame(const AVFrame* picture); 00088 00089 void writeRGB(const Image<PixRGB<byte> >& img); 00090 00091 void writeVideoFrame(const VideoFrame& frame); 00092 00093 virtual void writeFrame(const GenericFrame& f); 00094 00095 private: 00096 FILE* itsFile; 00097 AVCodecContext itsContext; 00098 AVFormatContext* itsFormatContext; 00099 AVStream* itsAVStream; 00100 int itsFrameNumber; 00101 const int itsOutbufSize; 00102 Range<int> itsFrameSizeRange; 00103 bool itsUseFormatContext; 00104 }; 00105 00106 #endif // HAVE_FFMPEG_AVCODEC_H 00107 00108 // ###################################################################### 00109 /* So things look consistent in everyone's emacs... */ 00110 /* Local Variables: */ 00111 /* mode: c++ */ 00112 /* indent-tabs-mode: nil */ 00113 /* End: */ 00114 00115 #endif // MEDIA_FFMPEGENCODER_H_DEFINED