runtime/complex-reduction.c

/*===-- flang/runtime/complex-reduction.c ---------------------------*- C -*-===
 *
 * Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
 * See https://llvm.org/LICENSE.txt for license information.
 * SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
 *
 * ===-----------------------------------------------------------------------===
 */

#include "complex-reduction.h"
#include "flang/Common/long-double.h"
#include <complex.h>

/* These are the C standard's names for _Complex constructors; not all C
 * compilers support them.
 */
#if !defined(CMPLXF) && defined(__clang__)
#define CMPLXF __builtin_complex
#define CMPLX __builtin_complex
#define CMPLXL __builtin_complex
#endif

struct CppComplexFloat {
  float r, i;
};
struct CppComplexDouble {
  double r, i;
};
struct CppComplexLongDouble {
  long double r, i;
};

/* RTNAME(SumComplex4) calls RTNAME(CppSumComplex4) with the same arguments
 * and converts the members of its C++ complex result to C _Complex.
 */

#define CPP_NAME(name) Cpp##name
#define ADAPT_REDUCTION(name, cComplex, cpptype, cmplxMacro) \
  struct cpptype RTNAME(CPP_NAME(name))(struct cpptype *, REDUCTION_ARGS); \
  cComplex RTNAME(name)(REDUCTION_ARGS) { \
    struct cpptype result; \
    RTNAME(CPP_NAME(name))(&result, REDUCTION_ARG_NAMES); \
    return cmplxMacro(result.r, result.i); \
  }

/* TODO: COMPLEX(2 & 3) */

/* SUM() */
ADAPT_REDUCTION(SumComplex4, float_Complex_t, CppComplexFloat, CMPLXF)
ADAPT_REDUCTION(SumComplex8, double_Complex_t, CppComplexDouble, CMPLX)
#if LONG_DOUBLE == 80
ADAPT_REDUCTION(
    SumComplex10, long_double_Complex_t, CppComplexLongDouble, CMPLXL)
#elif LONG_DOUBLE == 128
ADAPT_REDUCTION(
    SumComplex16, long_double_Complex_t, CppComplexLongDouble, CMPLXL)
#endif

/* PRODUCT() */
ADAPT_REDUCTION(ProductComplex4, float_Complex_t, CppComplexFloat, CMPLXF)
ADAPT_REDUCTION(ProductComplex8, double_Complex_t, CppComplexDouble, CMPLX)
#if LONG_DOUBLE == 80
ADAPT_REDUCTION(
    ProductComplex10, long_double_Complex_t, CppComplexLongDouble, CMPLXL)
#elif LONG_DOUBLE == 128
ADAPT_REDUCTION(
    ProductComplex16, long_double_Complex_t, CppComplexLongDouble, CMPLXL)
#endif