| /******************************************************************************* |
| * Copyright (c) 2006 Sybase, Inc. 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: |
| * Sybase, Inc. - initial API and implementation |
| *******************************************************************************/ |
| package org.eclipse.jst.pagedesigner.dnd.internal; |
| |
| import java.util.List; |
| |
| import org.eclipse.gef.EditPart; |
| import org.eclipse.gef.EditPartViewer; |
| import org.eclipse.gef.dnd.TemplateTransferDragSourceListener; |
| import org.eclipse.jst.pagedesigner.editors.palette.TagToolPaletteEntry; |
| |
| /** |
| * @author mengbo |
| */ |
| public class DesignerTemplateTransferDragSourceListener extends |
| TemplateTransferDragSourceListener { |
| |
| /** |
| * @param viewer |
| */ |
| public DesignerTemplateTransferDragSourceListener(EditPartViewer viewer) { |
| super(viewer); |
| } |
| |
| protected Object getTemplate() { |
| Object object = super.getTemplate(); |
| if (object == null) { |
| List selection = getViewer().getSelectedEditParts(); |
| if (selection.size() == 1) { |
| EditPart editpart = (EditPart) getViewer() |
| .getSelectedEditParts().get(0); |
| Object model = editpart.getModel(); |
| if (model instanceof TagToolPaletteEntry) { |
| return model; |
| } |
| } |
| } |
| return null; |
| } |
| } |