summaryrefslogtreecommitdiff
path: root/lib/Numerical/Complex.cpp
blob: ccc9dc695684e2a8d2b426b788dff564d20bec05 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
#include <complex>
#include "Numerical/Complex.h"

using namespace std;

static inline Complex8 cmplx(complex<double> Value) {
  Complex8 Result = { Value.real(), Value.imag() };
  return Result;
}

static inline Complex4 cmplx(complex<float> Value) {
  Complex4 Result = { Value.real(), Value.imag() };
  return Result;
}

static inline complex<double> generic(double Re, double Im) {
  return complex<double>(Re, Im);
}

static inline complex<float> generic(float Re, float Im) {
  return complex<float>(Re, Im);
}

LIBFLANG_ABI Complex8 libflang_cpow(double LHSRe, double LHSIm,
                                    double RHSRe, double RHSIm) {
  return cmplx(pow(generic(LHSRe, LHSIm),
                   generic(RHSRe, RHSIm)));
}
LIBFLANG_ABI Complex4 libflang_cpowf(float LHSRe, float LHSIm,
                                     float RHSRe, float RHSIm) {
  return cmplx(pow(generic(LHSRe, LHSIm),
                   generic(RHSRe, RHSIm)));
}

LIBFLANG_ABI Complex8 libflang_cpowi(double Re, double Im,
                                     int32_t power) {
  return cmplx(pow(generic(Re, Im), power));
}
LIBFLANG_ABI Complex4 libflang_cpowif(float Re, float Im,
                                      int32_t power) {
  return cmplx(complex<float>(pow(generic(Re, Im), power)));
}

// intrinsics

LIBFLANG_ABI double libflang_cabs(double Re, double Im) {
  return abs(generic(Re, Im));
}
LIBFLANG_ABI float libflang_cabsf(float Re, float Im) {
  return abs(generic(Re, Im));
}

LIBFLANG_ABI Complex8 libflang_csqrt(double Re, double Im) {
  return cmplx(sqrt(generic(Re, Im)));
}
LIBFLANG_ABI Complex4 libflang_csqrtf(float Re, float Im) {
  return cmplx(sqrt(generic(Re, Im)));
}

LIBFLANG_ABI Complex8 libflang_cexp(double Re, double Im) {
  return cmplx(exp(generic(Re, Im)));
}
LIBFLANG_ABI Complex4 libflang_cexpf(float Re, float Im) {
  return cmplx(exp(generic(Re, Im)));
}

LIBFLANG_ABI Complex8 libflang_clog(double Re, double Im) {
  return cmplx(log(generic(Re, Im)));
}
LIBFLANG_ABI Complex4 libflang_clogf(float Re, float Im) {
  return cmplx(log(generic(Re, Im)));
}

LIBFLANG_ABI Complex8 libflang_csin(double Re, double Im) {
  return cmplx(sin(generic(Re, Im)));
}
LIBFLANG_ABI Complex4 libflang_csinf(float Re, float Im) {
  return cmplx(sin(generic(Re, Im)));
}

LIBFLANG_ABI Complex8 libflang_ccos(double Re, double Im) {
  return cmplx(cos(generic(Re, Im)));
}
LIBFLANG_ABI Complex4 libflang_ccosf(float Re, float Im) {
  return cmplx(cos(generic(Re, Im)));
}

LIBFLANG_ABI Complex8 libflang_ctan(double Re, double Im) {
  return cmplx(tan(generic(Re, Im)));
}
LIBFLANG_ABI Complex4 libflang_ctanf(float Re, float Im) {
  return cmplx(tan(generic(Re, Im)));
}