Changeset 6896 for trunk/psModules/src/astrom/pmFPAWrite.c

Timestamp:

Apr 18, 2006, 5:24:09 PM (20 years ago)

Author:

Paul Price

Message:

Adding pmReadoutWriteNext, which is to be used in concert with pmReadoutReadNext. Updated pmFPAWrite, pmChipWrite, pmCellWrite to optionally not write the pixels (i.e., PHU only), so that an output file should always have a PHU. This necessitated some small changes to the way the 'concepts' are written, in the case that they don't exist.

File:

• trunk/psModules/src/astrom/pmFPAWrite.c (modified) (8 diffs)

trunk/psModules/src/astrom/pmFPAWrite.c

@@ -5 +5 @@
 #include "pmFPA.h"
 #include "pmHDU.h"
+#include "pmHDUUtils.h"
 #include "pmConcepts.h"
+
+
+bool pmReadoutWriteNext(pmReadout *readout, // Readout to write
+                        psFits *fits,   // FITS file to which to write
+                        int z           // Image plane to write
+                       )
+{
+    pmHDU *hdu = pmHDUFromReadout(readout); // The HDU to which to write
+    psMetadata *header = hdu->header;   // The FITS header
+    if (!header) {
+        psError(PS_ERR_IO, true, "No FITS header available in the HDU.\n");
+        return false;
+    }
+
+    // We have to rely to a great extent on the FITS header, because otherwise we simply don't know how many
+    // image planes there are (NAXIS3) or the size of the original image (NAXIS1, NAXIS2).
+    bool mdok = true;                   // Status of MD lookup
+    int naxis1 = psMetadataLookupS32(&mdok, header, "NAXIS1"); // Number of columns
+    if (!mdok || naxis1 <= 0) {
+        psError(PS_ERR_IO, true, "Can't find NAXIS1 in header.\n");
+        return false;
+    }
+    int naxis2 = psMetadataLookupS32(&mdok, header, "NAXIS2"); // Number of rows
+    if (!mdok || naxis2 <= 0) {
+        psError(PS_ERR_IO, true, "Can't find NAXIS2 in header.\n");
+        return false;
+    }
+    int naxis3 = psMetadataLookupS32(&mdok, header, "NAXIS3"); // Number of image planes
+    if (!mdok || naxis3 <= 0) {
+        naxis3 = 1;
+    }
+    if (z >= naxis3) {
+        psError(PS_ERR_IO, true, "Specified a plane number (%d) greater than NAXIS3 allows.\n", z);
+        return false;
+    }
+
+    psImage *image = hdu->images->data[z]; // The image from the HDU to write
+    if (readout->row0 == 0 && readout->col0 == 0 && z == 0) {
+        // Then we can assume that nothing has been written to the FITS file for now
+        if (naxis1 == image->numCols && naxis2 == image->numRows) {
+            // We can write the whole lot at once
+            return psFitsWriteImage(fits, header, image, z, hdu->extname);
+        }
+        // Create a dummy image so we can write something larger than we actually have
+        psImage *dummy = psImageAlloc(naxis1, naxis2, image->type.type); // Dummy image
+        psImageInit(dummy, 0);
+        psImageOverlaySection(dummy, image, 0, 0, "=");
+        bool result = psFitsWriteImage(fits, header, dummy, z, hdu->extname);
+        psFree(dummy);
+        return result;
+    }
+
+    // We can simply update an existing HDU
+    if (!psFitsMoveExtName(fits, hdu->extname)) {
+        psError(PS_ERR_IO, false, "Unable to move to extension %s\n", hdu->extname);
+        return false;
+    }
+    return psFitsUpdateImage(fits, image, readout->col0, readout->row0, z);
+}
+
+
 
 
 bool pmCellWrite(pmCell *cell,          // Cell to write
                  psFits *fits,          // FITS file to which to write
-                 psDB *db               // Database handle for "concepts" update
+                 psDB *db,              // Database handle for "concepts" update
+                 bool pixels            // Write the pixels?
                 )
 {
@@ -17 +80 @@
 
     pmHDU *hdu = cell->hdu;             // The HDU
-    if (hdu && !pmHDUWrite(hdu, fits)) {
+    if (hdu && ((!pixels && hdu->phu) || pixels) && !pmHDUWrite(hdu, fits)) {
         psError(PS_ERR_IO, false, "Unable to write HDU for Chip.\n");
         return false;
@@ -28 +91 @@
 bool pmChipWrite(pmChip *chip,          // Chip to write
                  psFits *fits,          // FITS file to which to write
-                 psDB *db               // Database handle for "concepts" update
+                 psDB *db,              // Database handle for "concepts" update
+                 bool pixels            // Write the pixels?
                 )
 {
@@ -35 +99 @@
 
     pmHDU *hdu = chip->hdu;             // The HDU
-    if (hdu && !pmHDUWrite(hdu, fits)) {
+    if (hdu && ((!pixels && hdu->phu) || pixels) && !pmHDUWrite(hdu, fits)) {
         psError(PS_ERR_IO, false, "Unable to write HDU for Chip.\n");
         return false;
@@ -41 +105 @@
 
     bool success = true;                // Success of writing
-    psArray *cells = chip->cells;       // Array of cells
-    for (int i = 0; i < cells->n; i++) {
-        pmCell *cell = cells->data[i];  // The cell of interest
-        success |= pmCellWrite(cell, fits, db);
+    if (pixels) {
+        psArray *cells = chip->cells;       // Array of cells
+        for (int i = 0; i < cells->n; i++) {
+            pmCell *cell = cells->data[i];  // The cell of interest
+            success |= pmCellWrite(cell, fits, db, true);
+        }
     }
 
@@ -55 +121 @@
 bool pmFPAWrite(pmFPA *fpa,             // FPA to write
                 psFits *fits,           // FITS file to which to write
-                psDB *db                // Database handle for "concepts" update
+                psDB *db,               // Database handle for "concepts" update
+                bool pixels             // Write the pixels?
                )
 {
@@ -62 +129 @@
 
     pmHDU *hdu = fpa->hdu;              // The HDU
-    if (hdu && !pmHDUWrite(hdu, fits)) {
+    if (hdu && ((!pixels && hdu->phu) || pixels) && !pmHDUWrite(hdu, fits)) {
         psError(PS_ERR_IO, false, "Unable to write HDU for FPA.\n");
         return false;
@@ -68 +135 @@
 
     bool success = true;                // Success of writing
-    psArray *chips = fpa->chips;        // Array of chips
-    for (int i = 0; i < chips->n; i++) {
-        pmChip *chip = chips->data[i];  // The chip of interest
-        success |= pmChipWrite(chip, fits, db);
+    if (pixels) {
+        psArray *chips = fpa->chips;        // Array of chips
+        for (int i = 0; i < chips->n; i++) {
+            pmChip *chip = chips->data[i];  // The chip of interest
+            success |= pmChipWrite(chip, fits, db, true);
+        }
     }