Missing includes.
[libdcp.git] / src / xyz_frame.cc
1 /*
2     Copyright (C) 2012-2013 Carl Hetherington <cth@carlh.net>
3
4     This program is free software; you can redistribute it and/or modify
5     it under the terms of the GNU General Public License as published by
6     the Free Software Foundation; either version 2 of the License, or
7     (at your option) any later version.
8
9     This program is distributed in the hope that it will be useful,
10     but WITHOUT ANY WARRANTY; without even the implied warranty of
11     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12     GNU General Public License for more details.
13
14     You should have received a copy of the GNU General Public License
15     along with this program; if not, write to the Free Software
16     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
17
18 */
19
20 #include <cassert>
21 #include <stdexcept>
22 #include <openjpeg.h>
23 #include "xyz_frame.h"
24
25 using namespace libdcp;
26
27 /** Construct an XYZFrame, taking ownership of the opj_image_t */
28 XYZFrame::XYZFrame (opj_image_t* image)
29         : _opj_image (image)
30 {
31         assert (_opj_image->numcomps == 3);
32 }
33
34 XYZFrame::XYZFrame (Size size)
35 {
36         opj_image_cmptparm_t cmptparm[3];
37         
38         for (int i = 0; i < 3; ++i) {
39                 cmptparm[i].dx = 1;
40                 cmptparm[i].dy = 1;
41                 cmptparm[i].w = size.width;
42                 cmptparm[i].h = size.height;
43                 cmptparm[i].x0 = 0;
44                 cmptparm[i].y0 = 0;
45                 cmptparm[i].prec = 12;
46                 cmptparm[i].bpp = 12;
47                 cmptparm[i].sgnd = 0;
48         }
49
50         /* XXX: is this _SRGB right? */
51         _opj_image = opj_image_create (3, &cmptparm[0], OPJ_CLRSPC_SRGB);
52         if (_opj_image == 0) {
53                 throw std::runtime_error ("could not create libopenjpeg image");
54         }
55
56         _opj_image->x0 = 0;
57         _opj_image->y0 = 0;
58         _opj_image->x1 = size.width;
59         _opj_image->y1 = size.height;
60 }
61
62 XYZFrame::~XYZFrame ()
63 {
64         opj_image_destroy (_opj_image);
65 }
66
67 int *
68 XYZFrame::data (int c) const
69 {
70         assert (c >= 0 && c < 3);
71         return _opj_image->comps[c].data;
72 }
73
74 libdcp::Size
75 XYZFrame::size () const
76 {
77         /* XXX: this may not be right; x0 and y0 can presumably be non-zero */
78         return libdcp::Size (_opj_image->x1, _opj_image->y1);
79 }