5566d70ef2
Original-commit: flang-compiler/f18@8227229f62 Reviewed-on: https://github.com/flang-compiler/f18/pull/183 Tree-same-pre-rewrite: false
83 lines
2.7 KiB
C++
83 lines
2.7 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_TOOLS_H_
|
|
#define FORTRAN_EVALUATE_TOOLS_H_
|
|
|
|
#include "expression.h"
|
|
#include "../common/idioms.h"
|
|
#include "../parser/message.h"
|
|
#include <optional>
|
|
#include <utility>
|
|
|
|
namespace Fortran::evaluate {
|
|
|
|
// Convert the second argument to the same type and kind of the first.
|
|
Expr<SomeReal> ConvertToTypeOf(
|
|
const Expr<SomeReal> &to, const Expr<SomeInteger> &from);
|
|
Expr<SomeReal> ConvertToTypeOf(
|
|
const Expr<SomeReal> &to, const Expr<SomeReal> &from);
|
|
|
|
// Ensure that both operands of an intrinsic REAL operation or CMPLX()
|
|
// are INTEGER or REAL, and convert them as necessary to the same REAL type.
|
|
using ConvertRealOperandsResult =
|
|
std::optional<std::pair<Expr<SomeReal>, Expr<SomeReal>>>;
|
|
ConvertRealOperandsResult ConvertRealOperands(
|
|
parser::ContextualMessages &, GenericExpr &&, GenericExpr &&);
|
|
|
|
template<TypeCategory CAT>
|
|
void ConvertToSameKind(Expr<SomeKind<CAT>> &x, Expr<SomeKind<CAT>> &y) {
|
|
std::visit(
|
|
[&](auto &xk, auto &yk) {
|
|
using xt = ResultType<decltype(xk)>;
|
|
using yt = ResultType<decltype(yk)>;
|
|
if constexpr (xt::kind < yt::kind) {
|
|
x.u = Expr<yt>{xk};
|
|
} else if constexpr (xt::kind > yt::kind) {
|
|
y.u = Expr<xt>{yk};
|
|
}
|
|
},
|
|
x.u, y.u);
|
|
}
|
|
|
|
template<typename A> Expr<ScalarValueType<A>> ScalarConstantToExpr(const A &x) {
|
|
return {x};
|
|
}
|
|
|
|
template<TypeCategory CAT, int KIND>
|
|
Expr<SomeKind<CAT>> ToSomeKindExpr(const Expr<Type<CAT, KIND>> &x) {
|
|
return Expr<SomeKind<CAT>>{x};
|
|
}
|
|
|
|
template<TypeCategory CAT>
|
|
Expr<SomeKind<CAT>> SomeKindScalarToExpr(const SomeKindScalar<CAT> &x) {
|
|
return std::visit(
|
|
[](const auto &c) { return ToSomeKindExpr(ScalarConstantToExpr(c)); },
|
|
x.u);
|
|
}
|
|
|
|
Expr<SomeType> GenericScalarToExpr(const Scalar<SomeType> &);
|
|
|
|
template<TypeCategory CAT>
|
|
Expr<SomeType> ToGenericExpr(const Expr<SomeKind<CAT>> &x) {
|
|
return Expr<SomeType>{x};
|
|
}
|
|
|
|
template<TypeCategory CAT>
|
|
Expr<SomeType> ToGenericExpr(Expr<SomeKind<CAT>> &&x) {
|
|
return Expr<SomeType>{std::move(x)};
|
|
}
|
|
|
|
} // namespace Fortran::evaluate
|
|
#endif // FORTRAN_EVALUATE_TOOLS_H_
|