GIScan/gisaxs.py at main · sstendahl/GIScan · GitHub

1
2
3
4
5
6
7
8
9
10
11
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
from PyQt5.QtWidgets import QFileDialog
from pathlib import Path
import os
import numpy as np
import cbf
import matplotlib.pyplot as plt
from matplotlib.backends.backend_qt5agg import NavigationToolbar2QT as NavigationToolbar
import scanning_tools as scan
from sample import Sample
import plottingtools
import settings

def get_labels():
    """Retrieve label names for current mapping."""
    mapping = settings.get_config("mapping")
    if mapping == "Angular":
        in_plane_label = r"In-plane scattering angle $\phi{_f}$ (°)"
        out_of_plane_label = r"Out-of-plane scattering angle $\alpha{_f}$ (°)"
    elif mapping == "q-space":
        in_plane_label = r"In-plane scattering vector q${_y}$ (Å$^{-1}$)"
        out_of_plane_label = r"Out-of-plane scattering vector q${_z}$ (Å$^{-1}$)"
    else:  # Pixels
        in_plane_label = "Horizontal detector position (pixels)"
        out_of_plane_label = "Vertical detector position (pixels)"
    return in_plane_label, out_of_plane_label

def loadEmpty(self):
    """Load empty graphs without any data for initial setup."""

    in_plane_label, out_of_plane_label = get_labels()
    gisaxsmap_canvas = plottingtools.PlotWidget(xlabel=in_plane_label, ylabel=out_of_plane_label,
                        title = "GISAXS data")

    horizontalscan_canvas = plottingtools.PlotWidget(xlabel=in_plane_label, ylabel="Intensity (arb. u)",
                        title = "Horizontal scan")
    verticalscan_canvas = plottingtools.PlotWidget(xlabel=out_of_plane_label, ylabel="Intensity (arb. u)",
                        title = "Vertical scan")
    create_layout(self, gisaxsmap_canvas, self.maplayout)
    create_layout(self, horizontalscan_canvas, self.horizontal_layout)
    create_layout(self, verticalscan_canvas, self.vertical_layout)

def create_layout(self, canvas, layout):
    """Create a layout to plot a graph on."""
    toolbar = NavigationToolbar(canvas, self)
    layout.addWidget(canvas)
    layout.addWidget(toolbar)

def loadMap_from_file_picker(self):
    """Load the GISAXS map after selecting it from the file picker."""
    path = getPath(self)
    loadMap(self, path)

def set_ROI_mode(self, mode = "ROI"):
    """Set ROI to either regular ROI or Background mode."""
    if mode == "ROI":
        setter1 = True
        setter2 = False
    elif mode == "background":
        setter1 = False
        setter2 = True

    self.ROI_button.setChecked(setter1)
    self.bg_ROI_button.setChecked(setter2)
    self.ROI_scan_rect.set_visible(setter1)
    self.ROI_scan_rect.set_active(setter1)
    self.ROI_background_rect.set_visible(setter2)
    self.ROI_background_rect.set_active(setter2)


def loadMap(self, file):
    """Load the selected GISAXS map."""

    if self.ROI_scan_rect == None:
        self.connectActions()
    self.firstRun = True
    self.holdVertical.setChecked(False)
    self.holdHorizontal.setChecked(False)
    self.figurecanvas = None
    self.sampledata = Sample()

    if file != "":
        path = os.path.dirname(file)
        filename = Path(file).name
        os.chdir(path)
        self.filename = filename
        contents = cbf.read(filename)
        data = contents.data
        self.sampledata.gisaxs_data = data
        self.sampledata.invalidate_cache()
        self.sampledata.path = file

        layout = self.maplayout
        self.clearLayout(self.maplayout)
        self.figurecanvas = plottingtools.singlePlotonCanvas(self, layout, data, title = filename, style = "default")
        try:
            self.define_rectangle()
            set_ROI_mode(self, "ROI")
            scan.find_specular(self)
            scan.detector_scan(self)
            self.holdHorizontal.setChecked(False)
            scan.YonedaScan(self)
        except:
            print("Error doing preset runs")
        self.firstRun = False


def getPath(self, documenttype="GISAXS data file (*.cbf);;All Files (*)"):
    """Get the path from the file picker."""
    options = QFileDialog.Options()
    path = QFileDialog.getOpenFileName(self,"Open GISAXS data file", "",documenttype, options=options)[0]
    return path