import numpy as np

from typing import Literal
from numpy.typing import NDArray

from spatz.simulations.csv_source import CSVSource


class ASTOSSource(CSVSource):
    def __init__(self, path: str, interpolation: Literal['linear'] = 'linear') -> None:
        super().__init__(path, 'time', interpolation)

    def get_length(self) -> float:
        pass

    def get_position(self) -> NDArray:
        return self.fetch_values(['x', 'y', 'z'])

    def get_velocity(self, frame: Literal['global', 'local']) -> NDArray:
        if frame == 'local':
            pass

        return self.fetch_values(['vx', 'vy', 'vz'])

    def get_acceleration(self, frame: Literal['global', 'local']) -> NDArray:
        if frame == 'local':
            pass

        return self.fetch_values(['ax', 'ay', 'az'])

    def get_attitude(self) -> NDArray:
        pass

    def local_to_global(self) -> NDArray:
        pass
    
    def global_to_local(self) -> NDArray:
        pass
    
    def get_angular_velocity(self) -> NDArray:
        pass
    
    def get_static_pressure(self) -> float:
        pass
    
    def get_longitude(self) -> float:
        pass
    
    def get_latitude(self) -> float:
        pass
    
    def get_altitude(self) -> float:
        pass