/******************************************************************************
 * Project:  PROJ.4
 * Purpose:  Convert between ellipsoidal, geodetic coordinates and
 *           cartesian, geocentric coordinates.
 *
 *           Formally, this functionality is also found in the PJ_geocent.c
 *           code.
 *
 *           Actually, however, the PJ_geocent transformations are carried
 *           out in concert between 2D stubs in PJ_geocent.c and 3D code
 *           placed in pj_transform.c.
 *
 *           For pipeline-style datum shifts, we do need direct access
 *           to the full 3D interface for this functionality.
 *
 *           Hence this code, which may look like "just another PJ_geocent"
 *           but really is something substantially different.
 *
 * Author:   Thomas Knudsen, thokn@sdfe.dk
 *
 ******************************************************************************
 * Copyright (c) 2016, Thomas Knudsen / SDFE
 *
 * Permission is hereby granted, free of charge, to any person obtaining a
 * copy of this software and associated documentation files (the "Software"),
 * to deal in the Software without restriction, including without limitation
 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
 * and/or sell copies of the Software, and to permit persons to whom the
 * Software is furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included
 * in all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
 * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
 * DEALINGS IN THE SOFTWARE.
 *****************************************************************************/

#include "proj_internal.h"
#include <math.h>

PROJ_HEAD(cart, "Geodetic/cartesian conversions");

/**************************************************************
                CARTESIAN / GEODETIC CONVERSIONS
***************************************************************
    This material follows:

    Bernhard Hofmann-Wellenhof & Helmut Moritz:
    Physical Geodesy, 2nd edition.
    Springer, 2005.

    chapter 5.6: Coordinate transformations
    (HM, below),

    and

    Wikipedia: Geographic Coordinate Conversion,
    https://en.wikipedia.org/wiki/Geographic_coordinate_conversion

    (WP, below).

    The cartesian-to-geodetic conversion is based on Bowring's
    celebrated method:

    B. R. Bowring:
    Transformation from spatial to geographical coordinates
    Survey Review 23(181), pp. 323-327, 1976

    (BB, below),

    but could probably use some TLC from a newer and faster
    algorithm:

    Toshio Fukushima:
    Transformation from Cartesian to Geodetic Coordinates
    Accelerated by Halley’s Method
    Journal of Geodesy, February 2006

    (TF, below).

    Close to the poles, we avoid singularities by switching to an
    approximation requiring knowledge of the geocentric radius
    at the given latitude. For this, we use an adaptation of the
    formula given in:

    Wikipedia: Earth Radius
    https://en.wikipedia.org/wiki/Earth_radius#Radius_at_a_given_geodetic_latitude
    (Derivation and commentary at https://gis.stackexchange.com/q/20200)

    (WP2, below)

    These routines are probably not as robust at those in
    geocent.c, at least they haven't been through as heavy
    use as their geocent sisters. Some care has been taken
    to avoid singularities, but extreme cases (e.g. setting
    es, the squared eccentricity, to 1), will cause havoc.

**************************************************************/

/*********************************************************************/
static double normal_radius_of_curvature(double a, double es, double sinphi) {
    /*********************************************************************/
    if (es == 0)
        return a;
    /* This is from WP.  HM formula 2-149 gives an a,b version */
    return a / sqrt(1 - es * sinphi * sinphi);
}

/*********************************************************************/
static double geocentric_radius(double a, double b_div_a, double cosphi,
                                double sinphi) {
    /*********************************************************************
        Return the geocentric radius at latitude phi, of an ellipsoid
        with semimajor axis a and semiminor axis b.

        This is from WP2, but uses hypot() for potentially better
        numerical robustness
    ***********************************************************************/
    // Non-optimized version:
    // const double b = a * b_div_a;
    // return hypot(a * a * cosphi, b * b * sinphi) /
    //        hypot(a * cosphi, b * sinphi);
    const double cosphi_squared = cosphi * cosphi;
    const double sinphi_squared = sinphi * sinphi;
    const double b_div_a_squared = b_div_a * b_div_a;
    const double b_div_a_squared_mul_sinphi_squared =
        b_div_a_squared * sinphi_squared;
    return a * sqrt((cosphi_squared +
                     b_div_a_squared * b_div_a_squared_mul_sinphi_squared) /
                    (cosphi_squared + b_div_a_squared_mul_sinphi_squared));
}

/*********************************************************************/
static PJ_XYZ cartesian(PJ_LPZ geod, PJ *P) {
    /*********************************************************************/
    PJ_XYZ xyz;

    const double cosphi = cos(geod.phi);
    const double sinphi = sin(geod.phi);
    const double N = normal_radius_of_curvature(P->a, P->es, sinphi);

    /* HM formula 5-27 (z formula follows WP) */
    xyz.x = (N + geod.z) * cosphi * cos(geod.lam);
    xyz.y = (N + geod.z) * cosphi * sin(geod.lam);
    xyz.z = (N * (1 - P->es) + geod.z) * sinphi;

    return xyz;
}

/*********************************************************************/
static PJ_LPZ geodetic(PJ_XYZ cart, PJ *P) {
    /*********************************************************************/
    PJ_LPZ lpz;

    // Normalize (x,y,z) to the unit sphere/ellipsoid.
#if (defined(__i386__) && !defined(__SSE__)) || defined(_M_IX86)
    // i386 (actually non-SSE) code path to make following test case of
    // testvarious happy
    // "echo 6378137.00 -0.00 0.00 | bin/cs2cs +proj=geocent +datum=WGS84 +to
    // +proj=latlong +datum=WGS84"
    const double x_div_a = cart.x / P->a;
    const double y_div_a = cart.y / P->a;
    const double z_div_a = cart.z / P->a;
#else
    const double x_div_a = cart.x * P->ra;
    const double y_div_a = cart.y * P->ra;
    const double z_div_a = cart.z * P->ra;
#endif

    /* Perpendicular distance from point to Z-axis (HM eq. 5-28) */
    const double p_div_a = sqrt(x_div_a * x_div_a + y_div_a * y_div_a);

#if 0
    /* HM eq. (5-37) */
    const double theta  =  atan2 (cart.z * P->a,  p * P->b);

    /* HM eq. (5-36) (from BB, 1976) */
    const double c  =  cos(theta);
    const double s  =  sin(theta);
#else
    const double b_div_a = 1 - P->f; // = P->b / P->a
    const double p_div_a_b_div_a = p_div_a * b_div_a;
    const double norm =
        sqrt(z_div_a * z_div_a + p_div_a_b_div_a * p_div_a_b_div_a);
    double c, s;
    if (norm != 0) {
        const double inv_norm = 1.0 / norm;
        c = p_div_a_b_div_a * inv_norm;
        s = z_div_a * inv_norm;
    } else {
        c = 1;
        s = 0;
    }
#endif

    const double y_phi = z_div_a + P->e2s * b_div_a * s * s * s;
    const double x_phi = p_div_a - P->es * c * c * c;
    const double norm_phi = sqrt(y_phi * y_phi + x_phi * x_phi);
    double cosphi, sinphi;
    if (norm_phi != 0) {
        const double inv_norm_phi = 1.0 / norm_phi;
        cosphi = x_phi * inv_norm_phi;
        sinphi = y_phi * inv_norm_phi;
    } else {
        cosphi = 1;
        sinphi = 0;
    }
    if (x_phi <= 0) {
        // this happen on non-sphere ellipsoid when x,y,z is very close to 0
        // there is no single solution to the cart->geodetic conversion in
        // that case, clamp to -90/90 deg and avoid a discontinuous boundary
        // near the poles
        lpz.phi = cart.z >= 0 ? M_HALFPI : -M_HALFPI;
        cosphi = 0;
        sinphi = cart.z >= 0 ? 1 : -1;
    } else {
        lpz.phi = atan(y_phi / x_phi);
    }
    lpz.lam = atan2(y_div_a, x_div_a);

    if (cosphi < 1e-6) {
        /* poleward of 89.99994 deg, we avoid division by zero   */
        /* by computing the height as the cartesian z value      */
        /* minus the geocentric radius of the Earth at the given */
        /* latitude                                              */
        const double r = geocentric_radius(P->a, b_div_a, cosphi, sinphi);
        lpz.z = fabs(cart.z) - r;
    } else {
        const double N = normal_radius_of_curvature(P->a, P->es, sinphi);
        lpz.z = P->a * p_div_a / cosphi - N;
    }

    return lpz;
}

/* In effect, 2 cartesian coordinates of a point on the ellipsoid. Rather
 * pointless, but... */
static PJ_XY cart_forward(PJ_LP lp, PJ *P) {
    PJ_COORD point;
    point.lp = lp;
    point.lpz.z = 0;

    const auto xyz = cartesian(point.lpz, P);
    point.xyz = xyz;
    return point.xy;
}

/* And the other way round. Still rather pointless, but... */
static PJ_LP cart_reverse(PJ_XY xy, PJ *P) {
    PJ_COORD point;
    point.xy = xy;
    point.xyz.z = 0;

    const auto lpz = geodetic(point.xyz, P);
    point.lpz = lpz;
    return point.lp;
}

/*********************************************************************/
PJ *PJ_CONVERSION(cart, 1) {
    /*********************************************************************/
    P->fwd3d = cartesian;
    P->inv3d = geodetic;
    P->fwd = cart_forward;
    P->inv = cart_reverse;
    P->left = PJ_IO_UNITS_RADIANS;
    P->right = PJ_IO_UNITS_CARTESIAN;
    return P;
}