Update zern_generator.py

zern_generator.py

@@ -36,6 +36,7 @@ def generate_zern_wavefront_fig(resolution, *zerns):
 
 
 def generate_batch(resolution):
+    term = ["PISTON","V-TILT","H-TILT","DEFOCUS0","O-PR.ASTIGMATISM","V-PR.ASTIGMATISM","V-PR.COMA","H-PR.COMA","V-TREFOIL","O-TREFOIL","P-SPHERICAL","O-SC.ASTIGMATISM","V-SC.ASTIGMATISM","V-QUADRAFOIL","O-QUADRAFOIL","V-SC.COMA","H-SC.COMA","V-SC-TREFOIL","O-SC-TREFOIL","O-PENTAFOIL","V-PENTAFOIL"]
     ddx = np.linspace(*span, resolution)
     ddy = np.linspace(*span, resolution)
     xv, yv = np.meshgrid(ddx, ddy)
@@ -51,7 +52,7 @@ def generate_batch(resolution):
             zern = [0] * (zern_index - 1) + [1]
             ax = plt.subplot2grid((5, 6), (i, j))
             Phi = generate_zern_wavefront(xv, yv, zern)
-            ax.set_title(f"Zernike {zern_index-1}")
+            ax.set_title(f"Zernike {zern_index-1} \n {term[i]} ")
             ax.imshow(Phi, extent=(*span, *span), cmap='gray')
 
             zern_index += 1