Search in sources :

Example 1 with NoteCreateCommand

use of org.jkiss.dbeaver.ext.erd.command.NoteCreateCommand in project dbeaver by serge-rider.

the class DiagramContainerEditPolicy method getCreateCommand.

/**
     * @see ContainerEditPolicy#getCreateCommand(org.eclipse.gef.requests.CreateRequest)
     */
@Override
protected Command getCreateCommand(CreateRequest request) {
    DiagramPart diagramPart = (DiagramPart) getHost();
    Point location = request.getLocation();
    diagramPart.getFigure().translateToRelative(location);
    Object newObject = request.getNewObject();
    if (newObject instanceof ERDNote) {
        return new NoteCreateCommand(diagramPart, (ERDNote) newObject, location);
    }
    Collection<ERDEntity> entities = null;
    if (newObject instanceof ERDEntity) {
        entities = Collections.singletonList((ERDEntity) newObject);
    } else if (newObject instanceof Collection) {
        entities = (Collection<ERDEntity>) newObject;
    }
    if (CommonUtils.isEmpty(entities)) {
        return null;
    }
    return new EntityAddCommand(diagramPart, entities, location);
}
Also used : NoteCreateCommand(org.jkiss.dbeaver.ext.erd.command.NoteCreateCommand) ERDEntity(org.jkiss.dbeaver.ext.erd.model.ERDEntity) ERDNote(org.jkiss.dbeaver.ext.erd.model.ERDNote) Collection(java.util.Collection) Point(org.eclipse.draw2d.geometry.Point) EntityAddCommand(org.jkiss.dbeaver.ext.erd.command.EntityAddCommand) DiagramPart(org.jkiss.dbeaver.ext.erd.part.DiagramPart)

Aggregations

Collection (java.util.Collection)1 Point (org.eclipse.draw2d.geometry.Point)1 EntityAddCommand (org.jkiss.dbeaver.ext.erd.command.EntityAddCommand)1 NoteCreateCommand (org.jkiss.dbeaver.ext.erd.command.NoteCreateCommand)1 ERDEntity (org.jkiss.dbeaver.ext.erd.model.ERDEntity)1 ERDNote (org.jkiss.dbeaver.ext.erd.model.ERDNote)1 DiagramPart (org.jkiss.dbeaver.ext.erd.part.DiagramPart)1