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

Change return type of lammps.get_natoms to int

Browse Source 
Closes PR #2370
This commit is contained in:
Richard Berger
2020-09-17 10:57:49 -04:00
parent 12582edfb7
commit 5dacfc47ca
3 changed files with 18 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
python/lammps.py
View File

@ -705,9 +705,9 @@ class lammps(object):
underlying :cpp:func:`lammps_get_natoms` function returning a double.
:return: number of atoms
:rtype: float
:rtype: int
"""
return self.lib.lammps_get_natoms(self.lmp)
return int(self.lib.lammps_get_natoms(self.lmp))
# -------------------------------------------------------------------------
@ -1140,7 +1140,7 @@ class lammps(object):
def gather_atoms(self,name,type,count):
if name: name = name.encode()
natoms = self.lib.lammps_get_natoms(self.lmp)
natoms = self.get_natoms()
if type == 0:
data = ((count*natoms)*c_int)()
self.lib.lammps_gather_atoms(self.lmp,name,type,count,data)
@ -1154,7 +1154,7 @@ class lammps(object):
def gather_atoms_concat(self,name,type,count):
if name: name = name.encode()
natoms = self.lib.lammps_get_natoms(self.lmp)
natoms = self.get_natoms()
if type == 0:
data = ((count*natoms)*c_int)()
self.lib.lammps_gather_atoms_concat(self.lmp,name,type,count,data)
@ -1206,7 +1206,7 @@ class lammps(object):
# e.g. for Python list or NumPy or ctypes
def gather(self,name,type,count):
if name: name = name.encode()
natoms = self.lib.lammps_get_natoms(self.lmp)
natoms = self.get_natoms()
if type == 0:
data = ((count*natoms)*c_int)()
self.lib.lammps_gather(self.lmp,name,type,count,data)
@ -1218,7 +1218,7 @@ class lammps(object):
def gather_concat(self,name,type,count):
if name: name = name.encode()
natoms = self.lib.lammps_get_natoms(self.lmp)
natoms = self.get_natoms()
if type == 0:
data = ((count*natoms)*c_int)()
self.lib.lammps_gather_concat(self.lmp,name,type,count,data)