Commit ef8c4609 authored by Fang Yuedong's avatar Fang Yuedong
Browse files

modified fits header etc.

parent 3f932427

ObservationSim/Config/ChipOutput.py

+10 −10
Original line number Diff line number Diff line 
import os



class ChipOutput(object):

    def __init__(self, config, focal_plane, chip, filt, imgKey0="", imgKey1="", imgKey2="", exptime=150., mjdTime="", ra_cen=None, dec_cen=None, pointing_ID='0', subdir="./", prefix=""):

    def __init__(self, config, focal_plane, chip, filt, imgKey0="", imgKey1="", imgKey2="", exptime=150., mjdTime="", ra_cen=None, dec_cen=None, pointing_type='MS', pointing_ID='0', subdir="./", prefix=""):

        self.focal_plane = focal_plane

        self.chip = chip

        self.filt = filt
@@ -25,18 +25,19 @@ class ChipOutput(object): 


        # hdr1  = "#ID ID_chip filter xImage yImage ra dec z mag flag SNR "

        hdr1  = "#ID ID_chip filter xImage yImage ra dec z mag flag "

        hdr2  = "thetaR bfrac hlr_disk hlr_bulge e1_disk e2_disk e1_bulge e2_bulge e1_total e2_total"

        hdr2  = "thetaR bfrac hlr_disk hlr_bulge e1_disk e2_disk e1_bulge e2_bulge "

        hdr3  = "e1PSF e2PSF e1 e2 g1 g2 e1OBS e2OBS "

        hdr4  = "sed_type av redden "

        hdr5  = "star_model teff logg feh\n"

        # fmt1  = "%10d %4d %5s %10.3f %10.3f %15.6f %15.6f %7.4f %8.4f %2d %9.2f "

        fmt1  = "%10d %4d %5s %10.3f %10.3f %15.6f %15.6f %7.4f %8.4f %2d "

        fmt2  = "%8.4f %8.4f %8.4f %8.4f %8.4f %8.4f %8.4f %8.4f %8.4f %8.4f"

        fmt2  = "%8.4f %8.4f %8.4f %8.4f %8.4f %8.4f %8.4f %8.4f "

        fmt3  = "%8.4f %8.4f %8.4f %8.4f %8.4f %8.4f %8.4f %8.4f "

        fmt4 = "%2d %8.4f %8.4f "

        fmt5 = "%10s %8.4f %8.4f %8.4f\n"

        self.hdr = hdr1 + hdr2 + hdr3 + hdr4 + hdr5

        self.fmt = fmt1 + fmt2 + fmt3 + fmt4 + fmt5

        if pointing_type == 'MS':

            self.cat = open(os.path.join(self.subdir, self.cat_name), "w")

            self.cat.write(self.hdr)
@@ -79,15 +80,14 @@ class ChipOutput(object): 
        e1, e2, g1, g2, e1OBS, e2OBS = obj.getObservedEll(g1, g2)

        if obj.type == 'galaxy':

            line = self.fmt%(obj.id, int(self.chipLabel), self.filt.filter_type, ximg, yimg, obj.ra, obj.dec, obj.z, obj.getMagFilter(self.filt), obj.param["star"], obj.thetaR, obj.bfrac, obj.hlr_disk, obj.hlr_bulge,

                obj.e1_disk, obj.e2_disk, obj.e1_bulge, obj.e2_bulge, obj.e1_total, obj.e2_total,

                obj.e1_disk, obj.e2_disk, obj.e1_bulge, obj.e2_bulge,

                pos_shear.g1, pos_shear.g2, e1, e2, g1, g2, e1OBS, e2OBS, obj.sed_type, obj.param['av'], obj.param['redden'], 'n', 0, 0, 0)

        elif obj.type == "quasar":

            line = self.fmt % (obj.id, int(self.chipLabel), self.filt.filter_type, ximg, yimg, obj.ra, obj.dec, obj.z,

                               obj.getMagFilter(self.filt), obj.param["star"], 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,

                               0.0, 0.0,

                               pos_shear.g1, pos_shear.g2, e1, e2, g1, g2, e1OBS, e2OBS, obj.sed_type, obj.param['av'], obj.param['redden'], 'n', 0.0, 0.0, 0.0)

        else:

            line = self.fmt%(obj.id, int(self.chipLabel), self.filt.filter_type, ximg, yimg, obj.ra, obj.dec, obj.z, obj.getMagFilter(self.filt), obj.param["star"], 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,

            line = self.fmt%(obj.id, int(self.chipLabel), self.filt.filter_type, ximg, yimg, obj.ra, obj.dec, obj.z, obj.getMagFilter(self.filt), obj.param["star"], 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 

                pos_shear.g1, pos_shear.g2, e1, e2, g1, g2, e1OBS, e2OBS, 0, 0.0, 0.0, obj.param['model_tag'], obj.param['teff'], obj.param['logg'],obj.param['feh'])

        # line = self.fmt%(obj.id, int(self.chipLabel), self.filt.filter_type, ximg, yimg, obj.ra, obj.dec, obj.z, obj.getMagFilter(self.filt), obj.param["star"], pos_shear.g1, pos_shear.g2, e1, e2, g1, g2, e1OBS, e2OBS)

        self.cat.write(line)

ObservationSim/Config/Config.py

+3 −3
Original line number Diff line number Diff line
@@ -24,9 +24,9 @@ def ConfigDir(cat_dir=None, work_dir=None, data_dir=None, config_file_path=None) 
    # path_dict["output_cat_dir"] = os.path.join(path_dict["work_dir"], "scat/")

    # if not os.path.exists(path_dict["output_cat_dir"]):

    #     os.system("mkdir %s"%path_dict["output_cat_dir"])

    path_dict["output_img_dir"] = os.path.join(path_dict["work_dir"], "simg/")

    if not os.path.exists(path_dict["output_img_dir"]):

        os.system("mkdir %s"%path_dict["output_img_dir"])

    # path_dict["output_img_dir"] = os.path.join(path_dict["work_dir"], "simg/")

    # if not os.path.exists(path_dict["output_img_dir"]):

    #     os.system("mkdir %s"%path_dict["output_img_dir"])

    

    # Data directory

    if data_dir == None:

ObservationSim/Config/Header/ImageHeader.py

+13 −4
Original line number Diff line number Diff line
@@ -233,7 +233,7 @@ def WCS_def(xlen = 9216, ylen = 9232, gapy = 898.0, gapx1 = 534, gapx2 = 1309, r 






def generatePrimaryHeader(xlen = 9216, ylen = 9232, pointNum = '1', ra = 60, dec = -40, psize = 0.074, row_num = 1, col_num = 1, date='200930', time_obs='120000', im_type = 'MS'):

def generatePrimaryHeader(xlen = 9216, ylen = 9232, pointNum = '1', ra = 60, dec = -40, psize = 0.074, row_num = 1, col_num = 1, date='200930', time_obs='120000', im_type = 'MS', exptime=150.):



    # array_size1, array_size2, flux, sigma = int(argv[1]), int(argv[2]), 1000.0, 5.0
@@ -262,12 +262,19 @@ def generatePrimaryHeader(xlen = 9216, ylen = 9232, pointNum = '1', ra = 60, dec 
    h_prim['FILENAME'] = 'MSC_' + im_type + '_' + date + time_obs + '_1' + pointNum.rjust(8,'0') + '_' +CCDID[k-1].rjust(2,'0')+'_raw'

    h_prim['DATE'] = '20'+date[0:2]+'-' + date[2:4]+'-'+date[4:6]

    h_prim['TIME'] = time_obs[0:2]+':'+time_obs[2:4]+':'+time_obs[4:6]

    h_prim['DATE-OBS'] = '20'+date[0:2]+'-' + date[2:4]+'-'+date[4:6]

    h_prim['TIME-OBS'] = time_obs[0:2]+':'+time_obs[2:4]+':'+time_obs[4:6]

    h_prim['DETECTOR'] = 'CCD'+CCDID[k-1].rjust(2,'0')

    h_prim['RA_OBJ'] = ra

    h_prim['DEC_OBJ'] = dec

    h_prim['OBJECT'] = '1'+ pointNum.rjust(8,'0')

    h_prim['OBSID'] = '1'+ pointNum.rjust(8,'0')

    h_prim['TELFOCUS'] = 'f/14'

    h_prim['EXPTIME'] = exptime

    

    # Define file types

    file_type = {'MS':'sci', 'CLB':'zero', 'CLD':'dark', 'CLF':'flat', 'CRS':'cosmic_ray', 'CRD':'cosmic_ray'}

    h_prim['FILETYPE'] = file_type[im_type]



    co = coord.SkyCoord(ra, dec, unit='deg')
@@ -277,13 +284,14 @@ def generatePrimaryHeader(xlen = 9216, ylen = 9232, pointNum = '1', ra = 60, dec 


    h_prim['RA_NOM'] = ra_hms

    h_prim['DEC_NOM'] = dec_hms

    h_prim['PIXSCAL1'] = str(psize)

    h_prim['PIXSCAL2'] = str(psize)

    h_prim['PIXSCAL1'] = psize

    h_prim['PIXSCAL2'] = psize



    ttt = h_prim['DATE'] + 'T' + h_prim['TIME']

    tstart = Time(ttt)

    h_prim['EXPSTART'] = round(tstart.mjd, 5)

    tend = Time(tstart.cxcsec + h_prim['EXPTIME'], format="cxcsec")

    # tend = Time(tstart.cxcsec + h_prim['EXPTIME'], format="cxcsec")

    tend = Time(tstart.mjd + h_prim['EXPTIME']/86400., format="mjd")

    h_prim['EXPEND'] = round(tend.mjd, 5)



    return h_prim
@@ -315,6 +323,7 @@ def generateExtensionHeader(xlen = 9216, ylen = 9232,ra = 60, dec = -40, pa = -2 
    h_ext['GAIN1'] = gain

    h_ext['RDNOISE1'] = readout

    h_ext['CCDCHIP'] =  'ccd' + CCDID[k-1].rjust(2,'0')

    h_ext['POS_ANG'] = pa

    header_wcs = WCS_def(xlen=xlen, ylen=ylen, gapy=898.0, gapx1=534, gapx2=1309, ra=ra, dec=dec, pa=pa, psize=psize,

                         row_num=row_num, col_num=col_num)