Commit ea4c8191 authored by Wei Chengliang's avatar Wei Chengliang
Browse files

correct ximg & yimg in .cat output; add --markdone flag

parent d612bd0f

observation_sim/_util.py

+5 −0
Original line number Diff line number Diff line
@@ -30,6 +30,11 @@ def parse_args(): 
        "-d", "--data_dir", type=str, help="Directory that houses the input data."

    )

    parser.add_argument("-w", "--work_dir", type=str, help="The path for output.")

    parser.add_argument(

            "--markdone",

            action="store_true",

            help="Mark config as .bak after successful run.",

        )

    return parser.parse_args()

observation_sim/config/ChipOutput.py

+2 −2
Original line number Diff line number Diff line
@@ -100,8 +100,8 @@ class ChipOutput(object): 
            self.cat.write(self.hdr)



    def cat_add_obj(self, obj, pos_img, pos_shear, ra_offset=0.0, dec_offset=0.0):

        ximg = obj.real_pos.x + 1.0

        yimg = obj.real_pos.y + 1.0

        ximg = obj.real_pos.x

        yimg = obj.real_pos.y



        line = self.fmt % (

            obj.id,

run_sim.py

+8 −0
Original line number Diff line number Diff line
@@ -6,6 +6,7 @@ import yaml 
import shutil

import datetime

import importlib

import mpi4py.MPI as MPI



import gc
@@ -123,6 +124,13 @@ def run_sim(): 
    # Run simulation

    obs.runExposure_MPI_PointingList(pointing_list=pointing_list)



    #move config to .bak

    if args.markdone:

        comm = MPI.COMM_WORLD

        ind_thread = comm.Get_rank()

        comm.Barrier()

        if ind_thread == 0:

            shutil.move(args.config_file, args.config_file+'.bak')



if __name__ == "__main__":

    run_sim()