| Line | Hits | Source |
|---|---|---|
| 1 | /* | |
| 2 | * Copyright (c) 2003, the JUNG Project and the Regents of the University | |
| 3 | * of California | |
| 4 | * All rights reserved. | |
| 5 | * | |
| 6 | * This software is open-source under the BSD license; see either | |
| 7 | * "license.txt" or | |
| 8 | * http://jung.sourceforge.net/license.txt for a description. | |
| 9 | */ | |
| 10 | /* | |
| 11 | * Created on Jan 6, 2002 | |
| 12 | * | |
| 13 | */ | |
| 14 | package edu.uci.ics.jung.io; | |
| 15 | import java.io.BufferedReader; | |
| 16 | import java.io.BufferedWriter; | |
| 17 | import java.io.FileReader; | |
| 18 | import java.io.FileWriter; | |
| 19 | import java.io.IOException; | |
| 20 | import java.text.ParseException; | |
| 21 | import java.util.ArrayList; | |
| 22 | import java.util.List; | |
| 23 | import java.util.StringTokenizer; | |
| 24 | ||
| 25 | import cern.colt.list.DoubleArrayList; | |
| 26 | import cern.colt.matrix.DoubleMatrix2D; | |
| 27 | import cern.colt.matrix.impl.SparseDoubleMatrix2D; | |
| 28 | import corejava.Format; | |
| 29 | import edu.uci.ics.jung.algorithms.GraphMatrixOperations; | |
| 30 | import edu.uci.ics.jung.exceptions.FatalException; | |
| 31 | import edu.uci.ics.jung.graph.Graph; | |
| 32 | /** | |
| 33 | * Basic I/O handler for ascii matrix files. An ascii matrix is simply | |
| 34 | * a square matrix where 0 values for cell (i,j) indicates no edge exists between | |
| 35 | * vertex i and vertex j and non-zero values indicates there is an edge. If | |
| 36 | * a non-null weight key is specified then it will be used to treat the non-zero | |
| 37 | * values as a weight stored in the edges' user data keyed off the specified weight key value. | |
| 38 | * <p> | |
| 39 | * When loading a graph from a file, a symmetric graph will result in the construction of | |
| 40 | * an undirected sparse graph while a non-symmetric graph will result in the construction of | |
| 41 | * a directed sparse graph. | |
| 42 | * <p> | |
| 43 | * For example the following ascii matrix when loaded using the code:<br><code> | |
| 44 | * MatrixFile mf = new MatrixFile(null); <br> | |
| 45 | * Graph g = mf.load(filename); </code><br> | |
| 46 | * will produce an undirected sparse matrix with no weights: <br> | |
| 47 | * <pre> | |
| 48 | * 0 1 0 1 | |
| 49 | * 1 0 0 1 | |
| 50 | * 0 0 0 0 | |
| 51 | * 1 1 0 0 | |
| 52 | * </pre><p> | |
| 53 | * whereas the following ascii matrix when loaded using the code:<br><code> | |
| 54 | * MatrixFile mf = new MatrixFile("WEIGHT"); <br> | |
| 55 | * Graph g = mf.load(filename); </code> <br> | |
| 56 | * will produce a directed sparse matrix with double weight values stored in | |
| 57 | * the edges user data under the key "WEIGHT" : <br> | |
| 58 | * <pre> | |
| 59 | * 0 .5 10 0 | |
| 60 | * 0 1 0 0 | |
| 61 | * 0 0 0 -30 | |
| 62 | * 5 0 0 0 | |
| 63 | * </pre> | |
| 64 | * @author Scott | |
| 65 | * | |
| 66 | */ | |
| 67 | public class MatrixFile implements GraphFile { | |
| 68 | private String mWeightKey; | |
| 69 | /** | |
| 70 | * Constructs MatrixFile instance. If weightKey is not null then, it will | |
| 71 | * attempt to use that key to store and retreive weights from the edges' | |
| 72 | * UserData. | |
| 73 | */ | |
| 74 | 2 | public MatrixFile(String weightKey) { |
| 75 | 2 | mWeightKey = weightKey; |
| 76 | 2 | } |
| 77 | ||
| 78 | /** | |
| 79 | * Loads a graph from an input reader | |
| 80 | * @param reader the input reader | |
| 81 | * @return the graph | |
| 82 | */ | |
| 83 | public Graph load(BufferedReader reader) { | |
| 84 | 2 | Graph graph = null; |
| 85 | try { | |
| 86 | 2 | DoubleMatrix2D matrix = createMatrixFromFile(reader); |
| 87 | 2 | graph = GraphMatrixOperations.matrixToGraph(matrix,mWeightKey); |
| 88 | 0 | } catch (Exception e) { |
| 89 | 0 | throw new FatalException( |
| 90 | "Fatal exception calling MatrixFile.load(...)", | |
| 91 | e); | |
| 92 | 2 | } |
| 93 | 2 | return graph; |
| 94 | } | |
| 95 | ||
| 96 | private DoubleMatrix2D createMatrixFromFile(BufferedReader reader) | |
| 97 | throws IOException, ParseException { | |
| 98 | 2 | List rows = new ArrayList(); |
| 99 | 2 | String currentLine = null; |
| 100 | 12 | while ((currentLine = reader.readLine()) != null) { |
| 101 | 10 | StringTokenizer tokenizer = new StringTokenizer(currentLine); |
| 102 | 10 | if (tokenizer.countTokens() == 0) { |
| 103 | 0 | break; |
| 104 | } | |
| 105 | 10 | DoubleArrayList currentRow = new DoubleArrayList(); |
| 106 | 60 | while (tokenizer.hasMoreTokens()) { |
| 107 | 50 | String token = tokenizer.nextToken(); |
| 108 | 50 | currentRow.add(Double.parseDouble(token)); |
| 109 | } | |
| 110 | 10 | rows.add(currentRow); |
| 111 | } | |
| 112 | 2 | int size = rows.size(); |
| 113 | 2 | DoubleMatrix2D matrix = new SparseDoubleMatrix2D(size, size); |
| 114 | 12 | for (int i = 0; i < size; i++) { |
| 115 | 10 | DoubleArrayList currentRow = (DoubleArrayList) rows.get(i); |
| 116 | 10 | if (currentRow.size() != size) { |
| 117 | 0 | throw new ParseException( |
| 118 | "Matrix must have the same number of rows as columns", | |
| 119 | 0); | |
| 120 | } | |
| 121 | 60 | for (int j = 0; j < size; j++) { |
| 122 | 50 | double currentVal = currentRow.get(j); |
| 123 | 50 | if (currentVal != 0) { |
| 124 | 12 | matrix.setQuick(i, j, currentVal); |
| 125 | } | |
| 126 | } | |
| 127 | } | |
| 128 | 2 | return matrix; |
| 129 | } | |
| 130 | /* | |
| 131 | * Loads a graph from a file | |
| 132 | * @see edu.uci.ics.jung.io.GraphFile#load(java.lang.String) | |
| 133 | */ | |
| 134 | public Graph load(String filename) { | |
| 135 | try { | |
| 136 | 2 | BufferedReader reader = |
| 137 | new BufferedReader(new FileReader(filename)); | |
| 138 | 2 | Graph graph = load(reader); |
| 139 | 2 | reader.close(); |
| 140 | 2 | return graph; |
| 141 | 0 | } catch (IOException ioe) { |
| 142 | 0 | throw new FatalException("Error in loading file " + filename, ioe); |
| 143 | } | |
| 144 | } | |
| 145 | /* | |
| 146 | * Saves a graph to a file | |
| 147 | * @see edu.uci.ics.jung.io.GraphFile#save(edu.uci.ics.jung.graph.Graph, java.lang.String) | |
| 148 | */ | |
| 149 | public void save(Graph graph, String filename) { | |
| 150 | try { | |
| 151 | 2 | BufferedWriter writer = |
| 152 | new BufferedWriter(new FileWriter(filename)); | |
| 153 | // Vertex currentVertex = null; | |
| 154 | 2 | DoubleMatrix2D matrix = GraphMatrixOperations.graphToSparseMatrix(graph,mWeightKey); |
| 155 | 2 | Format labelFormat = new Format("%4.2f"); |
| 156 | 12 | for (int i=0;i<matrix.rows();i++) { |
| 157 | 60 | for (int j=0;j<matrix.columns();j++) { |
| 158 | 50 | writer.write(labelFormat.format(matrix.getQuick(i,j)) + " "); |
| 159 | } | |
| 160 | 10 | writer.write("\n"); |
| 161 | } | |
| 162 | 2 | writer.close(); |
| 163 | 0 | } catch (Exception e) { |
| 164 | 0 | throw new FatalException("Error saving file: " + filename, e); |
| 165 | 2 | } |
| 166 | 2 | } |
| 167 | } |
|
this report was generated by version 1.0.5 of jcoverage. |
copyright © 2003, jcoverage ltd. all rights reserved. |