jianmu
/
lammps
forked from lijiext/lammps
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix Python 3 compatibility by encoding strings passed as c_char_p

This commit is contained in:
Richard Berger 2017-03-15 21:56:39 -04:00
parent 3a2da51a82
commit 0262a54ecf
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
python/lammps.py
View File

@ -190,12 +190,15 @@ class lammps(object):
# send a list of commands # send a list of commands
def commands_list(self,cmdlist): def commands_list(self,cmdlist):
args = (c_char_p * len(cmdlist))(*cmdlist) cmds = [x.encode() for x in cmdlist if type(x) is str]
args = (c_char_p * len(cmdlist))(*cmds)
self.lib.lammps_commands_list(self.lmp,len(cmdlist),args) self.lib.lammps_commands_list(self.lmp,len(cmdlist),args)
# send a string of commands # send a string of commands
def commands_string(self,multicmd): def commands_string(self,multicmd):
if type(multicmd) is str:
multicmd = multicmd.encode()
self.lib.lammps_commands_string(self.lmp,c_char_p(multicmd)) self.lib.lammps_commands_string(self.lmp,c_char_p(multicmd))
# extract global info # extract global info