Add amrex::Print() Like Function

Add `amr.Print(...)` that behaves like Python's `print()`, but only prints on the AMReX IOProcessor.
AMReX-Codes · Aug 14, 2023 · 7900f00 · 7900f00
1 parent d1598ba
commit 7900f00
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 0 deletions.
diff --git a/src/amrex/space1d/__init__.py b/src/amrex/space1d/__init__.py
@@ -32,3 +32,9 @@
 #
 def d_decl(x, y, z):
     return (x,)
+
+
+def Print(*args, **kwargs):
+    """Wrap amrex::Print() - only the IO processor writes"""
+    if ParallelDescriptor.IOProcessor():
+        print(*args, **kwargs)
diff --git a/src/amrex/space2d/__init__.py b/src/amrex/space2d/__init__.py
@@ -32,3 +32,9 @@
 #
 def d_decl(x, y, z):
     return (x, y)
+
+
+def Print(*args, **kwargs):
+    """Wrap amrex::Print() - only the IO processor writes"""
+    if ParallelDescriptor.IOProcessor():
+        print(*args, **kwargs)
diff --git a/src/amrex/space3d/__init__.py b/src/amrex/space3d/__init__.py
@@ -32,3 +32,9 @@
 #
 def d_decl(x, y, z):
     return (x, y, z)
+
+
+def Print(*args, **kwargs):
+    """Wrap amrex::Print() - only the IO processor writes"""
+    if ParallelDescriptor.IOProcessor():
+        print(*args, **kwargs)