Changeset 38330 for branches/eam_branches/ohana.20150429/src/opihi/cmd.data/wd.c

Timestamp:

May 28, 2015, 8:28:24 PM (11 years ago)

Author:

eugene

Message:

adding compression option to wd and write

File:

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

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

@@ -3 +3 @@
 int wd (int argc, char **argv) {
   
-  int N, Extend;
+  int N, Extend, Compress;
   int newUnsign, newBitpix, newScale, newZero;
   int outUnsign, outBitpix;
@@ -16 +16 @@
     remove_argument (N, &argc, argv);
     Extend  = TRUE;
+  }
+
+  Compress = FALSE;
+  if ((N = get_argument (argc, argv, "-compress"))) {
+    remove_argument (N, &argc, argv);
+    Compress = TRUE;
   }
 
@@ -162 +168 @@
   }
   
+  // not compatible with extend
+  if (Compress) {
+    Header myHeader;
+    Matrix myMatrix;
+
+    FILE *f = fopen (argv[2], "w");
+    if (!f) {
+      fprintf (stderr, "ERROR: cannot open image subset file for output %s\n", argv[2]);
+      return FALSE;
+    }
+    
+    gfits_init_header (&myHeader);
+    myHeader.extend = TRUE;
+    gfits_create_header (&myHeader);
+    gfits_create_matrix (&myHeader, &myMatrix);
+    gfits_fwrite_header  (f, &myHeader);
+    gfits_fwrite_matrix  (f, &myMatrix);
+    gfits_free_header (&myHeader);
+    gfits_free_matrix (&myMatrix);
+
+    FTable ftable;
+    Header theader;
+
+    ftable.header = &theader;
+
+    gfits_compress_image (&temp_header, &temp_matrix, &ftable, NULL, "GZIP_1");
+    gfits_byteswap_varlength_column (&ftable, 1);
+    
+    gfits_fwrite_Theader (f, &theader);
+    gfits_fwrite_table  (f, &ftable);
+    fclose (f);
+
+    gfits_free_header (&theader);
+    gfits_free_table (&ftable);
+    return (TRUE);
+  }
+
   /* the actual write-to-disk goes here */
   if (!gfits_write_header (argv[2], &temp_header)) {