Changeset 20859 for branches/eam_branch_20081124/Ohana/src/opihi/cmd.data/histogram.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/histogram.c (modified) (2 diffs)

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

@@ -4 +4 @@
   
   int i, bin, Nbins;
+  opihi_int *OUT;
   opihi_flt start, end, delta;
-  opihi_flt *V;
   Vector *xvec, *yvec;
 
@@ -31 +31 @@
   if ((xvec = SelectVector (argv[1], OLDVECTOR, TRUE)) == NULL) return (FALSE);
   if ((yvec = SelectVector (argv[2], ANYVECTOR, TRUE)) == NULL) return (FALSE);
-  yvec[0].Nelements = Nbins;
-    
-  REALLOCATE (yvec[0].elements.Flt, opihi_flt, yvec[0].Nelements);
-  bzero (yvec[0].elements.Flt, sizeof(opihi_flt)*yvec[0].Nelements);
-  if (Nbins < 1) {
-      return (TRUE);
+
+  ResetVector (yvec, OPIHI_INT, Nbins);
+  bzero (yvec[0].elements.Int, sizeof(opihi_int)*yvec[0].Nelements);
+  if (Nbins < 1) return (TRUE);
+  *OUT = yvec[0].elements.Int;
+
+  if (xvec[0].type == OPIHI_FLT) {
+    opihi_flt *V = xvec[0].elements.Flt;
+    for (i = 0; i < xvec[0].Nelements; i++, V++) {
+      if (isnan(*V)) continue;
+      bin = MIN (MAX (0, (*V - start) / delta), Nbins - 1);
+      OUT[bin]++;
+    }      
+  } else {
+    opihi_int *V = xvec[0].elements.Int;
+    for (i = 0; i < xvec[0].Nelements; i++, V++) {
+      if (isnan(*V)) continue;
+      bin = MIN (MAX (0, (*V - start) / delta), Nbins - 1);
+      OUT[bin]++;
+    }      
   }
-
-  V = xvec[0].elements.Flt;
-  for (i = 0; i < xvec[0].Nelements; i++, V++) {
-    if (isnan(*V)) continue;
-    bin = MIN (MAX (0, (*V - start) / delta), Nbins - 1);
-    yvec[0].elements.Flt[bin] += 1.0;
-  }      
 
   return (TRUE);