[flang][runtime] Complete list-directed character input with DECIMAL='COMMA'

Most of the infrastructure for DECIMAL='COMMA' mode was in place
in the I/O runtime support library, but I dropped the ball for
list-directed character input, which has its own detection of
input separators.  Finish the job.

Differential Revision: https://reviews.llvm.org/D129679

2 files changed, 18 insertions, 5 deletions

diff --git a/flang/runtime/edit-input.cpp b/flang/runtime/edit-input.cpp
index 5be6b07..f2ee328 100644
--- a/flang/runtime/edit-input.cpp
+++ b/flang/runtime/edit-input.cpp
@@ -654,15 +654,25 @@ static bool EditListDirectedCharacterInput(
   // in NextInField.
   std::optional<int> remaining{length > 0 ? maxUTF8Bytes : 0};
   while (std::optional<char32_t> next{io.NextInField(remaining, edit)}) {
+    bool isSep{false};
     switch (*next) {
     case ' ':
     case '\t':
+    case '/':
+      isSep = true;
+      break;
     case ',':
+      isSep = !(edit.modes.editingFlags & decimalComma);
+      break;
     case ';':
-    case '/':
-      remaining = 0; // value separator: stop
+      isSep = !!(edit.modes.editingFlags & decimalComma);
       break;
     default:
+      break;
+    }
+    if (isSep) {
+      remaining = 0;
+    } else {
       *x++ = *next;
       remaining = --length > 0 ? maxUTF8Bytes : 0;
     }
diff --git a/flang/runtime/io-stmt.cpp b/flang/runtime/io-stmt.cpp
index 96f5bb9..66b5e9c 100644
--- a/flang/runtime/io-stmt.cpp
+++ b/flang/runtime/io-stmt.cpp
@@ -630,7 +630,6 @@ std::optional<char32_t> IoStatementState::NextInField(
         switch (*next) {
         case ' ':
         case '\t':
-        case ';':
         case '/':
         case '(':
         case ')':
@@ -640,11 +639,15 @@ std::optional<char32_t> IoStatementState::NextInField(
         case '\n': // for stream access
           return std::nullopt;
         case ',':
+          if (!(edit.modes.editingFlags & decimalComma)) {
+            return std::nullopt;
+          }
+          break;
+        case ';':
           if (edit.modes.editingFlags & decimalComma) {
-            break;
-          } else {
             return std::nullopt;
           }
+          break;
         default:
           break;
         }