tests/org.eclipse.ui.tests/Eclipse UI Tests/org/eclipse/ui/tests/dnd/ViewDragSource.java

/*******************************************************************************
 * Copyright (c) 2004 IBM Corporation and others.
 * All rights reserved. This program and the accompanying materials 
 * are made available under the terms of the Common Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/cpl-v10.html
 * 
 * Contributors:
 *     IBM Corporation - initial API and implementation
 *******************************************************************************/
package org.eclipse.ui.tests.dnd;

import org.eclipse.swt.widgets.Display;
import org.eclipse.ui.IViewPart;
import org.eclipse.ui.internal.PartPane;
import org.eclipse.ui.internal.ViewStack;
import org.eclipse.ui.internal.ViewSite;
import org.eclipse.ui.internal.WorkbenchPage;
import org.eclipse.ui.internal.WorkbenchPlugin;
import org.eclipse.ui.internal.dnd.DragUtil;
import org.eclipse.ui.internal.dnd.TestDropLocation;
import org.eclipse.ui.internal.registry.IViewDescriptor;
import org.eclipse.ui.presentations.IPresentablePart;

/**
 * @since 3.0
 */
public class ViewDragSource extends TestDragSource {

	String targetPart;
	boolean wholeFolder;
	boolean maximized = false;
	
	public ViewDragSource(String part, boolean dragWholeFolder) {
		this(part, dragWholeFolder, false);
	}
	
	public ViewDragSource(String part, boolean dragWholeFolder, boolean maximized) {
		this.maximized = maximized;
		this.targetPart = part;
		
		wholeFolder = dragWholeFolder;
	}
	
	IViewPart getPart() {
		return getPage().findView(targetPart);
	}
	
	/* (non-Javadoc)
	 * @see org.eclipse.ui.tests.dnd.TestDragSource#getName()
	 */
	public String toString() {
		IViewDescriptor desc = WorkbenchPlugin.getDefault().getViewRegistry().find(targetPart);
		String title = desc.getLabel();
		
		if (wholeFolder) {
			title = title + " folder";
		}
		
		if (maximized) {
			title = "maximized " + title;
		}
		
		return title;
	}

	/* (non-Javadoc)
	 * @see org.eclipse.ui.tests.dnd.TestDragSource#drag(org.eclipse.swt.graphics.Point)
	 */
	public void drag(TestDropLocation target) {
		IViewPart part = getPart();
		
		WorkbenchPage page = getPage();
		PartPane pane = ((ViewSite)part.getSite()).getPane();
		if (maximized) {
			page.toggleZoom(pane.getPartReference());
		}
		
		DragUtil.forceDropLocation(target);
		ViewStack parent = ((ViewStack)(pane.getContainer()));
		
		IPresentablePart presentablePart = wholeFolder ? null : pane.getPresentablePart(); 
		parent.dragStart(presentablePart, Display.getDefault().getCursorLocation(), false);
		
		DragUtil.forceDropLocation(null);
	}
	
}