examples/read_dcp.cc

   1 /*
   2     Copyright (C) 2012-2021 Carl Hetherington <cth@carlh.net>
   3
   4     This file is part of libdcp.
   5
   6     libdcp 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     libdcp 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 libdcp.  If not, see <http://www.gnu.org/licenses/>.
  18 */
  19
  20 /* If you are using an installed libdcp, these #includes would need to be changed to
  21 #include <dcp/dcp.h>
  22 #include <dcp/picture_asset.h>
  23 ... etc. ...
  24 */
  25
  26 #include "dcp.h"
  27 #include "cpl.h"
  28 #include "reel.h"
  29 #include "reel_picture_asset.h"
  30 #include "mono_picture_frame.h"
  31 #include "mono_picture_asset.h"
  32 #include "mono_picture_asset_reader.h"
  33 #include "stereo_picture_asset.h"
  34 #include "sound_asset.h"
  35 #include "subtitle_asset.h"
  36 #include "openjpeg_image.h"
  37 #include "colour_conversion.h"
  38 #include "rgb_xyz.h"
  39 /* This DISABLE/ENABLE pair is just to ignore some warnings from Magick++.h; they
  40  * can be removed.
  41  */
  42 LIBDCP_DISABLE_WARNINGS
  43 #include <Magick++.h>
  44 LIBDCP_ENABLE_WARNINGS
  45
  46 /** @file examples/read_dcp.cc
  47  *  @brief Shows how to read a DCP.
  48  */
  49
  50 int
  51 main ()
  52 {
  53         /* Unless libdcp has been installed, you need to pass a path containing the "tags" folder here */
  54         dcp::init(boost::filesystem::path("."));
  55         Magick::InitializeMagick(nullptr);
  56
  57         /* Create a DCP, specifying where our existing data is */
  58         dcp::DCP dcp("/home/carl/Test_DCP");
  59         /* Read the DCP to find out about it */
  60         dcp.read();
  61
  62         if (dcp.all_encrypted()) {
  63                 std::cout << "DCP is encrypted.\n";
  64         } else if (dcp.any_encrypted()) {
  65                 std::cout << "DCP is partially encrypted.\n";
  66         } else {
  67                 std::cout << "DCP is not encrypted.\n";
  68         }
  69
  70         std::cout << "DCP has " << dcp.cpls().size() << " CPLs.\n";
  71         auto assets = dcp.assets();
  72         std::cout << "DCP has " << assets.size() << " assets.\n";
  73         for (auto i: assets) {
  74                 if (std::dynamic_pointer_cast<dcp::MonoPictureAsset>(i)) {
  75                         std::cout << "2D picture\n";
  76                 } else if (std::dynamic_pointer_cast<dcp::StereoPictureAsset>(i)) {
  77                         std::cout << "3D picture\n";
  78                 } else if (std::dynamic_pointer_cast<dcp::SoundAsset>(i)) {
  79                         std::cout << "Sound\n";
  80                 } else if (std::dynamic_pointer_cast<dcp::SubtitleAsset>(i)) {
  81                         std::cout << "Subtitle\n";
  82                 } else if (std::dynamic_pointer_cast<dcp::CPL>(i)) {
  83                         std::cout << "CPL\n";
  84                 }
  85                 std::cout << "\t" << i->file()->leaf().string() << "\n";
  86         }
  87
  88         /* Take the first CPL */
  89         auto cpl = dcp.cpls().front();
  90
  91         /* Get the picture asset in the first reel */
  92         auto picture_asset = std::dynamic_pointer_cast<dcp::MonoPictureAsset>(
  93                 cpl->reels()[0]->main_picture()->asset()
  94                 );
  95
  96         /* Get a reader for it */
  97         auto picture_asset_reader = picture_asset->start_read();
  98
  99         /* Get the 1000th frame of it */
 100         auto picture_frame_j2k = picture_asset_reader->get_frame(999);
 101
 102         /* Get the frame as an XYZ image */
 103         auto picture_image_xyz = picture_frame_j2k->xyz_image();
 104
 105         /* Convert to ARGB */
 106         std::vector<uint8_t> rgba(picture_image_xyz->size().width * picture_image_xyz->size().height * 4);
 107         dcp::xyz_to_rgba(picture_image_xyz, dcp::ColourConversion::srgb_to_xyz(), rgba.data(), picture_image_xyz->size().width * 4);
 108
 109         Magick::Image image(picture_image_xyz->size().width, picture_image_xyz->size().height, "BGRA", Magick::CharPixel, rgba.data());
 110         image.write("frame.png");
 111
 112         return 0;
 113 }