TSPLibReader.java

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