source/output/y4m.h

   1 /*****************************************************************************
   2  * Copyright (C) 2013 x265 project
   3  *
   4  * Authors: Steve Borho <steve@borho.org>
   5  *
   6  * This program is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License as published by
   8  * the Free Software Foundation; either version 2 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with this program; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02111, USA.
  19  *
  20  * This program is also available under a commercial proprietary license.
  21  * For more information, contact us at license @ x265.com.
  22  *****************************************************************************/
  23
  24 #ifndef X265_Y4M_H
  25 #define X265_Y4M_H
  26
  27 #include "output.h"
  28 #include <fstream>
  29
  30 namespace x265 {
  31 // private x265 namespace
  32
  33 class Y4MOutput : public Output
  34 {
  35 protected:
  36
  37     int width;
  38
  39     int height;
  40
  41     int colorSpace;
  42
  43     uint32_t frameSize;
  44
  45     std::ofstream ofs;
  46
  47     std::ofstream::pos_type header;
  48
  49     char *buf;
  50
  51     void writeHeader();
  52
  53 public:
  54
  55     Y4MOutput(const char *filename, int width, int height, uint32_t fpsNum, uint32_t fpsDenom, int csp);
  56
  57     virtual ~Y4MOutput();
  58
  59     const char *getName() const                   { return "y4m"; }
  60
  61     bool isFail() const                           { return ofs.fail(); }
  62
  63     void release()                                { delete this; }
  64
  65     bool writePicture(const x265_picture& pic);
  66 };
  67 }
  68
  69 #endif // ifndef X265_Y4M_H