PyThea 0.11.0__tar.gz → 0.13.0__tar.gz

pythea-0.13.0/.readthedocs.yaml

@@ -0,0 +1,29 @@
+# .readthedocs.yaml
+# Read the Docs configuration file
+# See https://docs.readthedocs.io/en/stable/config-file/v2.html for details
+
+# Required
+version: 2
+
+# Set the version of Python and other tools you might need
+build:
+  os: ubuntu-22.04
+  tools:
+    python: "3.9"
+    # You can also specify other tool versions:
+    # nodejs: "16"
+    # rust: "1.55"
+    # golang: "1.17"
+
+# Build documentation in the docs/ directory with Sphinx
+sphinx:
+   configuration: docs/source/conf.py
+
+# If using Sphinx, optionally build your docs in additional formats such as PDF
+# formats:
+#    - pdf
+
+# Optionally declare the Python requirements required to build your docs
+python:
+   install:
+   - requirements: docs/source/requirements.txt

{pythea-0.11.0 → pythea-0.13.0}/CHANGELOG.md

@@ -1,4 +1,43 @@
-# v0.11.0 (17-April-2024)
+# v0.13.0 (14-Jul-2024)
+
+## Features
+ - Adds imaging data from PSP/WISPR.
+ - Implements new feature of Parker spiral and HEK events visualization on the images.
+ - Adds a method to get directly the geometrical model from model_fittings.
+ - Adds tests for extensions, WISPR imaging.
+ - Adds and improves docstrings in utilities, modules, and geometrical models.
+ - Adds GCS and kinematic plots in documentation
+ - Adds function reference in the documentation
+
+## Minor Changes
+ - Changes some optional inputs for get_hek_flare, make_figure, plot_bodies, and plot_solar_reference_lines to be better utilised.
+
+# v0.12.0 (09-Jun-2024)
+
+## Features
+ - Adds more imaging data from SDO/AIA.
+ - Adds acceleration calculation in the kinematic plots.
+ - Implements acceleration plotting in the app.
+ - Includes fits file name in fitting files and dataframes.
+ - Highlights a row in the fitting table when fitting exist in table.
+
+## Major Changes
+ - Decouples the loading of fits files from download_fits function.
+
+## Minor Changes
+ - Improves configuration dictionary for selected imagers.
+ - Improves json warning message in the main app page.
+ - Adds documentation page link to the main app page.
+ - Updates README files.
+ - Improves plot_fitting_model and adds figure test.
+ - Improves application layout.
+
+## Bug Fixes
+ - Fixes a potential bug in filter_maps when no filtering is applied.
+ - Fixes a bug with progress bar with stqdm.
+ - Fixes a bug with image processing mode selection.
+
+# v0.11.0 (19-May-2024)
 
 ## Features
  - Adds tests for ellipsoid location on AIA and STEREO COR images.

{pythea-0.11.0/PyThea.egg-info → pythea-0.13.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: PyThea
-Version: 0.11.0
+Version: 0.13.0
 Summary: PyThea: A software package to reconstruct the 3D structure of CMEs and shock waves
 Home-page: https://github.com/AthKouloumvakos/PyThea
 Author: Athanasios Kouloumvakos
@@ -20,11 +20,11 @@ Classifier: Topic :: Scientific/Engineering :: Physics
 Requires-Python: >=3.8, <3.11
 Description-Content-Type: text/markdown
 License-File: LICENSE.md
-Requires-Dist: numpy
+Requires-Dist: numpy==1.26.4
 Requires-Dist: pandas
 Requires-Dist: scipy
 Requires-Dist: aiapy
-Requires-Dist: astropy
+Requires-Dist: astropy==6.0.0
 Requires-Dist: astroquery
 Requires-Dist: jplephem
 Requires-Dist: numexpr
@@ -35,7 +35,6 @@ Requires-Dist: matplotlib
 Requires-Dist: seaborn
 Requires-Dist: streamlit
 Requires-Dist: Pillow
-Requires-Dist: stqdm
 Requires-Dist: lxml
 Requires-Dist: zeep
 Requires-Dist: drms
@@ -63,8 +62,8 @@ We recommend, creating a virtual environment for _PyThea_ and installing the pac
 If you use Anaconda or Miniconda (which we also recommend) you can create a virtual environment using ```conda``` and then install _PyThea_ using from PyPI using ```pip```. In the terminal do the following:
 
 ```python
-# Create a virtual environment. Use python=3.8 or 3.9
-conda create --name PyThea python=3.9
+# Create a virtual environment. Use python>3.9
+conda create --name PyThea python=3.10
 
 # Activate the environment
 conda activate PyThea
@@ -138,7 +137,7 @@ If there is an error when running ```PyThea streamlit``` then you can manually r
 
 ## 📙 Usage
 
-Complete documentation of the _PyThea_ is under construction.
+Complete documentation of the _PyThea_ can be found in [https://www.pythea.org/](https://www.pythea.org/).
 
 ## 📦 Useful Python packages
 

{pythea-0.11.0 → pythea-0.13.0}/PyThea/PyThea_app.py

@@ -23,7 +23,6 @@ from copy import copy
 
 import astropy.units as u
 import numpy as np
-import stqdm  # See also https://github.com/tqdm/tqdm
 import streamlit as st
 from astropy.coordinates import Distance, SkyCoord, SphericalRepresentation
 from sunpy.coordinates import frames
@@ -38,8 +37,8 @@ from PyThea.modules import (date_and_event_selection, figure_streamlit,
                             fitting_and_slider_options_container,
                             fitting_sliders)
 from PyThea.sunpy_dev.map.maputils import get_closest, maps_clims
-from PyThea.utils import (download_fits, model_fittings, plot_fitting_model,
-                          single_imager_maps_process)
+from PyThea.utils import (download_fits, load_fits, model_fittings,
+                          plot_fitting_model, single_imager_maps_process)
 from PyThea.version import version
 
 
@@ -49,6 +48,13 @@ def delete_from_state(vars):
             del st.session_state[var]
 
 
+def highlight_row(row, row_index):
+    if row.name.strftime('%Y-%m-%dT%H:%M:%S.%f') == row_index.strftime('%Y-%m-%dT%H:%M:%S.%f').values[0]:
+        return ['background-color: lightpink']*len(row)
+    else:
+        return ['']*len(row)
+
+
 def footer_text():
     st.subheader('About this application:')
     st.markdown("""
@@ -65,16 +71,24 @@ def footer_text():
     right.markdown("""
                    **Github**: Find the latest version here
                                [![GitHub](https://img.shields.io/badge/github-%23121011.svg?style=for-the-badge&logo=github&logoColor=white)](https://github.com/AthKouloumvakos/PyThea) \n
-                   **Citation**: Cite PyThea as [![https://doi.org/10.5281/zenodo.5713659](https://zenodo.org/badge/DOI/10.5281/zenodo.5713659.svg)](https://doi.org/10.5281/zenodo.5713659) \n
+                   **Documentation Page**: https://www.pythea.org/
                    """ +
                    f"""
                    **Version**: {version} (latest release [![Version](https://img.shields.io/github/v/release/AthKouloumvakos/PyThea)](https://github.com/AthKouloumvakos/PyThea/releases))
                    """)
     left.image('https://github.com/AthKouloumvakos/PyThea/blob/master/docs/logo/pythea_logo.png?raw=true')
+    st.markdown("""
+                **Citation**: Please cite the following paper [![https://www.frontiersin.org/articles/10.3389/fspas.2022.974137/](https://img.shields.io/static/v1?label=Paper&message=Frontiers&color=red)](https://www.frontiersin.org/articles/10.3389/fspas.2022.974137/) and [![https://doi.org/10.5281/zenodo.5713659](https://zenodo.org/badge/DOI/10.5281/zenodo.5713659.svg)](https://doi.org/10.5281/zenodo.5713659)
+                """)
+    st.info('''
+            More imaging data have been added:
+            - SDO/AIA images from 211A channel.
+            - PSP/WISPR inner and outer telescope images.
+            ''', icon='ℹ️')
     st.warning('''
-           ⚠️ **NOTE: From PyThea >0.8.1 the JSON fitting files will be slightly different from the old ones.** ⚠️
-           * Due to a change in the fitting time input, the new fitting files may have slightly different times for the same images (see further information [here](https://github.com/AthKouloumvakos/PyThea/discussions/24)).
-           ''')
+           **NOTE: From PyThea >0.8.1 the JSON fitting files will be slightly different from the old ones.**
+           Due to a change in the fitting time input, the new fitting files may have slightly different times for the same images (see further information [here](https://github.com/AthKouloumvakos/PyThea/discussions/24)).
+           ''', icon='⚠️')
     st.markdown('---')
 
 
@@ -86,7 +100,10 @@ def run():
     #############################################################
     # set page config
     st.set_page_config(page_title='PyThea', page_icon=':rocket:',
-                       initial_sidebar_state='expanded')
+                       initial_sidebar_state='expanded',
+                       menu_items={
+                           'Get Help': 'https://www.pythea.org/',
+                           'Report a Bug': 'https://github.com/AthKouloumvakos/PyThea'})
 
     #############################################################
     # HTML Styles
@@ -94,7 +111,7 @@ def run():
 
     #############################################################
     # Main page information text
-    st.title('PyThea: Reconstruct CMEs & Shocks')
+    st.header('PyThea: Reconstruct CMEs & Shocks')
 
     #############################################################
     # Startup Variables
@@ -106,7 +123,6 @@ def run():
 
     if 'date_process' not in st.session_state:
         date_and_event_selection(st)
-        st.markdown(""" #### ⏻ Select a day & solar event. """)
     else:
         st.sidebar.markdown('## Processing Event|Date:')
         st.sidebar.info(f'{st.session_state.event_selected}')
@@ -182,7 +198,7 @@ def run():
                                                           key='imaging_time_range')
         select_timerange_form.form_submit_button(label='Submit',
                                                  on_click=delete_from_state,
-                                                 kwargs={'vars': ['map', 'map_', 'imagers_list_']})
+                                                 kwargs={'vars': ['map', 'map_', 'imagers_list_', 'hek_responses']})
 
     with st.sidebar.expander('Processing Options'):
         procoption_container = st.container()
@@ -194,7 +210,8 @@ def run():
                                                     on_change=delete_from_state,
                                                     kwargs={'vars': ['map', 'imagers_list_', 'clim', 'clim_manual_low', 'clim_manual_high']})
         if image_mode in ['Running Diff.', 'Base Diff.']:
-            diff_value = procoption_container.number_input('Select Step/Image for Running/Base Diff.', 1, 5, key='diff_value',
+            min_diff = 1 if image_mode == 'Running Diff.' else 0
+            diff_value = procoption_container.number_input('Select Step/Image for Running/Base Diff.', min_diff, 5, key='diff_value',
                                                            on_change=delete_from_state,
                                                            kwargs={'vars': ['map', 'imagers_list_', 'clim', 'clim_manual_low', 'clim_manual_high']})
         else:
@@ -206,32 +223,59 @@ def run():
         plotviewopt_container = st.container()
         plotviewopt_container.checkbox('Clip plot on image limits', value=True, key='clip_model')
         plt_supp_imagers = plotviewopt_container.checkbox('Supplementary Imaging', value=False)
-        plotviewopt_container.checkbox('View Bodies or s/c', key='star_field')
-        if st.session_state.star_field:
-            plotviewopt_container.multiselect('Select Bodies', options=selected_bodies.bodies_dict.keys(),
-                                              default=['Mercury', 'Venus', 'Jupiter'],
-                                              key='bodies_list')
-        plotviewopt_container.checkbox('View Limb and Meridians', key='plot_solar_reference_lines_')
-        if st.session_state.plot_solar_reference_lines_:
-            markdown = '''
-                **Limb**: plots the solar limb as observed  # noqa
-                for the selected observers.  # noqa
-                **Central Meridian**: plots the central meridian  # noqa
-                observed for the selected observers.  # noqa
-                **CR Meridan+Equator**: plots the primary meridian  # noqa
-                and equator of Carrington coordinate system.  # noqa
-                '''.strip()
-            plotviewopt_container.selectbox('Select plot option',
-                                            options=['Limb from Obs.', 'Central Meridian from Obs.', 'Carr. Prime Meridian+Solar Equator',
-                                                     'Stonyhurst Grid', 'Carrington Grid'],
-                                            help=markdown,
-                                            key='plot_solar_reference_lines_mode')
-            disabled = False if st.session_state.plot_solar_reference_lines_mode in ['Limb from Obs.', 'Central Meridian from Obs.'] else True
-            plotviewopt_container.multiselect('Select Bodies',
-                                              label_visibility='collapsed',
-                                              options=selected_bodies.bodies_dict.keys(),
-                                              default=['Earth'], disabled=disabled,
-                                              key='plot_solar_reference_lines_bodies_list')
+
+    #############################################################
+    # Magnetic Connectivity
+    st.sidebar.markdown('## Overlays')
+    with st.sidebar.expander('Bodies or s/c Location'):
+        bodies_container = st.container()
+        bodies_container.checkbox('View Bodies or s/c', key='star_field')
+        bodies_container.multiselect('Select Bodies', options=selected_bodies.bodies_dict.keys(),
+                                     default=['Mercury', 'Venus', 'Jupiter'],
+                                     key='bodies_list',
+                                     disabled=not st.session_state.star_field)
+    with st.sidebar.expander('Limb and Meridians'):
+        limb_meridians_container = st.container()
+        limb_meridians_container.checkbox('View Limb and Meridians', key='plot_solar_reference_lines_')
+        markdown = '''
+            **Limb**: plots the solar limb as observed
+            for the selected observers.
+            **Central Meridian**: plots the central meridian
+            observed for the selected observers.
+            **CR Meridan+Equator**: plots the primary meridian
+            and equator of Carrington coordinate system.
+            '''.strip()
+        limb_meridians_container.selectbox('Select plot option',
+                                           options=['Limb from Obs.', 'Central Meridian from Obs.', 'Carr. Prime Meridian+Solar Equator',
+                                                    'Stonyhurst Grid', 'Carrington Grid'],
+                                           help=markdown,
+                                           key='plot_solar_reference_lines_mode',
+                                           disabled=not st.session_state.plot_solar_reference_lines_)
+        disabled = False if (st.session_state.plot_solar_reference_lines_) and \
+            (st.session_state.plot_solar_reference_lines_mode in ['Limb from Obs.', 'Central Meridian from Obs.']) else True
+        limb_meridians_container.multiselect('Select Bodies',
+                                             label_visibility='collapsed',
+                                             options=selected_bodies.bodies_dict.keys(),
+                                             default=['Earth'], disabled=disabled,
+                                             key='plot_solar_reference_lines_bodies_list')
+    with st.sidebar.expander('Magnetic Connectivity'):
+        connectivity_container = st.container()
+        connectivity_container.checkbox('Plot Parker spirals', key='plot_parker_spirals')
+        connectivity_container.multiselect('Select bodies/spacecraft', options=selected_bodies.bodies_dict.keys(),
+                                           default=['Earth'], key='mag_bodies_list',
+                                           disabled=not st.session_state.plot_parker_spirals)
+        st.session_state.sw_speed_select = {}
+        for body in st.session_state.mag_bodies_list:
+            connectivity_container.number_input(f'{body} solar wind speed', min_value=200, max_value=800, value=350, step=50,
+                                                key=f'sw_speed_select_{body}',
+                                                disabled=not st.session_state.plot_parker_spirals)
+            st.session_state.sw_speed_select[body] = st.session_state[f'sw_speed_select_{body}'] * (u.km/u.second)
+    with st.sidebar.expander('HEK feature/events'):
+        hek_container = st.container()
+        hek_container.checkbox('Plot HEK feature/events', key='plot_hek', on_change=delete_from_state,
+                               kwargs={'vars': ['hek_responses', ]})
+        hek_container.multiselect('Select HEK feature/events', options=['Active Regions', 'Coronal Holes', 'Flares'],
+                                  default=['Flares'], key='hek_list', disabled=not st.session_state.plot_hek)
 
     #############################################################
     # Download and Process the Images
@@ -245,15 +289,16 @@ def run():
         st.session_state.map_ = {} if 'map_' not in st.session_state else st.session_state.map_
         st.session_state.map = {} if 'map' not in st.session_state else st.session_state.map
         st.session_state['imagers_list_'] = imagers_list
-        progress_bar = stqdm.stqdm(imager_added, desc='Preparing to Download data')
-        for imager in progress_bar:
-            progress_bar.desc = f'Downloaded {imager} images from VSO'
+        progress_bar = st.progress(0, text='Preparing to Download data')
+        for i, imager in enumerate(imager_added):
+            progress_bar.progress(i/len(imager_added), text=f'Download {imager} images from VSO')
             if imager not in st.session_state.map_:
                 timerange = a.Time(st.session_state.date_process + datetime.timedelta(hours=imaging_time_range[0]),
                                    st.session_state.date_process + datetime.timedelta(hours=imaging_time_range[1]))
-                st.session_state.map_[imager] = download_fits(timerange, imager)
+                downloaded_files = download_fits(timerange, imager)
+                st.session_state.map_[imager] = load_fits(downloaded_files)
                 st.session_state.map_[imager] = single_imager_maps_process(st.session_state.map_[imager],
-                                                                           **selected_imagers.imager_dict[imager][1],
+                                                                           **selected_imagers.imager_dict[imager]['process'],
                                                                            skip='sequence_processing')
             processed_images = single_imager_maps_process(st.session_state.map_[imager],
                                                           skip=['filter', 'prepare'],
@@ -265,6 +310,7 @@ def run():
             else:
                 st.session_state.imagers_list_.remove(imager)
                 st.session_state.map_colormap_limits.pop(imager, None)
+        progress_bar.empty()
 
     if imager_removed != []:
         st.session_state['imagers_list_'] = imagers_list
@@ -281,13 +327,14 @@ def run():
     col1, col2 = st.columns((1, 3))
     imager_select = col1.selectbox('Select an imager',
                                    options=st.session_state.imagers_list_,
-                                   on_change=delete_from_state, kwargs={'vars': ['clim', 'clim_manual_low', 'clim_manual_high']})
+                                   on_change=delete_from_state, kwargs={'vars': ['clim', 'clim_manual_low', 'clim_manual_high', 'hek_responses']})
 
     maps_date = [getattr(maps, 'date_average', None) or getattr(maps, 'date', None) for maps in st.session_state.map[imager_select]]
     if len(maps_date) > 1:
         running_map_date = col2.select_slider('Slide to the image time',
                                               options=maps_date, value=maps_date[0],
-                                              key='running_map_date')
+                                              key='running_map_date',
+                                              on_change=delete_from_state, kwargs={'vars': ['hek_responses']})
     else:
         running_map_date = maps_date[0]
 
@@ -350,6 +397,15 @@ def run():
     fig, axis = figure_streamlit(st, running_map, image_mode, imager_select, model)
     st.pyplot(fig)
 
+    if st.session_state.plot_hek and st.session_state.hek_responses['Flares']:
+        st.markdown('**HEK Flare List:**')
+        st.write(st.session_state.hek_responses['Flares'].to_pandas())
+        st.markdown('*Flares with no location have been removed from the table.')
+    if st.session_state.plot_hek and st.session_state.hek_responses['Active Regions']:
+        st.markdown('**HEK Active Regions List:**')
+        st.write(st.session_state.hek_responses['Active Regions'].to_pandas())
+        st.markdown('*Active Regions without NOAA number have been removed from the table.')
+
     if plt_supp_imagers:
         if len(st.session_state.imagers_list_) < 3:
             other_element = [element for element in st.session_state.imagers_list_ if element != imager_select][0]
@@ -374,6 +430,7 @@ def run():
     # Store the fitting
     single_fit = model.to_dataframe()
     single_fit['imager'] = imager_select
+    single_fit['fits_file'] = running_map.meta['fits_file']
 
     if store_fit_button_pressed:
         if 'model_fittings' not in st.session_state:
@@ -390,10 +447,13 @@ def run():
         st.markdown('---')
         st.markdown('### Parameters Table')
         st.markdown('**Running Fitting Table:**')
-        st.dataframe(single_fit)
+        col_formats = {col: '{:,.2f}'.format for col in single_fit.select_dtypes(include='number').columns}
+        st.dataframe(single_fit.style.format(col_formats))
         if 'model_fittings' in st.session_state:
             st.markdown('**Stored Fitting Table:**')
-            st.dataframe(st.session_state.model_fittings.parameters)
+            st.dataframe(st.session_state.model_fittings.parameters.style.apply(highlight_row,
+                                                                                row_index=single_fit.index, axis=1).format(col_formats)
+                         )
             col1, col2 = st.columns((2, 2))
             col2.selectbox('Select a fitting',
                            options=st.session_state.model_fittings.parameters.index,
@@ -413,7 +473,7 @@ def run():
         st.markdown('### Plots of kinematics ')
         col1, col2, col3 = st.columns(3)
         plt_kinematics_select = col1.selectbox('Select Plots',
-                                               options=['All', 'HeightT', 'SpeedT', 'Long-LatT'])
+                                               options=['All', 'HeightT', 'SpeedT', 'AccelerationT', 'Long-LatT'])
 
         fit_mode = col2.selectbox('Select Fitting Mode',
                                   options=['polynomial', 'spline', 'custom'],

{pythea-0.11.0 → pythea-0.13.0}/PyThea/_version.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.11.0'
-__version_tuple__ = version_tuple = (0, 11, 0)
+__version__ = version = '0.13.0'
+__version_tuple__ = version_tuple = (0, 13, 0)

{pythea-0.11.0 → pythea-0.13.0}/PyThea/callbacks.py

@@ -3,10 +3,24 @@ from sunpy.coordinates import frames
 
 
 def load_or_delete_fittings(st):
+    """
+    Load or delete fitting parameters from the model fittings based on user selection.
+
+    Parameters
+    ----------
+    st : Streamlit session state object
+        Streamlit session state object for managing UI components.
+
+    Returns
+    -------
+    None
+    """
     selected_row = str(st.session_state.fitting_select)
     dataframe = st.session_state.model_fittings.parameters
+
     if st.session_state.fit_action == 'Select':
         pass
+
     elif st.session_state.fit_action == 'Load':
         if st.session_state.coord_system == 'HGS':
             st.session_state.longit = float(dataframe.loc[selected_row, 'hgln'])
@@ -27,6 +41,7 @@ def load_or_delete_fittings(st):
                 # del st.session_state[key]
                 st.session_state[key] = float(dataframe.loc[selected_row, key])
         del st.session_state.fit_action
+
     elif st.session_state.fit_action == 'Delete':
         st.session_state.model_fittings.parameters = dataframe.drop(st.session_state.fitting_select)
         del st.session_state.fitting_select
@@ -36,6 +51,18 @@ def load_or_delete_fittings(st):
 
 
 def change_long_lat_sliders(st):
+    """
+    Update longitude and latitude sliders based on the selected coordinate system.
+
+    Parameters
+    ----------
+    st : Streamlit session state object
+        Streamlit session state object for managing UI components.
+
+    Returns
+    -------
+    None
+    """
     if st.session_state.coord_system == 'HGS':
         center_ = st.session_state.center.transform_to(frames.HeliographicStonyhurst)
     elif st.session_state.coord_system == 'HGC':
@@ -46,9 +73,22 @@ def change_long_lat_sliders(st):
 
 
 def change_fitting_sliders(st):
+    """
+    Update fitting sliders based on the selected fitting method.
+
+    Parameters
+    ----------
+    st : Streamlit session state object
+        Streamlit session state object for managing UI components.
+
+    Returns
+    -------
+    None
+    """
     st.session_state.startup['fitting'] = False
     fit_opt = st.session_state.fit_args_prime
     st.session_state.fit_mode = fit_opt['type']
+
     if fit_opt['type'] == 'polynomial':
         st.session_state.polyfit_order = fit_opt['order']
     elif fit_opt['type'] == 'spline':

{pythea-0.11.0 → pythea-0.13.0}/PyThea/config/app_styles.py

@@ -1,9 +1,4 @@
 def apply(st):
-    # Hide the menu button
-    st.markdown(""" <style>
-                #MainMenu {visibility: hidden;}
-                footer {visibility: hidden;}
-                </style> """, unsafe_allow_html=True)
     # Do some css styling tricks here (e.g. remove the padding)
     # https://discuss.streamlit.io/t/reduce-white-space-top-of-the-page/21737/3
     st.markdown("""

pythea-0.13.0/PyThea/config/selected_imagers.py

@@ -0,0 +1,73 @@
+'''
+ A dictionary of the selected imagers
+ How to use this:
+
+'''
+import astropy.units as u
+from sunpy.net import attrs as a
+
+imager_dict = {}
+
+imager_dict['LC2'] = {'fido': (a.Instrument.lasco, a.Detector.c2),
+                      'process': {'dimensions': (1024*u.pixel, 1024*u.pixel), 'polar': 'Clear', 'superpixel': 2},
+                      'source': 'SOHO', 'instrument': 'LASCO', 'detector': 'C2'}
+
+imager_dict['LC3'] = {'fido': (a.Instrument.lasco, a.Detector.c3),
+                      'process': {'dimensions': (1024*u.pixel, 1024*u.pixel), 'polar': 'Clear', 'superpixel': 2},
+                      'source': 'SOHO', 'instrument': 'LASCO', 'detector': 'C3'}
+
+imager_dict['AIA-193'] = {'fido': (a.Instrument.aia, a.Wavelength(19.3 * u.nm), a.Sample(1*u.minute)),
+                          'process': {'dimensions': (4096*u.pixel, 4096*u.pixel), 'superpixel': 8, 'exposure': 1.90},
+                          'source': 'SDO', 'instrument': 'AIA', 'wavelength': '193'}
+
+imager_dict['AIA-211'] = {'fido': (a.Instrument.aia, a.Wavelength(21.1 * u.nm), a.Sample(1*u.minute)),
+                          'process': {'dimensions': (4096*u.pixel, 4096*u.pixel), 'superpixel': 8, 'exposure': 1.90},
+                          'source': 'SDO', 'instrument': 'AIA', 'wavelength': '211'}
+
+imager_dict['COR2A'] = {'fido': (a.Source('STEREO_A'), a.Detector.cor2),
+                        'process': {'dimensions': (2048*u.pixel, 2048*u.pixel), 'polar': 1001, 'superpixel': 4},
+                        'source': 'STEREO_A', 'instrument': 'SECCHI', 'detector': 'COR2'}
+
+imager_dict['COR2B'] = {'fido': (a.Source('STEREO_B'), a.Detector.cor2),
+                        'process': {'dimensions': (2048*u.pixel, 2048*u.pixel), 'polar': 1001, 'superpixel': 4},
+                        'source': 'STEREO_B', 'instrument': 'SECCHI', 'detector': 'COR2'}
+
+imager_dict['EUVIA'] = {'fido': (a.Source('STEREO_A'), a.Detector.euvi, a.Wavelength(19.5 * u.nm)),
+                        'process': {'dimensions': (2048*u.pixel, 2048*u.pixel), 'superpixel': 4},
+                        'source': 'STEREO_A', 'instrument': 'SECCHI', 'detector': 'EUVI'}
+
+imager_dict['EUVIB'] = {'fido': (a.Source('STEREO_B'), a.Detector.euvi, a.Wavelength(19.5 * u.nm)),
+                        'process': {'dimensions': (2048*u.pixel, 2048*u.pixel), 'superpixel': 4},
+                        'source': 'STEREO_B', 'instrument': 'SECCHI', 'detector': 'EUVI'}
+
+imager_dict['COR1A'] = {'fido': (a.Source('STEREO_A'), a.Detector.cor1),
+                        'process': {'dimensions': (512*u.pixel, 512*u.pixel)},
+                        'source': 'STEREO_A', 'instrument': 'SECCHI', 'detector': 'COR1'}
+
+imager_dict['COR1B'] = {'fido': (a.Source('STEREO_B'), a.Detector.cor1),
+                        'process': {'dimensions': (512*u.pixel, 512*u.pixel)},
+                        'source': 'STEREO_B', 'instrument': 'SECCHI', 'detector': 'COR1'}
+
+imager_dict['HI1A'] = {'fido': (a.Source('STEREO_A'), a.Detector.hi1),
+                       'process': {'dimensions': (1024*u.pixel, 1024*u.pixel), 'superpixel': 2},
+                       'source': 'STEREO_A', 'instrument': 'SECCHI', 'detector': 'HI1'}
+
+imager_dict['HI1B'] = {'fido': (a.Source('STEREO_B'), a.Detector.hi1),
+                       'process': {'dimensions': (1024*u.pixel, 1024*u.pixel), 'superpixel': 2},
+                       'source': 'STEREO_B', 'instrument': 'SECCHI', 'detector': 'HI1'}
+
+imager_dict['HI2A'] = {'fido': (a.Source('STEREO_A'), a.Detector.hi2),
+                       'process': {'dimensions': (1024*u.pixel, 1024*u.pixel), 'superpixel': 2},
+                       'source': 'STEREO_A', 'instrument': 'SECCHI', 'detector': 'HI2'}
+
+imager_dict['HI2B'] = {'fido': (a.Source('STEREO_B'), a.Detector.hi2),
+                       'process': {'dimensions': (1024*u.pixel, 1024*u.pixel), 'superpixel': 2},
+                       'source': 'STEREO_B', 'instrument': 'SECCHI', 'detector': 'HI2'}
+
+imager_dict['WISPR1'] = {'fido': (a.Instrument.wispr, a.Detector.inner),
+                         'process': {'dimensions': (960*u.pixel, 1024*u.pixel), 'processing_level': 3, 'superpixel': 2},
+                         'source': 'PSP', 'instrument': 'WISPR', 'detector': 'Inner'}
+
+imager_dict['WISPR2'] = {'fido': (a.Instrument.wispr, a.Detector.outer),
+                         'process': {'dimensions': (960*u.pixel, 1024*u.pixel), 'processing_level': 3, 'superpixel': 2},
+                         'source': 'PSP', 'instrument': 'WISPR', 'detector': 'Outer'}

{pythea-0.11.0 → pythea-0.13.0}/PyThea/data/sample_data.py

@@ -26,10 +26,19 @@ stereo_sample_data = pooch.create(
     },
 )
 
+wispr_sample_data = pooch.create(
+    path=os.path.join(database_dir, 'sample_data'),  # The cache folder
+    base_url=f'{github_main_url}/raw/main/data/',  # The remote data url on Github
+    registry={
+        'psp_l3_wispr_20210808t103707_v1_1211.fits': 'sha256:85035693c4677e8fa8a3e61b8051d6f0a46e96806b5b0cbdb3efa77652308bda',
+        'psp_l3_wispr_20210808t104010_v1_2222.fits': 'sha256:5e55e17f1aed7dc96f88b79f092947c58e6ee0def1b72822dbf1a696d8df7e6f'
+    },
+)
+
 json_fitting_file_sample_data = pooch.create(
     path=os.path.join(database_dir, 'sample_data'),  # The cache folder
     base_url=f'{github_main_url}/raw/main/data/',  # The remote data url on Github
     registry={
-        'FLX1p0D20211028T153500MEllipsoid.json': 'sha256:700687ed982317400ec409390571747694237de9688a2816baa7b1770b75c9b1',
+        'FLX1p0D20211028T153500MEllipsoid.json': 'sha256:34fced8530875110117ba27a73541d3acd0c90bc8fca99367c1ec4cdfc05ce86',
     },
 )