def testDivisonOfGrid(): topo = topology.Topology() topoFile = "topology_200.txt" try: topo.importFromFile(topoFile) except: print "could not read topology file", topoFile habNM = simulation.makeNeighConnMap(topo) sects = simulation.divideGridToSection(habNM, GRID_SIZE, GRID_SIZE) print "sects ", sects k = 0 for habs in sects.values(): k = k + 1 X = pylab.ones((10, 10, 3)) * float(k * 0.4) print "habs", habs for hab in habs: hab = hab - 1 xo = int(hab % GRID_SIZE) * 2 yo = int(hab / GRID_SIZE) * 2 print "X ", xo print "Y ", yo plt.figimage(X, xo, yo, origin='lower') plt.show()
def testDivisonOfGrid(): topo = topology.Topology() topoFile="topology_200.txt" try: topo.importFromFile(topoFile) except: print "could not read topology file",topoFile habNM = simulation.makeNeighConnMap(topo) sects = simulation.divideGridToSection(habNM,GRID_SIZE,GRID_SIZE) print "sects ",sects k=0 for habs in sects.values(): k=k+1 X = pylab.ones((10,10,3))*float(k*0.4) print "habs" ,habs for hab in habs: hab=hab-1 xo=int(hab%GRID_SIZE)*2 yo=int(hab/GRID_SIZE)*2 print "X ",xo print "Y ",yo plt.figimage(X, xo, yo, origin='lower') plt.show()
from mpi4py import MPI import simulation import topology comm = MPI.COMM_WORLD size = comm.Get_size() rank = comm.Get_rank() if rank == 0: topo = topology.Topology() topoFile="topology_200.txt" try: topo.importFromFile(topoFile) except: print "could not read topology file",topoFile habNM = simulation.makeNeighConnMap(topo) secs = simulation.divideGridToSection(habNM,size,20) data =[] for i in range(size): data.append(secs[i]) print "before scattering ",data else: data = None data = comm.scatter(data, root=0) #assert data == (rank+1)**2 print "rank ",rank print "data ",data
from mpi4py import MPI import simulation import topology comm = MPI.COMM_WORLD size = comm.Get_size() rank = comm.Get_rank() if rank == 0: topo = topology.Topology() topoFile = "topology_200.txt" try: topo.importFromFile(topoFile) except: print "could not read topology file", topoFile habNM = simulation.makeNeighConnMap(topo) secs = simulation.divideGridToSection(habNM, size, 20) data = [] for i in range(size): data.append(secs[i]) print "before scattering ", data else: data = None data = comm.scatter(data, root=0) #assert data == (rank+1)**2 print "rank ", rank print "data ", data