src/sn-tex.c

   1 /**
   2  * libsortnetwork - src/sn-tex.c
   3  * Copyright (C) 2008-2010  Florian octo Forster
   4  *
   5  * This program is free software; you can redistribute it and/or modify it
   6  * under the terms of the GNU General Public License as published by the
   7  * Free Software Foundation; only version 2 of the License is applicable.
   8  *
   9  * This program is distributed in the hope that it will be useful, but
  10  * WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License along
  15  * with this program; if not, write to the Free Software Foundation, Inc.,
  16  * 51 Franklin St, Fifth Floor, Boston, MA  02110-1301 USA
  17  *
  18  * Authors:
  19  *   Florian octo Forster <ff at octo.it>
  20  **/
  21
  22 #ifndef _ISOC99_SOURCE
  23 # define _ISOC99_SOURCE
  24 #endif
  25 #ifndef _POSIX_C_SOURCE
  26 # define _POSIX_C_SOURCE 200112L
  27 #endif
  28
  29 #include <stdlib.h>
  30 #include <stdio.h>
  31 #include <unistd.h>
  32 #include <assert.h>
  33
  34 #include "sn_network.h"
  35
  36 /* 21cm (DIN-A4) - 2x3cm Rand = 15cm */
  37 static double output_width = 15.0;
  38 static double scale = 1.0;
  39 static double inner_spacing = 0.3;
  40 static double outer_spacing = 1.0;
  41 static double vertical_spacing = 0.8;
  42
  43 static double x_offset;
  44 static int next_vertex_number = 0;
  45 static int flip = 0;
  46
  47 #define INPUT_TO_Y(i) (((double) (i)) * vertical_spacing)
  48
  49 static void exit_usage (void) /* {{{ */
  50 {
  51   printf ("Usage: sn-tex [options] [file]\n"
  52       "\n"
  53       "Valid options are:\n"
  54       "  -w <width>   Specify the width of the graph (in TikZ default units).\n"
  55       "  -f           Flip the sorting network (min wire at the top, max wire at the bottom).\n"
  56       "\n");
  57   exit (EXIT_FAILURE);
  58 } /* }}} void exit_usage */
  59
  60 static int read_options (int argc, char **argv) /* {{{ */
  61 {
  62   int option;
  63
  64   while ((option = getopt (argc, argv, "w:fh?")) != -1)
  65   {
  66     switch (option)
  67     {
  68       case 'w':
  69       {
  70         double width = atof (optarg);
  71         if (width <= 0.0)
  72         {
  73           fprintf (stderr, "Invalid width argument: %s\n", optarg);
  74           exit (EXIT_FAILURE);
  75         }
  76         output_width = width;
  77         break;
  78       }
  79
  80       case 'f':
  81         flip = 1;
  82         break;
  83
  84       case 'h':
  85       case '?':
  86       default:
  87         exit_usage ();
  88     }
  89   }
  90
  91   return (0);
  92 } /* }}} int read_options */
  93
  94 static double determine_stage_width (sn_stage_t *s) /* {{{ */
  95 {
  96   int lines[s->comparators_num];
  97   int right[s->comparators_num];
  98   int lines_used = 0;
  99   int i;
 100
 101   if (SN_STAGE_COMP_NUM (s) == 0)
 102     return (0.0);
 103
 104   for (i = 0; i < SN_STAGE_COMP_NUM (s); i++)
 105   {
 106     lines[i] = -1;
 107     right[i] = -1;
 108   }
 109
 110   for (i = 0; i < SN_STAGE_COMP_NUM (s); i++)
 111   {
 112     int j;
 113     sn_comparator_t *c = SN_STAGE_COMP_GET (s, i);
 114
 115     for (j = 0; j < lines_used; j++)
 116       if (SN_COMP_LEFT (c) > right[j])
 117         break;
 118
 119     lines[i] = j;
 120     right[j] = SN_COMP_RIGHT (c);
 121     if (j >= lines_used)
 122       lines_used = j + 1;
 123   }
 124   assert (lines_used >= 1);
 125
 126   return (((double) (lines_used - 1)) * inner_spacing);
 127 } /* }}} double determine_stage_width */
 128
 129 static double determine_network_width (sn_network_t *n) /* {{{ */
 130 {
 131   double width;
 132   int i;
 133
 134   /* Spacing between stages and at the beginning and end of the network */
 135   width = (SN_NETWORK_STAGE_NUM (n) + 1) * outer_spacing;
 136
 137   /* Spacing required within a stage */
 138   for (i = 0; i < SN_NETWORK_STAGE_NUM (n); i++)
 139     width += determine_stage_width (SN_NETWORK_STAGE_GET (n, i));
 140
 141   return (width);
 142 } /* }}} double determine_network_width */
 143
 144 static int tex_show_stage (sn_stage_t *s) /* {{{ */
 145 {
 146   int lines[s->comparators_num];
 147   int right[s->comparators_num];
 148   int lines_used = 0;
 149   int i;
 150
 151   for (i = 0; i < s->comparators_num; i++)
 152   {
 153     lines[i] = -1;
 154     right[i] = -1;
 155   }
 156
 157   for (i = 0; i < s->comparators_num; i++)
 158   {
 159     int j;
 160     sn_comparator_t *c = s->comparators + i;
 161
 162     int min_num;
 163     int max_num;
 164
 165     min_num = next_vertex_number;
 166     next_vertex_number++;
 167
 168     max_num = next_vertex_number;
 169     next_vertex_number++;
 170
 171     for (j = 0; j < lines_used; j++)
 172       if (SN_COMP_LEFT (c) > right[j])
 173         break;
 174
 175     lines[i] = j;
 176     right[j] = SN_COMP_RIGHT (c);
 177     if (j >= lines_used)
 178       lines_used = j + 1;
 179
 180     printf ("\\node[vertex] (v%i) at (%.2f,%.2f) {};\n"
 181         "\\node[vertex] (v%i) at (%.2f,%.2f) {};\n"
 182         "\\path[comp] (v%i) -- (v%i);\n"
 183         "\n",
 184         min_num, x_offset + (j * inner_spacing), INPUT_TO_Y (c->min),
 185         max_num, x_offset + (j * inner_spacing), INPUT_TO_Y (c->max),
 186         min_num, max_num);
 187   }
 188
 189   x_offset = x_offset + ((lines_used - 1) * inner_spacing) + outer_spacing;
 190
 191   return (0);
 192 } /* }}} int tex_show_stage */
 193
 194 int main (int argc, char **argv) /* {{{ */
 195 {
 196   sn_network_t *n;
 197   FILE *fh = NULL;
 198   double orig_width;
 199   int i;
 200
 201   read_options (argc, argv);
 202
 203   if ((argc - optind) == 0)
 204     fh = stdin;
 205   else if ((argc - optind) == 1)
 206     fh = fopen (argv[optind], "r");
 207
 208   if (fh == NULL)
 209     exit_usage ();
 210
 211   n = sn_network_read (fh);
 212   if (n == NULL)
 213   {
 214     fprintf (stderr, "Unable to read network from file handle.\n");
 215     exit (EXIT_FAILURE);
 216   }
 217
 218   orig_width = determine_network_width (n);
 219   if (orig_width <= 0.0)
 220   {
 221     fprintf (stderr, "determine_network_width returned invalid value %g.\n",
 222         orig_width);
 223     exit (EXIT_FAILURE);
 224   }
 225
 226   scale = output_width / orig_width;
 227   inner_spacing *= scale;
 228   outer_spacing *= scale;
 229   vertical_spacing *= scale;
 230
 231   x_offset = outer_spacing;
 232
 233   if (flip)
 234     printf ("\\begin{tikzpicture}[auto,yscale=-1]\n");
 235   else
 236     printf ("\\begin{tikzpicture}[auto]\n");
 237
 238   for (i = 0; i < SN_NETWORK_STAGE_NUM (n); i++)
 239      tex_show_stage (SN_NETWORK_STAGE_GET (n, i));
 240
 241   for (i = 0; i < SN_NETWORK_INPUT_NUM (n); i++)
 242     printf ("\\path[edge] (0,%.2f) -- (%.2f,%.2f);\n",
 243         INPUT_TO_Y (i), x_offset, INPUT_TO_Y (i));
 244
 245   printf ("\\end{tikzpicture}\n");
 246
 247   return (0);
 248 } /* }}} int main */
 249
 250 /* vim: set shiftwidth=2 softtabstop=2 fdm=marker : */