classnode.java

UML设计测试工具

/*
 * USE - UML based specification environment
 * Copyright (C) 1999-2004 Mark Richters, University of Bremen
 *
 * This program 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.
 *
 * This program 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 this program; if not, write to the Free Software
 * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 */

/* $ProjectHeader: use 2-3-0-release.1 Mon, 12 Sep 2005 20:18:33 +0200 green $ */

package org.tzi.use.gui.views.diagrams.classdiagram;

import java.awt.Font;
import java.awt.FontMetrics;
import java.awt.Graphics;
import java.util.ArrayList;
import java.util.List;

import org.tzi.use.gui.main.ModelBrowserSorting;
import org.tzi.use.gui.main.ModelBrowserSorting.SortChangeEvent;
import org.tzi.use.gui.main.ModelBrowserSorting.SortChangeListener;
import org.tzi.use.gui.views.diagrams.DiagramOptions;
import org.tzi.use.gui.views.diagrams.NodeBase;
import org.tzi.use.uml.mm.MAttribute;
import org.tzi.use.uml.mm.MClass;
import org.tzi.use.uml.mm.MOperation;

/**
 * A node representing a class.
 * 
 * @version $ProjectVersion: 2-3-0-release.1 $
 * @author Fabian Gutsche
 */
public class ClassNode extends NodeBase implements SortChangeListener {
    
    private DiagramOptions fOpt;
    private MClass fClass;
    private String fLabel;
    private List fAttributes;
    private List fOperations;
    private String[] fAttrValues;
    private String[] fOprSignatures;
    
    ClassNode( MClass cls, DiagramOptions opt ) {
        fClass = cls;
        fOpt = opt;
        fLabel = cls.name();
        
        List allAttributes = cls.attributes(); //cls.allAttributes();
        final int N = allAttributes.size();
        fAttrValues = new String[N];
        fAttributes = (ArrayList) ModelBrowserSorting
        .getInstance()
        .sortAttributes(
                        allAttributes );
        List allOperations = cls.allOperations();
        final int M = allOperations.size();
        fOprSignatures = new String[M];
        fOperations = (ArrayList) ModelBrowserSorting.getInstance()
        .sortOperations( allOperations );
        ModelBrowserSorting.getInstance().addSortChangeListener( this );
    }
    
    public MClass cls() {
        return fClass;
    }
    
    public String name() {
        return fClass.name();
    }
    
    /**
     * After the occurence of an event the attribute list is updated.
     */
    public void stateChanged( SortChangeEvent e ) {
        fAttributes = (ArrayList) ModelBrowserSorting
        .getInstance()
        .sortAttributes(
                        fAttributes );
        fOperations = (ArrayList) ModelBrowserSorting
        .getInstance()
        .sortOperations(
                        fOperations );
    }
    
    /**
     * Sets the correct size of the width and height of this class node.
     * This method needs to be called before actually drawing the node.
     * (Width and height are needed from other methods before the nodes are
     * drawn.)
     */
    public void setRectangleSize( Graphics g ) { //FontMetrics fm ) {
        FontMetrics fm = g.getFontMetrics();
        Font normalFont = fm.getFont();
        if ( fClass.isAbstract() ) {
            Font italicFont = normalFont.deriveFont( Font.ITALIC );
            FontMetrics italicFm = g.getFontMetrics( italicFont );
            setWidth( italicFm.stringWidth( fLabel ) );
            setHeight( italicFm.getHeight() );
        } else {
            setWidth( fm.stringWidth( fLabel ) );
            setHeight( fm.getHeight() );
        }
        
        int attrHeight = 0;
        int oprHeight = 0;
        
        if ( fOpt.isShowAttributes() ) {
            for ( int i = 0; i < fAttributes.size(); i++ ) {
                MAttribute attr = (MAttribute) fAttributes.get( i );
                fAttrValues[i] = attr.toString();
                setWidth( Math.max( getWidth(), fm.stringWidth( fAttrValues[i] ) ) );
            }
            attrHeight = fm.getHeight() * fAttributes.size() + 3;
        }
        
        if ( fOpt.isShowOperations() ) {
            for ( int i = 0; i < fOperations.size(); i++ ) {
                MOperation opr = (MOperation) fOperations.get( i );
                fOprSignatures[i] = opr.signature();
                setWidth( Math.max( getWidth(), fm.stringWidth( fOprSignatures[i] ) ) );
            }
            oprHeight = fm.getHeight() * fOperations.size() + 3;
        }
        
        setWidth( getWidth() + 10 );
        setHeight( attrHeight + oprHeight + fm.getHeight() + 4 );
    }
    
    /**
     * Returns if this class node may be deleted.
     */
    public boolean isDeletable() {
        return true;
    }
    
    public String ident() {
        return "Class." + fClass.name();
    }
    public String identNodeEdge() {
        return "AssociationClass." + fClass.name();
    }
    
    /**
     * Draws a box with a label.
     */
    public void draw( Graphics g, FontMetrics fm ) {
        int x = (int) x();
        int y = (int) y();
        
        Font oldFont = g.getFont();
        if ( fClass.isAbstract() ) {
            g.setFont( oldFont.deriveFont( Font.ITALIC ) );
            fm = g.getFontMetrics();
        }
        
        int labelWidth = fm.stringWidth( fLabel );
        
        if ( isSelected() ) {
            g.setColor( fOpt.getNODE_SELECTED_COLOR() );
        } else {
            g.setColor( fOpt.getNODE_COLOR() );
        }
        g.fillRect( x - getWidth() / 2, y - getHeight() / 2, getWidth(), getHeight() );
        g.setColor( fOpt.getNODE_FRAME_COLOR() );
        g.drawRect( x - getWidth() / 2, y - getHeight() / 2, getWidth() - 1, getHeight() - 1 );
        
        x -= labelWidth / 2;
        y = y - ( getHeight() / 2 ) + fm.getAscent() + 2;
        g.setColor( fOpt.getNODE_LABEL_COLOR() );
        
        g.drawString( fLabel, x, y );
        // resetting font and fontMetrics if the class was abstract
        g.setFont( oldFont );
        fm = g.getFontMetrics();
        
        if ( fOpt.isShowAttributes() ) {
            // compartment divider
            x = (int) x();
            g.drawLine( x - getWidth() / 2, y + 3, x + getWidth() / 2 - 1, y + 3 );
            x -= ( getWidth() - 10 ) / 2;
            y += 3;
            for ( int i = 0; i < fAttributes.size(); i++ ) {
                y += fm.getHeight();
                g.drawString( fAttrValues[i], x, y );
            }
        }
        
        if ( fOpt.isShowOperations() ) {
            // compartment divider
            x = (int) x();
            g.drawLine( x - getWidth() / 2, y + 3, x + getWidth() / 2 - 1, y + 3 );
            x -= ( getWidth() - 10 ) / 2;
            y += 3;
            for ( int i = 0; i < fOperations.size(); i++ ) {
                y += fm.getHeight();
                g.drawString( fOprSignatures[i], x, y );
            }
        }
    }
}