Changeset 34529 for branches/eam_branches/ipp-20120905/Ohana/src/opihi/lib.data/SplineOps.c

Timestamp:

Oct 8, 2012, 2:30:52 PM (14 years ago)

Author:

eugene

Message:

add IO functions for splines

File:

• branches/eam_branches/ipp-20120905/Ohana/src/opihi/lib.data/SplineOps.c (modified) (4 diffs)

branches/eam_branches/ipp-20120905/Ohana/src/opihi/lib.data/SplineOps.c

@@ -27 +27 @@
 void InitSpline (Spline *spline, char *name, int Nknots) {
 
+  if (!spline[0].name) {
     spline[0].name = strcreate (name);
-
-    spline[0].Nknots = Nknots;
+  }
+  spline[0].Nknots = Nknots;
+  
+  if (spline[0].xk) {
+    REALLOCATE (spline[0].xk, opihi_flt, spline[0].Nknots);
+    REALLOCATE (spline[0].yk, opihi_flt, spline[0].Nknots);
+    REALLOCATE (spline[0].y2, opihi_flt, spline[0].Nknots);
+  } else {
     ALLOCATE (spline[0].xk, opihi_flt, spline[0].Nknots);
     ALLOCATE (spline[0].yk, opihi_flt, spline[0].Nknots);
     ALLOCATE (spline[0].y2, opihi_flt, spline[0].Nknots);
-    memset (spline[0].xk, 0, spline[0].Nknots * sizeof(opihi_flt));
-    memset (spline[0].yk, 0, spline[0].Nknots * sizeof(opihi_flt));
-    memset (spline[0].y2, 0, spline[0].Nknots * sizeof(opihi_flt));
+  }
+  memset (spline[0].xk, 0, spline[0].Nknots * sizeof(opihi_flt));
+  memset (spline[0].yk, 0, spline[0].Nknots * sizeof(opihi_flt));
+  memset (spline[0].y2, 0, spline[0].Nknots * sizeof(opihi_flt));
 }
 
@@ -76 +84 @@
 
   spline = FindSpline (name);
-  if (spline != NULL) return (spline);
+  if (spline != NULL) {
+    InitSpline (spline, name, Nknots);
+    return (spline);
+  }
 
   N = Nsplines;
@@ -82 +93 @@
   CHECK_REALLOCATE (splines, Spline *, NSPLINES, Nsplines, 16);
   ALLOCATE (spline, Spline, 1);
+  spline->name = NULL;
+  spline->xk = NULL;
+  spline->yk = NULL;
+  spline->y2 = NULL;
   InitSpline (spline, name, Nknots);
   splines[N] = spline;
@@ -126 +141 @@
   return;
 }
+
+// write the spline data to a FITS file.
+int SaveSpline (char *filename, char *name, int append) {
+  
+  Header header;
+  Matrix matrix;
+  Header theader;
+  FTable ftable;
+
+  Spline *myspline = FindSpline (name);
+  if (!myspline) {
+    gprint (GP_ERR, "can't find spline for write : %s\n", name);
+    return (FALSE);
+  }
+
+  FILE *f = NULL;
+
+  /* open file for output */
+  if (append) {
+    f = fopen (filename, "a");
+  } else {
+    f = fopen (filename, "w");
+  }
+  if (f == (FILE *) NULL) {
+    gprint (GP_ERR, "can't open file for write : %s\n", filename);
+    return (FALSE);
+  }
+
+  gfits_create_table_header (&theader, "BINTABLE", name);
+
+  gfits_define_bintable_column (&theader, "D", "X_KNOT", NULL, NULL, 1.0, 0.0);
+  gfits_define_bintable_column (&theader, "D", "Y_KNOT", NULL, NULL, 1.0, 0.0);
+  gfits_define_bintable_column (&theader, "D", "DY2_DX", NULL, NULL, 1.0, 0.0);
+
+  // generate the output array that carries the data
+  gfits_create_table (&theader, &ftable);
+
+  gfits_set_bintable_column_reformat (&theader, &ftable, "X_KNOT", "double", myspline->xk, myspline->Nknots);
+  gfits_set_bintable_column_reformat (&theader, &ftable, "Y_KNOT", "double", myspline->yk, myspline->Nknots);
+  gfits_set_bintable_column_reformat (&theader, &ftable, "DY2_DX", "double", myspline->y2, myspline->Nknots);
+
+  if (!append) {
+    gfits_init_header (&header);
+    header.extend = TRUE;
+
+    gfits_create_header (&header);
+    gfits_create_matrix (&header, &matrix);
+
+    gfits_fwrite_header  (f, &header);
+    gfits_fwrite_matrix  (f, &matrix);
+
+    gfits_free_header (&header);
+    gfits_free_matrix (&matrix);
+  }
+  gfits_fwrite_Theader (f, &theader);
+  gfits_fwrite_table  (f, &ftable);
+
+  gfits_free_header (&theader);
+  gfits_free_table (&ftable);
+
+  fclose (f);
+  fflush (f);
+  return (TRUE);
+}
+
+// read the spline data from a FITS file.
+// these functions (LoadSpline, SaveSpline) should probably take spline pointers not names
+int LoadSpline (char *filename, char *name) {
+  
+  int i, Ncol;
+  off_t Nrow;
+  char type[16];
+
+  Header theader;
+  FTable ftable;
+
+  FILE *f = NULL;
+
+  /* open file for input */
+  f = fopen (filename, "r");
+  if (f == (FILE *) NULL) {
+    gprint (GP_ERR, "can't open file for read : %s\n", filename);
+    return FALSE;
+  }
+
+  ftable.header = &theader;
+
+  // read the full table data into a buffer
+  if (!gfits_fread_ftable (f, &ftable, name)) {
+    fclose (f);
+    gprint (GP_ERR, "can't read FITS file : %s\n", filename);
+    return FALSE;
+  }
+
+  // XXX: need to handle case of spline data not existing...
+
+  // need to create and assign to flat-field correction
+  double *xk = gfits_get_bintable_column_data (&theader, &ftable, "X_KNOT", type, &Nrow, &Ncol);
+  assert (!strcmp(type, "double"));
+  
+  // need to create and assign to flat-field correction
+  double *yk = gfits_get_bintable_column_data (&theader, &ftable, "Y_KNOT", type, &Nrow, &Ncol);
+  assert (!strcmp(type, "double"));
+
+  // need to create and assign to flat-field correction
+  double *y2 = gfits_get_bintable_column_data (&theader, &ftable, "DY2_DX", type, &Nrow, &Ncol);
+  assert (!strcmp(type, "double"));
+
+  Spline *myspline = CreateSpline (name, Nrow);
+  for (i = 0; i < Nrow; i++) {
+    myspline->xk[i] = xk[i];
+    myspline->yk[i] = yk[i];
+    myspline->y2[i] = y2[i];
+  }
+  free (xk);
+  free (yk);
+  free (y2);
+
+  fclose (f);
+  return (TRUE);
+}