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