summaryrefslogtreecommitdiff
path: root/src/picture_frame.cc
blob: 87502e5ea75f3ef4c8e37c57b5d997e01415df12 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
/*
    Copyright (C) 2012 Carl Hetherington <cth@carlh.net>

    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation; either version 2 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program; if not, write to the Free Software
    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.

*/

#include <openjpeg.h>
#include "AS_DCP.h"
#include "KM_fileio.h"
#include "picture_frame.h"
#include "exceptions.h"
#include "argb_frame.h"
#include "lut.h"
#include "util.h"

using namespace std;
using namespace boost;
using namespace libdcp;

/** Make a picture frame from a 2D (monoscopic) asset.
 *  @param mxf_path Path to the asset's MXF file.
 *  @param n Frame within the asset, not taking EntryPoint into account.
 */
MonoPictureFrame::MonoPictureFrame (string mxf_path, int n)
{
	ASDCP::JP2K::MXFReader reader;
	if (ASDCP_FAILURE (reader.OpenRead (mxf_path.c_str()))) {
		throw FileError ("could not open MXF file for reading", mxf_path);
	}

	/* XXX: unfortunate guesswork on this buffer size */
	_buffer = new ASDCP::JP2K::FrameBuffer (4 * Kumu::Megabyte);

	if (ASDCP_FAILURE (reader.ReadFrame (n, *_buffer))) {
		throw DCPReadError ("could not read video frame");
	}
}

MonoPictureFrame::~MonoPictureFrame ()
{
	delete _buffer;
}

/** @return An ARGB representation of this frame.  This is ARGB in the
 *  Cairo sense, so that each pixel takes up 4 bytes; the first byte
 *  is blue, second green, third red and fourth alpha (always 255).
 */
shared_ptr<ARGBFrame>
MonoPictureFrame::argb_frame () const
{
	opj_image_t* xyz_frame = decompress_j2k (const_cast<uint8_t*> (_buffer->RoData()), _buffer->Size());
	assert (xyz_frame->numcomps == 3);
	shared_ptr<ARGBFrame> f = xyz_to_rgb (xyz_frame);
	opj_image_destroy (xyz_frame);
	return f;
}

/** Make a picture frame from a 3D (stereoscopic) asset.
 *  @param mxf_path Path to the asset's MXF file.
 *  @param n Frame within the asset, not taking EntryPoint into account.
 */
StereoPictureFrame::StereoPictureFrame (string mxf_path, int n)
{
	ASDCP::JP2K::MXFSReader reader;
	if (ASDCP_FAILURE (reader.OpenRead (mxf_path.c_str()))) {
		throw FileError ("could not open MXF file for reading", mxf_path);
	}

	/* XXX: unfortunate guesswork on this buffer size */
	_buffer = new ASDCP::JP2K::SFrameBuffer (4 * Kumu::Megabyte);

	if (ASDCP_FAILURE (reader.ReadFrame (n, *_buffer))) {
		throw DCPReadError ("could not read video frame");
	}
}

StereoPictureFrame::~StereoPictureFrame ()
{
	delete _buffer;
}

/** @return An ARGB representation of one of the eyes (left or right)
 *  of this frame.  This is ARGB in the Cairo sense, so that each
 *  pixel takes up 4 bytes; the first byte is blue, second green,
 *  third red and fourth alpha (always 255).
 *
 *  @param eye Eye to return (EYE_LEFT or EYE_RIGHT).
 */
shared_ptr<ARGBFrame>
StereoPictureFrame::argb_frame (Eye eye) const
{
	opj_image_t* xyz_frame = 0;
	switch (eye) {
	case LEFT:
		xyz_frame = decompress_j2k (const_cast<uint8_t*> (_buffer->Left.RoData()), _buffer->Left.Size());
		break;
	case RIGHT:
		xyz_frame = decompress_j2k (const_cast<uint8_t*> (_buffer->Right.RoData()), _buffer->Right.Size());
		break;
	}
	
	assert (xyz_frame->numcomps == 3);
	shared_ptr<ARGBFrame> f = xyz_to_rgb (xyz_frame);
	opj_image_destroy (xyz_frame);
	return f;
}