97 lines
2.8 KiB
Python
97 lines
2.8 KiB
Python
import enum
|
|
from typing import List
|
|
|
|
from spacepackets.ecss import PusTc
|
|
from tmtccmd.config import CmdTreeNode
|
|
from pydantic import BaseModel
|
|
from tmtccmd.tmtc import DefaultPusQueueHelper
|
|
|
|
from opssat_tmtc.common import EXPERIMENT_APID, UniqueId, make_action_cmd_header
|
|
|
|
|
|
class ActionId(enum.IntEnum):
|
|
DEFAULT_SINGLE = 1
|
|
BALANCED_SINGLE = 2
|
|
DEFAULT_SINGLE_FLATSAT = 3
|
|
BALANCED_SNGLE_FLATSAT = 4
|
|
CUSTOM_PARAMS = 5
|
|
|
|
|
|
class CameraParameters(BaseModel):
|
|
R: int
|
|
G: int
|
|
B: int
|
|
N: int
|
|
P: bool
|
|
E: int
|
|
W: int
|
|
|
|
|
|
def create_camera_node() -> CmdTreeNode:
|
|
cam_node = CmdTreeNode("cam", "OPS-SAT IMS1000 batch handler commands")
|
|
cam_node.add_child(
|
|
CmdTreeNode("default_single", "Default Single Image Camera Parameters")
|
|
)
|
|
cam_node.add_child(
|
|
CmdTreeNode("balanced_single", "Balanced Single Image Camera Parameters")
|
|
)
|
|
cam_node.add_child(
|
|
CmdTreeNode(
|
|
"default_single_flatsat",
|
|
"Default Single Image Camera Parameters for use on FlatSat",
|
|
)
|
|
)
|
|
cam_node.add_child(
|
|
CmdTreeNode(
|
|
"balanced_single_flatsat",
|
|
"Balanced Single Image Camera Parameters for use on FlatSat",
|
|
)
|
|
)
|
|
cam_node.add_child(
|
|
CmdTreeNode("custom_params", "Custom Camera Parameters as specified from file")
|
|
)
|
|
return cam_node
|
|
|
|
|
|
def create_cam_cmd(q: DefaultPusQueueHelper, cmd_path: List[str]):
|
|
|
|
assert len(cmd_path) >= 1
|
|
q.add_log_cmd(
|
|
"Sending PUS take image action request for command " + cmd_path[0] + " params."
|
|
)
|
|
data = bytearray()
|
|
if cmd_path[0] == "default_single":
|
|
data.extend(
|
|
make_action_cmd_header(UniqueId.CameraHandler, ActionId.DEFAULT_SINGLE)
|
|
)
|
|
elif cmd_path[0] == "balanced_single":
|
|
data.extend(
|
|
make_action_cmd_header(UniqueId.CameraHandler, ActionId.BALANCED_SINGLE)
|
|
)
|
|
elif cmd_path[0] == "default_single_flatsat":
|
|
data.extend(
|
|
make_action_cmd_header(
|
|
UniqueId.CameraHandler, ActionId.DEFAULT_SINGLE_FLATSAT
|
|
)
|
|
)
|
|
elif cmd_path[0] == "balanced_single_flatsat":
|
|
data.extend(
|
|
make_action_cmd_header(
|
|
UniqueId.CameraHandler, ActionId.BALANCED_SNGLE_FLATSAT
|
|
)
|
|
)
|
|
elif cmd_path[0] == "custom":
|
|
data.extend(
|
|
make_action_cmd_header(UniqueId.CameraHandler, ActionId.CUSTOM_PARAMS)
|
|
)
|
|
# TODO: Implement asking params from user.
|
|
|
|
# params = CameraParameters(8, 8, 8, 1, True, 200, 1000)
|
|
# data.extend(params.model_dump_json().encode())
|
|
raise NotImplementedError()
|
|
else:
|
|
raise ValueError("unknown camera action {}", cmd_path[0])
|
|
return q.add_pus_tc(
|
|
PusTc(service=8, subservice=128, apid=EXPERIMENT_APID, app_data=data)
|
|
)
|