add lsf.get_host_load_from_batch()

Author: liubo

examples/disp_gpu.py

@@ -6,7 +6,7 @@
    host_names = None
    num_hosts = lsf.new_intp()
    lsf.intp_assign(num_hosts, 0)
-   host_data = lsf.lsb_hostinfo_ex(host_names, num_hosts, "", 4096)
+   host_data = lsf.lsb_hostinfo_ex(host_names, num_hosts, "", 0)
    all_host_data = lsf.hostInfoEntArray_frompointer(host_data)
    for i in range(0, lsf.intp_value(num_hosts)):
      hostname = all_host_data[i].host

pythonlsf/lsf.i

@@ -416,6 +416,24 @@ PyObject * get_host_load(char *resreq, int index) {
     return result;
 }
 
+PyObject * get_host_load_from_batch (
+                               char * resreq, 
+                                int   index) {
+
+                   int   numhosts = 0; 
+    struct hostInfoEnt * hinfo    = NULL;
+
+    hinfo = lsb_hostinfo_ex (NULL, &numhosts, resreq, 0);
+    if (hinfo == NULL || numhosts > 1) { 
+        lsb_perror("lsb_hostinfo_ex"); 
+        exit(-1); 
+    }
+
+    PyObject *result = PyFloat_FromDouble(hinfo[0].load[index]);
+    return result;
+}
+
+
 PyObject * ls_load_py(char *resreq, int *numhosts, int options, char *fromhost) {
 
     struct hostLoad * hosts = NULL;