| 1 | //===-- runtime/connection.cpp --------------------------------------------===// |
|---|
| 2 | // |
|---|
| 3 | // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
|---|
| 4 | // See https://llvm.org/LICENSE.txt for license information. |
|---|
| 5 | // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
|---|
| 6 | // |
|---|
| 7 | //===----------------------------------------------------------------------===// |
|---|
| 8 | |
|---|
| 9 | #include "connection.h" |
|---|
| 10 | #include "environment.h" |
|---|
| 11 | #include "io-stmt.h" |
|---|
| 12 | #include <algorithm> |
|---|
| 13 | |
|---|
| 14 | namespace Fortran::runtime::io { |
|---|
| 15 | RT_OFFLOAD_API_GROUP_BEGIN |
|---|
| 16 | |
|---|
| 17 | RT_API_ATTRS std::size_t ConnectionState::RemainingSpaceInRecord() const { |
|---|
| 18 | auto recl{recordLength.value_or(openRecl.value_or( |
|---|
| 19 | executionEnvironment.listDirectedOutputLineLengthLimit))}; |
|---|
| 20 | return positionInRecord >= recl ? 0 : recl - positionInRecord; |
|---|
| 21 | } |
|---|
| 22 | |
|---|
| 23 | RT_API_ATTRS bool ConnectionState::NeedAdvance(std::size_t width) const { |
|---|
| 24 | return positionInRecord > 0 && width > RemainingSpaceInRecord(); |
|---|
| 25 | } |
|---|
| 26 | |
|---|
| 27 | RT_API_ATTRS bool ConnectionState::IsAtEOF() const { |
|---|
| 28 | return endfileRecordNumber && currentRecordNumber >= *endfileRecordNumber; |
|---|
| 29 | } |
|---|
| 30 | |
|---|
| 31 | RT_API_ATTRS bool ConnectionState::IsAfterEndfile() const { |
|---|
| 32 | return endfileRecordNumber && currentRecordNumber > *endfileRecordNumber; |
|---|
| 33 | } |
|---|
| 34 | |
|---|
| 35 | RT_API_ATTRS void ConnectionState::HandleAbsolutePosition(std::int64_t n) { |
|---|
| 36 | positionInRecord = std::max(n, std::int64_t{0}) + leftTabLimit.value_or(0); |
|---|
| 37 | } |
|---|
| 38 | |
|---|
| 39 | RT_API_ATTRS void ConnectionState::HandleRelativePosition(std::int64_t n) { |
|---|
| 40 | positionInRecord = std::max(leftTabLimit.value_or(0), positionInRecord + n); |
|---|
| 41 | } |
|---|
| 42 | |
|---|
| 43 | SavedPosition::SavedPosition(IoStatementState &io) : io_{io} { |
|---|
| 44 | ConnectionState &conn{io_.GetConnectionState()}; |
|---|
| 45 | saved_ = conn; |
|---|
| 46 | conn.pinnedFrame = true; |
|---|
| 47 | } |
|---|
| 48 | |
|---|
| 49 | SavedPosition::~SavedPosition() { |
|---|
| 50 | if (!cancelled_) { |
|---|
| 51 | ConnectionState &conn{io_.GetConnectionState()}; |
|---|
| 52 | while (conn.currentRecordNumber > saved_.currentRecordNumber) { |
|---|
| 53 | io_.BackspaceRecord(); |
|---|
| 54 | } |
|---|
| 55 | conn.leftTabLimit = saved_.leftTabLimit; |
|---|
| 56 | conn.furthestPositionInRecord = saved_.furthestPositionInRecord; |
|---|
| 57 | conn.positionInRecord = saved_.positionInRecord; |
|---|
| 58 | conn.pinnedFrame = saved_.pinnedFrame; |
|---|
| 59 | } |
|---|
| 60 | } |
|---|
| 61 | |
|---|
| 62 | RT_OFFLOAD_API_GROUP_END |
|---|
| 63 | } // namespace Fortran::runtime::io |
|---|
| 64 | |
|---|
