AbstractDiagramViewer.java

/*
 * Copyright 2017 Red Hat, Inc. and/or its affiliates.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.kie.workbench.common.stunner.client.widgets.presenters.diagram.impl;

import org.kie.workbench.common.stunner.client.widgets.presenters.canvas.AbstractCanvasViewer;
import org.kie.workbench.common.stunner.client.widgets.presenters.diagram.DiagramViewer;
import org.kie.workbench.common.stunner.client.widgets.views.WidgetWrapperView;
import org.kie.workbench.common.stunner.core.client.canvas.AbstractCanvas;
import org.kie.workbench.common.stunner.core.client.canvas.AbstractCanvasHandler;
import org.kie.workbench.common.stunner.core.client.service.ClientRuntimeError;
import org.kie.workbench.common.stunner.core.command.CommandResult;
import org.kie.workbench.common.stunner.core.command.util.CommandUtils;
import org.kie.workbench.common.stunner.core.diagram.Diagram;
import org.kie.workbench.common.stunner.core.graph.Graph;
import org.kie.workbench.common.stunner.core.graph.content.definition.DefinitionSet;
import org.kie.workbench.common.stunner.core.graph.util.GraphUtils;
import org.uberfire.mvp.ParameterizedCommand;

import static org.uberfire.commons.validation.PortablePreconditions.checkNotNull;

public abstract class AbstractDiagramViewer<D extends Diagram, H extends AbstractCanvasHandler>
        extends AbstractCanvasViewer<D, H, WidgetWrapperView, DiagramViewer.DiagramViewerCallback<D>>
        implements DiagramViewer<D, H> {

    public AbstractDiagramViewer(final WidgetWrapperView view) {
        super(view);
    }

    protected abstract AbstractCanvas getCanvas();

    @Override
    @SuppressWarnings("unchecked")
    public void open(final D item,
                     final DiagramViewer.DiagramViewerCallback<D> callback) {
        final int[] ds = getDiagramSize(item);
        // Open and initialize the canvas and its hander.
        openCanvas(getCanvas(),
                   ds[0],
                   ds[1]);
        // Notify listeners that the canvas and the handler are ready.
        callback.afterCanvasInitialized();
        // Loads and draw the diagram into the canvas handled instance.
        getHandler().draw(item,
                          new ParameterizedCommand<CommandResult<?>>() {
                              @Override
                              public void execute(CommandResult<?> result) {
                                  if (!CommandUtils.isError(result)) {
                                      callback.onSuccess();
                                  } else {
                                      callback.onError(new ClientRuntimeError("An error occurred while drawing the diagram " +
                                                                                      "[result=" + result + "]"));
                                  }
                              }
                          });
    }

    public void open(final D item,
                     final int width,
                     final int height,
                     final boolean keepAspectRatio,
                     final DiagramViewer.DiagramViewerCallback callback) {
        this.open(item,
                  new DiagramViewer.DiagramViewerCallback<D>() {
                      @Override
                      public void afterCanvasInitialized() {
                          callback.afterCanvasInitialized();
                      }

                      @Override
                      public void onSuccess() {
                          scale(width,
                                height,
                                keepAspectRatio);
                          callback.onSuccess();
                      }

                      @Override
                      public void onError(final ClientRuntimeError error) {
                          callback.onError(error);
                      }
                  });
    }

    @Override
    public void scale(final int toWidth,
                      final int toHeight) {
        this.scale(toWidth,
                   toHeight,
                   false);
    }

    @Override
    @SuppressWarnings("unchecked")
    public D getInstance() {
        return (D) getHandler().getDiagram();
    }

    protected void scale(final int toWidth,
                         final int toHeight,
                         final boolean keepAspectRatio) {
        checkNotNull("item",
                     getInstance());
        final int[] ds = getDiagramSize(getInstance());
        scale(ds[0],
              ds[1],
              toWidth,
              toHeight,
              keepAspectRatio);
    }

    private int[] getDiagramSize(final D diagram) {
        final Graph graph = diagram.getGraph();
        final double[] size = GraphUtils.getGraphSize((DefinitionSet) graph.getContent());
        final double w = size[0];
        final double h = size[1];
        return new int[]{(int) w, (int) h};
    }
}