org.codehaus.plexus.graph.visualize.TouchGraph Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of plexus-graph Show documentation
The newest version!
package org.codehaus.plexus.graph.visualize;

/*
 * Licensed to the Codehaus Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *  http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */

import org.codehaus.plexus.graph.DirectedGraph;
import org.codehaus.plexus.graph.Edge;
import org.codehaus.plexus.graph.Named;
import org.codehaus.plexus.graph.Vertex;
import org.codehaus.plexus.graph.WeightedGraph;

import java.awt.*;
import java.io.OutputStream;
import java.io.PrintWriter;
import java.util.Iterator;
import java.util.Random;

public class TouchGraph
{
    private Color vertexColor = Color.yellow;
    private Color textColor = Color.black;
    private Color edgeColor = Color.red;

    private double lengthFactor = 500.0;
    private double defaultLength = 1.0;
    private int fontSize = 18;

    private Random random = new Random();

    /** Constructor for the TouchGraph object */
    public TouchGraph()
    {
    }

    /** Gets the colorText attribute of the TouchGraph object */
    private String getColorText( Color color )
    {
        String RC = Integer.toHexString( color.getRGB() ).toUpperCase();
        return RC.substring( 2, 8 );
    }

    /** Sets the vertexColor attribute of the TouchGraph object */
    public void setVertexColor( Color vertexColor )
    {
        this.vertexColor = vertexColor;
    }

    /** Gets the vertexColor attribute of the TouchGraph object */
    public Color getVertexColor()
    {
        return this.vertexColor;
    }

    /** Gets the vertexColorText attribute of the TouchGraph object */
    private String getVertexColorText()
    {
        return getColorText( getVertexColor() );
    }

    /** Sets the textColor attribute of the TouchGraph object */
    public void setTextColor( Color vertexColor )
    {
        this.vertexColor = vertexColor;
    }

    /** Gets the textColor attribute of the TouchGraph object */
    public Color getTextColor()
    {
        return this.vertexColor;
    }

    /** Gets the textColorText attribute of the TouchGraph object */
    private String getTextColorText()
    {
        return getColorText( getTextColor() );
    }

    /** Sets the edgeColor attribute of the TouchGraph object */
    public void setEdgeColor( Color edgeColor )
    {
        this.edgeColor = edgeColor;
    }

    /** Gets the edgeColor attribute of the TouchGraph object */
    public Color getEdgeColor()
    {
        return this.edgeColor;
    }

    /** Gets the edgeColorText attribute of the TouchGraph object */
    private String getEdgeColorText()
    {
        return getColorText( getEdgeColor() );
    }

    /** Sets the fontSize attribute of the TouchGraph object */
    public void setFontSize( int size )
    {
        this.fontSize = size;
    }

    /** Gets the fontSize attribute of the TouchGraph object */
    public int getFontSize()
    {
        return fontSize;
    }

    /** Sets the defaultEdgeLength attribute of the TouchGraph object */
    public void setDefaultEdgeLength( int length )
    {
        this.defaultLength = length;
    }

    /** Gets the defaultEdgeLength attribute of the TouchGraph object */
    public double getDefaultEdgeLength()
    {
        return defaultLength;
    }

    /** Description of the Method */
    protected void writeNodeset( PrintWriter pw,
                                 DirectedGraph graph )
    {
        pw.println( "" );
        Iterator vertices = graph.getVertices().iterator();
        while ( vertices.hasNext() )
        {
            Vertex v = (Vertex) vertices.next();

            pw.println( "" );
            pw.println( "" );

            String label;
            if ( v instanceof Named )
            {
                label = ( (Named) v ).getName();
            }
            else
            {
                label = v.toString();
            }

            String backColor = null;
            String textColor = null;

            if ( v instanceof Colored )
            {
                backColor = getColorText( ( (Colored) v ).getBackgroundColor() );
                textColor = getColorText( ( (Colored) v ).getTextColor() );
            }
            else
            {
                backColor = getVertexColorText();
                textColor = getTextColorText();
            }

            pw.println( "" );

            pw.println( "" );
        }

        pw.println( "" );
    }

    /** Description of the Method */
    protected void writeEdgeset( PrintWriter pw,
                                 DirectedGraph graph )
    {
        pw.println( "" );

        Iterator edges = graph.getEdges().iterator();
        while ( edges.hasNext() )
        {
            Edge next = (Edge) edges.next();

            int length = new Double( lengthFactor * defaultLength ).intValue();

            if ( graph instanceof WeightedGraph )
            {
                length = new Double( lengthFactor * ( (WeightedGraph) graph )
                    .getWeight( next ) ).intValue();
            }

            pw.println( "" );

        }
        pw.println( "" );
    }

    /** Description of the Method */
    public void toXML( DirectedGraph graph,
                       OutputStream os )
    {
        PrintWriter pw = new PrintWriter( os );
        pw.println( "" );
        pw.println( "" );
        writeNodeset( pw, graph );
        writeEdgeset( pw, graph );
        pw.println( "" );
        pw.flush();

        return;
    }
}