# example illustrating reading an MFA and evaluating a point from it
# assumes that a 2-d domain was previously modeled and saved as "approx.out"
import diy
import mfa
import numpy as np
# MPI, DIY world and master
w = diy.mpi.MPIComm() # world
m = diy.Master(w) # master
# load the results and print them out
print("\n\nLoading blocks and printing them out\n")
a = diy.ContiguousAssigner(w.size, -1)
diy.read_blocks("approx.out", a, m, load=mfa.load_block)
m.foreach(lambda b, cp: b.print_block(cp, False))
# evaluate a point
param = np.array([0.5, 0.5]) # input parameters where to decode the point
pt = np.array([0.0, 0.0, 0.0]) # assigning fake values defines shape and type
m.foreach(lambda b, cp: b.decode_point(cp, param, pt))
print("\nThe point at [u =", param[0], ", v =", param[1], "] =", pt)
def recv(self, cp):
link = cp.link()
for i in range(len(link)):
gid = link.target(i).gid
o = cp.dequeue(gid)
dir = link.direction(i)
# print("%d received from %d: %s from direction %s" % (cp.gid(), gid, o, dir))
def add_block(gid, core, bounds, domain, link):
#print(gid, core, bounds, domain)
m.add(gid, Block(core), link)
nblocks = 16
domain = diy.DiscreteBounds([0,0,0], [100, 100, 100])
d = diy.DiscreteDecomposer(3, domain, nblocks)
a = diy.ContiguousAssigner(size, nblocks)
d.decompose(rank, a, add_block)
print(m)
# m.foreach(Block.show)
m.foreach(Block.send)
m.exchange()
# a little more mpi4py: pretend we want a barrier here
comm.Barrier()
m.foreach(Block.recv)
print("Success")
self.data = np.empty(n)
self.core = core
def show(self, cp):
print(w.rank, cp.gid(), self.core)
def add_block(gid, core, bounds, domain, link):
b = Block(core)
for i in range(b.data.shape[0]):
b.data[i] = i
m.add(gid, b, link)
nblocks = 16
domain = diy.DiscreteBounds([0,0,0], [100, 100, 100])
d = diy.DiscreteDecomposer(3, domain, nblocks)
a = diy.ContiguousAssigner(w.size, nblocks)
d.decompose(w.rank, a, add_block)
print(m)
diy.write_blocks("outfile", m)
m2 = diy.Master(w)
domain2 = diy.read_blocks("outfile", a, m2)
print(domain2)
# check that the data values match
for i in range(m.size()):
if np.any(m.block(i).data != m2.block(i).data):
print("Error: data values do not match. Aborting")
print(m.block(i).data)