xarpes 0.3.3__tar.gz → 0.4.0__tar.gz

xarpes-0.3.3/README.md → xarpes-0.4.0/PKG-INFO

@@ -1,3 +1,26 @@
+Metadata-Version: 2.4
+Name: xarpes
+Version: 0.4.0
+Summary: Extraction from angle resolved photoemission spectra
+Author: xARPES Developers
+Requires-Python: >=3.7.0
+Description-Content-Type: text/markdown
+Classifier: License :: OSI Approved :: GNU General Public License v3 (GPLv3)
+Classifier: Programming Language :: Python :: 3
+License-File: LICENSE
+Requires-Dist: igor2>=0.5.8
+Requires-Dist: jupyterlab
+Requires-Dist: jupytext
+Requires-Dist: matplotlib
+Requires-Dist: numpy
+Requires-Dist: scipy
+Requires-Dist: lmfit
+Requires-Dist: pyqt5
+Requires-Dist: ipympl>=0.9.3
+Requires-Dist: ipywidgets>=8.1.5
+Requires-Dist: ipykernel<6.32.0
+Project-URL: Documentation, https://xarpes.github.io
+
 # xARPES 
 
 ![xARPES](https://github.com/xARPES/xARPES/raw/main/logo/xarpes.svg)
@@ -10,7 +33,7 @@ This project is currently undergoing **beta testing**. Some of the functionaliti
 
 # Contributing
 
-Contributions to the code are most welcome. xARPES is intended to co-develop alongside the increasing complexity of experimental ARPES data sets. Contributions can be made by forking the code and creating a pull request. Importing of file formats from different beamlines is particularly encouraged.
+Contributions to the code are most welcome. xARPES is intended to co-develop alongside the increasing complexity of experimental ARPES data sets. Contributions can be made by forking the code and creating a pull request. Importing of file formats from different beamlines is particularly encouraged. Files useful for developers can be found in `/dev_tools`, such as the `Rmd2py.py` script for the development of examples.
 
 # Installation
 
@@ -19,6 +42,8 @@ xARPES installation can be divided into graphical package manager instructions,
 - via conda-forge, out-of-the-box or editable installation, sourcing the [conda-forge package](https://anaconda.org/conda-forge/xarpes).  
 - via Pip, out-of-the-box or editable installation, sourcing the [PyPI package](https://pypi.org/project/xarpes).
 
+We strongly recommend installing xARPES in a (conda/pip) virtual environment, and to activate the environment each time before activating xARPES.
+
 ## Graphical package manager installation
 
 Most IDEs and scientific Python distributions include a GUI-based package manager.  
@@ -61,11 +86,13 @@ Example for Linux:
     wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh  
     bash Miniconda3-latest-Linux-x86_64.sh
 
-Create and activate an environment:
+Answer `y` to questions. Create and activate a new environment:
 
-    conda create -n <my_env> -c defaults -c conda-forge  
+    conda create -n <my_env> -c conda-forge  
     conda activate <my_env>
 
+Where `<my_env>` must be replaced by your desired name. Package compatibility ssues may arise if conda installs from different channels. This can be prevented by appending `--strict-channel-priority` to the creation command.
+
 ### Installing xARPES
 
 #### Option A — Out-of-the-box installation (from conda-forge)
@@ -95,8 +122,8 @@ Install venv if necessary:
 
 Create and activate a virtual environment:
 
-    python3 -m venv <my_venv>  
-    source <my_venv>/bin/activate
+    python3 -m venv <my_env>  
+    source <my_env>/bin/activate
 
 Upgrade pip:
 
@@ -125,6 +152,8 @@ After installation of xARPES, the `examples/` folder can be downloaded to the cu
 
     python -c "import xarpes; xarpes.download_examples()"
 
+This attempts to download the examples from the version corresponding encountered in `__init__.py`. If no corresponding tagged version can be downloaded, the code attempts to download the latest examples instead.
+
 # Execution
 
 It is recommended to use JupyterLab to analyse data. JupyterLab is launched using:
@@ -143,3 +172,4 @@ Copyright (C) 2025 xARPES Developers
 This program is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License, version 3, as published by the Free Software Foundation.
 
 This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details.
+

xarpes-0.3.3/PKG-INFO → xarpes-0.4.0/README.md

@@ -1,24 +1,3 @@
-Metadata-Version: 2.1
-Name: xarpes
-Version: 0.3.3
-Summary: Extraction from angle resolved photoemission spectra
-Author: xARPES Developers
-Requires-Python: >=3.7.0
-Description-Content-Type: text/markdown
-Classifier: License :: OSI Approved :: GNU General Public License v3 (GPLv3)
-Classifier: Programming Language :: Python :: 3
-Requires-Dist: igor2>=0.5.8
-Requires-Dist: jupyterlab
-Requires-Dist: jupytext
-Requires-Dist: matplotlib
-Requires-Dist: numpy
-Requires-Dist: scipy
-Requires-Dist: lmfit
-Requires-Dist: pyqt5
-Requires-Dist: ipympl
-Requires-Dist: ipykernel<6.32.0
-Project-URL: Documentation, https://xarpes.github.io
-
 # xARPES 
 
 ![xARPES](https://github.com/xARPES/xARPES/raw/main/logo/xarpes.svg)
@@ -31,7 +10,7 @@ This project is currently undergoing **beta testing**. Some of the functionaliti
 
 # Contributing
 
-Contributions to the code are most welcome. xARPES is intended to co-develop alongside the increasing complexity of experimental ARPES data sets. Contributions can be made by forking the code and creating a pull request. Importing of file formats from different beamlines is particularly encouraged.
+Contributions to the code are most welcome. xARPES is intended to co-develop alongside the increasing complexity of experimental ARPES data sets. Contributions can be made by forking the code and creating a pull request. Importing of file formats from different beamlines is particularly encouraged. Files useful for developers can be found in `/dev_tools`, such as the `Rmd2py.py` script for the development of examples.
 
 # Installation
 
@@ -40,6 +19,8 @@ xARPES installation can be divided into graphical package manager instructions,
 - via conda-forge, out-of-the-box or editable installation, sourcing the [conda-forge package](https://anaconda.org/conda-forge/xarpes).  
 - via Pip, out-of-the-box or editable installation, sourcing the [PyPI package](https://pypi.org/project/xarpes).
 
+We strongly recommend installing xARPES in a (conda/pip) virtual environment, and to activate the environment each time before activating xARPES.
+
 ## Graphical package manager installation
 
 Most IDEs and scientific Python distributions include a GUI-based package manager.  
@@ -82,11 +63,13 @@ Example for Linux:
     wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh  
     bash Miniconda3-latest-Linux-x86_64.sh
 
-Create and activate an environment:
+Answer `y` to questions. Create and activate a new environment:
 
-    conda create -n <my_env> -c defaults -c conda-forge  
+    conda create -n <my_env> -c conda-forge  
     conda activate <my_env>
 
+Where `<my_env>` must be replaced by your desired name. Package compatibility ssues may arise if conda installs from different channels. This can be prevented by appending `--strict-channel-priority` to the creation command.
+
 ### Installing xARPES
 
 #### Option A — Out-of-the-box installation (from conda-forge)
@@ -116,8 +99,8 @@ Install venv if necessary:
 
 Create and activate a virtual environment:
 
-    python3 -m venv <my_venv>  
-    source <my_venv>/bin/activate
+    python3 -m venv <my_env>  
+    source <my_env>/bin/activate
 
 Upgrade pip:
 
@@ -146,6 +129,8 @@ After installation of xARPES, the `examples/` folder can be downloaded to the cu
 
     python -c "import xarpes; xarpes.download_examples()"
 
+This attempts to download the examples from the version corresponding encountered in `__init__.py`. If no corresponding tagged version can be downloaded, the code attempts to download the latest examples instead.
+
 # Execution
 
 It is recommended to use JupyterLab to analyse data. JupyterLab is launched using:
@@ -164,4 +149,3 @@ Copyright (C) 2025 xARPES Developers
 This program is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License, version 3, as published by the Free Software Foundation.
 
 This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details.
-

{xarpes-0.3.3 → xarpes-0.4.0}/examples/graphene/graphene.Rmd

@@ -8,7 +8,7 @@ jupyter:
       format_version: '1.2'
       jupytext_version: 1.15.2
   kernelspec:
-    display_name: 3.7_env
+    display_name: Python 3 (ipykernel)
     language: python
     name: python3
 ---
@@ -30,9 +30,6 @@ import os
 
 # Default plot configuration from xarpes.plotting.py
 xarpes.plot_settings('default')
-
-# If needed, close figures before running other cells, if figures show up in the wrong places.
-# get_ipython().events.register('pre_run_cell', lambda info: plt.close('all'))
 ```
 
 ```{python}
@@ -48,8 +45,7 @@ data_file_path = os.path.join(script_dir, dfld, flnm + extn)
 ```{python}
 # %matplotlib inline
 
-fig = plt.figure(figsize=(8, 5))
-ax = fig.gca()
+fig = plt.figure(figsize=(8, 5)); ax = fig.gca()
 
 bmap = xarpes.BandMap(data_file_path, energy_resolution=0.01,
                       angle_resolution=0.1, temperature=50)
@@ -64,13 +60,13 @@ fig = bmap.plot(abscissa='momentum', ordinate='kinetic_energy', ax=ax)
 
 fig, ax = plt.subplots(2, 1, figsize=(6, 8))
 
-fig = bmap.correct_fermi_edge(
+bmap.correct_fermi_edge(
       hnuminphi_guess=32, background_guess=1e2,
       integrated_weight_guess=1e3, angle_min=-10, angle_max=10,
       ekin_min=31.96, ekin_max=32.1, true_angle=0,
       ax=ax[0], show=False, fig_close=False)
 
-fig = bmap.plot(ordinate='electron_energy', abscissa='momentum',
+bmap.plot(ordinate='electron_energy', abscissa='momentum',
       ax=ax[1], show=False, fig_close=False)
 
 # Figure customization
@@ -124,9 +120,9 @@ fig = mdcs.visualize_guess(distributions=guess_dists, energy_value=energy_value,
 ```
 
 #### Note on interactive figures
-- Operability of the interactive figure is sensitive to configuration of the plotting backend.
-- With Python V>=3.13 in a conda environment, the figure may fail to show up more than once. In that case, the user could try to run "conda env config vars unset MPLBACKEND" inside the conda environment; in Python, this command can be preceded with an exclamation mark ("!conda env ...") such that the command is executed in the terminal. The reset should work after restarting the conda environment.
-- As a fallback, the user may switch "%matplotlib widget" to "%matplotlib qt", after which the figure should pop up in an external window.
+- The interactive figure might not work inside the Jupyter notebooks, despite our best efforts to ensure stability.
+- As a fallback, the user may switch from "%matplotlib widget" to "%matplotlib qt", after which the figure should pop up in an external window.
+- For some package versions, a static version of the interactive widget may spuriously show up inside other cells. In that case, uncomment the #get_ipython()... line in the first cell for your notebooks.
 
 ```{python}
 # %matplotlib widget
@@ -139,53 +135,29 @@ mdcs = xarpes.MDCs(*bmap.mdc_set(angle_min, angle_max, energy_range=energy_range
 fig = mdcs.fit_selection(distributions=guess_dists, ax=ax)
 ```
 
-```{python}
-# # %matplotlib widget
-
-# fig = plt.figure(figsize=(7, 5))
-# ax = fig.gca()
-
-# mdcs.plot(energy_range=[-0.15, -0.05], ax=ax)
-```
-
 ```{python}
 # %matplotlib inline
 
-self_energy = xarpes.SelfEnergy(*mdcs.expose_parameters(select_label='Linear_test_1',
-                                fermi_velocity=2.85, fermi_wavevector=0.358))
-```
-
-```{python}
-# %matplotlib inline
+plt.rcParams['lines.markersize'] = 0.8
 
 fig = plt.figure(figsize=(8, 5))
 ax = fig.gca()
 
-from xarpes.constants import stdv
+self_energy = xarpes.SelfEnergy(*mdcs.expose_parameters(select_label='Linear_test_1',
+                                fermi_velocity=2.85, fermi_wavevector=0.358))
 
-ax.errorbar(self_energy.peak_positions, self_energy.enel_range, 
-            xerr=stdv * self_energy.peak_positions_sigma,
-           markersize=2, color='tab:blue', label=self_energy.label)
+self_energies = xarpes.CreateSelfEnergies([self_energy])
 
-fig = bmap.plot(abscissa='momentum', ordinate='electron_energy', ax=ax)
+fig = bmap.plot(abscissa='momentum', ordinate='electron_energy', ax=ax,
+                self_energies=self_energies)
 ```
 
 ```{python}
 # %matplotlib inline
 
-fig = plt.figure(figsize=(7, 5))
-ax = fig.gca()
-
-from xarpes.constants import stdv
-
-ax.errorbar(self_energy.enel_range, self_energy.imag, 
-            yerr=stdv * self_energy.imag_sigma, label =r"$-\Sigma''(E)$")
-ax.errorbar(self_energy.enel_range, self_energy.real, 
-            yerr=stdv * self_energy.real_sigma, label =r"$\Sigma'(E)$")
-ax.set_xlabel(r'$E-\mu$ (eV)'); ax.set_ylabel(r"$\Sigma'(E), -\Sigma''(E)$ (eV)")
+fig = plt.figure(figsize=(7, 5)); ax = fig.gca()
 
-plt.legend()
-plt.show()
+fig = self_energy.plot_real(ax=ax)
 ```
 
 ```{python}
@@ -194,6 +166,8 @@ plt.show()
 angle_min2 = -1e6
 angle_max2 = 0
 
+plt.rcParams['lines.markersize'] = 3.0
+
 mdc2 = xarpes.MDCs(*bmap.mdc_set(angle_min2, angle_max2, energy_range=energy_range))
 
 guess_dists2 = xarpes.CreateDistributions([
@@ -202,8 +176,7 @@ xarpes.SpectralLinear(amplitude=450, peak=-7.25, broadening=0.01,
                       name='Linear_left', index='1'),
 ])
 
-fig = plt.figure(figsize=(8, 6))
-ax = fig.gca()
+fig = plt.figure(figsize=(8, 6)); ax = fig.gca()
 
 fig = mdc2.visualize_guess(distributions=guess_dists2, energy_value=0, ax=ax)
 ```
@@ -213,15 +186,55 @@ fig = mdc2.visualize_guess(distributions=guess_dists2, energy_value=0, ax=ax)
 
 fig = mdc2.fit_selection(distributions=guess_dists2, show=False, fig_close=True)
 
-self_left = xarpes.SelfEnergy(*mdc2.expose_parameters(select_label='Linear_left_1'))
+self_left = xarpes.SelfEnergy(*mdc2.expose_parameters(select_label='Linear_left_1',
+fermi_velocity=-2.67, fermi_wavevector=-0.354))
+```
+
+```{python}
+# %matplotlib inline
+
+fig = plt.figure(figsize=(8, 5)); ax = fig.gca()
+
+self_energies= xarpes.CreateSelfEnergies([
+    self_energy, self_left])
+
+fig = bmap.plot(abscissa='momentum', ordinate='electron_energy',
+                self_energies=self_energies, plot_dispersions='full',
+                ax=ax)
 ```
 
 ```{python}
+# %matplotlib inline
+fig = plt.figure(figsize=(9, 6)); ax = fig.gca()
+
+self_left.plot_both(ax=ax, show=False, fig_close=False)
+self_energy.plot_both(ax=ax, show=False, fig_close=False)
+
+ax.set_xlim([-0.275, 0.025])
+ax.set_ylim([-0.025, 0.275])
 
+plt.legend(); plt.show()
 ```
 
 ```{python}
+# %matplotlib inline
+fig = plt.figure(figsize=(8, 6)); ax = fig.gca()
+
+self_left.plot_both(ax=ax, show=False, fig_close=False)
+self_energy.plot_both(ax=ax, show=False, fig_close=False)
+
+ax.set_xlim([-0.275, 0.025]); ax.set_ylim([-0.025, 0.275])
 
+# Replace labels with custom labels
+left_real, left_imag, right_real, right_imag = ax.get_lines()
+
+labels = [
+    r"$\Sigma_{\mathrm{L}}'(E)$", r"$-\Sigma_{\mathrm{L}}''(E)$",
+    r"$\Sigma_{\mathrm{R}}'(E)$", r"$-\Sigma_{\mathrm{R}}''(E)$",
+]
+
+ax.legend([left_real, left_imag, right_real, right_imag], labels)
+plt.show()
 ```
 
 ```{python}
@@ -413,4 +426,6 @@ self_left = xarpes.SelfEnergy(*mdc2.expose_parameters(select_label='Linear_left_
 # ax = fig.gca()
 
 # fig = mdcs.plot(distributions=guess_dists, ax=ax, energy_value=-0.003)
+
+
 ```

{xarpes-0.3.3 → xarpes-0.4.0}/examples/graphene/graphene.py

@@ -17,8 +17,6 @@ import os
 # Default plot configuration from xarpes.plotting.py
 xarpes.plot_settings('default')
 
-# If needed, close figures before running other cells, if figures show up in the wrong places.
-
 script_dir = xarpes.set_script_dir()
 
 dfld = 'data_sets'    # Folder containing the data
@@ -28,8 +26,7 @@ extn = '.ibw'         # Extension of the file
 data_file_path = os.path.join(script_dir, dfld, flnm + extn)
 
 
-fig = plt.figure(figsize=(8, 5))
-ax = fig.gca()
+fig = plt.figure(figsize=(8, 5)); ax = fig.gca()
 
 bmap = xarpes.BandMap(data_file_path, energy_resolution=0.01,
                       angle_resolution=0.1, temperature=50)
@@ -41,13 +38,13 @@ fig = bmap.plot(abscissa='momentum', ordinate='kinetic_energy', ax=ax)
 
 fig, ax = plt.subplots(2, 1, figsize=(6, 8))
 
-fig = bmap.correct_fermi_edge(
+bmap.correct_fermi_edge(
       hnuminphi_guess=32, background_guess=1e2,
       integrated_weight_guess=1e3, angle_min=-10, angle_max=10,
       ekin_min=31.96, ekin_max=32.1, true_angle=0,
       ax=ax[0], show=False, fig_close=False)
 
-fig = bmap.plot(ordinate='electron_energy', abscissa='momentum',
+bmap.plot(ordinate='electron_energy', abscissa='momentum',
       ax=ax[1], show=False, fig_close=False)
 
 # Figure customization
@@ -94,9 +91,9 @@ ax = fig.gca()
 fig = mdcs.visualize_guess(distributions=guess_dists, energy_value=energy_value, ax=ax)
 
 # #### Note on interactive figures
-# - Operability of the interactive figure is sensitive to configuration of the plotting backend.
-# - With Python V>=3.13 in a conda environment, the figure may fail to show up more than once. In that case, the user could try to run "conda env config vars unset MPLBACKEND" inside the conda environment; in Python, this command can be preceded with an exclamation mark ("!conda env ...") such that the command is executed in the terminal. The reset should work after restarting the conda environment.
-# - As a fallback, the user may switch "%matplotlib widget" to "%matplotlib qt", after which the figure should pop up in an external window.
+# - The interactive figure might not work inside the Jupyter notebooks, despite our best efforts to ensure stability.
+# - As a fallback, the user may switch from "%matplotlib widget" to "%matplotlib qt", after which the figure should pop up in an external window.
+# - For some package versions, a static version of the interactive widget may spuriously show up inside other cells. In that case, uncomment the #get_ipython()... line in the first cell for your notebooks.
 
 
 fig = plt.figure(figsize=(7, 5))
@@ -107,46 +104,30 @@ mdcs = xarpes.MDCs(*bmap.mdc_set(angle_min, angle_max, energy_range=energy_range
 fig = mdcs.fit_selection(distributions=guess_dists, ax=ax)
 
 
-# fig = plt.figure(figsize=(7, 5))
-# ax = fig.gca()
-
-# mdcs.plot(energy_range=[-0.15, -0.05], ax=ax)
-
-
-self_energy = xarpes.SelfEnergy(*mdcs.expose_parameters(select_label='Linear_test_1',
-                                fermi_velocity=2.85, fermi_wavevector=0.358))
-
+plt.rcParams['lines.markersize'] = 0.8
 
 fig = plt.figure(figsize=(8, 5))
 ax = fig.gca()
 
-from xarpes.constants import stdv
-
-ax.errorbar(self_energy.peak_positions, self_energy.enel_range, 
-            xerr=stdv * self_energy.peak_positions_sigma,
-           markersize=2, color='tab:blue', label=self_energy.label)
-
-fig = bmap.plot(abscissa='momentum', ordinate='electron_energy', ax=ax)
+self_energy = xarpes.SelfEnergy(*mdcs.expose_parameters(select_label='Linear_test_1',
+                                fermi_velocity=2.85, fermi_wavevector=0.358))
 
+self_energies = xarpes.CreateSelfEnergies([self_energy])
 
-fig = plt.figure(figsize=(7, 5))
-ax = fig.gca()
+fig = bmap.plot(abscissa='momentum', ordinate='electron_energy', ax=ax,
+                self_energies=self_energies)
 
-from xarpes.constants import stdv
 
-ax.errorbar(self_energy.enel_range, self_energy.imag, 
-            yerr=stdv * self_energy.imag_sigma, label =r"$-\Sigma''(E)$")
-ax.errorbar(self_energy.enel_range, self_energy.real, 
-            yerr=stdv * self_energy.real_sigma, label =r"$\Sigma'(E)$")
-ax.set_xlabel(r'$E-\mu$ (eV)'); ax.set_ylabel(r"$\Sigma'(E), -\Sigma''(E)$ (eV)")
+fig = plt.figure(figsize=(7, 5)); ax = fig.gca()
 
-plt.legend()
-plt.show()
+fig = self_energy.plot_real(ax=ax)
 
 
 angle_min2 = -1e6
 angle_max2 = 0
 
+plt.rcParams['lines.markersize'] = 3.0
+
 mdc2 = xarpes.MDCs(*bmap.mdc_set(angle_min2, angle_max2, energy_range=energy_range))
 
 guess_dists2 = xarpes.CreateDistributions([
@@ -155,8 +136,7 @@ xarpes.SpectralLinear(amplitude=450, peak=-7.25, broadening=0.01,
                       name='Linear_left', index='1'),
 ])
 
-fig = plt.figure(figsize=(8, 6))
-ax = fig.gca()
+fig = plt.figure(figsize=(8, 6)); ax = fig.gca()
 
 fig = mdc2.visualize_guess(distributions=guess_dists2, energy_value=0, ax=ax)
 
@@ -164,11 +144,46 @@ fig = mdc2.visualize_guess(distributions=guess_dists2, energy_value=0, ax=ax)
 
 fig = mdc2.fit_selection(distributions=guess_dists2, show=False, fig_close=True)
 
-self_left = xarpes.SelfEnergy(*mdc2.expose_parameters(select_label='Linear_left_1'))
+self_left = xarpes.SelfEnergy(*mdc2.expose_parameters(select_label='Linear_left_1',
+fermi_velocity=-2.67, fermi_wavevector=-0.354))
+
+
+fig = plt.figure(figsize=(8, 5)); ax = fig.gca()
+
+self_energies= xarpes.CreateSelfEnergies([
+    self_energy, self_left])
+
+fig = bmap.plot(abscissa='momentum', ordinate='electron_energy',
+                self_energies=self_energies, plot_dispersions='full',
+                ax=ax)
+
+fig = plt.figure(figsize=(9, 6)); ax = fig.gca()
 
+self_left.plot_both(ax=ax, show=False, fig_close=False)
+self_energy.plot_both(ax=ax, show=False, fig_close=False)
 
+ax.set_xlim([-0.275, 0.025])
+ax.set_ylim([-0.025, 0.275])
 
+plt.legend(); plt.show()
 
+fig = plt.figure(figsize=(8, 6)); ax = fig.gca()
+
+self_left.plot_both(ax=ax, show=False, fig_close=False)
+self_energy.plot_both(ax=ax, show=False, fig_close=False)
+
+ax.set_xlim([-0.275, 0.025]); ax.set_ylim([-0.025, 0.275])
+
+# Replace labels with custom labels
+left_real, left_imag, right_real, right_imag = ax.get_lines()
+
+labels = [
+    r"$\Sigma_{\mathrm{L}}'(E)$", r"$-\Sigma_{\mathrm{L}}''(E)$",
+    r"$\Sigma_{\mathrm{R}}'(E)$", r"$-\Sigma_{\mathrm{R}}''(E)$",
+]
+
+ax.legend([left_real, left_imag, right_real, right_imag], labels)
+plt.show()
 
 
 # fig = plt.figure(figsize=(8, 5))
@@ -329,3 +344,5 @@ self_left = xarpes.SelfEnergy(*mdc2.expose_parameters(select_label='Linear_left_
 # ax = fig.gca()
 
 # fig = mdcs.plot(distributions=guess_dists, ax=ax, energy_value=-0.003)
+
+