public class CustomEditPartProvider extends AbstractEditPartProvider {

	String TrainCustomStyleID = "org.polarsys.capella.vp.guidewayscenario.TrainCustomStyle";

	@Override
	protected Class getNodeEditPartClass(View view) {
		if (view.getElement() instanceof CustomStyle) {
			CustomStyle customStyle = (CustomStyle) view.getElement();
			if (customStyle.getId().equals(TrainCustomStyleID)) {
				return CustomEditPartForTrain.class;
			}
		}

		if (view.getElement() instanceof DNodeSpec) {
			if (((DNodeSpec) view.getElement()).getTarget() instanceof Platform)
				return CustomSquareEditPartForPlatForm.class;

		}
		return super.getNodeEditPartClass(view);

	}
}

public class CustomSquareEditPartForPlatForm extends SquareEditPart {

	public CustomSquareEditPartForPlatForm(View view) {
		super(view);
		// TODO Auto-generated constructor stub
	}
	
	@Override
	protected void refreshVisuals() {
		// TODO Auto-generated method stub
		super.refreshVisuals();
		
	}

}

public class CustomEditPartForTrain extends AbstractNotSelectableShapeNodeEditPart implements IStyleEditPart {

	/**
	 * the content pane.
	 */
	protected IFigure contentPane;

	/**
	 * the primary shape.
	 */
////MyAttributeBasedImageFigure extends ImageFigure
	protected MyAttributeBasedImageFigure primaryShape;

	/**
	 * Create a new {@link ChangingImageEditPart}.
	 *
	 * @param view
	 *            the view.
	 */
	public CustomEditPartForTrain(View view) {
		super(view);

	}

	public DragTracker getDragTracker(Request request) {
		return getParent().getDragTracker(request);
	}

	protected NodeFigure createNodeFigure() {
		NodeFigure figure = createNodePlate();
		figure.setLayoutManager(new XYLayout());
		IFigure shape = createNodeShape();
		figure.add(shape);
		contentPane = setupContentPane(shape);
		return figure;
	}

	private NodeFigure createNodePlate() {
		DefaultSizeNodeFigure result = new AirStyleDefaultSizeNodeFigure(getMapMode().DPtoLP(40),
				getMapMode().DPtoLP(40));
		return result;
	}

	/**
	 * Create the instance role figure.
	 *
	 * @return the created figure.
	 */
	protected MyAttributeBasedImageFigure createNodeShape() {
		if (primaryShape == null) {

			primaryShape = new MyAttributeBasedImageFigure();
		}
		return primaryShape;
	}

	/**
	 * Return the instance role figure.
	 *
	 * @return the instance role figure.
	 */
	public MyAttributeBasedImageFigure getPrimaryShape() {
		return primaryShape;
	}

	/**
	 * Default implementation treats passed figure as content pane. Respects
	 * layout one may have set for generated figure.
	 *
	 * @param nodeShape
	 *            instance of generated figure class
	 * @return the figure
	 */
	protected IFigure setupContentPane(IFigure nodeShape) {

		return nodeShape; // use nodeShape itself as contentPane
	}

	public IFigure getContentPane() {
		if (contentPane != null) {
			return contentPane;
		}
		return super.getContentPane();
	}

	
	protected void refreshVisuals() {
		CustomStyle customStyle = (CustomStyle) this.resolveSemanticElement();
		if (customStyle.eContainer() instanceof DNode) {
			

			EObject train = ((DNode) customStyle.eContainer()).getTarget();
			if (train instanceof Train) {
//	           ((GraphicalEditPart) this.getParent()).setLayoutConstraint(this.getPrimaryChildEditPart(), this.getFigure(), new Rectangle(0, 0, getPrimaryShape().getPreferredSize().width, getPrimaryShape().getPreferredSize().height));

			}
		}

	}

	@Override
	protected void addSemanticListeners() {
		CustomStyle customStyle = (CustomStyle) this.resolveSemanticElement();
		if (customStyle.eContainer() instanceof DNode) {
			EObject train = ((DNode) customStyle.eContainer()).getTarget();
			if (train instanceof Train) {
				addListenerFilter(this.elementGuid, this, train);
			}
		}

		super.addSemanticListeners();
	}

	@Override
	protected void handleNotificationEvent(Notification notification) {
		if (notification.getNotifier() instanceof Train) {
			getPrimaryShape().updateFace();
		}
		super.handleNotificationEvent(notification);
	}

	protected void createDefaultEditPolicies() {

	}
}