added GPR
This commit is contained in:
@@ -4,8 +4,13 @@ from __future__ import annotations
|
||||
|
||||
from typing import Any
|
||||
|
||||
from python_app.models.run_config_schema import ComboModel, RunConfigModel
|
||||
from python_app.models.run_config_validation import load_ring_payload, load_switch_payload
|
||||
from python_app.models.run_config_schema import (
|
||||
ComboModel,
|
||||
GprRxGeometryModel,
|
||||
GprTxGeometryModel,
|
||||
RunConfigModel,
|
||||
)
|
||||
from python_app.models.run_config_validation import load_ring_payload, load_switch_payload, validate_gpr_model
|
||||
|
||||
|
||||
def _as_dict(value: Any, context: str) -> dict[str, Any]:
|
||||
@@ -29,6 +34,7 @@ def run_config_from_dict(payload: dict[str, Any]) -> RunConfigModel:
|
||||
port2_payload = _as_dict(switches_payload.get("port2"), "switches.port2")
|
||||
run_payload = _as_dict(payload.get("run"), "run")
|
||||
preprocess_payload = _as_dict(payload.get("preprocess"), "preprocess")
|
||||
gpr_payload = _as_dict(payload.get("gpr"), "gpr")
|
||||
rings_payload = _as_dict(payload.get("rings"), "rings")
|
||||
raw_ring_payload = _as_dict(rings_payload.get("raw"), "rings.raw")
|
||||
raw_tap_ring_payload = _as_dict(rings_payload.get("raw_tap"), "rings.raw_tap")
|
||||
@@ -68,6 +74,38 @@ def run_config_from_dict(payload: dict[str, Any]) -> RunConfigModel:
|
||||
preprocess_payload.get("reference_bundle_path", model.preprocess.reference_bundle_path)
|
||||
)
|
||||
|
||||
model.gpr.mode = str(gpr_payload.get("mode", model.gpr.mode))
|
||||
model.gpr.relative_permittivity = float(
|
||||
gpr_payload.get("relative_permittivity", model.gpr.relative_permittivity)
|
||||
)
|
||||
model.gpr.tx_geometry = []
|
||||
tx_geometry_payload = gpr_payload.get("tx_geometry", [])
|
||||
if isinstance(tx_geometry_payload, list):
|
||||
for entry in tx_geometry_payload:
|
||||
entry_payload = _as_dict(entry, "gpr.tx_geometry[]")
|
||||
model.gpr.tx_geometry.append(
|
||||
GprTxGeometryModel(
|
||||
output_pos=int(entry_payload.get("output_pos", 0)),
|
||||
x_m=float(entry_payload.get("x_m", 0.0)),
|
||||
)
|
||||
)
|
||||
model.gpr.rx_geometry = []
|
||||
rx_geometry_payload = gpr_payload.get("rx_geometry", [])
|
||||
if isinstance(rx_geometry_payload, list):
|
||||
for entry in rx_geometry_payload:
|
||||
entry_payload = _as_dict(entry, "gpr.rx_geometry[]")
|
||||
model.gpr.rx_geometry.append(
|
||||
GprRxGeometryModel(
|
||||
input_pos=int(entry_payload.get("input_pos", 0)),
|
||||
x_m=float(entry_payload.get("x_m", 0.0)),
|
||||
)
|
||||
)
|
||||
validate_gpr_model(
|
||||
model.gpr,
|
||||
input_switch_positions=model.input_switch.positions,
|
||||
output_switch_positions=model.output_switch.positions,
|
||||
)
|
||||
|
||||
load_ring_payload(raw_ring_payload, model.rings.raw)
|
||||
load_ring_payload(raw_tap_ring_payload, model.rings.raw_tap)
|
||||
load_ring_payload(pre_ring_payload, model.rings.preprocessed)
|
||||
@@ -145,6 +183,24 @@ def run_config_to_dict(model: RunConfigModel) -> dict[str, Any]:
|
||||
"calibration_bundle_path": model.preprocess.calibration_bundle_path,
|
||||
"reference_bundle_path": model.preprocess.reference_bundle_path,
|
||||
},
|
||||
"gpr": {
|
||||
"mode": model.gpr.mode,
|
||||
"relative_permittivity": model.gpr.relative_permittivity,
|
||||
"tx_geometry": [
|
||||
{
|
||||
"output_pos": entry.output_pos,
|
||||
"x_m": entry.x_m,
|
||||
}
|
||||
for entry in model.gpr.tx_geometry
|
||||
],
|
||||
"rx_geometry": [
|
||||
{
|
||||
"input_pos": entry.input_pos,
|
||||
"x_m": entry.x_m,
|
||||
}
|
||||
for entry in model.gpr.rx_geometry
|
||||
],
|
||||
},
|
||||
"rings": {
|
||||
"raw": {
|
||||
"name": model.rings.raw.name,
|
||||
|
||||
Reference in New Issue
Block a user