Changeset 20859 for branches/eam_branch_20081124/Ohana/src/opihi/cmd.data/gaussj.c

Timestamp:

Nov 30, 2008, 11:13:15 AM (18 years ago)

Author:

eugene

Message:

allow functions to use FLT or INT vectors as appropriate; document valid types

File:

• branches/eam_branch_20081124/Ohana/src/opihi/cmd.data/gaussj.c (modified) (3 diffs)

branches/eam_branch_20081124/Ohana/src/opihi/cmd.data/gaussj.c

@@ -4 +4 @@
 
   float *m;
-  opihi_flt *v;
+  opihi_flt *vf;
+  opihi_int *vi;
   double **a, **b;
   int i, j, N, status, QUIET;
@@ -32 +33 @@
   }
 
+  isFloat = (B[0].type == OPIHI_FLT);
+  vf = B[0].elements.Flt;
+  vi = B[0].elements.Int;
+
   m = (float *) A[0].matrix.buffer;
-  v = B[0].elements.Flt;
   for (i = 0; i < N; i++) {
     for (j = 0; j < N; j++) {
       a[i][j] = m[i+j*N];
     }
-    b[i][0] = v[i]; 
+    b[i][0] = isFloat ? vf[i] : vi[i]; 
   }
 
   status = dgaussjordan (a, b, N, 1);
+
+  // output vector needs to be float, so re-cast it
+  ResetVector (B, OPIHI_FLT, N);
+  vf = B[0].elements.Flt;
 
   for (i = 0; i < N; i++) {
@@ -47 +55 @@
        m[i+j*N] = a[i][j];
     }
-    v[i] = b[i][0]; 
+    vf[i] = b[i][0]; 
   }