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
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
|
# -*- coding: utf-8; -*-
"""
Copyright (c) 2019 Rolf Hempel, rolf6419@gmx.de
This file is part of the PlanetarySystemStacker tool (PSS).
https://github.com/Rolf-Hempel/PlanetarySystemStacker
PSS is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
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.
You should have received a copy of the GNU General Public License
along with PSS. If not, see <http://www.gnu.org/licenses/>.
"""
from numpy import arange
from PyQt5 import QtWidgets
import matplotlib
matplotlib.use('Agg')
import matplotlib.pyplot as plt
from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg as Canvas
from matplotlib.figure import Figure
from shift_distribution_viewer_gui import Ui_shift_distribution_viewer
class MatplotlibWidget(Canvas):
"""
This widget creates a plot of frame qualities, either sorted chronologically or by quality.
"""
def __init__(self, parent=None):
"""
Initialize the widget.
:param parent: Parent object
"""
super(MatplotlibWidget, self).__init__(Figure())
self.setParent(parent)
plt.rcParams.update({'font.size': 8})
self.fig, self.ax = plt.subplots()
def draw_distribution(self, shift_distribution):
"""
Draw the shift distribution.
:param shift_distribution: 1D Numpy array (int) with counts for each shift size.
:return: -
"""
# Find the last non-zero entry in the array. If no alignment points have been set,
# show a message in the widget title.
if max(shift_distribution) > 0:
max_index = [index for index, item in enumerate(shift_distribution) if item != 0][-1] + 1
plt.title('Frequency distribution of local warp sizes at alignment points')
else:
max_index = 1
plt.title('No alignment points, therefore no warp distribution')
pixels = arange(max_index)
plt.bar(pixels, shift_distribution[:max_index], align='center')
plt.xticks(pixels, pixels)
plt.xlabel('Warp size (pixels)')
plt.ylabel('Frequency')
self.fig.canvas.draw()
self.fig.canvas.flush_events()
class ShiftDistributionViewerWidget(QtWidgets.QFrame, Ui_shift_distribution_viewer):
"""
Display the distribution of frame shifts at alignment points. For each AP in each contributing
frame the total shift (in pixels) is rounded to the next integer value. For every integer number
up to the maximum frame shift the number of occurrences is displayed.
"""
def __init__(self, parent_gui, shift_distribution, shift_failure_percent, signal_finished,
parent=None):
"""
Initialize the Viewer. The widget has a fixed size and is rendered as a QFrame.
:param parent_gui: GUI object by which the viewer is invoked.
:param parent: Parent object
:param shift_distribution: 1D Numpy array (int) with counts for each shift size.
:param shift_failure_percent: Percent of failed warp shift measurements.
:param signal_finished: Qt signal with signature () telling the workflow thread that the
viewer has finished, or None (no signalling).
"""
QtWidgets.QFrame.__init__(self, parent)
self.setupUi(self)
self.setFrameShape(QtWidgets.QFrame.Panel)
self.setFrameShadow(QtWidgets.QFrame.Sunken)
self.setObjectName("shift_distribution_viewer")
self.setFixedSize(900, 600)
self.parent_gui = parent_gui
self.shift_distribution = shift_distribution
self.signal_finished = signal_finished
self.buttonBox.accepted.connect(self.accept)
# Create the Matplotlib widget showing the shift statistics.
self.matplotlib_widget = MatplotlibWidget()
self.verticalLayout.insertWidget(0, Canvas(self.matplotlib_widget.fig))
self.matplotlib_widget.draw_distribution(self.shift_distribution)
# Display the fraction of failed shift measurements. Use red ink if the value exceeds 5%.
if shift_failure_percent >= 0.:
self.failedShiftsLabel.setText(
"Failed shift measurements: " + str(shift_failure_percent) + "%")
if shift_failure_percent > 5.:
self.failedShiftsLabel.setStyleSheet('color: red')
else:
self.failedShiftsLabel.setStyleSheet('color: black')
else:
self.failedShiftsLabel.setText("")
def accept(self):
"""
:return: -
"""
# If a signal was passed in initialization, tell the workflow thread that the viewer has
# finished.
if self.signal_finished is not None:
self.signal_finished.emit()
self.close()
def closeEvent(self, event):
"""
Remove the widget from the parent GUI and close the viewer.
:param event: Close event object
:return: -
"""
# Remove the viewer widget from the main GUI and exit.
self.parent_gui.display_widget(None, display=False)
self.close()
|
