From fdbe1dbd2d06e0712ac041963e2bd223bed96a72 Mon Sep 17 00:00:00 2001
From: David Anderson <davea@ssl.berkeley.edu>
Date: Mon, 25 Jun 2007 23:56:03 +0000
Subject: [PATCH] - added hr_info.*

svn path=/trunk/boinc/; revision=13023
---
 checkin_notes   |   6 ++
 sched/hr_info.C | 150 ++++++++++++++++++++++++++++++++++++++++++++++++
 sched/hr_info.h |  25 ++++++++
 3 files changed, 181 insertions(+)
 create mode 100644 sched/hr_info.C
 create mode 100644 sched/hr_info.h

diff --git a/checkin_notes b/checkin_notes
index c37aeedb2b..edf55fe18f 100755
--- a/checkin_notes
+++ b/checkin_notes
@@ -6708,3 +6708,9 @@ David  25 June 2007
     clientgui/
         ViewWork.cpp
         ViewWorkGrid.cpp
+
+David  25 June 2007
+    - added hr_info.*
+
+    sched/
+        hr_info.C,h
diff --git a/sched/hr_info.C b/sched/hr_info.C
new file mode 100644
index 0000000000..8237134077
--- /dev/null
+++ b/sched/hr_info.C
@@ -0,0 +1,150 @@
+#include <stdio.h>
+#include <malloc.h>
+
+#include "error_numbers.h"
+#include "sched_msgs.h"
+
+#include "hr_info.h"
+
+int HR_INFO::write_file() {
+    int i, j;
+
+    FILE* f = fopen(HR_INFO_FILENAME, "w");
+    if (!f) return ERR_FOPEN;
+    for (i=1; i<HR_NTYPES; i++) {
+        fprintf(f, "--------- %s ----------\n", hr_names[i]);
+        for (j=0; j<hr_nclasses[i]; j++) {
+            fprintf(f, "%d %f\n", j, rac_per_class[i][j]);
+        }
+    }
+    fclose(f);
+    return 0;
+}
+
+int HR_INFO::read_file() {
+    char buf[256];
+    FILE* f = fopen(HR_INFO_FILENAME, "r");
+    if (!f) return ERR_FOPEN;
+    int i, j, jj;
+    double x;
+
+    for (i=1; i<HR_NTYPES; i++) {
+        char* p = fgets(buf, sizeof(buf), f);
+        if (!p) {
+            fprintf(stderr, "missing delimeter line in HR info");
+            exit(1);
+        }
+        for (j=0; j<hr_nclasses[i]; j++) {
+            char* p = fgets(buf, sizeof(buf), f);
+            if (!p) {
+                fprintf(stderr, "missing data line in HR info");
+                exit(1);
+            }
+            int n = sscanf(buf, "%d %lf", &jj, &x);
+            if (n!=2 || j!=jj) {
+                fprintf(stderr, "bad line %s in HR info: %d; %d != %d\n", buf, n, j, jj);
+                exit(1);
+            }
+            rac_per_class[i][j] = x;
+        }
+    }
+    fclose(f);
+    return 0;
+}
+
+void HR_INFO::init() {
+    int i;
+    for (i=1; i<HR_NTYPES; i++) {
+        rac_per_class[i] = (double*) calloc(hr_nclasses[i], sizeof(double));
+        max_slots[i] = (int*) calloc(hr_nclasses[i], sizeof(int));
+        cur_slots[i] = (int*) calloc(hr_nclasses[i], sizeof(int));
+    }
+}
+
+void HR_INFO::scan_db() {
+    DB_HOST host;
+    int retval;
+    int i;
+
+    while (1) {
+        retval = host.enumerate("where expavg_credit>1");
+        if (retval) break;
+        printf("host %d: %s | %s | %s\n", host.id, host.os_name, host.p_vendor, host.p_model);
+        for (i=1; i<HR_NTYPES; i++) {
+            if (hr_unknown_platform_type(host, i)) {
+                printf("type %d: unknown\n", i);
+                continue;
+            }
+            int hrc = hr_class(host, i);
+            printf("type %d: class %d\n", i, hrc);
+            if (!hrc) continue;
+            rac_per_class[i][hrc] += host.expavg_credit;
+        }
+    }
+    if (retval != ERR_DB_NOT_FOUND) {
+        fprintf(stderr, "host enum: %d", retval);
+        exit(1);
+    }
+}
+
+void HR_INFO::allocate(int total_slots) {
+    int i, j;
+
+    double weight_total = 0;
+    for (i=1; i<HR_NTYPES; i++) {
+        weight_total += type_weights[i];
+    }
+
+    // decide how many slots to allocate to each HR type
+    //
+    for (i=1; i<HR_NTYPES; i++) {
+        slots_per_type[i] = (int)(total_slots*type_weights[i]/weight_total);
+    }
+
+    // within each type, decide how many slots to allocate to each class
+    //
+    for (i=1; i<HR_NTYPES; i++) {
+        int nuncom = slots_per_type[i]/2;
+        int ncom = slots_per_type[i] - nuncom;
+        log_messages.printf(SCHED_MSG_LOG::MSG_DEBUG,
+            "nu %d nc %d spt %d\n", nuncom, ncom, slots_per_type[i]
+        );
+
+        double total_rac = 0;
+        for (j=1; j<hr_nclasses[i]; j++) {
+            total_rac += rac_per_class[i][j];
+        }
+        max_slots[i][0] = nuncom;
+        for (j=1; j<hr_nclasses[i]; j++) {
+            int n = (int)(ncom*rac_per_class[i][j]/total_rac);
+
+            // every HR class has a max of at least one,
+            // so that new classes can get "on the board".
+            // This means that the sum of the maxes can exceed the # of slots
+            //
+            if (n == 0) n = 1;
+            max_slots[i][j] = n;
+            if (n > 1) {
+                log_messages.printf(SCHED_MSG_LOG::MSG_DEBUG,
+                    "type %d class %d: allocating %d slots\n", i, j, n
+                );
+            }
+        }
+        for (j=0; j<hr_nclasses[i]; j++) {
+            cur_slots[i][j] = 0;
+        }
+    }
+}
+
+// Decide if job of the given HR type and class should be added to array,
+// and if to update counts
+//
+bool HR_INFO::accept(int hrt, int hrc) {
+    if (cur_slots[hrt][hrc] >= max_slots[hrt][hrc]) {
+        log_messages.printf(SCHED_MSG_LOG::MSG_DEBUG,
+            "skipping job because HR class (%d, %d) full\n", hrt, hrc
+        );
+        return false;
+    }
+    cur_slots[hrt][hrc]++;
+}
diff --git a/sched/hr_info.h b/sched/hr_info.h
new file mode 100644
index 0000000000..13717aa010
--- /dev/null
+++ b/sched/hr_info.h
@@ -0,0 +1,25 @@
+#include "hr.h"
+
+struct HR_INFO {
+    double *rac_per_class[HR_NTYPES];
+        // how much RAC per class
+    int *max_slots[HR_NTYPES];
+        // max # of work array slots per class
+    int *cur_slots[HR_NTYPES];
+        // estimate of current # of slots used for per class
+    double type_weights[HR_NTYPES];
+        // total app weight per type
+    int slots_per_type[HR_NTYPES];
+        // # of slots per type (fixed at start)
+    bool type_being_used[HR_NTYPES];
+        // whether any app is actually using this HR type
+    int write_file();
+    int read_file();
+    void scan_db();
+    void allot();
+    void init();
+    void allocate(int);
+    bool accept(int, int);
+};
+
+#define HR_INFO_FILENAME "../hr_info.txt"