forked from bioclipse/bioclipse.cheminformatics
/
PickMoleculeDialog.java
128 lines (110 loc) · 4.45 KB
/
PickMoleculeDialog.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
/*******************************************************************************
* Copyright (c) 2009 Jonathan Alvarsson <jonalv@users.sourceforge.net>
*
* 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
* www.eclipse.orgÑepl-v10.html <http://www.eclipse.org/legal/epl-v10.html>
*
* Contact: http://www.bioclipse.net/
******************************************************************************/
package net.bioclipse.chemoinformatics.dialogs;
import java.util.ArrayList;
import net.bioclipse.chemoinformatics.util.MoleculeContentTypeViewerFilter;
import net.bioclipse.chemoinformatics.util.SingleMoleculeContentTypeViewerFilter;
import org.eclipse.core.resources.IFile;
import org.eclipse.core.resources.ResourcesPlugin;
import org.eclipse.jface.dialogs.TitleAreaDialog;
import org.eclipse.jface.viewers.ISelectionChangedListener;
import org.eclipse.jface.viewers.IStructuredSelection;
import org.eclipse.jface.viewers.SelectionChangedEvent;
import org.eclipse.jface.viewers.TreeViewer;
import org.eclipse.swt.SWT;
import org.eclipse.swt.layout.GridData;
import org.eclipse.swt.layout.GridLayout;
import org.eclipse.swt.widgets.Composite;
import org.eclipse.swt.widgets.Control;
import org.eclipse.swt.widgets.Shell;
import org.eclipse.ui.model.WorkbenchContentProvider;
import org.eclipse.ui.model.WorkbenchLabelProvider;
/**
* A dialog for selecting a molecule somewhere in the workspace.
*
* @author jonalv
*
*/
public class PickMoleculeDialog extends TitleAreaDialog
implements ISelectionChangedListener {
private TreeViewer treeViewer;
private String title;
private String message;
private ArrayList<IFile> selectedFiles;
/**
* @param parentShell
* @wbp.parser.constructor
*/
public PickMoleculeDialog( Shell parentShell ) {
super( parentShell );
}
public PickMoleculeDialog( Shell parentShell,
String title,
String message ) {
super( parentShell );
this.title = title;
this.message = message;
}
@Override
protected Control createDialogArea(Composite parent) {
Composite dialogArea = (Composite) super.createDialogArea( parent );
GridLayout layout = new GridLayout();
dialogArea.setLayout( layout );
treeViewer = createViewer( dialogArea );
GridData data = new GridData( GridData.FILL_BOTH );
data.grabExcessHorizontalSpace = true;
data.grabExcessVerticalSpace = true;
data.heightHint = 400;
data.widthHint = 300;
treeViewer.getControl().setLayoutData( data );
treeViewer.addSelectionChangedListener( this );
setTitle( title == null ? "Select molecule"
: title);
setMessage( message == null ? "Select a molecule from your workspace"
: message );
return dialogArea;
}
protected TreeViewer createViewer(Composite parent) {
TreeViewer viewer = new TreeViewer( parent, SWT.SINGLE
| SWT.H_SCROLL
| SWT.V_SCROLL
| SWT.BORDER );
viewer.setUseHashlookup( true );
viewer.setContentProvider( new WorkbenchContentProvider() );
viewer.setLabelProvider( new WorkbenchLabelProvider() );
viewer.addFilter( new SingleMoleculeContentTypeViewerFilter() );
viewer.setInput( ResourcesPlugin.getWorkspace().getRoot() );
viewer.expandToLevel(2);
return viewer;
}
/**
* React on treeviewer changes to be able to return the selected resources
*/
public void selectionChanged( SelectionChangedEvent event ) {
IStructuredSelection selection
= (IStructuredSelection) event.getSelection();
selectedFiles = new ArrayList<IFile>();
for ( Object obj : selection.toList() ) {
if ( obj instanceof IFile ) {
IFile file = (IFile) obj;
selectedFiles.add( file );
}
}
}
public IFile getSelectedFile() {
assert selectedFiles.size() == 1;
return selectedFiles.get( 0 );
}
@Override
protected boolean isResizable() {
return true;
}
}