cSAXS/cdtools
0
0
Fork 0
mirror of https://github.com/cdtools-developers/cdtools.git synced 2026-09-17 00:19:55 +02:00
Code Issues Packages Projects Releases Wiki Activity

.

Browse Source
This commit is contained in:
Anastasiia Kutakh
2021-08-05 22:55:43 -04:00
parent 5f93deb11e
commit 3db2d0cdf6
2 changed files with 30 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
CDTools/tools/polarization/polarization.py
+3 -2
View File
@@ -12,7 +12,8 @@ __all__ = ['apply_linear_polarizer',
'apply_half_wave_plate',
'apply_quarter_wave_plate',
'apply_circular_polarizer',
'apply_jones_matrix']
'apply_jones_matrix',
'generate_linear_polarizer']
# Abe - split these into two functions
@@ -29,7 +30,7 @@ def generate_linear_polarizer(pol_angle):
pol_angle_rad = t.deg2rad(pol_angle)
jones_matrices = t.stack([t.tensor([[(t.cos(p)) ** 2, t.sin(p) * t.cos(p)],
[t.sin(p) * t.cos(p), (t.sin(p)) ** 2]])
for p in polarizer])
for p in pol_angle_rad])
if single_angle:
return jones_matrices[0].to(dtype=t.cfloat)
else: