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
|
# Copyright 2004-2009 Nanorex, Inc. See LICENSE file for details.
"""
ChunkProp.py
@author: Mark
@version: $Id$
@copyright: 2004-2009 Nanorex, Inc. See LICENSE file for details.
History: Original code from MoleculeProps.py and cleaned up by Mark.
"""
from PyQt4 import QtGui
from PyQt4.Qt import QDialog, SIGNAL, QColorDialog
from commands.ChunkProperties.ChunkPropDialog import Ui_ChunkPropDialog
from widgets.widget_helpers import RGBf_to_QColor, QColor_to_RGBf
class ChunkProp(QDialog, Ui_ChunkPropDialog):
def __init__(self, chunk):
QDialog.__init__(self)
self.setupUi(self)
self.chunk = chunk
self.glpane = chunk.glpane
self.connect(self.ok_btn,SIGNAL("clicked()"),self.accept)
self.connect(self.cancel_btn,SIGNAL("clicked()"),self.reject)
self.connect(self.reset_color_btn,SIGNAL("clicked()"),self.reset_chunk_color)
self.connect(self.choose_color_btn,SIGNAL("clicked()"),self.change_chunk_color)
self.connect(self.make_atoms_visible_btn,SIGNAL("clicked()"),self.make_atoms_visible)
self.setup()
def setup(self):
# Chunk color
self.original_color = self.chunk.color # Save original Chunk color in case of Cancel
if self.chunk.color: # Set colortile to chunk color (without border)
self.chunk_QColor = RGBf_to_QColor(self.chunk.color) # Used as default color by Color Chooser
else: # Set the colortile to the dialog's bg color (no color)
self.chunk_QColor =self.palette().color(QtGui.QPalette.Window)
plt = QtGui.QPalette()
plt.setColor(QtGui.QPalette.Active,QtGui.QPalette.Window,self.chunk_QColor)
plt.setColor(QtGui.QPalette.Inactive,QtGui.QPalette.Window,self.chunk_QColor)
plt.setColor(QtGui.QPalette.Disabled,QtGui.QPalette.Window,self.chunk_QColor)
self.chunk_color_frame.setPalette(plt)
self.nameLineEdit.setText(self.chunk.name)
self.atomsTextBrowser.setPlainText(self.get_chunk_props_info())
def get_chunk_props_info(self):
"""
Return chunk properties information.
"""
self.atomsTextBrowser.setReadOnly(True)
chunkInfoText = ""
natoms = len(self.chunk.atoms) # number of atoms in the chunk
# Determining the number of element types in this Chunk.
ele2Num = {}
for a in self.chunk.atoms.itervalues():
if not ele2Num.has_key(a.element.symbol):
ele2Num[a.element.symbol] = 1 # New element found
else:
ele2Num[a.element.symbol] += 1 # Increment element
# String construction for each element to be displayed.
nsinglets = 0
for item in ele2Num.iteritems():
if item[0] == "X": # It is a Singlet
nsinglets = int(item[1])
continue
else: eleStr = item[0] + ": " + str(item[1]) + "\n"
chunkInfoText += eleStr
if nsinglets:
eleStr = "\nBondpoints: " + str(nsinglets) + "\n"
chunkInfoText += eleStr
natoms -= nsinglets
header = "Total Atoms: " + str(natoms) + "\n"
return header + chunkInfoText
def change_chunk_color(self):
"""
Slot method to change the chunk's color.
"""
color = QColorDialog.getColor(self.chunk_QColor, self)
if color.isValid():
plt = QtGui.QPalette()
plt.setColor(QtGui.QPalette.Active,QtGui.QPalette.Window,color)
plt.setColor(QtGui.QPalette.Inactive,QtGui.QPalette.Window,color)
plt.setColor(QtGui.QPalette.Disabled,QtGui.QPalette.Window,color)
self.chunk_color_frame.setPalette(plt)
self.chunk_QColor = color
self.chunk.color = QColor_to_RGBf(color)
self.chunk.setcolor(self.chunk.color)
if self.chunk.hidden: # A hidden chunk has no glpane attr.
return
#Ninad 070321:
#Note:
#The chunk is NOT unpicked immediately after changing the color via
#chunk property dialog, This is intentional.
#BTW I don't know why it deselects the chunk after hitting OK or Cancel!
#(looks like an old Qt4 transition bug)
self.glpane.gl_update()
def reset_chunk_color(self):
"""
Slot method to reset the chunk's color.
"""
if not self.chunk.color:
return
self.chunk_QColor = self.palette().color(QtGui.QPalette.Window)
plt = QtGui.QPalette()
plt.setColor(QtGui.QPalette.Active,QtGui.QPalette.Window,self.chunk_QColor)
plt.setColor(QtGui.QPalette.Inactive,QtGui.QPalette.Window,self.chunk_QColor)
plt.setColor(QtGui.QPalette.Disabled,QtGui.QPalette.Window,self.chunk_QColor)
self.chunk_color_frame.setPalette(plt)
self.chunk.color = None
self.chunk.setcolor(self.chunk.color)
if self.chunk.hidden: # A hidden chunk has no glpane attr.
return
self.glpane.gl_update()
def make_atoms_visible(self):
"""
Makes any atoms in this chunk visible.
"""
self.chunk.show_invisible_atoms()
if self.chunk.hidden: # A hidden chunk has no glpane attr.
return
self.glpane.gl_update()
def accept(self):
"""
Slot for the 'OK' button
"""
self.chunk.try_rename(self.nameLineEdit.text())
self.chunk.assy.w.win_update() # Update model tree
self.chunk.assy.changed()
QDialog.accept(self)
def reject(self):
"""
Slot for the 'Cancel' button
"""
self.chunk.color = self.original_color
self.chunk.setcolor(self.chunk.color)
QDialog.reject(self)
# A hidden chunk has no glpane attr. This fixes bug 1137. Mark 051126.
if self.chunk.hidden:
return
self.glpane.gl_update()
|
