Whoops. Missed off the new script.

[rocl] / graph.c

/* -*-c-*-
 *
 * $Id: graph.c,v 1.1 2003/03/07 00:45:13 mdw Exp $
 *
 * Graph theory stuff
 *
 * (c) 2003 Mark Wooding
 */

/*----- Licensing notice --------------------------------------------------* 
 *
 * 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
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software Foundation,
 * Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
 */

/*----- Revision history --------------------------------------------------* 
 *
 * $Log: graph.c,v $
 * Revision 1.1  2003/03/07 00:45:13  mdw
 * Graph theory functions.
 *
 */

/*----- Header files ------------------------------------------------------*/

#include <assert.h>
#include <math.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include <tcl.h>

#include "vec.h"

/*----- Static variables --------------------------------------------------*/

#define INF ((unsigned long)-1)

/*----- Utility functions -------------------------------------------------*/

static int err(Tcl_Interp *ti, /*const*/ char *p)
{
  Tcl_SetResult(ti, p, TCL_STATIC);
  return (TCL_ERROR);
}

/* --- @import@ --- *
 *
 * Arguments:	@Tcl_Interp *ti@ = interpreter to leave errors in
 *		@vec *v@ = pointer to input adjacency matrix
 *		@unsigned long *tt@ = pointer to output adjacency matrix
 *		@size_t *nn@ = where to put the table size
 *
 * Returns:	Tcl return code.
 *
 * Use:		Imports an adjacency matrix.
 */

static int import(Tcl_Interp *ti, vec *v, unsigned long **tt, size_t *nn)
{
  size_t i;
  unsigned long *t;
  size_t n;

  /* --- Check the table is well-formed --- */

  if (v->ndim != 2)
    return (err(ti, "adjacency matrix must be two-dimensional"));
  if (v->dim[0].lo != 0 || v->dim[1].lo || v->dim[0].hi != v->dim[1].hi)
    return (err(ti, "adjacency matrix must be square and zero-origin"));
  n = *nn = v->dim[0].hi;

  /* --- Copy the data over --- */

  n *= n;
  assert(n == v->n);
  t = (void *)Tcl_Alloc(n * sizeof(*t));
  for (i = 0; i < n; i++) {
    long l;
    if (Tcl_GetLongFromObj(ti, v->v[i], &l) != TCL_OK) {
      Tcl_Free((void *)t);
      return (TCL_ERROR);
    }
    t[i] = l >= 0 ? l : INF;
  }
  *tt = t;
  return (TCL_OK);
}

/* --- @export@ --- *
 *
 * Arguments:	@Tcl_Interp *ti@ = interpreter to create output vector
 *		@unsigned long *t@ = pointer to table
 *		@size_t n@ = size of the table
 *
 * Returns:	A pointer to the vector, or null.
 *
 * Use:		Exports an adjacency matrix.
 */

static vec *export(Tcl_Interp *ti, unsigned long *t, size_t n)
{
  vec_bound b[2];
  vec *v;
  size_t i;
  Tcl_Obj *o;

  b[0].lo = b[1].lo = 0;
  b[0].hi = b[1].hi = n;
  if ((v = vec_create(ti, 2, b, 0)) == 0)
    return (0);
  o = Tcl_NewLongObj(-1);
  Tcl_IncrRefCount(o);
  for (i = 0; i < v->n; i++) {
    v->v[i] = t[i] == INF ? o : Tcl_NewLongObj(t[i]);
    Tcl_IncrRefCount(v->v[i]);
  }
  Tcl_DecrRefCount(o);
  return (v);
}

/*----- Floyd-Warshall all-points shortest path ---------------------------*/

/* --- @graph-shortest-path VEC@ --- *
 *
 * Returns a pair of vectors containing, respectively, the shortest path
 * length and the successor element in the shortest path.  If you say
 *
 *   destructure {len path} [graph-shortest-path $v]
 *
 * then [$len get I J] is the shortest path length from node I to node J, and
 * [$path get I J] is the first hop on that shortest path.  (To compute the
 * entire path, set K to be that first hop; the next hop is then [$path get K
 * J], and so on.)
 *
 * The adjacency matrix is given in VEC: negative entries indicate no path;
 * nonnegative entries are weights.  All entries must be integers.
 */

static int cmd_shortestpath(ClientData cd, Tcl_Interp *ti,
			    int objc, Tcl_Obj *const *objv)
{
  vec *v, *lv = 0, *pv = 0;
  size_t n, i, j, k;
  unsigned long *a = 0, *p = 0;
  Tcl_Obj *o;

  /* --- Read in the arguments --- */

  if (objc != 2) {
    err(ti, "usage: graph-shortest-path VEC");
    goto fail;
  }
  if ((v = vec_find(ti, objv[1])) == 0 || import(ti, v, &a, &n) != TCL_OK)
    goto fail;

  /* --- Set up the path table --- */

  p = (void *)Tcl_Alloc(n * n * sizeof(*p));
  for (i = 0; i < n; i++) {
    for (j = 0; j < n; j++)
      p[i * n + j] = j;
    p[i * n + i] = INF;
  }

  /* --- Do the main algorithm --- *
   *
   * Not so hard.  Just brute force and ignorance.
   */

  for (k = 0; k < n; k++) {
    for (i = 0; i < n; i++) {
      for (j = 0; j < n; j++) {
	if (a[i * n + k] != INF && a[k * n + j] != INF &&
	    a[i * n + k] + a[k * n + j] < a[i * n + j]) {
	  a[i * n + j] = a[i * n + k] + a[k * n + j];
	  p[i * n + j] = p[i * n + k];
	}
      }
    }
  }

  /* --- Wrap up --- */

  if ((lv = export(ti, a, n)) == 0 || (pv = export(ti, p, n)) == 0)
    goto fail;
  o = Tcl_NewListObj(0, 0);
  Tcl_ListObjAppendElement
    (ti, o, Tcl_NewStringObj(Tcl_GetCommandName(ti, lv->c), -1));
  Tcl_ListObjAppendElement
    (ti, o, Tcl_NewStringObj(Tcl_GetCommandName(ti, pv->c), -1));
  Tcl_SetObjResult(ti, o);
  Tcl_Free((void *)a);
  Tcl_Free((void *)p);
  return (TCL_OK);

fail:
  if (a) Tcl_Free((void *)a);
  if (p) Tcl_Free((void *)p);
  if (lv) vec_destroy(ti, lv);
  if (pv) vec_destroy(ti, pv);
  return (TCL_ERROR);
}

/*----- Travelling Salesman Problem ---------------------------------------*/

/* --- @rrange@ --- *
 *
 * Arguments:	@size_t max@ = maximum number wanted
 *
 * Returns:	An integer uniformly distributed on %$[0, max)$%.
 */

static size_t rrange(size_t max)
{
  size_t m, z, r;

  z = RAND_MAX/max;
  m = z * max;
  do {
    r = rand();
  } while (r > m);
  r /= z;
  return (r);
}

/* --- @graph-travelling-salesman [-OPTIONS] ADJ LIST@ --- *
 *
 * Solves the Travelling Salesman Problem approximately.  Returns a list
 * containing (firstly) the cost of the computed route, and secondly the
 * route itself.  Only the nodes in LIST are considered.  The OPTIONS affect
 * the algorithm in various ways.
 *
 * -cool FACTOR		Cooling factor.  Default is 1.001.  Must be greater
 *			than 1 for the simulated annealing to work.
 *
 * -dead COUNT		Give up after COUNT cycles with no improvement.
 *			Default is 200.
 *
 * -inner COUNT		Perform COUNT loops each cooling cycle.  Default is
 *			10000.
 *
 * -init TEMP		Set the initial temperature to TEMP.  Default is not
 *			very helpful.  Initial setting should be well above
 *			the maximum cost increase from a cycle.
 *
 * -cycle / -nocycle	If -cycle is set, solve the classical problem of
 *			finding a minimal cyclic path.  If -nocycle is set,
 *			then start at the first node in LIST, and minimize a
 *			tour without caring where the end goes.  The default
 *			is -cycle.
 */

static int cmd_tsp(ClientData cd, Tcl_Interp *ti,
		   int objc, Tcl_Obj *const *objv)
{
  /* --- Initial algorithm parameters --- */

  double cool = 1.001;
  double temp = 1024;
  long inner = 10000;
  long dead = 200;
  int cycle = 1;

  /* --- Other variables --- */

  vec *v;
  unsigned long *a = 0;
  size_t n;
  int nn;
  size_t *r = 0, *r_best = 0;
  unsigned long c_best = 0, c_curr, c;
  size_t i, j, t;
  long ii, d;
  int ok;
  int rc = TCL_ERROR;
  Tcl_Obj *o, *o2, **oo;

  /* --- Parse the command line --- */

  for (i = 1; i < objc; i++) {
    int len;
    char *p = Tcl_GetStringFromObj(objv[i], &len);
    if (strcmp(p, "-cool") == 0) {
      i++; if (i >= objc) goto args;
      if (Tcl_GetDoubleFromObj(ti, objv[i], &cool) != TCL_OK)
	goto done;
      if (cool <= 1) {
	err(ti, "cooling factor must be > 1");
	goto done;
      }
    } else if (strcmp(p, "-init") == 0) {
      i++; if (i >= objc) goto args;
      if (Tcl_GetDoubleFromObj(ti, objv[i], &temp) != TCL_OK)
	goto done;
      if (temp <= 0) {
	err(ti, "initial temperature must be > 0");
	goto done;
      }
    } else if (strcmp(p, "-inner") == 0) {
      i++; if (i >= objc) goto args;
      if (Tcl_GetLongFromObj(ti, objv[i], &inner) != TCL_OK)
	goto done;
      if (inner <= 0) {
	err(ti, "inner loop count must be > 0");
	goto done;
      }
    } else if (strcmp(p, "-dead") == 0) {
      i++; if (i >= objc) goto args;
      if (Tcl_GetLongFromObj(ti, objv[i], &dead) != TCL_OK)
	goto done;
      if (dead <= 0) {
	err(ti, "dead cycles count must be > 0");
	goto done;
      }
    } else if (strcmp(p, "-cycle") == 0)
      cycle = 1;
    else if (strcmp(p, "-nocycle") == 0)
      cycle = 0;
    else if (strcmp(p, "--") == 0) {
      i++; break;
    } else if (*p != '-')
      break;
    else {
      err(ti, "bad option for graph-travelling-salesman");
      goto done;
    }
  }

  /* --- Check the rest --- */

  if (i + 2 != objc) {
    err(ti, "usage: graph-travelling-salesman [-OPTIONS] ADJ LIST");
    goto done;
  }
  if ((v = vec_find(ti, objv[i])) == 0 || import(ti, v, &a, &n) != TCL_OK)
    goto done;
  if (Tcl_ListObjGetElements(ti, objv[i + 1], &nn, &oo) != TCL_OK)
    goto done;
  if (!nn)
    goto wrap;

  r = (void *)Tcl_Alloc(nn * sizeof(*r));
  r_best = (void *)Tcl_Alloc(nn * sizeof(*r_best));
  for (i = 0; i < nn; i++) {
    long l;
    if (Tcl_GetLongFromObj(ti, oo[i], &l) != TCL_OK)
      goto done;
    if (l < 0 || l >= n) {
      err(ti, "node index out of range");
      goto done;
    }
    r[i] = l;
  }

  /* --- The one and two node problems are trivial --- *
   *
   * Avoiding these prevents us from having to mess with special cases later.
   */

  if (nn <= 2) {
    memcpy(r_best, r, nn * sizeof(*r));
    if (n == 1)
      c_best = a[r[0] * n + r[0]];
    else
      c_best = a[r[0] * n + r[1]];
    goto wrap;
  }

  /* --- Randomize the initial vector --- *
   *
   * If we're not cycling, then nail the first item in place.
   */

  for (i = cycle ? 0 : 1; i < nn; i++) {
    j = rrange(nn - i);
    t = r[i]; r[i] = r[i + j]; r[i + j] = t;
  }

  /* --- Compute the initial cost --- *
   *
   * If we're not cycling, don't close off at the end.  The easiest way to do
   * that is to start at the end.  There are at least three elements.
   */

  if (cycle) { j = 0; i = nn - 1; }
  else { j = nn - 1; i = j - 1; }
  c = 0;
  for (;;) {
    c += a[r[i] * n + r[j]];
    if (!i)
      break;
    j = i;
    i--;
  }

/*   printf("*** initial cost = %lu\n", c_best); */
  c_curr = c_best = c;
  memcpy(r_best, r, nn * sizeof(*r));

  /* --- Embark on the main loop --- */

  d = dead;
  while (d) {
    ok = 0;
    for (ii = inner; ii; ii--) {
      size_t i, j, ilo, ihi, jlo, jhi;

      /* --- Decide on a change to make --- *
       *
       * We just swap two nodes around on the path.  This is simple and seems
       * to be effective.  Don't allow the first node to be moved if we're
       * not cycling.
       */

      if (cycle) {
	i = rrange(nn);
	j = rrange(nn);
      } else {
	i = rrange(nn - 1) + 1;
	j = rrange(nn - 1) + 1;
      }

      /* --- Compute the change in cost --- *
       *
       * Since we're only swapping two nodes, we can work out the change
       * without rescanning the entire path, by just looking at the local
       * effects.
       */

      if (i == j)
	continue; /* No change */
      if (j < i) { t = i; i = j; j = t; }
      ilo = (i - 1) % nn; ihi = (i + 1) % nn;
      jlo = (j - 1) % nn; jhi = (j + 1) % nn;

      c = c_curr;
      if (j == nn - 1) {

	/* --- This is where the algorithms differ --- *
	 *
	 * If we're producing a cycle, then we need the cost function to wrap
	 * around here.  Otherwise, it hits a barrier, and the last node only
	 * has a partial effect.
	 */

	if (cycle) {
	  if (i == 0) {
	    c -= (a[r[jlo] * n + r[j]] +
		  a[r[j] * n + r[i]] +
		  a[r[i] * n + r[ihi]]);
	    c += (a[r[jlo] * n + r[i]] +
		  a[r[i] * n + r[j]] +
		  a[r[j] * n + r[ihi]]);
	  } else goto std;
	} else {
	  if (i == j - 1) {
	    c -= a[r[ilo] * n + r[i]] + a[r[i] * n + r[j]];
	    c += a[r[ilo] * n + r[j]] + a[r[j] * n + r[i]];
	  } else {
	    c -= (a[r[ilo] * n + r[i]] +
		  a[r[i] * n + r[ihi]] +
		  a[r[jlo] * n + r[j]]);
	    c += (a[r[ilo] * n + r[j]] +
		  a[r[j] * n + r[ihi]] +
		  a[r[jlo] * n + r[i]]);
	  }
	}
      } else {

	/* --- Usual case --- *
	 *
	 * This splits into two subcases, depending on whether the areas
	 * overlap.
	 */

      std:
	if (i == j - 1) {
	  c -= (a[r[ilo] * n + r[i]] +
		a[r[i] * n + r[j]] +
		a[r[j] * n + r[jhi]]);
	  c += (a[r[ilo] * n + r[j]] +
		a[r[j] * n + r[i]] +
		a[r[i] * n + r[jhi]]);
	} else {
	  c -= (a[r[ilo] * n + r[i]] +
		a[r[i] * n + r[ihi]] +
		a[r[jlo] * n + r[j]] +
		a[r[j] * n + r[jhi]]);
	  c += (a[r[ilo] * n + r[j]] +
		a[r[j] * n + r[ihi]] +
		a[r[jlo] * n + r[i]] +
		a[r[i] * n + r[jhi]]);
	}
      }

      /* --- Decide what to do --- */

      if (c > c_curr &&
	  rrange(65536) >= (size_t)(exp(((double)c_curr -
					 (double)c)/temp) * 65536))
	continue;

      /* --- Accept the change --- */

      if (c < c_curr)
	ok = 1;
      c_curr = c;
      t = r[i]; r[i] = r[j]; r[j] = t;
      if (c_curr < c_best) {
	c_best = c_curr;
/* 	printf("*** new best = %lu\n", c_best); */
	memcpy(r_best, r, nn * sizeof(*r));
      }
    }
    temp /= cool;
    if (ok)
      d = dead;
    else
      d--;
  }

  /* --- Done --- */

wrap:
  o = Tcl_NewListObj(0, 0);
  o2 = Tcl_NewListObj(0, 0);
  Tcl_ListObjAppendElement(ti, o, Tcl_NewLongObj(c_best));
  for (i = 0; i < nn; i++)
    Tcl_ListObjAppendElement(ti, o2, Tcl_NewLongObj(r_best[i]));
  Tcl_ListObjAppendElement(ti, o, o2);
  Tcl_SetObjResult(ti, o);
  rc = TCL_OK;

  /* --- Tidy up --- */

done:
  if (a) Tcl_Free((void *)a);
  if (r) Tcl_Free((void *)r);
  if (r_best) Tcl_Free((void *)r_best);
  return (rc);

args:
  err(ti, "missing argument for option");
  goto done;
}

/*----- Initialization ----------------------------------------------------*/

int Graph_SafeInit(Tcl_Interp *ti)
{
  static const struct cmd {
    /*const*/ char *name;
    Tcl_ObjCmdProc *proc;
  } cmds[] = {
    { "graph-shortest-path",	cmd_shortestpath },
    { "graph-travelling-salesman", cmd_tsp },
    { 0,			0 }
  };

  const struct cmd *c;
  if (Tcl_PkgRequire(ti, "vector", "1.0.0", 0) == 0)
    return (TCL_ERROR);
  for (c = cmds; c->name; c++)
    Tcl_CreateObjCommand(ti, c->name, c->proc, 0, 0);
  if (Tcl_PkgProvide(ti, "graph", "1.0.0"))
    return (TCL_ERROR);
  return (TCL_OK);
}

int Graph_Init(Tcl_Interp *ti)
{
  return (Graph_SafeInit(ti));
}

/*----- That's all, folks -------------------------------------------------*/