LineWidthConnectionEndpointEditPolicy.java

/*******************************************************************************
 * MontiCore Language Workbench
 * Copyright (c) 2015, 2016, MontiCore, All rights reserved.
 *  
 * This project is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 3.0 of the License, or (at your option) any later version.
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this project. If not, see <http://www.gnu.org/licenses/>.
 *******************************************************************************/
package de.monticore.genericgraphics.controller.editpolicies.connections;

import org.eclipse.draw2d.PolylineConnection;
import org.eclipse.gef.GraphicalEditPart;
import org.eclipse.gef.editpolicies.ConnectionEndpointEditPolicy;

/**
 * <p>
 * An extension of {@link ConnectionEndpointEditPolicy}, adding some
 * functionality for highlighting selected connections.
 * </p>
 * <p>
 * When a connection is selected it's line is drawn with line width of 2. When
 * not selected anymore, the highlight is removed.
 * </p>
 * 
 * @author Tim Enger
 */
public class LineWidthConnectionEndpointEditPolicy extends ConnectionEndpointEditPolicy {
  
  private int lineWidth = 0;
  
  @Override
  protected void addSelectionHandles() {
    super.addSelectionHandles();
    lineWidth = getConnectionFigure().getLineWidth();
    getConnectionFigure().setLineWidth(lineWidth + 2);
  }
  
  @Override
  protected void removeSelectionHandles() {
    super.removeSelectionHandles();
    getConnectionFigure().setLineWidth(lineWidth);
  }
  
  protected PolylineConnection getConnectionFigure() {
    return ((PolylineConnection) ((GraphicalEditPart) getHost()).getFigure());
  }
}