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

Add background_color argument to IPyLammps.image

Browse Source
This commit is contained in:
Richard Berger
2020-09-15 15:19:39 -04:00
parent e5e449795a
commit 6cfad41be4
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
python/lammps.py
View File

@ -2226,7 +2226,7 @@ class IPyLammps(PyLammps):
super(IPyLammps, self).__init__(name=name,cmdargs=cmdargs,ptr=ptr,comm=comm) super(IPyLammps, self).__init__(name=name,cmdargs=cmdargs,ptr=ptr,comm=comm)
def image(self, filename="snapshot.png", group="all", color="type", diameter="type", def image(self, filename="snapshot.png", group="all", color="type", diameter="type",
size=None, view=None, center=None, up=None, zoom=1.0): size=None, view=None, center=None, up=None, zoom=1.0, background_color="white"):
cmd_args = [group, "image", filename, color, diameter] cmd_args = [group, "image", filename, color, diameter]
if size: if size:
@ -2255,7 +2255,7 @@ class IPyLammps(PyLammps):
if zoom: if zoom:
cmd_args += ["zoom", zoom] cmd_args += ["zoom", zoom]
cmd_args.append("modify backcolor white") cmd_args.append("modify backcolor " + background_color)
self.write_dump(*cmd_args) self.write_dump(*cmd_args)
from IPython.core.display import Image from IPython.core.display import Image