playground.johannes.studies.coopsim.SweepMerge2D.java Source code

Introduction

Here is the source code for playground.johannes.studies.coopsim.SweepMerge2D.java

Source

/* *********************************************************************** *
 * project: org.matsim.*
 * SweepMerge.java
 *                                                                         *
 * *********************************************************************** *
 *                                                                         *
 * copyright       : (C) 2011 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 playground.johannes.studies.coopsim;

import java.io.BufferedReader;
import java.io.File;
import java.io.FileReader;
import java.io.IOException;
import java.util.Arrays;
import java.util.Comparator;

import org.apache.commons.math.stat.descriptive.DescriptiveStatistics;
import org.apache.log4j.Logger;
import org.matsim.core.config.Config;
import org.matsim.core.config.ConfigReader;

import playground.johannes.studies.mcmc.KeyMatrix;

/**
 * @author illenberger
 * 
 */
public class SweepMerge2D {

    private static final Logger logger = Logger.getLogger(SweepMerge2D.class);

    /**
     * @param args
     * @throws IOException
     */
    public static void main(String[] args) throws IOException {
        File root = new File("/Volumes/cluster.math.tu-berlin.de/net/ils2/jillenberger/leisure/runs/run205/");
        String property = "d_trip_home";
        int valIdx = 1;
        String parameterKey1 = "traveling";
        String parameterKey2 = "performing";

        File analysis = new File(String.format("%1$s/analysis/", root.getAbsolutePath()));
        analysis.mkdirs();

        KeyMatrix<Double> matrix = new KeyMatrix<Double>();

        File tasks = new File(String.format("%1$s/tasks/", root.getAbsolutePath()));
        for (File file : tasks.listFiles()) {
            if (file.isDirectory()) {
                if (!file.getName().equals("analysis")) {
                    File output = new File(String.format("%1$s/output/", file.getAbsolutePath()));
                    String[] dirs = output.list();
                    if (dirs.length > 0) {

                        Arrays.sort(dirs, new Comparator<String>() {
                            @Override
                            public int compare(String o1, String o2) {
                                return Double.compare(Double.parseDouble(o1), Double.parseDouble(o2));
                            }
                        });
                        /*
                         * get parameter value
                         */
                        Config config = new Config();
                        ConfigReader creader = new ConfigReader(config);
                        creader.readFile(String.format("%1$s/config.xml", file.getAbsolutePath()));
                        double paramValue1 = Double.parseDouble(config.findParam("planCalcScore", parameterKey1));
                        double paramValue2 = Double.parseDouble(config.findParam("planCalcScore", parameterKey2));

                        int start = dirs.length - 10;
                        start = Math.max(0, start);
                        if (dirs.length < 10) {
                            logger.warn("Less than 10 samples.");
                        }
                        DescriptiveStatistics stat = new DescriptiveStatistics();
                        for (int i = start; i < dirs.length; i++) {
                            File statsFile = new File(
                                    String.format("%1$s/%2$s/statistics.txt", output.getAbsolutePath(), dirs[i]));
                            if (statsFile.exists()) {
                                /*
                                 * get property value
                                 */
                                BufferedReader reader = new BufferedReader(new FileReader(statsFile));
                                String line = reader.readLine();

                                while ((line = reader.readLine()) != null) {
                                    String[] tokens = line.split("\t");
                                    String key = tokens[0];
                                    double val = Double.parseDouble(tokens[valIdx]);

                                    if (key.equals(property)) {
                                        stat.addValue(val);
                                        break;
                                    }
                                }
                            }

                        }
                        matrix.putValue(stat.getMean(), Math.abs(paramValue1), Math.abs(paramValue2));
                    } else {
                        logger.warn("No samples.");
                    }
                }
            }
        }

        matrix.write(String.format("%1$s/%2$s.matrix.txt", analysis.getAbsolutePath(), property));
    }

}