tx258/lammps-gran-kokkos
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #3277 from erozic/feature-pylammps-verbose

Browse Source 
Verbose object-level option for PyLammps
This commit is contained in:
Axel Kohlmeyer
2022-05-25 10:57:58 -04:00
committed by GitHub
parent c6c0b269ec 1aaa9ca4ae
commit 7c2a548015
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
python/lammps/pylammps.py
View File

@ -449,6 +449,8 @@ class PyLammps(object):
:type ptr: pointer
:param comm: MPI communicator (as provided by `mpi4py <mpi4py_docs_>`_). ``None`` means use ``MPI_COMM_WORLD`` implicitly.
:type comm: MPI_Comm
:param verbose: print all LAMMPS output to stdout
:type verbose: bool
:ivar lmp: instance of original LAMMPS Python interface
:vartype lmp: :py:class:`lammps`
@ -457,8 +459,9 @@ class PyLammps(object):
:vartype run: list
"""
def __init__(self, name="", cmdargs=None, ptr=None, comm=None):
def __init__(self, name="", cmdargs=None, ptr=None, comm=None, verbose=False):
self.has_echo = False
self.verbose = verbose
if cmdargs:
if '-echo' in cmdargs:
@ -869,8 +872,8 @@ class PyLammps(object):
if comm:
output = self.lmp.comm.bcast(output, root=0)
if 'verbose' in kwargs and kwargs['verbose']:
print(output)
if self.verbose or ('verbose' in kwargs and kwargs['verbose']):
print(output, end = '')
lines = output.splitlines()