Add sporadic_task_ns_semi() helper function for semi-part algos.

2 files changed, 16 insertions, 3 deletions

diff --git a/include/litmus.h b/include/litmus.h
index 7382936..fd40e87 100644
--- a/include/litmus.h
+++ b/include/litmus.h
@@ -9,6 +9,7 @@ extern "C" {
  * This is required for the rt_param
  * and control_page structures.
  */
+#include <linux/threads.h>
 #include <litmus/rt_param.h>
 
 #include <sys/types.h>
@@ -50,6 +51,9 @@ int sporadic_task_ns_npsf(
                 int cpu, task_class_t cls, int npsf_id,
                 budget_policy_t budget_policy, int set_cpu_set);
 
+/* times are in ns, specific helper for semi-partitioned algos */
+int sporadic_task_ns_semi(struct rt_task *rt);
+
 /* budget enforcement off by default in these macros */
 #define sporadic_global(e, p) \
         sporadic_task(e, p, 0, 0, RT_CLASS_SOFT, NO_ENFORCEMENT, 0)
diff --git a/src/litmus.c b/src/litmus.c
index b0d7a16..c745a2c 100644
--- a/src/litmus.c
+++ b/src/litmus.c
@@ -29,8 +29,6 @@ task_class_t str2class(const char* str)
                 return -1;
 }
 
-#define NS_PER_MS 1000000
-
 /* only for best-effort execution: migrate to target_cpu */
 int be_migrate_to(int target_cpu)
 {
@@ -45,7 +43,7 @@ int sporadic_task(lt_t e, lt_t p, lt_t phase,
                   int cpu, task_class_t cls,
                   budget_policy_t budget_policy, int set_cpu_set)
 {
-        return sporadic_task_ns(e * NS_PER_MS, p * NS_PER_MS, phase * NS_PER_MS,
+        return sporadic_task_ns(e * __NS_PER_MS, p * __NS_PER_MS, phase * __NS_PER_MS,
                                 cpu, cls, budget_policy, set_cpu_set);
 }
 
@@ -128,6 +126,17 @@ int sporadic_task_ns_npsf(lt_t e, lt_t p, lt_t phase,
         return set_rt_task_param(gettid(), &param);
 }
 
+/* Sporadic task helper function for Semi-Partitioned algorithms. */
+int sporadic_task_ns_semi(struct rt_task *param)
+{
+        int ret;
+
+        ret = be_migrate_to(param->cpu);
+        check("migrate to cpu");
+
+        return set_rt_task_param(gettid(), param);
+}
+
 int init_kernel_iface(void);
 
 int init_litmus(void)