/*******************************************************************************
* This file is protected by Copyright.
* Please refer to the COPYRIGHT file distributed with this source distribution.
*
* This file is part of REDHAWK IDE.
*
* 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
*******************************************************************************/
package gov.redhawk.ide.graphiti.ui.palette;
import org.eclipse.gef.EditPart;
import org.eclipse.gef.palette.PaletteEntry;
import org.eclipse.gef.ui.palette.PaletteEditPartFactory;
import org.eclipse.gef.ui.palette.PaletteViewer;
/**
*
*/
public class RHGraphitiPaletteEditPartFactory extends PaletteEditPartFactory {
private PaletteViewer viewer;
public RHGraphitiPaletteEditPartFactory(PaletteViewer theViewer) {
viewer = theViewer;
}
@Override
public EditPart createEditPart(EditPart parentEditPart, Object model) {
if (model instanceof PaletteNamespaceFolder) {
return createNamespaceFolderEditPart(parentEditPart, model);
}
if ("paletteFilter".equals(((PaletteEntry) model).getType())) {
return createPaletteFilterEditPart(parentEditPart, model);
}
return super.createEditPart(parentEditPart, model);
}
protected EditPart createNamespaceFolderEditPart(EditPart parentEditPart, Object model) {
return new PaletteNamespaceFolderEditPart((PaletteNamespaceFolder) model);
}
protected EditPart createPaletteFilterEditPart(EditPart parentEditPart, Object model) {
return new RHGraphitiPaletteFilterEditPart(viewer);
}
}