jan/opencl_fdtd
2
2
Fork 1
Code Issues Pull Requests Releases Wiki Activity

Make some variables lower-case

Browse Source
This commit is contained in:
jan 2016-06-21 18:26:42 -07:00
parent 73aaa720b8
commit f4b2df2b2c
1 changed files with 15 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
fdtd/boundary.py
View File

@ -22,14 +22,14 @@ def conductor(direction: int,
uv = 'xyz'.replace(r, '') uv = 'xyz'.replace(r, '')
if polarity < 0: if polarity < 0:
bc_E = """ bc_e = """
if ({r} == 0) {{ if ({r} == 0) {{
E{r}[i] = 0; E{r}[i] = 0;
E{u}[i] = E{u}[i+di{r}]; E{u}[i] = E{u}[i+di{r}];
E{v}[i] = E{v}[i+di{r}]; E{v}[i] = E{v}[i+di{r}];
}} }}
""" """
bc_H = """ bc_h = """
if ({r} == 0) {{ if ({r} == 0) {{
H{r}[i] = H{r}[i+di{r}]; H{r}[i] = H{r}[i+di{r}];
H{u}[i] = 0; H{u}[i] = 0;
@ -38,7 +38,7 @@ if ({r} == 0) {{
""" """
elif polarity > 0: elif polarity > 0:
bc_E = """ bc_e = """
if ({r} == s{r} - 1) {{ if ({r} == s{r} - 1) {{
E{r}[i] = -E{r}[i-2*di{r}]; E{r}[i] = -E{r}[i-2*di{r}];
E{u}[i] = +E{u}[i-di{r}]; E{u}[i] = +E{u}[i-di{r}];
@ -47,7 +47,7 @@ if ({r} == s{r} - 1) {{
E{r}[i] = 0; E{r}[i] = 0;
}} }}
""" """
bc_H = """ bc_h = """
if ({r} == s{r} - 1) {{ if ({r} == s{r} - 1) {{
H{r}[i] = +H{r}[i-di{r}]; H{r}[i] = +H{r}[i-di{r}];
H{u}[i] = -H{u}[i-2*di{r}]; H{u}[i] = -H{u}[i-2*di{r}];
@ -61,7 +61,7 @@ if ({r} == s{r} - 1) {{
raise Exception() raise Exception()
replacements = {'r': r, 'u': uv[0], 'v': uv[1]} replacements = {'r': r, 'u': uv[0], 'v': uv[1]}
return [s.format(**replacements) for s in (bc_E, bc_H)] return [s.format(**replacements) for s in (bc_e, bc_h)]
def cpml(direction: int, def cpml(direction: int,
@ -105,12 +105,12 @@ def cpml(direction: int,
np = 'nVp'[numpy.sign(polarity)+1] np = 'nVp'[numpy.sign(polarity)+1]
uv = ['xyz'[i] for i in transverse] uv = ['xyz'[i] for i in transverse]
xE = numpy.arange(1, thickness+1, dtype=float)[::-1] xe = numpy.arange(1, thickness+1, dtype=float)[::-1]
xH = numpy.arange(1, thickness+1, dtype=float)[::-1] xh = numpy.arange(1, thickness+1, dtype=float)[::-1]
if polarity > 0: if polarity > 0:
xE -= 0.5 xe -= 0.5
elif polarity < 0: elif polarity < 0:
xH -= 0.5 xh -= 0.5
def par(x): def par(x):
sigma = ((x / thickness) ** m[0]) * sigma_max sigma = ((x / thickness) ** m[0]) * sigma_max
@ -118,8 +118,8 @@ def cpml(direction: int,
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
p0e, p1e = par(xE) p0e, p1e = par(xe)
p0h, p1h = par(xH) p0h, p1h = par(xh)
vals = {'r': r, vals = {'r': r,
'u': uv[0], 'u': uv[0],
@ -148,7 +148,7 @@ if ( (s{r} - 1) > {r} && {r} > (s{r} - 1) - ({th} + 1) ) {{
else: else:
raise Exception('Bad polarity (=0)') raise Exception('Bad polarity (=0)')
code_E = """ code_e = """
// pml parameters: // pml parameters:
const float p0[{th}] = {{ {p0e} }}; const float p0[{th}] = {{ {p0e} }};
const float p1[{th}] = {{ {p1e} }}; const float p1[{th}] = {{ {p1e} }};
@ -160,7 +160,7 @@ if ( (s{r} - 1) > {r} && {r} > (s{r} - 1) - ({th} + 1) ) {{
E{v}[i] {sh}= dt / eps{v}[i] * Psi_{r}{np}_E{v}[ip]; E{v}[i] {sh}= dt / eps{v}[i] * Psi_{r}{np}_E{v}[ip];
}} }}
""" """
code_H = """ code_h = """
// pml parameters: // pml parameters:
const float p0[{th}] = {{ {p0h} }}; const float p0[{th}] = {{ {p0h} }};
const float p1[{th}] = {{ {p1h} }}; const float p1[{th}] = {{ {p1h} }};
@ -174,8 +174,8 @@ if ( (s{r} - 1) > {r} && {r} > (s{r} - 1) - ({th} + 1) ) {{
""" """
pml_data = { pml_data = {
'E': (bounds_if + code_E).format(**vals), 'E': (bounds_if + code_e).format(**vals),
'H': (bounds_if + code_H).format(**vals), 'H': (bounds_if + code_h).format(**vals),
'psi_E': ['Psi_{r}{np}_E{u}'.format(**vals), 'psi_E': ['Psi_{r}{np}_E{u}'.format(**vals),
'Psi_{r}{np}_E{v}'.format(**vals)], 'Psi_{r}{np}_E{v}'.format(**vals)],
'psi_H': ['Psi_{r}{np}_H{u}'.format(**vals), 'psi_H': ['Psi_{r}{np}_H{u}'.format(**vals),