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

rename lib

Browse Source
This commit is contained in:
jan 2017-08-14 15:41:20 -07:00
parent c1750e9fde
commit 0d91f0d43e
8 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
fdtd.py
View File

@ -8,10 +8,10 @@ import sys
import time import time
import numpy import numpy
import lzma import lzma
import dill import dill
from fdtd.simulation import Simulation from opencl_fdtd import Simulation
from masque import Pattern, shapes from masque import Pattern, shapes
import gridlock import gridlock
import pcgen import pcgen
@ -133,7 +133,7 @@ def main():
def field_source(i): def field_source(i):
t0 = i * sim.dt - delay t0 = i * sim.dt - delay
return numpy.sin(w * t0) * numpy.exp(-alpha * t0**2) return numpy.sin(w * t0) * numpy.exp(-alpha * t0**2)
with open('sources.c', 'w') as f: with open('sources.c', 'w') as f:
f.write(sim.sources['E']) f.write(sim.sources['E'])
f.write('\n==========================================\n') f.write('\n==========================================\n')
@ -172,5 +172,6 @@ def main():
d['S'] = unvec(sim.S.get()) d['S'] = unvec(sim.S.get())
dill.dump(d, f) dill.dump(d, f)
if __name__ == '__main__': if __name__ == '__main__':
main() main()

0
fdtd/__init__.py
View File

1
opencl_fdtd/__init__.py Normal file
View File

@ -0,0 +1 @@
from .simulation import Simulation, type_to_C

0
fdtd/kernels/common.cl → opencl_fdtd/kernels/common.cl
View File

0
fdtd/kernels/update_e.cl → opencl_fdtd/kernels/update_e.cl
View File

0
fdtd/kernels/update_h.cl → opencl_fdtd/kernels/update_h.cl
View File

0
fdtd/kernels/update_s.cl → opencl_fdtd/kernels/update_s.cl
View File

6
fdtd/simulation.py → opencl_fdtd/simulation.py
View File

@ -152,7 +152,7 @@ class Simulation(object):
S_fields[ptr('oS')] = self.oS S_fields[ptr('oS')] = self.oS
S_fields[ptr('S')] = self.S S_fields[ptr('S')] = self.S
else: else:
S_fields = OrderedDict() S_fields = OrderedDict()
''' '''
PML PML
@ -167,7 +167,7 @@ class Simulation(object):
p0 = numpy.exp(-(sigma + alpha) * dt) p0 = numpy.exp(-(sigma + alpha) * dt)
p1 = sigma / (sigma + alpha) * (p0 - 1) p1 = sigma / (sigma + alpha) * (p0 - 1)
return p0, p1 return p0, p1
xen, xep, xhn, xhp = (numpy.arange(1, pml_thickness + 1, dtype=float_type)[::-1] for _ in range(4)) xen, xep, xhn, xhp = (numpy.arange(1, pml_thickness + 1, dtype=float_type)[::-1] for _ in range(4))
xep -= 0.5 xep -= 0.5
xhn -= 0.5 xhn -= 0.5
@ -190,7 +190,7 @@ class Simulation(object):
for ne, nh in zip(*psi_names): for ne, nh in zip(*psi_names):
pml_e_fields[ptr(ne)] = pyopencl.array.zeros(self.queue, tuple(psi_shape), dtype=self.arg_type) pml_e_fields[ptr(ne)] = pyopencl.array.zeros(self.queue, tuple(psi_shape), dtype=self.arg_type)
pml_h_fields[ptr(nh)] = pyopencl.array.zeros(self.queue, tuple(psi_shape), dtype=self.arg_type) pml_h_fields[ptr(nh)] = pyopencl.array.zeros(self.queue, tuple(psi_shape), dtype=self.arg_type)
self.pml_e_fields = pml_e_fields self.pml_e_fields = pml_e_fields
self.pml_h_fields = pml_h_fields self.pml_h_fields = pml_h_fields