GraphDrawer
/
NodeShuffler


			
				
					
						
						
							123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263
							package view;

import java.awt.BasicStroke;
import java.awt.Color;
import java.awt.Component;
import java.awt.Graphics;
import java.awt.Graphics2D;
import java.awt.event.ComponentEvent;
import java.awt.event.ComponentListener;
import java.awt.event.MouseEvent;
import java.awt.event.MouseListener;

import javax.swing.BorderFactory;
import javax.swing.JFrame;
import javax.swing.JPanel;
import javax.swing.SwingUtilities;
import javax.swing.border.Border;

import bk.LayoutType;
import graph.LayeredGraphNode;

/**
 * A drawable representation of a node.
 * @author kolja
 *
 */
public class NodeView extends JPanel implements AnnimatedView, MouseListener {
    private static final long serialVersionUID = 1L;
    private LayeredGraphNode model;
    private LayoutType layout;
    private JFrame mainView;
    private int originalWidth;
    private int originalHeight;

    public NodeView( LayeredGraphNode model, LayoutType lt, JFrame mv ) {
        mainView = mv;
        this.model = model;
        layout = lt;
        addMouseListener( this );
        originalWidth = (int)model.getWidth( lt );
        originalHeight = (int)model.getHeight( lt );
    }

    private synchronized void update()
    {
        SwingUtilities.invokeLater(new Runnable() {
            public void run() {
                for( ComponentListener l : mainView.getComponentListeners() )
                {
                    l.componentResized( new ComponentEvent(mainView, 0) );
                }
            }
        });
    }
    
    public int getScaledX( int x )
    {
        double scale = Math.min( (getWidth()-50) / (double)getVirtualWidth(), (getHeight()-50) / (double)getVirtualHeight());
        x *= scale;
        return x;
    }
    
    public int getScaledY( int y )
    {
        double scale = Math.min( (getWidth()-50) / (double)getVirtualWidth(), (getHeight()-50) / (double)getVirtualHeight());
        y *= scale;
        return y;
    }
    
    public void updateTooltipText() {
        if( layout != LayoutType.COMBINED )
        {
            setToolTipText( "<html>Name: " + model.toString() + 
                            "<br>Root: " + model.getRoot( layout ).toString() +
                            "<br>Shink: " + model.getSink( layout ).toString() + 
                            "<br>Shift: " + model.getShift( layout ) + "</html>" );
        }
        else
            setToolTipText( "Name: " + model.toString() );
        for( Component c : getComponents() )
        {
            if( !(c instanceof NodeView) )
                continue;
            ((NodeView)c).updateTooltipText();
        }
    }
    
    public int getXOffset()
    {
        int x = 0;
        double scale = Math.min( (getWidth()-50) / (double)getVirtualWidth(), (getHeight()-50) / (double)getVirtualHeight());
        x += (getWidth()-50) / 2 - (getVirtualWidth() * scale ) / 2 + 25;
        return x;
    }
    
    public int getYOffset()
    {
        int y = 0;
        double scale = Math.min( (getWidth()-50) / (double)getVirtualWidth(), (getHeight()-50) / (double)getVirtualHeight());
        y += (getHeight()-50) / 2 - (getVirtualHeight() * scale ) / 2 + 25;
        return y;
    }
    
    public int getOriginalWidth() {
        return originalWidth;
    }
    
    public int getOriginalHeight() {
        return originalHeight;
    }
    
    @Override
    public void doLayout() {
        double minX = Double.POSITIVE_INFINITY;
        for( Component c : getComponents() )
        {
            if( !(c instanceof AnnimatedView) )
                continue;
            minX = Math.min( ((AnnimatedView)c).getVirtualX(), minX);
        }
        int x = 0;
        int y = 0;
        double scale = Math.min( (getWidth()-50) / (double)getVirtualWidth(), (getHeight()-50) / (double)getVirtualHeight());
        y += (getHeight()-50) / 2 - (getVirtualHeight() * scale ) / 2 + 25;
        x += (getWidth()-50) / 2 - (getVirtualWidth() * scale ) / 2 + 25;
        for( Component c : getComponents() )
        {
            if( !(c instanceof AnnimatedView) )
                continue;
            AnnimatedView view = (AnnimatedView)c;
            c.setLocation( getScaledX( view.getVirtualX() - (int)minX ) + x, getScaledY( view.getVirtualY() ) + y);
            if( c instanceof NodeView )
                c.setSize( getScaledX( ((NodeView)c).getOriginalWidth() ), getScaledY( ((NodeView)c).getOriginalHeight() ) );
            else            
                c.setSize( getScaledX( view.getVirtualWidth() ), getScaledY( view.getVirtualHeight() ) );
            c.doLayout();
        }
    }
    
    @Override
    public void paint( Graphics g )
    {
        if( layout == LayoutType.COMBINED && model.getColor( null ) == null )
            return;
        updateTooltipText();
        paintComponent( g );
        double minX = Double.POSITIVE_INFINITY;
        for( Component c : getComponents() )
        {
            if( !(c instanceof AnnimatedView) )
                continue;
            minX = Math.min( ((AnnimatedView)c).getVirtualX(), minX);
        }
        int x = 0;
        double scale = Math.min( (getWidth()-50) / (double)getVirtualWidth(), (getHeight()-50) / (double)getVirtualHeight());
        x += (getWidth()-50) / 2 - (getVirtualWidth() * scale ) / 2 + 25;
        for( Component c : getComponents() )
        {
            if( c instanceof NodeView )
            {
                if( layout != LayoutType.COMBINED )
                {
                    NodeView v = (NodeView)c;
                    v.renderClass( g.create( getScaledX( v.getPlainVirtualX() - (int)minX - 12 ) + x, c.getY() - getScaledY(12), getScaledX( v.getPlainVirtualWidth() + 22 ), c.getHeight() + getScaledY(22) ) );
                }                
            }
            c.paint( g.create( c.getX(), c.getY(), c.getWidth(), c.getHeight() ) );
        }
    }
    
    public void renderClass( Graphics g ) {
        g.setColor( model.getRoot( layout ).getClassColor( layout ) );
        if( model.getContainedNodes().size() == 0 && model.getRoot( layout ).getClassColor( layout ) != Color.LIGHT_GRAY )
        {
            g.setColor( new Color( (g.getColor().getRed() + 500) / 3, (g.getColor().getGreen() + 500) / 3, (g.getColor().getBlue() + 500) / 3 ) );
            g.fillRect( 0, 0, g.getClipBounds().width, g.getClipBounds().height );
        }
    }

    @Override
    public void paintComponent( Graphics g )
    {
        Graphics2D g2 = (Graphics2D)g;
        g2.setColor( model.getColor( layout ) );
        g2.setStroke(new BasicStroke(5));
        if( model.getContainedNodes().size() == 0 )
        {
            if( model.getRoot( layout ) == model )
                g2.fillOval( 0, 0, getWidth(), getHeight() );
            else
                g2.fillRect( 0, 0, getWidth(), getHeight() );
        }
        boolean selected = model.isSelected( layout );
        if( selected )
        {
            g.setColor( Color.BLACK );
            if( model.getContainedNodes().size() > 0 )
                g.setColor( Color.GRAY );
            g.fillRect( 0, 0, getWidth(), getHeight() );
        }
        Border linebor = BorderFactory.createLineBorder(model.getColor( layout ), 5);
        if( model.getRoot( layout ) != model || model.getContainedNodes().size() != 0  )
            linebor.paintBorder( this, g2, 0, 0, getWidth(), getHeight() );
        if( model.isMouseOver() && model.getContainedNodes().size() == 0 )
        {
            g.setColor( Color.WHITE );
            g.fillOval( getWidth() / 4, getHeight() / 4, getWidth() / 2, getHeight() / 2 );
        }
    }

    public int getPlainVirtualX() {
        return (int)model.getX( layout );
    }

    @Override
    public int getVirtualX() {
        if( model.isDummyNode() )
            return (int)(model.getX( layout ) + getVirtualWidth() / (3/4.0) );
        return (int)model.getX( layout );
    }

    @Override
    public int getVirtualY() {
        return (int)model.getY( layout );
    }

    @Override
    public int getVirtualWidth() {
        if( model.isDummyNode() )
            return (int)(model.getWidth( layout ) / 3);
        return (int)model.getWidth( layout );
    }

    public int getPlainVirtualWidth() {
        return (int)model.getWidth( layout );
    }

    @Override
    public int getVirtualHeight() {
        return (int)model.getHeight( layout );
    }

    @Override
    public void mouseClicked(MouseEvent e) {}

    @Override
    public void mousePressed(MouseEvent e) {}

    @Override
    public void mouseReleased(MouseEvent e) {}

    @Override
    public void mouseEntered(MouseEvent e) {
        model.setMouseOver( true );
        update();
    }

    @Override
    public void mouseExited(MouseEvent e) {
        model.setMouseOver( false );
        update();
    }
}