1   /****************************************************************************
2    **
3    ** This file is part of yFiles-2.9. 
4    ** 
5    ** yWorks proprietary/confidential. Use is subject to license terms.
6    **
7    ** Redistribution of this file or of an unauthorized byte-code version
8    ** of this file is strictly forbidden.
9    **
10   ** Copyright (c) 2000-2011 by yWorks GmbH, Vor dem Kreuzberg 28, 
11   ** 72070 Tuebingen, Germany. All rights reserved.
12   **
13   ***************************************************************************/
14  package demo.io.graphml;
15  
16  
17  import y.base.DataMap;
18  import y.base.Edge;
19  import y.base.EdgeMap;
20  import y.base.Node;
21  import y.base.NodeMap;
22  import y.io.GraphMLIOHandler;
23  import y.io.graphml.KeyScope;
24  import y.io.graphml.KeyType;
25  import y.option.OptionHandler;
26  import y.view.EditMode;
27  import y.view.PopupMode;
28  import y.view.TooltipMode;
29  
30  import javax.swing.AbstractAction;
31  import javax.swing.JPopupMenu;
32  import java.awt.EventQueue;
33  import java.awt.event.ActionEvent;
34  import java.util.Locale;
35  
36  /**
37   * This demo shows how to configure GraphMLIOHandler to be able to handle
38   * extra node and edge data of simple type.
39   * Additional data for a node or an edge can be edited by right-clicking on the
40   * corresponding element.
41   * The element tool tip will show the currently set data values for each element.
42   *
43   * @see <a href="http://docs.yworks.com/yfiles/doc/developers-guide/graphml.html#graphml_extension">Section Reading and Writing Additional Data</a> in the yFiles for Java Developer's Guide
44   */
45  public class SimpleAttributesDemo extends GraphMLDemo {
46  
47    /** stores a boolean value for each node **/
48    private NodeMap node2BoolMap;
49  
50    /** stores an int value for each edge **/
51    private EdgeMap edge2IntMap;
52  
53  
54    protected void loadInitialGraph() {
55      loadGraph("resources/custom/simple-attributes.graphml");
56    }
57  
58    protected TooltipMode createTooltipMode() {
59      TooltipMode tooltipMode = new TooltipMode() {
60        /**
61         * Overwrites {@link TooltipMode#getNodeTip(y.base.Node)} to set a customized tooltip text.
62         * @param node the node for which the tooltip is set
63         * @return the tooltip string that is used by the <code>TooltipMode</code>
64         */
65        protected String getNodeTip(Node node) {
66          return "Node:BooleanValue=" + node2BoolMap.getBool(node);
67        }
68  
69        /**
70         * Overwrites {@link TooltipMode#getEdgeTip(y.base.Edge)} to set a customized tooltip text.
71         * @param edge the edge for which the tooltip is set
72         * @return the tooltip string that is used by the <code>TooltipMode</code>
73         */
74        protected String getEdgeTip(Edge edge) {
75          return "Edge:IntValue=" + edge2IntMap.getInt(edge);
76        }
77      };
78      return tooltipMode;
79    }
80  
81    /**
82     * Configures GraphMLIOHandler to read and write additional node and edge data
83     * of a simple type.
84     */
85    protected GraphMLIOHandler createGraphMLIOHandler() {
86        //Create maps that store the attributes
87      if (node2BoolMap == null) {
88        node2BoolMap = view.getGraph2D().createNodeMap();
89      }
90      if (edge2IntMap == null) {
91        edge2IntMap = view.getGraph2D().createEdgeMap();
92      }
93  
94      GraphMLIOHandler ioHandler = super.createGraphMLIOHandler();
95  
96      //  <key id="d1" for="node" attr.name="BooleanValue" attr.type="boolean"/>
97      ioHandler.getGraphMLHandler().addInputDataAcceptor("BooleanValue", node2BoolMap, KeyScope.NODE, KeyType.BOOLEAN);
98      ioHandler.getGraphMLHandler().addOutputDataProvider("BooleanValue", node2BoolMap, KeyScope.NODE, KeyType.BOOLEAN);
99  
100     //  <key id="d3" for="edge" attr.name="IntValue" attr.type="int"/>
101     ioHandler.getGraphMLHandler().addInputDataAcceptor("IntValue", edge2IntMap, KeyScope.EDGE, KeyType.INT);
102     ioHandler.getGraphMLHandler().addOutputDataProvider("IntValue", edge2IntMap, KeyScope.EDGE, KeyType.INT);
103     return ioHandler;
104   }
105 
106 
107   protected String[] getExampleResources() {
108     return null;
109   }
110 
111   /**
112    * Create an edit mode that displays a context-sensitive popup-menu when
113    * right-clicking on a node or an edge.
114    */
115   protected EditMode createEditMode() {
116     EditMode mode = super.createEditMode();
117 
118     mode.setPopupMode(new PopupMode() {
119       public JPopupMenu getNodePopup(Node v) {
120         JPopupMenu pm = new JPopupMenu();
121         pm.add(new EditAttributeAction("Edit Node Attribute...", v, node2BoolMap, KeyType.BOOLEAN));
122         return pm;
123       }
124 
125       public JPopupMenu getEdgePopup(Edge e) {
126         JPopupMenu pm = new JPopupMenu();
127         pm.add(new EditAttributeAction("Edit Edge Attribute...", e, edge2IntMap, KeyType.INT));
128         return pm;
129       }
130     });
131 
132     return mode;
133   }
134 
135 
136   /**
137    * Editor action for the additional node and edge attributes.
138    */
139   class EditAttributeAction extends AbstractAction {
140     private Object object;
141     private DataMap dataMap;
142     private KeyType dataType;
143 
144     private OptionHandler op;
145 
146     EditAttributeAction(String name, Object object, DataMap dataMap, KeyType dataType) {
147       super(name);
148       this.object = object;
149       this.dataMap = dataMap;
150       this.dataType = dataType;
151       op = new OptionHandler(name);
152       if (dataType == KeyType.BOOLEAN) {
153         op.addBool("Boolean Value", dataMap.getBool(object));
154       } else if (dataType == KeyType.INT) {
155         op.addInt("Integer Value", dataMap.getInt(object));
156       }
157     }
158 
159     public void actionPerformed(ActionEvent actionEvent) {
160       if (op.showEditor()) {
161         if (dataType == KeyType.BOOLEAN) {
162           dataMap.setBool(object, op.getBool("Boolean Value"));
163         } else if (dataType == KeyType.INT) {
164           dataMap.setInt(object, op.getInt("Integer Value"));          
165         }        
166         graphMLPane.updateGraphMLText(view.getGraph2D());
167       }
168     }
169   }
170 
171   /**
172    * Launches this demo.
173    */
174   public static void main(String[] args) {
175     EventQueue.invokeLater(new Runnable() {
176       public void run() {
177         Locale.setDefault(Locale.ENGLISH);
178         initLnF();
179         (new SimpleAttributesDemo()).start();
180       }
181     });
182   }
183 }
184