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
.StateSystemDisposedException
;
29 import org
.eclipse
.tracecompass
.statesystem
.core
.interval
.ITmfStateInterval
;
30 import org
.eclipse
.tracecompass
.tmf
.core
.trace
.ITmfTrace
;
31 import org
.eclipse
.tracecompass
.tmf
.core
.trace
.TmfTraceUtils
;
32 import org
.eclipse
.tracecompass
.tmf
.ui
.viewers
.tree
.AbstractTmfTreeViewer
;
33 import org
.eclipse
.tracecompass
.tmf
.ui
.viewers
.tree
.ITmfTreeColumnDataProvider
;
34 import org
.eclipse
.tracecompass
.tmf
.ui
.viewers
.tree
.ITmfTreeViewerEntry
;
35 import org
.eclipse
.tracecompass
.tmf
.ui
.viewers
.tree
.TmfTreeColumnData
;
36 import org
.eclipse
.tracecompass
.tmf
.ui
.viewers
.tree
.TmfTreeViewerEntry
;
39 * Tree viewer to select which process to display in the kernel memory usage
42 * @author Mahdi Zolnouri
43 * @author Wassim Nasrallah
44 * @author Najib Arbaoui
46 public class KernelMemoryUsageTreeViewer
extends AbstractTmfTreeViewer
{
48 private KernelMemoryAnalysisModule fModule
= null;
49 private String fSelectedThread
= null;
50 private static final String
[] COLUMN_NAMES
= new String
[] {
51 Messages
.KernelMemoryUsageComposite_ColumnTID
,
52 Messages
.KernelMemoryUsageComposite_ColumnProcess
55 /* A map that saves the mapping of a thread ID to its executable name */
56 private final Map
<String
, String
> fProcessNameMap
= new HashMap
<>();
58 /** Provides label for the Kernel memory usage tree viewer cells */
59 protected static class KernelMemoryLabelProvider
extends TreeLabelProvider
{
62 public String
getColumnText(Object element
, int columnIndex
) {
63 KernelMemoryUsageEntry obj
= (KernelMemoryUsageEntry
) element
;
64 if (columnIndex
== 0) {
66 } else if (columnIndex
== 1) {
67 return obj
.getProcessName();
69 return element
.toString();
77 * The parent composite that holds this viewer
79 public KernelMemoryUsageTreeViewer(Composite parent
) {
81 setLabelProvider(new KernelMemoryLabelProvider());
85 protected ITmfTreeColumnDataProvider
getColumnDataProvider() {
86 return new ITmfTreeColumnDataProvider() {
89 public List
<TmfTreeColumnData
> getColumnData() {
90 /* All columns are sortable */
91 List
<TmfTreeColumnData
> columns
= new ArrayList
<>();
92 TmfTreeColumnData column
= new TmfTreeColumnData(COLUMN_NAMES
[0]);
93 column
.setComparator(new ViewerComparator() {
95 public int compare(Viewer viewer
, Object e1
, Object e2
) {
96 KernelMemoryUsageEntry n1
= (KernelMemoryUsageEntry
) e1
;
97 KernelMemoryUsageEntry n2
= (KernelMemoryUsageEntry
) e2
;
99 return n1
.getTid().compareTo(n2
.getTid());
103 column
= new TmfTreeColumnData(COLUMN_NAMES
[1]);
104 column
.setComparator(new ViewerComparator() {
106 public int compare(Viewer viewer
, Object e1
, Object e2
) {
107 KernelMemoryUsageEntry n1
= (KernelMemoryUsageEntry
) e1
;
108 KernelMemoryUsageEntry n2
= (KernelMemoryUsageEntry
) e2
;
110 return n1
.getProcessName().compareTo(n2
.getProcessName());
119 // ------------------------------------------------------------------------
121 // ------------------------------------------------------------------------
124 protected void contentChanged(ITmfTreeViewerEntry rootEntry
) {
125 String selectedThread
= fSelectedThread
;
126 if (selectedThread
!= null) {
127 /* Find the selected thread among the inputs */
128 for (ITmfTreeViewerEntry entry
: rootEntry
.getChildren()) {
129 if (entry
instanceof KernelMemoryUsageEntry
) {
130 if (selectedThread
.equals(((KernelMemoryUsageEntry
) entry
).getTid())) {
131 List
<ITmfTreeViewerEntry
> list
= Collections
.singletonList(entry
);
132 super.setSelection(list
);
141 public void initializeDataSource() {
142 /* Should not be called while trace is still null */
143 ITmfTrace trace
= checkNotNull(getTrace());
145 fModule
= TmfTraceUtils
.getAnalysisModuleOfClass(trace
, KernelMemoryAnalysisModule
.class, KernelMemoryAnalysisModule
.ID
);
146 if (fModule
== null) {
150 fModule
.waitForInitialization();
151 fProcessNameMap
.clear();
155 protected ITmfTreeViewerEntry
updateElements(long start
, long end
, boolean isSelection
) {
156 if (isSelection
|| (start
== end
)) {
159 KernelMemoryAnalysisModule module
= fModule
;
160 if (getTrace() == null || module
== null) {
163 module
.waitForInitialization();
164 ITmfStateSystem ss
= module
.getStateSystem();
169 TmfTreeViewerEntry root
= new TmfTreeViewerEntry(""); //$NON-NLS-1$
170 List
<ITmfTreeViewerEntry
> entryList
= root
.getChildren();
173 List
<ITmfStateInterval
> memoryStates
= ss
.queryFullState(start
);
174 List
<Integer
> threadQuarkList
= ss
.getSubAttributes(ITmfStateSystem
.ROOT_ATTRIBUTE
, false);
176 for (Integer threadQuark
: threadQuarkList
) {
177 ITmfStateInterval threadMemoryInterval
= memoryStates
.get(threadQuark
);
178 if (threadMemoryInterval
.getEndTime() < end
) {
179 String tid
= ss
.getAttributeName(threadQuark
);
180 String procname
= getProcessName(tid
);
181 KernelMemoryUsageEntry obj
= new KernelMemoryUsageEntry(tid
, procname
);
185 } catch (StateSystemDisposedException e
) {
186 Activator
.getDefault().logError(e
.getMessage(), e
);
192 * Get the process name from its TID by using the LTTng kernel analysis
195 private String
getProcessName(String tid
) {
196 String execName
= fProcessNameMap
.get(tid
);
197 if (execName
!= null) {
200 if (tid
.equals(KernelMemoryStateProvider
.OTHER_TID
)) {
201 fProcessNameMap
.put(tid
, tid
);
204 ITmfTrace trace
= checkNotNull(getTrace());
205 KernelAnalysisModule kernelModule
= TmfTraceUtils
.getAnalysisModuleOfClass(trace
, KernelAnalysisModule
.class, KernelAnalysisModule
.ID
);
206 if (kernelModule
== null) {
209 execName
= KernelThreadInformationProvider
.getExecutableName(kernelModule
, Integer
.parseInt(tid
));
210 if (execName
== null) {
213 fProcessNameMap
.put(tid
, execName
);
218 * Set the currently selected thread ID
221 * The selected thread ID
223 public void setSelectedThread(String tid
) {
224 fSelectedThread
= tid
;