Formatting tidying; use vector instead of scoped_array.

[libdcp.git] / examples / read_dcp.cc

diff --git a/examples/read_dcp.cc b/examples/read_dcp.cc
index 43f5f0be4bc737d1f69d8c98aa5e08a323028958..f099409f81996acc5349349aac7478243b2ebe0c 100644 (file)
--- a/examples/read_dcp.cc
+++ b/examples/read_dcp.cc
@@ -1,30 +1,47 @@
 /*
-    Copyright (C) 2012-2014 Carl Hetherington <cth@carlh.net>
+    Copyright (C) 2012-2021 Carl Hetherington <cth@carlh.net>
 
-    This program is free software; you can redistribute it and/or modify
+    This file is part of libdcp.
+
+    libdcp 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,
+    libdcp 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.
-
+    along with libdcp.  If not, see <http://www.gnu.org/licenses/>.
 */
 
 /* If you are using an installed libdcp, these #includes would need to be changed to
-#include <libdcp/dcp.h>
-#include <libdcp/cpl.h>
-#include <libdcp/mono_picture_asset.h>
+#include <dcp/dcp.h>
+#include <dcp/picture_asset.h>
 ... etc. ...
 */
 
 #include "dcp.h"
+#include "cpl.h"
+#include "reel.h"
+#include "reel_picture_asset.h"
+#include "mono_picture_frame.h"
+#include "mono_picture_asset.h"
+#include "mono_picture_asset_reader.h"
+#include "stereo_picture_asset.h"
+#include "sound_asset.h"
+#include "subtitle_asset.h"
+#include "openjpeg_image.h"
+#include "colour_conversion.h"
+#include "rgb_xyz.h"
+/* This DISABLE/ENABLE pair is just to ignore some warnings from Magick++.h; they
+ * can be removed.
+ */
+LIBDCP_DISABLE_WARNINGS
+#include <Magick++.h>
+LIBDCP_ENABLE_WARNINGS
 
 /** @file examples/read_dcp.cc
  *  @brief Shows how to read a DCP.
@@ -33,19 +50,64 @@
 int
 main ()
 {
+       /* Unless libdcp has been installed, you need to pass a path containing the "tags" folder here */
+       dcp::init(boost::filesystem::path("."));
+       Magick::InitializeMagick(nullptr);
+
        /* Create a DCP, specifying where our existing data is */
-       dcp::DCP dcp ("/home/carl/diagonal.com/APPASSIONATA_TLR_F_UK-DEFR_CH_51_2K_LOK_20121115_DGL_OV");
+       dcp::DCP dcp("/home/carl/Test_DCP");
        /* Read the DCP to find out about it */
-       dcp.read ();
+       dcp.read();
 
-       if (dcp.encrypted ()) {
+       if (dcp.all_encrypted()) {
                std::cout << "DCP is encrypted.\n";
+       } else if (dcp.any_encrypted()) {
+               std::cout << "DCP is partially encrypted.\n";
        } else {
                std::cout << "DCP is not encrypted.\n";
        }
 
        std::cout << "DCP has " << dcp.cpls().size() << " CPLs.\n";
-       std::cout << "DCP has " << dcp.assets().size() << " assets.\n";
+       auto assets = dcp.assets();
+       std::cout << "DCP has " << assets.size() << " assets.\n";
+       for (auto i: assets) {
+               if (std::dynamic_pointer_cast<dcp::MonoPictureAsset>(i)) {
+                       std::cout << "2D picture\n";
+               } else if (std::dynamic_pointer_cast<dcp::StereoPictureAsset>(i)) {
+                       std::cout << "3D picture\n";
+               } else if (std::dynamic_pointer_cast<dcp::SoundAsset>(i)) {
+                       std::cout << "Sound\n";
+               } else if (std::dynamic_pointer_cast<dcp::SubtitleAsset>(i)) {
+                       std::cout << "Subtitle\n";
+               } else if (std::dynamic_pointer_cast<dcp::CPL>(i)) {
+                       std::cout << "CPL\n";
+               }
+               std::cout << "\t" << i->file()->leaf().string() << "\n";
+       }
+
+       /* Take the first CPL */
+       auto cpl = dcp.cpls().front();
+
+       /* Get the picture asset in the first reel */
+       auto picture_asset = std::dynamic_pointer_cast<dcp::MonoPictureAsset>(
+               cpl->reels()[0]->main_picture()->asset()
+               );
+
+       /* Get a reader for it */
+       auto picture_asset_reader = picture_asset->start_read();
+
+       /* Get the 1000th frame of it */
+       auto picture_frame_j2k = picture_asset_reader->get_frame(999);
+
+       /* Get the frame as an XYZ image */
+       auto picture_image_xyz = picture_frame_j2k->xyz_image();
+
+       /* Convert to ARGB */
+       std::vector<uint8_t> rgba(picture_image_xyz->size().width * picture_image_xyz->size().height * 4);
+       dcp::xyz_to_rgba(picture_image_xyz, dcp::ColourConversion::srgb_to_xyz(), rgba.data(), picture_image_xyz->size().width * 4);
+
+       Magick::Image image(picture_image_xyz->size().width, picture_image_xyz->size().height, "BGRA", Magick::CharPixel, rgba.data());
+       image.write("frame.png");
 
        return 0;
 }