PGM supportwip-mc2-new

1 files changed, 23 insertions, 0 deletions

diff --git a/litmus/sched_task_trace.c b/litmus/sched_task_trace.c
index 3a6756deb32f..784b18ecfd8b 100644
--- a/litmus/sched_task_trace.c
+++ b/litmus/sched_task_trace.c
@@ -238,3 +238,26 @@ feather_callback void do_sched_trace_action(unsigned long id,
                 put_record(rec);
         }
 }
+
+feather_callback void do_sched_trace_pgm_param(unsigned long id, unsigned long _task)
+{
+        struct task_struct *t = (struct task_struct*) _task;
+        struct st_event_record* rec = get_record(ST_PGM_PARAM, t);
+        if (rec) {
+                rec->data.pgm_param.node_type = tsk_rt(t)->task_params.pgm_type;
+                rec->data.pgm_param.graph_pid = t->tgid;
+                rec->data.pgm_param.expected_graph_etoe = tsk_rt(t)->task_params.pgm_expected_etoe;
+                put_record(rec);
+        }
+}
+
+feather_callback void do_sched_trace_pgm_release(unsigned long id, unsigned long _task)
+{
+        struct task_struct *t = (struct task_struct*) _task;
+        struct st_event_record* rec = get_record(ST_PGM_RELEASE, t);
+        if (rec) {
+                rec->data.pgm_release.release  = get_release(t);
+                rec->data.pgm_release.deadline = get_deadline(t);
+                put_record(rec);
+        }
+}