89 lines
3.1 KiB
Python
89 lines
3.1 KiB
Python
import struct
|
|
import datetime
|
|
import math
|
|
|
|
from spacepackets.ecss import PusService, PusTelecommand
|
|
from tmtccmd.config import CoreServiceList
|
|
from tmtccmd.config.tmtc import (
|
|
OpCodeEntry,
|
|
TmtcDefinitionWrapper,
|
|
tmtc_definitions_provider,
|
|
)
|
|
from tmtccmd.pus.s11_tc_sched import create_time_tagged_cmd
|
|
from tmtccmd.pus.s17_test import create_service_17_ping_command
|
|
from tmtccmd.tmtc import service_provider
|
|
from tmtccmd.tmtc.decorator import ServiceProviderParams
|
|
|
|
|
|
class OpCodes:
|
|
PING = "ping"
|
|
TRIGGER_EVENT = "trig_event"
|
|
PING_WITH_DATA = "ping_with_data"
|
|
SCHEDULE_PING = "sched_ping"
|
|
|
|
|
|
class Info:
|
|
PING = "Simple Ping and Connection Test"
|
|
TRIGGER_EVENT = "Trigger an event"
|
|
PING_WITH_DATA = "Ping with data. Size of sent data is sent back"
|
|
SCHEDULE_PING = "Schedule a ping"
|
|
|
|
|
|
@tmtc_definitions_provider
|
|
def add_test_defs(defs: TmtcDefinitionWrapper):
|
|
oce = OpCodeEntry()
|
|
oce.add(keys=OpCodes.PING, info=Info.PING)
|
|
oce.add(keys=OpCodes.TRIGGER_EVENT, info=Info.TRIGGER_EVENT)
|
|
oce.add(keys=OpCodes.PING_WITH_DATA, info=Info.PING_WITH_DATA)
|
|
oce.add(keys=OpCodes.SCHEDULE_PING, info=Info.SCHEDULE_PING)
|
|
|
|
defs.add_service(
|
|
name=CoreServiceList.SERVICE_17_ALT,
|
|
info="PUS 17 Test Service",
|
|
op_code_entry=oce,
|
|
)
|
|
|
|
|
|
@service_provider(CoreServiceList.SERVICE_17_ALT)
|
|
def pack_test_command(p: ServiceProviderParams):
|
|
info = p.info
|
|
q = p.queue_helper
|
|
if info.op_code == OpCodes.PING:
|
|
q.add_log_cmd("Sending PUS TC [17,1]")
|
|
q.add_pus_tc(create_service_17_ping_command())
|
|
if info.op_code == OpCodes.TRIGGER_EVENT:
|
|
q.add_log_cmd("Sending PUS TC Event Trigger [17, 128]")
|
|
q.add_pus_tc(PusTelecommand(service=PusService.S17_TEST, subservice=128))
|
|
if info.op_code == OpCodes.SCHEDULE_PING:
|
|
q.add_log_cmd("Sending scheduled PUS ping")
|
|
# Generate a UNIX timestamp 30 seconds in the future using the datetime API with a UTC timezone
|
|
now = datetime.datetime.now(tz=datetime.timezone.utc)
|
|
second_offset_to_now = input("Please specify offset to now in seconds: ")
|
|
now += datetime.timedelta(seconds=int(second_offset_to_now))
|
|
unix_stamp = struct.pack("!I", math.floor(now.timestamp()))
|
|
print(f"Sending ping scheuled at {now}")
|
|
ping = PusTelecommand(service=PusService.S17_TEST, subservice=128)
|
|
q.add_pus_tc(
|
|
create_time_tagged_cmd(
|
|
release_time=unix_stamp,
|
|
tc_to_insert=ping,
|
|
)
|
|
)
|
|
if info.op_code == OpCodes.PING_WITH_DATA:
|
|
q.add_log_cmd("Sending Ping With Data, Size Reported Back [17, 129]")
|
|
while True:
|
|
data_size = int(input("Please specify data size [0-1024]: "))
|
|
if data_size < 0 or data_size > 1024:
|
|
print("Invalid data size")
|
|
break
|
|
dummy_data = bytearray()
|
|
next_byte = True
|
|
for i in range(data_size):
|
|
dummy_data.append(int(next_byte))
|
|
next_byte = not next_byte
|
|
q.add_pus_tc(
|
|
PusTelecommand(
|
|
service=PusService.S17_TEST, subservice=130, app_data=dummy_data
|
|
)
|
|
)
|