/*

     This file is part of Combean.

     Combean 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 2 of the License, or

     (at your option) any later version.

     Combean 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 Combean; if not, write to the Free Software

     Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA

 */

 /*

  * Created on 19.03.2005

  *

  */

 package net.sourceforge.combean.adapters.jgraph;

 import org.jgraph.graph.AttributeMap;

 import org.jgraph.graph.DefaultEdge;

 import org.jgraph.graph.DefaultGraphCell;

 import org.jgraph.graph.DefaultPort;

 import net.sourceforge.combean.interfaces.graph.NodeIterator;

 /**

  * Utilities for working with JGraph classes

  * 

  * @author schickin

  *

  */

 public final class JGraphUtils {

     /**

      * Locate the first node in a JGraphModelAsGraph for which a

      * given attribute has a specified value

      * 

      * @param g the graph where the first matching node shall be located

      * @param attributeKey the attribute which is used for matching the node

      * @param queryValue a node matches if the given attribute has this value

      * @return the first matching node

      */

     public static JGraphCellAsNode locateFirstNodeWithAttribute(

             JGraphModelAsGraph g,

             Object attributeKey,

             Object queryValue )

     {

         JGraphCellAsNode result = null;

         NodeIterator itAllNodes = g.getAllNodesIterator();

         while (itAllNodes.hasNext()) {

             JGraphCellAsNode v = (JGraphCellAsNode) itAllNodes.next();

             AttributeMap attribs = v.getGraphCell().getAttributes();

             Object value = attribs.get(attributeKey);

             if (value != null &&

                 queryValue.toString().equals(value.toString())) {

                 result = v;

                 break;

             }

         }

         return result;

     }

     public static DefaultGraphCell[] createGraphCells(int num) {

         DefaultGraphCell[] result = new DefaultGraphCell[num];

         for (int i = 0; i < num; i++) {

             result[i] = new DefaultGraphCell();

             DefaultPort port = new DefaultPort();

             result[i].add(port);

         }

         return result;

     }

     public static DefaultEdge[] createEdges(DefaultGraphCell[] nodes, int[][] edges) {

         DefaultEdge[] result = new DefaultEdge[edges.length];

         for (int i = 0; i < edges.length; i++) {

             result[i] = createEdge(nodes, edges[i][0], edges[i][1]);

         }

         return result;

     }

     public static DefaultEdge createEdge(DefaultGraphCell[] nodes, int from, int to) {

         DefaultEdge result = new DefaultEdge();

         result.setSource(nodes[from].getChildAt(0));

         result.setTarget(nodes[to].getChildAt(0));

         return result;

     }

 }