net.bioclipse.cdk.ui.wizards.SelectFilesWizardPage.java Source code

Introduction

Here is the source code for net.bioclipse.cdk.ui.wizards.SelectFilesWizardPage.java

Source

/*
 * Copyright (C) 2005 Bioclipse Project
 *
 * 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:
 *     Stefan Kuhn - Implementation of new sdf wizard
 */
package net.bioclipse.cdk.ui.wizards;

import java.io.IOException;

import net.bioclipse.chemoinformatics.contentlabelproviders.MoleculeFileContentProvider;
import net.bioclipse.chemoinformatics.util.ChemoinformaticUtils;
import net.bioclipse.core.util.LogUtils;

import org.apache.log4j.Logger;
import org.eclipse.core.resources.IContainer;
import org.eclipse.core.resources.IFile;
import org.eclipse.core.resources.ResourcesPlugin;
import org.eclipse.core.runtime.CoreException;
import org.eclipse.jface.viewers.ISelection;
import org.eclipse.jface.viewers.ISelectionChangedListener;
import org.eclipse.jface.viewers.IStructuredSelection;
import org.eclipse.jface.viewers.SelectionChangedEvent;
import org.eclipse.jface.viewers.StructuredSelection;
import org.eclipse.jface.viewers.TreeViewer;
import org.eclipse.jface.viewers.ViewerFilter;
import org.eclipse.jface.wizard.WizardPage;
import org.eclipse.swt.SWT;
import org.eclipse.swt.events.SelectionEvent;
import org.eclipse.swt.events.SelectionListener;
import org.eclipse.swt.layout.GridData;
import org.eclipse.swt.layout.GridLayout;
import org.eclipse.swt.widgets.Button;
import org.eclipse.swt.widgets.Composite;
import org.eclipse.swt.widgets.Label;
import org.eclipse.ui.model.WorkbenchLabelProvider;

public class SelectFilesWizardPage extends WizardPage {
    private boolean withCheckbox;
    private IStructuredSelection selectedFiles = null;
    private Button includeRecursivlyButton;
    private static final Logger logger = Logger.getLogger(SelectFilesWizardPage.class);
    private ISelection sel;

    protected SelectFilesWizardPage(boolean withCheckbox, ISelection sel) {
        super("New SD File");
        this.withCheckbox = withCheckbox;
        setTitle("Select files to include");
        setDescription("All structures you select will be added to the sd file.");
        this.sel = sel;
    }

    public void createControl(Composite parent) {
        Composite container = new Composite(parent, SWT.NULL);
        GridLayout layout = new GridLayout();
        container.setLayout(layout);
        layout.numColumns = 2;
        layout.verticalSpacing = 9;
        final TreeViewer treeViewer = new TreeViewer(container);
        if (withCheckbox) {
            includeRecursivlyButton = new Button(container, SWT.CHECK);
            includeRecursivlyButton.addSelectionListener(new SelectionListener() {

                public void widgetDefaultSelected(SelectionEvent e) {
                    this.widgetSelected(e);
                }

                public void widgetSelected(SelectionEvent e) {
                    SelectFilesWizardPage.this.setPageComplete(false);
                    if (includeRecursivlyButton.getSelection()) {
                        ISelection sel = treeViewer.getSelection();
                        if (sel instanceof IStructuredSelection) {
                            selectedFiles = (IStructuredSelection) sel;
                            try {
                                if (containsMolecule(selectedFiles) || (includeRecursivlyButton.getSelection()
                                        && containsFolder(selectedFiles)))
                                    SelectFilesWizardPage.this.setPageComplete(true);
                            } catch (Exception e1) {
                                LogUtils.handleException(e1, logger);
                            }
                        }
                    }
                }
            });
            Label includeRecursivlyLabel = new Label(container, SWT.NULL);
            includeRecursivlyLabel.setText("Include directories recursivly");
        }
        treeViewer.setContentProvider(new MoleculeFileContentProvider());
        treeViewer.setLabelProvider(WorkbenchLabelProvider.getDecoratingWorkbenchLabelProvider());
        treeViewer.setUseHashlookup(true);
        ViewerFilter[] filters = new ViewerFilter[1];
        filters[0] = new DotViewerFilter();
        treeViewer.setFilters(filters);
        //Layout the tree viewer below the text field
        GridData layoutData = new GridData();
        layoutData.grabExcessHorizontalSpace = true;
        layoutData.grabExcessVerticalSpace = true;
        layoutData.horizontalAlignment = GridData.FILL;
        layoutData.verticalAlignment = GridData.FILL;
        layoutData.horizontalSpan = 3;
        treeViewer.getControl().setLayoutData(layoutData);
        treeViewer.setInput(ResourcesPlugin.getWorkspace().getRoot().findMember("."));
        treeViewer.expandToLevel(2);
        treeViewer.addSelectionChangedListener(new ISelectionChangedListener() {
            public void selectionChanged(SelectionChangedEvent event) {
                SelectFilesWizardPage.this.setPageComplete(false);
                ISelection sel = event.getSelection();
                if (sel instanceof IStructuredSelection) {
                    selectedFiles = (IStructuredSelection) sel;
                    try {
                        if (containsMolecule(selectedFiles)
                                || (includeRecursivlyButton.getSelection() && containsFolder(selectedFiles)))
                            SelectFilesWizardPage.this.setPageComplete(true);
                    } catch (Exception e) {
                        LogUtils.handleException(e, logger);
                    }
                }
            }
        });
        treeViewer.setSelection(sel, true);
        if (sel != null && !sel.isEmpty())
            setPageComplete(true);
        else
            setPageComplete(false);
        setControl(container);
    }

    public IStructuredSelection getSelectedRes() {
        return this.selectedFiles;
    }

    public boolean doRecursive() {
        return includeRecursivlyButton.getSelection();
    }

    private boolean containsMolecule(IStructuredSelection selectedFiles) throws CoreException, IOException {
        if (selectedFiles != null) {
            for (int i = 0; i < selectedFiles.toArray().length; i++) {
                if (selectedFiles.toArray()[i] instanceof IFile) {
                    if (ChemoinformaticUtils.isMolecule((IFile) selectedFiles.toArray()[i]))
                        return true;
                }
            }
        }
        return false;
    }

    protected boolean containsFolder(IStructuredSelection selectedFiles2) {
        if (selectedFiles != null) {
            for (int i = 0; i < selectedFiles.toArray().length; i++) {
                if (selectedFiles.toArray()[i] instanceof IContainer) {
                    return true;
                }
            }
        }
        return false;
    }
}