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

Use datatype constants instead of magic numbers

Browse Source
This commit is contained in:
Richard Berger
2020-08-27 14:39:20 -04:00
parent 0d8baff7a9
commit 26d09ea648
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
python/lammps.py
View File

@ -421,17 +421,17 @@ class lammps(object):
c_int_type = c_int c_int_type = c_int
if dim == 1: if dim == 1:
raw_ptr = self.lmp.extract_atom(name, 0) raw_ptr = self.lmp.extract_atom(name, LAMMPS_INT)
else: else:
raw_ptr = self.lmp.extract_atom(name, 1) raw_ptr = self.lmp.extract_atom(name, LAMMPS_INT2D)
return self.iarray(c_int_type, raw_ptr, nelem, dim) return self.iarray(c_int_type, raw_ptr, nelem, dim)
def extract_atom_darray(self, name, nelem, dim=1): def extract_atom_darray(self, name, nelem, dim=1):
if dim == 1: if dim == 1:
raw_ptr = self.lmp.extract_atom(name, 2) raw_ptr = self.lmp.extract_atom(name, LAMMPS_DOUBLE)
else: else:
raw_ptr = self.lmp.extract_atom(name, 3) raw_ptr = self.lmp.extract_atom(name, LAMMPS_DOUBLE2D)
return self.darray(raw_ptr, nelem, dim) return self.darray(raw_ptr, nelem, dim)