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}