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
  46 #define INPUT_TO_Y(i) (((double) (i)) * vertical_spacing)
  47
  48 static void exit_usage (void) /* {{{ */
  49 {
  50   printf ("Usage: sn-tex [options] [file]\n"
  51       "\n"
  52       "Valid options are:\n"
  53       "  -w <width>   Specify the width of the graph (in TikZ default units).\n"
  54       "\n");
  55   exit (EXIT_FAILURE);
  56 } /* }}} void exit_usage */
  57
  58 static int read_options (int argc, char **argv) /* {{{ */
  59 {
  60   int option;
  61
  62   while ((option = getopt (argc, argv, "w:h?")) != -1)
  63   {
  64     switch (option)
  65     {
  66       case 'w':
  67       {
  68         double width = atof (optarg);
  69         if (width <= 0.0)
  70         {
  71           fprintf (stderr, "Invalid width argument: %s\n", optarg);
  72           exit (EXIT_FAILURE);
  73         }
  74         output_width = width;
  75         break;
  76       }
  77
  78       case 'h':
  79       case '?':
  80       default:
  81         exit_usage ();
  82     }
  83   }
  84
  85   return (0);
  86 } /* }}} int read_options */
  87
  88 static double determine_stage_width (sn_stage_t *s) /* {{{ */
  89 {
  90   int lines[s->comparators_num];
  91   int right[s->comparators_num];
  92   int lines_used = 0;
  93   int i;
  94
  95   if (SN_STAGE_COMP_NUM (s) == 0)
  96     return (0.0);
  97
  98   for (i = 0; i < SN_STAGE_COMP_NUM (s); i++)
  99   {
 100     lines[i] = -1;
 101     right[i] = -1;
 102   }
 103
 104   for (i = 0; i < SN_STAGE_COMP_NUM (s); i++)
 105   {
 106     int j;
 107     sn_comparator_t *c = SN_STAGE_COMP_GET (s, i);
 108
 109     for (j = 0; j < lines_used; j++)
 110       if (SN_COMP_LEFT (c) > right[j])
 111         break;
 112
 113     lines[i] = j;
 114     right[j] = SN_COMP_RIGHT (c);
 115     if (j >= lines_used)
 116       lines_used = j + 1;
 117   }
 118   assert (lines_used >= 1);
 119
 120   return (((double) (lines_used - 1)) * inner_spacing);
 121 } /* }}} double determine_stage_width */
 122
 123 static double determine_network_width (sn_network_t *n) /* {{{ */
 124 {
 125   double width;
 126   int i;
 127
 128   /* Spacing between stages and at the beginning and end of the network */
 129   width = (SN_NETWORK_STAGE_NUM (n) + 1) * outer_spacing;
 130
 131   /* Spacing required within a stage */
 132   for (i = 0; i < SN_NETWORK_STAGE_NUM (n); i++)
 133     width += determine_stage_width (SN_NETWORK_STAGE_GET (n, i));
 134
 135   return (width);
 136 } /* }}} double determine_network_width */
 137
 138 static int tex_show_stage (sn_stage_t *s) /* {{{ */
 139 {
 140   int lines[s->comparators_num];
 141   int right[s->comparators_num];
 142   int lines_used = 0;
 143   int i;
 144
 145   for (i = 0; i < s->comparators_num; i++)
 146   {
 147     lines[i] = -1;
 148     right[i] = -1;
 149   }
 150
 151   for (i = 0; i < s->comparators_num; i++)
 152   {
 153     int j;
 154     sn_comparator_t *c = s->comparators + i;
 155
 156     int min_num;
 157     int max_num;
 158
 159     min_num = next_vertex_number;
 160     next_vertex_number++;
 161
 162     max_num = next_vertex_number;
 163     next_vertex_number++;
 164
 165     for (j = 0; j < lines_used; j++)
 166       if (SN_COMP_LEFT (c) > right[j])
 167         break;
 168
 169     lines[i] = j;
 170     right[j] = SN_COMP_RIGHT (c);
 171     if (j >= lines_used)
 172       lines_used = j + 1;
 173
 174     printf ("\\node[vertex] (v%i) at (%.2f,%.2f) {};\n"
 175         "\\node[vertex] (v%i) at (%.2f,%.2f) {};\n"
 176         "\\path[comp] (v%i) -- (v%i);\n"
 177         "\n",
 178         min_num, x_offset + (j * inner_spacing), INPUT_TO_Y (c->min),
 179         max_num, x_offset + (j * inner_spacing), INPUT_TO_Y (c->max),
 180         min_num, max_num);
 181   }
 182
 183   x_offset = x_offset + ((lines_used - 1) * inner_spacing) + outer_spacing;
 184
 185   return (0);
 186 } /* }}} int tex_show_stage */
 187
 188 int main (int argc, char **argv) /* {{{ */
 189 {
 190   sn_network_t *n;
 191   FILE *fh = NULL;
 192   double orig_width;
 193   int i;
 194
 195   read_options (argc, argv);
 196
 197   if ((argc - optind) == 0)
 198     fh = stdin;
 199   else if ((argc - optind) == 1)
 200     fh = fopen (argv[optind], "r");
 201
 202   if (fh == NULL)
 203     exit_usage ();
 204
 205   n = sn_network_read (fh);
 206   if (n == NULL)
 207   {
 208     fprintf (stderr, "Unable to read network from file handle.\n");
 209     exit (EXIT_FAILURE);
 210   }
 211
 212   orig_width = determine_network_width (n);
 213   if (orig_width <= 0.0)
 214   {
 215     fprintf (stderr, "determine_network_width returned invalid value %g.\n",
 216         orig_width);
 217     exit (EXIT_FAILURE);
 218   }
 219
 220   scale = output_width / orig_width;
 221   inner_spacing *= scale;
 222   outer_spacing *= scale;
 223   vertical_spacing *= scale;
 224
 225   x_offset = outer_spacing;
 226
 227   printf ("\\begin{tikzpicture}[auto]\n");
 228
 229   for (i = 0; i < SN_NETWORK_STAGE_NUM (n); i++)
 230      tex_show_stage (SN_NETWORK_STAGE_GET (n, i));
 231
 232   for (i = 0; i < SN_NETWORK_INPUT_NUM (n); i++)
 233     printf ("\\path[edge] (0,%.2f) -- (%.2f,%.2f);\n",
 234         INPUT_TO_Y (i), x_offset, INPUT_TO_Y (i));
 235
 236   printf ("\\end{tikzpicture}\n");
 237
 238   return (0);
 239 } /* }}} int main */
 240
 241 /* vim: set shiftwidth=2 softtabstop=2 fdm=marker : */