src/lib/dcp_decoder.cc

   1 /*
   2     Copyright (C) 2014 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 <dcp/dcp.h>
  21 #include <dcp/cpl.h>
  22 #include <dcp/reel.h>
  23 #include <dcp/mono_picture_mxf.h>
  24 #include <dcp/stereo_picture_mxf.h>
  25 #include <dcp/reel_picture_asset.h>
  26 #include <dcp/reel_sound_asset.h>
  27 #include <dcp/mono_picture_frame.h>
  28 #include <dcp/stereo_picture_frame.h>
  29 #include <dcp/sound_frame.h>
  30 #include "dcp_decoder.h"
  31 #include "dcp_content.h"
  32 #include "j2k_image_proxy.h"
  33 #include "image.h"
  34 #include "config.h"
  35
  36 using std::list;
  37 using std::cout;
  38 using boost::shared_ptr;
  39 using boost::dynamic_pointer_cast;
  40
  41 DCPDecoder::DCPDecoder (shared_ptr<const DCPContent> c, shared_ptr<Log> log)
  42         : VideoDecoder (c)
  43         , AudioDecoder (c)
  44         , SubtitleDecoder (c)
  45         , _log (log)
  46         , _dcp_content (c)
  47 {
  48         dcp::DCP dcp (c->directory ());
  49         dcp.read ();
  50         if (c->kdm ()) {
  51                 dcp.add (dcp::DecryptedKDM (c->kdm().get (), Config::instance()->decryption_private_key ()));
  52         }
  53         assert (dcp.cpls().size() == 1);
  54         _reels = dcp.cpls().front()->reels ();
  55         _reel = _reels.begin ();
  56 }
  57
  58 bool
  59 DCPDecoder::pass ()
  60 {
  61         if (_reel == _reels.end () || !_dcp_content->can_be_played ()) {
  62                 return true;
  63         }
  64
  65         float const vfr = _dcp_content->video_frame_rate ();
  66         int64_t const frame = _next.frames (vfr);
  67
  68         if ((*_reel)->main_picture ()) {
  69                 shared_ptr<dcp::PictureMXF> mxf = (*_reel)->main_picture()->mxf ();
  70                 shared_ptr<dcp::MonoPictureMXF> mono = dynamic_pointer_cast<dcp::MonoPictureMXF> (mxf);
  71                 shared_ptr<dcp::StereoPictureMXF> stereo = dynamic_pointer_cast<dcp::StereoPictureMXF> (mxf);
  72                 int64_t const entry_point = (*_reel)->main_picture()->entry_point ();
  73                 if (mono) {
  74                         video (shared_ptr<ImageProxy> (new J2KImageProxy (mono->get_frame (entry_point + frame), mxf->size(), _log)), frame);
  75                 } else {
  76                         video (
  77                                 shared_ptr<ImageProxy> (new J2KImageProxy (stereo->get_frame (entry_point + frame), mxf->size(), dcp::EYE_LEFT, _log)),
  78                                 frame
  79                                 );
  80
  81                         video (
  82                                 shared_ptr<ImageProxy> (new J2KImageProxy (stereo->get_frame (entry_point + frame), mxf->size(), dcp::EYE_RIGHT, _log)),
  83                                 frame
  84                                 );
  85                 }
  86         }
  87
  88         if ((*_reel)->main_sound ()) {
  89                 int64_t const entry_point = (*_reel)->main_sound()->entry_point ();
  90                 shared_ptr<const dcp::SoundFrame> sf = (*_reel)->main_sound()->mxf()->get_frame (entry_point + frame);
  91                 uint8_t const * from = sf->data ();
  92
  93                 int const channels = _dcp_content->audio_channels ();
  94                 int const frames = sf->size() / (3 * channels);
  95                 shared_ptr<AudioBuffers> data (new AudioBuffers (channels, frames));
  96                 for (int i = 0; i < frames; ++i) {
  97                         for (int j = 0; j < channels; ++j) {
  98                                 data->data()[j][i] = float (from[0] | (from[1] << 8) | (from[2] << 16)) / (1 << 23);
  99                                 from += 3;
 100                         }
 101                 }
 102
 103                 audio (data, _next);
 104         }
 105
 106         /* XXX: subtitle */
 107
 108         _next += ContentTime::from_frames (1, vfr);
 109
 110         if ((*_reel)->main_picture ()) {
 111                 if (_next.frames (vfr) >= (*_reel)->main_picture()->duration()) {
 112                         ++_reel;
 113                 }
 114         }
 115
 116         return false;
 117 }
 118
 119 void
 120 DCPDecoder::seek (ContentTime t, bool accurate)
 121 {
 122         VideoDecoder::seek (t, accurate);
 123         AudioDecoder::seek (t, accurate);
 124         SubtitleDecoder::seek (t, accurate);
 125
 126         _reel = _reels.begin ();
 127         while (_reel != _reels.end() && t >= ContentTime::from_frames ((*_reel)->main_picture()->duration(), _dcp_content->video_frame_rate ())) {
 128                 t -= ContentTime::from_frames ((*_reel)->main_picture()->duration(), _dcp_content->video_frame_rate ());
 129                 ++_reel;
 130         }
 131
 132         _next = t;
 133 }
 134
 135
 136 list<ContentTimePeriod>
 137 DCPDecoder::subtitles_during (ContentTimePeriod, bool starting) const
 138 {
 139         return list<ContentTimePeriod> ();
 140 }