org.eclipse.gmf.runtime.diagram.ui/src/org/eclipse/gmf/runtime/diagram/ui/internal/figures/CollapseFigure.java

/******************************************************************************
 * Copyright (c) 2002, 2010 IBM Corporation and others.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *    IBM Corporation - initial API and implementation 
 ****************************************************************************/

package org.eclipse.gmf.runtime.diagram.ui.internal.figures;

import org.eclipse.draw2d.Graphics;
import org.eclipse.draw2d.IFigure;
import org.eclipse.draw2d.RectangleFigure;
import org.eclipse.draw2d.geometry.Rectangle;
import org.eclipse.gmf.runtime.diagram.ui.internal.l10n.DiagramUIPluginImages;
import org.eclipse.swt.SWT;
import org.eclipse.swt.graphics.Image;

/**
 * @author sshaw
 *
 * Figure used to represent the collapse handle in the corner of a list compartment or shape compartment.
 */
public class CollapseFigure extends RectangleFigure {

	private boolean collapsed = false;
    private IFigure containerFigure = null;
    
    private static boolean isWinOS = SWT.getPlatform().equals("win32"); //$NON-NLS-1$
      
    /**
     * Constrcuts an instance
     */
    public CollapseFigure() {
        super();
    }

    /**
     * Constrcuts an instance
     * 
     * @param containerFigure
     *            containers's figure
     */
    public CollapseFigure(IFigure containerFigure) {
        super();
        this.containerFigure = containerFigure;
    }
	
	/*
     * (non-Javadoc)
     * 
     * @see org.eclipse.draw2d.Shape#fillShape(org.eclipse.draw2d.Graphics)
     */
	protected void fillShape(Graphics graphics) {
		
		Image img;
		
		if (isCollapsed()) {
			if (isWinOS) {
				img = DiagramUIPluginImages.get(DiagramUIPluginImages.IMG_HANDLE_EXPAND_WIN);
			} else {
				img = DiagramUIPluginImages.get(DiagramUIPluginImages.IMG_HANDLE_EXPAND);
			}
		} else {
			if (isWinOS) {
				img = DiagramUIPluginImages.get(DiagramUIPluginImages.IMG_HANDLE_COLLAPSE_WIN);
			} else {			
				img = DiagramUIPluginImages.get(DiagramUIPluginImages.IMG_HANDLE_COLLAPSE);
			}
		}
		graphics.drawImage(img, getBounds().x, getBounds().y);
	}
	
	/* (non-Javadoc)
	 * @see org.eclipse.draw2d.Shape#outlineShape(org.eclipse.draw2d.Graphics)
	 */
	protected void outlineShape(Graphics graphics) {
		// do nothing
	}
	
	/**
	 * isCollapsed
	 * Utility method to determine if the IFigure is collapse or not.
	 * 
	 * @return true if collapse, false otherwise.
	 */
	public boolean isCollapsed() {
		return collapsed;
	}

	/**
	 * setCollapsed
	 * Setter method to change collapsed state of the figure.  Will force update
	 * to repaint the figure to reflect the changes.
	 * 
	 * @param b boolean true to set collapsed, false to uncollapse.
	 */
	public void setCollapsed(boolean b) {
		collapsed = b;
		revalidate();
		repaint();
	}

    /*
     * (non-Javadoc)
     * 
     * @see org.eclipse.draw2d.Figure#isVisible()
     */
    public boolean isVisible() {
        boolean visibility = super.isVisible();
        if (visibility && containerFigure != null) {
            /*
             * Try to hide the handle if it's to be drawn outside of the
             * container edit part figure
             */
            Rectangle containerBounds = containerFigure.getClientArea()
                .getCopy();
            containerFigure.translateToAbsolute(containerBounds);
            translateToRelative(containerBounds);
            return containerBounds.contains(getBounds());
        }
        return visibility;
    }
    
    

}