81 lines
2.2 KiB
Python
81 lines
2.2 KiB
Python
from dataclasses import dataclass
|
|
from functools import cache, cached_property
|
|
import re
|
|
from typing import Collection, Mapping, Self, Sequence
|
|
|
|
from utils import lin_fun_interpolate, with_ops_scalar, with_ops_unary, with_ops_vector
|
|
|
|
MPH = 0.44704
|
|
|
|
@with_ops_unary('neg')
|
|
@with_ops_vector('sub', 'add')
|
|
@with_ops_scalar('mul', 'truediv')
|
|
@dataclass(frozen=True)
|
|
class PropOppoint:
|
|
speed: float
|
|
thrust: float
|
|
torque: float
|
|
rpm: float
|
|
|
|
def __neg__(self) -> Self: ...
|
|
|
|
def __truediv__(self, other: float) -> Self: ...
|
|
def __mul__(self, other: float) -> Self: ...
|
|
|
|
def __add__(self, other: Self) -> Self: ...
|
|
def __sub__(self, other: Self) -> Self: ...
|
|
|
|
@dataclass(frozen=True)
|
|
class ApcPerfdata:
|
|
oppoints: Collection[PropOppoint]
|
|
|
|
@cached_property
|
|
def rpms(self) -> Collection[float]:
|
|
return { op.rpm for op in self.oppoints }
|
|
|
|
def _rpm_serie(self, rpm: float) -> Sequence[PropOppoint]:
|
|
return sorted([op for op in self.oppoints if op.rpm == rpm], key=lambda op: op.speed)
|
|
|
|
@cached_property
|
|
def rpm_series(self) -> Mapping[float, Sequence[PropOppoint]]:
|
|
return { rpm: self._rpm_serie(rpm) for rpm in self.rpms }
|
|
|
|
def get_op_interpspeed(self, rpm: float, speed: float) -> PropOppoint:
|
|
s = self.rpm_series[rpm]
|
|
def mk_op(op):
|
|
return lambda: op
|
|
return lin_fun_interpolate([op.speed for op in s], [mk_op(op) for op in s], speed)
|
|
|
|
|
|
@classmethod
|
|
def from_file(cls, path: str):
|
|
with open(path, 'r') as file:
|
|
cont = file.read()
|
|
data = parse_apc_data(cont)
|
|
return cls(data)
|
|
|
|
|
|
def parse_apc_data(raw_data: str) -> Collection[PropOppoint]:
|
|
data_points = []
|
|
current_rpm = None
|
|
|
|
for line in raw_data.splitlines():
|
|
if 'PROP RPM' in line:
|
|
current_rpm = float(line.split()[-1])
|
|
elif current_rpm is None:
|
|
continue
|
|
try:
|
|
vals = list(map(float, line.split()))
|
|
except ValueError:
|
|
continue
|
|
if len(vals) != 15:
|
|
continue
|
|
|
|
speed = vals[0] * MPH
|
|
thrust = vals[9]
|
|
torque = vals[10]
|
|
data_points.append(PropOppoint(speed, thrust, torque, current_rpm))
|
|
|
|
return data_points
|
|
|