#!/usr/bin/env python3

import os
import sys
import pyfits


if __name__ == "__main__":

    fn = sys.argv[1]
    bn,_ = os.path.splitext(fn)

    hdulist = pyfits.open(fn)
    print("Reading input %s" % (fn))

    img = pyfits.PrimaryHDU(
        data=hdulist['IMAGE'].data,
        header=hdulist['IMAGE'].header
    )
    img_out_fn = bn+"_image.fits"
    print("Writing %s" % (img_out_fn))
    img.writeto(img_out_fn, clobber=True)

    mask = pyfits.PrimaryHDU(
        data = hdulist['MASK'].data,
        header = hdulist['MASK'].header
    )
    mask_out_fn = bn+"_mask.fits"
    print("writing %s" % (mask_out_fn))
    mask.writeto(mask_out_fn, clobber=True)

    var = pyfits.PrimaryHDU(
        data = hdulist['VARIANCE'].data,
        header = hdulist['VARIANCE'].header
    )
    var_out_fn = bn+"_var.fits"
    print("writing %s" % (var_out_fn))
    var.writeto(var_out_fn, clobber=True)

    print("all done!")