def grid2D(shape):
"""
:param `shape`: shape of the grid tuple (nb_rows,nb_columns)
"""
t = Topomesh(2)
positions = {}
NBI, NBJ = shape
NBCELLS = NBI * NBJ
xincr = 1. / NBI
yincr = 1. / NBJ
cell_grid = Grid((NBI, NBJ))
for ind in cell_grid:
cid = t.add_wisp(0, ind)
point_grid = Grid((NBI + 1, NBJ + 1))
for ind in point_grid:
i, j = point_grid.coordinates(ind)
eid = t.add_wisp(2, ind)
positions[eid] = Vector3(xincr * i, yincr * j, 0)
#murs verticaux
for j in xrange(NBJ):
for i in xrange(NBI + 1):
wid = t.add_wisp(1)
t.link(1, wid, point_grid.index((i, j)))
t.link(1, wid, point_grid.index((i, j + 1)))
if i < NBI:
t.link(0, cell_grid.index((i, j)), wid)
if i > 0:
t.link(0, cell_grid.index((i - 1, j)), wid)
#murs horizontaux
for i in xrange(NBI):
for j in xrange(NBJ + 1):
wid = t.add_wisp(1)
t.link(1, wid, point_grid.index((i, j)))
t.link(1, wid, point_grid.index((i + 1, j)))
if j < NBJ:
t.link(0, cell_grid.index((i, j)), wid)
if j > 0:
t.link(0, cell_grid.index((i, j - 1)), wid)
#et voila
return t, positions
def grid3D(shape):
"""
:param `shape`: shape of the grid tuple (nb_rows,nb_columns,nb_slices)
"""
from celltissue import PolyhedralTissue
from celltissue.geometry import Point, Segment
t = PolyhedralTissue(3)
positions = Mesh2D()
NBI, NBJ, NBK = shape
NBCELLS = NBI * NBJ * NBK
xincr = 1. / NBI
yincr = 1. / NBJ
zincr = 1. / NBK
cell_grid = Grid((NBI, NBJ, NBK))
for ind in xrange(len(cell_grid)):
cid = t.add_wisp(0, ind)
point_grid = Grid((NBI + 1, NBJ + 1, NBK + 1))
for ind in point_grid:
i, j, k = point_grid.coordinates(ind)
positions.add_point(xyz(xincr * i, yincr * j, zincr * k), ind)
#edges
JKedges = Grid((NBI, NBJ + 1, NBK + 1))
JK = {}
for ind in JKedges:
i, j, k = JKedges.coordinates(ind)
eid = t.add_wisp(2)
t.set_geometry(2,eid,Segment([Point(point_grid.index( (i,j,k) )),\
Point(point_grid.index( (i+1,j,k) ))] ))
JK[(i, j, k)] = eid
IKedges = Grid((NBI + 1, NBJ, NBK + 1))
IK = {}
for ind in IKedges:
i, j, k = IKedges.coordinates(ind)
eid = t.add_wisp(2)
t.set_geometry(2,eid,Segment([Point(point_grid.index( (i,j,k) )),\
Point(point_grid.index( (i,j+1,k) ))] ))
IK[(i, j, k)] = eid
IJedges = Grid((NBI + 1, NBJ + 1, NBK))
IJ = {}
for ind in IJedges:
i, j, k = IJedges.coordinates(ind)
eid = t.add_wisp(2)
t.set_geometry(2,eid,Segment([Point(point_grid.index( (i,j,k) )),\
Point(point_grid.index( (i,j,k+1) ))] ))
IJ[(i, j, k)] = eid
#murs
for k in xrange(NBK + 1):
for j in xrange(NBJ):
for i in xrange(NBI):
wid = t.add_wisp(1)
t.link(1, wid, IK[(i, j, k)])
t.link(1, wid, IK[(i + 1, j, k)])
t.link(1, wid, JK[(i, j, k)])
t.link(1, wid, JK[(i, j + 1, k)])
if k < NBK:
t.link(0, cell_grid.index((i, j, k)), wid)
if k > 0:
t.link(0, cell_grid.index((i, j, k - 1)), wid)
for k in xrange(NBK):
for j in xrange(NBJ + 1):
for i in xrange(NBI):
wid = t.add_wisp(1)
t.link(1, wid, IJ[(i, j, k)])
t.link(1, wid, IJ[(i + 1, j, k)])
t.link(1, wid, JK[(i, j, k)])
t.link(1, wid, JK[(i, j, k + 1)])
if j < NBJ:
t.link(0, cell_grid.index((i, j, k)), wid)
if j > 0:
t.link(0, cell_grid.index((i, j - 1, k)), wid)
for k in xrange(NBK):
for j in xrange(NBJ):
for i in xrange(NBI + 1):
wid = t.add_wisp(1)
t.link(1, wid, IK[(i, j, k)])
t.link(1, wid, IK[(i, j, k + 1)])
t.link(1, wid, IJ[(i, j, k)])
t.link(1, wid, IJ[(i, j + 1, k)])
if i < NBI:
t.link(0, cell_grid.index((i, j, k)), wid)
if i > 0:
t.link(0, cell_grid.index((i - 1, j, k)), wid)
return t, positions
from openalea.container import Grid
g=Grid( (2,3) )
print "dim",g.dim()
print "shape",g.shape()
print "size",g.size(),len(g)
for i in g :
print "iter",i,"coord",g.coordinates(i),"index",g.index(g.coordinates(i))
g=Grid( (2,3,4) )
print "dim",g.dim()
print "shape",g.shape()
print "size",g.size(),len(g)
for i in g :
print "iter",i,"coord",g.coordinates(i),"index",g.index(g.coordinates(i))
def grid2D (shape) :
"""
:param `shape`: shape of the grid tuple (nb_rows,nb_columns)
"""
t=Topomesh(2)
positions={}
NBI,NBJ=shape
NBCELLS=NBI*NBJ
xincr=1./NBI
yincr=1./NBJ
cell_grid=Grid( (NBI,NBJ) )
for ind in cell_grid :
cid=t.add_wisp(0,ind)
point_grid=Grid( (NBI+1,NBJ+1) )
for ind in point_grid :
i,j=point_grid.coordinates(ind)
eid=t.add_wisp(2,ind)
positions[eid]=Vector3(xincr*i,yincr*j,0)
#murs verticaux
for j in xrange(NBJ) :
for i in xrange(NBI+1) :
wid=t.add_wisp(1)
t.link(1,wid,point_grid.index( (i,j) ))
t.link(1,wid,point_grid.index( (i,j+1) ))
if i<NBI :
t.link(0,cell_grid.index( (i,j) ),wid)
if i>0 :
t.link(0,cell_grid.index( (i-1,j) ),wid)
#murs horizontaux
for i in xrange(NBI) :
for j in xrange(NBJ+1) :
wid=t.add_wisp(1)
t.link(1,wid,point_grid.index( (i,j) ))
t.link(1,wid,point_grid.index( (i+1,j) ))
if j<NBJ :
t.link(0,cell_grid.index( (i,j) ),wid)
if j>0 :
t.link(0,cell_grid.index( (i,j-1) ),wid)
#et voila
return t,positions
def grid3D (shape) :
"""
:param `shape`: shape of the grid tuple (nb_rows,nb_columns,nb_slices)
"""
from celltissue import PolyhedralTissue
from celltissue.geometry import Point,Segment
t=PolyhedralTissue(3)
positions=Mesh2D()
NBI,NBJ,NBK=shape
NBCELLS=NBI*NBJ*NBK
xincr=1./NBI
yincr=1./NBJ
zincr=1./NBK
cell_grid=Grid( (NBI,NBJ,NBK) )
for ind in xrange(len(cell_grid)) :
cid=t.add_wisp(0,ind)
point_grid=Grid( (NBI+1,NBJ+1,NBK+1) )
for ind in point_grid :
i,j,k=point_grid.coordinates(ind)
positions.add_point(xyz(xincr*i,yincr*j,zincr*k),ind)
#edges
JKedges=Grid( (NBI,NBJ+1,NBK+1) )
JK={}
for ind in JKedges :
i,j,k=JKedges.coordinates(ind)
eid=t.add_wisp(2)
t.set_geometry(2,eid,Segment([Point(point_grid.index( (i,j,k) )),\
Point(point_grid.index( (i+1,j,k) ))] ))
JK[(i,j,k)]=eid
IKedges=Grid( (NBI+1,NBJ,NBK+1) )
IK={}
for ind in IKedges :
i,j,k=IKedges.coordinates(ind)
eid=t.add_wisp(2)
t.set_geometry(2,eid,Segment([Point(point_grid.index( (i,j,k) )),\
Point(point_grid.index( (i,j+1,k) ))] ))
IK[(i,j,k)]=eid
IJedges=Grid( (NBI+1,NBJ+1,NBK) )
IJ={}
for ind in IJedges :
i,j,k=IJedges.coordinates(ind)
eid=t.add_wisp(2)
t.set_geometry(2,eid,Segment([Point(point_grid.index( (i,j,k) )),\
Point(point_grid.index( (i,j,k+1) ))] ))
IJ[(i,j,k)]=eid
#murs
for k in xrange(NBK+1) :
for j in xrange(NBJ) :
for i in xrange(NBI) :
wid=t.add_wisp(1)
t.link(1,wid,IK[(i,j,k)])
t.link(1,wid,IK[(i+1,j,k)])
t.link(1,wid,JK[(i,j,k)])
t.link(1,wid,JK[(i,j+1,k)])
if k<NBK :
t.link(0,cell_grid.index( (i,j,k) ),wid)
if k>0 :
t.link(0,cell_grid.index( (i,j,k-1) ),wid)
for k in xrange(NBK) :
for j in xrange(NBJ+1) :
for i in xrange(NBI) :
wid=t.add_wisp(1)
t.link(1,wid,IJ[(i,j,k)])
t.link(1,wid,IJ[(i+1,j,k)])
t.link(1,wid,JK[(i,j,k)])
t.link(1,wid,JK[(i,j,k+1)])
if j<NBJ :
t.link(0,cell_grid.index( (i,j,k) ),wid)
if j>0 :
t.link(0,cell_grid.index( (i,j-1,k) ),wid)
for k in xrange(NBK) :
for j in xrange(NBJ) :
for i in xrange(NBI+1) :
wid=t.add_wisp(1)
t.link(1,wid,IK[(i,j,k)])
t.link(1,wid,IK[(i,j,k+1)])
t.link(1,wid,IJ[(i,j,k)])
t.link(1,wid,IJ[(i,j+1,k)])
if i<NBI :
t.link(0,cell_grid.index( (i,j,k) ),wid)
if i>0 :
t.link(0,cell_grid.index( (i-1,j,k) ),wid)
return t,positions
from openalea.container import Grid
g = Grid((2, 3))
print "dim", g.dim()
print "shape", g.shape()
print "size", g.size(), len(g)
for i in g:
print "iter", i, "coord", g.coordinates(i), "index", g.index(
g.coordinates(i))
g = Grid((2, 3, 4))
print "dim", g.dim()
print "shape", g.shape()
print "size", g.size(), len(g)
for i in g:
print "iter", i, "coord", g.coordinates(i), "index", g.index(
g.coordinates(i))
