]> git.decadent.org.uk Git - videolink.git/blobdiff - webdvd.cpp
Moved xml_escape into a separate file.
[videolink.git] / webdvd.cpp
index a6064b2bf33bf65fc9b904fd9b54b023c0861ba5..43f54c7021a703b8be936cb8280548239dfee6f0 100644 (file)
@@ -1,7 +1,8 @@
-// Copyright 2005 Ben Hutchings <ben@decadentplace.org.uk>.
+// Copyright 2005-6 Ben Hutchings <ben@decadentplace.org.uk>.
 // See the file "COPYING" for licence details.
 
 #include <cassert>
+#include <cstring>
 #include <exception>
 #include <fstream>
 #include <iomanip>
@@ -13,7 +14,6 @@
 #include <string>
 
 #include <stdlib.h>
-#include <unistd.h>
 
 #include <boost/shared_ptr.hpp>
 
@@ -29,6 +29,7 @@
 #include <nsIContent.h>
 #include <nsIDocShell.h>
 #include <nsIDOMAbstractView.h>
+#include <nsIDOMBarProp.h>
 #include <nsIDOMDocumentEvent.h>
 #include <nsIDOMDocumentView.h>
 #include <nsIDOMElement.h>
 #include <nsIInterfaceRequestorUtils.h>
 #include <nsIURI.h> // required before nsILink.h
 #include <nsILink.h>
+#include <nsIPrefBranch.h>
+#include <nsIPrefService.h>
 #include <nsIPresContext.h>
 #include <nsIPresShell.h>
+#include <nsIServiceManagerUtils.h>
 #include <nsIWebBrowser.h>
 #include <nsString.h>
 
-#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 "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;
@@ -110,6 +116,7 @@ namespace
     {
        rectangle result;
 
+       // Start with this element's bounding box
        nsCOMPtr<nsIBoxObject> box;
        check(ns_doc->GetBoxObjectFor(elem, getter_AddRefs(box)));
        int width, height;
@@ -120,7 +127,8 @@ namespace
        result.right = result.left + width;
        result.bottom = result.top + height;
 
-       for (ChildIterator it = ChildIterator(elem), end; it != end; ++it)
+       // Merge bounding boxes of all child elements
+       for (child_iterator it = child_iterator(elem), end; it != end; ++it)
        {
            nsCOMPtr<nsIDOMNode> child_node(*it);
            PRUint16 child_type;
@@ -136,170 +144,219 @@ 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);
 
     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<ResourceType, int> ResourceEntry;
        video::frame_params frame_params_;
        std::string output_dir_;
-       BrowserWidget browser_widget_;
+       browser_widget browser_widget_;
        nsCOMPtr<nsIStyleSheet> stylesheet_;
+
+       dvd_contents contents_;
+       typedef std::map<std::string, dvd_contents::pgc_ref> resource_map_type;
+       resource_map_type resource_map_;
+
        std::queue<std::string> page_queue_;
-       std::map<std::string, ResourceEntry> resource_map_;
-       std::vector<std::vector<std::string> > page_links_;
-       std::vector<std::string> video_paths_;
-       bool loading_;
+       bool pending_window_update_;
        int pending_req_count_;
+       bool have_tweaked_page_;
        std::auto_ptr<temp_file> background_temp_;
-       struct link_state;
-       std::auto_ptr<link_state> link_state_;
-       std::vector<boost::shared_ptr<temp_file> > page_temp_files_;
+       struct page_state;
+       std::auto_ptr<page_state> page_state_;
     };
 
-    WebDvdWindow::WebDvdWindow(
+    webdvd_window::webdvd_window(
        const video::frame_params & frame_params,
        const std::string & main_page_uri,
        const std::string & output_dir)
            : frame_params_(frame_params),
              output_dir_(output_dir),
              stylesheet_(load_css("file://" WEBDVD_LIB_DIR "/webdvd.css")),
-             loading_(false),
-             pending_req_count_(0)
+             pending_window_update_(false),
+             pending_req_count_(0),
+             have_tweaked_page_(false)
     {
-       set_default_size(frame_params_.width, frame_params_.height);
+       set_size_request(frame_params_.width, frame_params_.height);
+       set_resizable(false);
+
        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)
+    dvd_contents::pgc_ref webdvd_window::add_menu(const std::string & uri)
     {
-       if (resource_map_.insert(
-               std::make_pair(uri, ResourceEntry(page_resource, 0)))
-           .second)
+       dvd_contents::pgc_ref next_menu(dvd_contents::menu_pgc,
+                                       contents_.menus.size());
+       std::pair<resource_map_type::iterator, bool> 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<resource_map_type::iterator, bool> insert_result(
+           resource_map_.insert(std::make_pair(uri, next_title)));
+
+       if (!insert_result.second)
        {
-           // FIXME: Should accept some slightly different URI prefixes
-           // (e.g. file://localhost/) and decode any URI-escaped
-           // characters in the path.
-           assert(uri.compare(0, 8, "file:///") == 0);
-           video_paths_.push_back(uri.substr(7));
+           return insert_result.first->second;
+       }
+       else
+       {
+           Glib::ustring hostname;
+           std::string path(Glib::filename_from_uri(uri, hostname));
+           // FIXME: Should check the hostname
+
+           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()
     {
-       loading_ = true;
-
        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)
     {
-       enum {
-           process_nothing,
-           process_new_page,
-           process_current_link
-       } action = process_nothing;
+#       ifdef DEBUG_ON_NET_STATE_CHANGE
+       std::cout << "webdvd_window::on_net_state_change(";
+       if (uri)
+           std::cout << '"' << uri << '"';
+       else
+           std::cout << "NULL";
+       std::cout << ", ";
+       {
+           gint flags_left = flags;
+           static const struct {
+               gint value;
+               const char * name;
+           } flag_names[] = {
+               { GTK_MOZ_EMBED_FLAG_START, "STATE_START" },
+               { GTK_MOZ_EMBED_FLAG_REDIRECTING, "STATE_REDIRECTING" },
+               { GTK_MOZ_EMBED_FLAG_TRANSFERRING, "STATE_TRANSFERRING" },
+               { GTK_MOZ_EMBED_FLAG_NEGOTIATING, "STATE_NEGOTIATING" },
+               { GTK_MOZ_EMBED_FLAG_STOP, "STATE_STOP" },
+               { GTK_MOZ_EMBED_FLAG_IS_REQUEST, "STATE_IS_REQUEST" },
+               { GTK_MOZ_EMBED_FLAG_IS_DOCUMENT, "STATE_IS_DOCUMENT" },
+               { GTK_MOZ_EMBED_FLAG_IS_NETWORK, "STATE_IS_NETWORK" },
+               { GTK_MOZ_EMBED_FLAG_IS_WINDOW, "STATE_IS_WINDOW" }
+           };
+           for (int i = 0; i != sizeof(flag_names)/sizeof(flag_names[0]); ++i)
+           {
+               if (flags & flag_names[i].value)
+               {
+                   std::cout << flag_names[i].name;
+                   flags_left -= flag_names[i].value;
+                   if (flags_left)
+                       std::cout << " | ";
+               }
+           }
+           if (flags_left)
+               std::cout << "0x" << std::setbase(16) << flags_left;
+       }
+       std::cout << ", " << "0x" << std::setbase(16) << status << ")\n";
+#       endif // DEBUG_ON_NET_STATE_CHANGE
 
        if (flags & GTK_MOZ_EMBED_FLAG_IS_REQUEST)
        {
            if (flags & GTK_MOZ_EMBED_FLAG_START)
                ++pending_req_count_;
+
            if (flags & GTK_MOZ_EMBED_FLAG_STOP)
            {
                assert(pending_req_count_ != 0);
                --pending_req_count_;
            }
-           if (pending_req_count_ == 0 && link_state_.get())
-               action = process_current_link;
        }
            
-       if (flags & GTK_MOZ_EMBED_FLAG_STOP
-           && flags & GTK_MOZ_EMBED_FLAG_IS_WINDOW)
-           action = process_new_page;
+       if (flags & GTK_MOZ_EMBED_FLAG_IS_DOCUMENT
+           && flags & GTK_MOZ_EMBED_FLAG_START)
+       {
+           pending_window_update_ = true;
+           have_tweaked_page_ = false;
+       }
 
-       if (action != process_nothing)
+       if (flags & GTK_MOZ_EMBED_FLAG_IS_WINDOW
+           && flags & GTK_MOZ_EMBED_FLAG_STOP)
        {
-           assert(loading_ && !page_queue_.empty());
-           assert(pending_req_count_ == 0);
+           // Check whether the load was successful, ignoring this
+           // pseudo-error.
+           if (status != NS_IMAGELIB_ERROR_LOAD_ABORTED)
+               check(status);
 
+           pending_window_update_ = false;
+       }
+
+       if (!browser_is_busy())
+       {
            try
            {
-               // Check whether the load was successful, ignoring this
-               // pseudo-error.
-               if (status != NS_IMAGELIB_ERROR_LOAD_ABORTED)
-                   check(status);
-
-               nsCOMPtr<nsIWebBrowser> browser(
-                   browser_widget_.get_browser());
-               nsCOMPtr<nsIDocShell> doc_shell(do_GetInterface(browser));
-               assert(doc_shell);
-               nsCOMPtr<nsIPresShell> pres_shell;
-               check(doc_shell->GetPresShell(getter_AddRefs(pres_shell)));
-               nsCOMPtr<nsIPresContext> pres_context;
-               check(doc_shell->GetPresContext(
-                         getter_AddRefs(pres_context)));
-               nsCOMPtr<nsIDOMWindow> dom_window;
-               check(browser->GetContentDOMWindow(
-                         getter_AddRefs(dom_window)));
-
-               if (action == process_new_page)
-               {
-                   apply_style_sheet(stylesheet_, pres_shell);
-                   save_screenshot();
-               }
-               process_links(pres_shell, pres_context, dom_window);
-               if (!link_state_.get())
-               {
-                   page_queue_.pop();
-                   if (page_queue_.empty())
-                   {
-                       generate_dvd();
-                       Gtk::Main::quit();
-                   }
-                   else
-                       load_next_page();
-               }
+               if (!process_page())
+                   Gtk::Main::quit();
            }
            catch (std::exception & e)
            {
@@ -310,10 +367,83 @@ 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 webdvd_window::process_page()
+    {
+       assert(!page_queue_.empty());
+
+       nsCOMPtr<nsIWebBrowser> browser(browser_widget_.get_browser());
+       nsCOMPtr<nsIDocShell> doc_shell(do_GetInterface(browser));
+       assert(doc_shell);
+       nsCOMPtr<nsIPresShell> pres_shell;
+       check(doc_shell->GetPresShell(getter_AddRefs(pres_shell)));
+       nsCOMPtr<nsIPresContext> pres_context;
+       check(doc_shell->GetPresContext(getter_AddRefs(pres_context)));
+       nsCOMPtr<nsIDOMWindow> dom_window;
+       check(browser->GetContentDOMWindow(getter_AddRefs(dom_window)));
+
+       // If we haven't done so already, apply the stylesheet and
+       // disable scrollbars.
+       if (!have_tweaked_page_)
+       {
+           apply_style_sheet(stylesheet_, pres_shell);
+
+           // This actually only needs to be done once.
+           nsCOMPtr<nsIDOMBarProp> dom_bar_prop;
+           check(dom_window->GetScrollbars(getter_AddRefs(dom_bar_prop)));
+           check(dom_bar_prop->SetVisible(false));
+
+           have_tweaked_page_ = true;
+
+           // Might need to wait a while for things to load or more
+           // likely for a re-layout.
+           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 menu, save a
+           // screenshot of its normal appearance.
+           if (!page_state_.get())
+               save_screenshot();
+
+           // Start or continue processing links.
+           process_links(pres_shell, pres_context, dom_window);
+
+           // If we've finished work on the links, move on to the
+           // next page, if any, or else generate the DVD filesystem.
+           if (!page_state_.get())
+           {
+               page_queue_.pop();
+               if (page_queue_.empty())
+               {
+                   generate_dvd(contents_, output_dir_);
+                   return false;
+               }
+               else
+               {
+                   load_next_page();
+               }
+           }
        }
+
+       return true;
     }
 
-    void WebDvdWindow::save_screenshot()
+    void webdvd_window::save_screenshot()
     {
        Glib::RefPtr<Gdk::Window> window(get_window());
        assert(window);
@@ -329,24 +459,39 @@ namespace
            ->save(background_temp_->get_name(), "png");
     }
 
-    struct WebDvdWindow::link_state
+    struct webdvd_window::page_state
     {
+       page_state(nsIDOMDocument * doc, int width, int height)
+               : diff_pixbuf(Gdk::Pixbuf::create(
+                                 Gdk::COLORSPACE_RGB,
+                                 true, 8, // has_alpha, bits_per_sample
+                                 width, height)),
+                 spumux_temp("webdvd-spumux-"),
+                 links_temp("webdvd-links-"),
+                 link_num(0),
+                 links_it(doc),
+                 link_changing(false)
+           {
+               spumux_temp.close();
+               links_temp.close();
+           }
+
        Glib::RefPtr<Gdk::Pixbuf> diff_pixbuf;
 
-       std::auto_ptr<temp_file> spumux_temp;
+       temp_file spumux_temp;
        std::ofstream spumux_file;
 
-       std::auto_ptr<temp_file> links_temp;
+       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<Gdk::Pixbuf> norm_pixbuf;
     };
 
-    void WebDvdWindow::process_links(nsIPresShell * pres_shell,
+    void webdvd_window::process_links(nsIPresShell * pres_shell,
                                     nsIPresContext * pres_context,
                                     nsIDOMWindow * dom_window)
     {
@@ -369,60 +514,52 @@ namespace
        check(doc_view->GetDefaultView(getter_AddRefs(view)));
 
        // Set up or recover our iteration state.
-       std::auto_ptr<link_state> state(link_state_);
+       std::auto_ptr<page_state> state(page_state_);
        if (!state.get())
        {
-           state.reset(new link_state);
-
-           state->diff_pixbuf = Gdk::Pixbuf::create(
-               Gdk::COLORSPACE_RGB,
-               true, 8, // has_alpha, bits_per_sample
-               frame_params_.width, frame_params_.height);
-
-           state->spumux_temp.reset(new temp_file("webdvd-spumux-"));
-           state->spumux_temp->close();
-
-           state->links_temp.reset(new temp_file("webdvd-links-"));
-           state->links_temp->close();
+           state.reset(
+               new page_state(
+                   basic_doc, frame_params_.width, frame_params_.height));
            
-           state->spumux_file.open(state->spumux_temp->get_name().c_str());
+           state->spumux_file.open(state->spumux_temp.get_name().c_str());
            state->spumux_file <<
                "<subpictures>\n"
                "  <stream>\n"
                "    <spu force='yes' start='00:00:00.00'\n"
-               "        highlight='" << state->links_temp->get_name()
-                              << "'\n"
-               "        select='" << state->links_temp->get_name()
-                              << "'>\n";
-
-           state->link_num = 0;
-           state->links_it = LinkIterator(basic_doc);
-           state->link_changing = false;
+               "        highlight='" << state->links_temp.get_name() << "'\n"
+               "        select='" << state->links_temp.get_name() << "'>\n";
        }
-           
+
        rectangle window_rect = {
            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<nsIDOMNode> node(*state->links_it);
 
-           // Find the link URI.
+           // Find the link URI and separate any fragment from it.
            nsCOMPtr<nsILink> link(do_QueryInterface(node));
            assert(link);
-           nsCOMPtr<nsIURI> uri;
-           check(link->GetHrefURI(getter_AddRefs(uri)));
-           std::string uri_string;
+           nsCOMPtr<nsIURI> 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)
@@ -437,44 +574,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 <<
+                   "      <button x0='" << state->link_rect.left << "'"
+                   " y0='" << state->link_rect.top << "'"
+                   " x1='" << state->link_rect.right - 1 << "'"
+                   " y1='" << state->link_rect.bottom - 1 << "'/>\n";
+
                // Check whether this is a link to a video or a page then
-               // add it to the known resources if not already seen.
-               nsCString path;
-               check(uri->GetPath(path));
+               // add it to the known resources if not already seen; then
+               // add it to the menu entries.
+               dvd_contents::pgc_ref target;
                // FIXME: This is a bit of a hack.  Perhaps we could decide
                // later based on the MIME type determined by Mozilla?
-               if (path.Length() > 4
-                   && std::strcmp(path.EndReading() - 4, ".vob") == 0)
+               if ((uri.size() > 4
+                    && uri.compare(uri.size() - 4, 4, ".vob") == 0)
+                   || (uri.size() > 8
+                       && uri.compare(uri.size() - 8, 8, ".voblist") == 0))
                {
                    PRBool is_file;
-                   check(uri->SchemeIs("file", &is_file));
+                   check(uri_iface->SchemeIs("file", &is_file));
                    if (!is_file)
                    {
                        std::cerr << "Links to video must use the file:"
                                  << " scheme\n";
                        continue;
                    }
-                   add_video(uri_sans_fragment);
+                   target = add_title(uri);
+                   target.sub_index =
+                       std::strtoul(fragment.c_str(), NULL, 10);
                }
                else
                {
-                   add_page(uri_sans_fragment);
+                   target = add_menu(uri);
+                   // TODO: If there's a fragment, work out which button
+                   // is closest and set target.sub_index.
                }
+               contents_.menus[menu_num].entries.push_back(target);
 
                nsCOMPtr<nsIContent> content(do_QueryInterface(node));
                assert(content);
@@ -523,10 +674,10 @@ namespace
                // We may have to exit and wait for image loading
                // to complete, at which point we will be called
                // again.
-               if (pending_req_count_ > 0)
+               if (browser_is_busy())
                {
                    state->link_changing = true;
-                   link_state_ = state;
+                   page_state_ = state;
                    return;
                }
            }
@@ -551,22 +702,13 @@ namespace
                state->link_rect.top,
                state->link_rect.right - state->link_rect.left,
                state->link_rect.bottom - state->link_rect.top);
-
-           state->spumux_file <<
-               "      <button x0='" << state->link_rect.left << "'"
-               " y0='" << state->link_rect.top << "'"
-               " x1='" << state->link_rect.right - 1 << "'"
-               " y1='" << state->link_rect.bottom - 1 << "'/>\n";
-
-           // Add to the page's links, ignoring any fragment (for now).
-           page_links_.back().push_back(uri_sans_fragment);
        }
 
        quantise_rgba_pixbuf(state->diff_pixbuf, dvd::button_n_colours);
 
-       std::cout << "saving " << state->links_temp->get_name()
+       std::cout << "saving " << state->links_temp.get_name()
                  << std::endl;
-       state->diff_pixbuf->save(state->links_temp->get_name(), "png");
+       state->diff_pixbuf->save(state->links_temp.get_name(), "png");
 
        state->spumux_file <<
            "    </spu>\n"
@@ -578,8 +720,6 @@ namespace
        // TODO: if (!state->spumux_file) throw ...
 
        {
-           boost::shared_ptr<temp_file> vob_temp(
-               new temp_file("webdvd-vob-"));
            std::ostringstream command_stream;
            command_stream << "pngtopnm "
                           << background_temp_->get_name()
@@ -592,8 +732,9 @@ namespace
                               " | mpeg2enc -v0 -f8 -a2 -o/dev/stdout"
                               " | mplex -v0 -f8 -o/dev/stdout /dev/stdin"
                               " | spumux -v0 -mdvd ")
-                          << state->spumux_temp->get_name()
-                          << " > " << vob_temp->get_name();
+                          << 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
@@ -610,237 +751,6 @@ namespace
                             &command_result);
            if (command_result != 0)
                throw std::runtime_error("spumux pipeline failed");
-
-           page_temp_files_.push_back(vob_temp);
-       }
-    }
-
-    void generate_page_dispatch(std::ostream &, int indent,
-                               int first_page, int last_page);
-
-    void WebDvdWindow::generate_dvd()
-    {
-       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:     link destination (when jumping to menu 1), then scratch
-       // g1:     current location
-       // g2-g11: location history (g2 = most recent)
-       // g12:    location that last linked 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.  This is
-       // chosen for compatibility with the encoding of the s8
-       // (button) register.
-       //
-       static const int link_mult = dvd::reg_s8_button_mult;
-       static const int page_mask = link_mult - 1;
-       static const int link_mask = (1 << dvd::reg_bits) - link_mult;
-
-       file <<
-           "<dvdauthor>\n"
-           "  <vmgm>\n"
-           "    <menus>\n";
-           
-       for (std::size_t page_num = 1;
-            page_num <= page_links_.size();
-            ++page_num)
-       {
-           std::vector<std::string> & page_links =
-               page_links_[page_num - 1];
-
-           if (page_num == 1)
-           {
-               // This is the first page (root menu) which needs to
-               // include initialisation and dispatch code.
-       
-               file <<
-                   "      <pgc entry='title'>\n"
-                   "        <pre>\n"
-                   // Has the location been set yet?
-                   "          if (g1 eq 0)\n"
-                   "          {\n"
-                   // Initialise the current location to first link on
-                   // this page.
-                   "            g1 = " << 1 * link_mult + 1 << ";\n"
-                   "          }\n"
-                   "          else\n"
-                   "          {\n"
-                   // Has the user selected a link?
-                   "            if (g0 ne 0)\n"
-                   "            {\n"
-                   // First update the history.
-                    // Does link go to the last page in the history?
-                   "              if (((g0 ^ g2) &amp; " << page_mask
-                    << ") == 0)\n"
-                    // It does; we treat this as going back and pop the old
-                   // location off the history stack into the current
-                   // location.  Clear the free stack slot.
-                   "              {\n"
-                   "                g1 = g2; g2 = g3; g3 = g4; g4 = g5;\n"
-                   "                g5 = g6; g6 = g7; g7 = g8; g8 = g9;\n"
-                   "                g9 = g10; g10 = g11; g11 = 0;\n"
-                   "              }\n"
-                    "              else\n"
-                   // Link goes to some other page, so push current
-                   // location onto the history stack and set the current
-                   // location to be exactly the target location.
-                   "              {\n"
-                   "                g11 = g10; g10 = g9; g9 = g8; g8 = g7;\n"
-                   "                g7 = g6; g6 = g5; g5 = g4; g4 = g3;\n"
-                   "                g3 = g2; g2 = g1; g1 = g0;\n"
-                   "              }\n"
-                   "            }\n"
-                   // Find the target page number.
-                   "            g0 = g1 &amp; " << page_mask << ";\n";
-               // There seems to be no way to perform a computed jump,
-               // so we generate all possible jumps and a binary search
-               // to select the correct one.
-               generate_page_dispatch(file, 12, 1, page_links_.size());
-               file <<
-                   "          }\n";
-           }
-           else // page_num != 1
-           {
-               file <<
-                   "      <pgc>\n"
-                   "        <pre>\n";
-           }
-
-           file <<
-               // Clear link indicator and highlight the
-               // appropriate link/button.
-               "          g0 = 0; s8 = g1 &amp; " << link_mask << ";\n"
-               "        </pre>\n"
-               "        <vob file='"
-                << page_temp_files_[page_num - 1]->get_name() << "'/>\n";
-
-           for (std::size_t link_num = 1;
-                link_num <= page_links.size();
-                ++link_num)
-           {
-               file <<
-                   "        <button>"
-                   // Update current location.
-                   " g1 = " << link_num * link_mult + page_num << ";";
-
-               // Jump to appropriate resource.
-               const ResourceEntry & resource_loc =
-                   resource_map_[page_links[link_num - 1]];
-               if (resource_loc.first == page_resource)
-                   file <<
-                       " g0 = " << 1 * link_mult + resource_loc.second << ";"
-                       " jump menu 1;";
-               else if (resource_loc.first == video_resource)
-                   file << " jump title " << resource_loc.second << ";";
-
-               file <<  " </button>\n";
-           }
-
-           file << "      </pgc>\n";
-       }
-
-       file <<
-           "    </menus>\n"
-           "  </vmgm>\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 <<
-               "  <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 page/menu.
-               "        <pre> g12 = g1; </pre>\n"
-               // FIXME: Should XML-escape the path
-               "        <vob file='" << video_paths_[video_num - 1]
-                << "'/>\n"
-               // 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.
-               "        <post> if (g1 eq g12) g1 = g1 + " << link_mult
-                << "; call menu; </post>\n"
-               "      </pgc>\n"
-               "    </titles>\n"
-               "  </titleset>\n";
-       }
-
-       file <<
-           "</dvdauthor>\n";
-
-       file.close();
-
-       {
-           const char * argv[] = {
-               "dvdauthor",
-               "-o", output_dir_.c_str(),
-               "-x", temp.get_name().c_str(),
-               0
-           };
-           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,
-                            SigC::Slot0<void>(),
-                            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);
        }
     }
 
@@ -867,75 +777,168 @@ namespace
            std::string("Invalid video standard: ").append(str));
     }
 
+    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");
+    }
+    
+    void set_browser_preferences()
+    {
+       nsCOMPtr<nsIPrefService> pref_service;
+       static const nsCID pref_service_cid = NS_PREFSERVICE_CID;
+       check(CallGetService<nsIPrefService>(pref_service_cid,
+                                            getter_AddRefs(pref_service)));
+       nsCOMPtr<nsIPrefBranch> 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
 
 int main(int argc, char ** argv)
 {
     try
     {
-       // Determine video frame parameters.
        video::frame_params frame_params = video::pal_params;
-       for (int argi = 1; argi != argc; ++argi)
+       bool preview_mode = false;
+       std::string menu_url;
+       std::string output_dir;
+
+       // Do initial option parsing.  We have to do this before
+       // letting Gtk parse the arguments since we may need to spawn
+       // Xvfb first.
+       int argi = 1;
+       while (argi != argc)
        {
            if (std::strcmp(argv[argi], "--") == 0)
+           {
                break;
-           if (std::strcmp(argv[argi], "--video-std") == 0)
+           }
+           else if (std::strcmp(argv[argi], "--help") == 0)
+           {
+               print_usage(std::cout, argv[0]);
+               return EXIT_SUCCESS;
+           }
+           else if (std::strcmp(argv[argi], "--preview") == 0)
+           {
+               preview_mode = true;
+               argi += 1;
+           }
+           else if (std::strcmp(argv[argi], "--video-std") == 0)
            {
                if (argi + 1 == argc)
                {
                    std::cerr << "Missing argument to --video-std\n";
+                   print_usage(std::cerr, argv[0]);
                    return EXIT_FAILURE;
                }
                frame_params = lookup_frame_params(argv[argi + 1]);
-               break;
+               argi += 2;
+           }
+           else
+           {
+               argi += 1;
            }
        }
-       
-       // 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.
-       FrameBuffer fb(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);
+
+       std::auto_ptr<x_frame_buffer> 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 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
        Gtk::Main kit(argc, argv);
 
-       // Check we have the right number of arguments.  We can't
-       // do this earlier because we need to let Gtk read and remove
-       // any of the many options it understands.
-       int argi = 1;
+       // Complete option parsing with Gtk's options out of the way.
+       argi = 1;
        while (argi != argc)
        {
-           if (std::strcmp(argv[argi], "--video-std") == 0)
+           if (std::strcmp(argv[argi], "--") == 0)
            {
-               argi += 2;
+               argi += 1;
+               break;
            }
-           else if (std::strcmp(argv[argi], "--") == 0)
+           else if (std::strcmp(argv[argi], "--preview") == 0)
            {
                argi += 1;
-               break;
+           }
+           else if (std::strcmp(argv[argi], "--video-std") == 0)
+           {
+               argi += 2;
            }
            else if (argv[argi][0] == '-')
            {
                std::cerr << "Invalid option: " << argv[argi] << "\n";
+               print_usage(std::cerr, argv[0]);
                return EXIT_FAILURE;
            }
            else
+           {
                break;
+           }
        }
-       if (argi != argc - 2)
+
+       // Look for a starting URL or filename and (except in preview
+       // mode) an output directory after the options.
+        if (argc - argi != (preview_mode ? 1 : 2))
        {
-           std::cerr << "Usage: " << argv[0]
-                     << (" [gtk-options] [--video-std std-name]"
-                         "front-page-url output-dir\n");
+           print_usage(std::cerr, argv[0]);
            return EXIT_FAILURE;
        }
+       if (std::strstr(argv[argi], "://"))
+       {
+           // It appears to be an absolute URL, so use it as-is.
+           menu_url = argv[argi];
+       }
+       else
+       {
+           // Assume it's a filename.  Resolve it to an absolute URL.
+           std::string path(argv[argi]);
+           if (!Glib::path_is_absolute(path))
+               path = Glib::build_filename(Glib::get_current_dir(), path);
+           menu_url = Glib::filename_to_uri(path);             
+       }
+       if (!preview_mode)
+           output_dir = argv[argi + 1];
 
        // Initialise Mozilla
-       BrowserWidget::init();
+       browser_widget::initialiser browser_init;
+       set_browser_preferences();
 
-       WebDvdWindow window(frame_params, argv[argi], argv[argi + 1]);
+       // Run the browser/converter
+       webdvd_window window(frame_params, menu_url, output_dir);
        Gtk::Main::run(window);
     }
     catch (std::exception & e)