Properly assign cpuwip-edf-os

1 files changed, 11 insertions, 14 deletions

diff --git a/bin/rt_launch_edfos.c b/bin/rt_launch_edfos.c
index 43ebee5..2b26fb2 100644
--- a/bin/rt_launch_edfos.c
+++ b/bin/rt_launch_edfos.c
@@ -50,9 +50,8 @@ int main(int argc, char** argv)
         lt_t wcet;
         lt_t period;
         /* [num,den] */
-        lt_t fracs[NR_CPUS][2];
-        int migrate = 0;
-        int cpu = 0;
+        lt_t fracs[NR_CPUS_EDF_OS][2];
+        int cpu = -1;
         int opt;
         int verbose = 0;
         int wait = 0;
@@ -60,7 +59,7 @@ int main(int argc, char** argv)
         startup_info_t info;
         task_class_t class = RT_CLASS_HARD;
 
-        for (i = 0; i < NR_CPUS; i++) {
+        for (i = 0; i < NR_CPUS_EDF_OS; i++) {
                 fracs[i][0] = 0;
                 fracs[i][1] = 0;
         }
@@ -77,7 +76,6 @@ int main(int argc, char** argv)
                         cpu = atoi(optarg);
                         fracs[cpu][0] = 1;
                         fracs[cpu][1] = 1;
-                        migrate = 1;
                         break;
                 case 'c':
                         class = str2class(optarg);
@@ -106,9 +104,11 @@ int main(int argc, char** argv)
                 fracs[cpu][0] = atoi(argv[i]);
                 fracs[cpu][1] = atoi(argv[i+1]);
         }
+        if (cpu == -1)
+                cpu = compute_initial_cpu_edfos(fracs);
         if (wcet <= 0)
-        usage("The worst-case execution time must be a "
-              "positive number.");
+                usage("The worst-case execution time must be a "
+                      "positive number.");
         if (period <= 0)
                 usage("The period must be a positive number.");
         if (wcet > period) {
@@ -118,17 +118,14 @@ int main(int argc, char** argv)
         info.exec_path = argv[optind + 8];
         info.argv      = argv + optind + 8;
         info.wait      = wait;
-        if (migrate) {
-                ret = be_migrate_to(cpu);
-                if (ret < 0)
-                        bail_out("could not migrate to target partition");
-        }
+        ret = be_migrate_to(cpu);
+        if (ret < 0)
+                bail_out("could not migrate to target partition");
         /* create in src/task.c a new wrapper for the __launch_rt_task
          * which takes the fraction and the cpus */
-        ret = __create_rt_task_edfos(launch, &info, wcet, period, fracs,
+        ret = __create_rt_task_edfos(launch, &info, cpu, wcet, period, fracs,
                         class);
 
-
         if (ret < 0)
                 bail_out("could not create rt child process");
         else if (verbose)