1 /**********************************************************************
2 * Copyright (c) 2016 Polytechnique de Montréal
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
8 **********************************************************************/
9 package org
.eclipse
.tracecompass
.internal
.analysis
.os
.linux
.ui
.views
.kernelmemoryusage
;
11 import static org
.eclipse
.tracecompass
.common
.core
.NonNullUtils
.checkNotNull
;
13 import java
.util
.ArrayList
;
14 import java
.util
.Collections
;
15 import java
.util
.HashMap
;
16 import java
.util
.List
;
19 import org
.eclipse
.jface
.viewers
.Viewer
;
20 import org
.eclipse
.jface
.viewers
.ViewerComparator
;
21 import org
.eclipse
.swt
.widgets
.Composite
;
22 import org
.eclipse
.tracecompass
.analysis
.os
.linux
.core
.kernel
.KernelAnalysisModule
;
23 import org
.eclipse
.tracecompass
.analysis
.os
.linux
.core
.kernel
.KernelThreadInformationProvider
;
24 import org
.eclipse
.tracecompass
.analysis
.os
.linux
.core
.kernelmemoryusage
.KernelMemoryAnalysisModule
;
25 import org
.eclipse
.tracecompass
.analysis
.os
.linux
.core
.kernelmemoryusage
.KernelMemoryStateProvider
;
26 import org
.eclipse
.tracecompass
.internal
.analysis
.os
.linux
.ui
.Activator
;
27 import org
.eclipse
.tracecompass
.statesystem
.core
.ITmfStateSystem
;
28 import org
.eclipse
.tracecompass
.statesystem
.core
.exceptions
.AttributeNotFoundException
;
29 import org
.eclipse
.tracecompass
.statesystem
.core
.exceptions
.StateSystemDisposedException
;
30 import org
.eclipse
.tracecompass
.statesystem
.core
.interval
.ITmfStateInterval
;
31 import org
.eclipse
.tracecompass
.tmf
.core
.trace
.ITmfTrace
;
32 import org
.eclipse
.tracecompass
.tmf
.core
.trace
.TmfTraceUtils
;
33 import org
.eclipse
.tracecompass
.tmf
.ui
.viewers
.tree
.AbstractTmfTreeViewer
;
34 import org
.eclipse
.tracecompass
.tmf
.ui
.viewers
.tree
.ITmfTreeColumnDataProvider
;
35 import org
.eclipse
.tracecompass
.tmf
.ui
.viewers
.tree
.ITmfTreeViewerEntry
;
36 import org
.eclipse
.tracecompass
.tmf
.ui
.viewers
.tree
.TmfTreeColumnData
;
37 import org
.eclipse
.tracecompass
.tmf
.ui
.viewers
.tree
.TmfTreeViewerEntry
;
40 * Tree viewer to select which process to display in the kernel memory usage
43 * @author Mahdi Zolnouri
44 * @author Wassim Nasrallah
45 * @author Najib Arbaoui
47 public class KernelMemoryUsageTreeViewer
extends AbstractTmfTreeViewer
{
49 private KernelMemoryAnalysisModule fModule
= null;
50 private String fSelectedThread
= null;
51 private static final String
[] COLUMN_NAMES
= new String
[] {
52 Messages
.KernelMemoryUsageComposite_ColumnTID
,
53 Messages
.KernelMemoryUsageComposite_ColumnProcess
56 /* A map that saves the mapping of a thread ID to its executable name */
57 private final Map
<String
, String
> fProcessNameMap
= new HashMap
<>();
59 /** Provides label for the Kernel memory usage tree viewer cells */
60 protected static class KernelMemoryLabelProvider
extends TreeLabelProvider
{
63 public String
getColumnText(Object element
, int columnIndex
) {
64 KernelMemoryUsageEntry obj
= (KernelMemoryUsageEntry
) element
;
65 if (columnIndex
== 0) {
67 } else if (columnIndex
== 1) {
68 return obj
.getProcessName();
70 return element
.toString();
78 * The parent composite that holds this viewer
80 public KernelMemoryUsageTreeViewer(Composite parent
) {
82 setLabelProvider(new KernelMemoryLabelProvider());
86 protected ITmfTreeColumnDataProvider
getColumnDataProvider() {
87 return new ITmfTreeColumnDataProvider() {
90 public List
<TmfTreeColumnData
> getColumnData() {
91 /* All columns are sortable */
92 List
<TmfTreeColumnData
> columns
= new ArrayList
<>();
93 TmfTreeColumnData column
= new TmfTreeColumnData(COLUMN_NAMES
[0]);
94 column
.setComparator(new ViewerComparator() {
96 public int compare(Viewer viewer
, Object e1
, Object e2
) {
97 KernelMemoryUsageEntry n1
= (KernelMemoryUsageEntry
) e1
;
98 KernelMemoryUsageEntry n2
= (KernelMemoryUsageEntry
) e2
;
100 return n1
.getTid().compareTo(n2
.getTid());
104 column
= new TmfTreeColumnData(COLUMN_NAMES
[1]);
105 column
.setComparator(new ViewerComparator() {
107 public int compare(Viewer viewer
, Object e1
, Object e2
) {
108 KernelMemoryUsageEntry n1
= (KernelMemoryUsageEntry
) e1
;
109 KernelMemoryUsageEntry n2
= (KernelMemoryUsageEntry
) e2
;
111 return n1
.getProcessName().compareTo(n2
.getProcessName());
120 // ------------------------------------------------------------------------
122 // ------------------------------------------------------------------------
125 protected void contentChanged(ITmfTreeViewerEntry rootEntry
) {
126 String selectedThread
= fSelectedThread
;
127 if (selectedThread
!= null) {
128 /* Find the selected thread among the inputs */
129 for (ITmfTreeViewerEntry entry
: rootEntry
.getChildren()) {
130 if (entry
instanceof KernelMemoryUsageEntry
) {
131 if (selectedThread
.equals(((KernelMemoryUsageEntry
) entry
).getTid())) {
132 List
<ITmfTreeViewerEntry
> list
= Collections
.singletonList(entry
);
133 super.setSelection(list
);
142 public void initializeDataSource() {
143 /* Should not be called while trace is still null */
144 ITmfTrace trace
= checkNotNull(getTrace());
146 fModule
= TmfTraceUtils
.getAnalysisModuleOfClass(trace
, KernelMemoryAnalysisModule
.class, KernelMemoryAnalysisModule
.ID
);
147 if (fModule
== null) {
151 fModule
.waitForInitialization();
152 fProcessNameMap
.clear();
156 protected ITmfTreeViewerEntry
updateElements(long start
, long end
, boolean isSelection
) {
157 if (isSelection
|| (start
== end
)) {
160 KernelMemoryAnalysisModule module
= fModule
;
161 if (getTrace() == null || module
== null) {
164 module
.waitForInitialization();
165 ITmfStateSystem ss
= module
.getStateSystem();
170 TmfTreeViewerEntry root
= new TmfTreeViewerEntry(""); //$NON-NLS-1$
171 List
<ITmfTreeViewerEntry
> entryList
= root
.getChildren();
174 List
<ITmfStateInterval
> memoryStates
= ss
.queryFullState(start
);
175 List
<Integer
> threadQuarkList
= ss
.getSubAttributes(ITmfStateSystem
.ROOT_ATTRIBUTE
, false);
177 for (Integer threadQuark
: threadQuarkList
) {
178 ITmfStateInterval threadMemoryInterval
= memoryStates
.get(threadQuark
);
179 if (threadMemoryInterval
.getEndTime() < end
) {
180 String tid
= ss
.getAttributeName(threadQuark
);
181 String procname
= getProcessName(tid
);
182 KernelMemoryUsageEntry obj
= new KernelMemoryUsageEntry(tid
, procname
);
186 } catch (StateSystemDisposedException
| AttributeNotFoundException e
) {
187 Activator
.getDefault().logError(e
.getMessage(), e
);
193 * Get the process name from its TID by using the LTTng kernel analysis
196 private String
getProcessName(String tid
) {
197 String execName
= fProcessNameMap
.get(tid
);
198 if (execName
!= null) {
201 if (tid
.equals(KernelMemoryStateProvider
.OTHER_TID
)) {
202 fProcessNameMap
.put(tid
, tid
);
205 ITmfTrace trace
= checkNotNull(getTrace());
206 KernelAnalysisModule kernelModule
= TmfTraceUtils
.getAnalysisModuleOfClass(trace
, KernelAnalysisModule
.class, KernelAnalysisModule
.ID
);
207 if (kernelModule
== null) {
210 execName
= KernelThreadInformationProvider
.getExecutableName(kernelModule
, Integer
.parseInt(tid
));
211 if (execName
== null) {
214 fProcessNameMap
.put(tid
, execName
);
219 * Set the currently selected thread ID
222 * The selected thread ID
224 public void setSelectedThread(String tid
) {
225 fSelectedThread
= tid
;