DrawnIconVertexDemo.java

/**
 * This file is part of CloudML [ http://cloudml.org ]
 *
 * Copyright (C) 2012 - SINTEF ICT
 * Contact: Franck Chauvel <franck.chauvel@sintef.no>
 *
 * Module: root
 *
 * CloudML is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as
 * published by the Free Software Foundation, either version 3 of
 * the License, or (at your option) any later version.
 *
 * CloudML 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 Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General
 * Public License along with CloudML. If not, see
 * <http://www.gnu.org/licenses/>.
 */
package org.cloudml.codecs;

import java.awt.Color;
import java.awt.Component;
import java.awt.Dimension;
import java.awt.Graphics;
import java.awt.Graphics2D;
import java.awt.Image;
import java.awt.Paint;
import java.awt.geom.Point2D;
import java.awt.image.BufferedImage;
import java.awt.image.ImageObserver;
import java.util.ArrayList;
import java.util.Collection;

import javax.imageio.ImageIO;
import javax.swing.Icon;
import javax.swing.ImageIcon;

import org.apache.commons.collections15.Transformer;
import org.cloudml.core.*;



import edu.uci.ics.jung.algorithms.layout.SpringLayout2;
import edu.uci.ics.jung.graph.DirectedSparseGraph;
import edu.uci.ics.jung.graph.Graph;
import edu.uci.ics.jung.visualization.VisualizationImageServer;
import edu.uci.ics.jung.visualization.VisualizationViewer;
import edu.uci.ics.jung.visualization.decorators.PickableVertexPaintTransformer;
import edu.uci.ics.jung.visualization.decorators.ToStringLabeller;
import edu.uci.ics.jung.visualization.renderers.DefaultEdgeLabelRenderer;
import edu.uci.ics.jung.visualization.renderers.DefaultVertexLabelRenderer;


import java.io.File;
import java.io.Serializable;

public class DrawnIconVertexDemo implements Serializable {

    /**
     * the graph
     */
    Graph<Vertex, Edge> graph;
    /**
     * the visual component and renderer for the graph
     */
    VisualizationViewer<Vertex, Edge> vv;
    private transient Deployment dmodel;
    private Transformer<Edge, Paint> edgeColor = new Transformer<Edge, Paint>() {
        public Paint transform(Edge e) {
            if (e.getType().equals("mandatory")) {
                return Color.RED;
            }
            if (vv.getPickedEdgeState().isPicked(e)) {
                return Color.lightGray;
            }
            return Color.BLACK;
        }
    };
    private Transformer<Vertex, Icon> vertexColor = new Transformer<Vertex, Icon>() {
        public Icon transform(final Vertex v) {
            return new Icon() {
                public int getIconHeight() {
                    return 40;
                }

                public int getIconWidth() {
                    return 40;
                }

                public void paintIcon(Component c, Graphics g,
                                      int x, int y) {
                    ImageIcon img;
                    if (v.getType() == "node") {
                        img = new ImageIcon(this.getClass().getResource("/server.png"));
                    }
                    else if (v.getType() == "platform") {
                        img = new ImageIcon(this.getClass().getResource("/dbms.png"));
                    }
                    else {
                        img = new ImageIcon(this.getClass().getResource("/soft.png"));
                    }
                    ImageObserver io = new ImageObserver() {
                        public boolean imageUpdate(Image img, int infoflags, int x, int y,
                                                   int width, int height) {
                            // TODO Auto-generated method stub
                            return false;
                        }
                    };
                    g.drawImage(img.getImage(), x, y, getIconHeight(), getIconWidth(), io);

                    if (!vv.getPickedVertexState().isPicked(v)) {
                        g.setColor(Color.black);
                    }
                    else {
                        g.setColor(Color.red);
                    }
                    g.drawString(v.getName(), x - 10, y + 50);
                }
            };
        }
    };

    public DrawnIconVertexDemo(final Deployment dm) {
        this.dmodel = dm;
        // create a simple graph for the demo
        graph = new DirectedSparseGraph<Vertex, Edge>();
        vv = new VisualizationViewer<Vertex, Edge>(new SpringLayout2<Vertex, Edge>(graph));

        vv.getRenderContext().setVertexLabelRenderer(new DefaultVertexLabelRenderer(Color.cyan));
        vv.getRenderContext().setEdgeLabelRenderer(new DefaultEdgeLabelRenderer(Color.cyan));

        vv.getRenderContext().setVertexIconTransformer(vertexColor);


        vv.getRenderContext().setVertexFillPaintTransformer(new PickableVertexPaintTransformer<Vertex>(vv.getPickedVertexState(), Color.white, Color.yellow));
        vv.getRenderContext().setEdgeDrawPaintTransformer(edgeColor);

        vv.setBackground(Color.white);

        // add my listener for ToolTips
        vv.setVertexToolTipTransformer(new ToStringLabeller<Vertex>());
    }

    public ArrayList<Vertex> drawFromDeploymentModel() {
        Collection<Edge> c = new ArrayList<Edge>(graph.getEdges());
        for (Edge e : c) {
            graph.removeEdge(e);
        }
        Collection<Vertex> vs = new ArrayList<Vertex>(graph.getVertices());
        for (Vertex ve : vs) {
            graph.removeVertex(ve);
        }

        ArrayList<Vertex> v = drawVerticesFromDeploymentModel(dmodel);
        drawEdgesFromDeploymentModel(dmodel, v);
        System.out.println(vv);
        return v;
    }

    public ArrayList<Vertex> drawVerticesFromDeploymentModel(Deployment dm) {
        ArrayList<Vertex> V = new ArrayList<Vertex>();
        for (ExternalComponentInstance n : dm.getComponentInstances().onlyExternals()) {
            Vertex v = new Vertex(n.getName(), "node", n);
            V.add(v);
            createVertice(v);
        }
        for (ComponentInstance x : dm.getComponentInstances()) {
            if (x instanceof InternalComponentInstance) {
                InternalComponentInstance ix = (InternalComponentInstance) x;
                if (ix.getRequiredExecutionPlatform() == null) {
                    Vertex v = new Vertex(x.getName(), "platform", ix);
                    V.add(v);
                    createVertice(v);
                }
                else {
                    Vertex v = new Vertex(x.getName(), "soft", ix);
                    V.add(v);
                    createVertice(v);
                }
            }//TODO else
        }
        return V;
    }

    public void drawEdgesFromDeploymentModel(Deployment dm, ArrayList<Vertex> v) {
        for (ComponentInstance x : dm.getComponentInstances()) {
            if (x instanceof InternalComponentInstance) {
                InternalComponentInstance ix = (InternalComponentInstance) x;
                if (ix.getRequiredExecutionPlatform() != null) {
                    Vertex v1 = findVertex(ix.getName(), v);
                    Vertex v2 = findVertex(ix.getRequiredExecutionPlatform().getName(), v);
                    Edge e = new Edge("dest" + ix.getName(), "destination");
                    createEdge(e, v1, v2);
                }
            }//TODO else
        }
        for (RelationshipInstance bi : dm.getRelationshipInstances()) {
            Vertex v1 = findVertex(bi.getRequiredEnd().getOwner().getName(), v);
            Vertex v2 = findVertex(bi.getProvidedEnd().getOwner().getName(), v);
            Edge e;
            if (!bi.getRequiredEnd().getType().isMandatory()) {
                e = new Edge(bi.getName(), "optional", bi);
            }
            else {
                e = new Edge(bi.getName(), "mandatory", bi);
            }
            createEdge(e, v1, v2);
        }
    }

    private Vertex findVertex(String name, ArrayList<Vertex> a) {
        for (Vertex v : a) {
            if (v.getName().equals(name)) {
                return v;
            }
        }
        return null;
    }

    private void createEdge(Edge e, Vertex v1, Vertex v2) {
        graph.addEdge(e, v1, v2);
    }

    private void createVertice(Vertex v) {
        graph.addVertex(v);
    }

    public void writeJPEGImage(File file) {
        int width = vv.getWidth();
        int height = vv.getHeight();

        BufferedImage bi = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
        Graphics2D graphics = bi.createGraphics();
        vv.paint(graphics);
        graphics.dispose();

        try {
            ImageIO.write(bi, "jpeg", file);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    public void writeServerJPEGImage(File file) {
        VisualizationImageServer<Vertex, Edge> vis = new VisualizationImageServer<Vertex, Edge>(vv.getGraphLayout(), vv.getGraphLayout().getSize());
        vis.setBackground(Color.WHITE);
        vis.getRenderContext().setEdgeDrawPaintTransformer(edgeColor);
        vis.getRenderContext().setVertexIconTransformer(vertexColor);

        BufferedImage image = (BufferedImage) vis.getImage(
                new Point2D.Double(vis.getWidth(), vis.getHeight()),
                new Dimension(vv.getGraphLayout().getSize()));

        try {
            ImageIO.write(image, "jpeg", file);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    public VisualizationViewer<Vertex, Edge> getVisualisationViewer() {
        return vv;
    }

    public Graph<Vertex, Edge> getGraph() {
        return graph;
    }

    public void setDeploymentModel(Deployment dm) {
        this.dmodel = dm;
    }
}