Commit | Line | Data |
---|---|---|
72b9787e JB |
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 |