org.jgrapht.nio.dimacs.DIMACSImporter Maven / Gradle / Ivy
/*
* (C) Copyright 2010-2020, by Michael Behrisch and Contributors.
*
* JGraphT : a free Java graph-theory library
*
* See the CONTRIBUTORS.md file distributed with this work for additional
* information regarding copyright ownership.
*
* This program and the accompanying materials are made available under the
* terms of the Eclipse Public License 2.0 which is available at
* http://www.eclipse.org/legal/epl-2.0, or the
* GNU Lesser General Public License v2.1 or later
* which is available at
* http://www.gnu.org/licenses/old-licenses/lgpl-2.1-standalone.html.
*
* SPDX-License-Identifier: EPL-2.0 OR LGPL-2.1-or-later
*/
package org.jgrapht.nio.dimacs;
import org.jgrapht.*;
import org.jgrapht.alg.util.*;
import org.jgrapht.nio.*;
import java.io.*;
import java.util.*;
import java.util.function.*;
/**
* Imports a graph specified in DIMACS format.
*
*
* See {@link DIMACSFormat} for a description of all the supported DIMACS formats.
*
*
* In summary, one of the most common DIMACS formats was used in the
* 2nd DIMACS challenge and follows
* the following structure:
*
*
* {@code
* DIMACS G {
* c ignored during parsing of the graph
* p edge
* e
* e
* e
* e
* ...
* }
* }
*
*
* Although not specified directly in the DIMACS format documentation, this implementation also
* allows for the a weighted variant:
*
*
* {@code
* e
* }
*
*
* Note: the current implementation does not fully implement the DIMACS specifications! Special
* (rarely used) fields specified as 'Optional Descriptors' are currently not supported (ignored).
*
* @author Michael Behrisch (adaptation of GraphReader class)
* @author Joris Kinable
* @author Dimitrios Michail
*
* @param the graph vertex type
* @param the graph edge type
*/
public class DIMACSImporter
extends
BaseEventDrivenImporter
implements
GraphImporter
{
private final double defaultWeight;
/**
* Construct a new DIMACSImporter
*
* @param defaultWeight default edge weight
*/
public DIMACSImporter(double defaultWeight)
{
super();
this.defaultWeight = defaultWeight;
}
/**
* Construct a new DIMACSImporter
*/
public DIMACSImporter()
{
this(Graph.DEFAULT_EDGE_WEIGHT);
}
/**
* Import a graph.
*
*
* The provided graph must be able to support the features of the graph that is read. For
* example if the file contains self-loops then the graph provided must also support self-loops.
* The same for multiple edges.
*
*
* If the provided graph is a weighted graph, the importer also reads edge weights. Otherwise
* edge weights are ignored.
*
* @param graph the output graph
* @param input the input reader
* @throws ImportException in case an error occurs, such as I/O or parse error
*/
@Override
public void importGraph(Graph graph, Reader input)
throws ImportException
{
DIMACSEventDrivenImporter genericImporter =
new DIMACSEventDrivenImporter().renumberVertices(false);
Consumers consumers = new Consumers(graph);
genericImporter.addVertexCountConsumer(consumers.nodeCountConsumer);
genericImporter.addEdgeConsumer(consumers.edgeConsumer);
genericImporter.importInput(input);
}
private class Consumers
{
private Graph graph;
private Integer nodeCount;
private Map map;
public Consumers(Graph graph)
{
this.graph = graph;
this.nodeCount = null;
this.map = new HashMap();
}
public final Consumer nodeCountConsumer = (n) -> {
this.nodeCount = n;
for (int i = 0; i < nodeCount; i++) {
map.put(Integer.valueOf(i), graph.addVertex());
}
};
public final Consumer> edgeConsumer = (t) -> {
int source = t.getFirst();
V from = map.get(t.getFirst());
if (from == null) {
throw new ImportException("Node " + source + " does not exist");
}
int target = t.getSecond();
V to = map.get(target);
if (to == null) {
throw new ImportException("Node " + target + " does not exist");
}
E e = graph.addEdge(from, to);
if (graph.getType().isWeighted()) {
double weight = t.getThird() == null ? defaultWeight : t.getThird();
graph.setEdgeWeight(e, weight);
}
};
}
}