Skip to content

sum_rules_tab

SumRules

Source code in mmg_toolbox/tkguis/xmcd_visualiser/sum_rules_tab.py 
 12
 13
 14
 15
 16
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
class SumRules:
    def __init__(self, root: tk.Misc, base: XMCDVisualiser, config: dict | None = None):
        self._base = base
        self.config = config
        self.root = root
        self.spectra: SpectraContainerSubtraction | None = None
        self.mode: str = ''
        self.output_name = tk.StringVar(self.root, '')

        # Average Tab
        self.root.columnconfigure(0, weight=0)
        self.root.columnconfigure(1, weight=1)
        self.root.columnconfigure(2, weight=0)
        self.root.rowconfigure(0, weight=1)
        grid_options = dict(padx=5, pady=5, sticky='nsew')

        # Column 1 - XAS plots
        col = ttk.Frame(self.root)
        col.grid(column=0, row=0, **grid_options)

        # Pol 1
        frm = ttk.Frame(col)
        frm.grid(column=0, row=0, **grid_options)
        self.pol1_figure = SimplePlot(
            root=frm,
            xdata=[],
            ydata=[],
            xlabel='Energy [eV]',
            ylabel='',
            title='Pol 1',
            config=self.config
        )
        self.pol1_figure.ax1.set_xlabel('Energy [eV]')
        self.pol1_figure.fig.tight_layout()

        # Pol 2
        frm = ttk.Frame(col)
        frm.grid(column=0, row=1, **grid_options)
        self.pol2_figure = SimplePlot(
            root=frm,
            xdata=[],
            ydata=[],
            xlabel='Energy [eV]',
            ylabel='',
            title='Pol 2',
            config=self.config
        )
        self.pol2_figure.ax1.set_xlabel('Energy [eV]')
        self.pol2_figure.fig.tight_layout()

        # Column 2 - Average
        frm = ttk.Frame(self.root)
        frm.grid(column=1, row=0, **grid_options)
        self.figure = SpectraPlotSlider(frm, config=self.config)

        # Column 3 - Buttons
        frm = ttk.Frame(self.root)
        frm.grid(column=2, row=0, **grid_options)
        ttk.Button(frm, text='Add Spectra to Comparison', command=self.btn_add_spectra).pack(side='top', fill='x')
        line = ttk.Frame(frm)
        line.pack(side='top', fill='x')
        ttk.Label(line, text='Filename:').pack(side='left', padx=5)
        ttk.Entry(line, textvariable=self.output_name).pack(side='left', fill='x', expand=True)
        ttk.Button(frm, text='Save NeXus', command=self.btn_nexus).pack(side='top', fill='x')
        ttk.Button(frm, text='Save CSV', command=self.btn_csv).pack(side='top', fill='x')

    def update_plot(self, spectra: SpectraContainerSubtraction, mode: str = 'tey'):
        self.spectra = spectra
        self.mode = mode

        self.pol1_figure.remove_lines()
        self.pol1_figure.reset_plot()
        spectra.add_edge_lines(self.pol1_figure.ax1)
        self.pol2_figure.remove_lines()
        for spec in spectra.spectra1.parents:
            spectrum = spec.spectra[self.mode]
            self.pol1_figure.plot(spectrum.energy, spectrum.signal, label=spec.name)
        self.pol1_figure.ax1.legend(frameon=False)
        self.pol1_figure.ax1.set_title(spectra.spectra1.name)
        self.pol1_figure.update_axes()

        self.pol2_figure.remove_lines()
        self.pol2_figure.reset_plot()
        spectra.add_edge_lines(self.pol2_figure.ax1)
        for spec in spectra.spectra2.parents:
            spectrum = spec.spectra[self.mode]
            self.pol2_figure.plot(spectrum.energy, spectrum.signal, label=spec.name)
        self.pol2_figure.ax1.legend(frameon=False)
        self.pol2_figure.ax1.set_title(spectra.spectra2.name)
        self.pol2_figure.update_axes()

        self.figure.update_spectra(spectra, self.mode)
        self.output_name.set(spectra.label())

    def btn_add_spectra(self):
        """Add spectra to different panel"""
        if self.spectra:
            self._base.comparison.treeview.add_spectra(self.spectra)

    def btn_nexus(self):
        output_name = self.output_name.get()
        filename = generate_output_filename(output_name, '.nxs', self.config)
        if self.spectra and output_name:
            self.spectra.write_nexus(filename)

    def btn_csv(self):
        output_name = self.output_name.get()
        filename = generate_output_filename(output_name, '.csv', self.config)
        if self.spectra and output_name and self.mode:
            self.spectra.write_csv(filename, self.mode)

btn_add_spectra()

Add spectra to different panel

Source code in mmg_toolbox/tkguis/xmcd_visualiser/sum_rules_tab.py 
106
107
108
109
def btn_add_spectra(self):
    """Add spectra to different panel"""
    if self.spectra:
        self._base.comparison.treeview.add_spectra(self.spectra)