yade-dev team mailing list archive

[noreply@xxxxxxxxxxxxx]

------------------------------------------------------------
revno: 3813
committer: Anton Gladky <gladky.anton@xxxxxxxxx>
timestamp: Thu 2016-03-24 23:07:41 +0100
message:
  Add import of polyhedras from the file.
modified:
  py/ymport.py


--
lp:yade
https://code.launchpad.net/~yade-pkg/yade/git-trunk

Your team Yade developers is subscribed to branch lp:yade.
To unsubscribe from this branch go to https://code.launchpad.net/~yade-pkg/yade/git-trunk/+edit-subscription

=== modified file 'py/ymport.py'
--- py/ymport.py	2016-01-07 17:49:32 +0000
+++ py/ymport.py	2016-03-24 22:07:41 +0000
@@ -3,7 +3,7 @@
 """
 
 from yade.wrapper import *
-from yade import utils
+from yade import utils, polyhedra_utils
 
 from minieigen import *
 
@@ -446,3 +446,43 @@
 		tetras[int(ls[0])-1] = utils.polyhedron([vertices[int(ls[j])-1] for j in (1,2,3,4)],**kw)
 	f.close()
 	return tetras
+
+def textPolyhedra(fileName,material,shift=Vector3.Zero,scale=1.0,**kw):
+	"""Load polyhedra from a text file.
+	
+	:param str filename: file name
+	:param [float,float,float] shift: [X,Y,Z] parameter moves the specimen.
+	:param float scale: factor scales the given data.
+	:param \*\*kw: (unused keyword arguments) is passed to :yref:`yade.polyhedra_utils.polyhedra`
+	:returns: list of polyhedras.
+
+	Lines starting with # are skipped
+	"""
+	infile = open(fileName,"r")
+	lines = infile.readlines()
+	infile.close()
+	ret=[]
+	i=-1
+	while (i < (len(lines)-1)):
+		i+=1
+		line = lines[i]
+		data = line.split()
+		if (data[0][0] == "#"): continue
+		
+		if (len(data)!=3):
+			raise RuntimeError("Check polyhedra input file! Number of parameters in the first line is not 3!");
+		else:
+			vertLoad = []
+			ids = int(data[0])
+			verts = int(data[1])
+			surfs = int(data[2])
+			i+=1
+			for d in range(verts):
+				dataV = lines[i].split()
+				pos = Vector3(float(dataV[0]),float(dataV[1]),float(dataV[2]))
+				vertLoad.append(pos)
+				i+=1
+			polR = polyhedra_utils.polyhedra(material=material,v=vertLoad,**kw)
+			ret.append(polR)
+			i= i + surfs - 1
+	return ret