ContainerLayoutStrategy.java

package husacct.graphics.task.modulelayout;

import husacct.graphics.domain.figures.BaseFigure;
import husacct.graphics.domain.figures.ParentFigure;
import husacct.graphics.task.modulelayout.layered.LayoutStrategy;

import java.awt.Rectangle;
import java.awt.geom.Point2D;
import java.awt.geom.Rectangle2D;

import org.jhotdraw.draw.Figure;

public class ContainerLayoutStrategy implements LayoutStrategy {
	private static final int	MIN_FIGURES_PER_ROW	= 3;
	private int					HORZ_SPACING = 30;
	private int					VERT_SPACING = 30;
	
	private ParentFigure		container			= null;
	private Point2D.Double		location;
	
	public ContainerLayoutStrategy(ParentFigure theParent) {
		container = theParent;
		location = new Point2D.Double(HORZ_SPACING, VERT_SPACING);
	}
	
	@Override
	public void doLayout() {
		BaseFigure[] figures = container.getChildFigures();
		Rectangle2D.Double parentBounds = container.getBounds();
		
		int maxItemsPerRow = MIN_FIGURES_PER_ROW;
		int itemsOnRow = 0;
		double heightOnRow = 0;
		
		if (figures.length > MIN_FIGURES_PER_ROW) maxItemsPerRow = (int) Math
				.ceil(Math.sqrt(figures.length));
		
		for (Figure f : figures) {
			BaseFigure figure = (BaseFigure) f;
			
			if (!figure.isLine()) {
				Rectangle2D.Double bounds = figure.getBounds();
				
				Point2D.Double anchor = new Point2D.Double(parentBounds.x
						+ location.x, parentBounds.y + location.y);
				Point2D.Double lead = new Point2D.Double(parentBounds.x
						+ location.x + bounds.width, parentBounds.y
						+ location.y + bounds.height);
				
				figure.setBounds(anchor, lead);
				
				location.x += bounds.width + HORZ_SPACING;
				itemsOnRow++;
				heightOnRow = Math.max(bounds.height, heightOnRow);
			}
			
			if (itemsOnRow >= maxItemsPerRow) {
				itemsOnRow = 0;
				location.y += heightOnRow + VERT_SPACING;
				location.x = HORZ_SPACING;
			}
		}
		
		// Calculate size container figure
		Rectangle newSize = new Rectangle();
		for (Figure f : figures) {
			Rectangle2D.Double bounds = f.getBounds();
			newSize.add(new Point2D.Double(bounds.x + bounds.width, bounds.y + bounds.height));
		}
		int minWidth = newSize.width + HORZ_SPACING;
		int minHeight = newSize.height + VERT_SPACING;
		//container.setMinimalSizes(minWidth, minHeight);
		
		Rectangle2D.Double bounds = container.getBounds();
		Point2D.Double anchor = new Point2D.Double(bounds.x, bounds.y);
		Point2D.Double lead = new Point2D.Double(anchor.x + minWidth + 10, anchor.y + minHeight + 10);
		container.updateBounds(anchor, lead);
	}
}