]> git.decadent.org.uk Git - videolink.git/blobdiff - webdvd.cpp
Renamed various types to fit lower_case_with_underscores convention.
[videolink.git] / webdvd.cpp
index 94ab121f890c3f043cca63277ecc7928e76933e1..9a5dd04e8b078af7c6de36d07c2e36e3076a73e1 100644 (file)
@@ -2,6 +2,7 @@
 // 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 "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 "xpcom_support.hpp"
 
 using xpcom_support::check;
@@ -110,6 +114,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 +125,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,10 +142,10 @@ 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);
@@ -149,79 +155,84 @@ namespace
        void add_video(const std::string & uri);
        void load_next_page();
        void on_net_state_change(const char * uri, gint flags, guint status);
+       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;
+       enum resource_type { page_resource, video_resource };
+       typedef std::pair<resource_type, int> resource_entry;
        video::frame_params frame_params_;
        std::string output_dir_;
-       BrowserWidget browser_widget_;
+       browser_widget browser_widget_;
        nsCOMPtr<nsIStyleSheet> stylesheet_;
        std::queue<std::string> page_queue_;
-       std::map<std::string, ResourceEntry> resource_map_;
+       std::map<std::string, resource_entry> 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_;
+       struct page_state;
+       std::auto_ptr<page_state> page_state_;
        std::vector<boost::shared_ptr<temp_file> > page_temp_files_;
     };
 
-    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);
        load_next_page();
     }
 
-    void WebDvdWindow::add_page(const std::string & uri)
+    void webdvd_window::add_page(const std::string & uri)
     {
        if (resource_map_.insert(
-               std::make_pair(uri, ResourceEntry(page_resource, 0)))
+               std::make_pair(uri, resource_entry(page_resource, 0)))
            .second)
        {
            page_queue_.push(uri);
        }
     }
 
-    void WebDvdWindow::add_video(const std::string & uri)
+    void webdvd_window::add_video(const std::string & uri)
     {
        if (resource_map_.insert(
-               std::make_pair(uri, ResourceEntry(video_resource,
-                                                 video_paths_.size() + 1)))
+               std::make_pair(uri, resource_entry(video_resource,
+                                                  video_paths_.size() + 1)))
            .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));
+           Glib::ustring hostname;
+           std::string filename(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);
        }
     }
 
-    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;
@@ -229,77 +240,88 @@ namespace
        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 (pending_req_count_ == 0 && !pending_window_update_)
+       {
            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)
            {
@@ -313,7 +335,71 @@ namespace
        }
     }
 
-    void WebDvdWindow::save_screenshot()
+    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 (pending_req_count_ > 0)
+               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
+           // 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();
+                   return false;
+               }
+               else
+               {
+                   load_next_page();
+               }
+           }
+       }
+
+       return true;
+    }
+
+    void webdvd_window::save_screenshot()
     {
        Glib::RefPtr<Gdk::Window> window(get_window());
        assert(window);
@@ -329,24 +415,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;
+       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,37 +470,22 @@ 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
        };
@@ -458,8 +544,11 @@ namespace
                check(uri->GetPath(path));
                // 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 ((path.Length() > 4
+                    && std::strcmp(path.EndReading() - 4, ".vob") == 0)
+                   || (path.Length() > 8
+                       && std::strcmp(path.EndReading() - 8, ".voblist")
+                          == 0))
                {
                    PRBool is_file;
                    check(uri->SchemeIs("file", &is_file));
@@ -526,7 +615,7 @@ namespace
                if (pending_req_count_ > 0)
                {
                    state->link_changing = true;
-                   link_state_ = state;
+                   page_state_ = state;
                    return;
                }
            }
@@ -564,9 +653,9 @@ namespace
 
        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"
@@ -592,7 +681,7 @@ namespace
                               " | mpeg2enc -v0 -f8 -a2 -o/dev/stdout"
                               " | mplex -v0 -f8 -o/dev/stdout /dev/stdin"
                               " | spumux -v0 -mdvd ")
-                          << state->spumux_temp->get_name()
+                          << state->spumux_temp.get_name()
                           << " > " << vob_temp->get_name();
            std::string command(command_stream.str());
            const char * argv[] = {
@@ -618,7 +707,7 @@ namespace
     void generate_page_dispatch(std::ostream &, int indent,
                                int first_page, int last_page);
 
-    void WebDvdWindow::generate_dvd()
+    void webdvd_window::generate_dvd()
     {
        temp_file temp("webdvd-dvdauthor-");
        temp.close();
@@ -729,7 +818,7 @@ namespace
                    " g1 = " << link_num * link_mult + page_num << ";";
 
                // Jump to appropriate resource.
-               const ResourceEntry & resource_loc =
+               const resource_entry & resource_loc =
                    resource_map_[page_links[link_num - 1]];
                if (resource_loc.first == page_resource)
                    file <<
@@ -766,10 +855,25 @@ namespace
                "    <titles>\n"
                "      <pgc>\n"
                // Record calling page/menu.
-               "        <pre> g12 = g1; </pre>\n"
+               "        <pre> g12 = g1; </pre>\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)
+           {
                // FIXME: Should XML-escape the path
-               "        <vob file='" << video_paths_[video_num - 1]
-                << "'/>\n"
+               file << "        <vob file='" << video_path << "'/>\n";
+           }
+           else
+           {
+               assert(video_path.compare(video_path.size() - 8, 8,
+                                         ".voblist") == 0);
+               // TODO: Validate the file contents;
+               file << Glib::file_get_contents(video_path);
+           }
+
+           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,
@@ -871,31 +975,65 @@ namespace
     {
        stream << "Usage: " << command_name
               << (" [gtk-options] [--video-std std-name]"
-                  " front-page-url output-dir\n");
+                  " [--preview] menu-url [output-dir]\n");
     }
     
+    void set_browser_preferences()
+    {
+       // 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)
+       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;
+       check(pref_service->GetDefaultBranch("layout",
+                                            getter_AddRefs(pref_branch)));
+       check(pref_branch->SetBoolPref(
+                 "fire_onload_after_image_background_loads",
+                 true));
+#      endif
+
+       // TODO: Set display resolution?  Unfortunately Mozilla doesn't
+       // support non-square pixels (and neither do fontconfig or Xft
+       // anyway).
+    }
+
 } // namespace
 
 int main(int argc, char ** argv)
 {
     try
     {
-       // Do initial argument parsing.  We have to do this before
-       // letting Gtk parse the arguments since we need to spawn Xvfb
-       // first and that's currently dependent on the frame
-       // parameters (though we could just make it large enough for
-       // any frame) and also an unnecessary expense in some cases.
        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], "--help") == 0)
+           }
+           else if (std::strcmp(argv[argi], "--help") == 0)
            {
                print_usage(std::cout, argv[0]);
                return EXIT_SUCCESS;
            }
-           if (std::strcmp(argv[argi], "--video-std") == 0)
+           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)
                {
@@ -904,32 +1042,46 @@ int main(int argc, char ** argv)
                    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);
 
-       // Complete argument parsing with Gtk's options out of the way.
-       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] == '-')
            {
@@ -938,18 +1090,40 @@ int main(int argc, char ** argv)
                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))
        {
            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)