X-Git-Url: https://git.decadent.org.uk/gitweb/?p=videolink.git;a=blobdiff_plain;f=webdvd.cpp;h=5fd5e6ff8774a34bbb70345a2c8d32e5af05da3b;hp=eb2a9b2743d28a2a59bdd369af6ab072c25ae16f;hb=e1c65f7da2e82bcd2e61886ae5b329a7211ba124;hpb=f95a8047fecc9bdde132b6b91a8c531febaefa65 diff --git a/webdvd.cpp b/webdvd.cpp index eb2a9b2..5fd5e6f 100644 --- a/webdvd.cpp +++ b/webdvd.cpp @@ -1,4 +1,4 @@ -// Copyright 2005 Ben Hutchings . +// Copyright 2005-6 Ben Hutchings . // See the file "COPYING" for licence details. #include @@ -53,18 +53,29 @@ #include "browser_widget.hpp" #include "child_iterator.hpp" #include "dvd.hpp" +#include "generate_dvd.hpp" #include "link_iterator.hpp" +#include "null_prompt_service.hpp" #include "pixbufs.hpp" #include "style_sheets.hpp" #include "temp_file.hpp" #include "video.hpp" #include "x_frame_buffer.hpp" +#include "xml_utils.hpp" #include "xpcom_support.hpp" using xpcom_support::check; namespace { + // We can try using any of these encoders to convert PNG to MPEG. + enum mpeg_encoder + { + mpeg_encoder_ffmpeg, // ffmpeg - doesn't work yet + mpeg_encoder_mjpegtools_old, // mjpegtools before version 1.8 + mpeg_encoder_mjpegtools_new // mjpegtools from version 1.8 + }; + struct rectangle { int left, top; // inclusive @@ -143,96 +154,16 @@ namespace } - std::string xml_escape(const std::string & str) - { - std::string result; - std::size_t begin = 0; - - for (;;) - { - std::size_t end = str.find_first_of("\"&'<>", begin); - result.append(str, begin, end - begin); - if (end == std::string::npos) - return result; - - const char * entity = NULL; - switch (str[end]) - { - case '"': entity = """; break; - case '&': entity = "&"; break; - case '\'': entity = "'"; break; - case '<': entity = "<"; break; - case '>': entity = ">"; break; - } - assert(entity); - result.append(entity); - - begin = end + 1; - } - } - - - struct dvd_contents - { - enum pgc_type { unknown_pgc, menu_pgc, title_pgc }; - - struct pgc_ref - { - explicit pgc_ref(pgc_type type = unknown_pgc, - int index = -1, - int sub_index = 0) - : type(type), index(index), sub_index(sub_index) - {} - bool operator==(const pgc_ref & other) const - { - return type == other.type && index == other.index; - } - bool operator!=(const pgc_ref & other) const - { - return !(*this == other); - } - - pgc_type type; // Menu or title reference? - unsigned index; // Menu or title index (within resp. vector) - unsigned sub_index; // Button or chapter number (1-based; 0 if - // unspecified; not compared!) - }; - - struct menu - { - menu() - : vob_temp(new temp_file("webdvd-vob-")) - { - vob_temp->close(); - } - - boost::shared_ptr vob_temp; - std::vector entries; - }; - - struct title - { - explicit title(const std::string & vob_list) - : vob_list(vob_list) - {} - - std::string vob_list; - }; - - std::vector menus; - std::vector titles; - }; - - void generate_dvd(const dvd_contents & contents, - const std::string & output_dir); - class webdvd_window : public Gtk::Window { public: webdvd_window( const video::frame_params & frame_params, const std::string & main_page_uri, - const std::string & output_dir); + const std::string & output_dir, + mpeg_encoder encoder); + + bool is_finished() const; private: dvd_contents::pgc_ref add_menu(const std::string & uri); @@ -251,6 +182,7 @@ namespace video::frame_params frame_params_; std::string output_dir_; + mpeg_encoder encoder_; browser_widget browser_widget_; nsCOMPtr<nsIStyleSheet> stylesheet_; @@ -265,18 +197,23 @@ namespace std::auto_ptr<temp_file> background_temp_; struct page_state; std::auto_ptr<page_state> page_state_; + + bool finished_; }; webdvd_window::webdvd_window( const video::frame_params & frame_params, const std::string & main_page_uri, - const std::string & output_dir) + const std::string & output_dir, + mpeg_encoder encoder) : frame_params_(frame_params), output_dir_(output_dir), + encoder_(encoder), stylesheet_(load_css("file://" WEBDVD_LIB_DIR "/webdvd.css")), pending_window_update_(false), pending_req_count_(0), - have_tweaked_page_(false) + have_tweaked_page_(false), + finished_(false) { set_size_request(frame_params_.width, frame_params_.height); set_resizable(false); @@ -290,6 +227,11 @@ namespace load_next_page(); } + bool webdvd_window::is_finished() const + { + return finished_; + } + dvd_contents::pgc_ref webdvd_window::add_menu(const std::string & uri) { dvd_contents::pgc_ref next_menu(dvd_contents::menu_pgc, @@ -323,32 +265,30 @@ namespace else { Glib::ustring hostname; - std::string filename(Glib::filename_from_uri(uri, hostname)); + std::string path(Glib::filename_from_uri(uri, hostname)); // FIXME: Should check the hostname - std::string vob_list; + vob_list list; // Store a reference to a linked VOB file, or the contents // of a linked VOB list file. - if (filename.compare(filename.size() - 4, 4, ".vob") == 0) + if (path.compare(path.size() - 4, 4, ".vob") == 0) { - if (!Glib::file_test(filename, Glib::FILE_TEST_IS_REGULAR)) + if (!Glib::file_test(path, Glib::FILE_TEST_IS_REGULAR)) throw std::runtime_error( - filename + " is missing or not a regular file"); - vob_list - .append("<vob file='") - .append(xml_escape(filename)) - .append("'/>\n"); + path + " is missing or not a regular file"); + vob_ref ref; + ref.file = path; + list.push_back(ref); } else { - assert(filename.compare(filename.size() - 8, 8, ".voblist") - == 0); - // TODO: Validate the file contents - vob_list.assign(Glib::file_get_contents(filename)); + assert(path.compare(path.size() - 8, 8, ".voblist") == 0); + read_vob_list(path).swap(list); } - contents_.titles.push_back(dvd_contents::title(vob_list)); + contents_.titles.resize(contents_.titles.size() + 1); + contents_.titles.back().swap(list); return next_title; } } @@ -439,7 +379,10 @@ namespace try { if (!process_page()) + { + finished_ = true; Gtk::Main::quit(); + } } catch (std::exception & e) { @@ -804,254 +747,61 @@ namespace { std::ostringstream command_stream; - command_stream << "pngtopnm " - << background_temp_->get_name() - << " | ppmtoy4m -v0 -n1 -F" - << frame_params_.rate_numer - << ":" << frame_params_.rate_denom - << " -A" << frame_params_.pixel_ratio_width - << ":" << frame_params_.pixel_ratio_height - << (" -Ip -S420_mpeg2" - " | mpeg2enc -v0 -f8 -a2 -o/dev/stdout" - " | mplex -v0 -f8 -o/dev/stdout /dev/stdin" - " | spumux -v0 -mdvd ") - << state->spumux_temp.get_name() - << " > " - << contents_.menus[menu_num].vob_temp->get_name(); - std::string command(command_stream.str()); - const char * argv[] = { - "/bin/sh", "-c", command.c_str(), 0 - }; - std::cout << "running " << argv[2] << std::endl; - int command_result; - Glib::spawn_sync(".", - Glib::ArrayHandle<std::string>( - argv, sizeof(argv)/sizeof(argv[0]), - Glib::OWNERSHIP_NONE), - Glib::SPAWN_STDOUT_TO_DEV_NULL, - SigC::Slot0<void>(), - 0, 0, - &command_result); - if (command_result != 0) - throw std::runtime_error("spumux pipeline failed"); - } - } - - void generate_dvd(const dvd_contents & contents, - const std::string & output_dir) - { - temp_file temp("webdvd-dvdauthor-"); - temp.close(); - std::ofstream file(temp.get_name().c_str()); - - // We generate code that uses registers in the following way: - // - // g0: scratch - // g1: current location - // g12: location that last jumped to a video - // - // All locations are divided into two bitfields: the least - // significant 10 bits are a page/menu number and the most - // significant 6 bits are a link/button number, and numbering - // starts at 1, not 0. This is done for compatibility with - // the encoding of the s8 (button) register. - // - static const int button_mult = dvd::reg_s8_button_mult; - static const int menu_mask = button_mult - 1; - static const int button_mask = (1 << dvd::reg_bits) - button_mult; - - file << - "<dvdauthor>\n" - " <vmgm>\n" - " <menus>\n"; - - for (unsigned menu_num = 0; - menu_num != contents.menus.size(); - ++menu_num) - { - const dvd_contents::menu & menu = contents.menus[menu_num]; - - if (menu_num == 0) + if (encoder_ == mpeg_encoder_ffmpeg) { - // This is the first (title) menu, displayed when the - // disc is first played. - file << - " <pgc entry='title'>\n" - " <pre>\n" - // Initialise the current location if it is not set - // (all general registers are initially 0). - " if (g1 eq 0)\n" - " g1 = " << 1 + button_mult << ";\n"; + command_stream + << "ffmpeg" + << " -f image2 -vcodec png -i " + << background_temp_->get_name() + << " -target " << frame_params_.name << "-dvd" + << " -vcodec mpeg2video -an -y /dev/stdout" + << " | spumux -v0 -mdvd " << state->spumux_temp.get_name() + << " > " << contents_.menus[menu_num].vob_temp->get_name(); } else { - file << - " <pgc>\n" - " <pre>\n"; - } - - // When a title finishes or the user presses the menu - // button, this always jumps to the titleset's root menu. - // We want to return the user to the last menu they used. - // So we arrange for each titleset's root menu to return - // to the vmgm title menu and then dispatch from there to - // whatever the correct menu is. We determine the correct - // menu by looking at the menu part of g1. - - file << " g0 = g1 & " << menu_mask << ";\n"; - - // There is a limit of 128 VM instructions in each PGC. - // Therefore in each menu's <pre> section we generate - // jumps to menus with numbers greater by 512, 256, 128, - // ..., 1 where (a) such a menu exists, (b) this menu - // number is divisible by twice that increment and (c) the - // correct menu is that or a later menu. Thus each menu - // has at most 10 such conditional jumps and is reachable - // by at most 10 jumps from the title menu. This chain of - // jumps might take too long on some players; this has yet - // to be investigated. - - for (std::size_t menu_incr = (menu_mask + 1) / 2; - menu_incr != 0; - menu_incr /= 2) - { - if (menu_num + menu_incr < contents.menus.size() - && (menu_num & (menu_incr * 2 - 1)) == 0) - { - file << - " if (g0 ge " << 1 + menu_num + menu_incr - << ")\n" - " jump menu " << 1 + menu_num + menu_incr - << ";\n"; - } - } - - file << - // Highlight the appropriate button. - " s8 = g1 & " << button_mask << ";\n" - " </pre>\n" - " <vob file='" << menu.vob_temp->get_name() << "'/>\n"; - - for (unsigned button_num = 0; - button_num != menu.entries.size(); - ++button_num) - { - const dvd_contents::pgc_ref & target = - menu.entries[button_num]; - - file << " <button> "; - - if (target.type == dvd_contents::menu_pgc) - { - unsigned target_button_num; - - if (target.sub_index) - { - target_button_num = target.sub_index; - } - else - { - // Look for a button on the new menu that links - // back to this one. If there is one, set that to - // be the highlighted button; otherwise, use the - // first button. - const std::vector<dvd_contents::pgc_ref> & - target_menu_entries = - contents.menus[target.index].entries; - dvd_contents::pgc_ref this_pgc(dvd_contents::menu_pgc, - menu_num); - target_button_num = target_menu_entries.size(); - while (target_button_num != 0 - && (target_menu_entries[--target_button_num] - != this_pgc)) - ; - target_button_num += 1; - } - - file << "g1 = " - << (1 + target.index - + target_button_num * button_mult) - << "; jump menu " << 1 + target.index << ";"; - } + assert(encoder_ == mpeg_encoder_mjpegtools_old + || encoder_ == mpeg_encoder_mjpegtools_new); + command_stream + << "pngtopnm " + << background_temp_->get_name() + << " | ppmtoy4m -v0 -n1 -F" + << frame_params_.rate_numer + << ":" << frame_params_.rate_denom + << " -A" << frame_params_.pixel_ratio_width + << ":" << frame_params_.pixel_ratio_height + << " -Ip "; + // The chroma subsampling keywords changed between + // versions 1.6.2 and 1.8 of mjpegtools. There is no + // keyword that works with both. + if (encoder_ == mpeg_encoder_mjpegtools_old) + command_stream << "-S420_mpeg2"; else - { - assert(target.type == dvd_contents::title_pgc); - - file << "g1 = " - << 1 + menu_num + (1 + button_num) * button_mult - << "; jump title " - << 1 + target.index; - if (target.sub_index) - file << " chapter " << target.sub_index; - file << ";"; - } - - file << " </button>\n"; + command_stream << "-S420mpeg2"; + command_stream + << (" | mpeg2enc -v0 -f8 -a2 -o/dev/stdout" + " | mplex -v0 -f8 -o/dev/stdout /dev/stdin" + " | spumux -v0 -mdvd ") + << state->spumux_temp.get_name() + << " > " + << contents_.menus[menu_num].vob_temp->get_name(); } - - file << " </pgc>\n"; - } - - file << - " </menus>\n" - " </vmgm>\n"; - - // Generate a titleset for each title. This appears to make - // jumping to titles a whole lot simpler (but limits us to 99 - // titles). - for (unsigned title_num = 0; - title_num != contents.titles.size(); - ++title_num) - { - file << - " <titleset>\n" - // Generate a dummy menu so that the menu button on the - // remote control will work. - " <menus>\n" - " <pgc entry='root'>\n" - " <pre> jump vmgm menu; </pre>\n" - " </pgc>\n" - " </menus>\n" - " <titles>\n" - " <pgc>\n" - // Record calling location. - " <pre> g12 = g1; </pre>\n" - << contents.titles[title_num].vob_list << - // If the menu location has not been changed during - // the title, set the location to be the following - // button in the menu. In any case, return to some - // menu. - " <post> if (g1 eq g12) g1 = g1 + " << button_mult - << "; call menu; </post>\n" - " </pgc>\n" - " </titles>\n" - " </titleset>\n"; - } - - file << - "</dvdauthor>\n"; - - file.close(); - - { + std::string command(command_stream.str()); const char * argv[] = { - "dvdauthor", - "-o", output_dir.c_str(), - "-x", temp.get_name().c_str(), - 0 + "/bin/sh", "-c", command.c_str(), 0 }; + std::cout << "running " << argv[2] << std::endl; int command_result; Glib::spawn_sync(".", Glib::ArrayHandle<std::string>( argv, sizeof(argv)/sizeof(argv[0]), Glib::OWNERSHIP_NONE), - Glib::SPAWN_SEARCH_PATH - | Glib::SPAWN_STDOUT_TO_DEV_NULL, + Glib::SPAWN_STDOUT_TO_DEV_NULL, SigC::Slot0<void>(), 0, 0, &command_result); if (command_result != 0) - throw std::runtime_error("dvdauthor failed"); + throw std::runtime_error("spumux pipeline failed"); } } @@ -1080,9 +830,11 @@ namespace void print_usage(std::ostream & stream, const char * command_name) { - stream << "Usage: " << command_name - << (" [gtk-options] [--video-std std-name]" - " [--preview] menu-url [output-dir]\n"); + stream << + "Usage: " << command_name << " [gtk-options] [--preview]\n" + " [--video-std {ntsc|pal|secam}]\n" + " [--encoder {mjpegtools|mjpegtools-old}]\n" + " menu-url [output-dir]\n"); } void set_browser_preferences() @@ -1120,6 +872,12 @@ namespace } // namespace +void fatal_error(const std::string & message) +{ + std::cerr << "Fatal error: " << message << "\n"; + Gtk::Main::quit(); +} + int main(int argc, char ** argv) { try @@ -1128,6 +886,7 @@ int main(int argc, char ** argv) bool preview_mode = false; std::string menu_url; std::string output_dir; + mpeg_encoder encoder = mpeg_encoder_mjpegtools_new; // Do initial option parsing. We have to do this before // letting Gtk parse the arguments since we may need to spawn @@ -1199,6 +958,40 @@ int main(int argc, char ** argv) { argi += 2; } + else if (std::strcmp(argv[argi], "--save-temps") == 0) + { + temp_file::keep_all(true); + argi += 1; + } + else if (std::strcmp(argv[argi], "--encoder") == 0) + { + if (argi + 1 == argc) + { + std::cerr << "Missing argument to --encoder\n"; + print_usage(std::cerr, argv[0]); + return EXIT_FAILURE; + } + if (std::strcmp(argv[argi + 1], "ffmpeg") == 0) + { + encoder = mpeg_encoder_ffmpeg; + } + else if (std::strcmp(argv[argi + 1], "mjpegtools-old") == 0) + { + encoder = mpeg_encoder_mjpegtools_old; + } + else if (std::strcmp(argv[argi + 1], "mjpegtools") == 0 + || std::strcmp(argv[argi + 1], "mjpegtools-new") == 0) + { + encoder = mpeg_encoder_mjpegtools_new; + } + else + { + std::cerr << "Invalid argument to --encoder\n"; + print_usage(std::cerr, argv[0]); + return EXIT_FAILURE; + } + argi += 2; + } else if (argv[argi][0] == '-') { std::cerr << "Invalid option: " << argv[argi] << "\n"; @@ -1237,16 +1030,20 @@ int main(int argc, char ** argv) // Initialise Mozilla browser_widget::initialiser browser_init; set_browser_preferences(); + if (!preview_mode) + null_prompt_service::install(); // Run the browser/converter - webdvd_window window(frame_params, menu_url, output_dir); + webdvd_window window(frame_params, menu_url, output_dir, encoder); Gtk::Main::run(window); + + return ((preview_mode || window.is_finished()) + ? EXIT_SUCCESS + : EXIT_FAILURE); } catch (std::exception & e) { std::cerr << "Fatal error: " << e.what() << "\n"; return EXIT_FAILURE; } - - return EXIT_SUCCESS; }