Merge pull request #910 from Skyhawk172/roman-docs-updates

Roman docs & figures updates

docs/index.rst

@@ -15,7 +15,8 @@ WebbPSF is a Python package that computes simulated point spread functions (PSFs
    :align: center
    :alt: Sample PSFs for the filters in the Roman WFI.
 
-   Sample PSFs for the filters in the Roman WFI.
+   Sample PSFs for the filters in the Roman WFI on SCA01. Note that
+   the prism and grism PSFs shown here are monochromatic.
 
 
 **Contributors:**

docs/roman.rst

@@ -16,7 +16,9 @@ Wide Field Instrument (WFI)
    :align: center
    :alt: Sample PSFs for the filters in the Roman WFI.
 
-   Sample PSFs for the filters in the Roman WFI. Angular scale in arcseconds, log-scaled intensity.
+   Sample PSFs for the filters in the Roman WFI on SCA01. Angular
+   scale in arcseconds, log-scaled intensity. Note that the prism and
+   grism PSFs shown here are monochromatic.
 
 The WFI model is based on the `Cycle 9 instrument reference information <https://roman.gsfc.nasa.gov/science/Roman_Reference_Information.html>`_ from the Roman team at Goddard Space Flight Center (GSFC). The reported jitter for the Roman observatory is 0.012 arcsec per axis, per `GSFC <https://roman.ipac.caltech.edu/sims/Param_db.html#telescope>`_.
 

docs/roman_figures/how_tos.py

@@ -2,35 +2,41 @@
 from webbpsf import display_psf, roman
 
 #### Create webbpsf-roman_page_header.png
+long = 4
+wide = 3
+
+fig = plt.figure(figsize=(10, 8))
+gs = fig.add_gridspec(wide, long, hspace=0.2, wspace=-0.15)
+ax = gs.subplots(sharey=True, sharex=True)
+axes = ax.flatten()
+
 wfi = roman.WFI()
 
 all_filters = [f for f in wfi.filter_list]
 
-long = 6
-wide = 2
+for i, ifilter in enumerate(sorted(all_filters)):
+    ax = axes[i]
+
+    wfi.filter = ifilter
 
-fig, axs = plt.subplots(wide, long, figsize=(12, 6), sharey=True)
+    nlambda = None  # use defaults
+    if wfi.filter in ['PRISM', 'GRISM0', 'GRISM1']:
+        nlambda = 1
 
-for i, filter in enumerate(sorted(all_filters)):
-    r = int(np.floor(i / long))
-    c = (i % long if i < long
-         else (i % long) + 1) # remove else for left-justified bottom row
-    ax = axs[r][c]
+    psf = wfi.calc_psf(oversample=4, nlambda=nlambda)
 
-    wfi.filter = filter
-    psf = wfi.calc_psf(oversample=4)
+    display_psf(psf, ax=ax, colorbar=False, title=ifilter)
+
+    if i not in [0, 4, 8]:
+        ax.tick_params(axis='y', length=0)
+    if i == 7:
+        ax.tick_params(axis='x', reset=True, top=False)
 
-    display_psf(psf, ax=ax, colorbar=False, title=filter)
-    ax.title.set_fontsize(20)
-    ax.tick_params(axis='both', labelsize=10)
     ax.xaxis.label.set_visible(False)
     ax.yaxis.label.set_visible(False)
 
+axes[-1].remove()
 
-axs[-1][0].remove() # change last index to -1 to justify left
-
-fig.tight_layout(w_pad=.1, h_pad=0)
-fig.tight_layout(w_pad=.1, h_pad=0) # calling twice somehow tightens h_pad
 # fig.savefig('webbpsf-roman_page_header.png', dpi=100, facecolor='w')
 
 #### Create compare_wfi_sca09_sca17.png