chiark / gitweb /
Refetch queue on pause/resume so that start times are always right (or
[disorder] / disobedience / choose.c
index cc9f3acdc94efef4419b59370af14ae7d05eeda3..ff5e72b52eabd4bfc4f605f225f3e5e24f1da226 100644 (file)
@@ -1,6 +1,7 @@
+
 /*
  * This file is part of DisOrder
- * Copyright (C) 2006, 2007 Richard Kettlewell
+ * Copyright (C) 2006-2008 Richard Kettlewell
  *
  * This program is free software; you can redistribute it and/or modify
  * it under the terms of the GNU General Public License as published by
  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
  * USA
  */
+/** @file disobedience/choose.c
+ * @brief Hierarchical track selection and search
+ *
+ * We don't use the built-in tree widgets because they require that you know
+ * the children of a node on demand, and we have to wait for the server to tell
+ * us.
+ */
 
 #include "disobedience.h"
+#include "timeval.h"
 
 /* Choose track ------------------------------------------------------------ */
 
+#if TDEBUG
+/* Timing */
+static struct {
+  struct timeval total;
+  struct timeval gtkbits;
+  struct timeval menuupdate;
+  struct timeval new_widgets;
+  struct timeval undisplay;
+  struct timeval colors;
+  struct timeval markers;
+  struct timeval location;
+  struct timeval selection;
+} times;
+
+#define BEGIN(WHAT) do {                        \
+  struct timeval started##WHAT, finished##WHAT; \
+  xgettimeofday(&started##WHAT, 0)
+
+#define END(WHAT)                                                       \
+  xgettimeofday(&finished##WHAT, 0);                                    \
+  times.WHAT = tvadd(times.WHAT, tvsub(finished##WHAT, started##WHAT)); \
+} while(0)
+
+#define INIT() memset(&times, 0, sizeof times)
+
+#define REPORT() do {                           \
+  fprintf(stderr, "total=%g\n"                  \
+          "gtkbits=%g\n"                        \
+          "menuupdate=%g\n"                     \
+          "new_widgets=%g\n"                    \
+          "undisplay=%g\n"                      \
+          "colors=%g\n"                         \
+          "markers=%g\n"                        \
+          "location=%g\n"                       \
+          "selection=%g\n"                      \
+          "accumulation=%g\n"                   \
+          "\n",                                 \
+          tvdouble(times.total),                \
+          tvdouble(times.gtkbits),              \
+          tvdouble(times.menuupdate),           \
+          tvdouble(times.new_widgets),          \
+          tvdouble(times.undisplay),            \
+          tvdouble(times.colors),               \
+          tvdouble(times.markers),              \
+          tvdouble(times.location),             \
+          tvdouble(times.selection),            \
+          (tvdouble(times.gtkbits)              \
+           + tvdouble(times.menuupdate)         \
+           + tvdouble(times.new_widgets)        \
+           + tvdouble(times.undisplay)          \
+           + tvdouble(times.colors)             \
+           + tvdouble(times.markers)            \
+           + tvdouble(times.location)           \
+           + tvdouble(times.selection)));       \
+} while(0)
+#else
+#define BEGIN(WHAT) do {
+#define END(WHAT) } while(0)
+#define INIT() ((void)0)
+#define REPORT() ((void)0)
+#endif
+
 WT(label);
 WT(event_box);
 WT(menu);
@@ -34,75 +105,104 @@ WT(button);
 WT(image);
 WT(entry);
 
-/* We don't use the built-in tree widgets because they require that you know
- * the children of a node on demand, and we have to wait for the server to tell
- * us. */
-
 /* Types */
 
 struct choosenode;
 
+/** @brief Accumulated information about the tree widget */
 struct displaydata {
-  guint width;                          /* total width required */
-  guint height;                         /* total height required */
+  /** @brief Maximum width required */
+  guint width;
+  /** @brief Maximum height required */
+  guint height;
 };
 
 /* instantiate the node vector type */
 
 VECTOR_TYPE(nodevector, struct choosenode *, xrealloc);
 
+/** @brief Signature of function called when a choosenode is filled */
+typedef void (when_filled_callback)(struct choosenode *cn,
+                                    void *wfu);
+
+/** @brief One node in the virtual filesystem */
 struct choosenode {
-  struct choosenode *parent;            /* parent node */
-  const char *path;                     /* full path or 0  */
-  const char *sort;                     /* sort key */
-  const char *display;                  /* display name */
-  int pending;                          /* pending resolve queries */
+  struct choosenode *parent;            /**< @brief parent node */
+  const char *path;                     /**< @brief full path or 0  */
+  const char *sort;                     /**< @brief sort key */
+  const char *display;                  /**< @brief display name */
+  int pending;                          /**< @brief pending resolve queries */
   unsigned flags;
-#define CN_EXPANDABLE 0x0001            /* node is expandable */
-#define CN_EXPANDED 0x0002              /* node is expanded */
-/* Expandable items are directories; non-expandable ones are files */
-#define CN_DISPLAYED 0x0004             /* widget is displayed in layout */
-#define CN_SELECTED 0x0008              /* node is selected */
-  struct nodevector children;           /* vector of children */
-  void (*fill)(struct choosenode *);    /* request child fill or 0 for leaf */
-  GtkWidget *container;                 /* the container for this row */
-  GtkWidget *hbox;                      /* the hbox for this row */
-  GtkWidget *arrow;                     /* arrow widget or 0 */
-  GtkWidget *label;                     /* text label for this node */
-  GtkWidget *marker;                    /* queued marker */
+#define CN_EXPANDABLE 0x0001            /**< @brief node is expandable */
+#define CN_EXPANDED 0x0002              /**< @brief node is expanded
+                                         *
+                                         * Expandable items are directories;
+                                         * non-expandable ones are files. */
+#define CN_DISPLAYED 0x0004             /**< @brief widget is displayed in layout */
+#define CN_SELECTED 0x0008              /**< @brief node is selected */
+#define CN_GETTING_FILES 0x0010         /**< @brief files inbound */
+#define CN_RESOLVING_FILES 0x0020       /**< @brief resolved files inbound */
+#define CN_GETTING_DIRS 0x0040          /**< @brief directories inbound */
+#define CN_GETTING_ANY 0x0070           /**< @brief getting something */
+#define CN_CONTINGENT 0x0080            /**< @brief expansion contingent on search */
+  struct nodevector children;           /**< @brief vector of children */
+  void (*fill)(struct choosenode *);    /**< @brief request child fill or 0 for leaf */
+  GtkWidget *container;                 /**< @brief the container for this row */
+  GtkWidget *hbox;                      /**< @brief the hbox for this row */
+  GtkWidget *arrow;                     /**< @brief arrow widget or 0 */
+  GtkWidget *label;                     /**< @brief text label for this node */
+  GtkWidget *marker;                    /**< @brief queued marker */
+
+  when_filled_callback *whenfilled;     /**< @brief called when filled or 0 */
+  void *wfu;                            /**< @brief passed to @c whenfilled */
+  int ymin;                             /**< @brief least Y value */
+  int ymax;                             /**< @brief greatest Y value */
 };
 
-struct menuitem {
+/** @brief One item in the popup menu */
+struct choose_menuitem {
   /* Parameters */
-  const char *name;                     /* name */
+  const char *name;                     /**< @brief name */
 
   /* Callbacks */
   void (*activate)(GtkMenuItem *menuitem, gpointer user_data);
-  /* Called to activate the menu item.  The user data is the choosenode the
-   * pointer is over. */
+  /**< @brief Called to activate the menu item.
+   *
+   * @p user_data is the choosenode the mouse pointer is over. */
 
   gboolean (*sensitive)(struct choosenode *cn);
-  /* Called to determine whether the menu item should be sensitive.  TODO */
+  /* @brief Called to determine whether the menu item should be sensitive.
+   *
+   * TODO? */
 
   /* State */
-  gulong handlerid;                     /* signal handler ID */
-  GtkWidget *w;                         /* menu item widget */
+  gulong handlerid;                     /**< @brief signal handler ID */
+  GtkWidget *w;                         /**< @brief menu item widget */
 };
 
 /* Variables */
 
 static GtkWidget *chooselayout;
-static GtkWidget *searchentry;          /* search terms */
+static GtkAdjustment *vadjust;
+static GtkWidget *searchentry;          /**< @brief search terms */
+static GtkWidget *nextsearch;           /**< @brief next search result */
+static GtkWidget *prevsearch;           /**< @brief previous search result */
 static struct choosenode *root;
-static struct choosenode *realroot;
-static GtkWidget *menu;                 /* our popup menu */
-static struct choosenode *last_click;   /* last clicked node for selection */
-static int files_visible;               /* total files visible */
-static int files_selected;              /* total files selected */
-static int search_in_flight;            /* a search is underway */
-static int search_obsolete;             /* the current search is void */
-static char **searchresults;            /* search results */
-static int nsearchresults;              /* number of results */
+static GtkWidget *track_menu;           /**< @brief track popup menu */
+static GtkWidget *dir_menu;             /**< @brief directory popup menu */
+static struct choosenode *last_click;   /**< @brief last clicked node for selection */
+static int files_visible;               /**< @brief total files visible */
+static int files_selected;              /**< @brief total files selected */
+static int gets_in_flight;              /**< @brief total gets in flight */
+static int search_in_flight;            /**< @brief a search is underway */
+static int search_obsolete;             /**< @brief the current search is void */
+static char **searchresults;            /**< @brief search results */
+static int nsearchresults;              /**< @brief number of results */
+static int nsearchvisible;      /**< @brief number of search results visible */
+static struct hash *searchhash;         /**< @brief hash of search results */
+static struct progress_window *spw;     /**< @brief progress window */
+static struct choosenode **searchnodes; /**< @brief choosenodes of search results */
+static int suppress_redisplay;          /**< @brief suppress redisplay */
 
 /* Forward Declarations */
 
@@ -114,55 +214,104 @@ static struct choosenode *newnode(struct choosenode *parent,
                                   unsigned flags,
                                   void (*fill)(struct choosenode *));
 static void fill_root_node(struct choosenode *cn);
-static void fill_letter_node(struct choosenode *cn);
 static void fill_directory_node(struct choosenode *cn);
-static void got_files(void *v, int nvec, char **vec);
-static void got_resolved_file(void *v, const char *track);
-static void got_dirs(void *v, int nvec, char **vec);
+static void got_files(void *v, const char *error, int nvec, char **vec);
+static void got_resolved_file(void *v, const char *error, const char *track);
+static void got_dirs(void *v, const char *error, int nvec, char **vec);
 
-static void expand_node(struct choosenode *cn);
+static void expand_node(struct choosenode *cn, int contingent);
 static void contract_node(struct choosenode *cn);
-static void updated_node(struct choosenode *cn, int redisplay);
+static void updated_node(struct choosenode *cn, int redisplay,
+                         const char *why);
 
 static void display_selection(struct choosenode *cn);
 static void clear_selection(struct choosenode *cn);
 
-static void redisplay_tree(void);
+static void redisplay_tree(const char *why);
 static struct displaydata display_tree(struct choosenode *cn, int x, int y);
 static void undisplay_tree(struct choosenode *cn);
 static void initiate_search(void);
 static void delete_widgets(struct choosenode *cn);
+static void expand_from(struct choosenode *cn);
+static struct choosenode *first_search_result(struct choosenode *cn);
 
 static void clicked_choosenode(GtkWidget attribute((unused)) *widget,
                                GdkEventButton *event,
                                gpointer user_data);
 
-static void activate_play(GtkMenuItem *menuitem, gpointer user_data);
-#if 0
-static void activate_remove(GtkMenuItem *menuitem, gpointer user_data);
-#endif
-static void activate_properties(GtkMenuItem *menuitem, gpointer user_data);
+static void activate_track_play(GtkMenuItem *menuitem, gpointer user_data);
+static void activate_track_properties(GtkMenuItem *menuitem, gpointer user_data);
 
-static gboolean sensitive_play(struct choosenode *cn);
-#if 0
-static gboolean sensitive_remove(struct choosenode *cn);
-#endif
-static gboolean sensitive_properties(struct choosenode *cn);
+static gboolean sensitive_track_play(struct choosenode *cn);
+static gboolean sensitive_track_properties(struct choosenode *cn);
 
-static struct menuitem menuitems[] = {
-  { "Play track", activate_play, sensitive_play, 0, 0 },
-#if 0
-  /* Not implemented yet */
-  { "Remove", activate_remove, sensitive_remove, 0, 0 },
-#endif
-  { "Track properties", activate_properties, sensitive_properties, 0, 0 },
+static void activate_dir_play(GtkMenuItem *menuitem, gpointer user_data);
+static void activate_dir_properties(GtkMenuItem *menuitem, gpointer user_data);
+static void activate_dir_select(GtkMenuItem *menuitem, gpointer user_data);
+
+static gboolean sensitive_dir_play(struct choosenode *cn);
+static gboolean sensitive_dir_properties(struct choosenode *cn);
+static gboolean sensitive_dir_select(struct choosenode *cn);
+
+/** @brief Track menu items */
+static struct choose_menuitem track_menuitems[] = {
+  { "Play track", activate_track_play, sensitive_track_play, 0, 0 },
+  { "Track properties", activate_track_properties, sensitive_track_properties, 0, 0 },
+  { 0, 0, 0, 0, 0 }
 };
 
-#define NMENUITEMS (int)(sizeof menuitems / sizeof *menuitems)
+/** @brief Directory menu items */
+static struct choose_menuitem dir_menuitems[] = {
+  { "Play all tracks", activate_dir_play, sensitive_dir_play, 0, 0 },
+  { "Track properties", activate_dir_properties, sensitive_dir_properties, 0, 0 },
+  { "Select all tracks", activate_dir_select, sensitive_dir_select, 0, 0 },
+  { 0, 0, 0, 0, 0 }
+};
 
 /* Maintaining the data structure ------------------------------------------ */
 
-/* Create a new node */
+static char *cnflags(const struct choosenode *cn) {
+  unsigned f = cn->flags, n;
+  struct dynstr d[1];
+  
+  static const char *bits[] = {
+    "expandable",
+    "expanded",
+    "displayed",
+    "selected",
+    "getting_files",
+    "resolving_files",
+    "getting_dirs",
+    "contingent"
+  };
+#define NBITS (sizeof bits / sizeof *bits)
+
+  dynstr_init(d);
+  if(!f)
+    dynstr_append(d, '0');
+  else {
+    for(n = 0; n < NBITS; ++n) {
+      const unsigned bit = 1 << n;
+      if(f & bit) {
+        if(d->nvec)
+          dynstr_append(d, '|');
+        dynstr_append_string(d, bits[n]);
+        f ^= bit;
+      }
+    }
+    if(f) {
+      char buf[32];
+      if(d->nvec)
+        dynstr_append(d, '|');
+      sprintf(buf, "%#x", f);
+      dynstr_append_string(d, buf);
+    }
+  }
+  dynstr_terminate(d);
+  return d->vec;
+}
+
+/** @brief Create a new node */
 static struct choosenode *newnode(struct choosenode *parent,
                                   const char *path,
                                   const char *display,
@@ -188,35 +337,41 @@ static struct choosenode *newnode(struct choosenode *parent,
   return n;
 }
 
-/* Fill the root */
-static void fill_root_node(struct choosenode *cn) {
-  int ch;
-  char *name;
-  struct callbackdata *cbd;
+/** @brief Called when a node has been filled
+ *
+ * Response for calling @c whenfilled.
+ */
+static void filled(struct choosenode *cn) {
+  when_filled_callback *const whenfilled = cn->whenfilled;
 
+  if(whenfilled) {
+    cn->whenfilled = 0;
+    whenfilled(cn, cn->wfu);
+  }
+  if(nsearchvisible < nsearchresults) {
+    /* There is still search expansion work to do */
+    D(("filled %s %d/%d", cn->path, nsearchvisible, nsearchresults));
+    expand_from(cn);
+  }
+  if(gets_in_flight == 0 && nsearchvisible < nsearchresults)
+    expand_from(root);
+}
+
+/** @brief Fill the root */
+static void fill_root_node(struct choosenode *cn) {
   D(("fill_root_node"));
   clear_children(cn);
-  if(choosealpha) {
-    if(!cn->children.nvec) {              /* Only need to do this once */
-      for(ch = 'A'; ch <= 'Z'; ++ch) {
-        byte_xasprintf(&name, "%c", ch);
-        newnode(cn, "<letter>", name, name, CN_EXPANDABLE, fill_letter_node);
-      }
-      newnode(cn, "<letter>", "*", "~", CN_EXPANDABLE, fill_letter_node);
-    }
-    updated_node(cn, 1);
-  } else {
-    /* More de-duping possible here */
-    gtk_label_set_text(GTK_LABEL(report_label), "getting files");
-    cbd = xmalloc(sizeof *cbd);
-    cbd->u.choosenode = cn;
-    disorder_eclient_dirs(client, got_dirs, "", 0, cbd);
-    cbd = xmalloc(sizeof *cbd);
-    cbd->u.choosenode = cn;
-    disorder_eclient_files(client, got_files, "", 0, cbd);
-  }
+  /* More de-duping possible here */
+  if(cn->flags & CN_GETTING_ANY)
+    return;
+  gtk_label_set_text(GTK_LABEL(report_label), "getting files");
+  disorder_eclient_dirs(client, got_dirs, "", 0, cn);
+  disorder_eclient_files(client, got_files, "", 0, cn);
+  cn->flags |= CN_GETTING_FILES|CN_GETTING_DIRS;
+  gets_in_flight += 2;
 }
 
+/** @brief Delete all the widgets owned by @p cn */
 static void delete_cn_widgets(struct choosenode *cn) {
   if(cn->arrow) {
     DW(arrow);
@@ -245,7 +400,11 @@ static void delete_cn_widgets(struct choosenode *cn) {
   }
 }
 
-/* Clear all the children of CN */
+/** @brief Recursively clear all the children of @p cn
+ *
+ * All the widgets at or below @p cn are deleted.  All choosenodes below
+ * it are emptied. i.e. we prune the tree at @p cn.
+ */
 static void clear_children(struct choosenode *cn) {
   int n;
 
@@ -258,98 +417,107 @@ static void clear_children(struct choosenode *cn) {
   cn->children.nvec = 0;
 }
 
-/* Fill a child node */
-static void fill_letter_node(struct choosenode *cn) {
-  const char *regexp;
-  struct callbackdata *cbd;
-
-  D(("fill_letter_node %s", cn->display));
-  switch(cn->display[0]) {
-  default:
-    byte_xasprintf((char **)&regexp, "^(the )?%c", tolower(cn->display[0]));
-    break;
-  case 'T':
-    regexp = "^(?!the [^t])t";
-    break;
-  case '*':
-    regexp = "^[^a-z]";
-    break;
-  }
-  /* TODO: caching */
-  /* TODO: de-dupe against fill_directory_node */
-  gtk_label_set_text(GTK_LABEL(report_label), "getting files");
-  clear_children(cn);
-  cbd = xmalloc(sizeof *cbd);
-  cbd->u.choosenode = cn;
-  disorder_eclient_dirs(client, got_dirs, "", regexp, cbd);
-  cbd = xmalloc(sizeof *cbd);
-  cbd->u.choosenode = cn;
-  disorder_eclient_files(client, got_files, "", regexp, cbd);
-}
-
-/* Called with a list of files just below some node */
-static void got_files(void *v, int nvec, char **vec) {
-  struct callbackdata *cbd = v;
-  struct choosenode *cn = cbd->u.choosenode;
+/** @brief Called with a list of files just below some node */
+static void got_files(void *v, const char *error, int nvec, char **vec) {
+  struct choosenode *cn = v;
   int n;
 
-  D(("got_files %d files for %s", nvec, cn->path));
-  /* Complicated by the need to resolve aliases.  We can save a bit of effort
-   * by re-using cbd though. */
-  cn->pending = nvec;
-  for(n = 0; n < nvec; ++n)
-    disorder_eclient_resolve(client, got_resolved_file, vec[n], cbd);
+  if(error) {
+    popup_protocol_error(0, error);
+    return;
+  }
+
+  D(("got_files %d files for %s %s", nvec, cn->path, cnflags(cn)));
+  /* Complicated by the need to resolve aliases. */
+  cn->flags &= ~CN_GETTING_FILES;
+  --gets_in_flight;
+  if((cn->pending = nvec)) {
+    cn->flags |= CN_RESOLVING_FILES;
+    for(n = 0; n < nvec; ++n) {
+      disorder_eclient_resolve(client, got_resolved_file, vec[n], cn);
+      ++gets_in_flight;
+    }
+  }
+  /* If there are no files and the directories are all read by now, we're
+   * done */
+  if(!(cn->flags & CN_GETTING_ANY))
+    filled(cn);
+  if(!gets_in_flight)
+    redisplay_tree("got_files");
 }
 
-static void got_resolved_file(void *v, const char *track) {
-  struct callbackdata *cbd = v;
-  struct choosenode *cn = cbd->u.choosenode, *file_cn;
-
-  file_cn = newnode(cn, track,
-                    trackname_transform("track", track, "display"),
-                    trackname_transform("track", track, "sort"),
-                    0/*flags*/, 0/*fill*/);
-  /* Only bother updating when we've got the lot */
-  if(--cn->pending == 0)
-    updated_node(cn, 1);
+/** @brief Called with an alias resolved filename */
+static void got_resolved_file(void *v, const char *error, const char *track) {
+  struct choosenode *const cn = v, *file_cn;
+
+  if(error) {
+    popup_protocol_error(0, error);
+  } else {
+    D(("resolved %s %s %d left", cn->path, cnflags(cn), cn->pending - 1));
+    /* TODO as below */
+    file_cn = newnode(cn, track,
+                      trackname_transform("track", track, "display"),
+                      trackname_transform("track", track, "sort"),
+                      0/*flags*/, 0/*fill*/);
+    --gets_in_flight;
+    /* Only bother updating when we've got the lot */
+    if(--cn->pending == 0) {
+      cn->flags &= ~CN_RESOLVING_FILES;
+      updated_node(cn, gets_in_flight == 0, "got_resolved_file");
+      if(!(cn->flags & CN_GETTING_ANY))
+        filled(cn);
+    }
+  }
 }
 
-/* Called with a list of directories just below some node */
-static void got_dirs(void *v, int nvec, char **vec) {
-  struct callbackdata *cbd = v;
-  struct choosenode *cn = cbd->u.choosenode;
+/** @brief Called with a list of directories just below some node */
+static void got_dirs(void *v, const char *error, int nvec, char **vec) {
+  struct choosenode *cn = v;
   int n;
+  
+  if(error) {
+    popup_protocol_error(0, error);
+    return;
+  }
 
-  D(("got_dirs %d dirs for %s", nvec, cn->path));
+  D(("got_dirs %d dirs for %s %s", nvec, cn->path, cnflags(cn)));
+  /* TODO this depends on local configuration for trackname_transform().
+   * This will work, since the defaults are now built-in, but it'll be
+   * (potentially) different to the server's configured settings.
+   *
+   * Really we want a variant of files/dirs that produces both the
+   * raw filename and the transformed name for a chosen context.
+   */
+  --gets_in_flight;
   for(n = 0; n < nvec; ++n)
     newnode(cn, vec[n],
             trackname_transform("dir", vec[n], "display"),
             trackname_transform("dir", vec[n], "sort"),
             CN_EXPANDABLE, fill_directory_node);
-  updated_node(cn, 1);
+  updated_node(cn, gets_in_flight == 0, "got_dirs");
+  cn->flags &= ~CN_GETTING_DIRS;
+  if(!(cn->flags & CN_GETTING_ANY))
+    filled(cn);
 }
   
-/* Fill a child node */
+/** @brief Fill a child node */
 static void fill_directory_node(struct choosenode *cn) {
-  struct callbackdata *cbd;
-
   D(("fill_directory_node %s", cn->path));
   /* TODO: caching */
-  /* TODO: de-dupe against fill_letter_node */
+  if(cn->flags & CN_GETTING_ANY)
+    return;
   assert(report_label != 0);
   gtk_label_set_text(GTK_LABEL(report_label), "getting files");
   clear_children(cn);
-  cbd = xmalloc(sizeof *cbd);
-  cbd->u.choosenode = cn;
-  disorder_eclient_dirs(client, got_dirs, cn->path, 0, cbd);
-  cbd = xmalloc(sizeof *cbd);
-  cbd->u.choosenode = cn;
-  disorder_eclient_files(client, got_files, cn->path, 0, cbd);
+  disorder_eclient_dirs(client, got_dirs, cn->path, 0, cn);
+  disorder_eclient_files(client, got_files, cn->path, 0, cn);
+  cn->flags |= CN_GETTING_FILES|CN_GETTING_DIRS;
+  gets_in_flight += 2;
 }
 
-/* Expand a node */
-static void expand_node(struct choosenode *cn) {
-  D(("expand_node %s", cn->path));
+/** @brief Expand a node */
+static void expand_node(struct choosenode *cn, int contingent) {
+  D(("expand_node %s %d %s", cn->path, contingent, cnflags(cn)));
   assert(cn->flags & CN_EXPANDABLE);
   /* If node is already expanded do nothing. */
   if(cn->flags & CN_EXPANDED) return;
@@ -357,17 +525,82 @@ static void expand_node(struct choosenode *cn) {
    * completed it will called updated_node() and we can redraw at that
    * point. */
   cn->flags |= CN_EXPANDED;
+  if(contingent)
+    cn->flags |= CN_CONTINGENT;
+  else
+    cn->flags &= ~CN_CONTINGENT;
+  /* If this node is not contingently expanded, mark all its parents back to
+   * the root as not contingent either, so they won't be contracted when the
+   * search results change */
+  if(!contingent) {
+    struct choosenode *cnp;
+
+    for(cnp = cn->parent; cnp; cnp = cnp->parent)
+      cnp->flags &= ~CN_CONTINGENT;
+  }
   /* TODO: visual feedback */
   cn->fill(cn);
 }
 
-/* Contract a node */
+/** @brief Make sure all the search results below @p cn are expanded
+ * @param cn Node to start at
+ */
+static void expand_from(struct choosenode *cn) {
+  int n;
+
+  if(nsearchvisible == nsearchresults)
+    /* We're done */
+    return;
+  /* Are any of the search tracks at/below this point? */
+  if(!(cn == root || hash_find(searchhash, cn->path)))
+    return;
+  D(("expand_from %d/%d visible %s", 
+     nsearchvisible, nsearchresults, cn->path));
+  if(cn->flags & CN_EXPANDABLE) {
+    if(cn->flags & CN_EXPANDED)
+      /* This node is marked as expanded already.  children.nvec might be 0,
+       * indicating that expansion is still underway.  We should get another
+       * callback when it is expanded. */
+      for(n = 0; n < cn->children.nvec && gets_in_flight < 10; ++n)
+        expand_from(cn->children.vec[n]);
+    else {
+      /* This node is not expanded yet */
+      expand_node(cn, 1);
+    }
+  } else {
+    /* This is an actual search result */
+    ++nsearchvisible;
+    progress_window_progress(spw, nsearchvisible, nsearchresults);
+    if(nsearchvisible == nsearchresults) {
+      if(suppress_redisplay) {
+        suppress_redisplay = 0;
+        redisplay_tree("all search results visible");
+      }
+      /* We've got the lot.  We make sure the first result is visible. */
+      cn = first_search_result(root);
+      gtk_adjustment_clamp_page(vadjust, cn->ymin, cn->ymax);
+    }
+  }
+}
+
+/** @brief Contract all contingently expanded nodes below @p cn */
+static void contract_contingent(struct choosenode *cn) {
+  int n;
+
+  if(cn->flags & CN_CONTINGENT)
+    contract_node(cn);
+  else
+    for(n = 0; n < cn->children.nvec; ++n)
+      contract_contingent(cn->children.vec[n]);
+}
+
+/** @brief Contract a node */
 static void contract_node(struct choosenode *cn) {
   D(("contract_node %s", cn->path));
   assert(cn->flags & CN_EXPANDABLE);
   /* If node is already contracted do nothing. */
   if(!(cn->flags & CN_EXPANDED)) return;
-  cn->flags &= ~CN_EXPANDED;
+  cn->flags &= ~(CN_EXPANDED|CN_CONTINGENT);
   /* Clear selection below this node */
   clear_selection(cn);
   /* Zot children.  We never used to do this but the result would be that over
@@ -376,10 +609,10 @@ static void contract_node(struct choosenode *cn) {
    * anyone complains we can make it an option. */
   clear_children(cn);
   /* We can contract a node immediately. */
-  redisplay_tree();
+  redisplay_tree("contract_node");
 }
 
-/* qsort callback for ordering choosenodes */
+/** @brief qsort() callback for ordering choosenodes */
 static int compare_choosenode(const void *av, const void *bv) {
   const struct choosenode *const *aa = av, *const *bb = bv;
   const struct choosenode *a = *aa, *b = *bb;
@@ -389,8 +622,9 @@ static int compare_choosenode(const void *av, const void *bv) {
                        a->path, b->path);
 }
 
-/* Called when an expandable node is updated.   */
-static void updated_node(struct choosenode *cn, int redisplay) {
+/** @brief Called when an expandable node is updated.   */
+static void updated_node(struct choosenode *cn, int redisplay,
+                         const char *why) {
   D(("updated_node %s", cn->path));
   assert(cn->flags & CN_EXPANDABLE);
   /* It might be that the node has been de-expanded since we requested the
@@ -399,143 +633,122 @@ static void updated_node(struct choosenode *cn, int redisplay) {
   /* Sort children */
   qsort(cn->children.vec, cn->children.nvec, sizeof (struct choosenode *),
         compare_choosenode);
-  if(redisplay)
-    redisplay_tree();
-}
-
-/* Searching --------------------------------------------------------------- */
+  if(redisplay) {
+    char whywhy[1024];
 
-static int compare_track_for_qsort(const void *a, const void *b) {
-  return compare_path(*(char **)a, *(char **)b);
+    snprintf(whywhy, sizeof whywhy, "updated_node %s", why);
+    redisplay_tree(whywhy);
+  }
 }
 
-/* Return true iff FILE is a child of DIR */
-static int is_child(const char *dir, const char *file) {
-  const size_t dlen = strlen(dir);
-
-  return (!strncmp(file, dir, dlen)
-          && file[dlen] == '/'
-          && strchr(file + dlen + 1, '/') == 0);
-}
+/* Searching --------------------------------------------------------------- */
 
-/* Return true iff FILE is a descendant of DIR */
-static int is_descendant(const char *dir, const char *file) {
-  const size_t dlen = strlen(dir);
+/** @brief Return true if @p track is a search result
+ *
+ * In particular the return value is one more than the index of the track
+ * @p searchresults.
+ */
+static int is_search_result(const char *track) {
+  void *r;
 
-  return !strncmp(file, dir, dlen) && file[dlen] == '/';
+  if(searchhash && (r = hash_find(searchhash, track)))
+    return 1 + *(int *)r;
+  else
+    return 0;
 }
 
-/* Called to fill a node in the search results tree */
-static void fill_search_node(struct choosenode *cn) {
+/** @brief Return the first search result at or below @p cn */
+static struct choosenode *first_search_result(struct choosenode *cn) {
   int n;
-  const size_t plen = strlen(cn->path);
-  const char *s;
-  char *dir, *last = 0;
+  struct choosenode *r;
 
-  D(("fill_search_node %s", cn->path));
-  /* We depend on the search results being sorted as by compare_path(). */
-  clear_children(cn);
-  for(n = 0; n < nsearchresults; ++n) {
-    /* We only care about descendants of CN */
-    if(!is_descendant(cn->path, searchresults[n]))
-       continue;
-    s = strchr(searchresults[n] + plen + 1, '/');
-    if(s) {
-      /* We've identified a subdirectory of CN. */
-      dir = xstrndup(searchresults[n], s - searchresults[n]);
-      if(!last || strcmp(dir, last)) {
-        /* Not a duplicate */
-        last = dir;
-        newnode(cn, dir,
-                trackname_transform("dir", dir, "display"),
-                trackname_transform("dir", dir, "sort"),
-                CN_EXPANDABLE, fill_search_node);
-      }
-    } else {
-      /* We've identified a file in CN */
-      newnode(cn, searchresults[n],
-              trackname_transform("track", searchresults[n], "display"),
-              trackname_transform("track", searchresults[n], "sort"),
-              0/*flags*/, 0/*fill*/);
-    }
-  }
-  updated_node(cn, 1);
+  if(cn->flags & CN_EXPANDABLE) {
+    for(n = 0; n < cn->children.nvec; ++n)
+      if((r = first_search_result(cn->children.vec[n])))
+        return r;
+  } else if(is_search_result(cn->path))
+    return cn;
+  return 0;
 }
 
-/* This is called from eclient with a (possibly empty) list of search results,
+/** @brief Called with a list of search results
+ *
+ * This is called from eclient with a (possibly empty) list of search results,
  * and also from initiate_seatch with an always empty list to indicate that
  * we're not searching for anything in particular. */
 static void search_completed(void attribute((unused)) *v,
+                             const char *error,
                              int nvec, char **vec) {
-  struct choosenode *cn;
   int n;
-  const char *dir;
+  char *s;
 
+  if(error) {
+    popup_protocol_error(0, error);
+    return;
+  }
+  
   search_in_flight = 0;
+  /* Contract any choosenodes that were only expanded to show search
+   * results */
+  suppress_redisplay = 1;
+  contract_contingent(root);
+  suppress_redisplay = 0;
   if(search_obsolete) {
     /* This search has been obsoleted by user input since it started.
      * Therefore we throw away the result and search again. */
     search_obsolete = 0;
     initiate_search();
   } else {
+    /* Stash the search results */
+    searchresults = vec;
+    nsearchresults = nvec;
     if(nvec) {
-      /* We will replace the choose tree with a tree structured view of search
-       * results.  First we must disabled the choose tree's widgets. */
-      delete_widgets(root);
-      /* Put the tracks into order, grouped by directory.  They'll probably
-       * come back this way anyway in current versions of the server, but it's
-       * cheap not to rely on it (compared with the massive effort we expend
-       * later on) */
-      qsort(vec, nvec, sizeof(char *), compare_track_for_qsort);
-      searchresults = vec;
-      nsearchresults = nvec;
-      cn = root = newnode(0/*parent*/, "", "Search results", "",
-                  CN_EXPANDABLE|CN_EXPANDED, fill_search_node);
-      /* Construct the initial tree.  We do this in a single pass and expand
-       * everything, so you can actually see your search results. */
-      for(n = 0; n < nsearchresults; ++n) {
-        /* Firstly we might need to go up a few directories to each an ancestor
-         * of this track */
-        while(!is_descendant(cn->path, searchresults[n])) {
-          /* We report the update on each node the last time we see it (With
-           * display=0, the main purpose of this is to get the order of the
-           * children right.) */
-          updated_node(cn, 0);
-          cn = cn->parent;
-        }
-        /* Secondly we might need to insert some new directories */
-        while(!is_child(cn->path, searchresults[n])) {
-          /* Figure out the subdirectory */
-          dir = xstrndup(searchresults[n],
-                         strchr(searchresults[n] + strlen(cn->path) + 1,
-                                '/') - searchresults[n]);
-          cn = newnode(cn, dir,
-                       trackname_transform("dir", dir, "display"),
-                       trackname_transform("dir", dir, "sort"),
-                       CN_EXPANDABLE|CN_EXPANDED, fill_search_node);
+      /* Create a new search hash for fast identification of results */
+      searchhash = hash_new(sizeof(int));
+      for(n = 0; n < nvec; ++n) {
+        int *const ip = xmalloc(sizeof (int *));
+        static const int minus_1 = -1;
+        *ip = n;
+        /* The filename itself lives in the hash */
+        hash_add(searchhash, vec[n], ip, HASH_INSERT_OR_REPLACE);
+        /* So do its ancestor directories */
+        for(s = vec[n] + 1; *s; ++s) {
+          if(*s == '/') {
+            *s = 0;
+            hash_add(searchhash, vec[n], &minus_1, HASH_INSERT_OR_REPLACE);
+            *s = '/';
+          }
         }
-        /* Finally we can insert the track as a child of the current
-         * directory */
-        newnode(cn, searchresults[n],
-                trackname_transform("track", searchresults[n], "display"),
-                trackname_transform("track", searchresults[n], "sort"),
-                0/*flags*/, 0/*fill*/);
       }
-      while(cn) {
-        /* Update all the nodes back up to the root */
-        updated_node(cn, 0);
-        cn = cn->parent;
+      /* We don't yet know that the results are visible */
+      nsearchvisible = 0;
+      if(spw) {
+        progress_window_progress(spw, 0, 0);
+        spw = 0;
       }
-      /* Now it's worth displaying the tree */
-      redisplay_tree();
-    } else if(root != realroot) {
-      delete_widgets(root);
-      root = realroot;
-      redisplay_tree();
+      if(nsearchresults > 50)
+        spw = progress_window_new("Fetching search results");
+      /* Initiate expansion */
+      expand_from(root);
+      /* The search results buttons are usable */
+      gtk_widget_set_sensitive(nextsearch, 1);
+      gtk_widget_set_sensitive(prevsearch, 1);
+      suppress_redisplay = 1;           /* avoid lots of redisplays */
+    } else {
+      searchhash = 0;                   /* for the gc */
+      redisplay_tree("no search results"); /* remove search markers */
+      /* The search results buttons are not usable */
+      gtk_widget_set_sensitive(nextsearch, 0);
+      gtk_widget_set_sensitive(prevsearch, 0);
     }
   }
 }
 
+/** @brief Initiate a search 
+ *
+ * If a search is underway we set @ref search_obsolete and restart the search
+ * in search_completed() above.
+ */
 static void initiate_search(void) {
   char *terms, *e;
 
@@ -558,25 +771,71 @@ static void initiate_search(void) {
       /* The search terms are bad!  We treat this as if there were no search
        * terms at all.  Some kind of feedback would be handy. */
       fprintf(stderr, "bad terms [%s]\n", terms); /* TODO */
-      search_completed(0, 0, 0);
+      search_completed(0, 0, 0, 0);
     } else {
       search_in_flight = 1;
     }
   } else {
     /* No search terms - we want to see all tracks */
-    search_completed(0, 0, 0);
+    search_completed(0, 0, 0, 0);
   }
 }
 
-/* Called when the cancel search button is clicked */
+/** @brief Called when the cancel search button is clicked */
 static void clearsearch_clicked(GtkButton attribute((unused)) *button,
                                 gpointer attribute((unused)) userdata) {
   gtk_entry_set_text(GTK_ENTRY(searchentry), "");
+  /* Put the input focus back */
+  gtk_widget_grab_focus(searchentry);
+}
+
+/** @brief Called when the 'next search result' button is clicked */
+static void next_clicked(GtkButton attribute((unused)) *button,
+                         gpointer attribute((unused)) userdata) {
+  /* We want to find the highest (lowest ymax) track that is below the current
+   * visible range */
+  int n;
+  const gdouble bottom = gtk_adjustment_get_value(vadjust) + vadjust->page_size;
+  const struct choosenode *candidate = 0;
+
+  for(n = 0; n < nsearchresults; ++n) {
+    const struct choosenode *const cn = searchnodes[n];
+
+    if(cn
+       && cn->ymax > bottom
+       && (candidate == 0
+           || cn->ymax < candidate->ymax))
+      candidate = cn;
+  }
+  if(candidate)
+    gtk_adjustment_clamp_page(vadjust, candidate->ymin, candidate->ymax);
+}
+
+/** @brief Called when the 'previous search result' button is clicked */
+static void prev_clicked(GtkButton attribute((unused)) *button,
+                         gpointer attribute((unused)) userdata) {
+  /* We want to find the lowest (greated ymax) track that is above the current
+   * visible range */
+  int n;
+  const gdouble top = gtk_adjustment_get_value(vadjust);
+  const struct choosenode *candidate = 0;
+
+  for(n = 0; n < nsearchresults; ++n) {
+    const struct choosenode *const cn = searchnodes[n];
+
+    if(cn
+       && cn->ymin  < top
+       && (candidate == 0
+           || cn->ymax > candidate->ymax))
+      candidate = cn;
+  }
+  if(candidate)
+    gtk_adjustment_clamp_page(vadjust, candidate->ymin, candidate->ymax);
 }
 
 /* Display functions ------------------------------------------------------- */
 
-/* Delete all the widgets in the tree */
+/** @brief Delete all the widgets in the tree */
 static void delete_widgets(struct choosenode *cn) {
   int n;
 
@@ -587,19 +846,34 @@ static void delete_widgets(struct choosenode *cn) {
   files_selected = 0;
 }
 
-/* Update the display */
-static void redisplay_tree(void) {
+/** @brief Update the display */
+static void redisplay_tree(const char *why) {
   struct displaydata d;
   guint oldwidth, oldheight;
 
-  D(("redisplay_tree"));
+  D(("redisplay_tree %s", why));
+  if(suppress_redisplay) {
+    /*fprintf(stderr, "redisplay_tree %s suppressed\n", why);*/
+    return;
+  }
+  if(gets_in_flight) {
+    /*fprintf(stderr, "redisplay_tree %s suppressed (gets_in_flight)\n", why);*/
+    return;
+  }
+  INIT();
+  BEGIN(total);
+  /*fprintf(stderr, "redisplay_tree %s   *** NOT SUPPRESSED ***\n", why);*/
   /* We'll count these up empirically each time */
   files_selected = 0;
   files_visible = 0;
   /* Correct the layout and find out how much space it uses */
   MTAG_PUSH("display_tree");
+  searchnodes = nsearchresults ? xcalloc(nsearchresults, 
+                                         sizeof (struct choosenode *)) : 0;
   d = display_tree(root, 0, 0);
   MTAG_POP();
+
+  BEGIN(gtkbits);
   /* We must set the total size or scrolling will not work (it wouldn't be hard
    * for GtkLayout to figure it out for itself but presumably you're supposed
    * to be able to have widgets off the edge of the layuot.)
@@ -613,17 +887,29 @@ static void redisplay_tree(void) {
   if(oldwidth > d.width || oldheight > d.height)
     gtk_widget_queue_draw(chooselayout);
   gtk_layout_set_size(GTK_LAYOUT(chooselayout), d.width, d.height);
+  END(gtkbits);
   /* Notify the main menu of any recent changes */
+  BEGIN(menuupdate);
   menu_update(-1);
+  END(menuupdate);
+  END(total);
+  REPORT();
 }
 
-/* Make sure all displayed widgets from CN down exist and are in their proper
- * place and return the vertical space used. */
+/** @brief Recursive step for redisplay_tree()
+ * @param cn Node to display
+ * @param x X coordinate for @p cn
+ * @param y Y coordinate for @p cn
+ *
+ * Makes sure all displayed widgets from CN down exist and are in their proper
+ * place and return the maximum space used.
+ */
 static struct displaydata display_tree(struct choosenode *cn, int x, int y) {
   int n, aw;
   GtkRequisition req;
   struct displaydata d, cd;
   GdkPixbuf *pb;
+  const int search_result = is_search_result(cn->path);
   
   D(("display_tree %s %d,%d", cn->path, x, y));
 
@@ -633,6 +919,7 @@ static struct displaydata display_tree(struct choosenode *cn, int x, int y) {
    * A non-expandable item has just a text label and no arrow.
    */
   if(!cn->container) {
+    BEGIN(new_widgets);
     MTAG_PUSH("make_widgets_1");
     /* Widgets need to be created */
     NW(hbox);
@@ -669,14 +956,24 @@ static struct displaydata display_tree(struct choosenode *cn, int x, int y) {
     g_signal_connect(cn->container, "button-press-event", 
                      G_CALLBACK(clicked_choosenode), cn);
     g_object_ref(cn->container);
-    gtk_widget_set_name(cn->label, "choose");
-    gtk_widget_set_name(cn->container, "choose");
     /* Show everything by default */
     gtk_widget_show_all(cn->container);
     MTAG_POP();
+    END(new_widgets);
   }
   assert(cn->container);
+  /* Set colors */
+  BEGIN(colors);
+  if(search_result) {
+    gtk_widget_set_style(cn->container, search_style);
+    gtk_widget_set_style(cn->label, search_style);
+  } else {
+    gtk_widget_set_style(cn->container, layout_style);
+    gtk_widget_set_style(cn->label, layout_style);
+  }
+  END(colors);
   /* Make sure the icon is right */
+  BEGIN(markers);
   if(cn->flags & CN_EXPANDABLE)
     gtk_arrow_set(GTK_ARROW(cn->arrow),
                   cn->flags & CN_EXPANDED ? GTK_ARROW_DOWN : GTK_ARROW_RIGHT,
@@ -685,7 +982,9 @@ static struct displaydata display_tree(struct choosenode *cn, int x, int y) {
     /* Make sure the queued marker is right */
     /* TODO: doesn't always work */
     (queued(cn->path) ? gtk_widget_show : gtk_widget_hide)(cn->marker);
+  END(markers);
   /* Put the widget in the right place */
+  BEGIN(location);
   if(cn->flags & CN_DISPLAYED)
     gtk_layout_move(GTK_LAYOUT(chooselayout), cn->container, x, y);
   else {
@@ -694,13 +993,18 @@ static struct displaydata display_tree(struct choosenode *cn, int x, int y) {
     /* Now chooselayout has a ref to the container */
     g_object_unref(cn->container);
   }
+  END(location);
   /* Set the widget's selection status */
+  BEGIN(selection);
   if(!(cn->flags & CN_EXPANDABLE))
     display_selection(cn);
+  END(selection);
   /* Find the size used so we can get vertical positioning right. */
   gtk_widget_size_request(cn->container, &req);
   d.width = x + req.width;
   d.height = y + req.height;
+  cn->ymin = y;
+  cn->ymax = d.height;
   if(cn->flags & CN_EXPANDED) {
     /* We'll offset children by the size of the arrow whatever it might be. */
     assert(cn->arrow);
@@ -713,21 +1017,26 @@ static struct displaydata display_tree(struct choosenode *cn, int x, int y) {
       d.height = cd.height;
     }
   } else {
+    BEGIN(undisplay);
     for(n = 0; n < cn->children.nvec; ++n)
       undisplay_tree(cn->children.vec[n]);
+    END(undisplay);
   }
   if(!(cn->flags & CN_EXPANDABLE)) {
     ++files_visible;
     if(cn->flags & CN_SELECTED)
       ++files_selected;
   }
+  /* update the search results array */
+  if(search_result)
+    searchnodes[search_result - 1] = cn;
   /* report back how much space we used */
   D(("display_tree %s %d,%d total size %dx%d", cn->path, x, y,
      d.width, d.height));
   return d;
 }
 
-/* Remove widgets for newly hidden nodes */
+/** @brief Remove widgets for newly hidden nodes */
 static void undisplay_tree(struct choosenode *cn) {
   int n;
 
@@ -744,6 +1053,7 @@ static void undisplay_tree(struct choosenode *cn) {
 
 /* Selection --------------------------------------------------------------- */
 
+/** @brief Mark the widget @p cn according to its selection state */
 static void display_selection(struct choosenode *cn) {
   /* Need foreground and background colors */
   gtk_widget_set_state(cn->label, (cn->flags & CN_SELECTED
@@ -752,8 +1062,9 @@ static void display_selection(struct choosenode *cn) {
                                        ? GTK_STATE_SELECTED : GTK_STATE_NORMAL));
 }
 
-/* Set the selection state of a widget.  Directories can never be selected, we
- * just ignore attempts to do so. */
+/** @brief Set the selection state of a widget
+ *
+ * Directories can never be selected, we just ignore attempts to do so. */
 static void set_selection(struct choosenode *cn, int selected) {
   unsigned f = selected ? CN_SELECTED : 0;
 
@@ -771,7 +1082,7 @@ static void set_selection(struct choosenode *cn, int selected) {
   }
 }
 
-/* Recursively clear all selection bits from CN down */
+/** @brief Recursively clear all selection bits from CN down */
 static void clear_selection(struct choosenode *cn) {
   int n;
 
@@ -782,7 +1093,17 @@ static void clear_selection(struct choosenode *cn) {
 
 /* User actions ------------------------------------------------------------ */
 
-/* Clicked on something */
+/** @brief Called when disorder_eclient_play completes */
+void play_completed(void attribute((unused)) *v,
+                    const char *error) {
+  if(error)
+    popup_protocol_error(0, error);
+}
+
+/** @brief Clicked on something
+ *
+ * This implements playing, all the modifiers for selection, etc.
+ */
 static void clicked_choosenode(GtkWidget attribute((unused)) *widget,
                                GdkEventButton *event,
                                gpointer user_data) {
@@ -798,7 +1119,7 @@ static void clicked_choosenode(GtkWidget attribute((unused)) *widget,
       if(cn->flags & CN_EXPANDED)
         contract_node(cn);
       else
-        expand_node(cn);
+        expand_node(cn, 0/*!contingent*/);
       last_click = 0;
     } else {
       /* This is a file.  Adjust selection status */
@@ -841,6 +1162,8 @@ static void clicked_choosenode(GtkWidget attribute((unused)) *widget,
               last_click = cn;
           }
         }
+        /* TODO trying to select a range that doesn't share a single parent
+         * currently does not work, but it ought to. */
         break;
       }
     }
@@ -851,11 +1174,15 @@ static void clicked_choosenode(GtkWidget attribute((unused)) *widget,
       clear_selection(root);
       set_selection(cn, 1);
       gtk_label_set_text(GTK_LABEL(report_label), "adding track to queue");
-      disorder_eclient_play(client, cn->path, 0, 0);
+      disorder_eclient_play(client, cn->path, play_completed, 0);
       last_click = 0;
     }
   } else if(event->type == GDK_BUTTON_PRESS
      && event->button == 3) {
+    struct choose_menuitem *const menuitems =
+      (cn->flags & CN_EXPANDABLE ? dir_menuitems : track_menuitems);
+    GtkWidget *const menu =
+      (cn->flags & CN_EXPANDABLE ? dir_menu : track_menu);
     /* Right click.  Pop up a menu. */
     /* If the current file isn't selected, switch the selection to just that.
      * (If we're looking at a directory then leave the selection alone.) */
@@ -865,7 +1192,7 @@ static void clicked_choosenode(GtkWidget attribute((unused)) *widget,
       last_click = cn;
     }
     /* Set the item sensitivity and callbacks */
-    for(n = 0; n < NMENUITEMS; ++n) {
+    for(n = 0; menuitems[n].name; ++n) {
       if(menuitems[n].handlerid)
         g_signal_handler_disconnect(menuitems[n].w,
                                     menuitems[n].handlerid);
@@ -874,6 +1201,7 @@ static void clicked_choosenode(GtkWidget attribute((unused)) *widget,
       menuitems[n].handlerid = g_signal_connect
         (menuitems[n].w, "activate", G_CALLBACK(menuitems[n].activate), cn);
     }
+    set_tool_colors(menu);
     /* Pop up the menu */
     gtk_widget_show_all(menu);
     gtk_menu_popup(GTK_MENU(menu), 0, 0, 0, 0,
@@ -881,13 +1209,15 @@ static void clicked_choosenode(GtkWidget attribute((unused)) *widget,
   }
 }
 
+/** @brief Called BY GTK+ to tell us the search entry box has changed */
 static void searchentry_changed(GtkEditable attribute((unused)) *editable,
                                 gpointer attribute((unused)) user_data) {
   initiate_search();
 }
 
-/* Menu items -------------------------------------------------------------- */
+/* Track menu items -------------------------------------------------------- */
 
+/** @brief Recursive step for gather_selected() */
 static void recurse_selected(struct choosenode *cn, struct vector *v) {
   int n;
 
@@ -901,84 +1231,199 @@ static void recurse_selected(struct choosenode *cn, struct vector *v) {
   }
 }
 
-static char **gather_selected(int *ntracks) {
+/*** @brief Get a list of all the selected tracks */
+static const char **gather_selected(int *ntracks) {
   struct vector v;
 
   vector_init(&v);
   recurse_selected(root, &v);
   vector_terminate(&v);
   if(ntracks) *ntracks = v.nvec;
-  return v.vec;
+  return (const char **)v.vec;
 }
 
-static void activate_play(GtkMenuItem attribute((unused)) *menuitem,
-                          gpointer attribute((unused)) user_data) {
-  char **tracks = gather_selected(0);
+/** @brief Called when the track menu's play option is activated */
+static void activate_track_play(GtkMenuItem attribute((unused)) *menuitem,
+                                gpointer attribute((unused)) user_data) {
+  const char **tracks = gather_selected(0);
   int n;
   
   gtk_label_set_text(GTK_LABEL(report_label), "adding track to queue");
   for(n = 0; tracks[n]; ++n)
-    disorder_eclient_play(client, tracks[n], 0, 0);
-}
-
-#if 0
-static void activate_remove(GtkMenuItem attribute((unused)) *menuitem,
-                            gpointer attribute((unused)) user_data) {
-  /* TODO remove all selected tracks */
+    disorder_eclient_play(client, tracks[n], play_completed, 0);
 }
-#endif
 
-static void activate_properties(GtkMenuItem attribute((unused)) *menuitem,
-                                gpointer attribute((unused)) user_data) {
+/** @brief Called when the menu's properties option is activated */
+static void activate_track_properties(GtkMenuItem attribute((unused)) *menuitem,
+                                      gpointer attribute((unused)) user_data) {
   int ntracks;
-  char **tracks = gather_selected(&ntracks);
+  const char **tracks = gather_selected(&ntracks);
 
   properties(ntracks, tracks);
 }
 
-static gboolean sensitive_play(struct choosenode attribute((unused)) *cn) {
+/** @brief Determine whether the menu's play option should be sensitive */
+static gboolean sensitive_track_play(struct choosenode attribute((unused)) *cn) {
   return (!!files_selected
-          && (disorder_eclient_state(client) & DISORDER_CONNECTED));
+          && (disorder_eclient_state(client) & DISORDER_CONNECTED)
+          && (last_rights & RIGHT_PLAY));
 }
 
-#if 0
-static gboolean sensitive_remove(struct choosenode attribute((unused)) *cn) {
-  return FALSE;                         /* not implemented yet */
+/** @brief Determine whether the menu's properties option should be sensitive */
+static gboolean sensitive_track_properties(struct choosenode attribute((unused)) *cn) {
+  return (!!files_selected
+          && (disorder_eclient_state(client) & DISORDER_CONNECTED)
+          && (last_rights & RIGHT_PREFS));
 }
-#endif
 
-static gboolean sensitive_properties(struct choosenode attribute((unused)) *cn) {
-  return !!files_selected && (disorder_eclient_state(client) & DISORDER_CONNECTED);
+/* Directory menu items ---------------------------------------------------- */
+
+/** @brief Return the file children of @p cn
+ *
+ * The list is terminated by a null pointer.
+ */
+static const char **dir_files(struct choosenode *cn, int *nfiles) {
+  const char **files = xcalloc(cn->children.nvec + 1, sizeof (char *));
+  int n, m;
+
+  for(n = m = 0; n < cn->children.nvec; ++n) 
+    if(!(cn->children.vec[n]->flags & CN_EXPANDABLE))
+      files[m++] = cn->children.vec[n]->path;
+  files[m] = 0;
+  if(nfiles) *nfiles = m;
+  return files;
 }
 
-/* Main menu plumbing ------------------------------------------------------ */
+static void play_dir(struct choosenode *cn,
+                     void attribute((unused)) *wfu) {
+  int ntracks, n;
+  const char **tracks = dir_files(cn, &ntracks);
+  
+  gtk_label_set_text(GTK_LABEL(report_label), "adding track to queue");
+  for(n = 0; n < ntracks; ++n)
+    disorder_eclient_play(client, tracks[n], play_completed, 0);
+}
 
-static int choose_properties_sensitive(GtkWidget attribute((unused)) *w) {
-  return !!files_selected && (disorder_eclient_state(client) & DISORDER_CONNECTED);
+static void properties_dir(struct choosenode *cn,
+                           void attribute((unused)) *wfu) {
+  int ntracks;
+  const char **tracks = dir_files(cn, &ntracks);
+  
+  properties(ntracks, tracks);
+}
+
+static void select_dir(struct choosenode *cn,
+                       void attribute((unused)) *wfu) {
+  int n;
+
+  clear_selection(root);
+  for(n = 0; n < cn->children.nvec; ++n) 
+    set_selection(cn->children.vec[n], 1);
+}
+
+/** @brief Ensure @p cn is expanded and then call @p callback */
+static void call_with_dir(struct choosenode *cn,
+                          when_filled_callback *whenfilled,
+                          void *wfu) {
+  if(!(cn->flags & CN_EXPANDABLE))
+    return;                             /* something went wrong */
+  if(cn->flags & CN_EXPANDED)
+    /* @p cn is already open */
+    whenfilled(cn, wfu);
+  else {
+    /* @p cn is not open, arrange for the callback to go off when it is
+     * opened */
+    cn->whenfilled = whenfilled;
+    cn->wfu = wfu;
+    expand_node(cn, 0/*not contingnet upon search*/);
+  }
+}
+
+/** @brief Called when the directory menu's play option is activated */
+static void activate_dir_play(GtkMenuItem attribute((unused)) *menuitem,
+                              gpointer user_data) {
+  struct choosenode *const cn = (struct choosenode *)user_data;
+
+  call_with_dir(cn, play_dir, 0);
+}
+
+/** @brief Called when the directory menu's properties option is activated */
+static void activate_dir_properties(GtkMenuItem attribute((unused)) *menuitem,
+                                    gpointer user_data) {
+  struct choosenode *const cn = (struct choosenode *)user_data;
+
+  call_with_dir(cn, properties_dir, 0);
+}
+
+/** @brief Called when the directory menu's select option is activated */
+static void activate_dir_select(GtkMenuItem attribute((unused)) *menuitem,
+                                gpointer user_data) {
+  struct choosenode *const cn = (struct choosenode *)user_data;
+
+  call_with_dir(cn, select_dir,  0);
 }
 
-static int choose_selectall_sensitive(GtkWidget attribute((unused)) *w) {
-  return FALSE;                         /* TODO */
+/** @brief Determine whether the directory menu's play option should be sensitive */
+static gboolean sensitive_dir_play(struct choosenode attribute((unused)) *cn) {
+  return !!(disorder_eclient_state(client) & DISORDER_CONNECTED);
 }
 
-static void choose_properties_activate(GtkWidget attribute((unused)) *w) {
-  activate_properties(0, 0);
+/** @brief Determine whether the directory menu's properties option should be sensitive */
+static gboolean sensitive_dir_properties(struct choosenode attribute((unused)) *cn) {
+  return !!(disorder_eclient_state(client) & DISORDER_CONNECTED);
 }
 
-static void choose_selectall_activate(GtkWidget attribute((unused)) *w) {
-  /* TODO */
+/** @brief Determine whether the directory menu's select option should be sensitive */
+static gboolean sensitive_dir_select(struct choosenode attribute((unused)) *cn) {
+  return TRUE;
 }
 
+/* Main menu plumbing ------------------------------------------------------ */
+
+/** @brief Determine whether the edit menu's properties option should be sensitive */
+static int choose_properties_sensitive(void attribute((unused)) *extra) {
+  return !!files_selected && (disorder_eclient_state(client) & DISORDER_CONNECTED);
+}
+
+/** @brief Called when the edit menu's properties option is activated */
+static void choose_properties_activate(void attribute((unused)) *extra) {
+  activate_track_properties(0, 0);
+}
+
+/** @brief Called when the choose tab is selected */
+static void choose_tab_selected(void) {
+  gtk_widget_grab_focus(searchentry);
+}
+
+/** @brief Main menu callbacks for Choose screen */
 static const struct tabtype tabtype_choose = {
   choose_properties_sensitive,
-  choose_selectall_sensitive,
+  NULL/*choose_selectall_sensitive*/,
+  NULL/*choose_selectnone_sensitive*/,
   choose_properties_activate,
-  choose_selectall_activate,
+  NULL/*choose_selectall_activate*/,
+  NULL/*choose_selectnone_activate*/,
+  choose_tab_selected,
+  0
 };
 
 /* Public entry points ----------------------------------------------------- */
 
-/* Create a track choice widget */
+/** @brief Called when we have just logged in
+ *
+ * Entirely resets the choose tab.
+ */
+static void choose_logged_in(const char attribute((unused)) *event,
+                             void attribute((unused)) *eventdata,
+                             void attribute((unused)) *callbackdata) {
+  if(root)
+    undisplay_tree(root);
+  root = newnode(0/*parent*/, "<root>", "All files", "",
+                 CN_EXPANDABLE, fill_root_node);
+  expand_node(root, 0);                 /* will call redisplay_tree */
+}
+
+/** @brief Create a track choice widget */
 GtkWidget *choose_widget(void) {
   int n;
   GtkWidget *scrolled;
@@ -1007,41 +1452,76 @@ GtkWidget *choose_widget(void) {
   /* Text entry box for search terms */
   NW(entry);
   searchentry = gtk_entry_new();
+  gtk_widget_set_style(searchentry, tool_style);
   g_signal_connect(searchentry, "changed", G_CALLBACK(searchentry_changed), 0);
+  gtk_tooltips_set_tip(tips, searchentry, "Enter search terms here; search is automatic", "");
 
   /* Cancel button to clear the search */
   NW(button);
   clearsearch = gtk_button_new_from_stock(GTK_STOCK_CANCEL);
+  gtk_widget_set_style(clearsearch, tool_style);
   g_signal_connect(G_OBJECT(clearsearch), "clicked",
                    G_CALLBACK(clearsearch_clicked), 0);
-
-  /* hbox packs the search box and the cancel button together on a line */
+  gtk_tooltips_set_tip(tips, clearsearch, "Clear search terms", "");
+
+  /* Up and down buttons to find previous/next results; initially they are not
+   * usable as there are no search results. */
+  prevsearch = iconbutton("up.png", "Previous search result");
+  g_signal_connect(G_OBJECT(prevsearch), "clicked",
+                   G_CALLBACK(prev_clicked), 0);
+  gtk_widget_set_style(prevsearch, tool_style);
+  gtk_widget_set_sensitive(prevsearch, 0);
+  nextsearch = iconbutton("down.png", "Next search result");
+  g_signal_connect(G_OBJECT(nextsearch), "clicked",
+                   G_CALLBACK(next_clicked), 0);
+  gtk_widget_set_style(nextsearch, tool_style);
+  gtk_widget_set_sensitive(nextsearch, 0);
+
+  /* hbox packs the search tools button together on a line */
   NW(hbox);
   hbox = gtk_hbox_new(FALSE/*homogeneous*/, 1/*spacing*/);
   gtk_box_pack_start(GTK_BOX(hbox), searchentry,
                      TRUE/*expand*/, TRUE/*fill*/, 0/*padding*/);
-  gtk_box_pack_end(GTK_BOX(hbox), clearsearch,
-                   FALSE/*expand*/, FALSE/*fill*/, 0/*padding*/);
+  gtk_box_pack_start(GTK_BOX(hbox), prevsearch,
+                     FALSE/*expand*/, FALSE/*fill*/, 0/*padding*/);
+  gtk_box_pack_start(GTK_BOX(hbox), nextsearch,
+                     FALSE/*expand*/, FALSE/*fill*/, 0/*padding*/);
+  gtk_box_pack_start(GTK_BOX(hbox), clearsearch,
+                     FALSE/*expand*/, FALSE/*fill*/, 0/*padding*/);
   
   /* chooselayout contains the currently visible subset of the track
    * namespace */
   NW(layout);
   chooselayout = gtk_layout_new(0, 0);
-  root = newnode(0/*parent*/, "<root>", "All files", "",
-                 CN_EXPANDABLE, fill_root_node);
-  realroot = root;
-  expand_node(root);                    /* will call redisplay_tree */
-  /* Create the popup menu */
+  gtk_widget_set_style(chooselayout, layout_style);
+  choose_logged_in(0, 0, 0);
+  event_register("logged-in", choose_logged_in, 0);
+  /* Create the popup menus */
+  NW(menu);
+  track_menu = gtk_menu_new();
+  g_signal_connect(track_menu, "destroy", G_CALLBACK(gtk_widget_destroyed),
+                   &track_menu);
+  for(n = 0; track_menuitems[n].name; ++n) {
+    NW(menu_item);
+    track_menuitems[n].w = 
+      gtk_menu_item_new_with_label(track_menuitems[n].name);
+    gtk_menu_attach(GTK_MENU(track_menu), track_menuitems[n].w,
+                    0, 1, n, n + 1);
+  }
   NW(menu);
-  menu = gtk_menu_new();
-  g_signal_connect(menu, "destroy", G_CALLBACK(gtk_widget_destroyed), &menu);
-  for(n = 0; n < NMENUITEMS; ++n) {
+  dir_menu = gtk_menu_new();
+  g_signal_connect(dir_menu, "destroy", G_CALLBACK(gtk_widget_destroyed),
+                   &dir_menu);
+  for(n = 0; dir_menuitems[n].name; ++n) {
     NW(menu_item);
-    menuitems[n].w = gtk_menu_item_new_with_label(menuitems[n].name);
-    gtk_menu_attach(GTK_MENU(menu), menuitems[n].w, 0, 1, n, n + 1);
+    dir_menuitems[n].w = 
+      gtk_menu_item_new_with_label(dir_menuitems[n].name);
+    gtk_menu_attach(GTK_MENU(dir_menu), dir_menuitems[n].w,
+                    0, 1, n, n + 1);
   }
   /* The layout is scrollable */
-  scrolled = scroll_widget(GTK_WIDGET(chooselayout), "choose");
+  scrolled = scroll_widget(chooselayout);
+  vadjust = gtk_layout_get_vadjustment(GTK_LAYOUT(chooselayout));
 
   /* The scrollable layout and the search hbox go together in a vbox */
   NW(vbox);
@@ -1055,9 +1535,9 @@ GtkWidget *choose_widget(void) {
   return vbox;
 }
 
-/* Called when something we care about here might have changed */
+/** @brief Called when something we care about here might have changed */
 void choose_update(void) {
-  redisplay_tree();
+  redisplay_tree("choose_update");
 }
 
 /*