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
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
|
# Copyright 2004-2007 Nanorex, Inc. See LICENSE file for details.
"""
ESPImageProp.py
$Id$
"""
from PyQt4.Qt import QWidget
from PyQt4.Qt import SIGNAL
from PyQt4.Qt import QColorDialog
from PyQt4.Qt import QDialog
from analysis.ESP.ESPImagePropDialog import Ui_ESPImagePropDialog
from widgets.widget_helpers import RGBf_to_QColor, QColor_to_RGBf, get_widget_with_color_palette
class ESPImageProp(QDialog, Ui_ESPImagePropDialog):
def __init__(self, esp_image, glpane):
QWidget.__init__(self)
self.setupUi(self)
self.connect(self.ok_btn,SIGNAL("clicked()"),self.accept)
self.connect(self.cancel_btn,SIGNAL("clicked()"),self.reject)
self.connect(self.choose_border_color_btn,SIGNAL("clicked()"),self.change_border_color)
self.connect(self.choose_fill_color_btn,SIGNAL("clicked()"),self.change_fill_color)
self.connect(self.show_esp_bbox_checkbox,SIGNAL("toggled(bool)"),self.show_esp_bbox)
self.connect(self.select_atoms_btn,SIGNAL("clicked()"),self.select_atoms_inside_esp_bbox)
self.connect(self.highlight_atoms_in_bbox_checkbox,SIGNAL("toggled(bool)"),self.highlight_atoms_in_bbox)
self.connect(self.calculate_esp_btn,SIGNAL("clicked()"),self.calculate_esp)
self.connect(self.rotate_ccw_btn,SIGNAL("clicked()"),self.rotate_90)
self.connect(self.rotate_cw_btn,SIGNAL("clicked()"),self.rotate_neg_90)
self.connect(self.flip_btn,SIGNAL("clicked()"),self.flip_esp_image)
self.connect(self.mirror_btn,SIGNAL("clicked()"),self.mirror_esp_image)
self.connect(self.opacity_slider,SIGNAL("valueChanged(int)"),self.change_opacity)
self.connect(self.choose_file_btn,SIGNAL("clicked()"),self.change_esp_image)
self.connect(self.load_btn,SIGNAL("clicked()"),self.load_esp_image)
self.connect(self.clear_btn,SIGNAL("clicked()"),self.clear_esp_image)
self.connect(self.xaxis_spinbox,SIGNAL("valueChanged(int)"),self.change_xaxisOrient)
self.connect(self.yaxis_spinbox,SIGNAL("valueChanged(int)"),self.change_yaxisOrient)
self.connect(self.width_linedit,SIGNAL("returnPressed()"),self.change_jig_size)
self.connect(self.edge_offset_linedit,SIGNAL("returnPressed()"),self.change_jig_size)
self.connect(self.image_offset_linedit,SIGNAL("returnPressed()"),self.change_jig_size)
self.jig = esp_image
self.glpane = glpane
def setup(self):
#Default state for atoms selection
self.selected = False
self.jig_attrs = self.jig.copyable_attrs_dict() # Save the jig's attributes in case of Cancel.
# Jig color
self.fill_QColor = RGBf_to_QColor(self.jig.fill_color) # Used as default color by Color Chooser
self.border_QColor = RGBf_to_QColor(self.jig.normcolor) # Used as default color by Color Chooser
self.fill_color_pixmap = get_widget_with_color_palette(
self.fill_color_pixmap, self.fill_QColor)
self.border_color_pixmap = get_widget_with_color_palette(
self.border_color_pixmap,self.border_QColor)
self.name_linedit.setText(self.jig.name)
self.width_linedit.setText(str(self.jig.width))
self.image_offset_linedit.setText(str(self.jig.image_offset))
self.edge_offset_linedit.setText(str(self.jig.edge_offset))
self.resolution_spinbox.setValue(self.jig.resolution)
self.show_esp_bbox_checkbox.setChecked(self.jig.show_esp_bbox)
opacity = '%1.2f ' % self.jig.opacity
self.opacity_linedit.setText(opacity)
self.opacity_slider.setValue(int (self.jig.opacity * 100))
self.highlight_atoms_in_bbox_checkbox.setChecked(self.jig.highlightChecked)
self._updateSelectButtonText()
self.xaxis_spinbox.setValue(self.jig.xaxis_orient)
self.yaxis_spinbox.setValue(self.jig.yaxis_orient)
self.png_fname_linedit.setText(self.jig.espimage_file)
def change_fill_color(self):
"""
Slot method to change fill color.
"""
color = QColorDialog.getColor(self.fill_QColor, self)
if color.isValid():
self.fill_QColor = color
self.fill_color_pixmap = get_widget_with_color_palette(
self.fill_color_pixmap, self.fill_QColor)
self.jig.fill_color = QColor_to_RGBf(color)
self.glpane.gl_update()
def change_border_color(self):
"""
Slot method change border color.
"""
color = QColorDialog.getColor(self.border_QColor, self)
if color.isValid():
self.border_QColor = color
self.border_color_pixmap = get_widget_with_color_palette(
self.border_color_pixmap,self.border_QColor)
self.jig.color = self.jig.normcolor = QColor_to_RGBf(color)
self.glpane.gl_update()
def change_jig_size(self, gl_update=True):
"""
Slot method to change the jig's length, radius and/or spoke radius.
"""
self.jig.width = float(str(self.width_linedit.text()))
self.jig.image_offset = float(str(self.image_offset_linedit.text()))
self.jig.edge_offset = float(str(self.edge_offset_linedit.text()))
if gl_update:
self.glpane.gl_update()
def change_opacity(self, val):
"""
Slot for opacity slider
"""
self.jig.opacity = val/100.0
opacity = '%1.2f ' % self.jig.opacity
self.opacity_linedit.setText(opacity)
self.glpane.gl_update()
def show_esp_bbox(self, val):
"""
Slot for Show ESP Image Volume checkbox
"""
self.jig.show_esp_bbox = val
self.glpane.gl_update()
def highlight_atoms_in_bbox(self, val):
"""
Slot for Highlight Atoms Inside Volume checkbox
"""
self.jig.highlightChecked = val
self.glpane.gl_update()
def _updateSelectButtonText(self):
if self.selected:
self.select_atoms_btn.setText("Deselect Atoms Inside Volume")
else:
self.select_atoms_btn.setText("Select Atoms Inside Volume")
def select_atoms_inside_esp_bbox(self):
"""
Slot for Select Atoms Inside Volume button,
which selects all the atoms inside the bbox
"""
self.selected = not self.selected
self._updateSelectButtonText()
self.jig.pickSelected(self.selected)
self.glpane.gl_update()
def calculate_esp(self):
self.jig.calculate_esp()
def load_esp_image(self):
self.jig.load_espimage_file()
self.png_fname_linedit.setText(self.jig.espimage_file)
def change_esp_image(self):
self.jig.load_espimage_file(choose_new_image = True, parent=self)
self.png_fname_linedit.setText(self.jig.espimage_file)
def clear_esp_image(self):
self.jig.clear_esp_image()
def change_xaxisOrient(self, val):
self.jig.xaxis_orient = val
def change_yaxisOrient(self, val):
self.jig.yaxis_orient = val
def rotate_90(self):
"""
CCW rotate 90 degrees.
"""
self.jig.rotate_esp_image(90)
def rotate_neg_90(self):
"""
CW rotate 90 degrees.
"""
self.jig.rotate_esp_image(-90)
def flip_esp_image(self):
"""
Flip esp image.
"""
self.jig.flip_esp_image()
def mirror_esp_image(self):
"""
Mirror esp image.
"""
self.jig.mirror_esp_image()
def accept(self):
"""
Slot for the 'OK' button
"""
self.jig.cancelled = False
self.jig.try_rename(self.name_linedit.text())
self.change_jig_size(gl_update=False)
self.jig.resolution = float(self.resolution_spinbox.value())
# Before exit the dialog, turn off the highlighting and selection
#self.jig.highlightChecked = False # Let's try leaving highlighted atoms on. It's useful. Mark 051004.
self.jig.pickSelected(False)
self.jig.assy.w.win_update() # Update model tree
self.jig.assy.changed()
QDialog.accept(self)
def reject(self):
"""
Slot for the 'Cancel' button
"""
self.jig.attr_update(self.jig_attrs) # Restore attributes of the jig.
# Before exit the dialog, turn off the highlighting and selection
self.jig.highlightChecked = False
self.jig.pickSelected(False)
self.glpane.gl_update()
QDialog.reject(self)
|
