forked from OSchip/llvm-project
116 lines
4.3 KiB
C++
116 lines
4.3 KiB
C++
//===-- runtime/edit-output.h -----------------------------------*- 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef FORTRAN_RUNTIME_EDIT_OUTPUT_H_
|
|
#define FORTRAN_RUNTIME_EDIT_OUTPUT_H_
|
|
|
|
// Output data editing templates implementing the FORMAT data editing
|
|
// descriptors E, EN, ES, EX, D, F, and G for REAL data (and COMPLEX
|
|
// components, I and G for INTEGER, and B/O/Z for both.
|
|
// See subclauses in 13.7.2.3 of Fortran 2018 for the
|
|
// detailed specifications of these descriptors.
|
|
// List-directed output (13.10.4) for numeric types is also done here.
|
|
// Drives the same fast binary-to-decimal formatting templates used
|
|
// in the f18 front-end.
|
|
|
|
#include "format.h"
|
|
#include "io-stmt.h"
|
|
#include "flang/Common/uint128.h"
|
|
#include "flang/Decimal/decimal.h"
|
|
|
|
namespace Fortran::runtime::io {
|
|
|
|
// I, B, O, Z, and G output editing for INTEGER.
|
|
// The DataEdit reference is const here (and elsewhere in this header) so that
|
|
// one edit descriptor with a repeat factor may safely serve to edit
|
|
// multiple elements of an array.
|
|
template <typename INT = std::int64_t, typename UINT = std::uint64_t>
|
|
bool EditIntegerOutput(IoStatementState &, const DataEdit &, INT);
|
|
|
|
// Encapsulates the state of a REAL output conversion.
|
|
class RealOutputEditingBase {
|
|
protected:
|
|
explicit RealOutputEditingBase(IoStatementState &io) : io_{io} {}
|
|
|
|
static bool IsInfOrNaN(const decimal::ConversionToDecimalResult &res) {
|
|
const char *p{res.str};
|
|
if (!p || res.length < 1) {
|
|
return false;
|
|
}
|
|
if (*p == '-' || *p == '+') {
|
|
if (res.length == 1) {
|
|
return false;
|
|
}
|
|
++p;
|
|
}
|
|
return *p < '0' || *p > '9';
|
|
}
|
|
|
|
const char *FormatExponent(int, const DataEdit &edit, int &length);
|
|
bool EmitPrefix(const DataEdit &, std::size_t length, std::size_t width);
|
|
bool EmitSuffix(const DataEdit &);
|
|
|
|
IoStatementState &io_;
|
|
int trailingBlanks_{0}; // created when Gw editing maps to Fw
|
|
char exponent_[16];
|
|
};
|
|
|
|
template <int KIND> class RealOutputEditing : public RealOutputEditingBase {
|
|
public:
|
|
static constexpr int binaryPrecision{common::PrecisionOfRealKind(KIND)};
|
|
using BinaryFloatingPoint =
|
|
decimal::BinaryFloatingPointNumber<binaryPrecision>;
|
|
template <typename A>
|
|
RealOutputEditing(IoStatementState &io, A x)
|
|
: RealOutputEditingBase{io}, x_{x} {}
|
|
bool Edit(const DataEdit &);
|
|
|
|
private:
|
|
// The DataEdit arguments here are const references or copies so that
|
|
// the original DataEdit can safely serve multiple array elements when
|
|
// it has a repeat count.
|
|
bool EditEorDOutput(const DataEdit &);
|
|
bool EditFOutput(const DataEdit &);
|
|
DataEdit EditForGOutput(DataEdit); // returns an E or F edit
|
|
bool EditEXOutput(const DataEdit &);
|
|
bool EditListDirectedOutput(const DataEdit &);
|
|
|
|
bool IsZero() const { return x_.IsZero(); }
|
|
|
|
decimal::ConversionToDecimalResult Convert(
|
|
int significantDigits, const DataEdit &, int flags = 0);
|
|
|
|
BinaryFloatingPoint x_;
|
|
char buffer_[BinaryFloatingPoint::maxDecimalConversionDigits +
|
|
EXTRA_DECIMAL_CONVERSION_SPACE];
|
|
};
|
|
|
|
bool ListDirectedLogicalOutput(
|
|
IoStatementState &, ListDirectedStatementState<Direction::Output> &, bool);
|
|
bool EditLogicalOutput(IoStatementState &, const DataEdit &, bool);
|
|
bool ListDirectedDefaultCharacterOutput(IoStatementState &,
|
|
ListDirectedStatementState<Direction::Output> &, const char *, std::size_t);
|
|
bool EditDefaultCharacterOutput(
|
|
IoStatementState &, const DataEdit &, const char *, std::size_t);
|
|
|
|
extern template bool EditIntegerOutput<std::int64_t, std::uint64_t>(
|
|
IoStatementState &, const DataEdit &, std::int64_t);
|
|
extern template bool EditIntegerOutput<common::uint128_t, common::uint128_t>(
|
|
IoStatementState &, const DataEdit &, common::uint128_t);
|
|
|
|
extern template class RealOutputEditing<2>;
|
|
extern template class RealOutputEditing<3>;
|
|
extern template class RealOutputEditing<4>;
|
|
extern template class RealOutputEditing<8>;
|
|
extern template class RealOutputEditing<10>;
|
|
// TODO: double/double
|
|
extern template class RealOutputEditing<16>;
|
|
|
|
} // namespace Fortran::runtime::io
|
|
#endif // FORTRAN_RUNTIME_EDIT_OUTPUT_H_
|