Commit a6c54cf9 authored by niejuzi's avatar niejuzi
Browse files

updated pipeline

parent 8fdb4e50

csst/msc/pipeline.py

+35 −35
Original line number Diff line number Diff line
@@ -69,41 +69,41 @@ def do_one_exposure(ver_sim="C5.1", dir_l0="", dir_l1="", dir_pcref="", path_aux 
    # Step 1. Correct instrumental effect

    os.chdir(dir_l1)



    img_list = []

    wht_list = []

    flg_list = []

    fn_list = []

    for this_ccd_id in ccd_ids:

        print("processing CCD {}".format(this_ccd_id))

        fp_raw = dm.l0_sci(ccd_id=this_ccd_id)



        # read data with CsstMscImgData.read

        raw = CsstMscImgData.read(fp_raw)



        # in the future, use get_* functions grab

        bias = dm.get_bias(this_ccd_id)

        dark = dm.get_dark(this_ccd_id)

        flat = dm.get_flat(this_ccd_id)



        # initialize Instrument Processor

        instProc = CsstMscInstrumentProc()

        instProc.prepare(n_jobs=n_jobs)

        img, wht, flg = instProc.run(raw, bias, dark, flat, ver_sim)

        instProc.cleanup()

        fp_img = img[0].header["FILENAME"] + '.fits'



        # save img, wht, flg to somewhere

        img.writeto(dm.l1_sci(ccd_id=this_ccd_id, suffix="img", ext="fits"), overwrite=True)

        wht.writeto(dm.l1_sci(ccd_id=this_ccd_id, suffix="wht", ext="fits"), overwrite=True)

        flg.writeto(dm.l1_sci(ccd_id=this_ccd_id, suffix="flg", ext="fits"), overwrite=True)

        # save header

        img[1].header.tofile(dm.l1_sci(ccd_id=this_ccd_id, suffix="img", ext="head"), overwrite=True)



        # append img, wht, flg list

        img_list.append(img)

        wht_list.append(wht)

        flg_list.append(flg)

        fn_list.append(fp_img)

    #img_list = []

    #wht_list = []

    #flg_list = []

    #fn_list = []

    #for this_ccd_id in ccd_ids:

    #    print("processing CCD {}".format(this_ccd_id))

    #    fp_raw = dm.l0_sci(ccd_id=this_ccd_id)



    #    # read data with CsstMscImgData.read

    #    raw = CsstMscImgData.read(fp_raw)



    #    # in the future, use get_* functions grab

    #    bias = dm.get_bias(this_ccd_id)

    #    dark = dm.get_dark(this_ccd_id)

    #    flat = dm.get_flat(this_ccd_id)



    #    # initialize Instrument Processor

    #    instProc = CsstMscInstrumentProc()

    #    instProc.prepare(n_jobs=n_jobs)

    #    img, wht, flg = instProc.run(raw, bias, dark, flat, ver_sim)

    #    instProc.cleanup()

    #    fp_img = img[0].header["FILENAME"] + '.fits'



    #    # save img, wht, flg to somewhere

    #    img.writeto(dm.l1_sci(ccd_id=this_ccd_id, suffix="img", ext="fits"), overwrite=True)

    #    wht.writeto(dm.l1_sci(ccd_id=this_ccd_id, suffix="wht", ext="fits"), overwrite=True)

    #    flg.writeto(dm.l1_sci(ccd_id=this_ccd_id, suffix="flg", ext="fits"), overwrite=True)

    #    # save header

    #    img[1].header.tofile(dm.l1_sci(ccd_id=this_ccd_id, suffix="img", ext="head"), overwrite=True)



    #    # append img, wht, flg list

    #    img_list.append(img)

    #    wht_list.append(wht)

    #    flg_list.append(flg)

    #    fn_list.append(fp_img)



    # Step 2. Calibrate Position

    pcProc = CsstProcMscPositionCalibration()