from __future__ import annotations import time from typing import List from config.definitions import CustomServiceList from config.object_ids import get_object_ids from pus_tc.system.tcs import pack_tcs_sys_commands from tmtccmd.config import ( QueueCommands, ServiceOpCodeDictT, add_op_code_entry, add_service_op_code_entry, ) from tmtccmd.tc.definitions import TcQueueT from tmtccmd.tc.pus_11_tc_sched import ( generate_time_tagged_cmd, generate_enable_tc_sched_cmd, generate_reset_tc_sched_cmd, ) from tmtccmd.tc.pus_3_fsfw_hk import * import config.object_ids as oids from pus_tc.system.tcs import OpCodes as TcsOpCodes from pus_tc.devs.bpx_batt import BpxSetIds from pus_tc.system.core import SetIds as CoreSetIds from gomspace.gomspace_common import SetIds as GsSetIds from pus_tc.devs.rad_sensor import SetIds as RadSetIds from pus_tc.devs.mgms import MgmLis3SetIds as MgmLis3SetIds_0_2 from pus_tc.devs.mgms import MgmRm3100SetIds as MgmRm3100SetIds_1_3 from pus_tc.devs.gyros import AdisGyroSetIds as AdisGyroSetIds_0_2 from pus_tc.devs.gyros import L3gGyroSetIds as L3gGyroSetIds_1_3 from pus_tc.devs.gps import SetIds as GpsSetIds from pus_tc.devs.imtq import ImtqSetIds from pus_tc.devs.sus import SetIds from pus_tc.devs.star_tracker import SetIds as StrSetIds from pus_tc.devs.reaction_wheels import RwSetIds from pus_tc.system.controllers import pack_cmd_ctrl_to_off, pack_cmd_ctrl_to_nml from pus_tc.system.acs import pack_acs_command, pack_sus_cmds from pus_tc.devs.imtq import pack_imtq_test_into, pack_dipole_command from pus_tc.devs.star_tracker import pack_star_tracker_commands from pus_tc.devs.reaction_wheels import pack_rw_ass_cmds, pack_set_speed_command class OpCodes: TV_SETUP_TCS_FT_ON = ["s", "tcs-ft-on"] RESET_SCHED = ["reset-sched", "rs"] HEATER = ["heater"] BAT_FT = ["bat-ft"] CORE_FT = ["core-ft"] PCDU_FT = ["pcdu-ft"] RAD_SEN_FT = ["rad-sen-ft"] ACS_FT = ["acs-ft"] MGT_FT = ["mgt-ft"] MGT_FT_DP = ["mgt-ft-dp"] SUS_FT = ["sus-ft"] STR_FT = ["str-ft"] RW_FT_ONE_RW = ["rw-ft-one-rw"] RW_FT_TWO_RWS = ["rw-ft-two-rws"] TV_TEARDOWN_TCS_FT_OFF = ["t", "tcs-ft-off"] class KeyAndInfo: TV_SETUP_TCS_FT_ON = ["TCS Act.", "TCS functional test activation"] RESET_SCHED = ["Reset Sched", "Reset/Clear TC schedule"] HEATER = ["Heater", "heater procedure"] BAT_FT = ["BPX Battery", "battery functional test"] CORE_FT = ["OBC", "OBC functional test"] PCDU_FT = ["PCDU", "PCDU functional test"] RAD_SEN_FT = ["Radiation Sensor", "Radiation Sensor functional test"] ACS_FT = ["ACS", "ACS functional test"] MGT_FT = ["MGT", "MGT functional test"] MGT_FT_DP = ["MGT dipole", "MGT functional test with dipole"] SUS_FT = ["SUS", "SUS functional test"] STR_FT = ["STR", "STR functional test"] RW_FT_ONE_RW = ["RW ONE RW", "RW functional test with one RW"] RW_FT_TWO_RWS = ["RW TWO RWS", "RW functional test with two RWs"] TV_TEARDOWN_TCS_FT_OFF = ["TCS Deact.", "TCS functional test deactivation"] KAI = KeyAndInfo PROC_INFO_DICT = { KAI.TV_SETUP_TCS_FT_ON[0]: [OpCodes.TV_SETUP_TCS_FT_ON, KAI.TV_SETUP_TCS_FT_ON[1], 120.0, 10.0], KAI.TV_TEARDOWN_TCS_FT_OFF[0]: [ OpCodes.TV_TEARDOWN_TCS_FT_OFF, KAI.TV_TEARDOWN_TCS_FT_OFF[1], 120.0, 10.0, ], KAI.BAT_FT[0]: [OpCodes.BAT_FT, KAI.BAT_FT[1], 120.0, 10.0], KAI.CORE_FT[0]: [OpCodes.CORE_FT, KAI.CORE_FT[1], 120.0, 10.0], KAI.PCDU_FT[0]: [OpCodes.PCDU_FT, KAI.PCDU_FT[1], 120.0, 10.0], KAI.RAD_SEN_FT[0]: [OpCodes.RAD_SEN_FT, KAI.RAD_SEN_FT[1], 120.0, 10.0], KAI.ACS_FT[0]: [OpCodes.ACS_FT, KAI.ACS_FT[1], 120.0, 10.0], KAI.MGT_FT[0]: [OpCodes.MGT_FT, KAI.MGT_FT[1], 120.0, 10.0], # collection_time for KAI.MGT_FT_DP maybe be reduced as a full 120 # seconds is not needed after MGTs are tested KAI.MGT_FT_DP[0]: [OpCodes.MGT_FT_DP, KAI.MGT_FT_DP[1], 10.0, 10.0], KAI.SUS_FT[0]: [OpCodes.SUS_FT, KAI.SUS_FT[1], 120.0, 10.0], KAI.STR_FT[0]: [OpCodes.STR_FT, KAI.STR_FT[1], 120.0, 10.0], # collection_time for KAI.RW_FT_ONE_RW maybe be reduced as a full 120 # seconds is not needed after RWs are tested KAI.RW_FT_ONE_RW[0]: [OpCodes.RW_FT_ONE_RW, KAI.RW_FT_ONE_RW[1], 10.0, 1.0], # collection_time for KAI.RW_FT_ONE_RW maybe be reduced as a full 120 # seconds is not needed after RWs are tested KAI.RW_FT_TWO_RWS[0]: [OpCodes.RW_FT_TWO_RWS, KAI.RW_FT_TWO_RWS[1], 10.0, 1.0], } class GenericHkListeningCfg: def __init__(self, mgt: bool = False, one_rw: bool = False, two_rws: bool = False): self.use_tc_sched = False self.mgt = mgt self.one_rw = one_rw self.two_rws = two_rws @classmethod def default(cls) -> GenericHkListeningCfg: return GenericHkListeningCfg(False, False, False) def generic_print(tc_queue: TcQueueT, info: dict): tc_queue.appendleft( (QueueCommands.PRINT, f"Executing {info[1]} Procedure (OpCodes: {info[0]})") ) def add_proc_cmds(cmd_dict: ServiceOpCodeDictT): op_code_dict = dict() for proc_entry in PROC_INFO_DICT.values(): add_op_code_entry( op_code_dict=op_code_dict, keys=proc_entry[0], info=proc_entry[1] ) add_service_op_code_entry( srv_op_code_dict=cmd_dict, name=CustomServiceList.PROCEDURE.value, info="TV Test Procedures", op_code_entry=op_code_dict, ) def pack_generic_hk_listening_cmds( tc_queue: TcQueueT, proc_key: str, sid_list: list[bytearray], diag_list: list[bool], cfg: GenericHkListeningCfg, ): info = PROC_INFO_DICT[proc_key] collection_time = info[2] generic_print(tc_queue=tc_queue, info=info) for i in range(len(sid_list)): enable_listen_to_hk_for_x_seconds( diag=diag_list[i], tc_queue=tc_queue, device=proc_key, sid=sid_list[i], interval_seconds=info[3], ) if not cfg.use_tc_sched: tc_queue.appendleft((QueueCommands.WAIT, 2.0)) if cfg.mgt: activate_mgts_alternately( tc_queue=tc_queue, ) elif cfg.one_rw: activate_all_rws_in_sequence( tc_queue=tc_queue, test_speed=20000, test_ramp_time=10000, init_ssc=0 ) elif cfg.two_rws: activate_all_rws_two_consecutively(tc_queue=tc_queue, init_ssc=0) else: pass if not cfg.use_tc_sched: tc_queue.appendleft((QueueCommands.WAIT, collection_time)) disable_cmd_list = [] for i in range(len(sid_list)): disable_cmd_list.append( gen_disable_listen_to_hk_for_x_seconds( diag=diag_list[i], tc_queue=tc_queue, device=proc_key, sid=sid_list[i], ) ) current_time = time.time() current_time += collection_time if not cfg.use_tc_sched: for cmd in disable_cmd_list: tc_queue.appendleft(cmd.pack_command_tuple()) else: for cmd in disable_cmd_list: tc_queue.appendleft( generate_time_tagged_cmd( release_time=struct.pack("!I", current_time), tc_to_insert=cmd, ssc=0, ) ) sid_list.clear() diag_list.clear() def pack_proc_commands(tc_queue: TcQueueT, op_code: str): sid_list = [] obj_id_dict = get_object_ids() if op_code in OpCodes.RESET_SCHED: tc_queue.appendleft((QueueCommands.PRINT, "Resetting/Clearing TC schedule")) tc_queue.appendleft(generate_reset_tc_sched_cmd().pack_command_tuple()) if op_code in OpCodes.BAT_FT: key = KAI.BAT_FT[0] sid_list.append(make_sid(oids.BPX_HANDLER_ID, BpxSetIds.GET_HK_SET)) diag_list = [False] pack_generic_hk_listening_cmds( tc_queue=tc_queue, proc_key=key, sid_list=sid_list, diag_list=diag_list, cfg=GenericHkListeningCfg.default(), ) if op_code in OpCodes.CORE_FT: key = KAI.CORE_FT[0] sid_list.append(make_sid(oids.CORE_CONTROLLER_ID, CoreSetIds.HK)) diag_list = [False] pack_generic_hk_listening_cmds( tc_queue=tc_queue, proc_key=key, sid_list=sid_list, diag_list=diag_list, cfg=GenericHkListeningCfg.default(), ) if op_code in OpCodes.PCDU_FT: key = KAI.PCDU_FT[0] pcdu_pairs = [ (oids.P60_DOCK_HANDLER, GsSetIds.P60_CORE), (oids.PDU_1_HANDLER_ID, GsSetIds.PDU_1_CORE), (oids.PDU_2_HANDLER_ID, GsSetIds.PDU_2_CORE), (oids.ACU_HANDLER_ID, GsSetIds.ACU_CORE), (oids.P60_DOCK_HANDLER, GsSetIds.P60_AUX), (oids.PDU_1_HANDLER_ID, GsSetIds.PDU_1_AUX), (oids.PDU_2_HANDLER_ID, GsSetIds.PDU_2_AUX), (oids.ACU_HANDLER_ID, GsSetIds.ACU_AUX), ] diag_list = [ False, True, True, True, False, False, False, False, ] for pcdu_dev in pcdu_pairs: oid = pcdu_dev[0] set_id = pcdu_dev[1] sid_list.append(make_sid(oid, set_id)) pack_generic_hk_listening_cmds( tc_queue=tc_queue, proc_key=key, sid_list=sid_list, diag_list=diag_list, cfg=GenericHkListeningCfg.default(), ) if op_code in OpCodes.RAD_SEN_FT: key = KAI.RAD_SEN_FT[0] sid_list.append(make_sid(oids.RAD_SENSOR_ID, RadSetIds.HK)) diag_list = [False] pack_generic_hk_listening_cmds( tc_queue=tc_queue, proc_key=key, sid_list=sid_list, diag_list=diag_list, cfg=GenericHkListeningCfg.default(), ) if op_code in OpCodes.TV_SETUP_TCS_FT_ON: # Enable scheduling tc_queue.appendleft(generate_enable_tc_sched_cmd().pack_command_tuple()) # check whether tcs_assembly also has to be commanded to NORMAL Mode pack_tcs_sys_commands( tc_queue=tc_queue, op_code=TcsOpCodes.TCS_BOARD_ASS_NORMAL[0] ) pack_cmd_ctrl_to_nml( tc_queue=tc_queue, object_id=obj_id_dict.get(oids.THERMAL_CONTROLLER_ID) ) if op_code in OpCodes.TV_TEARDOWN_TCS_FT_OFF: # TCS board should always be on anyway, do not command it off here pack_cmd_ctrl_to_off( tc_queue=tc_queue, object_id=obj_id_dict.get(oids.THERMAL_CONTROLLER_ID) ) if op_code in OpCodes.ACS_FT: key = KAI.ACS_FT[0] a_side_pairs = [ (oids.MGM_0_LIS3_HANDLER_ID, MgmLis3SetIds_0_2.CORE_HK), (oids.MGM_1_RM3100_HANDLER_ID, MgmRm3100SetIds_1_3.CORE_HK), (oids.GYRO_0_ADIS_HANDLER_ID, AdisGyroSetIds_0_2.CORE_HK), (oids.GYRO_1_L3G_HANDLER_ID, L3gGyroSetIds_1_3.CORE_HK), (oids.GPS_CONTROLLER, GpsSetIds.HK), ] b_side_pairs = [ (oids.MGM_2_LIS3_HANDLER_ID, MgmLis3SetIds_0_2.CORE_HK), (oids.MGM_3_RM3100_HANDLER_ID, MgmRm3100SetIds_1_3.CORE_HK), (oids.GYRO_2_ADIS_HANDLER_ID, AdisGyroSetIds_0_2.CORE_HK), (oids.GYRO_3_L3G_HANDLER_ID, L3gGyroSetIds_1_3.CORE_HK), (oids.GPS_CONTROLLER, GpsSetIds.HK), ] d_side_pairs = a_side_pairs + b_side_pairs diag_list = [ False, False, False, False, False, ] pack_acs_command(tc_queue=tc_queue, op_code="acs-a") for a_side_dev in a_side_pairs: oid = a_side_dev[0] set_id = a_side_dev[1] sid_list.append(make_sid(oid, set_id)) pack_generic_hk_listening_cmds( tc_queue=tc_queue, proc_key=key, sid_list=sid_list, diag_list=diag_list, cfg=GenericHkListeningCfg.default(), ) pack_acs_command(tc_queue=tc_queue, op_code="acs-off") tc_queue.appendleft((QueueCommands.WAIT, 5.0)) pack_acs_command(tc_queue=tc_queue, op_code="acs-b") for b_side_dev in b_side_pairs: oid = b_side_dev[0] set_id = b_side_dev[1] sid_list.append(make_sid(oid, set_id)) pack_generic_hk_listening_cmds( tc_queue=tc_queue, proc_key=key, sid_list=sid_list, diag_list=diag_list, cfg=GenericHkListeningCfg.default(), ) pack_acs_command(tc_queue=tc_queue, op_code="acs-off") tc_queue.appendleft((QueueCommands.WAIT, 5.0)) pack_acs_command(tc_queue=tc_queue, op_code="acs-d") for d_side_dev in d_side_pairs: oid = d_side_dev[0] set_id = d_side_dev[1] sid_list.append(make_sid(oid, set_id)) diag_list = diag_list + diag_list pack_generic_hk_listening_cmds( tc_queue=tc_queue, proc_key=key, sid_list=sid_list, diag_list=diag_list, cfg=GenericHkListeningCfg.default(), ) pack_acs_command(tc_queue=tc_queue, op_code="acs-off") if op_code in OpCodes.MGT_FT: key = KAI.MGT_FT[0] imtq_pairs = [ (oids.IMTQ_HANDLER_ID, ImtqSetIds.ENG_HK_SET), (oids.IMTQ_HANDLER_ID, ImtqSetIds.CAL_MTM_SET), (oids.IMTQ_HANDLER_ID, ImtqSetIds.RAW_MTM_SET), ] diag_list = [ False, False, False, ] # mgt 1: imtq und hk pack_imtq_test_into(oids.IMTQ_HANDLER_ID, tc_queue=tc_queue, op_code="2") for imtq_dev in imtq_pairs: oid = imtq_dev[0] set_id = imtq_dev[1] sid_list.append(make_sid(oid, set_id)) pack_generic_hk_listening_cmds( tc_queue=tc_queue, proc_key=key, sid_list=sid_list, diag_list=diag_list, cfg=GenericHkListeningCfg.default() ) pack_imtq_test_into(oids.IMTQ_HANDLER_ID, tc_queue=tc_queue, op_code="0") if op_code in OpCodes.MGT_FT_DP: key = KAI.MGT_FT_DP[0] a_side_pairs = [ (oids.MGM_0_LIS3_HANDLER_ID, MgmLis3SetIds_0_2.CORE_HK), (oids.MGM_1_RM3100_HANDLER_ID, MgmRm3100SetIds_1_3.CORE_HK), (oids.GYRO_0_ADIS_HANDLER_ID, AdisGyroSetIds_0_2.CORE_HK), (oids.GYRO_1_L3G_HANDLER_ID, L3gGyroSetIds_1_3.CORE_HK), (oids.GPS_CONTROLLER, GpsSetIds.HK), ] b_side_pairs = [ (oids.MGM_2_LIS3_HANDLER_ID, MgmLis3SetIds_0_2.CORE_HK), (oids.MGM_3_RM3100_HANDLER_ID, MgmRm3100SetIds_1_3.CORE_HK), (oids.GYRO_2_ADIS_HANDLER_ID, AdisGyroSetIds_0_2.CORE_HK), (oids.GYRO_3_L3G_HANDLER_ID, L3gGyroSetIds_1_3.CORE_HK), (oids.GPS_CONTROLLER, GpsSetIds.HK), ] imtq_pairs = [ (oids.IMTQ_HANDLER_ID, ImtqSetIds.ENG_HK_SET), (oids.IMTQ_HANDLER_ID, ImtqSetIds.CAL_MTM_SET), (oids.IMTQ_HANDLER_ID, ImtqSetIds.RAW_MTM_SET), ] d_side_and_imtq_pairs = a_side_pairs + b_side_pairs + imtq_pairs diag_list = [ False, False, False, False, False, False, False, False, False, False, False, False, False, ] pack_acs_command(tc_queue=tc_queue, op_code="acs-d") # mgt 2.: imtq + dual side + dipole pack_imtq_test_into( object_id=oids.IMTQ_HANDLER_ID, tc_queue=tc_queue, op_code="2" ) for d_side_and_imtq_dev in d_side_and_imtq_pairs: oid = d_side_and_imtq_dev[0] set_id = d_side_and_imtq_dev[1] sid_list.append(make_sid(oid, set_id)) cfg = GenericHkListeningCfg.default() cfg.mgt = True pack_generic_hk_listening_cmds( tc_queue=tc_queue, proc_key=key, sid_list=sid_list, diag_list=diag_list, cfg=cfg, ) pack_imtq_test_into(oids.IMTQ_HANDLER_ID, tc_queue=tc_queue, op_code="0") pack_acs_command(tc_queue=tc_queue, op_code="acs-off") if op_code in OpCodes.SUS_FT: key = KAI.SUS_FT[0] pack_sus_cmds(tc_queue=tc_queue, op_code="sus-nom") sus_n_ids = [ oids.SUS_0_N_LOC_XFYFZM_PT_XF, oids.SUS_1_N_LOC_XBYFZM_PT_XB, oids.SUS_2_N_LOC_XFYBZB_PT_YB, oids.SUS_3_N_LOC_XFYBZF_PT_YF, oids.SUS_4_N_LOC_XMYFZF_PT_ZF, oids.SUS_5_N_LOC_XFYMZB_PT_ZB, ] sus_r_ids = [ oids.SUS_6_R_LOC_XFYBZM_PT_XF, oids.SUS_7_R_LOC_XBYBZM_PT_XB, oids.SUS_8_R_LOC_XBYBZB_PT_YB, oids.SUS_9_R_LOC_XBYBZB_PT_YF, oids.SUS_10_R_LOC_XMYBZF_PT_ZF, oids.SUS_11_R_LOC_XBYMZB_PT_ZB, ] diag_list = [ False, False, False, False, False, False, ] # SUSs for nom_sus in sus_n_ids: sid_list.append(make_sid(nom_sus, SetIds.HK)) pack_generic_hk_listening_cmds( tc_queue=tc_queue, proc_key=key, sid_list=sid_list, diag_list=diag_list, cfg=GenericHkListeningCfg.default(), ) pack_acs_command(tc_queue=tc_queue, op_code="sus-off") tc_queue.appendleft((QueueCommands.WAIT, 5.0)) pack_sus_cmds(tc_queue=tc_queue, op_code="sus-red") # SUSs for red_sus in sus_r_ids: sid_list.append(make_sid(red_sus, SetIds.HK)) pack_generic_hk_listening_cmds( tc_queue=tc_queue, proc_key=key, sid_list=sid_list, diag_list=diag_list, cfg=GenericHkListeningCfg.default(), ) pack_acs_command(tc_queue=tc_queue, op_code="sus-off") tc_queue.appendleft((QueueCommands.WAIT, 5.0)) pack_sus_cmds(tc_queue=tc_queue, op_code="sus-d") # SUSs for nom_sus in sus_n_ids: sid_list.append(make_sid(nom_sus, SetIds.HK)) for red_sus in sus_r_ids: sid_list.append(make_sid(red_sus, SetIds.HK)) diag_list = diag_list + diag_list pack_generic_hk_listening_cmds( tc_queue=tc_queue, proc_key=key, sid_list=sid_list, diag_list=diag_list, cfg=GenericHkListeningCfg.default(), ) pack_acs_command(tc_queue=tc_queue, op_code="sus-off") if op_code in OpCodes.STR_FT: key = KAI.STR_FT[0] pack_star_tracker_commands( object_id=oids.STAR_TRACKER_ID, tc_queue=tc_queue, op_code="2" ) # STR sid_list.append(make_sid(oids.STAR_TRACKER_ID, StrSetIds.TEMPERATURE)) diag_list = [False] pack_generic_hk_listening_cmds( tc_queue=tc_queue, proc_key=key, sid_list=sid_list, diag_list=diag_list, cfg=GenericHkListeningCfg.default(), ) pack_star_tracker_commands( object_id=oids.STAR_TRACKER_ID, tc_queue=tc_queue, op_code="3" ) if op_code in OpCodes.RW_FT_ONE_RW: key = KAI.RW_FT_ONE_RW[0] rw_pairs = [ (oids.RW1_ID, RwSetIds.STATUS_SET_ID), (oids.RW1_ID, RwSetIds.TEMPERATURE_SET_ID), (oids.RW1_ID, RwSetIds.LAST_RESET), (oids.RW1_ID, RwSetIds.TM_SET), (oids.RW2_ID, RwSetIds.STATUS_SET_ID), (oids.RW2_ID, RwSetIds.TEMPERATURE_SET_ID), (oids.RW2_ID, RwSetIds.LAST_RESET), (oids.RW2_ID, RwSetIds.TM_SET), (oids.RW3_ID, RwSetIds.STATUS_SET_ID), (oids.RW3_ID, RwSetIds.TEMPERATURE_SET_ID), (oids.RW3_ID, RwSetIds.LAST_RESET), (oids.RW3_ID, RwSetIds.TM_SET), (oids.RW4_ID, RwSetIds.STATUS_SET_ID), (oids.RW4_ID, RwSetIds.TEMPERATURE_SET_ID), (oids.RW4_ID, RwSetIds.LAST_RESET), (oids.RW4_ID, RwSetIds.TM_SET), ] diag_list = [ False, False, False, False, False, False, False, False, False, False, False, False, False, False, False, False, ] # RW NORMAL pack_rw_ass_cmds(object_id=oids.RW_ASSEMBLY, tc_queue=tc_queue, op_code="nml") # RW HK für alle RWs nur einzeln for rw_dev in rw_pairs: oid = rw_dev[0] set_id = rw_dev[1] sid_list.append(make_sid(oid, set_id)) pack_generic_hk_listening_cmds( tc_queue=tc_queue, proc_key=key, sid_list=sid_list, diag_list=diag_list, cfg=GenericHkListeningCfg(mgt=False, one_rw=True, two_rws=False), ) # RW OFF pack_rw_ass_cmds(object_id=oids.RW_ASSEMBLY, tc_queue=tc_queue, op_code="off") # ass command with 2 rws to speed if op_code in OpCodes.RW_FT_TWO_RWS: key = KAI.RW_FT_TWO_RWS[0] rw_pairs = [ (oids.RW1_ID, RwSetIds.STATUS_SET_ID), (oids.RW1_ID, RwSetIds.TEMPERATURE_SET_ID), (oids.RW1_ID, RwSetIds.LAST_RESET), (oids.RW1_ID, RwSetIds.TM_SET), (oids.RW2_ID, RwSetIds.STATUS_SET_ID), (oids.RW2_ID, RwSetIds.TEMPERATURE_SET_ID), (oids.RW2_ID, RwSetIds.LAST_RESET), (oids.RW2_ID, RwSetIds.TM_SET), (oids.RW3_ID, RwSetIds.STATUS_SET_ID), (oids.RW3_ID, RwSetIds.TEMPERATURE_SET_ID), (oids.RW3_ID, RwSetIds.LAST_RESET), (oids.RW3_ID, RwSetIds.TM_SET), (oids.RW4_ID, RwSetIds.STATUS_SET_ID), (oids.RW4_ID, RwSetIds.TEMPERATURE_SET_ID), (oids.RW4_ID, RwSetIds.LAST_RESET), (oids.RW4_ID, RwSetIds.TM_SET), ] diag_list = [ False, False, False, False, False, False, False, False, False, False, False, False, False, False, False, False, ] # RW NORMAL pack_rw_ass_cmds(object_id=oids.RW_ASSEMBLY, tc_queue=tc_queue, op_code="nml") # RW for rw_dev in rw_pairs: oid = rw_dev[0] set_id = rw_dev[1] sid_list.append(make_sid(oid, set_id)) pack_generic_hk_listening_cmds( tc_queue=tc_queue, proc_key=key, sid_list=sid_list, diag_list=diag_list, cfg=GenericHkListeningCfg(mgt=False, one_rw=False, two_rws=True), ) # RW OFF pack_rw_ass_cmds(object_id=oids.RW_ASSEMBLY, tc_queue=tc_queue, op_code="off") def enable_listen_to_hk_for_x_seconds( tc_queue: TcQueueT, diag: bool, device: str, sid: bytes, interval_seconds: float, ): tc_queue.appendleft((QueueCommands.PRINT, f"Enabling periodic HK for {device}")) cmd_tuple = enable_periodic_hk_command_with_interval( diag=diag, sid=sid, interval_seconds=interval_seconds, ssc=0 ) for cmd in cmd_tuple: tc_queue.appendleft(cmd.pack_command_tuple()) def gen_disable_listen_to_hk_for_x_seconds( tc_queue: TcQueueT, diag: bool, device: str, sid: bytes, ) -> PusTelecommand: tc_queue.appendleft((QueueCommands.PRINT, f"Disabling periodic HK for {device}")) return disable_periodic_hk_command(diag=diag, sid=sid, ssc=0) def activate_mgts_alternately( tc_queue: TcQueueT, ): command = pack_dipole_command( object_id=oids.IMTQ_HANDLER_ID, x_dipole=2000, y_dipole=0, z_dipole=0, duration=30000, ) tc_queue.appendleft(command.pack_command_tuple()) tc_queue.appendleft((QueueCommands.WAIT, 40.0)) command = pack_dipole_command( object_id=oids.IMTQ_HANDLER_ID, x_dipole=-2000, y_dipole=0, z_dipole=0, duration=30000, ) tc_queue.appendleft(command.pack_command_tuple()) tc_queue.appendleft((QueueCommands.WAIT, 40.0)) command = pack_dipole_command( object_id=oids.IMTQ_HANDLER_ID, x_dipole=0, y_dipole=2000, z_dipole=0, duration=30000, ) tc_queue.appendleft(command.pack_command_tuple()) tc_queue.appendleft((QueueCommands.WAIT, 40.0)) command = pack_dipole_command( object_id=oids.IMTQ_HANDLER_ID, x_dipole=0, y_dipole=-2000, z_dipole=0, duration=30000, ) tc_queue.appendleft(command.pack_command_tuple()) tc_queue.appendleft((QueueCommands.WAIT, 40.0)) command = pack_dipole_command( object_id=oids.IMTQ_HANDLER_ID, x_dipole=0, y_dipole=0, z_dipole=2000, duration=30000, ) tc_queue.appendleft(command.pack_command_tuple()) tc_queue.appendleft((QueueCommands.WAIT, 40.0)) command = pack_dipole_command( object_id=oids.IMTQ_HANDLER_ID, x_dipole=0, y_dipole=0, z_dipole=-2000, duration=30000, ) tc_queue.appendleft(command.pack_command_tuple()) tc_queue.appendleft((QueueCommands.WAIT, 40.0)) def rw_speed_cmd_single( tc_queue: TcQueueT, oid: bytes, init_ssc: int, speed: int, ramp_time: int ) -> int: command = pack_set_speed_command( object_id=oid, speed=speed, ramp_time_ms=ramp_time, ssc=init_ssc ) init_ssc += 1 tc_queue.appendleft(command.pack_command_tuple()) tc_queue.appendleft((QueueCommands.WAIT, 70.0)) command = pack_set_speed_command( object_id=oid, speed=0, ramp_time_ms=ramp_time, ssc=init_ssc ) tc_queue.appendleft(command.pack_command_tuple()) return init_ssc + 1 def rw_speed_up_cmd_consec( tc_queue: TcQueueT, obids: List[bytes], init_ssc: int, speed: int, ramp_time: int ) -> int: for oid in obids: command = pack_set_speed_command( object_id=oid, speed=speed, ramp_time_ms=ramp_time, ssc=init_ssc ) tc_queue.appendleft(command.pack_command_tuple()) init_ssc += 1 return init_ssc def rw_speed_down_cmd_consec( tc_queue: TcQueueT, obids: List[bytes], init_ssc: int, ramp_time: int ) -> int: for oid in obids: command = pack_set_speed_command( object_id=oid, speed=0, ramp_time_ms=ramp_time, ssc=init_ssc ) tc_queue.appendleft(command.pack_command_tuple()) init_ssc += 1 return init_ssc def activate_all_rws_in_sequence( tc_queue: TcQueueT, init_ssc: int, test_speed: int, test_ramp_time: int ) -> int: new_ssc = init_ssc # RW1 speed cmd tc_queue.appendleft((QueueCommands.WAIT, 5.0)) new_ssc = rw_speed_cmd_single( tc_queue, oids.RW1_ID, new_ssc, test_speed, test_ramp_time ) tc_queue.appendleft((QueueCommands.WAIT, 15.0)) new_ssc = rw_speed_cmd_single( tc_queue, oids.RW2_ID, new_ssc, test_speed, test_ramp_time ) tc_queue.appendleft((QueueCommands.WAIT, 15.0)) new_ssc = rw_speed_cmd_single( tc_queue, oids.RW3_ID, new_ssc, test_speed, test_ramp_time ) tc_queue.appendleft((QueueCommands.WAIT, 15.0)) new_ssc = rw_speed_cmd_single( tc_queue, oids.RW4_ID, new_ssc, test_speed, test_ramp_time ) tc_queue.appendleft((QueueCommands.WAIT, 15.0)) return new_ssc def activate_all_rws_two_consecutively(tc_queue: TcQueueT, init_ssc: int) -> int: new_ssc = init_ssc # RW1+3 speed cmd tc_queue.appendleft((QueueCommands.WAIT, 5.0)) new_ssc = rw_speed_up_cmd_consec( tc_queue, [oids.RW1_ID, oids.RW3_ID], new_ssc, -20000, 10000 ) tc_queue.appendleft((QueueCommands.WAIT, 70.0)) new_ssc = rw_speed_down_cmd_consec( tc_queue, [oids.RW1_ID, oids.RW3_ID], new_ssc, 10000 ) tc_queue.appendleft((QueueCommands.WAIT, 15.0)) # RW2+4 speed cmd new_ssc = rw_speed_up_cmd_consec( tc_queue, [oids.RW2_ID, oids.RW4_ID], new_ssc, -20000, 10000 ) tc_queue.appendleft((QueueCommands.WAIT, 70.0)) new_ssc = rw_speed_down_cmd_consec( tc_queue, [oids.RW2_ID, oids.RW4_ID], new_ssc, 10000 ) tc_queue.appendleft((QueueCommands.WAIT, 15.0)) return new_ssc