/
qt_utils.py
190 lines (169 loc) · 6.58 KB
/
qt_utils.py
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
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
###########################################################################
#
# Copyright 2015-2018 Robert B. Lowrie (http://github.com/lowrie)
#
# This file is part of pyRouterJig.
#
# pyRouterJig 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.
#
# pyRouterJig 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
# pyRouterJig; see the file LICENSE. If not, see <http://www.gnu.org/licenses/>.
#
###########################################################################
'''
Contains utilities for Qt functionality
'''
from __future__ import print_function
import os
import glob
import operator
from PyQt5 import QtCore, QtWidgets
import router
def set_router_value(line_edit, obj, attr, setter, is_float=False, bit=None):
'''
Sets the value contained in the line_edit object.
'''
if not line_edit.isModified():
return None
line_edit.setModified(False)
text = str(line_edit.text())
units = getattr(obj, 'units')
try:
# Call the appropriate setter function. This is the function that will
# throw an exception.
if bit is None:
getattr(obj, setter)(text)
else:
getattr(obj, setter)(bit, text)
# Set the string value of the new value
new_value = getattr(obj, attr)
if is_float:
text = str(new_value)
else:
text = units.increments_to_string(new_value)
line_edit.setText(text)
return text
except router.Router_Exception as e:
# Notify the user of the error, and set the line editor back to its
# original value
QtWidgets.QMessageBox.warning(line_edit.parentWidget(), 'Error', e.msg)
old_value = getattr(obj, attr)
text = units.increments_to_string(old_value)
line_edit.setText(text)
return None
class PreviewComboBox(QtWidgets.QComboBox):
'''
This comboxbox emits "activated" when hidePopup is called. This allows
for a combobox with a preview mode, so that as each selection is
highlighted with the popup open, the figure can be updated. Once the
popup is closed, this hidePopup ensures that the figure is redrawn with
the current actual selection.
'''
def __init__(self, parent):
QtWidgets.QComboBox.__init__(self, parent)
def hidePopup(self):
QtWidgets.QComboBox.hidePopup(self)
# print('hidePopup')
self.activated.emit(self.currentIndex())
class ShadowTextLineEdit(QtWidgets.QLineEdit):
'''
This line edit sets a grayed shadow text, until focus is received and text
is entered. Shadow text is the text displayed when the line edit is empty.
'''
def __init__(self, parent, shadow_text):
QtWidgets.QLineEdit.__init__(self, parent)
self.shadow_text = shadow_text
self.has_real_text = False
self.initialize_shadow()
def initialize_shadow(self):
self.setText(self.shadow_text)
self.setStyleSheet('color: gray;')
self.has_real_text = False
def focusInEvent(self, event):
QtWidgets.QLineEdit.focusInEvent(self, event)
# If no real text, clear the shadow text and darken the text
if not self.has_real_text:
self.clear()
self.setStyleSheet('color: black;')
def focusOutEvent(self, event):
QtWidgets.QLineEdit.focusOutEvent(self, event)
# If there's no text, set it back to the shadow
if str(self.text()) == '':
self.initialize_shadow()
else:
self.has_real_text = True
def set_line_style(line):
'''Sets the style for create_vline() and create_hline()'''
line.setFrameShadow(QtWidgets.QFrame.Raised)
line.setLineWidth(1)
line.setMidLineWidth(1)
line.setSizePolicy(QtWidgets.QSizePolicy.Preferred, QtWidgets.QSizePolicy.Preferred)
def create_vline():
'''Creates a vertical line'''
vline = QtWidgets.QFrame()
vline.setFrameStyle(QtWidgets.QFrame.VLine)
set_line_style(vline)
return vline
def create_hline():
'''Creates a horizontal line'''
hline = QtWidgets.QFrame()
hline.setFrameStyle(QtWidgets.QFrame.HLine)
set_line_style(hline)
return hline
def create_wood_dict(wood_images, transl):
'''
Creates a dictionary {wood_name : wood_image_filename} by parsing the
directory wood_images. The wood_name is formed by taking the prefix of the
wood_image_filename.
'''
# form woods dictionary, which is empty if wood_images directory does not exist
woods = {}
if os.path.isdir(wood_images):
globber = os.path.join(wood_images, '*')
files = glob.glob(globber)
for f in files:
name = os.path.basename(f)
i = name.rfind('.')
if i > 0:
name = transl.tr(name[0:i])
path = os.path.abspath(f)
woods[name] = path
# Set the available patterns
patterns = {transl.tr('DiagCrossPattern'): QtCore.Qt.DiagCrossPattern,
transl.tr('BDiagPattern'): QtCore.Qt.BDiagPattern,
transl.tr('FDiagPattern'): QtCore.Qt.FDiagPattern,
transl.tr('Dense1Pattern'): QtCore.Qt.Dense1Pattern,
transl.tr('Dense5Pattern'): QtCore.Qt.Dense5Pattern,
transl.tr('Solid Fill'): QtCore.Qt.SolidPattern,
transl.tr('No Fill'): None}
return (woods, patterns)
def create_lang_dict():
'''
Creates a dictionary {lang_name : locale_id} by parsing the
directory ./ts. The language name is formed by taking the prefix of the
language file. The default language en_US added by default
'''
lang_path = '.\\ts\\'
langs = {}
langs['English'] = 'en_US' # set default language
if os.path.isdir(lang_path):
globber = os.path.join(lang_path, 'pyRouterJig_*.qm')
files = glob.glob(globber)
for f in files:
name = os.path.basename(f)
i = name.rfind('.')
if i > 0:
name = name[12:i]
locale = QtCore.QLocale(name)
name = locale.name()
if name != 'C':
langs[locale.languageToString(locale.language())] = name
langs = sorted(langs.items(), key=operator.itemgetter(0))
return langs