org.yaoqiang.graph.shape.DataObjectShape Maven / Gradle / Ivy

Go to download
package org.yaoqiang.graph.shape;

import java.awt.Polygon;
import java.awt.Rectangle;
import java.awt.Shape;
import java.util.Map;

import org.yaoqiang.graph.util.Constants;

import com.mxgraph.canvas.mxGraphics2DCanvas;
import com.mxgraph.shape.mxBasicShape;
import com.mxgraph.util.mxUtils;
import com.mxgraph.view.mxCellState;

/**
 * DataObjectShape
 * 
 * @author Shi Yaoqiang([email protected])
 */
public class DataObjectShape extends mxBasicShape {

	public void paintShape(mxGraphics2DCanvas canvas, mxCellState state) {
		super.paintShape(canvas, state);
		
		// Paint Image
		Map style = state.getStyle();
		double scale = canvas.getScale();
		double imgWidth = 16 * scale;
		double imgHeight = 16 * scale;

		Rectangle imageBounds = state.getRectangle();
		imageBounds.setRect(imageBounds.getX() + (imageBounds.getWidth() - imgWidth) / 4, imageBounds.getY(), imgWidth, imgHeight);

		canvas.drawImage(imageBounds, canvas.getImageForStyle(style));
		
		// Paint marker
		String marker = mxUtils.getString(style, Constants.STYLE_LOOP_IMAGE, "");
		if (marker.length() != 0) {
			Rectangle markerImageBounds = state.getRectangle();
			markerImageBounds.setRect(markerImageBounds.getX() + (markerImageBounds.getWidth() - imgWidth) / 2, markerImageBounds.getY()
					+ markerImageBounds.getHeight() - imgHeight, imgWidth, imgHeight);
			canvas.drawImage(markerImageBounds, marker);
		}
		
		Rectangle rec = state.getRectangle();
		int x = rec.x;
		int y = rec.y;
		int w = rec.width;
		int a = rec.width/3;
		
		canvas.getGraphics().drawLine(x + w - a, y, x + w - a, y + a);
		canvas.getGraphics().drawLine(x + w - a, y + a, x + w, y + a);
		
	}

	public Shape createShape(mxGraphics2DCanvas canvas, mxCellState state) {
		Rectangle rec = state.getRectangle();
		int x = rec.x;
		int y = rec.y;
		int w = rec.width;
		int h = rec.height;
		int a = rec.width/3;

		Polygon dataObject = new Polygon();
		dataObject.addPoint(x, y);
		dataObject.addPoint(x, y + h);
		dataObject.addPoint(x + w, y + h);
		dataObject.addPoint(x + w, y + a);
		dataObject.addPoint(x + w - a, y);
		
		return dataObject;
	}
}