source/input/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 "input.h"
  28 #include "threading.h"
  29 #include <fstream>
  30
  31 #define QUEUE_SIZE 5
  32
  33 namespace x265 {
  34 // x265 private namespace
  35
  36 class Y4MInput : public Input, public Thread
  37 {
  38 protected:
  39
  40     uint32_t rateNum;
  41
  42     uint32_t rateDenom;
  43
  44     uint32_t sarWidth;
  45
  46     uint32_t sarHeight;
  47
  48     size_t framesize;
  49
  50     int depth;
  51
  52     int width;
  53
  54     int height;
  55
  56     int colorSpace;
  57
  58     bool threadActive;
  59
  60     ThreadSafeInteger readCount;
  61
  62     ThreadSafeInteger writeCount;
  63
  64     char* buf[QUEUE_SIZE];
  65
  66     std::istream *ifs;
  67
  68     bool parseHeader();
  69
  70     void threadMain();
  71
  72     bool populateFrameQueue();
  73
  74 public:
  75
  76     Y4MInput(InputFileInfo& info);
  77
  78     virtual ~Y4MInput();
  79
  80     void release();
  81
  82     bool isEof() const            { return ifs && ifs->eof();  }
  83
  84     bool isFail()                 { return !(ifs && !ifs->fail() && threadActive); }
  85
  86     void startReader();
  87
  88     bool readPicture(x265_picture&);
  89
  90     const char *getName() const   { return "y4m"; }
  91 };
  92 }
  93
  94 #endif // ifndef X265_Y4M_H