ICODE-ADC/lammps
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added verbose class option + removed empty output lines

Browse Source
This commit is contained in:
Eugen Rožić
2022-05-24 11:37:44 +02:00
parent 18df50356e
commit b66702c285
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
python/lammps/pylammps.py
View File

@ -457,8 +457,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 +870,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()
@ -984,4 +985,4 @@ class IPyLammps(PyLammps):
:rtype: :py:class:`IPython.display.HTML`
"""
from IPython.display import HTML
return HTML("<video controls><source src=\"" + filename + "\"></video>")
return HTML("<video controls><source src=\"" + filename + "\"></video>")