001/*
002 * Copyright 2006 - 2013
003 *      Stefan Balev    <stefan.balev@graphstream-project.org>
004 *      Julien Baudry   <julien.baudry@graphstream-project.org>
005 *      Antoine Dutot   <antoine.dutot@graphstream-project.org>
006 *      Yoann Pigné         <yoann.pigne@graphstream-project.org>
007 *      Guilhelm Savin  <guilhelm.savin@graphstream-project.org>
008 *  
009 * GraphStream is a library whose purpose is to handle static or dynamic
010 * graph, create them from scratch, file or any source and display them.
011 * 
012 * This program is free software distributed under the terms of two licenses, the
013 * CeCILL-C license that fits European law, and the GNU Lesser General Public
014 * License. You can  use, modify and/ or redistribute the software under the terms
015 * of the CeCILL-C license as circulated by CEA, CNRS and INRIA at the following
016 * URL <http://www.cecill.info> or under the terms of the GNU LGPL as published by
017 * the Free Software Foundation, either version 3 of the License, or (at your
018 * option) any later version.
019 * 
020 * This program is distributed in the hope that it will be useful, but WITHOUT ANY
021 * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A
022 * PARTICULAR PURPOSE.  See the GNU Lesser General Public License for more details.
023 * 
024 * You should have received a copy of the GNU Lesser General Public License
025 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
026 * 
027 * The fact that you are presently reading this means that you have had
028 * knowledge of the CeCILL-C and LGPL licenses and that you accept their terms.
029 */
030package org.graphstream.stream.file.gexf;
031
032import javax.xml.stream.XMLStreamException;
033
034public class GEXFGraph implements GEXFElement {
035        GEXF root;
036
037        DefaultEdgeType defaultEdgeType;
038        IDType idType;
039        Mode mode;
040
041        GEXFAttributes nodesAttributes;
042        GEXFAttributes edgesAttributes;
043        GEXFElement nodes;
044        GEXFElement edges;
045
046        public GEXFGraph(GEXF root) {
047                this(root, Mode.DYNAMIC);
048        }
049
050        public GEXFGraph(GEXF root, Mode mode) {
051                this.root = root;
052
053                this.defaultEdgeType = DefaultEdgeType.UNDIRECTED;
054                this.idType = IDType.STRING;
055                this.mode = mode;
056
057                nodesAttributes = new GEXFAttributes(root, ClassType.NODE);
058                edgesAttributes = new GEXFAttributes(root, ClassType.EDGE);
059                nodes = new GEXFNodes(root);
060                edges = new GEXFEdges(root);
061        }
062
063        public void export(SmartXMLWriter stream) throws XMLStreamException {
064                Mode realMode = mode;
065
066                if (!root.isExtensionEnable(Extension.DYNAMICS))
067                        realMode = Mode.STATIC;
068
069                stream.startElement("graph");
070
071                stream.stream.writeAttribute("idtype", idType.qname);
072                stream.stream.writeAttribute("mode", realMode.qname);
073                stream.stream.writeAttribute("defaultedgetype", defaultEdgeType.qname);
074
075                if (root.isExtensionEnable(Extension.DYNAMICS))
076                        stream.stream.writeAttribute("timeformat",
077                                        root.getTimeFormat().qname);
078
079                nodesAttributes.export(stream);
080                edgesAttributes.export(stream);
081                nodes.export(stream);
082                edges.export(stream);
083
084                stream.endElement(); // GRAPH
085        }
086
087}