complex.cpp
3.9 KB
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
94
95
96
97
98
99
100
101
102
103
104
105
106
//===-- lib/Evaluate/complex.cpp ------------------------------------------===//
//
// 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 "flang/Evaluate/complex.h"
#include "llvm/Support/raw_ostream.h"
namespace Fortran::evaluate::value {
template <typename R>
ValueWithRealFlags<Complex<R>> Complex<R>::Add(
const Complex &that, Rounding rounding) const {
RealFlags flags;
Part reSum{re_.Add(that.re_, rounding).AccumulateFlags(flags)};
Part imSum{im_.Add(that.im_, rounding).AccumulateFlags(flags)};
return {Complex{reSum, imSum}, flags};
}
template <typename R>
ValueWithRealFlags<Complex<R>> Complex<R>::Subtract(
const Complex &that, Rounding rounding) const {
RealFlags flags;
Part reDiff{re_.Subtract(that.re_, rounding).AccumulateFlags(flags)};
Part imDiff{im_.Subtract(that.im_, rounding).AccumulateFlags(flags)};
return {Complex{reDiff, imDiff}, flags};
}
template <typename R>
ValueWithRealFlags<Complex<R>> Complex<R>::Multiply(
const Complex &that, Rounding rounding) const {
// (a + ib)*(c + id) -> ac - bd + i(ad + bc)
RealFlags flags;
Part ac{re_.Multiply(that.re_, rounding).AccumulateFlags(flags)};
Part bd{im_.Multiply(that.im_, rounding).AccumulateFlags(flags)};
Part ad{re_.Multiply(that.im_, rounding).AccumulateFlags(flags)};
Part bc{im_.Multiply(that.re_, rounding).AccumulateFlags(flags)};
Part acbd{ac.Subtract(bd, rounding).AccumulateFlags(flags)};
Part adbc{ad.Add(bc, rounding).AccumulateFlags(flags)};
return {Complex{acbd, adbc}, flags};
}
template <typename R>
ValueWithRealFlags<Complex<R>> Complex<R>::Divide(
const Complex &that, Rounding rounding) const {
// (a + ib)/(c + id) -> [(a+ib)*(c-id)] / [(c+id)*(c-id)]
// -> [ac+bd+i(bc-ad)] / (cc+dd)
// -> ((ac+bd)/(cc+dd)) + i((bc-ad)/(cc+dd))
// but to avoid overflows, scale by d/c if c>=d, else c/d
Part scale; // <= 1.0
RealFlags flags;
bool cGEd{that.re_.ABS().Compare(that.im_.ABS()) != Relation::Less};
if (cGEd) {
scale = that.im_.Divide(that.re_, rounding).AccumulateFlags(flags);
} else {
scale = that.re_.Divide(that.im_, rounding).AccumulateFlags(flags);
}
Part den;
if (cGEd) {
Part dS{scale.Multiply(that.im_, rounding).AccumulateFlags(flags)};
den = dS.Add(that.re_, rounding).AccumulateFlags(flags);
} else {
Part cS{scale.Multiply(that.re_, rounding).AccumulateFlags(flags)};
den = cS.Add(that.im_, rounding).AccumulateFlags(flags);
}
Part aS{scale.Multiply(re_, rounding).AccumulateFlags(flags)};
Part bS{scale.Multiply(im_, rounding).AccumulateFlags(flags)};
Part re1, im1;
if (cGEd) {
re1 = re_.Add(bS, rounding).AccumulateFlags(flags);
im1 = im_.Subtract(aS, rounding).AccumulateFlags(flags);
} else {
re1 = aS.Add(im_, rounding).AccumulateFlags(flags);
im1 = bS.Subtract(re_, rounding).AccumulateFlags(flags);
}
Part re{re1.Divide(den, rounding).AccumulateFlags(flags)};
Part im{im1.Divide(den, rounding).AccumulateFlags(flags)};
return {Complex{re, im}, flags};
}
template <typename R> std::string Complex<R>::DumpHexadecimal() const {
std::string result{'('};
result += re_.DumpHexadecimal();
result += ',';
result += im_.DumpHexadecimal();
result += ')';
return result;
}
template <typename R>
llvm::raw_ostream &Complex<R>::AsFortran(llvm::raw_ostream &o, int kind) const {
re_.AsFortran(o << '(', kind);
im_.AsFortran(o << ',', kind);
return o << ')';
}
template class Complex<Real<Integer<16>, 11>>;
template class Complex<Real<Integer<16>, 8>>;
template class Complex<Real<Integer<32>, 24>>;
template class Complex<Real<Integer<64>, 53>>;
template class Complex<Real<Integer<80>, 64>>;
template class Complex<Real<Integer<128>, 113>>;
} // namespace Fortran::evaluate::value