Add amrex::Print() Like Function

Add `amr.Print(...)` that behaves like Python's `print()`, but only prints on the AMReX IOProcessor.
ax3l · Aug 11, 2023 · e83530b · e83530b
1 parent 7b8055f
commit e83530b
Show file tree

Hide file tree

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