com.archimatetool.editor.views.navigator.NavigatorViewer.java Source code

Introduction

Here is the source code for com.archimatetool.editor.views.navigator.NavigatorViewer.java

Source

/**
 * This program and the accompanying materials
 * are made available under the terms of the License
 * which accompanies this distribution in the file LICENSE.txt
 */
package com.archimatetool.editor.views.navigator;

import org.eclipse.emf.ecore.EObject;
import org.eclipse.jface.viewers.ITreeContentProvider;
import org.eclipse.jface.viewers.LabelProvider;
import org.eclipse.jface.viewers.TreeViewer;
import org.eclipse.jface.viewers.Viewer;
import org.eclipse.jface.viewers.ViewerSorter;
import org.eclipse.swt.SWT;
import org.eclipse.swt.graphics.Image;
import org.eclipse.swt.widgets.Composite;

import com.archimatetool.editor.ui.ArchimateLabelProvider;
import com.archimatetool.model.IArchimateElement;
import com.archimatetool.model.IRelationship;
import com.archimatetool.model.util.ArchimateModelUtils;

/**
 * Tree Viewer for Navigator View
 * 
 * @author Phillip Beauvoir
 */
public class NavigatorViewer extends TreeViewer {

    private boolean fShowTargetElements = true;

    public NavigatorViewer(Composite parent, int style) {
        super(parent, style | SWT.MULTI);

        setContentProvider(new NavigatorViewerContentProvider());
        setLabelProvider(new NavigatorViewerLabelProvider());
        setAutoExpandLevel(3);

        setSorter(new ViewerSorter());
    }

    public Object getActualInput() {
        return getActualInput(getInput());
    }

    public Object getActualInput(Object input) {
        if (input instanceof Object[] && ((Object[]) input).length == 1) {
            input = ((Object[]) input)[0];
        }
        return input;
    }

    public void setShowTargetElements(boolean set) {
        if (fShowTargetElements != set) {
            fShowTargetElements = set;
            refresh();
        }
    }

    /**
     *  Content Provider
     */
    private class NavigatorViewerContentProvider implements ITreeContentProvider {

        public void inputChanged(Viewer v, Object oldInput, Object newInput) {
        }

        public void dispose() {
        }

        public Object[] getElements(Object parent) {
            if (parent instanceof Object[]) {
                // Check if it was deleted
                Object input = getActualInput(parent);
                if (input instanceof EObject && ((EObject) input).eContainer() == null) {
                    return new Object[0];
                }

                return (Object[]) parent;
            }
            return new Object[0];
        }

        public Object[] getChildren(Object parent) {
            if (parent instanceof IRelationship) {
                IRelationship relation = (IRelationship) parent;
                if (fShowTargetElements) {
                    return new Object[] { relation.getTarget() };
                } else {
                    return new Object[] { relation.getSource() };
                }
            } else if (parent instanceof IArchimateElement) {
                IArchimateElement element = (IArchimateElement) parent;
                if (fShowTargetElements) {
                    return ArchimateModelUtils.getSourceRelationships(element).toArray();
                } else {
                    return ArchimateModelUtils.getTargetRelationships(element).toArray();
                }
            }

            return new Object[0];
        }

        public Object getParent(Object element) {
            return null;
        }

        public boolean hasChildren(Object element) {
            return getChildren(element).length > 0;
        }

    }

    /**
     * Label Provider
     */
    private class NavigatorViewerLabelProvider extends LabelProvider {

        @Override
        public String getText(Object element) {
            return ArchimateLabelProvider.INSTANCE.getLabel(element);
        }

        @Override
        public Image getImage(Object element) {
            return ArchimateLabelProvider.INSTANCE.getImage(element);
        }
    }

}