shuren/opencl_fdtd
1
0
Fork 0
forked from jan/opencl_fdtd
Code Issues Pull Requests Releases Wiki Activity

Use logging package for output

Browse Source
This commit is contained in:
Jan Petykiewicz 2017-08-12 19:20:29 -07:00
parent c1750e9fde
commit 97d9901e4b
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
fdtd.py
View File

@ -6,6 +6,7 @@ See main() for simulation setup.
import sys import sys
import time import time
import logging
import numpy import numpy
import lzma import lzma
@ -20,6 +21,9 @@ import fdfd_tools
__author__ = 'Jan Petykiewicz' __author__ = 'Jan Petykiewicz'
logging.basicConfig(level=logging.DEBUG)
logger = logging.getLogger(__name__)
def perturbed_l3(a: float, radius: float, **kwargs) -> Pattern: def perturbed_l3(a: float, radius: float, **kwargs) -> Pattern:
""" """
@ -120,7 +124,7 @@ def main():
eps=n_air**2, eps=n_air**2,
polygons=mask.as_polygons()) polygons=mask.as_polygons())
print('grid shape: {}'.format(grid.shape)) logger.info('grid shape: {}'.format(grid.shape))
# #### Create the simulation grid #### # #### Create the simulation grid ####
sim = Simulation(grid.grids, do_poynting=True, pml_thickness=8) sim = Simulation(grid.grids, do_poynting=True, pml_thickness=8)
@ -157,7 +161,7 @@ def main():
e.wait() e.wait()
if t % 100 == 0: if t % 100 == 0:
print('iteration {}: average {} iterations per sec'.format(t, (t+1)/(time.perf_counter()-start))) logger.info('iteration {}: average {} iterations per sec'.format(t, (t+1)/(time.perf_counter()-start)))
sys.stdout.flush() sys.stdout.flush()
with lzma.open('saved_simulation', 'wb') as f: with lzma.open('saved_simulation', 'wb') as f: