geodetic-to-ecef.py

# SPDX-FileCopyrightText: Steven Ward
# SPDX-License-Identifier: OSL-3.0

# pylint: disable=invalid-name

'''
Read Geodetic coordinates from stdin and convert them to ECEF coordinates.

If 2 fields are given, they are interpreted as
    geodetic latitude (degrees)
    ellipsoid height (meters)

    The output coordinate represents
        W (meters)
        Z (meters)

If 3 fields are given, they are interpreted as
    geodetic latitude (degrees)
    geodetic longitude (degrees)
    ellipsoid height (meters)

    The output coordinate represents
        X (meters)
        Y (meters)
        Z (meters)
'''

__author__ = 'Steven Ward'
__license__ = 'OSL-3.0'

import sys

from ellipsoid import WGS84

for line in sys.stdin:
    fields = line.strip().split()

    if len(fields) == 2:
        lat_deg = float(fields[0])
        ht = float(fields[1])
        (w, z) = WGS84.geodetic_2d_to_ecef(lat_deg, ht)
        print(f"{w} {z}")

    elif len(fields) == 3:
        lat_deg = float(fields[0])
        lon_deg = float(fields[1])
        ht = float(fields[2])
        (x, y, z) = WGS84.geodetic_to_ecef(lat_deg, lon_deg, ht)
        print(f"{x} {y} {z}")