Improve TmfTrace test coverage

[deliverable/tracecompass.git] / org.eclipse.linuxtools.lttng2.kernel.ui / src / org / eclipse / linuxtools / internal / lttng2 / kernel / ui / views / resources / ResourcesView.java

/*******************************************************************************\r
 * Copyright (c) 2012 Ericsson\r
 * \r
 * All rights reserved. This program and the accompanying materials are\r
 * made available under the terms of the Eclipse Public License v1.0 which\r
 * accompanies this distribution, and is available at\r
 * http://www.eclipse.org/legal/epl-v10.html\r
 * \r
 * Contributors:\r
 *   Patrick Tasse - Initial API and implementation\r
 *******************************************************************************/\r
\r
package org.eclipse.linuxtools.internal.lttng2.kernel.ui.views.resources;\r
\r
import java.util.ArrayList;\r
import java.util.HashMap;\r
import java.util.Iterator;\r
import java.util.List;\r
import java.util.Map;\r
\r
import org.eclipse.linuxtools.lttng2.kernel.core.trace.Attributes;\r
import org.eclipse.linuxtools.lttng2.kernel.core.trace.CtfKernelTrace;\r
import org.eclipse.linuxtools.tmf.core.event.ITmfEvent;\r
import org.eclipse.linuxtools.tmf.core.event.TmfEvent;\r
import org.eclipse.linuxtools.tmf.core.exceptions.AttributeNotFoundException;\r
import org.eclipse.linuxtools.tmf.core.exceptions.StateValueTypeException;\r
import org.eclipse.linuxtools.tmf.core.exceptions.TimeRangeException;\r
import org.eclipse.linuxtools.tmf.core.experiment.TmfExperiment;\r
import org.eclipse.linuxtools.tmf.core.interval.ITmfStateInterval;\r
import org.eclipse.linuxtools.tmf.core.signal.TmfExperimentSelectedSignal;\r
import org.eclipse.linuxtools.tmf.core.signal.TmfSignalHandler;\r
import org.eclipse.linuxtools.tmf.core.statesystem.IStateSystemQuerier;\r
import org.eclipse.linuxtools.tmf.core.trace.ITmfTrace;\r
import org.eclipse.linuxtools.tmf.ui.views.TmfView;\r
import org.eclipse.linuxtools.tmf.ui.widgets.timegraph.TimeGraphProvider;\r
import org.eclipse.linuxtools.tmf.ui.widgets.timegraph.TimeGraphViewer;\r
import org.eclipse.linuxtools.tmf.ui.widgets.timegraph.model.ITimeEvent;\r
import org.eclipse.linuxtools.tmf.ui.widgets.timegraph.model.ITimeGraphEntry;\r
import org.eclipse.linuxtools.tmf.ui.widgets.timegraph.model.TimeEvent;\r
import org.eclipse.swt.SWT;\r
import org.eclipse.swt.widgets.Composite;\r
import org.eclipse.swt.widgets.Display;\r
\r
public class ResourcesView extends TmfView {\r
\r
    // ------------------------------------------------------------------------\r
    // Constants\r
    // ------------------------------------------------------------------------\r
\r
    /**\r
     * View ID.\r
     */\r
    public static final String ID = "org.eclipse.linuxtools.lttng2.kernel.ui.views.resources"; //$NON-NLS-1$\r
\r
    // ------------------------------------------------------------------------\r
    // Fields\r
    // ------------------------------------------------------------------------\r
\r
    // The time graph viewer\r
    TimeGraphViewer fTimeGraphViewer;\r
\r
    // The selected experiment\r
    private TmfExperiment<ITmfEvent> fSelectedExperiment;\r
\r
    // The time graph entry list\r
    private ArrayList<ITimeGraphEntry> fEntryList;\r
\r
    // The start time\r
    private long fStartTime;\r
\r
    // The end time\r
    private long fEndTime;\r
\r
    // The display width\r
    private int fDisplayWidth;\r
\r
    // ------------------------------------------------------------------------\r
    // Classes\r
    // ------------------------------------------------------------------------\r
\r
    private class GroupEntry implements ITimeGraphEntry {\r
        public ITimeGraphEntry fParent;\r
        public ArrayList<ITimeGraphEntry> fChildren;\r
        public String fName;\r
\r
        public GroupEntry(ITimeGraphEntry parent, String name) {\r
            fParent = parent;\r
            fChildren = new ArrayList<ITimeGraphEntry>();\r
            fName = name;\r
        }\r
\r
        @Override\r
        public ITimeGraphEntry getParent() {\r
            return fParent;\r
        }\r
\r
        @Override\r
        public boolean hasChildren() {\r
            return fChildren != null && fChildren.size() > 0;\r
        }\r
\r
        @Override\r
        public ITimeGraphEntry[] getChildren() {\r
            return fChildren.toArray(new ITimeGraphEntry[0]);\r
        }\r
\r
        @Override\r
        public String getName() {\r
            return fName;\r
        }\r
\r
        @Override\r
        public long getStartTime() {\r
            return -1;\r
        }\r
\r
        @Override\r
        public long getStopTime() {\r
            return -1;\r
        }\r
\r
        @Override\r
        public Iterator<ITimeEvent> getTimeEventsIterator() {\r
            return null;\r
        }\r
\r
        @Override\r
        public <T extends ITimeEvent> Iterator<T> getTimeEventsIterator(long startTime, long stopTime, long visibleDuration) {\r
            return null;\r
        }\r
\r
        public void addChild(ITimeGraphEntry entry) {\r
            fChildren.add(entry);\r
        }\r
    }\r
\r
    // ------------------------------------------------------------------------\r
    // Constructors\r
    // ------------------------------------------------------------------------\r
\r
    public ResourcesView() {\r
        super(ID);\r
        fDisplayWidth = Display.getDefault().getBounds().width;\r
    }\r
\r
    // ------------------------------------------------------------------------\r
    // ViewPart\r
    // ------------------------------------------------------------------------\r
\r
    /* (non-Javadoc)\r
     * @see org.eclipse.linuxtools.tmf.ui.views.TmfView#createPartControl(org.eclipse.swt.widgets.Composite)\r
     */\r
    @Override\r
    public void createPartControl(Composite parent) {\r
        fTimeGraphViewer = new TimeGraphViewer(parent, SWT.NONE);\r
\r
        fTimeGraphViewer.setTimeGraphProvider(new TimeGraphProvider() {\r
            @Override\r
            public String getTraceClassName(ITimeGraphEntry trace) {\r
                return "trace class"; //$NON-NLS-1$\r
            }\r
            \r
            @Override\r
            public String getStateName(StateColor color) {\r
                return "state name"; //$NON-NLS-1$\r
            }\r
            \r
            @Override\r
            public String getEventName(ITimeEvent event, boolean upper, boolean extInfo) {\r
                return "event name"; //$NON-NLS-1$\r
            }\r
            \r
            @Override\r
            public Map<String, String> getEventHoverToolTipInfo(ITimeEvent event) {\r
                return new HashMap<String, String>();\r
            }\r
            \r
            @Override\r
            public StateColor getEventColor(ITimeEvent event) {\r
                if (event.getTime() % 2 == 0) {\r
                    return StateColor.BLACK;\r
                } else {\r
                    return StateColor.GRAY;\r
                }\r
            }\r
        });\r
\r
        final Thread thread = new Thread() {\r
            @Override\r
            public void run() {\r
                if (TmfExperiment.getCurrentExperiment() != null) {\r
                    selectExperiment(TmfExperiment.getCurrentExperiment());\r
                }\r
            }\r
        };\r
        thread.start();\r
    }\r
\r
    /* (non-Javadoc)\r
     * @see org.eclipse.ui.part.WorkbenchPart#setFocus()\r
     */\r
    @Override\r
    public void setFocus() {\r
        fTimeGraphViewer.setFocus();\r
    }\r
\r
    // ------------------------------------------------------------------------\r
    // Signal handlers\r
    // ------------------------------------------------------------------------\r
\r
    @TmfSignalHandler\r
    public void experimentSelected(final TmfExperimentSelectedSignal<? extends TmfEvent> signal) {\r
        if (signal.getExperiment().equals(fSelectedExperiment)) {\r
            return;\r
        }\r
\r
        final Thread thread = new Thread() {\r
            @Override\r
            public void run() {\r
                selectExperiment(signal.getExperiment());\r
            }};\r
        thread.start();\r
    }\r
\r
    @SuppressWarnings("unchecked")\r
    private void selectExperiment(TmfExperiment<?> experiment) {\r
        fStartTime = Long.MAX_VALUE;\r
        fEndTime = Long.MIN_VALUE;\r
        fSelectedExperiment = (TmfExperiment<ITmfEvent>) experiment;\r
        fEntryList = new ArrayList<ITimeGraphEntry>();\r
        for (ITmfTrace<?> trace : experiment.getTraces()) {\r
            GroupEntry groupEntry = new GroupEntry(null, trace.getPath());\r
            fEntryList.add(groupEntry);\r
            refresh();\r
            if (trace instanceof CtfKernelTrace) {\r
                CtfKernelTrace ctfKernelTrace = (CtfKernelTrace) trace;\r
                IStateSystemQuerier ssq = ctfKernelTrace.getStateSystem();\r
                long start = ssq.getStartTime();\r
                long end = ssq.getCurrentEndTime();\r
                fStartTime = Math.min(fStartTime, start);\r
                fEndTime = Math.max(fEndTime, end);\r
                List<Integer> cpuQuarks = ssq.getQuarks(Attributes.CPUS, "*"); //$NON-NLS-1$\r
                for (int cpuQuark : cpuQuarks) {\r
                    String cpuName = "CPU " + ssq.getAttributeName(cpuQuark);\r
                    ResourcesEntry entry = new ResourcesEntry(groupEntry, ctfKernelTrace, cpuName);\r
                    try {\r
                        int currentThreadQuark = ssq.getQuarkRelative(cpuQuark, Attributes.CURRENT_THREAD);\r
                        long resolution = (end - start) / fDisplayWidth;\r
                        List<ITmfStateInterval> currentThreadIntervals = ssq.queryHistoryRange(currentThreadQuark, start, end, resolution);\r
                        for (ITmfStateInterval currentThreadInterval : currentThreadIntervals) {\r
                            if (!currentThreadInterval.getStateValue().isNull() && currentThreadInterval.getStateValue().getType() == 0) {\r
                                int currentThread = currentThreadInterval.getStateValue().unboxInt();\r
                                long startTime = currentThreadInterval.getStartTime();\r
                                long endTime = currentThreadInterval.getEndTime();\r
                                entry.addTraceEvent(new TimeEvent(entry, startTime, endTime - startTime));\r
                            }\r
                        }\r
                    } catch (AttributeNotFoundException e) {\r
                        e.printStackTrace();\r
                    } catch (TimeRangeException e) {\r
                        e.printStackTrace();\r
                    } catch (StateValueTypeException e) {\r
                        e.printStackTrace();\r
                    }\r
                    groupEntry.addChild(entry);\r
                    refresh();\r
                }\r
            }\r
        }\r
    }\r
\r
    private void refresh() {\r
        Display.getDefault().asyncExec(new Runnable() {\r
            @Override\r
            public void run() {\r
                if (fTimeGraphViewer.getControl().isDisposed()) {\r
                    return;\r
                }\r
                fTimeGraphViewer.setInput(fEntryList.toArray(new ITimeGraphEntry[0]));\r
                fTimeGraphViewer.setTimeBounds(fStartTime, fEndTime);\r
                fTimeGraphViewer.setStartFinishTime(fStartTime, fEndTime);\r
            }\r
        });\r
    }\r
\r
}\r