[padb-devel] [padb] r165 committed - Re-instate RMS support, it's un-tested by me but is tested by...

[codesite-noreply at google.com]

Revision: 165
Author: apittman
Date: Tue Sep  1 13:06:21 2009
Log: Re-instate RMS support, it's un-tested by me but is tested by
somebody with access to a RMS machine.

http://code.google.com/p/padb/source/detail?r=165

Modified:
  /trunk/src/padb

=======================================
--- /trunk/src/padb	Wed Aug 26 05:04:08 2009
+++ /trunk/src/padb	Tue Sep  1 13:06:21 2009
@@ -2124,6 +2124,8 @@

      my $ncpus = rms_job_to_ncpus($job);

+    my $nhosts = rms_job_to_nhosts($job);
+
      if ( $res eq "" ) {
          printf("Job '$job' doesn't have a associated resource\n");
          return undef;
@@ -2153,7 +2155,7 @@

      my $cmd = "prun -i /dev/null -T $res";

-    return ( $cmd, $ncpus );
+    return ( $cmd, $ncpus, $nhosts );
  }

  # Not exported...
@@ -2186,6 +2188,55 @@

      return $ncpus;
  }
+
+sub rms_job_to_nhosts {
+    my $job      = shift;
+    my $nodeSpec = `rmsquery "select hostnames from jobs where  
name=\'$job\'"`;
+
+    chomp $nodeSpec;
+    my $i;
+    my @nodeList;
+    my $prefix;
+    my $suffix;
+
+    # deal with multiple entries
+    foreach ( split( " ", $nodeSpec ) ) {
+        if (m/([^\[]+)\[([0-9-,]+)\]([^\[]*)/) {
+            $prefix = $1;
+            $suffix = $3;
+
+            foreach ( split( ",", $2 ) ) {
+                if ( !m/([0-9]+)-?([0-9]+)?/ ) {
+                    print "malformed nodespec '$_'\n";
+                    exit(1);
+                }
+
+                if ( defined($2) ) {
+
+                    # square braces with range, eg 		    'machine[0-3]'
+                    for ( $i = $1 ; $i <= $2 ; $i++ ) {
+                        push( @nodeList, $prefix . $i . $suffix );
+                    }
+                } else {
+
+                    # no range, just suffix
+                    push( @nodeList, $prefix . $1 . $suffix );
+                }
+            }
+        } else {
+
+            # no square braces, just node name, eg 'machine0'
+            if ( !m/([^\[]+)([0-9]+)([^\[]*)/ ) {
+                print "malformed nodespec '$_'\n";
+                exit(1);
+            }
+
+            push( @nodeList, $1 . $2 . $3 );
+        }
+    }
+
+    return $#nodeList + 1;
+}

  sub rms_res_to_partition {
      my $res  = shift;