org.eclipse.nebula.widgets.nattable.core/src/org/eclipse/nebula/widgets/nattable/selection/event/SelectionLayerStructuralChangeEventHandler.java

/*******************************************************************************
 * Copyright (c) 2012, 2013 Original authors and others.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 * 
 * Contributors:
 *     Original authors and others - initial API and implementation
 ******************************************************************************/
package org.eclipse.nebula.widgets.nattable.selection.event;

import java.util.Collection;

import org.eclipse.swt.graphics.Rectangle;

import org.eclipse.nebula.widgets.nattable.coordinate.Range;
import org.eclipse.nebula.widgets.nattable.coordinate.RangeList;
import org.eclipse.nebula.widgets.nattable.layer.event.ILayerEventHandler;
import org.eclipse.nebula.widgets.nattable.layer.event.IStructuralChangeEvent;
import org.eclipse.nebula.widgets.nattable.layer.event.StructuralDiff;
import org.eclipse.nebula.widgets.nattable.layer.event.StructuralDiff.DiffTypeEnum;
import org.eclipse.nebula.widgets.nattable.selection.ISelectionModel;
import org.eclipse.nebula.widgets.nattable.selection.SelectionLayer;


public class SelectionLayerStructuralChangeEventHandler implements ILayerEventHandler<IStructuralChangeEvent> {

	private ISelectionModel selectionModel;
	private final SelectionLayer selectionLayer;
	
	public SelectionLayerStructuralChangeEventHandler(SelectionLayer selectionLayer, ISelectionModel selectionModel) {
		this.selectionLayer = selectionLayer;
		this.selectionModel = selectionModel;
	}

	@Override
	public Class<IStructuralChangeEvent> getLayerEventClass() {
		return IStructuralChangeEvent.class;
	}

	@Override
	public void handleLayerEvent(IStructuralChangeEvent event) {
		if (event.isHorizontalStructureChanged()) {
			// TODO handle column deletion
		}
		
		if (event.isVerticalStructureChanged()) {
			//if there are no row diffs, it seems to be a complete refresh
			if (event.getRowDiffs() == null) {
				Collection<Rectangle> rectangles = event.getChangedPositionRectangles();
				for (Rectangle rectangle : rectangles) {
					Range changedRange = new Range(rectangle.y, rectangle.y + rectangle.height);
					if (selectedRowModified(changedRange)) {
						selectionLayer.clear();
						break;
					}
				}
			}
			else {
				//there are row diffs so we try to determine the diffs to process
				for (StructuralDiff diff : event.getRowDiffs()) {
					//DiffTypeEnum.CHANGE is used for resizing and shouldn't result in clearing the selection
					if (diff.getDiffType() != DiffTypeEnum.CHANGE) {
						if (selectedRowModified(diff.getBeforePositionRange())) {
							selectionLayer.clear();
							break;
						}
					}
				}
			}
		}
	}
	
	private boolean selectedRowModified(Range changedRange){
		final RangeList selectedRows = selectionModel.getSelectedRowPositions();
		for (final Range rowRange : selectedRows) {
			if (rowRange.overlap(changedRange)){
				return true;
			}
		}
		return false;
	}

	public void setSelectionModel(ISelectionModel selectionModel) {
		this.selectionModel = selectionModel;
	}

}