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 moviemaker_h
  24 #define moviemaker_h
  25
  26 #ifndef PACKAGE
  27 # error config.h must be included first in each C++ source file
  28 #endif
  29
  30 #include <stdio.h>
  31
  32 struct AVCodecContext;
  33 struct AVFormatContext;
  34 struct AVStream;
  35 struct AVFrame;
  36 struct AVPicture;
  37 struct SwsContext;
  38
  39 class MovieMaker {
  40 #ifdef WITH_LIBAV
  41     AVFormatContext *oc;
  42     AVStream *video_st;
  43 # ifndef HAVE_AVCODEC_ENCODE_VIDEO2
  44     int out_size; // Legacy-only.
  45 # endif
  46     AVCodecContext *context;
  47     AVFrame *frame;
  48     unsigned char *outbuf; // Legacy-only.
  49 # ifndef HAVE_AVCODEC_ENCODE_VIDEO2
  50     AVPicture *out; // Legacy-only.
  51 # endif
  52     unsigned char *pixels;
  53     SwsContext *sws_ctx;
  54     int averrno;
  55     FILE* fh_to_close;
  56
  57     int encode_frame(AVFrame* frame);
  58     void release();
  59 #endif
  60
  61 public:
  62     MovieMaker();
  63     bool Open(FILE* fh, const char* ext, int width, int height);
  64     unsigned char * GetBuffer() const;
  65     int GetWidth() const;
  66     int GetHeight() const;
  67     bool AddFrame();
  68     bool Close();
  69     ~MovieMaker();
  70     const char * get_error_string() const;
  71 };
  72
  73 #endif