tests/org.eclipse.graphiti.testtool.sketch/src/org/eclipse/graphiti/testtool/sketch/features/ModifyControlPointsFeature.java - graphiti/org.eclipse.graphiti - Git at Google

 /*********************************************************************
 * Copyright (c) 2005, 2019 SAP SE
 *
 * This program and the accompanying materials are made
 * available under the terms of the Eclipse Public License 2.0
 * which is available at https://www.eclipse.org/legal/epl-2.0/
 *
 * Contributors:
 *    SAP SE - initial API, implementation and documentation
 *
 * SPDX-License-Identifier: EPL-2.0
 **********************************************************************/
 package org.eclipse.graphiti.testtool.sketch.features;

 import org.eclipse.core.runtime.OperationCanceledException;
 import org.eclipse.graphiti.features.IFeatureProvider;
 import org.eclipse.graphiti.features.context.IContext;
 import org.eclipse.graphiti.features.context.ICustomContext;
 import org.eclipse.graphiti.features.custom.AbstractCustomFeature;
 import org.eclipse.graphiti.mm.algorithms.Polyline;
 import org.eclipse.graphiti.mm.algorithms.styles.PrecisionPoint;
 import org.eclipse.graphiti.mm.algorithms.styles.StylesFactory;
 import org.eclipse.graphiti.mm.pictograms.CurvedConnection;
 import org.eclipse.graphiti.mm.pictograms.Diagram;
 import org.eclipse.graphiti.mm.pictograms.PictogramElement;
 import org.eclipse.jface.dialogs.IInputValidator;
 import org.eclipse.jface.dialogs.InputDialog;
 import org.eclipse.jface.viewers.ILabelProvider;
 import org.eclipse.jface.viewers.ILabelProviderListener;
 import org.eclipse.jface.viewers.ISelection;
 import org.eclipse.jface.viewers.ISelectionChangedListener;
 import org.eclipse.jface.viewers.IStructuredContentProvider;
 import org.eclipse.jface.viewers.IStructuredSelection;
 import org.eclipse.jface.viewers.SelectionChangedEvent;
 import org.eclipse.jface.viewers.StructuredSelection;
 import org.eclipse.jface.viewers.Viewer;
 import org.eclipse.jface.window.Window;
 import org.eclipse.swt.SWT;
 import org.eclipse.swt.events.SelectionEvent;
 import org.eclipse.swt.events.SelectionListener;
 import org.eclipse.swt.graphics.Image;
 import org.eclipse.swt.layout.GridData;
 import org.eclipse.swt.layout.GridLayout;
 import org.eclipse.swt.widgets.Button;
 import org.eclipse.swt.widgets.Composite;
 import org.eclipse.swt.widgets.Control;
 import org.eclipse.swt.widgets.Display;
 import org.eclipse.swt.widgets.Shell;
 import org.eclipse.ui.dialogs.ListDialog;

 /**
  * The Class ModifyControlPointsFeature.
  */
 public class ModifyControlPointsFeature extends AbstractCustomFeature {

 	public ModifyControlPointsFeature(IFeatureProvider fp) {
 		super(fp);
 	}

 	@Override
 	public String getName() {
 		return "Modify control points...";
 	}

 	@Override
 	public boolean canExecute(ICustomContext context) {
 		return true;
 	}

 	public void execute(ICustomContext context) {
 		PictogramElement[] pes = context.getPictogramElements();
 		if (pes != null && pes.length > 0 && pes[0] instanceof CurvedConnection) {
 			CurvedConnection connection = (CurvedConnection) pes[0];

 			final ListDialogWithButtons dialog = new ListDialogWithButtons(Display.getCurrent().getActiveShell(),
 					connection);
 			dialog.setTitle("Control Points");
 			dialog.setContentProvider(new IStructuredContentProvider() {
 				public void inputChanged(Viewer viewer, Object oldInput, Object newInput) {
 				}

 				public void dispose() {
 				}

 				public Object[] getElements(Object inputElement) {
 					if (inputElement instanceof CurvedConnection) {
 						return ((CurvedConnection) inputElement).getControlPoints().toArray();
 					}
 					return null;
 				}
 			});
 			dialog.setLabelProvider(new ILabelProvider() {

 				public void removeListener(ILabelProviderListener listener) {
 				}

 				public boolean isLabelProperty(Object element, String property) {
 					return false;
 				}

 				public void dispose() {
 				}

 				public void addListener(ILabelProviderListener listener) {
 				}

 				public String getText(Object element) {
 					if (element instanceof PrecisionPoint) {
 						return ((PrecisionPoint) element).getX() + ", " + ((PrecisionPoint) element).getY();
 					}
 					return null;
 				}

 				public Image getImage(Object element) {
 					return null;
 				}
 			});
 			dialog.setInput(connection);
 			if (dialog.open() != Window.OK) {
 				throw new OperationCanceledException("Cancelled by user");
 			}

 		}
 	}

 	@Override
 	public boolean isAvailable(IContext context) {
 		if (context instanceof ICustomContext) {
 			PictogramElement[] pes = ((ICustomContext) context).getPictogramElements();
 			if (pes == null || pes.length > 1) {
 				return false;
 			}
 			PictogramElement pe = pes[0];
 			if (pe instanceof CurvedConnection && pe.getGraphicsAlgorithm() instanceof Polyline
 					&& !(pe instanceof Diagram)) {
 				return true;
 			}
 		}
 		return false;
 	}

 	private class ListDialogWithButtons extends ListDialog implements ISelectionChangedListener {

 		private Button addButton;
 		private Button removeButton;
 		private Button upButton;
 		private Button downButton;
 		private CurvedConnection connection;

 		public ListDialogWithButtons(Shell parent, CurvedConnection connection) {
 			super(parent);
 			this.connection = connection;
 		}

 		@Override
 		protected Control createDialogArea(Composite container) {
 			Composite composite = new Composite(container, SWT.None);
 			GridLayout layout = new GridLayout();
 			layout.numColumns = 2;
 			composite.setLayout(layout);
 			GridData data = new GridData(SWT.FILL, SWT.FILL, true, true);
 			composite.setLayoutData(data);

 			Control area = super.createDialogArea(composite);
 			area.setLayoutData(new GridData(SWT.FILL, SWT.TOP, true, true));

 			Composite buttonsComposite = new Composite(composite, SWT.None);
 			layout = new GridLayout();
 			layout.numColumns = 1;
 			buttonsComposite.setLayout(layout);
 			data = new GridData(SWT.FILL, SWT.FILL, true, true);
 			buttonsComposite.setLayoutData(data);

 			addButton = new Button(buttonsComposite, SWT.PUSH);
 			addButton.setLayoutData(new GridData(SWT.FILL, SWT.TOP, true, false));
 			addButton.setText("Add...");
 			addButton.addSelectionListener(new SelectionListener() {
 				public void widgetSelected(SelectionEvent event) {
 					InputDialog inputDialog = new InputDialog(getShell(), "New Control Point",
 							"Enter a new control point in the format '1.0,1.0'.", "", new IInputValidator() {
 								public String isValid(String newText) {
 									int index = newText.indexOf(',');
 									if (index != -1) {
 										String first = newText.substring(0, index);
 										String second = newText.substring(index + 1);
 										try {
 											Double.parseDouble(first);
 											Double.parseDouble(second);
 										} catch (NumberFormatException e) {
 											return "No valid number";
 										}
 										return null;
 									}

 									return "No valid input, use comma to seperate x and y.";
 								}
 							});
 					if (inputDialog.open() == OK) {
 						String value = inputDialog.getValue();
 						int index = value.indexOf(',');
 						double firstValue = 0;
 						double secondValue = 0;
 						if (index != -1) {
 							String first = value.substring(0, index);
 							String second = value.substring(index + 1);
 							try {
 								firstValue = Double.parseDouble(first);
 								secondValue = Double.parseDouble(second);
 							} catch (NumberFormatException e) {
 								return;
 							}
 						}
 						PrecisionPoint newPoint = StylesFactory.eINSTANCE.createPrecisionPoint();
 						newPoint.setX(firstValue);
 						newPoint.setY(secondValue);
 						connection.getControlPoints().add(newPoint);
 						getTableViewer().refresh();
 						getTableViewer().setSelection(new StructuredSelection(newPoint));
 					}
 				}

 				public void widgetDefaultSelected(SelectionEvent e) {
 					widgetSelected(e);
 				}
 			});

 			removeButton = new Button(buttonsComposite, SWT.PUSH);
 			removeButton.setLayoutData(new GridData(SWT.FILL, SWT.TOP, true, false));
 			removeButton.setText("Remove");
 			removeButton.setEnabled(false);
 			removeButton.addSelectionListener(new SelectionListener() {
 				public void widgetSelected(SelectionEvent event) {
 					ISelection selection = getTableViewer().getSelection();
 					if (selection instanceof IStructuredSelection) {
 						Object firstElement = ((IStructuredSelection) selection).getFirstElement();
 						if (firstElement instanceof PrecisionPoint) {
 							PrecisionPoint point = (PrecisionPoint) firstElement;
 							CurvedConnection connection = (CurvedConnection) point.eContainer();
 							connection.getControlPoints().remove(point);
 							getTableViewer().refresh();
 							getTableViewer().setSelection(new StructuredSelection(point));
 						}
 					}
 				}

 				public void widgetDefaultSelected(SelectionEvent e) {
 					widgetSelected(e);
 				}
 			});

 			upButton = new Button(buttonsComposite, SWT.UP);
 			upButton.setLayoutData(new GridData(SWT.FILL, SWT.TOP, true, false));
 			upButton.setText("Up");
 			upButton.setEnabled(false);
 			upButton.addSelectionListener(new SelectionListener() {
 				public void widgetSelected(SelectionEvent event) {
 					ISelection selection = getTableViewer().getSelection();
 					if (selection instanceof IStructuredSelection) {
 						Object firstElement = ((IStructuredSelection) selection).getFirstElement();
 						if (firstElement instanceof PrecisionPoint) {
 							PrecisionPoint point = (PrecisionPoint) firstElement;
 							CurvedConnection connection = (CurvedConnection) point.eContainer();
 							int index = connection.getControlPoints().indexOf(point);
 							connection.getControlPoints().move(index - 1, point);
 							getTableViewer().refresh();
 							getTableViewer().setSelection(new StructuredSelection(point));
 						}
 					}
 				}

 				public void widgetDefaultSelected(SelectionEvent e) {
 					widgetSelected(e);
 				}
 			});

 			downButton = new Button(buttonsComposite, SWT.DOWN);
 			downButton.setLayoutData(new GridData(SWT.FILL, SWT.TOP, true, false));
 			downButton.setText("Down");
 			downButton.setEnabled(false);
 			downButton.addSelectionListener(new SelectionListener() {
 				public void widgetSelected(SelectionEvent event) {
 					ISelection selection = getTableViewer().getSelection();
 					if (selection instanceof IStructuredSelection) {
 						Object firstElement = ((IStructuredSelection) selection).getFirstElement();
 						if (firstElement instanceof PrecisionPoint) {
 							PrecisionPoint point = (PrecisionPoint) firstElement;
 							CurvedConnection connection = (CurvedConnection) point.eContainer();
 							int index = connection.getControlPoints().indexOf(point);
 							connection.getControlPoints().move(index + 1, point);
 							getTableViewer().refresh();
 							getTableViewer().setSelection(new StructuredSelection(point));
 						}
 					}
 				}

 				public void widgetDefaultSelected(SelectionEvent e) {
 					widgetSelected(e);
 				}
 			});

 			getTableViewer().addSelectionChangedListener(this);

 			return composite;
 		}

 		public void selectionChanged(SelectionChangedEvent event) {
 			removeButton.setEnabled(false);
 			upButton.setEnabled(false);
 			downButton.setEnabled(false);
 			Object selection = event.getSelection();
 			if (selection instanceof IStructuredSelection) {
 				Object firstElement = ((IStructuredSelection) selection).getFirstElement();
 				if (firstElement instanceof PrecisionPoint) {
 					PrecisionPoint point = (PrecisionPoint) firstElement;
 					CurvedConnection connection = (CurvedConnection) point.eContainer();
 					int index = connection.getControlPoints().indexOf(point);
 					removeButton.setEnabled(index != -1);
 					upButton.setEnabled(index > 0);
 					downButton.setEnabled(index < connection.getControlPoints().size() - 1);
 				}
 			}
 		}
 	}
 }
	/*********************************************************************
	* Copyright (c) 2005, 2019 SAP SE
	*
	* This program and the accompanying materials are made
	* available under the terms of the Eclipse Public License 2.0
	* which is available at https://www.eclipse.org/legal/epl-2.0/
	*
	* Contributors:
	* SAP SE - initial API, implementation and documentation
	*
	* SPDX-License-Identifier: EPL-2.0
	**********************************************************************/
	package org.eclipse.graphiti.testtool.sketch.features;

	import org.eclipse.core.runtime.OperationCanceledException;
	import org.eclipse.graphiti.features.IFeatureProvider;
	import org.eclipse.graphiti.features.context.IContext;
	import org.eclipse.graphiti.features.context.ICustomContext;
	import org.eclipse.graphiti.features.custom.AbstractCustomFeature;
	import org.eclipse.graphiti.mm.algorithms.Polyline;
	import org.eclipse.graphiti.mm.algorithms.styles.PrecisionPoint;
	import org.eclipse.graphiti.mm.algorithms.styles.StylesFactory;
	import org.eclipse.graphiti.mm.pictograms.CurvedConnection;
	import org.eclipse.graphiti.mm.pictograms.Diagram;
	import org.eclipse.graphiti.mm.pictograms.PictogramElement;
	import org.eclipse.jface.dialogs.IInputValidator;
	import org.eclipse.jface.dialogs.InputDialog;
	import org.eclipse.jface.viewers.ILabelProvider;
	import org.eclipse.jface.viewers.ILabelProviderListener;
	import org.eclipse.jface.viewers.ISelection;
	import org.eclipse.jface.viewers.ISelectionChangedListener;
	import org.eclipse.jface.viewers.IStructuredContentProvider;
	import org.eclipse.jface.viewers.IStructuredSelection;
	import org.eclipse.jface.viewers.SelectionChangedEvent;
	import org.eclipse.jface.viewers.StructuredSelection;
	import org.eclipse.jface.viewers.Viewer;
	import org.eclipse.jface.window.Window;
	import org.eclipse.swt.SWT;
	import org.eclipse.swt.events.SelectionEvent;
	import org.eclipse.swt.events.SelectionListener;
	import org.eclipse.swt.graphics.Image;
	import org.eclipse.swt.layout.GridData;
	import org.eclipse.swt.layout.GridLayout;
	import org.eclipse.swt.widgets.Button;
	import org.eclipse.swt.widgets.Composite;
	import org.eclipse.swt.widgets.Control;
	import org.eclipse.swt.widgets.Display;
	import org.eclipse.swt.widgets.Shell;
	import org.eclipse.ui.dialogs.ListDialog;

	/**
	* The Class ModifyControlPointsFeature.
	*/
	public class ModifyControlPointsFeature extends AbstractCustomFeature {

	public ModifyControlPointsFeature(IFeatureProvider fp) {
	super(fp);
	}

	@Override
	public String getName() {
	return "Modify control points...";
	}

	@Override
	public boolean canExecute(ICustomContext context) {
	return true;
	}

	public void execute(ICustomContext context) {
	PictogramElement[] pes = context.getPictogramElements();
	if (pes != null && pes.length > 0 && pes[0] instanceof CurvedConnection) {
	CurvedConnection connection = (CurvedConnection) pes[0];

	final ListDialogWithButtons dialog = new ListDialogWithButtons(Display.getCurrent().getActiveShell(),
	connection);
	dialog.setTitle("Control Points");
	dialog.setContentProvider(new IStructuredContentProvider() {
	public void inputChanged(Viewer viewer, Object oldInput, Object newInput) {
	}

	public void dispose() {
	}

	public Object[] getElements(Object inputElement) {
	if (inputElement instanceof CurvedConnection) {
	return ((CurvedConnection) inputElement).getControlPoints().toArray();
	}
	return null;
	}
	});
	dialog.setLabelProvider(new ILabelProvider() {

	public void removeListener(ILabelProviderListener listener) {
	}

	public boolean isLabelProperty(Object element, String property) {
	return false;
	}

	public void dispose() {
	}

	public void addListener(ILabelProviderListener listener) {
	}

	public String getText(Object element) {
	if (element instanceof PrecisionPoint) {
	return ((PrecisionPoint) element).getX() + ", " + ((PrecisionPoint) element).getY();
	}
	return null;
	}

	public Image getImage(Object element) {
	return null;
	}
	});
	dialog.setInput(connection);
	if (dialog.open() != Window.OK) {
	throw new OperationCanceledException("Cancelled by user");
	}

	}
	}

	@Override
	public boolean isAvailable(IContext context) {
	if (context instanceof ICustomContext) {
	PictogramElement[] pes = ((ICustomContext) context).getPictogramElements();
	if (pes == null \|\| pes.length > 1) {
	return false;
	}
	PictogramElement pe = pes[0];
	if (pe instanceof CurvedConnection && pe.getGraphicsAlgorithm() instanceof Polyline
	&& !(pe instanceof Diagram)) {
	return true;
	}
	}
	return false;
	}

	private class ListDialogWithButtons extends ListDialog implements ISelectionChangedListener {

	private Button addButton;
	private Button removeButton;
	private Button upButton;
	private Button downButton;
	private CurvedConnection connection;

	public ListDialogWithButtons(Shell parent, CurvedConnection connection) {
	super(parent);
	this.connection = connection;
	}

	@Override
	protected Control createDialogArea(Composite container) {
	Composite composite = new Composite(container, SWT.None);
	GridLayout layout = new GridLayout();
	layout.numColumns = 2;
	composite.setLayout(layout);
	GridData data = new GridData(SWT.FILL, SWT.FILL, true, true);
	composite.setLayoutData(data);

	Control area = super.createDialogArea(composite);
	area.setLayoutData(new GridData(SWT.FILL, SWT.TOP, true, true));

	Composite buttonsComposite = new Composite(composite, SWT.None);
	layout = new GridLayout();
	layout.numColumns = 1;
	buttonsComposite.setLayout(layout);
	data = new GridData(SWT.FILL, SWT.FILL, true, true);
	buttonsComposite.setLayoutData(data);

	addButton = new Button(buttonsComposite, SWT.PUSH);
	addButton.setLayoutData(new GridData(SWT.FILL, SWT.TOP, true, false));
	addButton.setText("Add...");
	addButton.addSelectionListener(new SelectionListener() {
	public void widgetSelected(SelectionEvent event) {
	InputDialog inputDialog = new InputDialog(getShell(), "New Control Point",
	"Enter a new control point in the format '1.0,1.0'.", "", new IInputValidator() {
	public String isValid(String newText) {
	int index = newText.indexOf(',');
	if (index != -1) {
	String first = newText.substring(0, index);
	String second = newText.substring(index + 1);
	try {
	Double.parseDouble(first);
	Double.parseDouble(second);
	} catch (NumberFormatException e) {
	return "No valid number";
	}
	return null;
	}

	return "No valid input, use comma to seperate x and y.";
	}
	});
	if (inputDialog.open() == OK) {
	String value = inputDialog.getValue();
	int index = value.indexOf(',');
	double firstValue = 0;
	double secondValue = 0;
	if (index != -1) {
	String first = value.substring(0, index);
	String second = value.substring(index + 1);
	try {
	firstValue = Double.parseDouble(first);
	secondValue = Double.parseDouble(second);
	} catch (NumberFormatException e) {
	return;
	}
	}
	PrecisionPoint newPoint = StylesFactory.eINSTANCE.createPrecisionPoint();
	newPoint.setX(firstValue);
	newPoint.setY(secondValue);
	connection.getControlPoints().add(newPoint);
	getTableViewer().refresh();
	getTableViewer().setSelection(new StructuredSelection(newPoint));
	}
	}

	public void widgetDefaultSelected(SelectionEvent e) {
	widgetSelected(e);
	}
	});

	removeButton = new Button(buttonsComposite, SWT.PUSH);
	removeButton.setLayoutData(new GridData(SWT.FILL, SWT.TOP, true, false));
	removeButton.setText("Remove");
	removeButton.setEnabled(false);
	removeButton.addSelectionListener(new SelectionListener() {
	public void widgetSelected(SelectionEvent event) {
	ISelection selection = getTableViewer().getSelection();
	if (selection instanceof IStructuredSelection) {
	Object firstElement = ((IStructuredSelection) selection).getFirstElement();
	if (firstElement instanceof PrecisionPoint) {
	PrecisionPoint point = (PrecisionPoint) firstElement;
	CurvedConnection connection = (CurvedConnection) point.eContainer();
	connection.getControlPoints().remove(point);
	getTableViewer().refresh();
	getTableViewer().setSelection(new StructuredSelection(point));
	}
	}
	}

	public void widgetDefaultSelected(SelectionEvent e) {
	widgetSelected(e);
	}
	});

	upButton = new Button(buttonsComposite, SWT.UP);
	upButton.setLayoutData(new GridData(SWT.FILL, SWT.TOP, true, false));
	upButton.setText("Up");
	upButton.setEnabled(false);
	upButton.addSelectionListener(new SelectionListener() {
	public void widgetSelected(SelectionEvent event) {
	ISelection selection = getTableViewer().getSelection();
	if (selection instanceof IStructuredSelection) {
	Object firstElement = ((IStructuredSelection) selection).getFirstElement();
	if (firstElement instanceof PrecisionPoint) {
	PrecisionPoint point = (PrecisionPoint) firstElement;
	CurvedConnection connection = (CurvedConnection) point.eContainer();
	int index = connection.getControlPoints().indexOf(point);
	connection.getControlPoints().move(index - 1, point);
	getTableViewer().refresh();
	getTableViewer().setSelection(new StructuredSelection(point));
	}
	}
	}

	public void widgetDefaultSelected(SelectionEvent e) {
	widgetSelected(e);
	}
	});

	downButton = new Button(buttonsComposite, SWT.DOWN);
	downButton.setLayoutData(new GridData(SWT.FILL, SWT.TOP, true, false));
	downButton.setText("Down");
	downButton.setEnabled(false);
	downButton.addSelectionListener(new SelectionListener() {
	public void widgetSelected(SelectionEvent event) {
	ISelection selection = getTableViewer().getSelection();
	if (selection instanceof IStructuredSelection) {
	Object firstElement = ((IStructuredSelection) selection).getFirstElement();
	if (firstElement instanceof PrecisionPoint) {
	PrecisionPoint point = (PrecisionPoint) firstElement;
	CurvedConnection connection = (CurvedConnection) point.eContainer();
	int index = connection.getControlPoints().indexOf(point);
	connection.getControlPoints().move(index + 1, point);
	getTableViewer().refresh();
	getTableViewer().setSelection(new StructuredSelection(point));
	}
	}
	}

	public void widgetDefaultSelected(SelectionEvent e) {
	widgetSelected(e);
	}
	});

	getTableViewer().addSelectionChangedListener(this);

	return composite;
	}

	public void selectionChanged(SelectionChangedEvent event) {
	removeButton.setEnabled(false);
	upButton.setEnabled(false);
	downButton.setEnabled(false);
	Object selection = event.getSelection();
	if (selection instanceof IStructuredSelection) {
	Object firstElement = ((IStructuredSelection) selection).getFirstElement();
	if (firstElement instanceof PrecisionPoint) {
	PrecisionPoint point = (PrecisionPoint) firstElement;
	CurvedConnection connection = (CurvedConnection) point.eContainer();
	int index = connection.getControlPoints().indexOf(point);
	removeButton.setEnabled(index != -1);
	upButton.setEnabled(index > 0);
	downButton.setEnabled(index < connection.getControlPoints().size() - 1);
	}
	}
	}
	}
	}