git-svn-id: svn://svn.icms.temple.edu/lammps-ro/trunk@15811 f3b2605a-c512-4ea7-a41b-209d697bcdaa

python/examples/simple.py

@@ -51,17 +51,39 @@ for line in lines: lmp.command(line)
 
 lmp.command("run 10")
 x = lmp.gather_atoms("x",1,3)
+v = lmp.gather_atoms("v",1,3)
 epsilon = 0.1
 x[0] += epsilon
 lmp.scatter_atoms("x",1,3,x)
 lmp.command("run 1");
 
+# extract force on single atom
+
 f = lmp.extract_atom("f",3)
+print("F",f)
 print("Force on 1 atom via extract_atom: ",f[0][0])
 
 fx = lmp.extract_variable("fx","all",1)
+print("FX",fx)
 print("Force on 1 atom via extract_variable:",fx[0])
 
+# use commands_list() to invoke more commands
+
+#strtwo = ["run 10","run 20"]
+#lmp.commands_list(strtwo)
+#lmp.commands_list(strtwo)
+
+# delete all atoms
+# create_atoms() to create new ones with old coords, vels
+# initial thermo should be same as step 20
+
+#natoms = lmp.get_natoms()
+#type = natoms*[1]
+
+#lmp.command("delete_atoms group all");
+#lmp.create_atoms(natoms,None,type,x,v);
+#lmp.command("run 10");
+
 # uncomment if running in parallel via Pypar
 #print("Proc %d out of %d procs has" % (me,nprocs), lmp)
 #pypar.finalize()

python/lammps.py

@@ -172,6 +172,8 @@ class lammps(object):
     if file: file = file.encode()
     self.lib.lammps_file(self.lmp,file)
 
+  # send a single command
+    
   def command(self,cmd):
     if cmd: cmd = cmd.encode()
     self.lib.lammps_command(self.lmp,cmd)
@@ -185,6 +187,13 @@ class lammps(object):
         raise MPIAbortException(error_msg)
       raise Exception(error_msg)
 
+  # send a list of commands, one at a time
+
+  def commands_list(self,cmdlist):
+    for cmd in cmdlist: self.command(cmd)
+
+  # extract global info
+    
   def extract_global(self,name,type):
     if name: name = name.encode()
     if type == 0:
@@ -195,6 +204,8 @@ class lammps(object):
     ptr = self.lib.lammps_extract_global(self.lmp,name)
     return ptr[0]
 
+  # extract per-atom info
+  
   def extract_atom(self,name,type):
     if name: name = name.encode()
     if type == 0:
@@ -209,6 +220,8 @@ class lammps(object):
     ptr = self.lib.lammps_extract_atom(self.lmp,name)
     return ptr
 
+  # extract compute info
+  
   def extract_compute(self,id,style,type):
     if id: id = id.encode()
     if type == 0:
@@ -226,6 +239,7 @@ class lammps(object):
       return ptr
     return None
 
+  # extract fix info
   # in case of global datum, free memory for 1 double via lammps_free()
   # double was allocated by library interface function
 
@@ -249,6 +263,7 @@ class lammps(object):
     else:
       return None
 
+  # extract variable info
   # free memory for 1 double or 1 vector of doubles via lammps_free()
   # for vector, must copy nlocal returned values to local c_double vector
   # memory was allocated by library interface function
@@ -296,7 +311,14 @@ class lammps(object):
     return self.lib.lammps_get_natoms(self.lmp)
 
   # return vector of atom properties gathered across procs, ordered by atom ID
-
+  # name = atom property recognized by LAMMPS in atom->extract()
+  # type = 0 for integer values, 1 for double values
+  # count = number of per-atom valus, 1 for type or charge, 3 for x or f
+  # returned data is a 1d vector - doc how it is ordered?
+  # NOTE: how could we insure are converting to correct Python type
+  #   e.g. for Python list or NumPy, etc
+  #   ditto for extact_atom() above
+  
   def gather_atoms(self,name,type,count):
     if name: name = name.encode()
     natoms = self.lib.lammps_get_natoms(self.lmp)
@@ -310,12 +332,38 @@ class lammps(object):
     return data
 
   # scatter vector of atom properties across procs, ordered by atom ID
-  # assume vector is of correct type and length, as created by gather_atoms()
-
+  # name = atom property recognized by LAMMPS in atom->extract()
+  # type = 0 for integer values, 1 for double values
+  # count = number of per-atom valus, 1 for type or charge, 3 for x or f
+  # assume data is of correct type and length, as created by gather_atoms()
+  # NOTE: how could we insure are passing correct type to LAMMPS
+  # e.g. for Python list or NumPy, etc
+    
   def scatter_atoms(self,name,type,count,data):
     if name: name = name.encode()
     self.lib.lammps_scatter_atoms(self.lmp,name,type,count,data)
 
+  # create N atoms on all procs
+  # N = global number of atoms
+  # id = ID of each atom (optional, can be None)
+  # type = type of each atom (1 to Ntypes) (required)
+  # x = coords of each atom as (N,3) array (required)
+  # v = velocity of each atom as (N,3) array (optional, can be None)
+  # NOTE: how could we insure are passing correct type to LAMMPS
+  #   e.g. for Python list or NumPy, etc
+  #   ditto for gather_atoms() above
+
+  def create_atoms(self,n,id,type,x,v):
+    if id:
+      id_lmp = (c_int * n)()
+      id_lmp[:] = id
+    else: id_lmp = id
+    type_lmp = (c_int * n)()
+    type_lmp[:] = type
+    self.lib.lammps_create_atoms(self.lmp,n,id_lmp,type_lmp,x,v)
+
+  # document this?
+    
   @property
   def uses_exceptions(self):
     try: