QPMS
/
qpms
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

More cosmetics 

Former-commit-id: 2ffd8031569cf35cb913dd30fa0046884df611f0
This commit is contained in:
Marek Nečada 2017-05-17 13:44:57 +03:00
parent 43702a00c8
commit a5163c7e8d
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
misc/finitesqlatzsym-scatter.py
View File

@ -1,6 +1,6 @@
#!/usr/bin/env python3 #!/usr/bin/env python3
import argparse, re, random, string import argparse, re, random, string, sys
import subprocess import subprocess
from scipy.constants import hbar, e as eV, pi, c from scipy.constants import hbar, e as eV, pi, c
@ -115,7 +115,7 @@ btime=_time_b(verbose)
import qpms import qpms
import numpy as np import numpy as np
import os, sys, warnings, math import os, warnings, math
from scipy import interpolate from scipy import interpolate
nx = None nx = None
s3 = math.sqrt(3) s3 = math.sqrt(3)
@ -298,6 +298,7 @@ for action in actions:
scat.prepare_partial(action, verbose=verbose) scat.prepare_partial(action, verbose=verbose)
actionstring = '.TM' if action else '.TE' actionstring = '.TM' if action else '.TE'
for chunki in range(chunkn): for chunki in range(chunkn):
sbtime = _time_b(verbose, step='Solving the scattering problem, chunk %d'%chunki+actionstring)
if pargs.nosuffix: if pargs.nosuffix:
outfile = pargs.output_prefix + actionstring + ( outfile = pargs.output_prefix + actionstring + (
('.%03d' % chunki) if chunkn > 1 else '') ('.%03d' % chunki) if chunkn > 1 else '')
@ -336,6 +337,7 @@ for action in actions:
if action == 1 or action is None: if action == 1 or action is None:
pq = np.array(qpms.plane_pq_y(lMax, kdir, xu)).ravel()[TMč] * phases[:, nx] pq = np.array(qpms.plane_pq_y(lMax, kdir, xu)).ravel()[TMč] * phases[:, nx]
zresult[i] = scat.scatter_partial(1, pq) zresult[i] = scat.scatter_partial(1, pq)
_time_e(sbtime, verbose, step='Solving the scattering problem, chunk %d'%chunki+actionstring)
metadata[()]['chunki'] = chunki metadata[()]['chunki'] = chunki
if action is None: if action is None: