WSFileDialog.java

Go to the documentation of this file.

package org.ros.model.ros.diagram;



import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;

import org.eclipse.core.resources.IResource;
import org.eclipse.core.resources.ResourcesPlugin;
import org.eclipse.core.runtime.IAdaptable;
import org.eclipse.jface.dialogs.Dialog;
import org.eclipse.jface.viewers.DecoratingLabelProvider;
import org.eclipse.jface.viewers.ISelection;
import org.eclipse.jface.viewers.IStructuredSelection;
import org.eclipse.jface.viewers.TreeViewer;
import org.eclipse.jface.viewers.Viewer;
import org.eclipse.jface.viewers.ViewerFilter;
import org.eclipse.swt.SWT;
import org.eclipse.swt.layout.GridData;
import org.eclipse.swt.widgets.Composite;
import org.eclipse.swt.widgets.Control;
import org.eclipse.swt.widgets.Shell;
import org.eclipse.ui.IWorkingSet;
import org.eclipse.ui.ResourceWorkingSetFilter;
import org.eclipse.ui.internal.ide.IDEWorkbenchPlugin;
import org.eclipse.ui.model.WorkbenchContentProvider;
import org.eclipse.ui.model.WorkbenchLabelProvider;
import org.eclipse.ui.views.navigator.ResourceNavigator;


public class WSFileDialog extends Dialog {

        private TreeViewer viewer;
        private IResource rootElement;
        private boolean expand;
        private String[] extensions;
    private FilePatternFilter patternFilter = new FilePatternFilter();
    private ResourceWorkingSetFilter workingSetFilter = new ResourceWorkingSetFilter();
        private IWorkingSet workingSet;
        private int selectionStyle;
        private IResource[] result;
        private String title;

        public WSFileDialog(Shell parentShell, int selectionStyle, String title, IResource rootElement, boolean expand, String[] extensions, IWorkingSet workingSet) {
                super(parentShell);
                setShellStyle(getShellStyle() | SWT.RESIZE);
                this.title = title;
                this.rootElement = rootElement;
                this.expand = expand;
                this.extensions = extensions;
                this.workingSet = workingSet;
                this.selectionStyle = selectionStyle;
        }
        
        public WSFileDialog(Shell parentShell, int selectionStyle, String title) {
                this(
                                parentShell,
                                selectionStyle,
                                title,
                                ResourcesPlugin.getWorkspace().getRoot(),
                                true,
                                null,
                                null);
        }

        public void setExtensions(String[] extensions) {
                this.extensions = extensions;
        }

        /* (non-Javadoc)
         * @see org.eclipse.jface.dialogs.Dialog#createDialogArea(org.eclipse.swt.widgets.Composite)
         */
        @Override
        protected Control createDialogArea(Composite parent) {
                Composite comp = (Composite) super.createDialogArea(parent);
                getShell().setText(title);
                TreeViewer viewer = createViewer(comp);
                GridData data = new GridData(GridData.FILL_BOTH);
                data.grabExcessHorizontalSpace=true;
                data.grabExcessVerticalSpace=true;
                data.heightHint = 400;
                data.widthHint = 300;
                viewer.getControl().setLayoutData(data);
                this.viewer = viewer;
                return comp;
        }
        
        protected TreeViewer createViewer(Composite parent) {
                ResourceNavigator nav = new ResourceNavigator();
                TreeViewer viewer =
                        new TreeViewer(parent, selectionStyle | SWT.H_SCROLL | SWT.V_SCROLL | SWT.BORDER);
                viewer.setUseHashlookup(true);
                initContentProvider(viewer);
                initLabelProvider(viewer);
                initFilters(viewer);
                viewer.setInput(rootElement);
                if (expand) {
                        viewer.expandToLevel(2);
                }
                return viewer;
        }
        
        protected void initFilters(TreeViewer viewer) {
                viewer.addFilter(patternFilter);
                if (workingSet != null) {
                        workingSetFilter.setWorkingSet(workingSet);
                        viewer.addFilter(workingSetFilter);
                }
        }

        protected void initContentProvider(TreeViewer viewer) {
                viewer.setContentProvider(new WorkbenchContentProvider());
        }
        
        protected void initLabelProvider(TreeViewer viewer) {
                viewer.setLabelProvider(
                        new DecoratingLabelProvider(
                                new WorkbenchLabelProvider(),
                                IDEWorkbenchPlugin.getDefault().getWorkbench().getDecoratorManager().getLabelDecorator()));
        }

        @Override
        protected void okPressed() {
                ISelection selection = viewer.getSelection();
                List data = new ArrayList();
                if (!selection.isEmpty()) {
                        if (selection instanceof IStructuredSelection) {
                                IStructuredSelection sel = (IStructuredSelection)selection;
                                for (Iterator i = sel.iterator();i.hasNext();) {
                                        Object next = i.next();
                                        IResource resource= null;                       
                                        if (next instanceof IResource)
                                                resource= (IResource)next;
                                        else if (next instanceof IAdaptable) {
                                                if (resource == null)
                                                        resource= (IResource)((IAdaptable)next).getAdapter(IResource.class);
                                        }
                                        if (resource != null) {
                                                data.add(resource);
                                        }
                                }
                        }
                }
                result = (IResource[])data.toArray(new IResource[]{});
                super.okPressed();
        }
        
        public IResource getSingleResult() {
                if (getReturnCode() == OK) {
                        return getMultiResult()[0];
                } else return null;
        }
        
        public IResource[] getMultiResult() {
                if (getReturnCode() == OK) {
                        return result;
                } else return null;
        }
        
        private class FilePatternFilter extends ViewerFilter {

                @Override
                public boolean select(Viewer viewer, Object parentElement, Object element) {
                        if (extensions == null || extensions.length == 0) return true;
                IResource resource = null;
                if (element instanceof IResource) {
                    resource = (IResource) element;
                } else if (element instanceof IAdaptable) {
                    IAdaptable adaptable = (IAdaptable) element;
                    resource = (IResource) adaptable.getAdapter(IResource.class);
                }
                if (resource != null && !resource.isDerived() && !resource.getName().equals("CVS")) {
                           if (resource.getType() != IResource.FILE) return true;
                    String extension = resource.getFileExtension();
                    if (extension == null) return true;
                    for (int i = 0; i < extensions.length;i++) {
                                if (extension.equalsIgnoreCase(extensions[i])) return true;
                    }
                }
                return false;
                }
        }
}