ERDiagramPropertySourceProvider.java

package org.insightech.er.editor.view.property_source;

import org.eclipse.ui.views.properties.IPropertySource;
import org.eclipse.ui.views.properties.IPropertySourceProvider;
import org.insightech.er.editor.controller.editpart.element.ERDiagramEditPart;
import org.insightech.er.editor.controller.editpart.element.node.ERTableEditPart;
import org.insightech.er.editor.model.ERDiagram;
import org.insightech.er.editor.model.diagram_contents.element.node.table.ERTable;

public class ERDiagramPropertySourceProvider implements IPropertySourceProvider {

	public IPropertySource getPropertySource(Object object) {
		if (object instanceof ERDiagramEditPart) {
			ERDiagram diagram = (ERDiagram) ((ERDiagramEditPart) object)
					.getModel();
			return new ERDiagramPropertySource(diagram);

		} else if (object instanceof ERTableEditPart) {
			ERTable table = (ERTable) ((ERTableEditPart) object).getModel();
			return new ERTablePropertySource(table);

		}

		return null;
	}
}