science.atlarge.graphalytics.powergraph.Utils.java Source code

Introduction

Here is the source code for science.atlarge.graphalytics.powergraph.Utils.java

Source

/*
 * Copyright 2015 Delft University of Technology
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *         http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package science.atlarge.graphalytics.powergraph;

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.File;
import java.io.FileReader;
import java.io.FileWriter;
import java.io.IOException;
import java.util.HashMap;
import java.util.Map;

import org.apache.commons.configuration.Configuration;
import org.apache.commons.configuration.ConfigurationException;
import org.apache.commons.configuration.PropertiesConfiguration;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;

import science.atlarge.graphalytics.validation.GraphStructure;

public class Utils {
    private static final Logger LOG = LogManager.getLogger();

    public static void writeVerticesToFile(GraphStructure graph, File f) throws IOException {
        BufferedWriter w = new BufferedWriter(new FileWriter(f));

        for (long v : graph.getVertices()) {
            w.write(String.format("%d\n", v));
        }

        w.flush();
        w.close();
    }

    public static void writeEdgeToFile(GraphStructure graph, boolean directed, File f) throws IOException {
        BufferedWriter w = new BufferedWriter(new FileWriter(f));

        for (long v : graph.getVertices()) {
            for (long u : graph.getEdgesForVertex(v)) {
                if (directed || v < u) {
                    w.write(String.format("%d %d\n", v, u));
                }
            }
        }

        w.flush();
        w.close();
    }

    public static <T> Map<Long, T> readResults(File f, Class<T> clazz) throws Exception {
        Map<Long, T> results = new HashMap<Long, T>();

        BufferedReader r = new BufferedReader(new FileReader(f));
        String line;

        while ((line = r.readLine()) != null) {
            String tokens[] = line.split(" ", 2);

            Long vertex = Long.valueOf(tokens[0]);
            T value = clazz.getDeclaredConstructor(String.class).newInstance(tokens[1]);
            results.put(vertex, value);
        }

        return results;
    }

    public static Configuration loadConfiguration() {
        try {
            return new PropertiesConfiguration(PowergraphPlatform.BENCHMARK_PROPERTIES_FILE);
        } catch (ConfigurationException e) {
            LOG.warn("failed to load " + PowergraphPlatform.BENCHMARK_PROPERTIES_FILE, e);
            return new PropertiesConfiguration();
        }
    }
}