FillAreaAfterDeletionDemo.java

/****************************************************************************
 **
 ** This demo file is part of yFiles for JavaFX 3.6.
 **
 ** Copyright (c) 2000-2023 by yWorks GmbH, Vor dem Kreuzberg 28,
 ** 72070 Tuebingen, Germany. All rights reserved.
 **
 ** yFiles demo files exhibit yFiles for JavaFX functionalities. Any redistribution
 ** of demo files in source code or binary form, with or without
 ** modification, is not permitted.
 **
 ** Owners of a valid software license for a yFiles for JavaFX version that this
 ** demo is shipped with are allowed to use the demo source code as basis
 ** for their own yFiles for JavaFX powered applications. Use of such programs is
 ** governed by the rights and conditions as set out in the yFiles for JavaFX
 ** license agreement.
 **
 ** THIS SOFTWARE IS PROVIDED ''AS IS'' AND ANY EXPRESS OR IMPLIED
 ** WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
 ** MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN
 ** NO EVENT SHALL yWorks BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 ** SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED
 ** TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
 ** PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
 ** LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
 ** NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
 ** SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 **
 ***************************************************************************/
package layout.fillarea;

import com.yworks.yfiles.geometry.RectD;
import com.yworks.yfiles.graph.IEdge;
import com.yworks.yfiles.graph.IModelItem;
import com.yworks.yfiles.graph.INode;
import com.yworks.yfiles.layout.LayoutExecutor;
import com.yworks.yfiles.layout.hierarchic.HierarchicLayout;
import com.yworks.yfiles.layout.partial.ComponentAssignmentStrategy;
import com.yworks.yfiles.layout.partial.FillAreaLayout;
import com.yworks.yfiles.view.GraphControl;
import com.yworks.yfiles.view.ISelectionModel;
import com.yworks.yfiles.view.input.GraphEditorInputMode;
import com.yworks.yfiles.view.input.SelectionEventArgs;
import javafx.fxml.FXML;
import javafx.scene.control.ComboBox;
import javafx.scene.paint.Color;
import javafx.scene.web.WebView;
import toolkit.DemoApplication;
import toolkit.DemoStyles;
import toolkit.WebViewUtils;

import java.io.IOException;
import java.time.Duration;

public class FillAreaAfterDeletionDemo extends DemoApplication {
  public WebView helpView;
  public GraphControl graphControl;

  private LayoutExecutor layoutExecutor;

  private ComponentAssignmentStrategy componentAssignmentStrategy = ComponentAssignmentStrategy.SINGLE;

  @FXML
  private ComboBox<ComponentAssignmentStrategy> strategyComboBox;

  private boolean layoutRunning;

  /**
   * Initializes the demo.
   */
  public void initialize() {
    WebViewUtils.initHelp(helpView, this);

    strategyComboBox.getItems().addAll(ComponentAssignmentStrategy.SINGLE, ComponentAssignmentStrategy.CONNECTED, ComponentAssignmentStrategy.CLUSTERING);
    strategyComboBox.getSelectionModel().select(0);
    strategyComboBox.valueProperty().addListener((observable, oldValue, newValue) -> componentAssignmentStrategy = newValue);

    graphControl.getGraph().setUndoEngineEnabled(true);

    DemoStyles.initDemoStyles(graphControl.getGraph());

    initializeInputModes();

    loadGraph();
  }

  private void initializeInputModes() {
    // create a GraphEditorInputMode instance and install the edit mode into the canvas.
    GraphEditorInputMode graphEditorInputMode = new GraphEditorInputMode();
    graphControl.setInputMode(graphEditorInputMode);

    // registers handlers which are called when selected nodes are deleted
    graphEditorInputMode.addDeletingSelectionListener(this::onDeletingSelection);
    graphEditorInputMode.addDeletedSelectionListener(this::onDeletedSelection);
  }

  private void onDeletingSelection(Object source, SelectionEventArgs<IModelItem> args) {
    FillAreaLayout fillAreaLayout = new FillAreaLayout();
    fillAreaLayout.setComponentAssignmentStrategy(componentAssignmentStrategy);
    fillAreaLayout.setArea(getBounds(args.getSelection()).toYRectangle());

    // configure the LayoutExecutor that will perform the layout and morph the result
    layoutExecutor = new LayoutExecutor(graphControl, fillAreaLayout);
    layoutExecutor.setDuration(Duration.ofMillis(100));
  }

  private void onDeletedSelection(Object source, SelectionEventArgs<IModelItem> args) {
    layoutExecutor.start();
  }


  /**
   * Calculates the bounds including all nodes in a selection
   * @param selection the selection
   * @return the bounds of all nodes in the selection unified
   */
  public RectD getBounds(ISelectionModel<IModelItem> selection) {
    RectD bounds = RectD.EMPTY;

    for (IModelItem item : selection) {
      if (item instanceof INode) {
        bounds = RectD.add(bounds, ((INode) item).getLayout().toRectD());
      }
      else if (item instanceof IEdge) {
        IEdge edge = (IEdge) item;
        if (edge.getSourcePort() != null) {
          bounds = RectD.add(bounds, edge.getSourcePort().getLocation());
        }
        if (edge.getTargetPort() != null) {
          bounds = RectD.add(bounds, edge.getTargetPort().getLocation());
        }
      }
    }
    return bounds;
  }


  @Override
  protected void onLoaded() {
    graphControl.fitGraphBounds();
  }

  /**
   * Loads the graph from the sample graphml file
   */
  private void loadGraph() {
    try {
      graphControl.importFromGraphML(getClass().getResource("resources/hierarchic.graphml"));
    } catch (IOException e) {
      e.printStackTrace();
    }
  }


  /**
   * Applies the {@link HierarchicLayout} to the graph
   */
  public void layoutHierarchical() {
    if (layoutRunning) {
      return;
    }
    layoutRunning = true;

    HierarchicLayout layout = new HierarchicLayout();
    layout.setOrthogonalRoutingEnabled(true);

    LayoutExecutor executor = new LayoutExecutor(graphControl, layout);
    executor.setDuration(Duration.ofMillis(500));
    executor.setContentRectUpdatingEnabled(true);

    executor.addLayoutFinishedListener((source, args) -> {
      layoutRunning = false;
    });

    executor.start();
  }

  public static void main(String[] args) {
    launch(args);
  }
}