protoas/TSPLibReader.java

   1 package protoas;
   2
   3 import java.util.HashMap;
   4 import java.io.File;
   5 import java.io.FileReader;
   6 import java.io.IOException;
   7 import java.io.InputStream;
   8 import java.io.BufferedReader;
   9 import java.io.FileInputStream;
  10 import java.io.StreamTokenizer;
  11 import java.io.BufferedInputStream;
  12
  13 class TSPLibReader {
  14
  15   final static long serialVersionUID = (long)1.0;
  16
  17   protected StreamTokenizer streamtokenizer;
  18   protected HashMap<Integer, CoordinatePair<Integer, Integer>> coordinates;
  19
  20   TSPLibReader(String filename) throws IOException {
  21     this(new File(filename));
  22   }
  23
  24   TSPLibReader(File file) throws IOException {
  25     coordinates =
  26       new HashMap<Integer, CoordinatePair<Integer, Integer>>(countLines(new FileInputStream(file)));
  27
  28     streamtokenizer = new StreamTokenizer(new BufferedReader(new FileReader(file)));
  29     streamtokenizer.eolIsSignificant(true);
  30     streamtokenizer.wordChars('_','_');
  31   }
  32
  33   public HashMap<Integer, CoordinatePair<Integer, Integer>> getCoordinates() {
  34     try {
  35       this.parseLines();
  36     } catch (IOException e) {
  37       System.out.println(e.toString() + "(" + e.getClass() + "): " + e);
  38       return null;
  39     }
  40     return coordinates;
  41   }
  42
  43   protected void parseLines() throws IOException {
  44
  45     boolean SeenDisplayDataSection = false;
  46     int i = 0, x = 0, y = 0, City = 0;
  47
  48     do {
  49
  50       if (!SeenDisplayDataSection) {
  51         streamtokenizer.nextToken();
  52         //System.out.println(streamtokenizer);
  53         if (streamtokenizer.ttype == StreamTokenizer.TT_WORD)
  54           if (streamtokenizer.sval.equals("DISPLAY_DATA_SECTION") ||
  55               streamtokenizer.sval.equals("NODE_COORD_SECTION"))
  56             SeenDisplayDataSection = true;
  57
  58         continue;
  59       }
  60
  61       streamtokenizer.nextToken();
  62       if (streamtokenizer.ttype == StreamTokenizer.TT_NUMBER) {
  63         i++;
  64       } else {
  65         i = 0;
  66         continue;
  67       }
  68
  69       streamtokenizer.nextToken();
  70       if (streamtokenizer.ttype == StreamTokenizer.TT_NUMBER) {
  71         x = (int)streamtokenizer.nval;
  72         i++;
  73       } else {
  74         i = 0;
  75         continue;
  76       }
  77
  78       streamtokenizer.nextToken();
  79       if (streamtokenizer.ttype == StreamTokenizer.TT_NUMBER) {
  80         y = (int)streamtokenizer.nval;
  81         i++;
  82       } else {
  83         i = 0;
  84         continue;
  85       }
  86
  87       streamtokenizer.nextToken();
  88       if (streamtokenizer.ttype == StreamTokenizer.TT_EOL && i == 3)
  89         coordinates.put(City++, new CoordinatePair<Integer, Integer>(x, y));
  90
  91       i = 0;
  92
  93     } while (streamtokenizer.ttype != StreamTokenizer.TT_EOF);
  94
  95   }
  96
  97   protected int countLines(FileInputStream fis) throws IOException {
  98     InputStream is = new BufferedInputStream(fis);
  99     byte[] c = new byte[1024];
 100     int count = 0;
 101     int readChars = 0;
 102     while ((readChars = is.read(c)) != -1) {
 103       for (int i = 0; i < readChars; ++i) {
 104         if (c[i] == '\n')
 105           ++count;
 106       }
 107     }
 108     return count;
 109   }
 110
 111   public static void main(String[] args) {
 112     try {
 113       TSPLibReader tlr = new TSPLibReader(args[0]);
 114
 115       HashMap<Integer, CoordinatePair<Integer, Integer>> coordinates =
 116         tlr.getCoordinates();
 117
 118       System.out.println("Entries in HashMap: " + coordinates.size());
 119
 120       for (int k : coordinates.keySet())
 121         System.out.println("City " + k + " with Coordinates " +
 122             coordinates.get(k).getFirst() +
 123             "/" +
 124             coordinates.get(k).getSecond());
 125
 126     } catch (IOException e) {
 127       System.out.println(e.toString() + "(" + e.getClass() + "): " + e);
 128       System.exit(1);
 129     }
 130   }
 131
 132 }