ArrowNodeStyleHandleProvider.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 style.arrownodestyle;

import com.yworks.yfiles.graph.INode;
import com.yworks.yfiles.graph.styles.ArrowNodeStyle;
import com.yworks.yfiles.graph.styles.ArrowStyleShape;
import com.yworks.yfiles.view.input.IHandle;
import com.yworks.yfiles.view.input.IHandleProvider;
import com.yworks.yfiles.view.input.IInputModeContext;

import java.util.ArrayList;
import java.util.List;

/**
 * An {@link IHandleProvider} for nodes using an {@link ArrowNodeStyle} that provides an
 * {@link ArrowNodeStyleAngleHandle}, an {@link ArrowNodeStyleShaftRatioHandle} and further handles
 * provided by a delegating provider.
 * The {@link ArrowNodeStyleShaftRatioHandle} is only provided for {@link ArrowStyleShape#ARROW},
 * {@link ArrowStyleShape.DOUBLE_ARROW"/>, and {@link ArrowStyleShape.NOTCHED_ARROW"/>.
 */
public class ArrowNodeStyleHandleProvider implements IHandleProvider {
  private final INode node;
  private final IHandleProvider delegateProvider;
  private final Runnable stylePropertyChanged;

  /**
   * Creates a new instance of {@link ArrowNodeStyleHandleProvider} with the given
   * <code>stylePropertyChanged</code> action and an optional <code>delegateProvider</code> whose
   * handles are also returned.
   * @param node                 The node to provide handles for.
   * @param stylePropertyChanged The wrapped <see cref="IHandleProvider"/> implementation.
   * @param delegateProvider     An action that is called when the handle is moved.
   */
  public ArrowNodeStyleHandleProvider(INode node, Runnable stylePropertyChanged, IHandleProvider delegateProvider) {
    this.node = node;
    this.delegateProvider = delegateProvider;
    this.stylePropertyChanged = stylePropertyChanged;
  }

  @Override
  public Iterable<IHandle> getHandles(IInputModeContext context) {
    List<IHandle> result = new ArrayList<>();
    if (delegateProvider != null) {
      delegateProvider.getHandles(context).forEach(result::add);
    }
    if (node.getStyle() instanceof ArrowNodeStyle) {
      result.add(new ArrowNodeStyleAngleHandle(node, stylePropertyChanged));

      ArrowStyleShape shape = ((ArrowNodeStyle) node.getStyle()).getShape();
      if (shape == ArrowStyleShape.ARROW ||
          shape == ArrowStyleShape.DOUBLE_ARROW ||
          shape == ArrowStyleShape.NOTCHED_ARROW) {
        result.add(new ArrowNodeStyleShaftRatioHandle(node, stylePropertyChanged));
      }
    }
    return result;
  }
}