import numpy as np
from observation_sim.instruments import Chip, Filter, FilterParam
import astropy.io.fits as fits
from observation_sim.mock_objects import FlatLED

# 输入打开的LED名字列表led_type_list和对应的打开时间列表exp_t_list


def get_LED_Img(chipID=8, led_type_list=['LED1'], exp_t_list=[0.1]):
    chip = Chip(chipID)
    filter_id, filter_type = chip.getChipFilter()
    filt = Filter(
        filter_id=filter_id,
        filter_type=filter_type,
        filter_param=FilterParam())
    led_obj = FlatLED(chip, filt)
    led_flat, ledstat, letts = led_obj.drawObj_LEDFlat(
        led_type_list=led_type_list, exp_t_list=exp_t_list)
    return led_flat


if __name__ == "__main__":
    chipid = 7
    led_type_list = ['LED5', 'LED6', 'LED7', 'LED8']
    exp_t_list = [0.1, 8, 9, 10]

    led_img = get_LED_Img(
        chipID=chipid, led_type_list=led_type_list, exp_t_list=exp_t_list)
    fits.writeto('test_led.fits', led_img)