Import lttng.kernel.core plugins from Scope

[deliverable/tracecompass.git] / lttng / org.lttng.scope.lttng.kernel.core / src / org / lttng / scope / lttng / kernel / core / analysis / os / handlers / internal / SchedWakeupHandler.java

diff --git a/lttng/org.lttng.scope.lttng.kernel.core/src/org/lttng/scope/lttng/kernel/core/analysis/os/handlers/internal/SchedWakeupHandler.java b/lttng/org.lttng.scope.lttng.kernel.core/src/org/lttng/scope/lttng/kernel/core/analysis/os/handlers/internal/SchedWakeupHandler.java
new file mode 100644 (file)
index 0000000..d52db37
--- /dev/null
+++ b/lttng/org.lttng.scope.lttng.kernel.core/src/org/lttng/scope/lttng/kernel/core/analysis/os/handlers/internal/SchedWakeupHandler.java
@@ -0,0 +1,83 @@
+/*******************************************************************************
+ * Copyright (c) 2015 Ericsson
+ *
+ * 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
+ *
+ * Contributors:
+ *    Matthew Khouzam - Initial API and implementation
+ *******************************************************************************/
+
+package org.lttng.scope.lttng.kernel.core.analysis.os.handlers.internal;
+
+import org.eclipse.tracecompass.tmf.core.event.ITmfEvent;
+import org.lttng.scope.lttng.kernel.core.analysis.os.Attributes;
+import org.lttng.scope.lttng.kernel.core.analysis.os.StateValues;
+import org.lttng.scope.lttng.kernel.core.trace.layout.ILttngKernelEventLayout;
+
+import ca.polymtl.dorsal.libdelorean.ITmfStateSystemBuilder;
+import ca.polymtl.dorsal.libdelorean.exceptions.AttributeNotFoundException;
+import ca.polymtl.dorsal.libdelorean.statevalue.ITmfStateValue;
+import ca.polymtl.dorsal.libdelorean.statevalue.TmfStateValue;
+
+/**
+ * Waking/wakeup handler.
+ *
+ * "sched_waking" and "sched_wakeup" tracepoints contain the same fields, and
+ * apply the same state transitions in our model, so they can both use this
+ * handler.
+ */
+public class SchedWakeupHandler extends KernelEventHandler {
+
+    /**
+     * Constructor
+     * @param layout event layout
+     */
+    public SchedWakeupHandler(ILttngKernelEventLayout layout) {
+        super(layout);
+    }
+
+    @Override
+    public void handleEvent(ITmfStateSystemBuilder ss, ITmfEvent event) throws AttributeNotFoundException {
+        Integer cpu = KernelEventHandlerUtils.getCpu(event);
+        final int tid = ((Long) event.getContent().getField(getLayout().fieldTid()).getValue()).intValue();
+        final int prio = ((Long) event.getContent().getField(getLayout().fieldPrio()).getValue()).intValue();
+        Long targetCpu = event.getContent().getFieldValue(Long.class, getLayout().fieldTargetCpu());
+
+        String threadAttributeName = Attributes.buildThreadAttributeName(tid, cpu);
+        if (cpu == null || targetCpu == null || threadAttributeName == null) {
+            return;
+        }
+
+        final int threadNode = ss.getQuarkRelativeAndAdd(KernelEventHandlerUtils.getNodeThreads(ss), threadAttributeName);
+
+        /*
+         * The process indicated in the event's payload is now ready to run.
+         * Assign it to the "wait for cpu" state, but only if it was not already
+         * running.
+         */
+        int status = ss.queryOngoingState(threadNode).unboxInt();
+        ITmfStateValue value = null;
+        long timestamp = KernelEventHandlerUtils.getTimestamp(event);
+        if (status != StateValues.PROCESS_STATUS_RUN_SYSCALL &&
+                status != StateValues.PROCESS_STATUS_RUN_USERMODE) {
+            value = StateValues.PROCESS_STATUS_WAIT_FOR_CPU_VALUE;
+            ss.modifyAttribute(timestamp, value, threadNode);
+        }
+
+        /* Set the thread's target run queue */
+        int quark = ss.getQuarkRelativeAndAdd(threadNode, Attributes.CURRENT_CPU_RQ);
+        value = TmfStateValue.newValueInt(targetCpu.intValue());
+        ss.modifyAttribute(timestamp, value, quark);
+
+        /*
+         * When a user changes a threads prio (e.g. with pthread_setschedparam),
+         * it shows in ftrace with a sched_wakeup.
+         */
+        quark = ss.getQuarkRelativeAndAdd(threadNode, Attributes.PRIO);
+        value = TmfStateValue.newValueInt(prio);
+        ss.modifyAttribute(timestamp, value, quark);
+    }
+}