src/moviemaker.h

   1 //
   2 //  moviemaker.h
   3 //
   4 //  Class for writing movies from Aven.
   5 //
   6 //  Copyright (C) 2004,2010,2011,2013,2014,2016 Olly Betts
   7 //
   8 //  This program is free software; you can redistribute it and/or modify
   9 //  it under the terms of the GNU General Public License as published by
  10 //  the Free Software Foundation; either version 2 of the License, or
  11 //  (at your option) any later version.
  12 //
  13 //  This program is distributed in the hope that it will be useful,
  14 //  but WITHOUT ANY WARRANTY; without even the implied warranty of
  15 //  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16 //  GNU General Public License for more details.
  17 //
  18 //  You should have received a copy of the GNU General Public License
  19 //  along with this program; if not, write to the Free Software
  20 //  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
  21 //
  22
  23 #ifndef PACKAGE
  24 # error config.h must be included first in each C++ source file
  25 #endif
  26
  27 #include <stdio.h>
  28
  29 struct AVFormatContext;
  30 struct AVStream;
  31 struct AVFrame;
  32 struct AVPicture;
  33 struct SwsContext;
  34
  35 class MovieMaker {
  36 #ifdef WITH_LIBAV
  37     AVFormatContext *oc;
  38     AVStream *video_st;
  39 # ifndef HAVE_AVCODEC_ENCODE_VIDEO2
  40     int out_size;
  41 # endif
  42     AVFrame *frame;
  43     unsigned char *outbuf;
  44 # ifndef HAVE_AVCODEC_ENCODE_VIDEO2
  45     AVPicture *out;
  46 # endif
  47     unsigned char *pixels;
  48     SwsContext *sws_ctx;
  49     int averrno;
  50     FILE* fh_to_close;
  51
  52     void release();
  53 #endif
  54
  55 public:
  56     MovieMaker();
  57     bool Open(FILE* fh, const char* ext, int width, int height);
  58     unsigned char * GetBuffer() const;
  59     int GetWidth() const;
  60     int GetHeight() const;
  61     bool AddFrame();
  62     bool Close();
  63     ~MovieMaker();
  64     const char * get_error_string() const;
  65 };