from ObservationSim.Config.Header import generatePrimaryHeader, generateExtensionHeader

def prepare_headers(self, chip, pointing):
    self.h_prim = generatePrimaryHeader(
        xlen=chip.npix_x, 
        ylen=chip.npix_y, 
        pointing_id = pointing.obs_id,
        pointing_type_code = pointing.pointing_type_code, 
        ra=pointing.ra, 
        dec=pointing.dec, 
        pixel_scale=chip.pix_scale,
        time_pt = pointing.timestamp,
        exptime=pointing.exp_time,
        im_type=pointing.pointing_type,
        sat_pos=[pointing.sat_x, pointing.sat_y, pointing.sat_z],
        sat_vel=[pointing.sat_vx, pointing.sat_vy, pointing.sat_vz],
        project_cycle=self.overall_config["project_cycle"],
        run_counter=self.overall_config["run_counter"],
        chip_name=str(chip.chipID).rjust(2, '0'))
    self.h_ext = generateExtensionHeader(
        chip=chip,
        xlen=chip.npix_x, 
        ylen=chip.npix_y, 
        ra=pointing.ra, 
        dec=pointing.dec, 
        pa=pointing.img_pa.deg, 
        gain=chip.gain, 
        readout=chip.read_noise, 
        dark=chip.dark_noise, 
        saturation=90000, 
        pixel_scale=chip.pix_scale, 
        pixel_size=chip.pix_size,
        xcen=chip.x_cen,
        ycen=chip.y_cen,
        extName=pointing.pointing_type,
        timestamp = pointing.timestamp,
        exptime = pointing.exp_time,
        readoutTime = chip.readout_time,
        t_shutter_open = pointing.t_shutter_open, 
        t_shutter_close = pointing.t_shutter_close)

    return self.h_prim, self.h_ext

def updateHeaderInfo(self,header_flag='prim', keys = ['key'], values = [0]):
    if header_flag == 'prim':
        for key,value in zip(keys, values):
            self.h_prim[key] = value
    if header_flag == 'ext':
        for key,value in zip(keys, values):
            self.h_ext[key] = value