Added task level to records

3 files changed, 8 insertions, 1 deletions

diff --git a/include/litmus/sched_mc.h b/include/litmus/sched_mc.h
index 0a1b3f0c718c..740cc11be5d7 100644
--- a/include/litmus/sched_mc.h
+++ b/include/litmus/sched_mc.h
@@ -34,6 +34,7 @@ struct mc_data {
 #define tsk_mc_data(t)   (tsk_rt(t)->mc_data)
 #define tsk_mc_crit(t)   (tsk_mc_data(t)->mc_task.crit)
 #define is_ghost(t)      (tsk_mc_data(t)->mc_job.is_ghost)
+
 #define TS "(%s/%d:%d:%s)"
 #define TA(t) (t) ? tsk_mc_data(t) ? is_ghost(t) ? "ghost" : t->comm \
                                                  : t->comm : "NULL", \
diff --git a/include/litmus/sched_trace.h b/include/litmus/sched_trace.h
index 7179fc2745ad..5cc05be8332f 100644
--- a/include/litmus/sched_trace.h
+++ b/include/litmus/sched_trace.h
@@ -24,7 +24,8 @@ struct st_param_data {		/* regular params */
         u32     phase;
         u8      partition;
         u8      class;
-        u8      __unused[2];
+        u8      level;
+        u8      __unused[1];
 };
 
 struct st_release_data {        /* A job is was/is going to be released. */
diff --git a/litmus/sched_task_trace.c b/litmus/sched_task_trace.c
index 5ef8d09ab41f..768a2a0f9d2f 100644
--- a/litmus/sched_task_trace.c
+++ b/litmus/sched_task_trace.c
@@ -14,7 +14,11 @@
 #include <litmus/sched_trace.h>
 #include <litmus/feather_trace.h>
 #include <litmus/ftdev.h>
+#include <litmus/rt_domain.h>
+#include <litmus/domain.h>
+#include <litmus/event_group.h>
 
+#include <litmus/sched_mc.h>
 
 #define NO_EVENTS               (1 << CONFIG_SCHED_TASK_TRACE_SHIFT)
 
@@ -132,6 +136,7 @@ feather_callback void do_sched_trace_task_param(unsigned long id, unsigned long
                 rec->data.param.phase     = get_rt_phase(t);
                 rec->data.param.partition = get_partition(t);
                 rec->data.param.class     = get_class(t);
+                rec->data.param.level     = (tsk_mc_data(t) ? tsk_mc_crit(t) : -1);
                 put_record(rec);
         }
 }