src/picture_frame.cc

   1 /*
   2     Copyright (C) 2012 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 <openjpeg.h>
  21 #include "AS_DCP.h"
  22 #include "KM_fileio.h"
  23 #include "picture_frame.h"
  24 #include "exceptions.h"
  25 #include "argb_frame.h"
  26 #include "lut.h"
  27 #include "util.h"
  28
  29 using std::string;
  30 using boost::shared_ptr;
  31 using namespace libdcp;
  32
  33 /** Make a picture frame from a 2D (monoscopic) asset.
  34  *  @param mxf_path Path to the asset's MXF file.
  35  *  @param n Frame within the asset, not taking EntryPoint into account.
  36  */
  37 MonoPictureFrame::MonoPictureFrame (string mxf_path, int n)
  38 {
  39         ASDCP::JP2K::MXFReader reader;
  40         if (ASDCP_FAILURE (reader.OpenRead (mxf_path.c_str()))) {
  41                 throw FileError ("could not open MXF file for reading", mxf_path);
  42         }
  43
  44         /* XXX: unfortunate guesswork on this buffer size */
  45         _buffer = new ASDCP::JP2K::FrameBuffer (4 * Kumu::Megabyte);
  46
  47         if (ASDCP_FAILURE (reader.ReadFrame (n, *_buffer))) {
  48                 throw DCPReadError ("could not read video frame");
  49         }
  50 }
  51
  52 MonoPictureFrame::~MonoPictureFrame ()
  53 {
  54         delete _buffer;
  55 }
  56
  57 /** @param reduce a factor by which to reduce the resolution
  58  *  of the image, expressed as a power of two (pass 0 for no
  59  *  reduction).
  60  *
  61  *  @return An ARGB representation of this frame.  This is ARGB in the
  62  *  Cairo sense, so that each pixel takes up 4 bytes; the first byte
  63  *  is blue, second green, third red and fourth alpha (always 255).
  64  *
  65  */
  66 shared_ptr<ARGBFrame>
  67 MonoPictureFrame::argb_frame (int reduce) const
  68 {
  69         opj_image_t* xyz_frame = decompress_j2k (const_cast<uint8_t*> (_buffer->RoData()), _buffer->Size(), reduce);
  70         assert (xyz_frame->numcomps == 3);
  71         shared_ptr<ARGBFrame> f = xyz_to_rgb (xyz_frame);
  72         opj_image_destroy (xyz_frame);
  73         return f;
  74 }
  75
  76 /** Make a picture frame from a 3D (stereoscopic) asset.
  77  *  @param mxf_path Path to the asset's MXF file.
  78  *  @param n Frame within the asset, not taking EntryPoint into account.
  79  */
  80 StereoPictureFrame::StereoPictureFrame (string mxf_path, int n)
  81 {
  82         ASDCP::JP2K::MXFSReader reader;
  83         if (ASDCP_FAILURE (reader.OpenRead (mxf_path.c_str()))) {
  84                 throw FileError ("could not open MXF file for reading", mxf_path);
  85         }
  86
  87         /* XXX: unfortunate guesswork on this buffer size */
  88         _buffer = new ASDCP::JP2K::SFrameBuffer (4 * Kumu::Megabyte);
  89
  90         if (ASDCP_FAILURE (reader.ReadFrame (n, *_buffer))) {
  91                 throw DCPReadError ("could not read video frame");
  92         }
  93 }
  94
  95 StereoPictureFrame::~StereoPictureFrame ()
  96 {
  97         delete _buffer;
  98 }
  99
 100 /** @param reduce a factor by which to reduce the resolution
 101  *  of the image, expressed as a power of two (pass 0 for no
 102  *  reduction).
 103  *
 104  *  @param eye Eye to return (EYE_LEFT or EYE_RIGHT).
 105  *
 106  *  @return An ARGB representation of one of the eyes (left or right)
 107  *  of this frame.  This is ARGB in the Cairo sense, so that each
 108  *  pixel takes up 4 bytes; the first byte is blue, second green,
 109  *  third red and fourth alpha (always 255).
 110  *
 111  */
 112 shared_ptr<ARGBFrame>
 113 StereoPictureFrame::argb_frame (Eye eye, int reduce) const
 114 {
 115         opj_image_t* xyz_frame = 0;
 116         switch (eye) {
 117         case LEFT:
 118                 xyz_frame = decompress_j2k (const_cast<uint8_t*> (_buffer->Left.RoData()), _buffer->Left.Size(), reduce);
 119                 break;
 120         case RIGHT:
 121                 xyz_frame = decompress_j2k (const_cast<uint8_t*> (_buffer->Right.RoData()), _buffer->Right.Size(), reduce);
 122                 break;
 123         }
 124
 125         assert (xyz_frame->numcomps == 3);
 126         shared_ptr<ARGBFrame> f = xyz_to_rgb (xyz_frame);
 127         opj_image_destroy (xyz_frame);
 128         return f;
 129 }