Source code for LOGS.Entities.TrackXYComplexData

from typing import Optional

from numpy import ndarray, vectorize

from LOGS.Entities.DatatrackNumericArray import DatatrackNumericArray
from LOGS.Entities.TrackData import TrackData




class TrackXYComplexData(TrackData):
    _x: Optional[DatatrackNumericArray] = None
    _re: Optional[DatatrackNumericArray] = None
    _im: Optional[DatatrackNumericArray] = None



    def fetchFull(self):
        if self.x:
            self.x.fetchFull()
        if self.re:
            self.re.fetchFull()
        if self.im:
            self.im.fetchFull()


    @property
    def x(self) -> Optional[DatatrackNumericArray]:
        return self._x

    @x.setter
    def x(self, value):
        self._x = self.checkAndConvertNullable(value, DatatrackNumericArray, "x")

    @property
    def re(self) -> Optional[DatatrackNumericArray]:
        return self._re

    @re.setter
    def re(self, value):
        self._re = self.checkAndConvertNullable(value, DatatrackNumericArray, "re")

    @property
    def im(self) -> Optional[DatatrackNumericArray]:
        return self._im

    @im.setter
    def im(self, value):
        self._im = self.checkAndConvertNullable(value, DatatrackNumericArray, "im")

    @property
    def y(self) -> Optional[ndarray]:
        if not self.re or not self.im:
            return None
        return vectorize(complex)(self.re.data, self.im.data)