Allow internal I/O to support non-default kinds of CHARACTER. The I/O runtime design anticipated this standard feature, but this patch is somewhat larger than I thought it would be because many code sites had to have assumptions about units (characters vs. bytes) brought into harmony, and some encoding utilities had to be pulled out of IoStatementState and templatized into their own new header file so that they are available to formatted output code without having to "thread" an IoStatementState reference through many call chains. Differential Revision: https://reviews.llvm.org/D131107
57 lines
1.9 KiB
C++
57 lines
1.9 KiB
C++
//===-- runtime/internal-unit.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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
// Fortran internal I/O "units"
|
|
|
|
#ifndef FORTRAN_RUNTIME_IO_INTERNAL_UNIT_H_
|
|
#define FORTRAN_RUNTIME_IO_INTERNAL_UNIT_H_
|
|
|
|
#include "connection.h"
|
|
#include "flang/Runtime/descriptor.h"
|
|
#include <cinttypes>
|
|
#include <type_traits>
|
|
|
|
namespace Fortran::runtime::io {
|
|
|
|
class IoErrorHandler;
|
|
|
|
// Points to (but does not own) a CHARACTER scalar or array for internal I/O.
|
|
// Does not buffer.
|
|
template <Direction DIR> class InternalDescriptorUnit : public ConnectionState {
|
|
public:
|
|
using Scalar =
|
|
std::conditional_t<DIR == Direction::Input, const char *, char *>;
|
|
InternalDescriptorUnit(Scalar, std::size_t chars, int kind);
|
|
InternalDescriptorUnit(const Descriptor &, const Terminator &);
|
|
void EndIoStatement();
|
|
|
|
bool Emit(const char *, std::size_t, IoErrorHandler &);
|
|
std::size_t GetNextInputBytes(const char *&, IoErrorHandler &);
|
|
bool AdvanceRecord(IoErrorHandler &);
|
|
void BackspaceRecord(IoErrorHandler &);
|
|
|
|
private:
|
|
Descriptor &descriptor() { return staticDescriptor_.descriptor(); }
|
|
const Descriptor &descriptor() const {
|
|
return staticDescriptor_.descriptor();
|
|
}
|
|
Scalar CurrentRecord() const {
|
|
return descriptor().template ZeroBasedIndexedElement<char>(
|
|
currentRecordNumber - 1);
|
|
}
|
|
void BlankFill(char *, std::size_t);
|
|
void BlankFillOutputRecord();
|
|
|
|
StaticDescriptor<maxRank, true /*addendum*/> staticDescriptor_;
|
|
};
|
|
|
|
extern template class InternalDescriptorUnit<Direction::Output>;
|
|
extern template class InternalDescriptorUnit<Direction::Input>;
|
|
} // namespace Fortran::runtime::io
|
|
#endif // FORTRAN_RUNTIME_IO_INTERNAL_UNIT_H_
|