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

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

@@ -43 +43 @@
   }
 
-  zvec[0].Nelements = xvec[0].Nelements;
-  REALLOCATE (zvec[0].elements.Flt, opihi_flt, zvec[0].Nelements);
+  ResetVector (zvec, OPIHI_FLT, xvec[0].Nelements);
   bzero (zvec[0].elements.Flt, sizeof(opihi_flt)*zvec[0].Nelements);
-  x = xvec[0].elements.Flt;
-  y = yvec[0].elements.Flt;
   z = zvec[0].elements.Flt;
 
-  for (n = 0; n < xvec[0].Nelements; n++, x++, y++, z++) {
-    Y = X = 1;
-    for (j = 0; j < order + 1; j++) {
-      X = Y;
-      for (i = 0; i < order + 1 - j; i++) {
-        *z += C[i][j]*X;
-        X = X * (*x);
+  // we have four cases (x == flt or int) and (y == flt or int)
+  if ((xvec[0].type == OPIHI_FLT) && (yvec[0].type == OPIHI_FLT)) {
+    opihi_flt *x = xvec[0].elements.Flt;
+    opihi_flt *y = yvec[0].elements.Flt;
+    for (n = 0; n < xvec[0].Nelements; n++, x++, y++, z++) {
+      Y = X = 1;
+      for (j = 0; j < order + 1; j++) {
+        X = Y;
+        for (i = 0; i < order + 1 - j; i++) {
+          *z += C[i][j]*X;
+          X = X * (*x);
+        }
+        Y = Y * (*y);
       }
-      Y = Y * (*y);
     }
   }
+  if ((xvec[0].type == OPIHI_FLT) && (yvec[0].type == OPIHI_INT)) {
+    opihi_flt *x = xvec[0].elements.Flt;
+    opihi_int *y = yvec[0].elements.Int;
+    for (n = 0; n < xvec[0].Nelements; n++, x++, y++, z++) {
+      Y = X = 1;
+      for (j = 0; j < order + 1; j++) {
+        X = Y;
+        for (i = 0; i < order + 1 - j; i++) {
+          *z += C[i][j]*X;
+          X = X * (*x);
+        }
+        Y = Y * (*y);
+      }
+    }
+  }
+  if ((xvec[0].type == OPIHI_INT) && (yvec[0].type == OPIHI_FLT)) {
+    opihi_int *x = xvec[0].elements.Int;
+    opihi_flt *y = yvec[0].elements.Flt;
+    for (n = 0; n < xvec[0].Nelements; n++, x++, y++, z++) {
+      Y = X = 1;
+      for (j = 0; j < order + 1; j++) {
+        X = Y;
+        for (i = 0; i < order + 1 - j; i++) {
+          *z += C[i][j]*X;
+          X = X * (*x);
+        }
+        Y = Y * (*y);
+      }
+    }
+  }
+  if ((xvec[0].type == OPIHI_INT) && (yvec[0].type == OPIHI_INT)) {
+    opihi_int *x = xvec[0].elements.Int;
+    opihi_int *y = yvec[0].elements.Int;
+    for (n = 0; n < xvec[0].Nelements; n++, x++, y++, z++) {
+      Y = X = 1;
+      for (j = 0; j < order + 1; j++) {
+        X = Y;
+        for (i = 0; i < order + 1 - j; i++) {
+          *z += C[i][j]*X;
+          X = X * (*x);
+        }
+        Y = Y * (*y);
+      }
+    }
+  }
+
   for (i = 0; i < order + 1; i++) free (C[i]);
   free (C);