

/*-
 * Copyright (c) 2005 Bruce D. Evans and Steven G. Kargl
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice unmodified, this list of conditions, and the following
 *    disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

/* adapted from FreeBSD:
 *    lib/msun/src/s_ccoshf.c
 */


#pragma once

#include <cusp/detail/thrust/complex.h>
#include <cusp/detail/thrust/detail/complex/math_private.h>

namespace thrust {
namespace detail {
namespace complex {

using thrust::complex;

__host__ __device__ inline
complex<float> ccoshf(const complex<float>& z) {
    float x, y, h;
    uint32_t hx, hy, ix, iy;
    const float huge = 1.70141183460469231731687303716e+38; //0x1p127;


    x = z.real();
    y = z.imag();

    get_float_word(hx, x);
    get_float_word(hy, y);

    ix = 0x7fffffff & hx;
    iy = 0x7fffffff & hy;
    if (ix < 0x7f800000 && iy < 0x7f800000) {
        if (iy == 0) {
            return (complex<float>(coshf(x), x * y));
        }
        if (ix < 0x41100000) {	/* small x: normal case */
            return (complex<float>(coshf(x) * cosf(y), sinhf(x) * sinf(y)));
        }
        /* |x| >= 9, so cosh(x) ~= exp(|x|) */
        if (ix < 0x42b17218) {
            /* x < 88.7: expf(|x|) won't overflow */
            h = expf(fabsf(x)) * 0.5f;
            return (complex<float>(h * cosf(y), copysignf(h, x) * sinf(y)));
        } else if (ix < 0x4340b1e7) {
            /* x < 192.7: scale to avoid overflow */
            thrust::complex<float> z_;
            z_ = ldexp_cexpf(complex<float>(fabsf(x), y), -1);
            return (complex<float>(z_.real(), z_.imag() * copysignf(1.0f, x)));
        } else {
            /* x >= 192.7: the result always overflows */
            h = huge * x;
            return (complex<float>(h * h * cosf(y), h * sinf(y)));
        }
    }

    if (ix == 0 && iy >= 0x7f800000) {
        return (complex<float>(y - y, copysignf(0.0f, x * (y - y))));
    }
    if (iy == 0 && ix >= 0x7f800000) {
        if ((hx & 0x7fffff) == 0)
            return (complex<float>(x * x, copysignf(0.0f, x) * y));
        return (complex<float>(x * x, copysignf(0.0f, (x + x) * y)));
    }

    if (ix < 0x7f800000 && iy >= 0x7f800000) {
        return (complex<float>(y - y, x * (y - y)));
    }

    if (ix >= 0x7f800000 && (hx & 0x7fffff) == 0) {
        if (iy >= 0x7f800000)
            return (complex<float>(x * x, x * (y - y)));
        return (complex<float>((x * x) * cosf(y), x * sinf(y)));
    }
    return (complex<float>((x * x) * (y - y), (x + x) * (y - y)));
}

__host__ __device__ inline
complex<float> ccosf(const complex<float>& z) {
    return (ccoshf(complex<float>(-z.imag(), z.real())));
}

} // namespace complex

} // namespace detail

template <>
__host__ __device__
inline complex<float> cos(const complex<float>& z) {
    return detail::complex::ccosf(z);
}

template <>
__host__ __device__
inline complex<float> cosh(const complex<float>& z) {
    return detail::complex::ccoshf(z);
}

} // namespace thrust
