X-Git-Url: https://git.decadent.org.uk/gitweb/?p=videolink.git;a=blobdiff_plain;f=webdvd.cpp;h=5fd5e6ff8774a34bbb70345a2c8d32e5af05da3b;hp=ceefb7804d446614ce69676bb8e6bbc05982aca7;hb=e1c65f7da2e82bcd2e61886ae5b329a7211ba124;hpb=e8e58cc644b3df580cece15eecd2524329c43405 diff --git a/webdvd.cpp b/webdvd.cpp index ceefb78..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 @@ -42,26 +42,40 @@ #include #include // required before nsILink.h #include +#include +#include #include #include +#include #include #include -#include "browserwidget.hpp" -#include "childiterator.hpp" +#include "browser_widget.hpp" +#include "child_iterator.hpp" #include "dvd.hpp" -#include "framebuffer.hpp" -#include "linkiterator.hpp" +#include "generate_dvd.hpp" +#include "link_iterator.hpp" +#include "null_prompt_service.hpp" #include "pixbufs.hpp" -#include "stylesheets.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 @@ -123,7 +137,7 @@ namespace result.bottom = result.top + height; // Merge bounding boxes of all child elements - for (ChildIterator it = ChildIterator(elem), end; it != end; ++it) + for (child_iterator it = child_iterator(elem), end; it != end; ++it) { nsCOMPtr child_node(*it); PRUint16 child_type; @@ -139,55 +153,67 @@ namespace return result; } - class WebDvdWindow : public Gtk::Window + + class webdvd_window : public Gtk::Window { public: - WebDvdWindow( + 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: - void add_page(const std::string & uri); - void add_video(const std::string & uri); + dvd_contents::pgc_ref add_menu(const std::string & uri); + dvd_contents::pgc_ref add_title(const std::string & uri); void load_next_page(); void on_net_state_change(const char * uri, gint flags, guint status); + bool browser_is_busy() const + { + return pending_window_update_ || pending_req_count_; + } bool process_page(); void save_screenshot(); void process_links(nsIPresShell * pres_shell, nsIPresContext * pres_context, nsIDOMWindow * dom_window); - void generate_dvd(); - enum ResourceType { page_resource, video_resource }; - typedef std::pair ResourceEntry; video::frame_params frame_params_; std::string output_dir_; - BrowserWidget browser_widget_; + mpeg_encoder encoder_; + browser_widget browser_widget_; nsCOMPtr stylesheet_; + + dvd_contents contents_; + typedef std::map resource_map_type; + resource_map_type resource_map_; + std::queue page_queue_; - std::map resource_map_; - std::vector > page_links_; - std::vector video_paths_; bool pending_window_update_; int pending_req_count_; bool have_tweaked_page_; std::auto_ptr background_temp_; struct page_state; std::auto_ptr page_state_; - std::vector > page_temp_files_; + + bool finished_; }; - WebDvdWindow::WebDvdWindow( + 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); @@ -195,57 +221,92 @@ namespace add(browser_widget_); browser_widget_.show(); browser_widget_.signal_net_state().connect( - SigC::slot(*this, &WebDvdWindow::on_net_state_change)); + SigC::slot(*this, &webdvd_window::on_net_state_change)); - add_page(main_page_uri); + add_menu(main_page_uri); load_next_page(); } - void WebDvdWindow::add_page(const std::string & uri) + bool webdvd_window::is_finished() const { - if (resource_map_.insert( - std::make_pair(uri, ResourceEntry(page_resource, 0))) - .second) + return finished_; + } + + dvd_contents::pgc_ref webdvd_window::add_menu(const std::string & uri) + { + dvd_contents::pgc_ref next_menu(dvd_contents::menu_pgc, + contents_.menus.size()); + std::pair insert_result( + resource_map_.insert(std::make_pair(uri, next_menu))); + + if (!insert_result.second) + { + return insert_result.first->second; + } + else { page_queue_.push(uri); + contents_.menus.resize(contents_.menus.size() + 1); + return next_menu; } } - void WebDvdWindow::add_video(const std::string & uri) + dvd_contents::pgc_ref webdvd_window::add_title(const std::string & uri) { - if (resource_map_.insert( - std::make_pair(uri, ResourceEntry(video_resource, - video_paths_.size() + 1))) - .second) + dvd_contents::pgc_ref next_title(dvd_contents::title_pgc, + contents_.titles.size()); + std::pair insert_result( + resource_map_.insert(std::make_pair(uri, next_title))); + + if (!insert_result.second) + { + return insert_result.first->second; + } + 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 - if (!Glib::file_test(filename, Glib::FILE_TEST_IS_REGULAR)) - throw std::runtime_error( - filename + " is missing or not a regular file"); - video_paths_.push_back(filename); + + vob_list list; + + // Store a reference to a linked VOB file, or the contents + // of a linked VOB list file. + if (path.compare(path.size() - 4, 4, ".vob") == 0) + { + if (!Glib::file_test(path, Glib::FILE_TEST_IS_REGULAR)) + throw std::runtime_error( + path + " is missing or not a regular file"); + vob_ref ref; + ref.file = path; + list.push_back(ref); + } + else + { + assert(path.compare(path.size() - 8, 8, ".voblist") == 0); + read_vob_list(path).swap(list); + } + + contents_.titles.resize(contents_.titles.size() + 1); + contents_.titles.back().swap(list); + return next_title; } } - void WebDvdWindow::load_next_page() + void webdvd_window::load_next_page() { assert(!page_queue_.empty()); const std::string & uri = page_queue_.front(); std::cout << "loading " << uri << std::endl; - std::size_t page_count = page_links_.size(); - resource_map_[uri].second = ++page_count; - page_links_.resize(page_count); - browser_widget_.load_uri(uri); } - void WebDvdWindow::on_net_state_change(const char * uri, + void webdvd_window::on_net_state_change(const char * uri, gint flags, guint status) { # ifdef DEBUG_ON_NET_STATE_CHANGE - std::cout << "WebDvdWindow::on_net_state_change("; + std::cout << "webdvd_window::on_net_state_change("; if (uri) std::cout << '"' << uri << '"'; else @@ -313,12 +374,15 @@ namespace pending_window_update_ = false; } - if (pending_req_count_ == 0 && !pending_window_update_) + if (!browser_is_busy()) { try { if (!process_page()) + { + finished_ = true; Gtk::Main::quit(); + } } catch (std::exception & e) { @@ -329,10 +393,19 @@ namespace std::cerr << ": " << e.what() << "\n"; Gtk::Main::quit(); } + catch (Glib::Exception & e) + { + std::cerr << "Fatal error"; + if (!page_queue_.empty()) + std::cerr << " while processing <" << page_queue_.front() + << ">"; + std::cerr << ": " << e.what() << "\n"; + Gtk::Main::quit(); + } } } - bool WebDvdWindow::process_page() + bool webdvd_window::process_page() { assert(!page_queue_.empty()); @@ -361,14 +434,14 @@ namespace // Might need to wait a while for things to load or more // likely for a re-layout. - if (pending_req_count_ > 0) + if (browser_is_busy()) return true; } // All further work should only be done if we're not in preview mode. if (!output_dir_.empty()) { - // If we haven't already started work on this page, save a + // If we haven't already started work on this menu, save a // screenshot of its normal appearance. if (!page_state_.get()) save_screenshot(); @@ -383,7 +456,7 @@ namespace page_queue_.pop(); if (page_queue_.empty()) { - generate_dvd(); + generate_dvd(contents_, output_dir_); return false; } else @@ -396,7 +469,7 @@ namespace return true; } - void WebDvdWindow::save_screenshot() + void webdvd_window::save_screenshot() { Glib::RefPtr window(get_window()); assert(window); @@ -412,7 +485,7 @@ namespace ->save(background_temp_->get_name(), "png"); } - struct WebDvdWindow::page_state + struct webdvd_window::page_state { page_state(nsIDOMDocument * doc, int width, int height) : diff_pixbuf(Gdk::Pixbuf::create( @@ -436,15 +509,15 @@ namespace temp_file links_temp; - int link_num; - LinkIterator links_it, links_end; + unsigned link_num; + link_iterator links_it, links_end; rectangle link_rect; bool link_changing; Glib::RefPtr norm_pixbuf; }; - void WebDvdWindow::process_links(nsIPresShell * pres_shell, + void webdvd_window::process_links(nsIPresShell * pres_shell, nsIPresContext * pres_context, nsIDOMWindow * dom_window) { @@ -487,25 +560,32 @@ namespace 0, 0, frame_params_.width, frame_params_.height }; + unsigned menu_num = resource_map_[page_queue_.front()].index; + for (/* no initialisation */; state->links_it != state->links_end; ++state->links_it) { nsCOMPtr node(*state->links_it); - // Find the link URI. + // Find the link URI and separate any fragment from it. nsCOMPtr link(do_QueryInterface(node)); assert(link); - nsCOMPtr uri; - check(link->GetHrefURI(getter_AddRefs(uri))); - std::string uri_string; + nsCOMPtr uri_iface; + check(link->GetHrefURI(getter_AddRefs(uri_iface))); + std::string uri_and_fragment, uri, fragment; { - nsCString uri_ns_string; - check(uri->GetSpec(uri_ns_string)); - uri_string.assign(uri_ns_string.BeginReading(), - uri_ns_string.EndReading()); + nsCString uri_and_fragment_ns; + check(uri_iface->GetSpec(uri_and_fragment_ns)); + uri_and_fragment.assign(uri_and_fragment_ns.BeginReading(), + uri_and_fragment_ns.EndReading()); + + std::size_t hash_pos = uri_and_fragment.find('#'); + uri.assign(uri_and_fragment, 0, hash_pos); + if (hash_pos != std::string::npos) + fragment.assign(uri_and_fragment, + hash_pos + 1, std::string::npos); } - std::string uri_sans_fragment(uri_string, 0, uri_string.find('#')); // Is this a new link? if (!state->link_changing) @@ -520,47 +600,58 @@ namespace if (state->link_rect.empty()) { std::cerr << "Ignoring invisible link to " - << uri_string << "\n"; + << uri_and_fragment << "\n"; continue; } ++state->link_num; - if (state->link_num >= dvd::menu_buttons_max) + if (state->link_num >= unsigned(dvd::menu_buttons_max)) { - if (state->link_num == dvd::menu_buttons_max) + if (state->link_num == unsigned(dvd::menu_buttons_max)) std::cerr << "No more than " << dvd::menu_buttons_max - << " buttons can be placed on a page\n"; - std::cerr << "Ignoring link to " << uri_string << "\n"; + << " buttons can be placed on a menu\n"; + std::cerr << "Ignoring link to " << uri_and_fragment + << "\n"; continue; } + state->spumux_file << + " \n"; - } - - file << " \n"; - } - - file << - " \n" - " \n"; - - // Generate a titleset for each video. This appears to make - // jumping to titles a whole lot simpler. - for (std::size_t video_num = 1; - video_num <= video_paths_.size(); - ++video_num) - { - file << - " \n" - // Generate a dummy menu so that the menu button on the - // remote control will work. - " \n" - " \n" - " 
 jump vmgm menu;
\n" - " \n" - " \n" - " \n" - " \n" - // Record calling page/menu. - " 
 g12 = g1;
\n"; - - // Write a reference to a linked VOB file, or the contents - // of a linked VOB list file. - const std::string & video_path = video_paths_[video_num - 1]; - if (video_path.compare(video_path.size() - 4, 4, ".vob") == 0) + if (encoder_ == mpeg_encoder_ffmpeg) { - // FIXME: Should XML-escape the path - file << " \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 { - assert(video_path.compare(video_path.size() - 8, 8, - ".voblist") == 0); - // TODO: Validate the file contents; - file << Glib::file_get_contents(video_path); + 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 + 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 << - // If page/menu location has not been changed during the - // video, change the location to be the following - // link/button when returning to it. In any case, - // return to a page/menu. - " if (g1 eq g12) g1 = g1 + " << link_mult - << "; call menu; \n" - " \n" - " \n" - " \n"; - } - - file << - "\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( 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(), 0, 0, &command_result); if (command_result != 0) - throw std::runtime_error("dvdauthor failed"); - } - } - - void generate_page_dispatch(std::ostream & file, int indent, - int first_page, int last_page) - { - if (first_page == 1 && last_page == 1) - { - // The dispatch code is *on* page 1 so we must not dispatch to - // page 1 since that would cause an infinite loop. This case - // should be unreachable if there is more than one page due - // to the following case. - } - else if (first_page == 1 && last_page == 2) - { - // dvdauthor doesn't allow empty blocks or null statements so - // when selecting between pages 1 and 2 we don't use an "else" - // part. We must use braces so that a following "else" will - // match the right "if". - file << std::setw(indent) << "" << "{\n" - << std::setw(indent) << "" << "if (g0 eq 2)\n" - << std::setw(indent + 2) << "" << "jump menu 2;\n" - << std::setw(indent) << "" << "}\n"; - } - else if (first_page == last_page) - { - file << std::setw(indent) << "" - << "jump menu " << first_page << ";\n"; - } - else - { - int middle = (first_page + last_page) / 2; - file << std::setw(indent) << "" << "if (g0 le " << middle << ")\n"; - generate_page_dispatch(file, indent + 2, first_page, middle); - file << std::setw(indent) << "" << "else\n"; - generate_page_dispatch(file, indent + 2, middle + 1, last_page); + throw std::runtime_error("spumux pipeline failed"); } } @@ -970,13 +830,54 @@ 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() + { + nsCOMPtr pref_service; + static const nsCID pref_service_cid = NS_PREFSERVICE_CID; + check(CallGetService(pref_service_cid, + getter_AddRefs(pref_service))); + nsCOMPtr pref_branch; + + // Disable IE-compatibility kluge that causes backgrounds to + // sometimes/usually be missing from snapshots. This is only + // effective from Mozilla 1.8 onward. +# if MOZ_VERSION_MAJOR > 1 \ + || (MOZ_VERSION_MAJOR == 1 && MOZ_VERSION_MINOR >= 8) + check(pref_service->GetDefaultBranch("layout", + getter_AddRefs(pref_branch))); + check(pref_branch->SetBoolPref( + "fire_onload_after_image_background_loads", + true)); +# endif + + // Set display resolution. With standard-definition video we + // will be fitting ~600 pixels across a screen typically + // ranging from 10 to 25 inches wide, for a resolution of + // 24-60 dpi. I therefore declare the average horizontal + // resolution to be 40 dpi. The vertical resolution will be + // slightly higher (PAL/SECAM) or lower (NTSC), but + // unfortunately Mozilla doesn't support non-square pixels + // (and neither do fontconfig or Xft anyway). + check(pref_service->GetDefaultBranch("browser.display", + getter_AddRefs(pref_branch))); + check(pref_branch->SetIntPref("screen_resolution", 40)); + } + } // namespace +void fatal_error(const std::string & message) +{ + std::cerr << "Fatal error: " << message << "\n"; + Gtk::Main::quit(); +} + int main(int argc, char ** argv) { try @@ -985,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 @@ -1023,16 +925,17 @@ int main(int argc, char ** argv) } } - std::auto_ptr fb; + std::auto_ptr fb; if (!preview_mode) { // Spawn Xvfb and set env variables so that Xlib will use it // Use 8 bits each for RGB components, which should translate into // "enough" bits for YUV components. - fb.reset(new FrameBuffer(frame_params.width, frame_params.height, - 3 * 8)); - setenv("XAUTHORITY", fb->get_x_authority().c_str(), true); - setenv("DISPLAY", fb->get_x_display().c_str(), true); + fb.reset(new x_frame_buffer(frame_params.width, + frame_params.height, + 3 * 8)); + setenv("XAUTHORITY", fb->get_authority().c_str(), true); + setenv("DISPLAY", fb->get_display().c_str(), true); } // Initialise Gtk @@ -1055,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"; @@ -1091,17 +1028,22 @@ int main(int argc, char ** argv) output_dir = argv[argi + 1]; // Initialise Mozilla - BrowserWidget::init(); + browser_widget::initialiser browser_init; + set_browser_preferences(); + if (!preview_mode) + null_prompt_service::install(); // Run the browser/converter - WebDvdWindow 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; }