cz.cuni.mff.ksi.jinfer.autoeditor.automatonvisualizer.layouts.LayoutHelperFactory.java Source code

Introduction

Here is the source code for cz.cuni.mff.ksi.jinfer.autoeditor.automatonvisualizer.layouts.LayoutHelperFactory.java

Source

/*
 *  Copyright (C) 2010 rio
 *
 *  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 3 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
package cz.cuni.mff.ksi.jinfer.autoeditor.automatonvisualizer.layouts;

import cz.cuni.mff.ksi.jinfer.autoeditor.automatonvisualizer.layouts.properties.LayoutPropertiesPanel;
import cz.cuni.mff.ksi.jinfer.base.automaton.Automaton;
import cz.cuni.mff.ksi.jinfer.base.automaton.State;
import cz.cuni.mff.ksi.jinfer.base.automaton.Step;
import cz.cuni.mff.ksi.jinfer.base.utils.ModuleSelectionHelper;
import cz.cuni.mff.ksi.jinfer.base.utils.RunningProject;
import edu.uci.ics.jung.algorithms.layout.Layout;
import edu.uci.ics.jung.graph.DirectedSparseMultigraph;
import edu.uci.ics.jung.graph.Graph;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Properties;
import java.util.Set;
import org.apache.commons.collections15.Transformer;

/**
 * Factory of custom JUNG {@link Layout}s.
 *
 * @author rio
 */
public final class LayoutHelperFactory {

    private LayoutHelperFactory() {
    }

    /**
     * Creates {@link Layout} which user selected in preferences.
     *
     * @param <T> Type parameter of specified automaton.
     * @param automaton Automaton to create layout from.
     * @param edgeLabelTransformer Transformer that can transform edge to string, which is displayed as its label.
     * @return Layout by user selection.
     */
    public static <T> Layout<State<T>, Step<T>> createUserLayout(final Automaton<T> automaton,
            final Transformer<Step<T>, String> edgeLabelTransformer) throws InterruptedException {
        final Properties p = RunningProject.getActiveProjectProps(LayoutPropertiesPanel.NAME);

        final LayoutFactory f = ModuleSelectionHelper.lookupImpl(LayoutFactory.class,
                p.getProperty(LayoutPropertiesPanel.LAYOUT_PROP, LayoutPropertiesPanel.LAYOUT_DEFAULT));
        return f.createLayout(automaton, createGraph(automaton), edgeLabelTransformer);
    }

    /**
     * Creates JUNG representation of {@link Automaton}.
     *
     * @param <T> Type parameter of specified automaton.
     * @param automaton Automaton instance.
     * @return Automaton as JUNG graph.
     */
    private static <T> Graph<State<T>, Step<T>> createGraph(final Automaton<T> automaton) {
        final DirectedSparseMultigraph<State<T>, Step<T>> graph = new DirectedSparseMultigraph<State<T>, Step<T>>();
        final Map<State<T>, Set<Step<T>>> automatonDelta = automaton.getDelta();

        // Get vertices = states of automaton
        for (final Entry<State<T>, Set<Step<T>>> entry : automatonDelta.entrySet()) {
            graph.addVertex(entry.getKey());
        }

        // Get edges of automaton
        for (Entry<State<T>, Set<Step<T>>> entry : automatonDelta.entrySet()) {
            for (Step<T> step : entry.getValue()) {
                graph.addEdge(step, step.getSource(), step.getDestination());
            }
        }

        return graph;
    }
}