org.eclipse.gmf.runtime.diagram.ui.editparts.DiagramEditPart Java Examples

public byte [] copyToImageByteArray(final Diagram diagram,final Element elementToDraw, final int maxWidth, final int maxHeight, final ImageFileFormat format, final IProgressMonitor monitor, final PreferencesHint preferencesHint, final boolean useMargins) throws Exception {
	final Shell shell = new Shell();
	DiagramEditPart diagramEditPart = null ;
	EditPart ep = null ;
	try {
		diagramEditPart = createDiagramEditPart(diagram,
				shell, preferencesHint);
		Assert.isNotNull(diagramEditPart);
		ep = GMFTools.findEditPart(diagramEditPart,elementToDraw) ;


		if(ep == null) {
               throw new Exception("Element to draw not found") ;
           }

		return copyToImageByteArray(diagramEditPart, Collections.singletonList(ep), maxWidth, maxHeight, format, monitor, useMargins);
	}catch(final Exception e){
		e.printStackTrace();
		throw e ;
	} finally {
		shell.dispose();
	}
}
 

/**
* @generated
*/
private static void createMarkers(IFile target, IStatus validationStatus, DiagramEditPart diagramEditPart) {
	if (validationStatus.isOK()) {
		return;
	}
	final IStatus rootStatus = validationStatus;
	List allStatuses = new ArrayList();
	ProcessDiagramEditorUtil.LazyElement2ViewMap element2ViewMap = new ProcessDiagramEditorUtil.LazyElement2ViewMap(
			diagramEditPart.getDiagramView(),
			collectTargetElements(rootStatus, new HashSet<EObject>(), allStatuses));
	for (Iterator it = allStatuses.iterator(); it.hasNext();) {
		IConstraintStatus nextStatus = (IConstraintStatus) it.next();
		View view = ProcessDiagramEditorUtil.findView(diagramEditPart, nextStatus.getTarget(), element2ViewMap);
		addMarker(diagramEditPart.getViewer(), target, view.eResource().getURIFragment(view),
				EMFCoreUtil.getQualifiedName(nextStatus.getTarget(), true), nextStatus.getMessage(),
				nextStatus.getSeverity());
	}
}
 

/**
* @generated
*/
private static void createMarkers(IFile target, IStatus validationStatus, DiagramEditPart diagramEditPart) {
	if (validationStatus.isOK()) {
		return;
	}
	final IStatus rootStatus = validationStatus;
	List allStatuses = new ArrayList();
	CrossflowDiagramEditorUtil.LazyElement2ViewMap element2ViewMap = new CrossflowDiagramEditorUtil.LazyElement2ViewMap(
			diagramEditPart.getDiagramView(),
			collectTargetElements(rootStatus, new HashSet<EObject>(), allStatuses));
	for (Iterator it = allStatuses.iterator(); it.hasNext();) {
		IConstraintStatus nextStatus = (IConstraintStatus) it.next();
		View view = CrossflowDiagramEditorUtil.findView(diagramEditPart, nextStatus.getTarget(), element2ViewMap);
		addMarker(diagramEditPart.getViewer(), target, view.eResource().getURIFragment(view),
				EMFCoreUtil.getQualifiedName(nextStatus.getTarget(), true), nextStatus.getMessage(),
				nextStatus.getSeverity());
	}
}
 

/**
* @generated
*/
private static void createMarkers(IFile target, Diagnostic emfValidationStatus, DiagramEditPart diagramEditPart) {
	if (emfValidationStatus.getSeverity() == Diagnostic.OK) {
		return;
	}
	final Diagnostic rootStatus = emfValidationStatus;
	List allDiagnostics = new ArrayList();
	CrossflowDiagramEditorUtil.LazyElement2ViewMap element2ViewMap = new CrossflowDiagramEditorUtil.LazyElement2ViewMap(
			diagramEditPart.getDiagramView(),
			collectTargetElements(rootStatus, new HashSet<EObject>(), allDiagnostics));
	for (Iterator it = emfValidationStatus.getChildren().iterator(); it.hasNext();) {
		Diagnostic nextDiagnostic = (Diagnostic) it.next();
		List data = nextDiagnostic.getData();
		if (data != null && !data.isEmpty() && data.get(0) instanceof EObject) {
			EObject element = (EObject) data.get(0);
			View view = CrossflowDiagramEditorUtil.findView(diagramEditPart, element, element2ViewMap);
			addMarker(diagramEditPart.getViewer(), target, view.eResource().getURIFragment(view),
					EMFCoreUtil.getQualifiedName(element, true), nextDiagnostic.getMessage(),
					diagnosticToStatusSeverity(nextDiagnostic.getSeverity()));
		}
	}
}
 

public static  void updateActiviesSize(DiagramEditPart root) {
    resizeActivities(root);
    for(Object o : root.getChildren()){
        if(o instanceof PoolEditPart){
            for(Object child :((PoolEditPart) o).getChildren()){
                if(child instanceof PoolPoolCompartmentEditPart){
                    resizeActivities((EditPart) child);
                    for(Object laneChild : ((PoolPoolCompartmentEditPart) child).getChildren()){
                        if(laneChild instanceof LaneEditPart){
                            for(Object child2 : ((LaneEditPart) laneChild).getChildren()){
                                if(child2 instanceof LaneLaneCompartmentEditPart){
                                    resizeActivities((EditPart) child2);
                                }
                            }
                        }
                    }
                }
            }
        }
    }
}
 

public void createMarkers(final IFile
        target, final Diagnostic emfValidationStatus, final DiagramEditPart diagramEditPart) {
    if (emfValidationStatus.getSeverity() == Diagnostic.OK) {
        return;
    }
    final Diagnostic rootStatus = emfValidationStatus;
    final List allDiagnostics = new ArrayList();
    final ProcessDiagramEditorUtil.LazyElement2ViewMap element2ViewMap =
            new ProcessDiagramEditorUtil.LazyElement2ViewMap(
                    diagramEditPart.getDiagramView(),
                    collectTargetElements(rootStatus, new HashSet<EObject>(), allDiagnostics));
    for (final Iterator it = emfValidationStatus.getChildren().iterator(); it.hasNext();) {
        final Diagnostic nextDiagnostic = (Diagnostic) it.next();
        final List data = nextDiagnostic.getData();
        if (data != null && !data.isEmpty() && data.get(0) instanceof EObject) {
            final EObject element = (EObject) data.get(0);
            final View view = ProcessDiagramEditorUtil.findView(
                    diagramEditPart, element, element2ViewMap);
            addMarker(diagramEditPart, null, diagramEditPart.getViewer(), target, view.eResource().getURIFragment(view),
                    EMFCoreUtil.getQualifiedName(element, true),
                    nextDiagnostic.getMessage(), diagnosticToStatusSeverity(nextDiagnostic.getSeverity()));
        }
    }
}
 

public static void removeMessageFlow(TransactionalEditingDomain editingDomain, AbstractCatchMessageEvent target,
        DiagramEditPart dep) {

    EditPart ep = findEditPart(dep, target);
    CompositeCommand command = new CompositeCommand("Remove MessageFlows");

    for (Object connection : ((AbstractGraphicalEditPart) ep).getTargetConnections()) {
        if (connection instanceof MessageFlowEditPart) {
            MessageFlowEditPart connectionPart = (MessageFlowEditPart) connection;
            MessageFlow flow = (MessageFlow) connectionPart.resolveSemanticElement();
            SetCommand c = new SetCommand(editingDomain, flow.getTarget(),
                    ProcessPackage.Literals.ABSTRACT_CATCH_MESSAGE_EVENT__EVENT, null);
            editingDomain.getCommandStack().execute(c);
            command.add(new DeleteCommand(editingDomain, connectionPart.getPrimaryView()));
            DestroyElementRequest req = new DestroyElementRequest(editingDomain, flow, false);
            DestroyElementCommand rmComd = new DestroyElementCommand(req);
            command.add(rmComd);
        }
    }

    dep.getDiagramEditDomain().getDiagramCommandStack().execute(new ICommandProxy(command.reduce()));
    dep.getDiagramEditDomain().getDiagramCommandStack().flush();
    dep.refresh();
}
 

/**
* @generated
*/
private static void validate(DiagramEditPart diagramEditPart, View view) {
	IFile target = view.eResource() != null ? WorkspaceSynchronizer.getFile(view.eResource()) : null;
	if (target != null) {
		ProcessMarkerNavigationProvider.deleteMarkers(target);
	}
	Diagnostic diagnostic = runEMFValidator(view);
	createMarkers(target, diagnostic, diagramEditPart);
	IBatchValidator validator = (IBatchValidator) ModelValidationService.getInstance()
			.newValidator(EvaluationMode.BATCH);
	validator.setIncludeLiveConstraints(true);
	if (view.isSetElement() && view.getElement() != null) {
		IStatus status = validator.validate(view.getElement());
		createMarkers(target, status, diagramEditPart);
	}
}
 

@Override
protected void addElementsToDiagram(Diagram diagram, IProgressMonitor monitor) {
	Element container = diagramManager.getDiagramContainer(diagram);
	AbstractDiagramElementsManager diagramElementsManager;
	DiagramEditPart diagep = diagramManager.getActiveDiagramEditPart();
	if(diagram.getType().equals("PapyrusUMLClassDiagram")){					
		diagramElementsManager = new ClassDiagramElementsManager(diagep);
	}else if(diagram.getType().equals("PapyrusUMLActivityDiagram")){
		diagramElementsManager = new ActivityDiagramElementsManager(diagep);
	}else if(diagram.getType().equals("PapyrusUMLStateMachineDiagram")){
		diagramElementsManager = new StateMachineDiagramElementsManager(diagep);
	}else{
		return;
	}
	
	List<Element> baseElements = modelManager.getAllChildrenOfPackage(container);
	diagramElementsManager.addElementsToDiagram(baseElements);
}
 

@SuppressWarnings("unchecked")
public static void editMessageFlowName(Message message, AbstractCatchMessageEvent target, DiagramEditPart dep) {
    EditPart targetEP = findEditPart(dep, target);
    Optional<MessageFlowEditPart> connectionPart = ((AbstractGraphicalEditPart) targetEP).getTargetConnections().stream()
            .filter(MessageFlowEditPart.class::isInstance)
            .map(MessageFlowEditPart.class::cast)
            .findFirst();
    if (connectionPart.isPresent()) {
        MessageFlow messageFlowToEdit = (MessageFlow) connectionPart.get().resolveSemanticElement();
        EMFModelUpdater<MessageFlow> updater = new EMFModelUpdater<>();
        MessageFlow workingCopy = updater.from(messageFlowToEdit).getWorkingCopy();
        workingCopy.setName(message.getName());
        updater.update();
        dep.refresh();
    }
}
 

/**
 * @Generated BonitaSoft
 */
@Override
public Object getAdapter(Class key) {

	if (key == SnapToHelper.class) {
		EditPart parent = getParent();
		while (!(parent instanceof DiagramEditPart)) {
			parent = parent.getParent();
		}
		return GMFTools.getSnapHelper((GraphicalEditPart) parent);
	}

	return super.getAdapter(key);
}
 

protected DiagramEditPart createDiagramEditPart(final Diagram d) {
    if (d != null && d.eResource() != null) {
        DiagramEditPart offscreenDiagramEditPart = null;
        try {
            offscreenDiagramEditPart = offscreenEditPartFactory.createOffscreenDiagramEditPart(d);
        } catch (final Exception e) {
            //Need to call twice because sometimes for unknown reasons, the first time is not working
            offscreenDiagramEditPart = offscreenEditPartFactory.createOffscreenDiagramEditPart(d);
        }
        return offscreenDiagramEditPart;
    }
    return null;
}
 

/**
 * @Generated BonitaSoft
 */
@Override
public Object getAdapter(Class key) {

	if (key == SnapToHelper.class) {
		EditPart parent = getParent();
		while (!(parent instanceof DiagramEditPart)) {
			parent = parent.getParent();
		}
		return GMFTools.getSnapHelper((GraphicalEditPart) parent);
	}

	return super.getAdapter(key);
}
 

/**
 * @Generated BonitaSoft
 */
@Override
public Object getAdapter(Class key) {

	if (key == SnapToHelper.class) {
		EditPart parent = getParent();
		while (!(parent instanceof DiagramEditPart)) {
			parent = parent.getParent();
		}
		return GMFTools.getSnapHelper((GraphicalEditPart) parent);
	}

	return super.getAdapter(key);
}
 

/**
 * @Generated BonitaSoft
 */
@Override
public Object getAdapter(Class key) {

	if (key == SnapToHelper.class) {
		EditPart parent = getParent();
		while (!(parent instanceof DiagramEditPart)) {
			parent = parent.getParent();
		}
		return GMFTools.getSnapHelper((GraphicalEditPart) parent);
	}

	return super.getAdapter(key);
}
 

private DiagramEditPart getCurrentDiagramEditPart() {
	IEditorPart editor = PlatformUI.getWorkbench().getActiveWorkbenchWindow().getActivePage().getActiveEditor() ;
	if(editor instanceof DiagramEditor){
		return ((DiagramEditor)editor).getDiagramEditPart();
	}
	return null;
}
 

/**
 * @Generated BonitaSoft
 */
@Override
public java.lang.Object getAdapter(Class key) {

	if (key == SnapToHelper.class) {
		EditPart parent = getParent();
		while (!(parent instanceof DiagramEditPart)) {
			parent = parent.getParent();
		}
		return GMFTools.getSnapHelper((GraphicalEditPart) parent);
	}

	return super.getAdapter(key);
}
 

/**
 * @Generated BonitaSoft
 */
@Override
public java.lang.Object getAdapter(Class key) {

	if (key == SnapToHelper.class) {
		EditPart parent = getParent();
		while (!(parent instanceof DiagramEditPart)) {
			parent = parent.getParent();
		}
		return GMFTools.getSnapHelper((GraphicalEditPart) parent);
	}

	return super.getAdapter(key);
}
 

/**
 * @Generated BonitaSoft
 */
@Override
public Object getAdapter(Class key) {

	if (key == SnapToHelper.class) {
		EditPart parent = getParent();
		while (!(parent instanceof DiagramEditPart)) {
			parent = parent.getParent();
		}
		return GMFTools.getSnapHelper((GraphicalEditPart) parent);
	}

	return super.getAdapter(key);
}
 

@Override
public Object getAdapter(Class key) {

    if (key == SnapToHelper.class) {
        EditPart parent = getParent();
        while (!(parent instanceof DiagramEditPart)) {
            parent = parent.getParent();
        }
        return GMFTools.getSnapHelper((GraphicalEditPart) parent);
    }

    return super.getAdapter(key);
}
 

protected DiagramEditPart retrieveEditPartFromOpenedEditors(final Diagram d) {
    final IWorkbenchPage activePage = getActivePage();
    if (activePage != null) {
        for (final IEditorReference ep : activePage.getEditorReferences()) {
            final IEditorPart editor = ep.getEditor(false);
            if (editor instanceof DiagramEditor
                    && ((DiagramEditor) editor).getDiagram().equals(d)
                    && ((DiagramEditor) editor).getDiagramEditPart() != null) {
                    return ((DiagramEditor) editor).getDiagramEditPart();
                }
        }
    }
    return null;
}
 

/**
 * @Generated BonitaSoft
 */
@Override
public Object getAdapter(Class key) {

	if (key == SnapToHelper.class) {
		EditPart parent = getParent();
		while (!(parent instanceof DiagramEditPart)) {
			parent = parent.getParent();
		}
		return GMFTools.getSnapHelper((GraphicalEditPart) parent);
	}

	return super.getAdapter(key);
}
 

protected void validate(final DiagramEditPart diagramEditPart, final View view, final IProgressMonitor monitor) {
    final IFile target = view.eResource() != null ? WorkspaceSynchronizer.getFile(view.eResource()) : null;
    final Diagnostic diagnostic = validationMarkerProvider.runEMFValidator(view);
    validationMarkerProvider.createMarkers(target, diagnostic, diagramEditPart);
    final IBatchValidator validator = (IBatchValidator) ModelValidationService.getInstance().newValidator(
            EvaluationMode.BATCH);
    validator.setIncludeLiveConstraints(true);
    if (view.isSetElement() && view.getElement() != null && view.getElement().eResource() != null) {
        final IStatus status = validator.validate(view.getElement(), monitor);
        validationMarkerProvider.createMarkers(target, status, diagramEditPart);
    }
}
 

/**
 * @Generated BonitaSoft
 */
@Override
public Object getAdapter(Class key) {

	if (key == SnapToHelper.class) {
		EditPart parent = getParent();
		while (!(parent instanceof DiagramEditPart)) {
			parent = parent.getParent();
		}
		return GMFTools.getSnapHelper((GraphicalEditPart) parent);
	}

	return super.getAdapter(key);
}
 

/**
 * @Generated BonitaSoft
 */
@Override
public Object getAdapter(Class key) {

	if (key == SnapToHelper.class) {
		EditPart parent = getParent();
		while (!(parent instanceof DiagramEditPart)) {
			parent = parent.getParent();
		}
		return GMFTools.getSnapHelper((GraphicalEditPart) parent);
	}

	return super.getAdapter(key);
}
 

/**
 * @Generated BonitaSoft
 */
@Override
public Object getAdapter(Class key) {

	if (key == SnapToHelper.class) {
		EditPart parent = getParent();
		while (!(parent instanceof DiagramEditPart)) {
			parent = parent.getParent();
		}
		return GMFTools.getSnapHelper((GraphicalEditPart) parent);
	}

	return super.getAdapter(key);
}
 

/**
 * @Generated BonitaSoft
 */
@Override
public Object getAdapter(Class key) {

	if (key == SnapToHelper.class) {
		EditPart parent = getParent();
		while (!(parent instanceof DiagramEditPart)) {
			parent = parent.getParent();
		}
		return GMFTools.getSnapHelper((GraphicalEditPart) parent);
	}

	return super.getAdapter(key);
}
 

/**
 * @Generated BonitaSoft
 */
@Override
public Object getAdapter(Class key) {

	if (key == SnapToHelper.class) {
		EditPart parent = getParent();
		while (!(parent instanceof DiagramEditPart)) {
			parent = parent.getParent();
		}
		return GMFTools.getSnapHelper((GraphicalEditPart) parent);
	}

	return super.getAdapter(key);
}
 

public DiagramEditPart createOffscreenDiagramEditPart(final Diagram d) {
    final EObject element = d.getElement();
    if (element != null) {
        final OffscreenDiagramEditPartRunnable runnable = new OffscreenDiagramEditPartRunnable(d, factory);
        if (inUIThread()) {
            runnable.run();
        } else {
            runInUI(runnable);
        }

        toDispose.add(runnable.getDisposable());
        return runnable.getDiagramEditPart();
    }
    return null;
}
 

/**
* @generated
*/
private static int findElementsInDiagramByID(DiagramEditPart diagramPart, EObject element,
		List<EditPart> editPartCollector) {
	IDiagramGraphicalViewer viewer = (IDiagramGraphicalViewer) diagramPart.getViewer();
	final int intialNumOfEditParts = editPartCollector.size();

	if (element instanceof View) { // support notation element lookup
		EditPart editPart = (EditPart) viewer.getEditPartRegistry().get(element);
		if (editPart != null) {
			editPartCollector.add(editPart);
			return 1;
		}
	}

	String elementID = EMFCoreUtil.getProxyID(element);
	@SuppressWarnings("unchecked")
	List<EditPart> associatedParts = viewer.findEditPartsForElement(elementID, IGraphicalEditPart.class);
	// perform the possible hierarchy disjoint -> take the top-most parts only
	for (EditPart nextPart : associatedParts) {
		EditPart parentPart = nextPart.getParent();
		while (parentPart != null && !associatedParts.contains(parentPart)) {
			parentPart = parentPart.getParent();
		}
		if (parentPart == null) {
			editPartCollector.add(nextPart);
		}
	}

	if (intialNumOfEditParts == editPartCollector.size()) {
		if (!associatedParts.isEmpty()) {
			editPartCollector.add(associatedParts.get(0));
		} else {
			if (element.eContainer() != null) {
				return findElementsInDiagramByID(diagramPart, element.eContainer(), editPartCollector);
			}
		}
	}
	return editPartCollector.size() - intialNumOfEditParts;
}