[flang-commits] [flang] 0c37529 - [flang] COMMAND_ARGUMENT_COUNT runtime implementation
Diana Picus via flang-commits
flang-commits at lists.llvm.org
Fri Sep 3 00:30:16 PDT 2021
Author: Diana Picus
Date: 2021-09-03T07:29:39Z
New Revision: 0c375296cc771af7279427e942a207e3cfe7695f
URL: https://github.com/llvm/llvm-project/commit/0c375296cc771af7279427e942a207e3cfe7695f
DIFF: https://github.com/llvm/llvm-project/commit/0c375296cc771af7279427e942a207e3cfe7695f.diff
LOG: [flang] COMMAND_ARGUMENT_COUNT runtime implementation
Grab whatever ProgramStart has stored in executionEnvironment.argc and
subtract 1 (based on the assumption that ProgramStart is called with
a C-style argc that counts the command name as an argument).
Spoiler alert: The tests will evolve into fixtures when we implement
GET_COMMAND_ARGUMENT etc.
Differential Revision: https://reviews.llvm.org/D109048
Added:
flang/runtime/command.cpp
flang/unittests/Runtime/CommandTest.cpp
Modified:
flang/runtime/CMakeLists.txt
flang/unittests/Runtime/CMakeLists.txt
Removed:
################################################################################
diff --git a/flang/runtime/CMakeLists.txt b/flang/runtime/CMakeLists.txt
index bc5c2d1e27643..b3e96faede706 100644
--- a/flang/runtime/CMakeLists.txt
+++ b/flang/runtime/CMakeLists.txt
@@ -35,6 +35,7 @@ add_flang_library(FortranRuntime
allocatable.cpp
assign.cpp
buffer.cpp
+ command.cpp
complex-reduction.c
copy.cpp
character.cpp
diff --git a/flang/runtime/command.cpp b/flang/runtime/command.cpp
new file mode 100644
index 0000000000000..603da15b5699b
--- /dev/null
+++ b/flang/runtime/command.cpp
@@ -0,0 +1,21 @@
+//===-- runtime/command.cpp -----------------------------------------------===//
+//
+// 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
+//
+//===----------------------------------------------------------------------===//
+
+#include "command.h"
+#include "environment.h"
+
+namespace Fortran::runtime {
+CppTypeFor<TypeCategory::Integer, 4> RTNAME(ArgumentCount)() {
+ int argc{executionEnvironment.argc};
+ if (argc > 1) {
+ // C counts the command name as one of the arguments, but Fortran doesn't.
+ return argc - 1;
+ }
+ return 0;
+}
+} // namespace Fortran::runtime
diff --git a/flang/unittests/Runtime/CMakeLists.txt b/flang/unittests/Runtime/CMakeLists.txt
index 6ae2f6443d67e..33f1bd086ded0 100644
--- a/flang/unittests/Runtime/CMakeLists.txt
+++ b/flang/unittests/Runtime/CMakeLists.txt
@@ -1,6 +1,7 @@
add_flang_unittest(FlangRuntimeTests
BufferTest.cpp
CharacterTest.cpp
+ CommandTest.cpp
CrashHandlerFixture.cpp
ExternalIOTest.cpp
Format.cpp
diff --git a/flang/unittests/Runtime/CommandTest.cpp b/flang/unittests/Runtime/CommandTest.cpp
new file mode 100644
index 0000000000000..9c67329edfc02
--- /dev/null
+++ b/flang/unittests/Runtime/CommandTest.cpp
@@ -0,0 +1,32 @@
+//===-- flang/unittests/RuntimeGTest/CommandTest.cpp ----------------------===//
+//
+// 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
+//
+//===----------------------------------------------------------------------===//
+
+#include "../../runtime/command.h"
+#include "gtest/gtest.h"
+#include "../../runtime/descriptor.h"
+#include "../../runtime/main.h"
+
+using namespace Fortran::runtime;
+
+TEST(ArgumentCount, ZeroArguments) {
+ const char *argv[]{"aProgram"};
+ RTNAME(ProgramStart)(1, argv, {});
+ EXPECT_EQ(0, RTNAME(ArgumentCount)());
+}
+
+TEST(ArgumentCount, OneArgument) {
+ const char *argv[]{"aProgram", "anArgument"};
+ RTNAME(ProgramStart)(2, argv, {});
+ EXPECT_EQ(1, RTNAME(ArgumentCount)());
+}
+
+TEST(ArgumentCount, SeveralArguments) {
+ const char *argv[]{"aProgram", "arg1", "arg2", "arg3", "arg4"};
+ RTNAME(ProgramStart)(5, argv, {});
+ EXPECT_EQ(4, RTNAME(ArgumentCount)());
+}
More information about the flang-commits
mailing list