code: def plot_waves(self, ax=None, fontsize=12, **kwargs):
          
          ax, fig, plt = get_ax_fig_plt(ax)
  
          ax.grid(True)
          ax.set_xlabel("r [Bohr]")
          ax.set_ylabel(r"$r\phi,\, r\tilde\phi\, [Bohr]^{-\frac{1}{2}}$")
  
          #ax.axvline(x=self.paw_radius, linewidth=2, color='k', linestyle="--")
          #ax.annotate("$r_c$", xy=(self.paw_radius + 0.1, 0.1))
  
          for state, rfunc in self.pseudo_partial_waves.items():
              ax.plot(rfunc.mesh, rfunc.mesh * rfunc.values, lw=2, label="PS-WAVE: " + state)
  
          for state, rfunc in self.ae_partial_waves.items():
              ax.plot(rfunc.mesh, rfunc.mesh * rfunc.values, lw=2, label="AE-WAVE: " + state)
  
          ax.legend(loc="best", shadow=True, fontsize=fontsize)
  
          return fig
docs: Plot the AE and the pseudo partial waves.
  
          Args:
              ax: matplotlib :class:`Axes` or None if a new figure should be created.
              fontsize: fontsize for legends and titles
  
          Returns: `matplotlib` figure