1 /*******************************************************************************
2 * Copyright (c) 2009, 2010, 2011 Ericsson
4 * All rights reserved. This program and the accompanying materials are
5 * made available under the terms of the Eclipse Public License v1.0 which
6 * accompanies this distribution, and is available at
7 * http://www.eclipse.org/legal/epl-v10.html
10 * Francois Chouinard - Initial API and implementation
11 *******************************************************************************/
13 package org
.eclipse
.linuxtools
.tmf
.ui
.project
.wizards
;
15 import java
.util
.HashMap
;
18 import java
.util
.Vector
;
20 import org
.eclipse
.core
.resources
.IFile
;
21 import org
.eclipse
.core
.resources
.IFolder
;
22 import org
.eclipse
.core
.resources
.IResource
;
23 import org
.eclipse
.core
.resources
.IWorkspace
;
24 import org
.eclipse
.core
.resources
.ResourcesPlugin
;
25 import org
.eclipse
.core
.runtime
.CoreException
;
26 import org
.eclipse
.core
.runtime
.IPath
;
27 import org
.eclipse
.core
.runtime
.QualifiedName
;
28 import org
.eclipse
.jface
.viewers
.CheckboxTableViewer
;
29 import org
.eclipse
.jface
.wizard
.WizardPage
;
30 import org
.eclipse
.linuxtools
.internal
.tmf
.ui
.TmfUiPlugin
;
31 import org
.eclipse
.linuxtools
.tmf
.core
.TmfCommonConstants
;
32 import org
.eclipse
.linuxtools
.tmf
.ui
.project
.model
.ITmfProjectModelElement
;
33 import org
.eclipse
.linuxtools
.tmf
.ui
.project
.model
.TmfExperimentElement
;
34 import org
.eclipse
.linuxtools
.tmf
.ui
.project
.model
.TmfProjectElement
;
35 import org
.eclipse
.linuxtools
.tmf
.ui
.project
.model
.TmfTraceElement
;
36 import org
.eclipse
.linuxtools
.tmf
.ui
.project
.model
.TmfTraceFolder
;
37 import org
.eclipse
.linuxtools
.tmf
.ui
.project
.model
.TraceFolderContentProvider
;
38 import org
.eclipse
.linuxtools
.tmf
.ui
.project
.model
.TraceFolderLabelProvider
;
39 import org
.eclipse
.swt
.SWT
;
40 import org
.eclipse
.swt
.layout
.FormAttachment
;
41 import org
.eclipse
.swt
.layout
.FormData
;
42 import org
.eclipse
.swt
.layout
.FormLayout
;
43 import org
.eclipse
.swt
.widgets
.Composite
;
44 import org
.eclipse
.swt
.widgets
.Table
;
45 import org
.eclipse
.swt
.widgets
.TableColumn
;
48 * <b><u>SelectTracesWizardPage</u></b>
51 public class SelectTracesWizardPage
extends WizardPage
{
53 // ------------------------------------------------------------------------
55 // ------------------------------------------------------------------------
57 private final TmfProjectElement fProject
;
58 private final TmfExperimentElement fExperiment
;
59 private Map
<String
, TmfTraceElement
> fPreviousTraces
;
60 private CheckboxTableViewer fCheckboxTableViewer
;
62 // ------------------------------------------------------------------------
64 // ------------------------------------------------------------------------
66 protected SelectTracesWizardPage(TmfProjectElement project
, TmfExperimentElement experiment
) {
67 super(""); //$NON-NLS-1$
68 setTitle(Messages
.SelectTracesWizardPage_WindowTitle
);
69 setDescription(Messages
.SelectTracesWizardPage_Description
);
71 fExperiment
= experiment
;
74 // ------------------------------------------------------------------------
76 // ------------------------------------------------------------------------
79 public void createControl(Composite parent
) {
80 Composite container
= new Composite(parent
, SWT
.NULL
);
81 container
.setLayout(new FormLayout());
82 setControl(container
);
84 fCheckboxTableViewer
= CheckboxTableViewer
.newCheckList(container
, SWT
.BORDER
);
85 fCheckboxTableViewer
.setContentProvider(new TraceFolderContentProvider());
86 fCheckboxTableViewer
.setLabelProvider(new TraceFolderLabelProvider());
88 final Table table
= fCheckboxTableViewer
.getTable();
89 final FormData formData
= new FormData();
90 formData
.bottom
= new FormAttachment(100, 0);
91 formData
.right
= new FormAttachment(100, 0);
92 formData
.top
= new FormAttachment(0, 0);
93 formData
.left
= new FormAttachment(0, 0);
94 table
.setLayoutData(formData
);
95 table
.setHeaderVisible(true);
97 final TableColumn tableColumn
= new TableColumn(table
, SWT
.NONE
);
98 tableColumn
.setWidth(200);
99 tableColumn
.setText(Messages
.SelectTracesWizardPage_TraceColumnHeader
);
101 // Get the list of traces already part of the experiment
102 fPreviousTraces
= new HashMap
<String
, TmfTraceElement
>();
103 for (ITmfProjectModelElement child
: fExperiment
.getChildren()) {
104 if (child
instanceof TmfTraceElement
) {
105 TmfTraceElement trace
= (TmfTraceElement
) child
;
106 String name
= trace
.getResource().getName();
107 fPreviousTraces
.put(name
, trace
);
111 // Populate the list of traces to choose from
112 Set
<String
> keys
= fPreviousTraces
.keySet();
113 TmfTraceFolder traceFolder
= fProject
.getTracesFolder();
114 fCheckboxTableViewer
.setInput(traceFolder
);
116 // Set the checkbox for the traces already included
118 Object element
= fCheckboxTableViewer
.getElementAt(index
++);
119 while (element
!= null) {
120 if (element
instanceof TmfTraceElement
) {
121 TmfTraceElement trace
= (TmfTraceElement
) element
;
122 if (keys
.contains(trace
.getResource().getName())) {
123 fCheckboxTableViewer
.setChecked(element
, true);
126 element
= fCheckboxTableViewer
.getElementAt(index
++);
130 public boolean performFinish() {
132 IFolder experiment
= fExperiment
.getResource();
134 // Add the selected traces to the experiment
135 Set
<String
> keys
= fPreviousTraces
.keySet();
136 TmfTraceElement
[] traces
= getSelection();
137 for (TmfTraceElement trace
: traces
) {
138 String name
= trace
.getResource().getName();
139 if (keys
.contains(name
)) {
140 fPreviousTraces
.remove(name
);
142 IResource resource
= trace
.getResource();
143 IPath location
= resource
.getLocation();
144 createLink(experiment
, trace
, resource
, location
);
148 // Remove traces that were unchecked (thus left in fPreviousTraces)
149 keys
= fPreviousTraces
.keySet();
150 for (String key
: keys
) {
151 fExperiment
.removeChild(fPreviousTraces
.get(key
));
152 IResource resource
= experiment
.findMember(key
);
154 resource
.delete(true, null);
155 } catch (CoreException e
) {
156 TmfUiPlugin
.getDefault().logError("Error selecting traces for experiment " + experiment
.getName(), e
); //$NON-NLS-1$
165 * Create a link to the actual trace and set the trace type
167 private void createLink(IFolder experiment
, TmfTraceElement trace
, IResource resource
, IPath location
) {
168 IWorkspace workspace
= ResourcesPlugin
.getWorkspace();
170 Map
<QualifiedName
, String
> properties
= trace
.getResource().getPersistentProperties();
171 String bundleName
= properties
.get(TmfCommonConstants
.TRACEBUNDLE
);
172 String traceType
= properties
.get(TmfCommonConstants
.TRACETYPE
);
173 String iconUrl
= properties
.get(TmfCommonConstants
.TRACEICON
);
175 if (resource
instanceof IFolder
) {
176 IFolder folder
= experiment
.getFolder(trace
.getName());
177 if (workspace
.validateLinkLocation(folder
, location
).isOK()) {
178 folder
.createLink(location
, IResource
.REPLACE
, null);
179 setProperties(folder
, bundleName
, traceType
, iconUrl
);
182 TmfUiPlugin
.getDefault().logError("Error creating link. Invalid trace location " + location
); //$NON-NLS-1$
185 IFile file
= experiment
.getFile(trace
.getName());
186 if (workspace
.validateLinkLocation(file
, location
).isOK()) {
187 file
.createLink(location
, IResource
.REPLACE
, null);
188 setProperties(file
, bundleName
, traceType
, iconUrl
);
190 TmfUiPlugin
.getDefault().logError("Error creating link. Invalid trace location " + location
); //$NON-NLS-1$
193 } catch (CoreException e
) {
194 TmfUiPlugin
.getDefault().logError("Error creating link to location " + location
, e
); //$NON-NLS-1$
198 private void setProperties(IResource resource
, String bundleName
, String traceType
, String iconUrl
) throws CoreException
{
199 resource
.setPersistentProperty(TmfCommonConstants
.TRACEBUNDLE
, bundleName
);
200 resource
.setPersistentProperty(TmfCommonConstants
.TRACETYPE
, traceType
);
201 resource
.setPersistentProperty(TmfCommonConstants
.TRACEICON
, iconUrl
);
205 * Get the list of selected traces
207 private TmfTraceElement
[] getSelection() {
208 Vector
<TmfTraceElement
> traces
= new Vector
<TmfTraceElement
>();
209 Object
[] selection
= fCheckboxTableViewer
.getCheckedElements();
210 for (Object sel
: selection
) {
211 if (sel
instanceof TmfTraceElement
)
212 traces
.add((TmfTraceElement
) sel
);
214 TmfTraceElement
[] result
= new TmfTraceElement
[traces
.size()];
215 traces
.toArray(result
);