Changeset 38340 for branches/eam_branches/ohana.20150429/src/opihi/cmd.data/read_vectors.c

Timestamp:

May 30, 2015, 7:59:48 PM (11 years ago)

Author:

eugene

Message:

more work on compression : cfitsio assumes gzip, not zlib (I have made uncompress general )

File:

• branches/eam_branches/ohana.20150429/src/opihi/cmd.data/read_vectors.c (modified) (8 diffs)

branches/eam_branches/ohana.20150429/src/opihi/cmd.data/read_vectors.c

@@ -412 +412 @@
   table.header = &header;
 
+  int IsCompressed = FALSE;
+
   /* load appropriate extension (if extname is a number, use count) */
   if (Nextend > -1) {
@@ -430 +432 @@
     if (!gfits_load_header (f, &header)) ESCAPE ("error reading header for extension %d\n", Nextend);
 
-    if (getSizes) {
+    IsCompressed = gfits_extension_is_compressed_table (&header);
+
+    if (getSizes && !IsCompressed) {
       read_table_sizes (&header);
       if (CCDKeyword != NULL) free (CCDKeyword); 
       gfits_free_header (&header); 
       return TRUE;
+    }
+    if (IsCompressed) {
+      if (getSizes) {
+        gprint (GP_ERR, "%s[%s] is compressed: -sizes is invalid\n", filename, extname);
+        gfits_free_header (&header); 
+        return FALSE;
+      }
+      if ((start > 0) || (Nrows > -1)) {
+        gprint (GP_ERR, "%s[%s] is compressed: must read entire table\n", filename, extname);
+        gfits_free_header (&header); 
+        return FALSE;
+      }
     }
 
@@ -483 +499 @@
       }
 
-      if (getSizes) {
+      IsCompressed = gfits_extension_is_compressed_table (&header);
+
+      if (getSizes && !IsCompressed) {
         read_table_sizes (&header);
         if (CCDKeyword != NULL) free (CCDKeyword); 
@@ -489 +507 @@
         return TRUE;
       }
+      if (IsCompressed) {
+        if (getSizes) {
+          gprint (GP_ERR, "%s[%s] is compressed: -sizes is invalid\n", filename, extname);
+          gfits_free_header (&header); 
+          return FALSE;
+        }
+        if ((start > 0) || (Nrows > -1)) {
+          gprint (GP_ERR, "%s[%s] is compressed: must read entire table\n", filename, extname);
+          gfits_free_header (&header); 
+          return FALSE;
+        }
+      }
 
       if (Nrows == -1) {
@@ -503 +533 @@
 
       if (!gfits_fread_ftable_range (f, padIfShort, &table, start, Nrows)) ESCAPE ("error reading table for extension %d\n", Nextend);
-
-      // if (!gfits_fread_ftable_data (f, &table, padIfShort)) ESCAPE ("error reading table for extension\n");
-
       break;
     }
@@ -518 +545 @@
   Binary = !strcmp (type, "BINTABLE");
   Ny = header.Naxis[1];
+
+  Header *outheader = &header;
+  FTable *outtable  = &table;
+
+  Header rawheader;
+  FTable rawtable;
+  if (IsCompressed) {
+    rawtable.header = &rawheader;
+    if (!gfits_uncompress_table (&table, &rawtable)) ESCAPE ("failed to uncompress table");
+    outheader = &rawheader;
+    outtable  = &rawtable;
+    gfits_free_header (&header);
+    gfits_free_table (&table);
+    Ny = rawheader.Naxis[1];
+  }
 
   /* find columns which match requested vectors */
@@ -527 +569 @@
     Nval = 0;
     if (Binary) {
-      if (!gfits_get_bintable_column_type (&header, argv[i], type, &Nval)) ESCAPE ("requested field not found");
-      if (!gfits_get_bintable_column (&header, &table, argv[i], &data)) ESCAPE ("error reading data from specified field");
+      if (!gfits_get_bintable_column_type (outheader, argv[i], type, &Nval)) ESCAPE ("requested field not found");
+      if (!gfits_get_bintable_column (outheader, outtable, argv[i], &data)) ESCAPE ("error reading data from specified field");
     } else {
-      if (!gfits_get_table_column_type (&header, argv[i], type)) ESCAPE ("requested field not found");
-      if (!gfits_get_table_column (&header, &table, argv[i], &data)) ESCAPE ("error reading data from specified field");
+      if (!gfits_get_table_column_type (outheader, argv[i], type)) ESCAPE ("requested field not found");
+      if (!gfits_get_table_column (outheader, outtable, argv[i], &data)) ESCAPE ("error reading data from specified field");
     }
     if (Nval == 0) ESCAPE ("no data for field in table");
@@ -594 +636 @@
   }
   if (CCDKeyword != NULL) free (CCDKeyword);
-  gfits_free_table (&table);
-  gfits_free_header (&header);
+  gfits_free_table (outtable);
+  gfits_free_header (outheader);
   return (TRUE);
 }