74a77dd2ed
Original-commit: flang-compiler/f18@57b0ebb558 Reviewed-on: https://github.com/flang-compiler/f18/pull/273
106 lines
3.5 KiB
C++
106 lines
3.5 KiB
C++
// Copyright (c) 2018, NVIDIA CORPORATION. All rights reserved.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
#ifndef FORTRAN_EVALUATE_COMPLEX_H_
|
|
#define FORTRAN_EVALUATE_COMPLEX_H_
|
|
|
|
#include "real.h"
|
|
#include <string>
|
|
|
|
namespace Fortran::evaluate::value {
|
|
|
|
template<typename REAL_TYPE> class Complex {
|
|
public:
|
|
using Part = REAL_TYPE;
|
|
static constexpr int bits{2 * Part::bits};
|
|
|
|
constexpr Complex() {} // (+0.0, +0.0)
|
|
constexpr Complex(const Complex &) = default;
|
|
constexpr Complex(const Part &r, const Part &i) : re_{r}, im_{i} {}
|
|
explicit constexpr Complex(const Part &r) : re_{r} {}
|
|
constexpr Complex &operator=(const Complex &) = default;
|
|
constexpr Complex &operator=(Complex &&) = default;
|
|
|
|
constexpr bool operator==(const Complex &that) const {
|
|
return re_ == that.re_ && im_ == that.im_;
|
|
}
|
|
|
|
constexpr const Part &REAL() const { return re_; }
|
|
constexpr const Part &AIMAG() const { return im_; }
|
|
constexpr Complex CONJG() const { return {re_, im_.Negate()}; }
|
|
constexpr Complex Negate() const { return {re_.Negate(), im_.Negate()}; }
|
|
|
|
constexpr bool Equals(const Complex &that) const {
|
|
return re_.Compare(that.re_) == Relation::Equal &&
|
|
im_.Compare(that.im_) == Relation::Equal;
|
|
}
|
|
|
|
constexpr bool IsZero() const { return re_.IsZero() || im_.IsZero(); }
|
|
|
|
constexpr bool IsInfinite() const {
|
|
return re_.IsInfinite() || im_.IsInfinite();
|
|
}
|
|
|
|
constexpr bool IsNotANumber() const {
|
|
return re_.IsNotANumber() || im_.IsNotANumber();
|
|
}
|
|
|
|
constexpr bool IsSignalingNaN() const {
|
|
return re_.IsSignalingNaN() || im_.IsSignalingNaN();
|
|
}
|
|
|
|
template<typename INT>
|
|
static ValueWithRealFlags<Complex> FromInteger(
|
|
const INT &n, Rounding rounding = defaultRounding) {
|
|
ValueWithRealFlags<Complex> result;
|
|
result.value.re_ =
|
|
Part::FromInteger(n, rounding).AccumulateFlags(result.flags);
|
|
return result;
|
|
}
|
|
|
|
ValueWithRealFlags<Complex> Add(
|
|
const Complex &, Rounding rounding = defaultRounding) const;
|
|
ValueWithRealFlags<Complex> Subtract(
|
|
const Complex &, Rounding rounding = defaultRounding) const;
|
|
ValueWithRealFlags<Complex> Multiply(
|
|
const Complex &, Rounding rounding = defaultRounding) const;
|
|
ValueWithRealFlags<Complex> Divide(
|
|
const Complex &, Rounding rounding = defaultRounding) const;
|
|
|
|
constexpr Complex FlushSubnormalToZero() const {
|
|
return {re_.FlushSubnormalToZero(), im_.FlushSubnormalToZero()};
|
|
}
|
|
|
|
static constexpr Complex NotANumber() {
|
|
return {Part::NotANumber(), Part::NotANumber()};
|
|
}
|
|
|
|
std::string DumpHexadecimal() const;
|
|
std::ostream &AsFortran(std::ostream &, int kind) const;
|
|
|
|
// TODO: (C)ABS once Real::HYPOT is done
|
|
// TODO: unit testing
|
|
|
|
private:
|
|
Part re_, im_;
|
|
};
|
|
|
|
extern template class Complex<Real<Integer<16>, 11>>;
|
|
extern template class Complex<Real<Integer<16>, 8>>;
|
|
extern template class Complex<Real<Integer<32>, 24>>;
|
|
extern template class Complex<Real<Integer<64>, 53>>;
|
|
extern template class Complex<Real<Integer<80>, 64, false>>;
|
|
extern template class Complex<Real<Integer<128>, 112>>;
|
|
}
|
|
#endif // FORTRAN_EVALUATE_COMPLEX_H_
|