| /********************************************************************** |
| * Copyright (c) 2016, 2017 Polytechnique de Montréal |
| * |
| * 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 |
| **********************************************************************/ |
| package org.eclipse.tracecompass.internal.analysis.os.linux.ui.views.memory; |
| |
| import java.util.Comparator; |
| |
| import org.eclipse.jdt.annotation.Nullable; |
| import org.eclipse.swt.graphics.Image; |
| import org.eclipse.swt.widgets.Composite; |
| import org.eclipse.tracecompass.analysis.os.linux.core.memory.MemoryUsageTreeModel; |
| import org.eclipse.tracecompass.internal.provisional.tmf.core.model.filters.FilterTimeQueryFilter; |
| import org.eclipse.tracecompass.internal.provisional.tmf.core.model.filters.TimeQueryFilter; |
| import org.eclipse.tracecompass.tmf.ui.viewers.tree.AbstractSelectTreeViewer; |
| import org.eclipse.tracecompass.tmf.ui.viewers.tree.ITmfTreeColumnDataProvider; |
| import org.eclipse.tracecompass.tmf.ui.viewers.tree.TmfGenericTreeEntry; |
| import org.eclipse.tracecompass.tmf.ui.viewers.tree.TmfTreeColumnData; |
| |
| import com.google.common.collect.ImmutableList; |
| |
| /** |
| * Tree viewer to select which process to display in the kernel memory usage |
| * chart. |
| * |
| * @since 2.2 |
| * @author Mahdi Zolnouri |
| * @author Wassim Nasrallah |
| * @author Najib Arbaoui |
| */ |
| public class MemoryUsageTreeViewer extends AbstractSelectTreeViewer { |
| |
| /** Provides label for the Kernel memory usage tree viewer cells */ |
| private class MemoryLabelProvider extends TreeLabelProvider { |
| |
| @Override |
| public String getColumnText(Object element, int columnIndex) { |
| if (!(element instanceof TmfGenericTreeEntry)) { |
| return null; |
| } |
| TmfGenericTreeEntry<MemoryUsageTreeModel> obj = (TmfGenericTreeEntry<MemoryUsageTreeModel>) element; |
| if (columnIndex == 0) { |
| return obj.getName(); |
| } else if (columnIndex == 1) { |
| int tid = obj.getModel().getTid(); |
| if (tid < 0) { |
| return Messages.MemoryUsageTree_Total; |
| } |
| return Integer.toString(tid); |
| } |
| return null; |
| } |
| |
| @Override |
| public Image getColumnImage(Object element, int columnIndex) { |
| if (columnIndex == 2 && element instanceof TmfGenericTreeEntry) { |
| TmfGenericTreeEntry<MemoryUsageTreeModel> entry = (TmfGenericTreeEntry<MemoryUsageTreeModel>) element; |
| int tid = entry.getModel().getTid(); |
| if (entry.getParent() instanceof TmfGenericTreeEntry && isChecked(element)) { |
| return getLegendImage(entry.getParent().getName() + ':' + tid); |
| } else if (tid < 0) { |
| return getLegendImage(entry.getName() + MemoryUsageTreeModel.TOTAL_SUFFIX); |
| } |
| } |
| return null; |
| } |
| } |
| |
| // view is filtered by default |
| private boolean fFiltered = true; |
| |
| /** |
| * Constructor |
| * |
| * @param parent |
| * The parent composite that holds this viewer |
| * @param id |
| * The memory usage data provider ID. |
| */ |
| public MemoryUsageTreeViewer(Composite parent, String id) { |
| super(parent, 2, id); |
| setLabelProvider(new MemoryLabelProvider()); |
| } |
| |
| @Override |
| protected @Nullable TimeQueryFilter getFilter(long start, long end, boolean isSelection) { |
| return new FilterTimeQueryFilter(start, end, 2, fFiltered); |
| } |
| |
| @Override |
| protected ITmfTreeColumnDataProvider getColumnDataProvider() { |
| return () -> { |
| Comparator<TmfGenericTreeEntry<MemoryUsageTreeModel>> compareTid = Comparator.comparingInt(c -> c.getModel().getTid()); |
| return ImmutableList.of( |
| createColumn(Messages.MemoryUsageTree_ColumnProcess, Comparator.comparing(TmfGenericTreeEntry::getName)), |
| createColumn(Messages.MemoryUsageTree_ColumnTID, compareTid), |
| new TmfTreeColumnData(Messages.MemoryUsageTree_Legend)); |
| }; |
| } |
| |
| /** |
| * Set the view to filter active threads or not. |
| * |
| * @param isFiltered |
| * if we filter the active threads or not. |
| */ |
| public void setFiltered(boolean isFiltered) { |
| fFiltered = isFiltered; |
| updateContent(getWindowStartTime(), getWindowEndTime(), false); |
| } |
| } |