Update generate_Laguerre_Gauss_SLM.py

Function that generates grating is corrected.
khoroshyy · Sep 28, 2016 · f675150 · f675150
1 parent 4b9598c
commit f675150
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/generate_Laguerre_Gauss_SLM.py b/generate_Laguerre_Gauss_SLM.py
@@ -93,10 +93,10 @@ def generate_displaced_LG_Mask(beamCharge, gratingPeriod):
         image = np.zeros([ImgResY, ImgResX])
     # if period > 0 ... shift beam to the RIGHT wrt period = 0
     elif gratingPeriod > 0:             
-        image = np.angle(np.exp((2*np.pi*X/gratingPeriod)*1j)+np.exp((beamCharge*np.angle(X+Y*1j))*1j))
+        image = np.angle(np.exp((2*np.pi*X/gratingPeriod)*1j)*np.exp((beamCharge*np.angle(X+Y*1j))*1j))
     # if period < 0 ... shift beam to the LEFT wrt period = 0
     elif gratingPeriod < 0:             
-        image = np.angle(np.exp((2*np.pi*X/gratingPeriod+np.pi)*1j)+np.exp((beamCharge*np.angle(X+Y*1j))*1j))
+        image = np.angle(np.exp((2*np.pi*X/gratingPeriod+np.pi)*1j)*np.exp((beamCharge*np.angle(X+Y*1j))*1j))
 
     image8bit = normalize_image(image)