public class CreateGoalFeature extends AbstractCreateNodeFeature {

	public static final String NAME = "Goal";

	public static final String DESCRIPTION = "Create Goal";
	
	public CreateGoalFeature(IFeatureProvider fp) {
		super(fp, NAME, DESCRIPTION);
	}

	@Override
	public Object[] create(ICreateContext context) {
		Goal goal = XxegsnFactory.eINSTANCE.createGoal();
		goal.setIdentifier(getUniqueId(IIdentifierConstants.GOAL));		
		getDiagram().eResource().getContents().add(goal);
		addGraphicalRepresentation(context, goal);

		return new Object[] { goal };
	}

	@Override
	public String getCreateImageId() {
		return IImageConstants.GOAL;
	}
}

	@Override
	public PictogramElement add(IAddContext context) {
		Diagram diagram = (Diagram) context.getTargetContainer();

		Goal goal = (Goal) context.getNewObject();

		IPeCreateService peCreateService = getPeCreateService();
		IGaService gaService = getGaService();

		// Container-shape
		ContainerShape containerShape = peCreateService.createContainerShape(diagram, true);
		peCreateService.createChopboxAnchor(containerShape);
		link(containerShape, goal);

		// Outer-rectangle
		Rectangle outerRectangle = gaService.createInvisibleRectangle(containerShape);
		gaService.setLocationAndSize(outerRectangle, context.getX(), context.getY(), context.getWidth(), context.getHeight());

		// Inner-rectangle
		Rectangle innerRectangle = gaService.createRectangle(outerRectangle);
		innerRectangle.setStyle(getStyle(IModelConstants.Goal));
		
		addIdentifier(containerShape);
		addDescription(containerShape);

		updatePictogramElement(containerShape);
		layoutPictogramElement(containerShape);
		
		activateDirectEdit(containerShape, IDiagramConstants.description);

		return containerShape;
	}