X-Git-Url: https://main.carlh.net/gitweb/?p=dcpomatic.git;a=blobdiff_plain;f=src%2Flib%2Fcontent_factory.cc;h=0e8b04eda21c48b9b22f975a2e7e4b126485a417;hp=16340adb43b0e74f8f14d4b2bc039e258d8e58a9;hb=6ef1fc5f40567650ca9ef2b7644e4fdd97640ae6;hpb=3b48d5494c3cae7743d283203f5c8021860ab81f diff --git a/src/lib/content_factory.cc b/src/lib/content_factory.cc index 16340adb4..0e8b04eda 100644 --- a/src/lib/content_factory.cc +++ b/src/lib/content_factory.cc @@ -1,19 +1,20 @@ /* - Copyright (C) 2013-2014 Carl Hetherington + Copyright (C) 2013-2016 Carl Hetherington - This program is free software; you can redistribute it and/or modify + This file is part of DCP-o-matic. + + DCP-o-matic 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, + DCP-o-matic 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 DCP-o-matic. If not, see . */ @@ -21,73 +22,172 @@ * @brief Methods to create content objects. */ -#include #include "ffmpeg_content.h" +#include "audio_content.h" #include "image_content.h" -#include "sndfile_content.h" -#include "subrip_content.h" +#include "atmos_mxf_content.h" +#include "string_text_file_content.h" #include "dcp_content.h" #include "dcp_subtitle_content.h" #include "util.h" +#include "ffmpeg_audio_stream.h" +#include "video_mxf_content.h" +#include "film.h" +#include "log_entry.h" +#include "dcpomatic_log.h" +#include "log.h" +#include "compose.hpp" +#include +#include +#include + +#include "i18n.h" using std::string; using std::list; using boost::shared_ptr; +using boost::optional; /** Create a Content object from an XML node. - * @param film Film that the content will be in. * @param node XML description. * @param version XML state version. * @param notes A list to which is added descriptions of any non-critial warnings / messages. * @return Content object, or 0 if no content was recognised in the XML. */ shared_ptr -content_factory (shared_ptr film, cxml::NodePtr node, int version, list& notes) +content_factory (cxml::ConstNodePtr node, int version, list& notes) { string const type = node->string_child ("Type"); boost::shared_ptr content; - + if (type == "FFmpeg") { - content.reset (new FFmpegContent (film, node, version, notes)); + /* SndfileContent is now handled by the FFmpeg code rather than by + separate libsndfile-based code. + */ + content.reset (new FFmpegContent (node, version, notes)); } else if (type == "Image") { - content.reset (new ImageContent (film, node, version)); + content.reset (new ImageContent (node, version)); } else if (type == "Sndfile") { - content.reset (new SndfileContent (film, node, version)); - } else if (type == "SubRip") { - content.reset (new SubRipContent (film, node, version)); + /* SndfileContent is now handled by the FFmpeg code rather than by + separate libsndfile-based code. + */ + content.reset (new FFmpegContent (node, version, notes)); + + content->audio->set_stream ( + AudioStreamPtr ( + new FFmpegAudioStream ( + "Stream", 0, + node->number_child ("AudioFrameRate"), + node->number_child ("AudioLength"), + AudioMapping (node->node_child ("AudioMapping"), version) + ) + ) + ); + + } else if (type == "SubRip" || type == "TextSubtitle") { + content.reset (new StringTextFileContent (node, version)); } else if (type == "DCP") { - content.reset (new DCPContent (film, node, version)); + content.reset (new DCPContent (node, version)); } else if (type == "DCPSubtitle") { - content.reset (new DCPSubtitleContent (film, node, version)); + content.reset (new DCPSubtitleContent (node, version)); + } else if (type == "VideoMXF") { + content.reset (new VideoMXFContent (node, version)); + } else if (type == "AtmosMXF") { + content.reset (new AtmosMXFContent (node, version)); } return content; } -/** Create a Content object from a file, depending on its extension. - * @param film Film that the content will be in. - * @param path File's path. - * @return Content object. +/** Create some Content objects from a file or directory. + * @param path File or directory. + * @return Content objects. */ -shared_ptr -content_factory (shared_ptr film, boost::filesystem::path path) +list > +content_factory (boost::filesystem::path path) { - shared_ptr content; - - string ext = path.extension().string (); - transform (ext.begin(), ext.end(), ext.begin(), ::tolower); - - if (valid_image_file (path)) { - content.reset (new ImageContent (film, path)); - } else if (SndfileContent::valid_file (path)) { - content.reset (new SndfileContent (film, path)); - } else if (ext == ".srt") { - content.reset (new SubRipContent (film, path)); - } else if (ext == ".xml") { - content.reset (new DCPSubtitleContent (film, path)); + list > content; + + if (boost::filesystem::is_directory (path)) { + + LOG_GENERAL ("Look in directory %1", path); + + if (boost::filesystem::is_empty (path)) { + return content; + } + + /* See if this is a set of images or a set of sound files */ + + int image_files = 0; + int sound_files = 0; + int read = 0; + for (boost::filesystem::directory_iterator i(path); i != boost::filesystem::directory_iterator() && read < 10; ++i) { + + LOG_GENERAL ("Checking file %1", i->path()); + + if (boost::starts_with (i->path().leaf().string(), ".")) { + /* We ignore hidden files */ + LOG_GENERAL ("Ignored %1 (starts with .)", i->path()); + continue; + } + + if (!boost::filesystem::is_regular_file(i->path())) { + /* Ignore things which aren't files (probably directories) */ + LOG_GENERAL ("Ignored %1 (not a regular file)", i->path()); + continue; + } + + if (valid_image_file (i->path ())) { + ++image_files; + } + + if (valid_sound_file (i->path ())) { + ++sound_files; + } + + ++read; + } + + if (image_files > 0 && sound_files == 0) { + content.push_back (shared_ptr (new ImageContent(path))); + } else if (image_files == 0 && sound_files > 0) { + for (boost::filesystem::directory_iterator i(path); i != boost::filesystem::directory_iterator(); ++i) { + content.push_back (shared_ptr (new FFmpegContent(i->path()))); + } + } + } else { - content.reset (new FFmpegContent (film, path)); + + shared_ptr single; + + string ext = path.extension().string (); + transform (ext.begin(), ext.end(), ext.begin(), ::tolower); + + if (valid_image_file (path)) { + single.reset (new ImageContent(path)); + } else if (ext == ".srt" || ext == ".ssa" || ext == ".ass" || ext == ".stl") { + single.reset (new StringTextFileContent(path)); + } else if (ext == ".xml") { + cxml::Document doc; + doc.read_file (path); + if (doc.root_name() == "DCinemaSecurityMessage") { + throw KDMAsContentError (); + } + single.reset (new DCPSubtitleContent(path)); + } else if (ext == ".mxf" && dcp::SMPTESubtitleAsset::valid_mxf (path)) { + single.reset (new DCPSubtitleContent(path)); + } else if (ext == ".mxf" && VideoMXFContent::valid_mxf (path)) { + single.reset (new VideoMXFContent(path)); + } else if (ext == ".mxf" && AtmosMXFContent::valid_mxf (path)) { + single.reset (new AtmosMXFContent(path)); + } + + if (!single) { + single.reset (new FFmpegContent(path)); + } + + content.push_back (single); } return content;