flang/runtime/connection.cpp

   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