pyTEMlib 0.2024.9.0__py3-none-any.whl → 0.2025.2.2__py3-none-any.whl

pyTEMlib/info_widget.py

@@ -2,6 +2,7 @@ from typing import Any
 
 import numpy as np
 import os
+import sys
 import ipywidgets
 import matplotlib.pylab as plt
 import matplotlib
@@ -40,15 +41,15 @@ def get_image_sidebar() -> Any:
                                           layout=ipywidgets.Layout(width='auto', grid_area='header'),
                                           style=ipywidgets.ButtonStyle(button_color='lightblue'))
     row += 1
-    side_bar[row, :2] = ipywidgets.FloatText(value=7.5, description='Conv.Angle:', disabled=False, color='black',
+    side_bar[row, :2] = ipywidgets.FloatText(value=-1, description='Conv.Angle:', disabled=False, color='black',
                                              layout=ipywidgets.Layout(width='200px'))
     side_bar[row, 2] = ipywidgets.widgets.Label(value="mrad", layout=ipywidgets.Layout(width='100px'))
     row += 1
-    side_bar[row, :2] = ipywidgets.FloatText(value=0.1, description='Coll.Angle:', disabled=False, color='black',
+    side_bar[row, :2] = ipywidgets.FloatText(value=-0.1, description='Coll.Angle:', disabled=False, color='black',
                                              layout=ipywidgets.Layout(width='200px'))
     side_bar[row, 2] = ipywidgets.widgets.Label(value="mrad", layout=ipywidgets.Layout(width='100px'))
     row += 1
-    side_bar[row, :2] = ipywidgets.FloatText(value=0.1, description='Acc Voltage:', disabled=False, color='black',
+    side_bar[row, :2] = ipywidgets.FloatText(value=.1, description='Acc Voltage:', disabled=False, color='black',
                                              layout=ipywidgets.Layout(width='200px'))
     side_bar[row, 2] = ipywidgets.widgets.Label(value="keV", layout=ipywidgets.Layout(width='100px'))
     
@@ -123,11 +124,11 @@ def get_info_sidebar() -> Any:
                                           layout=ipywidgets.Layout(width='auto', grid_area='header'),
                                           style=ipywidgets.ButtonStyle(button_color='lightblue'))
     row += 1
-    side_bar[row, :2] = ipywidgets.FloatText(value=7.5, description='Conv.Angle:', disabled=False, color='black',
+    side_bar[row, :2] = ipywidgets.FloatText(value=-1, description='Conv.Angle:', disabled=False, color='black',
                                              layout=ipywidgets.Layout(width='200px'))
     side_bar[row, 2] = ipywidgets.widgets.Label(value="mrad", layout=ipywidgets.Layout(width='100px'))
     row += 1
-    side_bar[row, :2] = ipywidgets.FloatText(value=0.1, description='Coll.Angle:', disabled=False, color='black',
+    side_bar[row, :2] = ipywidgets.FloatText(value=-0.1, description='Coll.Angle:', disabled=False, color='black',
                                              layout=ipywidgets.Layout(width='200px'))
     side_bar[row, 2] = ipywidgets.widgets.Label(value="mrad", layout=ipywidgets.Layout(width='100px'))
     row += 1
@@ -174,9 +175,11 @@ def get_info_sidebar() -> Any:
     side_bar[row, 1] = ipywidgets.Button(description='Shift Spec', disabled=True,
                                          layout=ipywidgets.Layout(width='auto'),
                                          style=ipywidgets.ButtonStyle(button_color='lightblue'))
+    """ 
     side_bar[row, 2] = ipywidgets.Button(description='Res.Fct.', disabled=True,
                                          layout=ipywidgets.Layout(width='auto'),
                                          style=ipywidgets.ButtonStyle(button_color='lightblue'))
+    """
     
     row += 1
     side_bar[row, :3] = ipywidgets.Button(description='Spectrum Image',
@@ -187,7 +190,7 @@ def get_info_sidebar() -> Any:
     side_bar[row, :2] = ipywidgets.IntText(value=1, description='bin X:', disabled=False, color='black',
                                            layout=ipywidgets.Layout(width='200px'))
     row += 1
-    side_bar[row, :2] = ipywidgets.IntText(value=1, description='bin X:', disabled=False, color='black',
+    side_bar[row, :2] = ipywidgets.IntText(value=1, description='bin Y:', disabled=False, color='black',
                                            layout=ipywidgets.Layout(width='200px'))
     
     for i in range(15, 18):
@@ -230,13 +233,18 @@ def get_file_widget_ui():
     side_bar[row, 0] = ipywidgets.Button(description='Save',
                                          layout=ipywidgets.Layout(width='100px'),
                                          style=ipywidgets.ButtonStyle(button_color='lightblue'))
-
+    side_bar[row, 1] = ipywidgets.Text(
+                                        value='Test.hf5',
+                                        placeholder='Type something',
+                                        description='File:',
+                                        disabled=False   
+                                    )
     return side_bar  
 
 
 class EELSBaseWidget(object):
 
-    def __init__(self, datasets, sidebar, tab_title=None):
+    def __init__(self, datasets=None, sidebar=None, tab_title=None):
         
         self.datasets = datasets
         self.dataset = None
@@ -248,6 +256,15 @@ class EELSBaseWidget(object):
         self.image_list = ['Sum']
         self.dir_name = file_tools.get_last_path()
 
+        self.key = None
+        self.new_info = False
+        self.image = 'Sum'
+        if 'google.colab' in sys.modules:
+            self.google = True
+        else:
+            self.google = False
+        # self.google = True
+
         self.save_path = True
 
         if not os.path.isdir(self.dir_name):
@@ -269,25 +286,42 @@ class EELSBaseWidget(object):
         self.end_channel = -2
 
         self.file_bar = get_file_widget_ui()
+        children = [self.file_bar]
+        titles = ['File']
         if isinstance(sidebar, dict):
-            tab = ipywidgets.Tab()
-            children = [self.file_bar]
-            titles = ['File']
             for sidebar_key, sidebar_gui in sidebar.items():
                 children.append(sidebar_gui)
                 titles.append(sidebar_key)
-            tab.children = children
-            tab.titles = titles
         elif not isinstance(sidebar, list):
-            tab = ipywidgets.Tab()
-            tab.children = [self.file_bar, sidebar]
-            tab.titles = ['File', 'Info']
+            children = [self.file_bar, sidebar]
+            titles = ['File', 'Info']
+        if sidebar is None:
+            children = [self.file_bar]
+            titles = ['File']
+
+        if self.google:
+            self.buttons = []
+            for i in range(len(children)):
+                self.buttons.append(ipywidgets.Button(description=titles[i],
+                                    disabled=False,
+                                    button_style='',  # 'success', 'info', 'warning', 'danger' or ''
+                                    layout=ipywidgets.Layout(width='800px')))
+
+
+            self.tab_buttons = ipywidgets.ToggleButtons(options=titles, description='', disabled=False,
+                                                        layout=ipywidgets.Layout(width='auto'),
+                                                        style={"button_width": "auto"})
+            tab = ipywidgets.VBox([self.tab_buttons, self.file_bar])
+            self.children = children
+
         else:
-            tab = sidebar
-        self.tab = tab
+            tab = ipywidgets.Tab()
+            tab.children = children
+            tab.titles = titles
+
         with plt.ioff():
             self.figure = plt.figure()
-        
+        self.tab =tab
         self.figure.canvas.toolbar_position = 'right'
         self.figure.canvas.toolbar_visible = True
 
@@ -295,11 +329,17 @@ class EELSBaseWidget(object):
                                                  layout=ipywidgets.Layout(width='200px'))
         self.end_cursor = ipywidgets.FloatText(value=0, description='End:', disabled=False, color='black',
                                                layout=ipywidgets.Layout(width='200px'))
+        self.statusbar = ipywidgets.Text(value='Starting',
+                                    placeholder='Type something',
+                                    description='Status:',
+                                    disabled=True,
+                                    layout=ipywidgets.Layout(width='100%'))
         self.panel = ipywidgets.VBox([ipywidgets.HBox([ipywidgets.Label('', layout=ipywidgets.Layout(width='100px')),
                                                        ipywidgets.Label('Cursor:'),
                                                        self.start_cursor, ipywidgets.Label('eV'),
                                                        self.end_cursor, ipywidgets.Label('eV')]),
-                                      self.figure.canvas])
+                                     self.figure.canvas,
+                                     self.statusbar])
         
         self.app_layout = ipywidgets.AppLayout(
             left_sidebar=tab,
@@ -329,9 +369,30 @@ class EELSBaseWidget(object):
         select_button.on_click(self.select_main)
         add_button.on_click(self.add_dataset)
         self.loaded_datasets.observe(self.select_dataset, names='value')
-        self.file_bar[4, 0].observe(self.plot, names='value')
+        self.file_bar[4, 0].observe(self.set_image, names='value')
+        self.file_bar[5, 0].observe(self.set_survey_image, names='value')
 
-    def set_image(self, key=None):
+        self.file_bar[6, 0].observe(self.save_datasets)
+        
+    def save_datasets(self, value=0):
+        import warnings
+        file_name = self.file_bar[6, 1].value
+        path = self.path_choice.options[0]
+        with warnings.catch_warnings():
+            warnings.simplefilter("ignore")
+        h5_group = file_tools.save_dataset(self.datasets, os.path.join(path, file_name))
+        h5_group.file.close()
+        self.status_message(' File saved')
+
+    def status_message(self, out: str):
+        self.statusbar.value = out 
+
+    def set_survey_image(self, key=None):
+        
+        self.datasets['_relationship']['survey_image'] = self.file_bar[5, 0].value
+        # ToDo: Find boundaries of scan. 
+        
+    def get_image(self):
         if self.file_bar[4, 0].value == 'Sum':
             spec_dim = self.dataset.get_dimensions_by_type(sidpy.DimensionType.SPECTRAL)
             if len(spec_dim) != 1:
@@ -346,9 +407,15 @@ class EELSBaseWidget(object):
                 self.image = self.dataset.mean(axis=(spec_dim[0], channel_dim[0]))
             else:
                 self.image = self.dataset.mean(axis=(spec_dim[0]))
+            image_key = 'Sum'
         else:
             image_key = self.file_bar[4, 0].value.split(':')[0]
             self.image = self.datasets[image_key]
+        self.datasets['_relationship']['image'] = image_key
+        
+    def set_image(self, key=None):
+        self.get_image()
+        self.plot()
 
     def plot(self, scale=True):
         self.figure.clear()
@@ -358,6 +425,7 @@ class EELSBaseWidget(object):
             if self.dataset.data_type.name == 'SPECTRUM':
                 self.axis = self.figure.subplots(ncols=1)
             else:
+                self.get_image()
                 self.plot_spectrum_image() 
                 self.axis = self.axes[-1]
             self.spectrum = self.get_spectrum()
@@ -373,13 +441,27 @@ class EELSBaseWidget(object):
         for dim, axis in self.dataset._axes.items():
             if axis.dimension_type in [sidpy.DimensionType.SPATIAL, sidpy.DimensionType.RECIPROCAL]:
                 image_dims.append(dim)
-        self.img = self.axis.imshow(self.dataset, extent=self.dataset.get_extent(image_dims))
+        self.img = self.axis.imshow(self.dataset.T, extent=self.dataset.get_extent(image_dims))
         self.axis.set_xlabel(self.dataset.labels[image_dims[0]])
         self.axis.set_ylabel(self.dataset.labels[image_dims[1]])
         
         cbar = self.figure.colorbar(self.img)
         cbar.set_label(self.dataset.data_descriptor)
-
+        if 'annotations' in self.dataset.metadata:
+            kwargs={'edgecolor': 'red', 'facecolor': 'None'}
+                    
+            annotations = self.dataset.metadata['annotations']
+            for key in annotations:
+                if annotations[key]['type'] == 'spectral_image':
+                    rectangle = matplotlib.patches.Rectangle(annotations[key]['position'], annotations[key]['width'], annotations[key]['height'], **kwargs)
+                    self.axis.text(annotations[key]['position'][0], annotations[key]['position'][1], annotations[key]['label'], color='r')
+                    self.axis.add_artist(rectangle)
+                elif annotations[key]['type'] == 'text':
+                    self.axis.text(annotations[key]['position'][0], annotations[key]['position'][1], annotations[key]['label'], color='r')
+                elif annotations[key]['type'] == 'circle':
+                    circle = matplotlib.patches.Circle(annotations[key]['position'], annotations[key]['radius'], **kwargs)
+                    self.axis.add_artist(circle)
+        self.figure.tight_layout()
         self.axis.ticklabel_format(style='sci', scilimits=(-2, 3))
         self.figure.tight_layout()
         self.figure.canvas.draw_idle()
@@ -424,7 +506,8 @@ class EELSBaseWidget(object):
         self.get_spectrum()
         if len(self.energy_scale) != self.spectrum.shape[0]:
             self.spectrum = self.spectrum.T
-        self.axis.plot(self.energy_scale, self.spectrum.compute(), label='experiment')
+        # self.axis.plot(self.energy_scale, self.spectrum.compute(), label='experiment')
+        self.axis.plot(self.energy_scale, self.spectrum, label='experiment')
 
         self.axis.set_title(f'spectrum {self.x}, {self.y}')
         self.figure.tight_layout()
@@ -438,8 +521,12 @@ class EELSBaseWidget(object):
         self.axis.set_xlabel(self.xlabel)
         self.axis.set_ylabel(self.ylabel)
         self.change_y_scale = 1.0
+        self.update_tab_spectra()
         self.figure.canvas.draw_idle()
 
+    def update_tab_spectra(self):
+        pass
+
     def _onclick(self, event):
         self.event = event
         if event.inaxes in [self.axes[0]]:
@@ -513,7 +600,6 @@ class EELSBaseWidget(object):
         self.axes = self.figure.subplots(ncols=2)
         self.axis = self.axes[-1]
 
-        self.set_image()
         self.rect = matplotlib.patches.Rectangle((0, 0), self.bin_x, self.bin_y, linewidth=1, edgecolor='r',
                                                  facecolor='red', alpha=0.2)
         image_dims = self.dataset.get_image_dims()
@@ -587,6 +673,7 @@ class EELSBaseWidget(object):
         # self.loaded_datasets.options = self.dataset_list
         
         self.datasets = file_tools.open_file(self.file_name)
+        
         file_tools.save_path(self.file_name)
         self.dataset_list = []
         self.image_list = ['Sum']
@@ -606,6 +693,7 @@ class EELSBaseWidget(object):
         
         self.key = self.dataset_list[0].split(':')[0]
         self.dataset = self.datasets[self.key]
+        self.new_info = True
 
         self.selected_dataset = self.dataset
         if len(self.image_list) > 0:
@@ -619,12 +707,27 @@ class EELSBaseWidget(object):
         self.file_bar[3, 0].options = self.dataset_list
         self.loaded_datasets.options = self.dataset_list
         self.loaded_datasets.value = self.dataset_list[0]
+        path, filename = os.path.split(self.file_name)
+        name, extension = os.path.splitext(filename)
+        self.file_bar[6, 1].value = name+'.hf5'
+        self.status_message(' New file loaded')
 
     def add_dataset(self, value=0):
         key = file_tools.add_dataset_from_file(self.datasets, self.file_name, 'Channel')
         self.dataset_list.append(f'{key}: {self.datasets[key].title}')
         self.loaded_datasets.options = self.dataset_list
         self.loaded_datasets.value = self.dataset_list[-1]
+        
+        if 'SPECTR' in self.datasets[key].data_type.name:
+            self.spectral_list.append(f'{key}: {self.datasets[key].title}')
+        if 'IMAGE' == self.datasets[key].data_type.name:
+            if 'survey' in self.datasets[key].title.lower():
+                self.survey_list.append(f'{key}: {self.datasets[key].title}')
+                self.file_bar[5, 0].options = self.survey_list
+            else:
+                self.image_list.append(f'{key}: {self.datasets[key].title}')
+                self.file_bar[4, 0].options = self.image_list
+                self.status_message(' image list file loaded')
 
     def get_directory(self, directory='.'):
         self.dir_name = directory
@@ -691,7 +794,11 @@ class EELSBaseWidget(object):
 
 class EELSWidget(EELSBaseWidget):
     def __init__(self, datasets=None):
-        
+        self.lowloss_key = 'None'
+        self.coreloss_key = 'None'
+        self.info_key = 'None'
+        self.tabval = 0
+
         sidebar = {'Spec.': get_info_sidebar(),
                    'LowLoss': get_low_loss_sidebar(),
                    'CoreLoss': get_core_loss_sidebar()}
@@ -707,18 +814,30 @@ class EELSWidget(EELSBaseWidget):
         self.set_action()
     
     def set_action(self):
+        if self.google:
+            self.tab_buttons.observe(self.tab_activated)
         self.tab.observe(self.tab_activated)
 
     def tab_activated(self, val=0):
-        if isinstance(val.new, int):
-            self.tabval = val.new
-            self.update_sidebars()
-            if val.new == 1:
-                self.info.update_dataset()
-            elif val.new == 2:
-                self.low_loss.update_ll_dataset()
-            elif val.new == 3:
-                self.core_loss.update_cl_sidebar()
+        if self.google:
+            self.tab.children = [self.tab_buttons, self.children[self.tab_buttons.index]]  # update sidebar gui
+            self.tabval = self.tab_buttons.index
+        else:
+            if isinstance(val.new, int):
+                self.tabval = val.new
+            # self.update_sidebars()
+        if self.tabval == 1:
+            self.info.update_dataset()
+        elif self.tabval == 2:
+            self.low_loss.update_ll_sidebar()
+        elif self.tabval == 3:
+            self.core_loss.update_cl_sidebar()
+
+    
+    def update_tab_spectra(self):
+        if self.tabval == 2:
+            self.low_loss._update()
+
 
     def update_sidebars(self):
         if hasattr(self, 'info'):
@@ -732,7 +851,7 @@ class Info(object):
     def __init__(self, sidebar=None, parent=None):
         self.parent = parent
         self.info_tab = sidebar
-        self.key = self.parent.key
+        self.key = self.parent.info_key
         self.update_sidebar()        
         self.set_action()
         self.count =0
@@ -749,10 +868,10 @@ class Info(object):
     def set_y_scale(self, value):
         self.count += 1
         self.parent.change_y_scale = 1.0 / self.parent.y_scale
-        if self.parent.datasets[self.key].metadata['experiment']['flux_ppm'] > 1e-12:
+        if self.parent.datasets[self.parent.key].metadata['experiment']['flux_ppm'] > 1e-12:
             if self.info_tab[9, 2].value:
-                dispersion = self.parent.datasets[self.key].get_dimension_slope(self.energy_scale)
-                self.parent.y_scale = 1 / self.parent.datasets[self.key].metadata['experiment']['flux_ppm'] * dispersion
+                dispersion = self.parent.datasets[self.parent.key].get_dimension_slope(self.parent.energy_scale)
+                self.parent.y_scale = 1 / self.parent.datasets[self.parent.key].metadata['experiment']['flux_ppm'] * dispersion
                 self.parent.ylabel = 'scattering probability (ppm)'
             else:
                 self.parent.y_scale = 1.0
@@ -763,10 +882,11 @@ class Info(object):
     def set_flux(self, value):
         # self.parent.datasets[self.key].metadata['experiment']['exposure_time'] = self.info_tab[10, 0].value
         if self.info_tab[9, 0].value == 'None':
-            self.parent.datasets[self.key].metadata['experiment']['flux_ppm'] = 0.
+            self.parent.datasets[self.parent.key].metadata['experiment']['flux_ppm'] = 0.
         else:
             ll_key = self.info_tab[9, 0].value.split(':')[0]
             self.parent.datasets['_relationship']['low_loss'] = ll_key
+            self.parent.lowloss_key = ll_key
             spectrum_dimensions = self.parent.dataset.get_spectral_dims()
 
             number_of_pixels = 1
@@ -780,18 +900,19 @@ class Info(object):
                     self.parent.datasets[ll_key].metadata['experiment']['exposure_time'] = (self.parent.datasets[ll_key].metadata['experiment']['single_exposure_time'] *
                                                                                      self.parent.datasets[ll_key].metadata['experiment']['number_of_frames'])
 
-            self.parent.datasets[self.key].metadata['experiment']['flux_ppm'] = ((np.array(self.parent.datasets[ll_key])*1e-6).sum() /
+            self.parent.datasets[self.parent.key].metadata['experiment']['flux_ppm'] = ((np.array(self.parent.datasets[ll_key])*1e-6).sum() /
                                                                           self.parent.datasets[ll_key].metadata['experiment']['exposure_time'] /
                                                                           number_of_pixels)
-            self.parent.datasets[self.key].metadata['experiment']['flux_ppm'] *= self.parent.datasets[self.key].metadata['experiment']['exposure_time']
+            self.parent.datasets[self.parent.key].metadata['experiment']['flux_ppm'] *= self.parent.datasets[self.parent.key].metadata['experiment']['exposure_time']
             if 'SPECT' in self.parent.datasets[ll_key].data_type.name:
                 self.info_tab[14, 0].disabled = False
-        self.info_tab[11, 0].value = np.round(self.parent.datasets[self.key].metadata['experiment']['flux_ppm'], 2)
+        self.info_tab[11, 0].value = np.round(self.parent.datasets[self.parent.key].metadata['experiment']['flux_ppm'], 2)
 
     def set_microscope_parameter(self, value):
-        self.parent.datasets[self.key].metadata['experiment']['convergence_angle'] = self.info_tab[5, 0].value
-        self.parent.datasets[self.key].metadata['experiment']['collection_angle'] = self.info_tab[6, 0].value
-        self.parent.datasets[self.key].metadata['experiment']['acceleration_voltage'] = self.info_tab[7, 0].value*1000
+        if not self.parent.new_info:
+            self.parent.datasets[self.key].metadata['experiment']['convergence_angle'] = self.info_tab[5, 0].value
+            self.parent.datasets[self.key].metadata['experiment']['collection_angle'] = self.info_tab[6, 0].value
+            self.parent.datasets[self.key].metadata['experiment']['acceleration_voltage'] = self.info_tab[7, 0].value*1000
     
     def cursor2energy_scale(self, value):
         self.energy_scale = self.parent.datasets[self.key].get_spectral_dims(return_axis=True)[0]
@@ -823,10 +944,12 @@ class Info(object):
             if self.bin_y > self.parent.dataset.shape[image_dims[1]]:
                 self.bin_y = self.parent.dataset.shape[image_dims[1]]
                 self.info_tab[17, 0].value = self.bin_y
+            self.parent.bin_x = self.bin_x
+            self.parent.bin_y = self.bin_y
 
             self.parent.datasets[self.key].metadata['experiment']['SI_bin_x'] = self.bin_x
             self.parent.datasets[self.key].metadata['experiment']['SI_bin_y'] = self.bin_y
-            self.plot()
+            self.parent.plot()
 
     
 
@@ -834,8 +957,11 @@ class Info(object):
         spectrum_list = ['None']
         reference_list = ['None']
         data_list = []
-        self.key = self.parent.key
+        
+        self.key = self.info_key = self.parent.info_key
+        
         spectrum_data = False
+        info_index= 0
         for key in self.parent.datasets.keys():
             if isinstance(self.parent.datasets[key], sidpy.Dataset):
                 if key[0] != '_' :
@@ -843,20 +969,29 @@ class Info(object):
                     if 'SPECTR' in self.parent.datasets[key].data_type.name:
                         spectrum_data = True
                         spectrum_list.append(f'{key}: {self.parent.datasets[key].title}')
+                        if self.info_key == key:
+                            info_index = len(spectrum_list)-1
                 reference_list.append(f'{key}: {self.parent.datasets[key].title}')
         self.info_tab[0, 0].options = spectrum_list
         self.info_tab[9, 0].options = reference_list
-
+        self.info_tab[0, 0].value = spectrum_list[info_index]
+            
         if 'SPECTRUM' in self.parent.dataset.data_type.name:
             for i in range(15, 18):
                 self.info_tab[i, 0].layout.display = "none"
         else:
             for i in range(15, 18):
                 self.info_tab[i, 0].layout.display = "flex"
-        
-        if spectrum_data:
-            offset = self.parent.datasets[self.key].energy_loss[0]
-            dispersion = self.parent.datasets[self.key].energy_loss[1] - offset
+
+        if 'None' not in self.key:
+            self.parent.new_info = True
+            energy_scale = self.parent.datasets[self.key].get_spectral_dims(return_axis=True)
+            if len(energy_scale) == 0:
+                return
+            energy_scale = energy_scale[0]
+            offset = energy_scale[0]
+            # dispersion = self.parent.datasets[self.key].get_dimension_slope(energy_scale)
+            dispersion = energy_scale[1] - offset
             
             # self.info_tab[0,0].value = dataset_index #f'{self.key}: {self.parent.datasets[self.key].title}'
             self.info_tab[2, 0].value = np.round(offset, 3)
@@ -864,6 +999,7 @@ class Info(object):
             self.info_tab[5, 0].value = np.round(self.parent.datasets[self.key].metadata['experiment']['convergence_angle'], 1)
             self.info_tab[6, 0].value = np.round(self.parent.datasets[self.key].metadata['experiment']['collection_angle'], 1)
             self.info_tab[7, 0].value = np.round(self.parent.datasets[self.key].metadata['experiment']['acceleration_voltage']/1000, 1)
+            # print(self.parent.datasets[self.key].metadata['experiment']['acceleration_voltage'])
             self.info_tab[10, 0].value = np.round(self.parent.datasets[self.key].metadata['experiment']['exposure_time'], 4)
             if 'flux_ppm' not in self.parent.datasets[self.key].metadata['experiment']:
                 self.parent.datasets[self.key].metadata['experiment']['flux_ppm'] = 0
@@ -879,16 +1015,32 @@ class Info(object):
                 if 'low_loss' in self.parent.datasets['_relationship']:
                     ll_key = self.parent.datasets['_relationship']['low_loss']  
                     ll_key = f'{ll_key}: {self.parent.datasets[ll_key].title}'
+                    self.lowloss_key = ll_key
             self.info_tab[9, 0].value = ll_key
+            self.parent.new_info = False
 
     def update_dataset(self, value=0):
-        key = self.info_tab[0, 0].value.split(':')[0]
-        self.parent.set_dataset(key)
-
+        self.key = self.info_tab[0, 0].value.split(':')[0]
+        
+        self.info_key = self.key
+        self.parent.info_key = self.key
+        
+        if self.info_key != 'None':
+            self.parent.set_dataset(self.info_key)
+        self.parent.status_message(self.key+'  ,  '+ self.parent.info_key)
+        if '_relationship' in self.parent.datasets.keys():
+            self.parent.datasets['_relationship']['spectrum'] = self.info_key
+        self.update_sidebar()
+        self.parent._update(0)
+        
     def shift_low_loss(self,  value=0):
         if 'low_loss' in self.parent.datasets['_relationship']:
             low_loss = self.parent.datasets[self.parent.datasets['_relationship']['low_loss']]
-            self.parent.datasets[self.parent.datasets['_relationship']['low_loss']] = eels_tools.align_zero_loss(low_loss)
+
+            self.parent.datasets['shifted_low_loss']  = eels_tools.align_zero_loss(low_loss)
+            self.parent.datasets['shifted_low_loss'].title = self.parent.dataset.title + '_shifted'
+            self.parent.datasets['_relationship']['low_loss'] = 'shifted_low_loss'
+            self.update_sidebar()
            
         if 'low_loss' in self.parent.datasets['_relationship']:
             if 'zero_loss' in self.parent.datasets[self.parent.datasets['_relationship']['low_loss']].metadata:
@@ -928,6 +1080,7 @@ class Info(object):
         self.info_tab[10, 0].observe(self.set_flux)
         self.info_tab[14, 0].on_click(self.shift_low_loss)
         self.info_tab[14, 1].on_click(self.shift_spectrum)
+        self.info_tab[14, 1].on_click(self.shift_spectrum)
         
         self.info_tab[16, 0].observe(self.set_binning)
         self.info_tab[17, 0].observe(self.set_binning)