node = plasma.Grid(conf.Nx, conf.Ny)
node.setGridLims(conf.xmin, conf.xmax, conf.ymin, conf.ymax)
#node.initMpi()
#loadMpiXStrides(node)
loadCells(node, conf)
#insert initial current into the grid
injectRingCurrent(node, conf)
updateBoundaries(node)
#plot initial condition
plotNode(axs[0], node, conf)
plotEfield(axs[1], node, conf)
saveVisz(0, node, conf)
#main loop
for lap in range(1, 51):
print("---lap: {}".format(lap))
#B field
updateBoundaries(node)
for cid in node.getCellIds():
c = node.getCellPtr(cid)
c.pushHalfB()
c.pushHalfB()
#E field
updateBoundaries(node)
for cid in node.getCellIds():
def test_boundary_construction(self):
conf = Conf()
# set up plotting and figure
try:
plt.fig = plt.figure(1, figsize=(3, 3))
plt.rc('font', family='serif', size=12)
plt.rc('xtick')
plt.rc('ytick')
gs = plt.GridSpec(1, 1)
gs.update(hspace=0.5)
axs = []
for ai in range(1):
axs.append(plt.subplot(gs[ai]))
if not os.path.exists(conf.outdir):
os.makedirs(conf.outdir)
except:
pass
grid = pycorgi.twoD.Grid(conf.Nx, conf.Ny)
grid.set_grid_lims(conf.xmin, conf.xmax, conf.ymin, conf.ymax)
# one tile surrounded by other rank
refGrid = np.ones((conf.Nx, conf.Ny), np.int)
refGrid[1, 1] = 0
#setup grid configuration
if grid.master():
for j in range(grid.get_Ny()):
for i in range(grid.get_Nx()):
val = refGrid[i, j]
grid.set_mpi_grid(i, j, val)
grid.bcast_mpi_grid()
# add tiles
rank = grid.rank()
for i in range(grid.get_Nx()):
for j in range(grid.get_Ny()):
if rank == refGrid[i, j]:
c = pycorgi.twoD.Tile()
grid.add_tile(c, (i, j))
try:
plotNode(axs[0], grid, conf)
saveVisz(0, grid, conf)
except:
pass
if grid.size() > 1:
grid.analyze_boundaries()
#print(grid.rank(), ":sq ", grid.send_queue)
#print(grid.rank(), ":sqa", grid.send_queue_address)
if grid.size() > 1:
grid.send_tiles()
grid.recv_tiles()
try:
plotNode(axs[0], grid, conf)
saveVisz(1, grid, conf)
except:
pass
cur = readGrid(grid, conf)
#print(cur)
if grid.size() > 1:
for i in range(grid.get_Nx()):
for j in range(grid.get_Ny()):
self.assertEqual(refGrid[i, j], cur[i, j])
#static setup; communicate neighbor info once
grid.analyze_boundaries()
grid.send_tiles()
grid.recv_tiles()
initialize_virtuals(grid, conf)
timer.stop("init")
timer.stats("init")
# end of initialization
##################################################
# visualize initial condition
#try:
plotNode(axs[0], grid, conf)
#plotXmesh(axs[1], grid, conf, 0, "x")
saveVisz(-1, grid, conf)
#except:
# print()
# pass
Nsamples = conf.Nt
fldprop = pyfld.FDTD2()
pusher = pypic.BorisPusher()
fintp = pypic.LinearInterpolator()
currint = pypic.ZigZag()
analyzer = pypic.Analyzator()
flt = pytools.Filter(conf.NxMesh, conf.NyMesh)
flt.init_gaussian_kernel(2.0, 2.0)
#simulation loop
