Write experimental data to result file.

Optionally, with parameter encoding.

1 files changed, 52 insertions, 13 deletions

diff --git a/bin/cache_cost.c b/bin/cache_cost.c
index dc4c0fe..195e23d 100644
--- a/bin/cache_cost.c
+++ b/bin/cache_cost.c
@@ -8,6 +8,7 @@
 #include <sched.h>
 
 #include <sys/io.h>
+#include <sys/utsname.h>
 
 #include "litmus.h"
 #include "asm.h"
@@ -125,7 +126,8 @@ static int touch_mem(int *mem, int wss, int write_cycle)
         return sum;
 }
 
-static void do_random_experiment(int num_cpus, int wss,
+static void do_random_experiment(FILE* outfile,
+                                 int num_cpus, int wss,
                                  int sleep_min, int sleep_max,
                                  int write_cycle)
 {
@@ -185,15 +187,16 @@ static void do_random_experiment(int num_cpus, int wss,
 
                 /* run, write ratio, wss, delay, from, to, cold, hot1, hot2,
                  * hot3, after_resume */
-                printf("%6ld, %3d, %6d, %6d, %3d, %3d, "
-                       "%" CYCLES_FMT ", "
-                       "%" CYCLES_FMT ", "
-                       "%" CYCLES_FMT ", "
-                       "%" CYCLES_FMT ", "
-                       "%" CYCLES_FMT "\n",
-                       counter++, write_cycle,
-                       wss, delay, last_cpu, next_cpu, cold, hot1, hot2, hot3,
-                       after_resume);
+                fprintf(outfile,
+                        "%6ld, %3d, %6d, %6d, %3d, %3d, "
+                        "%" CYCLES_FMT ", "
+                        "%" CYCLES_FMT ", "
+                        "%" CYCLES_FMT ", "
+                        "%" CYCLES_FMT ", "
+                        "%" CYCLES_FMT "\n",
+                        counter++, write_cycle,
+                        wss, delay, last_cpu, next_cpu, cold, hot1, hot2, hot3,
+                        after_resume);
                 last_cpu = next_cpu;
                 deallocate(mem);
         }
@@ -201,7 +204,7 @@ static void do_random_experiment(int num_cpus, int wss,
 
 static void on_sigalarm(int signo)
 {
-        fprintf(stderr, "SIGALARM\n");
+        /*fprintf(stderr, "SIGALARM\n");*/
         exit(0);
 }
 
@@ -211,7 +214,7 @@ static void usage(char *error) {
 }
 
 
-#define OPTSTR "m:w:l:s:"
+#define OPTSTR "m:w:l:s:o:x:y:n"
 
 int main(int argc, char** argv)
 {
@@ -221,6 +224,10 @@ int main(int argc, char** argv)
         int sleep_max = 1000;
         int exit_after = 0; /* seconds */
         int write_cycle = 0; /* every nth cycle is a write; 0 means read-only  */
+        FILE* out = stdout;
+        char fname[255];
+        struct utsname utsname;
+        int auto_name_file = 0;
         int opt;
 
         while ((opt = getopt(argc, argv, OPTSTR)) != -1) {
@@ -237,6 +244,20 @@ int main(int argc, char** argv)
                 case 'l':
                         exit_after = atoi(optarg);
                         break;
+                case 'o':
+                        out = fopen(optarg, "w");
+                        if (out == NULL)
+                                usage("could not open file");
+                        break;
+                case 'n':
+                        auto_name_file = 1;
+                        break;
+                case 'x':
+                        sleep_min = atoi(optarg);
+                        break;
+                case 'y':
+                        sleep_max = atoi(optarg);
+                        break;
                 case ':':
                         usage("Argument missing.");
                         break;
@@ -253,6 +274,9 @@ int main(int argc, char** argv)
         if (wss <= 0)
                 usage("The working set size must be positive.");
 
+        if (sleep_min < 0 || sleep_min > sleep_max)
+                usage("Invalid minimum sleep time");
+
         if (write_cycle < 0)
                 usage("Write cycle may not be negative.");
 
@@ -265,11 +289,26 @@ int main(int argc, char** argv)
         if (lock_memory() != 0)
                 die("Could not lock memory.");
 
+        if (auto_name_file) {
+                uname(&utsname);
+                snprintf(fname, 255,
+                         "pmo_host=%s_wss=%d_wcyle=%d_smin=%d_smax=%d.csv",
+                         utsname.nodename, wss, write_cycle, sleep_min, sleep_max);
+                out = fopen(fname, "w");
+                if (out == NULL) {
+                        fprintf(stderr, "Can't open %s.", fname);
+                        die("I/O");
+                }
+        }
+
         if (exit_after > 0) {
                 signal(SIGALRM, on_sigalarm);
                 alarm(exit_after);
         }
 
-        do_random_experiment(num_cpus, wss, sleep_min, sleep_max, write_cycle);
+        do_random_experiment(out,
+                             num_cpus, wss, sleep_min,
+                             sleep_max, write_cycle);
+        fclose(out);
         return 0;
 }