VisumMatrixWriter.java

Go to the documentation of this file.

/* *********************************************************************** *
 * project: org.matsim.*
 * VisumMatrixWriter.java
 *                                                                         *
 * *********************************************************************** *
 *                                                                         *
 * copyright       : (C) 2007 by the members listed in the COPYING,        *
 *                   LICENSE and WARRANTY file.                            *
 * email           : info at matsim dot org                                *
 *                                                                         *
 * *********************************************************************** *
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *   See also COPYING, LICENSE and WARRANTY file                           *
 *                                                                         *
 * *********************************************************************** */

package org.matsim.visum;

import java.io.BufferedWriter;
import java.io.FileWriter;
import java.io.IOException;
import java.util.Set;
import java.util.TreeSet;

import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import org.matsim.matrices.Entry;
import org.matsim.matrices.Matrix;

public class VisumMatrixWriter {

    /*package*/ final static Logger log = LogManager.getLogger(VisumMatrixWriter.class);

    Matrix matrix;
    Set<String> ids;

    public VisumMatrixWriter(final Matrix matrix) {
        super();
        this.matrix = matrix;
        this.ids = new TreeSet<>();
        this.ids.addAll(matrix.getFromLocations().keySet());
        this.ids.addAll(matrix.getToLocations().keySet());
    }

    public void setIds(final Set<String> ids) {
        this.ids = ids;
    }

    public void writeFile(final String filename) {

        try (BufferedWriter out = new BufferedWriter(new FileWriter(filename))) {
            out.write("$VN;Y5\n");
            out.write("*\n");
            out.write("*\tAnzahl Bezirke\n");
            out.write(this.ids.size() + "\n");

            out.write("*\tBezirksNummern\n");
            int cnt = 0;
            for (String value : this.ids) {
                cnt++;
                if (cnt > 1) {
                    out.write("\t");
                }
                out.write(value);
            }
            out.write("\n");

            for (String from : this.ids) {
                out.write("*\t" + from + "\n");
                cnt = 0;
                for (String to : this.ids) {
                    cnt++;
                    Entry e = this.matrix.getEntry(from, to);
                    if (cnt > 1) {
                        out.write("\t");
                    }
                    if (e == null) {
                        out.write("0");
                    } else {
                        out.write(Double.toString(e.getValue()));
                    }
                }
                out.write("\n");
            }

            out.write("\n");

        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}