001/* Copyright (C) 2013 TU Dortmund 002 * This file is part of AutomataLib, http://www.automatalib.net/. 003 * 004 * AutomataLib is free software; you can redistribute it and/or 005 * modify it under the terms of the GNU Lesser General Public 006 * License version 3.0 as published by the Free Software Foundation. 007 * 008 * AutomataLib is distributed in the hope that it will be useful, 009 * but WITHOUT ANY WARRANTY; without even the implied warranty of 010 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 011 * Lesser General Public License for more details. 012 * 013 * You should have received a copy of the GNU Lesser General Public 014 * License along with AutomataLib; if not, see 015 * http://www.gnu.de/documents/lgpl.en.html. 016 */ 017package net.automatalib.automata.dot; 018 019import java.util.Collection; 020import java.util.Map; 021 022import net.automatalib.automata.Automaton; 023import net.automatalib.automata.graphs.TransitionEdge; 024import net.automatalib.graphs.dot.DefaultDOTHelper; 025 026 027public class DefaultDOTHelperAutomaton<S, I, T, A extends Automaton<S, I, T>> 028 extends DefaultDOTHelper<S,TransitionEdge<I,T>> { 029 030 protected final A automaton; 031 032 public DefaultDOTHelperAutomaton(A automaton) { 033 this.automaton = automaton; 034 } 035 036 037 /* 038 * (non-Javadoc) 039 * @see net.automatalib.graphs.dot.DefaultDOTHelper#initialNodes() 040 */ 041 @Override 042 protected Collection<? extends S> initialNodes() { 043 return automaton.getInitialStates(); 044 } 045 046 047 /* 048 * (non-Javadoc) 049 * @see net.automatalib.graphs.dot.DefaultDOTHelper#getEdgeProperties(java.lang.Object, java.util.Map) 050 */ 051 @Override 052 public boolean getEdgeProperties(S src, TransitionEdge<I, T> edge, S tgt, Map<String,String> properties) { 053 if(!super.getEdgeProperties(src, edge, tgt, properties)) 054 return false; 055 String label = String.valueOf(edge.getInput()); 056 properties.put("label", label); 057 return true; 058 } 059 060}