def __init__(self, *args):
QWidget.__init__(self, *args)
# init layout
self._lo = QVBoxLayout(self)
self._lo.setContentsMargins(0, 0, 0, 0)
self._lo.setSpacing(0)
# init internal state
self._currier = PersistentCurrier()
self._z0 = 0
# z-depth of first image, the rest count down from it
self._updating_imap = False
self._locked_display_range = False
self._imagecons = []
self._imagecon_loadorder = []
self._center_image = None
self._plot = None
self._border_pen = None
self._drawing_key = None
self._load_image_dialog = None
self._model_imagecons = set()
# init menu and standard actions
self._menu = QMenu("&Image", self)
qag = QActionGroup(self)
# exclusive controls for plotting topmost or all images
self._qa_plot_top = qag.addAction("Display topmost image only")
self._qa_plot_all = qag.addAction("Display all images")
self._qa_plot_top.setCheckable(True)
self._qa_plot_all.setCheckable(True)
self._qa_plot_top.setChecked(True)
QObject.connect(self._qa_plot_all, SIGNAL("toggled(bool)"),
self._displayAllImages)
self._closing = False
self._qa_load_clipboard = None
self._clipboard_mode = QClipboard.Clipboard
QObject.connect(QApplication.clipboard(),
SIGNAL("changed(QClipboard::Mode)"),
self._checkClipboardPath)
# populate the menu
self._repopulateMenu()
def __init__(self, *args):
QWidget.__init__(self, *args)
# init layout
self._lo = QVBoxLayout(self)
self._lo.setContentsMargins(0, 0, 0, 0)
self._lo.setSpacing(0)
# init internal state
self._currier = PersistentCurrier()
self._z0 = 0; # z-depth of first image, the rest count down from it
self._updating_imap = False
self._locked_display_range = False
self._imagecons = []
self._imagecon_loadorder = []
self._center_image = None
self._plot = None
self._border_pen = None
self._drawing_key = None
self._load_image_dialog = None
self._model_imagecons = set()
# init menu and standard actions
self._menu = QMenu("&Image", self)
qag = QActionGroup(self)
# exclusive controls for plotting topmost or all images
self._qa_plot_top = qag.addAction("Display topmost image only")
self._qa_plot_all = qag.addAction("Display all images")
self._qa_plot_top.setCheckable(True)
self._qa_plot_all.setCheckable(True)
self._qa_plot_top.setChecked(True)
QObject.connect(self._qa_plot_all, SIGNAL("toggled(bool)"), self._displayAllImages)
self._closing = False
self._qa_load_clipboard = None
self._clipboard_mode = QClipboard.Clipboard
QObject.connect(QApplication.clipboard(), SIGNAL("changed(QClipboard::Mode)"), self._checkClipboardPath)
# populate the menu
self._repopulateMenu()
class ThemeMenu(QMenu): # {{{
def __init__(self, parent):
QMenu.__init__(self, _('Choose theme (needs restart)'))
parent.addMenu(self)
self.group = QActionGroup(self)
current = prefs['theme']
alls = list(sorted(get_all_styles()))
if current not in alls:
current = prefs['theme'] = 'default'
self.actions = []
for style in alls:
ac = self.group.addAction(style)
ac.setCheckable(True)
if current == style:
ac.setChecked(True)
self.actions.append(ac)
ac.triggered.connect(partial(self.set_theme, style))
self.addAction(ac)
def set_theme(self, style, *args):
prefs['theme'] = style
class BuildAtomsFlyout(Ui_AbstractFlyout):
def _action_in_controlArea_to_show_this_flyout(self):
"""
Required action in the 'Control Area' as a reference for this
flyout toolbar. See superclass method for documentation and todo note.
"""
return self.win.toolsDepositAtomAction
def _getExitActionText(self):
"""
Overrides superclass method.
@see: self._createActions()
"""
return "Exit Atoms"
def getFlyoutActionList(self):
"""
Returns a tuple that contains mode spcific actionlists in the
added in the flyout toolbar of the mode.
CommandToolbar._createFlyoutToolBar method calls this
@return: params: A tuple that contains 3 lists:
(subControlAreaActionList, commandActionLists, allActionsList)
"""
#ninad070330 This implementation may change in future.
#'allActionsList' returns all actions in the flyout toolbar
#including the subcontrolArea actions.
allActionsList = []
#Action List for subcontrol Area buttons.
#For now this list is just used to set a different palette to thisaction
#buttons in the toolbar
subControlAreaActionList =[]
#Append subcontrol area actions to subControlAreaActionList
#The 'Exit button' althought in the subcontrol area, would
#look as if its in the Control area because of the different color
#palette
#and the separator after it. This is intentional.
subControlAreaActionList.append(self.exitModeAction)
separator1 = QAction(self.win)
separator1.setSeparator(True)
subControlAreaActionList.append(separator1)
subControlAreaActionList.append(self.atomsToolAction)
subControlAreaActionList.append(self.bondsToolAction)
separator = QAction(self.win)
separator.setSeparator(True)
subControlAreaActionList.append(separator)
#Also add the subcontrol Area actions to the 'allActionsList'
for a in subControlAreaActionList:
allActionsList.append(a)
##actionlist.append(self.win.modifyAdjustSelAction)
##actionlist.append(self.win.modifyAdjustAllAction)
#Ninad 070330:
#Command Actions : These are the actual actions that will respond the
#a button pressed in the 'subControlArea (if present).
#Each Subcontrol area button can have its own 'command list'
#The subcontrol area buuton and its command list form a 'key:value pair
#in a python dictionary object
#In Build mode, as of 070330 we have 3 subcontrol area buttons
#(or 'actions)'
commandActionLists = []
#Append empty 'lists' in 'commandActionLists equal to the
#number of actions in subControlArea
for i in range(len(subControlAreaActionList)):
lst = []
commandActionLists.append(lst)
#Command list for the subcontrol area button 'Atoms Tool'
#For others, this list will remain empty for now -- ninad070330
depositAtomsCmdLst = []
depositAtomsCmdLst.append(self.win.modifyHydrogenateAction)
depositAtomsCmdLst.append(self.win.modifyDehydrogenateAction)
depositAtomsCmdLst.append(self.win.modifyPassivateAction)
separatorAfterPassivate = QAction(self.win)
separatorAfterPassivate.setSeparator(True)
depositAtomsCmdLst.append(separatorAfterPassivate)
depositAtomsCmdLst.append(self.transmuteAtomsAction)
separatorAfterTransmute = QAction(self.win)
separatorAfterTransmute.setSeparator(True)
depositAtomsCmdLst.append(separatorAfterTransmute)
depositAtomsCmdLst.append(self.win.modifyDeleteBondsAction)
depositAtomsCmdLst.append(self.win.modifySeparateAction)
depositAtomsCmdLst.append(self.win.makeChunkFromSelectedAtomsAction)
commandActionLists[2].extend(depositAtomsCmdLst)
##for action in self.win.buildToolsMenu.actions():
##commandActionLists[2].append(action)
# Command list for the subcontrol area button 'Bonds Tool'
# For others, this list will remain empty for now -- ninad070405
bondsToolCmdLst = []
bondsToolCmdLst.append(self.bond1Action)
bondsToolCmdLst.append(self.bond2Action)
bondsToolCmdLst.append(self.bond3Action)
bondsToolCmdLst.append(self.bondaAction)
bondsToolCmdLst.append(self.bondgAction)
bondsToolCmdLst.append(self.cutBondsAction)
commandActionLists[3].extend(bondsToolCmdLst)
params = (subControlAreaActionList, commandActionLists, allActionsList)
return params
def _createActions(self, parentWidget):
"""
Define flyout toolbar actions for this mode.
"""
#@NOTE: In Build mode, some of the actions defined in this method are also
#used in Build Atoms PM. (e.g. bond actions) So probably better to rename
#it as _init_modeActions. Not doing that change in mmkit code cleanup
#commit(other modes still implement a method by same name)-ninad20070717
_superclass._createActions(self, parentWidget)
# The subControlActionGroup is the parent of all flyout QActions.
self.subControlActionGroup = QActionGroup(parentWidget)
# Shouldn't the parent be self.win?
# Isn't parentWidget = BuildAtomsPropertyManager.
# Ask Bruce about this. --Mark 2008-12-07.
self.subControlActionGroup.setExclusive(True)
#Following Actions are added in the Flyout toolbar.
#Defining them outside that method as those are being used
#by the subclasses of deposit mode (testmode.py as of 070410) -- ninad
self.atomsToolAction = \
NE1_QWidgetAction(self.subControlActionGroup, win = self.win)
self.atomsToolAction.setText("Atoms Tool")
self.atomsToolAction.setIcon(geticon(
"ui/actions/Command Toolbar/BuildAtoms/AtomsTool.png"))
self.atomsToolAction.setCheckable(True)
self.atomsToolAction.setChecked(True)
self.atomsToolAction.setObjectName('ACTION_ATOMS_TOOL')
self.bondsToolAction = \
NE1_QWidgetAction(self.subControlActionGroup, win = self.win)
self.bondsToolAction.setText("Bonds Tool")
self.bondsToolAction.setIcon(geticon(
"ui/actions/Command Toolbar/BuildAtoms/BondsTool.png"))
self.bondsToolAction.setCheckable(True)
self.bondsToolAction.setObjectName('ACTION_BOND_TOOL')
self.subControlActionGroup.addAction(self.atomsToolAction)
self.subControlActionGroup.addAction(self.bondsToolAction)
self.transmuteAtomsAction = \
NE1_QWidgetAction(self.subControlActionGroup, win = self.win)
self.transmuteAtomsAction.setText("Transmute Atoms")
self.transmuteAtomsAction.setIcon(geticon(
"ui/actions/Command Toolbar/BuildAtoms/TransmuteAtoms.png"))
self.transmuteAtomsAction.setCheckable(False)
self._createBondToolActions(parentWidget)
def _createBondToolActions(self, parentWidget):
"""
Create the actions to be included in flyout toolbar , when the 'bonds
tool' is active. Note that the object names of these action will be
be used to find the Bond Tool type to which each action corresponds to.
@see: self._createActions() where this is called.
"""
self.bondToolsActionGroup = QActionGroup(parentWidget)
self.bondToolsActionGroup.setExclusive(True)
self.bond1Action = \
NE1_QWidgetAction(self.bondToolsActionGroup, win = self.win)
self.bond1Action.setText("Single")
self.bond1Action.setIcon(geticon("ui/actions/Command Toolbar/BuildAtoms/SingleBond.png"))
self.bond1Action.setObjectName('ACTION_SINGLE_BOND_TOOL')
self.bond2Action = \
NE1_QWidgetAction(self.bondToolsActionGroup, win = self.win)
self.bond2Action.setText("Double")
self.bond2Action.setIcon(geticon("ui/actions/Command Toolbar/BuildAtoms/DoubleBond.png"))
self.bond2Action.setObjectName('ACTION_DOUBLE_BOND_TOOL')
self.bond3Action = \
NE1_QWidgetAction(self.bondToolsActionGroup, win = self.win)
self.bond3Action.setText("Triple")
self.bond3Action.setIcon(geticon("ui/actions/Command Toolbar/BuildAtoms/TripleBond.png"))
self.bond3Action.setObjectName('ACTION_TRIPLE_BOND_TOOL')
self.bondaAction = \
NE1_QWidgetAction(self.bondToolsActionGroup, win = self.win)
self.bondaAction.setText("Aromatic")
self.bondaAction.setIcon(geticon("ui/actions/Command Toolbar/BuildAtoms/AromaticBond.png"))
self.bondaAction.setObjectName('ACTION_AROMATIC_BOND_TOOL')
self.bondgAction = \
NE1_QWidgetAction(self.bondToolsActionGroup, win = self.win)
self.bondgAction.setText("Graphitic")
self.bondgAction.setIcon(geticon("ui/actions/Command Toolbar/BuildAtoms/GraphiticBond.png"))
self.bondgAction.setObjectName('ACTION_GRAPHITIC_BOND_TOOL')
self.cutBondsAction = \
NE1_QWidgetAction(self.bondToolsActionGroup, win = self.win)
self.cutBondsAction.setText("Cut Bonds")
self.cutBondsAction.setIcon(geticon("ui/actions/Command Toolbar/BuildAtoms/CutBonds.png"))
self.cutBondsAction.setObjectName('ACTION_DELETE_BOND_TOOL')
for action in [self.bond1Action,
self.bond2Action,
self.bond3Action,
self.bondaAction,
self.bondgAction,
self.cutBondsAction
]:
self.bondToolsActionGroup.addAction(action)
action.setCheckable(True)
return
def _addWhatsThisText(self):
"""
Add 'What's This' help text for all actions on toolbar.
"""
from ne1_ui.WhatsThisText_for_CommandToolbars import whatsThisTextForAtomsCommandToolbar
whatsThisTextForAtomsCommandToolbar(self)
return
def _addToolTipText(self):
"""
Add 'Tool tip' help text for all actions on toolbar.
"""
from ne1_ui.ToolTipText_for_CommandToolbars import toolTipTextForAtomsCommandToolbar
toolTipTextForAtomsCommandToolbar(self)
return
def connect_or_disconnect_signals(self, isConnect):
"""
Connect or disconnect widget signals sent to their slot methods.
This can be overridden in subclasses. By default it does nothing.
@param isConnect: If True the widget will send the signals to the slot
method.
@type isConnect: boolean
@see: self.activateFlyoutToolbar, self.deActivateFlyoutToolbar
"""
if isConnect:
change_connect = self.win.connect
else:
change_connect = self.win.disconnect
#Ui_AbstractFlyout connects the self.exitmodeAction, so call it first.
_superclass.connect_or_disconnect_signals(self, isConnect)
#Atom , Bond Tools Groupbox
change_connect(self.bondToolsActionGroup,
SIGNAL("triggered(QAction *)"),
self.command.changeBondTool)
change_connect(self.transmuteAtomsAction,
SIGNAL("triggered()"),self.command.transmutePressed)
change_connect(self.subControlActionGroup,
SIGNAL("triggered(QAction *)"),
self.updateCommandToolbar)
change_connect(self.bondsToolAction,
SIGNAL("triggered()"),
self._activateBondsTool)
change_connect(self.atomsToolAction,
SIGNAL("triggered()"),
self._activateAtomsTool)
def get_cursor_id_for_active_tool(self):
"""
Provides a cursor id (int) for updating cursor in graphics mode,
based on the checked action in its flyout toolbar. (i.e. based on the
active tool)
@see: BuildAtoms_GraphicsMode.update_cursor_for_no_MB_selection_filter_disabled
"""
if self.atomsToolAction.isChecked():
cursor_id = 0
elif self.bond1Action.isChecked():
cursor_id = 1
elif self.bond2Action.isChecked():
cursor_id = 2
elif self.bond3Action.isChecked():
cursor_id = 3
elif self.bondaAction.isChecked():
cursor_id = 4
elif self.bondgAction.isChecked():
cursor_id = 5
elif self.cutBondsAction.isChecked():
cursor_id = 6
else:
cursor_id = 0
return cursor_id
def _activateAtomsTool(self):
"""
Activate the atoms tool of the Build Atoms mode
hide only the Atoms Tools groupbox in the Build Atoms Property manager
and show all others the others.
"""
self.command.activateAtomsTool()
def _activateBondsTool(self):
"""
Activate the bond tool of the Build Atoms mode
Show only the Bond Tools groupbox in the Build Atoms Property manager
and hide the others.
@see:self._convert_bonds_bet_selected_atoms()
"""
self.command.activateBondsTool()
def getBondToolActions(self):
return self.bondToolsActionGroup.actions()
def getCheckedBondToolAction(self):
return self.bondToolsActionGroup.checkedAction()
def resetStateOfActions(self):
"""
Default implementation does nothing. Overridden in subclasses.
R Resets the state of actions in the flyout toolbar.
Generally, it unchecks all the actions except the ExitModeAction.
This is called while resuming a command.
Example: if exits is in Insert > Dna command,
the Build > Dna command is resumed. When this happens, program needs to
make sure that the Insert > dna button in the flyout is unchecked.
It is done by using this method.
@see: self.deActivateFlyoutToolbar()
@see: self.activateBreakStrands_Command()
@see: baseCommand.command_update_flyout() which calls this method.
@see: BuildAtoms_Command.command_update_state()
@see: BuildAtoms_Command.activateAtomsTool()
"""
self.atomsToolAction.setChecked(True)
##self.bondsToolAction.setChecked(False)
for action in self.bondToolsActionGroup.actions():
action.setChecked(False)
class LocationManager(QObject): # {{{
locations_changed = pyqtSignal()
unmount_device = pyqtSignal()
location_selected = pyqtSignal(object)
configure_device = pyqtSignal()
update_device_metadata = pyqtSignal()
def __init__(self, parent=None):
QObject.__init__(self, parent)
self.free = [-1, -1, -1]
self.count = 0
self.location_actions = QActionGroup(self)
self.location_actions.setExclusive(True)
self.current_location = 'library'
self._mem = []
self.tooltips = {}
self.all_actions = []
def ac(name, text, icon, tooltip):
icon = QIcon(I(icon))
ac = self.location_actions.addAction(icon, text)
setattr(self, 'location_'+name, ac)
ac.setAutoRepeat(False)
ac.setCheckable(True)
receiver = partial(self._location_selected, name)
ac.triggered.connect(receiver)
self.tooltips[name] = tooltip
m = QMenu(parent)
self._mem.append(m)
a = m.addAction(icon, tooltip)
a.triggered.connect(receiver)
if name != 'library':
self._mem.append(a)
a = m.addAction(QIcon(I('eject.png')), _('Eject this device'))
a.triggered.connect(self._eject_requested)
self._mem.append(a)
a = m.addAction(QIcon(I('config.png')), _('Configure this device'))
a.triggered.connect(self._configure_requested)
self._mem.append(a)
a = m.addAction(QIcon(I('sync.png')), _('Update cached metadata on device'))
a.triggered.connect(lambda x : self.update_device_metadata.emit())
self._mem.append(a)
else:
ac.setToolTip(tooltip)
ac.setMenu(m)
ac.calibre_name = name
self.all_actions.append(ac)
return ac
self.library_action = ac('library', _('Library'), 'lt.png',
_('Show books in calibre library'))
ac('main', _('Device'), 'reader.png',
_('Show books in the main memory of the device'))
ac('carda', _('Card A'), 'sd.png',
_('Show books in storage card A'))
ac('cardb', _('Card B'), 'sd.png',
_('Show books in storage card B'))
def set_switch_actions(self, quick_actions, rename_actions, delete_actions,
switch_actions, choose_action):
self.switch_menu = self.library_action.menu()
if self.switch_menu:
self.switch_menu.addSeparator()
else:
self.switch_menu = QMenu()
self.switch_menu.addAction(choose_action)
self.cs_menus = []
for t, acs in [(_('Quick switch'), quick_actions),
(_('Rename library'), rename_actions),
(_('Delete library'), delete_actions)]:
if acs:
self.cs_menus.append(QMenu(t))
for ac in acs:
self.cs_menus[-1].addAction(ac)
self.switch_menu.addMenu(self.cs_menus[-1])
self.switch_menu.addSeparator()
for ac in switch_actions:
self.switch_menu.addAction(ac)
if self.switch_menu != self.library_action.menu():
self.library_action.setMenu(self.switch_menu)
def _location_selected(self, location, *args):
if location != self.current_location and hasattr(self,
'location_'+location):
self.current_location = location
self.location_selected.emit(location)
getattr(self, 'location_'+location).setChecked(True)
def _eject_requested(self, *args):
self.unmount_device.emit()
def _configure_requested(self):
self.configure_device.emit()
def update_devices(self, cp=(None, None), fs=[-1, -1, -1], icon=None):
if icon is None:
icon = I('reader.png')
self.location_main.setIcon(QIcon(icon))
had_device = self.has_device
if cp is None:
cp = (None, None)
if isinstance(cp, (str, unicode)):
cp = (cp, None)
if len(fs) < 3:
fs = list(fs) + [0]
self.free[0] = fs[0]
self.free[1] = fs[1]
self.free[2] = fs[2]
cpa, cpb = cp
self.free[1] = fs[1] if fs[1] is not None and cpa is not None else -1
self.free[2] = fs[2] if fs[2] is not None and cpb is not None else -1
self.update_tooltips()
if self.has_device != had_device:
self.location_library.setChecked(True)
self.locations_changed.emit()
if not self.has_device:
self.location_library.trigger()
def update_tooltips(self):
for i, loc in enumerate(('main', 'carda', 'cardb')):
t = self.tooltips[loc]
if self.free[i] > -1:
t += u'\n\n%s '%human_readable(self.free[i]) + _('available')
ac = getattr(self, 'location_'+loc)
ac.setToolTip(t)
ac.setWhatsThis(t)
ac.setStatusTip(t)
@property
def has_device(self):
return max(self.free) > -1
@property
def available_actions(self):
ans = [self.location_library]
for i, loc in enumerate(('main', 'carda', 'cardb')):
if self.free[i] > -1:
ans.append(getattr(self, 'location_'+loc))
return ans
class LocationManager(QObject): # {{{
locations_changed = pyqtSignal()
unmount_device = pyqtSignal()
location_selected = pyqtSignal(object)
configure_device = pyqtSignal()
def __init__(self, parent=None):
QObject.__init__(self, parent)
self.free = [-1, -1, -1]
self.count = 0
self.location_actions = QActionGroup(self)
self.location_actions.setExclusive(True)
self.current_location = 'library'
self._mem = []
self.tooltips = {}
self.all_actions = []
def ac(name, text, icon, tooltip):
icon = QIcon(I(icon))
ac = self.location_actions.addAction(icon, text)
setattr(self, 'location_' + name, ac)
ac.setAutoRepeat(False)
ac.setCheckable(True)
receiver = partial(self._location_selected, name)
ac.triggered.connect(receiver)
self.tooltips[name] = tooltip
m = QMenu(parent)
self._mem.append(m)
a = m.addAction(icon, tooltip)
a.triggered.connect(receiver)
if name != 'library':
self._mem.append(a)
a = m.addAction(QIcon(I('eject.png')), _('Eject this device'))
a.triggered.connect(self._eject_requested)
self._mem.append(a)
a = m.addAction(QIcon(I('config.png')),
_('Configure this device'))
a.triggered.connect(self._configure_requested)
self._mem.append(a)
else:
ac.setToolTip(tooltip)
ac.setMenu(m)
ac.calibre_name = name
self.all_actions.append(ac)
return ac
self.library_action = ac('library', _('Library'), 'lt.png',
_('Show books in calibre library'))
ac('main', _('Device'), 'reader.png',
_('Show books in the main memory of the device'))
ac('carda', _('Card A'), 'sd.png', _('Show books in storage card A'))
ac('cardb', _('Card B'), 'sd.png', _('Show books in storage card B'))
def set_switch_actions(self, quick_actions, rename_actions, delete_actions,
switch_actions, choose_action):
self.switch_menu = self.library_action.menu()
if self.switch_menu:
self.switch_menu.addSeparator()
else:
self.switch_menu = QMenu()
self.switch_menu.addAction(choose_action)
self.cs_menus = []
for t, acs in [(_('Quick switch'), quick_actions),
(_('Rename library'), rename_actions),
(_('Delete library'), delete_actions)]:
if acs:
self.cs_menus.append(QMenu(t))
for ac in acs:
self.cs_menus[-1].addAction(ac)
self.switch_menu.addMenu(self.cs_menus[-1])
self.switch_menu.addSeparator()
for ac in switch_actions:
self.switch_menu.addAction(ac)
if self.switch_menu != self.library_action.menu():
self.library_action.setMenu(self.switch_menu)
def _location_selected(self, location, *args):
if location != self.current_location and hasattr(
self, 'location_' + location):
self.current_location = location
self.location_selected.emit(location)
getattr(self, 'location_' + location).setChecked(True)
def _eject_requested(self, *args):
self.unmount_device.emit()
def _configure_requested(self):
self.configure_device.emit()
def update_devices(self, cp=(None, None), fs=[-1, -1, -1], icon=None):
if icon is None:
icon = I('reader.png')
self.location_main.setIcon(QIcon(icon))
had_device = self.has_device
if cp is None:
cp = (None, None)
if isinstance(cp, (str, unicode)):
cp = (cp, None)
if len(fs) < 3:
fs = list(fs) + [0]
self.free[0] = fs[0]
self.free[1] = fs[1]
self.free[2] = fs[2]
cpa, cpb = cp
self.free[1] = fs[1] if fs[1] is not None and cpa is not None else -1
self.free[2] = fs[2] if fs[2] is not None and cpb is not None else -1
self.update_tooltips()
if self.has_device != had_device:
self.location_library.setChecked(True)
self.locations_changed.emit()
if not self.has_device:
self.location_library.trigger()
def update_tooltips(self):
for i, loc in enumerate(('main', 'carda', 'cardb')):
t = self.tooltips[loc]
if self.free[i] > -1:
t += u'\n\n%s ' % human_readable(self.free[i]) + _('available')
ac = getattr(self, 'location_' + loc)
ac.setToolTip(t)
ac.setWhatsThis(t)
ac.setStatusTip(t)
@property
def has_device(self):
return max(self.free) > -1
@property
def available_actions(self):
ans = [self.location_library]
for i, loc in enumerate(('main', 'carda', 'cardb')):
if self.free[i] > -1:
ans.append(getattr(self, 'location_' + loc))
return ans
class CookieCtrlPanel(CookiePropertyManager):
"""
This class provides GUI controls to the cookie-cutter command.
"""
def __init__(self, parentMode):
"""
"""
self.w = parentMode.w
self.cookieCommand = parentMode
#bruce 071008, probably redundant with some other attribute
self.pw = None # pw is active part window
self._init_flyoutActions()
CookiePropertyManager.__init__(self, parentMode)
self._makeConnections()
def _init_flyoutActions(self):
"""
Define flyout toolbar actions for this command
"""
#Create an action group and add all the cookie selection shape buttons to it
self.cookieSelectionGroup = QActionGroup(self.w)
#Action List for subcontrol Area buttons.
#In cookie cutter, there is really no subcontrol area.
#We will treat subcontrol area same as 'command area'
#(subcontrol area buttons will have an empty list as their command area
#list). We will set the Comamnd Area palette background color to the
#subcontrol area. This list will be used in getFlyoutActionList
self.subControlAreaActionList =[]
self.exitCrystalAction = NE1_QWidgetAction(self.w,
win = self.w)
self.exitCrystalAction.setText("Exit Crystal")
self.exitCrystalAction.setCheckable(True)
self.exitCrystalAction.setChecked(True)
self.exitCrystalAction.setIcon(
geticon('ui/actions/Toolbars/Smart/Exit.png'))
self.subControlAreaActionList.append(self.exitCrystalAction)
separator = QtGui.QAction(self.w)
separator.setSeparator(True)
self.subControlAreaActionList.append(separator)
self.DefaultSelAction = NE1_QWidgetAction(self.w, win = self.w)
self.DefaultSelAction.setObjectName("DEFAULT")
self.DefaultSelAction.setText("Default")
self.subControlAreaActionList.append(self.DefaultSelAction)
self.DefaultSelAction.setToolTip( "Default Selection (D)")
self.DefaultSelAction.setWhatsThis(
"""<b>Default </b>
<p>
Defines the crystal shape as a polygon with the user specifying the
sides
</p>""")
self.CircleSelAction = NE1_QWidgetAction(self.w, win = self.w)
self.CircleSelAction.setObjectName("CIRCLE")
self.CircleSelAction.setText("Circle")
self.subControlAreaActionList.append(self.CircleSelAction)
self.CircleSelAction.setToolTip( "Circle (C)")
self.CircleSelAction.setWhatsThis(
"""<b>Circle </b>
<p>
Draws the crystal geometry as a circle
</p>""")
self.RectCtrSelAction = NE1_QWidgetAction(self.w, win = self.w)
self.RectCtrSelAction.setObjectName("RECTANGLE")
self.RectCtrSelAction.setText("RectCenter")
self.subControlAreaActionList.append(self.RectCtrSelAction)
self.RectCtrSelAction.setToolTip( "Rectangular Center (R)")
self.RectCtrSelAction.setWhatsThis(
"""<b>Rectangle - Center Select</b>
<p>
Draws the crystal geometry as a rectangle with the cursor defining
the center of the rectangle
</p>""")
self.HexagonSelAction = NE1_QWidgetAction(self.w, win = self.w)
self.HexagonSelAction.setObjectName("HEXAGON")
self.HexagonSelAction.setText("Hexagon")
self.subControlAreaActionList.append(self.HexagonSelAction)
self.HexagonSelAction.setToolTip( "Hexagon (H)")
self.HexagonSelAction.setWhatsThis(
"""<b>Hexagon </b>
<p>
Draws the crystal geometry as a hexagon
</p>""")
self.TriangleSelAction = NE1_QWidgetAction(self.w, win = self.w)
self.TriangleSelAction.setObjectName("TRIANGLE")
self.TriangleSelAction.setText("Triangle")
self.subControlAreaActionList.append(self.TriangleSelAction)
self.TriangleSelAction.setToolTip( "Triangle (T)")
self.TriangleSelAction.setWhatsThis(
"""<b>Triangle </b>
<p>
Draws the crystal geometry as a triangle
</p>""")
self.RectCornerSelAction = NE1_QWidgetAction(self.w, win = self.w)
self.RectCornerSelAction.setObjectName("RECT_CORNER")
self.RectCornerSelAction.setText("RectCorners")
self.subControlAreaActionList.append(self.RectCornerSelAction)
self.RectCornerSelAction.setToolTip( "Rectangular Corner (Shift+R)")
self.RectCornerSelAction.setWhatsThis(
"""<b>Rectangle - Corner Select</b>
<p>
Draws the crystal geometry as a rectangle with the cursor defining
the initial corner
</p>""")
self.LassoSelAction = NE1_QWidgetAction(self.w, win = self.w)
self.LassoSelAction.setObjectName("LASSO")
self.LassoSelAction.setText("Lasso")
self.subControlAreaActionList.append(self.LassoSelAction)
self.LassoSelAction.setToolTip( "Lasso (L)")
self.LassoSelAction.setWhatsThis(
"""<b>Lasso</b>
<p>
Can be used to draw irregular crystal geometries
</p>""")
self.DiamondSelAction = NE1_QWidgetAction(self.w, win = self.w)
self.DiamondSelAction.setObjectName("DIAMOND")
self.DiamondSelAction.setText("Diamond")
self.subControlAreaActionList.append(self.DiamondSelAction)
self.DiamondSelAction.setToolTip( "Diamond (D)")
self.DiamondSelAction.setWhatsThis(
"""<b>Diamond</b>
<p>
Draws the crystal geometry as a diamond
</p>""")
self.SquareSelAction = NE1_QWidgetAction(self.w, win = self.w)
self.SquareSelAction.setObjectName("SQUARE")
self.SquareSelAction.setText("Square")
self.subControlAreaActionList.append(self.SquareSelAction)
self.SquareSelAction.setToolTip( "Square(S)")
self.SquareSelAction.setWhatsThis(
"""<b>Square</b>
<p>
Draws the crystal geometry as a square
</p>""")
for action in self.subControlAreaActionList[1:]:
if isinstance(action, QtGui.QWidgetAction):
action.setCheckable(True)
self.cookieSelectionGroup.addAction(action)
iconpath = "ui/actions/Toolbars/Smart/" + str(action.text()) + ".png"
action.setIcon(geticon(iconpath))
if not self.cookieSelectionGroup.checkedAction():
self.DefaultSelAction.setChecked(True)
def getFlyoutActionList(self):
"""
Returns a tuple that contains mode-specific actionlists in the
added in the flyout toolbar of the mode.
CommandToolbar._createFlyoutToolBar method calls this
@return: params: A tuple that contains 3 lists:
(subControlAreaActionList, commandActionLists, allActionsList).
"""
#'allActionsList' returns all actions in the flyout toolbar
#including the subcontrolArea actions
allActionsList = []
#Action List for subcontrol Area buttons.
#In cookie cutter, there is really no subcontrol area.
#We will treat subcontrol area same as 'command area'
#(subcontrol area buttons will have an empty list as their command area
#list). We will set the Comamnd Area palette background color to the
#subcontrol area.
subControlAreaActionList = self.subControlAreaActionList
allActionsList.extend(subControlAreaActionList)
#Empty actionlist for the 'Command Area'
commandActionLists = []
#Append empty 'lists' in 'commandActionLists equal to the
#number of actions in subControlArea
for i in range(len(subControlAreaActionList)):
lst = []
commandActionLists.append(lst)
#The subcontrol area buuton and its command list form a 'key:value pair
#in a python dictionary object
flyoutDictionary = {}
counter = 0
for k in subControlAreaActionList:
# counter is used to sort the keys in the order in which they
#were added
key = (counter, k)
flyoutDictionary[key] = commandActionLists[counter]
#Also add command actions to the 'allActionsList'
allActionsList.extend(commandActionLists[counter])
counter +=1
params = (subControlAreaActionList, commandActionLists, allActionsList)
return params
def updateCommandToolbar(self, bool_entering = True):
"""
Update the Command Toolbar.
"""
if bool_entering:
action = self.w.toolsCookieCutAction
else:
action = None
# object that needs its own flyout toolbar. In this case it is just
# the mode itself.
# [later, bruce 071009: self is now a PM -- is that what this should
# now say, instead of 'mode'?]
obj = self
self.w.commandToolbar.updateCommandToolbar(action,
obj,
entering = bool_entering)
def _makeConnections(self):
"""Connect signal to slots """
self.connect(self.latticeCBox, SIGNAL("activated ( int )"), self.changeLatticeType)
self.connect(self.orientButtonGroup, SIGNAL("buttonClicked(int)"), self.changeGridOrientation)
self.connect(self.rotGridAntiClockwiseButton, SIGNAL("clicked()"), self.antiRotateView)
self.connect(self.rotGridClockwiseButton, SIGNAL("clicked()"), self.rotateView)
self.connect(self.addLayerButton,SIGNAL("clicked()"), self.addLayer)
self.connect(self.currentLayerComboBox,SIGNAL("activated(int)"), self.changeLayer)
self.connect(self.layerCellsSpinBox,SIGNAL("valueChanged(int)"), self.setThickness)
#self.connect(self.gridColorButton,SIGNAL("clicked()"),self.changeGridColor)
self.connect(self.gridLineCheckBox,SIGNAL("toggled(bool)"),self.showGridLine)
self.connect(self.freeViewCheckBox,SIGNAL("toggled(bool)"),self.setFreeView)
self.connect(self.fullModelCheckBox, SIGNAL("toggled(bool)"),self.toggleFullModel)
self.connect(self.snapGridCheckBox, SIGNAL("toggled(bool)"), self.setGridSnap)
self.connect(self.dispModeComboBox, SIGNAL("activated(const QString &)"),
self.changeDispMode)
self.connect(self.cookieSelectionGroup, SIGNAL("triggered(QAction *)"),
self.changeSelectionShape)
self.connect(self.exitCrystalAction, SIGNAL("triggered()"),
self.w.toolsDone)
def _setAutoShapeAcclKeys(self, on):
"""
If <on>, then set the acceleration keys for autoshape selection
in this command; otherwise, like when exit. set it to empty.
"""
if on:
self.DefaultSelAction.setShortcut('D')
self.CircleSelAction.setShortcut('C')
self.RectCtrSelAction.setShortcut('R')
self.HexagonSelAction.setShortcut('H')
self.TriangleSelAction.setShortcut('T')
self.RectCornerSelAction.setShortcut('SHIFT+R')
self.LassoSelAction.setShortcut('L')
self.DiamondSelAction.setShortcut('SHIFT+D')
self.SquareSelAction.setShortcut('S')
else:
for btn in self.cookieSelectionGroup.actions():
btn.setShortcut('')
def initGui(self):
"""
This is used to initialize GUI items which need
to change every time the command becomes active.
"""
# WARNING: the docstring said "every time when the mode is on"
# and I am only guessing that it meant "every time the command becomes active".
# [bruce 071009]
self.w.toolsCookieCutAction.setChecked(1) # toggle on the Cookie Cutter icon
#always show Exit Crystal button checked. (this implementation may change in future --ninad 070131)
self.exitCrystalAction.setChecked(True)
self.show()
self.updateCommandToolbar(bool_entering = True)
self.latticeCBox.setEnabled(True)
#Set projection to ortho, display them
self.w.setViewOrthoAction.setChecked(True)
self.w.setViewOrthoAction.setEnabled(False)
self.w.setViewPerspecAction.setEnabled(False)
# Other things that have been lost at this point:
# self.layerThicknessLineEdit
self.layerCellsSpinBox.setValue(2)
self.rotateGridByAngleSpinBox.setValue(45)
self.currentLayerComboBox.clear()
self.currentLayerComboBox.addItem("1") #QString(str(len(self.layers[0])))) ? ? ?
self.addLayerButton.setEnabled(False)
# Disable some action items in the main window.
self.w.zoomToAreaAction.setEnabled(0) # Disable "Zoom to Area"
self.w.setViewZoomtoSelectionAction.setEnabled(0) # Disable Zoom to Selection
self.w.viewOrientationAction.setEnabled(0) #Disable Orientation Window
# Temporarily disable the global display style combobox since this
# has its own special "crystal" display styles (i.e. spheres and tubes).
# I decided to leave them enabled since the user might want to see the
# entire model and change the global display style. --Mark 2008-03-16
#self.win.statusBar().dispbarLabel.setEnabled(False)
#self.win.statusBar().globalDisplayStylesComboBox.setEnabled(False)
# Disable these toolbars
self.w.buildToolsToolBar.setEnabled(False)
self.w.simulationToolBar.setEnabled(False)
# Set acceleration keys for auto-selection shape
self._setAutoShapeAcclKeys(True)
def restoreGui(self):
"""
Restore GUI items when exiting from the PM (command).
"""
self.updateCommandToolbar(bool_entering = False)
self.w.toolsCookieCutAction.setChecked(0) # Toggle cookie cutter icon
self.close()
self.w.zoomToAreaAction.setEnabled(1) # Enable "Zoom to Area"
self.w.setViewZoomtoSelectionAction.setEnabled(1) # Enable Zoom to Selection
self.w.viewOrientationAction.setEnabled(1) #Enable Orientation Window
# See note above in initGui() about these. Mark 2008-01-30.
#self.w.panToolAction.setEnabled(1) # Enable "Pan Tool"
#self.w.rotateToolAction.setEnabled(1) # Enable "Rotate Tool"
# Enable these toolbars
self.w.buildToolsToolBar.setEnabled(True)
self.w.simulationToolBar.setEnabled(True)
# Enable all those view options
self.enableViewChanges(True)
# Restore global display style label and combobox.
# (see note above in initGui(). )
#self.win.statusBar().dispbarLabel.setEnabled(False)
#self.win.statusBar().globalDisplayStylesComboBox.setEnabled(False)
# Restore view projection, enable them.
self.w.setViewOrthoAction.setEnabled(True)
self.w.setViewPerspecAction.setEnabled(True)
# Turn off acclerating keys
self._setAutoShapeAcclKeys(False)
def enableViewChanges(self, enableFlag):
"""Turn on or off view changes depending on <param> 'enableFlag'.
Turn off view changes is needed during the cookie-cutting stage. """
for c in self.orientButtonGroup.buttons():
c.setEnabled(enableFlag)
self.rotateGridByAngleSpinBox.setEnabled(enableFlag)
self.rotGridAntiClockwiseButton.setEnabled(enableFlag)
self.rotGridClockwiseButton.setEnabled(enableFlag)
self.w.enableViews(enableFlag) # Mark 051122.
def changeSelectionShape(self, action):
"""Slot method that is called when user changes selection shape by GUI. """
cookieCommand = self.cookieCommand
if not cookieCommand.isCurrentCommand(): # [bruce 071008]
return
sShape = action.objectName()
cookieCommand.changeSelectionShape(sShape)
return
def getSelectionShape(self):
"""Return the current selection shape that is checked. """
selectionShape = self.cookieSelectionGroup.checkedAction().objectName()
return selectionShape
def setThickness(self, value):
self.cookieCommand.setThickness(value)
def addLayer(self):
self.addLayerButton.setEnabled(False)
layerId = self.cookieCommand.addLayer()
self.currentLayerComboBox.addItem(QString(str(layerId)))
self.currentLayerComboBox.setCurrentIndex(layerId-1)
self.w.glpane.gl_update()
def changeLayer(self, value):
"""Change current layer to <value> layer """
self.cookieCommand.change2Layer(value)
def setFreeView(self, freeView):
"""Slot function to switch between free view/cookie selection states """
self.cookieCommand.setFreeView(freeView)
def toggleFullModel(self, showFullModel):
"""Slot function for the check box of 'Full Model' in cookie-cutter dashboard """
self.cookieCommand.toggleFullModel(showFullModel)
def showGridLine(self, show):
"""Slot function"""
self.cookieCommand.showGridLine(show)
def setGridSnap(self, snap):
"""Turn on/off the grid snap option """
self.cookieCommand.gridSnap = snap
pass
def changeGridColor(self):
"""Open the stand color chooser dialog to change grid line color """
c = QColorDialog.getColor(QColor(222,148,0), self)
if c.isValid():
self.gridColorLabel.setPaletteBackgroundColor(c)
self.cookieCommand.setGridLineColor(c)
def changeLatticeType(self, lType):
self.cookieCommand.changeLatticeType(lType)
if lType != 0: #Changes to other lattice type
#Disable the snap to grid feature
self.setGridSnap(False)
self.snapGridCheckBox.setEnabled(False)
else:
self.snapGridCheckBox.setEnabled(True)
def changeDispMode(self, display_style):
self.cookieCommand.changeDispMode(display_style)
def changeGridOrientation(self, value):
if value == 0: self._orient100()
elif value == 1: self._orient110()
elif value == 2: self._orient111()
def _rotView(self, direction):
"""Rotate the view anti-clockwise or clockWise.
If <direction> == True, anti-clockwise rotate, otherwise,
clockwise rotate"""
from math import pi
from geometry.VQT import Q, V
angle = self.rotateGridByAngleSpinBox.value()
if not direction: angle = -angle
angle = pi * angle/180.0
glpane = self.w.glpane
glpane.quat += Q(V(0, 0, 1), angle)
glpane.gl_update()
def antiRotateView(self):
"""Anti-clockwise rotatation """
self._rotView(True)
def rotateView(self):
"""clock-wise rotation """
self._rotView(False)
def _orient100(self):
""" Along one axis """
self.cookieCommand.setOrientSurf(0)
self.w.glpane.snapquat100()
def _orient110(self):
"""halfway between two axes"""
self.cookieCommand.setOrientSurf(1)
self.w.glpane.snapquat110()
def _orient111(self):
"""equidistant from three axes """
self.cookieCommand.setOrientSurf(2)
self.w.glpane.snapquat111()
class ProteinFlyout(Ui_AbstractFlyout):
def _action_in_controlArea_to_show_this_flyout(self):
"""
Required action in the 'Control Area' as a reference for this
flyout toolbar. See superclass method for documentation and todo note.
"""
return self.win.buildProteinAction
def _getExitActionText(self):
"""
Overrides superclass method.
@see: self._createActions()
"""
return "Exit Protein"
def getFlyoutActionList(self):
"""
Returns a tuple that contains mode spcific actionlists in the
added in the flyout toolbar of the mode.
CommandToolbar._createFlyoutToolBar method calls this
@return: params: A tuple that contains 3 lists:
(subControlAreaActionList, commandActionLists, allActionsList)
"""
allActionsList = []
subControlAreaActionList =[]
#Append subcontrol area actions to subControlAreaActionList
#The 'Exit button' althought in the subcontrol area, would
#look as if its in the Control area because of the different color
#palette
#and the separator after it. This is intentional.
subControlAreaActionList.append(self.exitModeAction)
separator1 = QAction(self.win)
separator1.setSeparator(True)
subControlAreaActionList.append(separator1)
subControlAreaActionList.append(self.modelProteinAction)
subControlAreaActionList.append(self.simulateProteinAction)
separator = QAction(self.win)
separator.setSeparator(True)
subControlAreaActionList.append(separator)
#Also add the subcontrol Area actions to the 'allActionsList'
for a in subControlAreaActionList:
allActionsList.append(a)
commandActionLists = []
#Append empty 'lists' in 'commandActionLists equal to the
#number of actions in subControlArea
for i in range(len(subControlAreaActionList)):
lst = []
commandActionLists.append(lst)
#Command list for the subcontrol area button 'Model Proteins Tool'
modelProteinsCmdLst = []
modelProteinsCmdLst.append(self.buildPeptideAction)
modelProteinsCmdLst.append(self.compareProteinsAction)
modelProteinsCmdLst.append(self.displayProteinStyleAction)
commandActionLists[2].extend(modelProteinsCmdLst)
# Command list for the subcontrol area button 'Simulate Proteins Tool'
if self.rosetta_enabled:
simulateProteinsCmdLst = []
simulateProteinsCmdLst.append(self.rosetta_fixedbb_design_Action)
simulateProteinsCmdLst.append(self.rosetta_backrub_Action)
simulateProteinsCmdLst.append(self.editResiduesAction)
simulateProteinsCmdLst.append(self.rosetta_score_Action)
commandActionLists[3].extend(simulateProteinsCmdLst)
else:
modelProteinsCmdLst = []
modelProteinsCmdLst.append(self.buildPeptideAction)
modelProteinsCmdLst.append(self.compareProteinsAction)
modelProteinsCmdLst.append(self.displayProteinStyleAction)
commandActionLists[0].extend(modelProteinsCmdLst)
params = (subControlAreaActionList, commandActionLists, allActionsList)
return params
def _createActions(self, parentWidget):
"""
Define flyout toolbar actions for this mode.
"""
_superclass._createActions(self, parentWidget)
#Urmi 20080814: show this flyout toolbar only when rosetta plugin path
# is there
#Probably only rosetta_enabled_prefs_key check would have sufficed
plugin_name = "ROSETTA"
plugin_prefs_keys = (rosetta_enabled_prefs_key, rosetta_path_prefs_key)
errorcode, errortext_or_path = \
checkPluginPreferences(plugin_name, plugin_prefs_keys, ask_for_help = False)
#print "Error code =", errorcode, errortext_or_path
if errorcode == 0:
self.rosetta_enabled = True
else:
self.rosetta_enabled = False
# The subControlActionGroup is the parent of all flyout QActions.
self.subControlActionGroup = QActionGroup(parentWidget)
# Shouldn't the parent be self.win?
# Isn't parentWidget = BuildProteinPropertyManager?
# Ask Bruce about this. --Mark 2008-12-07.
self.subControlActionGroup.setExclusive(True)
self.modelProteinAction = \
NE1_QWidgetAction(self.subControlActionGroup, win = self.win)
self.modelProteinAction.setText("Model")
self.modelProteinAction.setIcon(geticon(
'ui/actions/Command Toolbar/BuildProtein/ModelProtein.png'))
self.modelProteinAction.setCheckable(True)
self.modelProteinAction.setChecked(True)
self.modelProteinAction.setObjectName('ACTION_MODEL_PROTEINS')
self.simulateProteinAction = \
NE1_QWidgetAction(self.subControlActionGroup, win = self.win)
self.simulateProteinAction.setText("Simulate")
self.simulateProteinAction.setIcon(geticon(
"ui/actions/Command Toolbar/BuildProtein/Simulate.png"))
self.simulateProteinAction.setCheckable(True)
self.simulateProteinAction.setObjectName('ACTION_SIMULATE_PROTEINS')
self.subControlActionGroup.addAction(self.modelProteinAction)
self.subControlActionGroup.addAction(self.simulateProteinAction)
self._createModelProteinsActions(parentWidget)
self._createSimulateProteinsActions(parentWidget)
#else:
# self._createModelProteinsActions(parentWidget)
#return
def _createModelProteinsActions(self, parentWidget):
"""
Create the actions to be included in flyout toolbar, when the 'Model
Proteins' is active.
@see: self._createActions() where this is called.
"""
self.subControlActionGroupForModelProtein = QActionGroup(parentWidget)
# Shouldn't the parent be self.win?
# Isn't parentWidget = BuildProteinPropertyManager?
# Ask Bruce about this. --Mark 2008-12-07.
self.subControlActionGroupForModelProtein.setExclusive(True)
self.buildPeptideAction = \
NE1_QWidgetAction(self.subControlActionGroupForModelProtein, win = self.win)
self.buildPeptideAction.setText("Insert Peptide")
self.buildPeptideAction.setCheckable(True)
self.buildPeptideAction.setIcon(
geticon("ui/actions/Command Toolbar/BuildProtein/InsertPeptide.png"))
self.compareProteinsAction = \
NE1_QWidgetAction(self.subControlActionGroupForModelProtein, win = self.win)
self.compareProteinsAction.setText("Compare")
self.compareProteinsAction.setCheckable(True)
self.compareProteinsAction.setIcon(
geticon("ui/actions/Command Toolbar/BuildProtein/Compare.png"))
self.displayProteinStyleAction = \
NE1_QWidgetAction(self.subControlActionGroupForModelProtein, win = self.win)
self.displayProteinStyleAction.setText("Edit Style")
self.displayProteinStyleAction.setCheckable(True)
self.displayProteinStyleAction.setIcon(
geticon("ui/actions/Command Toolbar/BuildProtein/EditProteinDisplayStyle.png"))
self.subControlActionGroupForModelProtein.addAction(self.buildPeptideAction)
self.subControlActionGroupForModelProtein.addAction(self.displayProteinStyleAction)
self.subControlActionGroupForModelProtein.addAction(self.compareProteinsAction)
return
def _createSimulateProteinsActions(self, parentWidget):
"""
Create the actions to be included in flyout toolbar, when the 'Simulate
Proteins' is active.
@see: self._createActions() where this is called.
"""
self.subControlActionGroupForSimulateProtein = QActionGroup(parentWidget)
# Shouldn't the parent be self.win?
# Isn't parentWidget = BuildProteinPropertyManager?
# Ask Bruce about this. --Mark 2008-12-07.
self.subControlActionGroupForSimulateProtein.setExclusive(True)
self.rosetta_fixedbb_design_Action = \
NE1_QWidgetAction(self.subControlActionGroupForSimulateProtein, win = self.win)
self.rosetta_fixedbb_design_Action.setText("Fixed BB")
self.rosetta_fixedbb_design_Action.setCheckable(True)
self.rosetta_fixedbb_design_Action.setIcon(
geticon("ui/actions/Command Toolbar/BuildProtein/FixedBackbone.png"))
self.rosetta_backrub_Action = \
NE1_QWidgetAction(self.subControlActionGroupForSimulateProtein, win = self.win)
self.rosetta_backrub_Action.setText("Backrub")
self.rosetta_backrub_Action.setCheckable(True)
self.rosetta_backrub_Action.setIcon(
geticon("ui/actions/Command Toolbar/BuildProtein/Backrub.png"))
self.editResiduesAction = \
NE1_QWidgetAction(self.subControlActionGroupForSimulateProtein, win = self.win)
self.editResiduesAction.setText("Residues")
self.editResiduesAction.setCheckable(True)
self.editResiduesAction.setIcon(
geticon("ui/actions/Command Toolbar/BuildProtein/Residues.png"))
self.rosetta_score_Action = \
NE1_QWidgetAction(self.subControlActionGroupForSimulateProtein, win = self.win)
self.rosetta_score_Action.setText("Score")
self.rosetta_score_Action.setCheckable(True)
self.rosetta_score_Action.setIcon(
geticon("ui/actions/Command Toolbar/BuildProtein/Score.png"))
self.subControlActionGroupForSimulateProtein.addAction(self.rosetta_fixedbb_design_Action)
self.subControlActionGroupForSimulateProtein.addAction(self.rosetta_backrub_Action)
self.subControlActionGroupForSimulateProtein.addAction(self.editResiduesAction)
self.subControlActionGroupForSimulateProtein.addAction(self.rosetta_score_Action)
return
def _addWhatsThisText(self):
"""
Add 'What's This' help text for all actions on toolbar.
"""
from ne1_ui.WhatsThisText_for_CommandToolbars import whatsThisTextForProteinCommandToolbar
whatsThisTextForProteinCommandToolbar(self)
return
def _addToolTipText(self):
"""
Add 'Tool tip' help text for all actions on toolbar.
"""
from ne1_ui.ToolTipText_for_CommandToolbars import toolTipTextForProteinCommandToolbar
toolTipTextForProteinCommandToolbar(self)
return
def connect_or_disconnect_signals(self, isConnect):
"""
Connect or disconnect widget signals sent to their slot methods.
This can be overridden in subclasses. By default it does nothing.
@param isConnect: If True the widget will send the signals to the slot
method.
@type isConnect: boolean
@see: self.activateFlyoutToolbar, self.deActivateFlyoutToolbar
"""
if isConnect:
change_connect = self.win.connect
else:
change_connect = self.win.disconnect
#Ui_AbstractFlyout connects the self.exitmodeAction, so call it first.
_superclass.connect_or_disconnect_signals(self, isConnect)
change_connect(self.modelProteinAction,
SIGNAL("triggered()"),self._activateModelProteins)
change_connect(self.subControlActionGroup,
SIGNAL("triggered(QAction *)"),
self.testRosettaAndUpdateCommandToolbar)
change_connect(self.simulateProteinAction,
SIGNAL("triggered()"),
self._activateSimulateProteins)
change_connect(self.buildPeptideAction,
SIGNAL("triggered(bool)"),
self.activateInsertPeptide_EditCommand)
change_connect(self.editResiduesAction,
SIGNAL("triggered(bool)"),
self.activateEditResidues_EditCommand)
change_connect(self.compareProteinsAction,
SIGNAL("triggered(bool)"),
self.activateCompareProteins_EditCommand)
change_connect(self.displayProteinStyleAction,
SIGNAL("triggered(bool)"),
self.activateProteinDisplayStyle_Command)
change_connect(self.rosetta_fixedbb_design_Action,
SIGNAL("triggered(bool)"),
self.activateRosettaFixedBBDesign_Command)
change_connect(self.rosetta_score_Action,
SIGNAL("triggered(bool)"),
self.activateRosettaScore_Command)
change_connect(self.rosetta_backrub_Action,
SIGNAL("triggered(bool)"),
self.activateRosettaBackrub_Command)
return
def activateRosettaBackrub_Command(self, isChecked):
"""
Activate Rosetta sequence design with backrub motion command
"""
self.win.enterOrExitTemporaryCommand('BACKRUB_PROTEIN_SEQUENCE_DESIGN')
for action in self.subControlActionGroupForSimulateProtein.actions():
if action is not self.rosetta_backrub_Action and action.isChecked():
action.setChecked(False)
return
def activateRosettaScore_Command(self, isChecked):
"""
Score the current protein sequence
"""
proteinChunk = self.win.assy.getSelectedProteinChunk()
if not proteinChunk:
msg = "You must select a single protein to run a Rosetta <i>Score</i> calculation."
self.updateMessage(msg)
return
otherOptionsText = ""
numSim = 1
argList = [numSim, otherOptionsText, proteinChunk.name]
from simulation.ROSETTA.rosetta_commandruns import rosettaSetup_CommandRun
if argList[0] > 0:
cmdrun = rosettaSetup_CommandRun(self.win, argList, "ROSETTA_SCORE")
cmdrun.run()
for action in self.subControlActionGroupForSimulateProtein.actions():
if action is not self.rosetta_score_Action and action.isChecked():
action.setChecked(False)
return
def activateRosettaFixedBBDesign_Command(self, isChecked):
"""
Activate fixed backbone rosetta sequence design mode
"""
self.win.enterOrExitTemporaryCommand('FIXED_BACKBONE_PROTEIN_SEQUENCE_DESIGN')
for action in self.subControlActionGroupForSimulateProtein.actions():
if action is not self.rosetta_fixedbb_design_Action and action.isChecked():
action.setChecked(False)
return
def _activateModelProteins(self):
"""
Activate the model proteins action of the Proteins mode
"""
self.command.setCurrentCommandMode('MODEL_PROTEIN')
self.command.enterModelOrSimulateCommand('MODEL_PROTEIN')
for action in self.subControlActionGroupForModelProtein.actions():
action.setChecked(False)
return
def _activateSimulateProteins(self):
"""
Activate the simulation tool of the Build Protein mode
"""
self.command.setCurrentCommandMode('SIMULATE_PROTEIN')
self.command.enterModelOrSimulateCommand('SIMULATE_PROTEIN')
for action in self.subControlActionGroupForSimulateProtein.actions():
action.setChecked(False)
return
def activateInsertPeptide_EditCommand(self, isChecked):
"""
Slot for inserting a peptide action.
"""
self.win.insertPeptide(isChecked)
for action in self.subControlActionGroupForModelProtein.actions():
if action is not self.buildPeptideAction and action.isChecked():
action.setChecked(False)
def activateEditResidues_EditCommand(self, isChecked):
"""
Slot for B{EditResidues} action.
"""
self.win.enterEditResiduesCommand(isChecked)
for action in self.subControlActionGroupForSimulateProtein.actions():
if action is not self.editResiduesAction and action.isChecked():
action.setChecked(False)
def activateCompareProteins_EditCommand(self, isChecked):
"""
Slot for B{CompareProteins} action.
"""
self.win.enterCompareProteinsCommand(isChecked)
for action in self.subControlActionGroupForModelProtein.actions():
if action is not self.compareProteinsAction and action.isChecked():
action.setChecked(False)
def activateProteinDisplayStyle_Command(self, isChecked):
"""
Call the method that enters DisplayStyle_Command.
(After entering the command) Also make sure that
all the other actions on the DnaFlyout toolbar are unchecked AND
the DisplayStyle Action is checked.
"""
self.win.enterProteinDisplayStyleCommand(isChecked)
#Uncheck all the actions except the (DNA) display style action
#in the flyout toolbar (subcontrol area)
for action in self.subControlActionGroupForModelProtein.actions():
if action is not self.displayProteinStyleAction and action.isChecked():
action.setChecked(False)
def resetStateOfActions(self):
"""
See superclass for documentation.
@see: self.deActivateFlyoutToolbar()
"""
# Uncheck all the actions in the 'command area' , corresponding to the
#'subControlArea button checked.
#Note :the attrs self.subControlActionGroupFor* are misnamed. Those
#should be somethin like commandAreaActionGroup*
# [-- Ninad 2008-09-17 revised this method for USE_CMMAND_STACK]
actionGroup = None
current_active_tool = self.command.getCurrentActiveTool()
if current_active_tool == 'MODEL_PROTEIN':
actionGroup = self.subControlActionGroupForModelProtein
elif current_active_tool == 'SIMULATE_PROTEIN':
actionGroup = self.subControlActionGroupForSimulateProtein
if actionGroup is None:
return
for action in actionGroup.actions():
action.setChecked(False)
return
def testRosettaAndUpdateCommandToolbar(self, action):
if action == self.simulateProteinAction:
if not self.rosetta_enabled:
#print "Rosetta is not enabled"
plugin_name = "ROSETTA"
plugin_prefs_keys = (rosetta_enabled_prefs_key, rosetta_path_prefs_key)
errorcode, errortext_or_path = \
checkPluginPreferences(plugin_name, plugin_prefs_keys, ask_for_help = True)
#print "Error code =", errorcode, errortext_or_path
if errorcode == 0:
self.rosetta_enabled = True
else:
self.rosetta_enabled = False
self.modelProteinAction.setChecked(True)
self.simulateProteinAction.setChecked(False)
return
self.updateCommandToolbar()
class BuildAtomsFlyout(Ui_AbstractFlyout):
def _action_in_controlArea_to_show_this_flyout(self):
"""
Required action in the 'Control Area' as a reference for this
flyout toolbar. See superclass method for documentation and todo note.
"""
return self.win.toolsDepositAtomAction
def _getExitActionText(self):
"""
Overrides superclass method.
@see: self._createActions()
"""
return "Exit Atoms"
def getFlyoutActionList(self):
"""
Returns a tuple that contains mode spcific actionlists in the
added in the flyout toolbar of the mode.
CommandToolbar._createFlyoutToolBar method calls this
@return: params: A tuple that contains 3 lists:
(subControlAreaActionList, commandActionLists, allActionsList)
"""
#ninad070330 This implementation may change in future.
#'allActionsList' returns all actions in the flyout toolbar
#including the subcontrolArea actions.
allActionsList = []
#Action List for subcontrol Area buttons.
#For now this list is just used to set a different palette to thisaction
#buttons in the toolbar
subControlAreaActionList = []
#Append subcontrol area actions to subControlAreaActionList
#The 'Exit button' althought in the subcontrol area, would
#look as if its in the Control area because of the different color
#palette
#and the separator after it. This is intentional.
subControlAreaActionList.append(self.exitModeAction)
separator1 = QAction(self.win)
separator1.setSeparator(True)
subControlAreaActionList.append(separator1)
subControlAreaActionList.append(self.atomsToolAction)
subControlAreaActionList.append(self.bondsToolAction)
separator = QAction(self.win)
separator.setSeparator(True)
subControlAreaActionList.append(separator)
#Also add the subcontrol Area actions to the 'allActionsList'
for a in subControlAreaActionList:
allActionsList.append(a)
##actionlist.append(self.win.modifyAdjustSelAction)
##actionlist.append(self.win.modifyAdjustAllAction)
#Ninad 070330:
#Command Actions : These are the actual actions that will respond the
#a button pressed in the 'subControlArea (if present).
#Each Subcontrol area button can have its own 'command list'
#The subcontrol area buuton and its command list form a 'key:value pair
#in a python dictionary object
#In Build mode, as of 070330 we have 3 subcontrol area buttons
#(or 'actions)'
commandActionLists = []
#Append empty 'lists' in 'commandActionLists equal to the
#number of actions in subControlArea
for i in range(len(subControlAreaActionList)):
lst = []
commandActionLists.append(lst)
#Command list for the subcontrol area button 'Atoms Tool'
#For others, this list will remain empty for now -- ninad070330
depositAtomsCmdLst = []
depositAtomsCmdLst.append(self.win.modifyHydrogenateAction)
depositAtomsCmdLst.append(self.win.modifyDehydrogenateAction)
depositAtomsCmdLst.append(self.win.modifyPassivateAction)
separatorAfterPassivate = QAction(self.win)
separatorAfterPassivate.setSeparator(True)
depositAtomsCmdLst.append(separatorAfterPassivate)
depositAtomsCmdLst.append(self.transmuteAtomsAction)
separatorAfterTransmute = QAction(self.win)
separatorAfterTransmute.setSeparator(True)
depositAtomsCmdLst.append(separatorAfterTransmute)
depositAtomsCmdLst.append(self.win.modifyDeleteBondsAction)
depositAtomsCmdLst.append(self.win.modifySeparateAction)
depositAtomsCmdLst.append(self.win.makeChunkFromSelectedAtomsAction)
commandActionLists[2].extend(depositAtomsCmdLst)
##for action in self.win.buildToolsMenu.actions():
##commandActionLists[2].append(action)
# Command list for the subcontrol area button 'Bonds Tool'
# For others, this list will remain empty for now -- ninad070405
bondsToolCmdLst = []
bondsToolCmdLst.append(self.bond1Action)
bondsToolCmdLst.append(self.bond2Action)
bondsToolCmdLst.append(self.bond3Action)
bondsToolCmdLst.append(self.bondaAction)
bondsToolCmdLst.append(self.bondgAction)
bondsToolCmdLst.append(self.cutBondsAction)
commandActionLists[3].extend(bondsToolCmdLst)
params = (subControlAreaActionList, commandActionLists, allActionsList)
return params
def _createActions(self, parentWidget):
"""
Define flyout toolbar actions for this mode.
"""
#@NOTE: In Build mode, some of the actions defined in this method are also
#used in Build Atoms PM. (e.g. bond actions) So probably better to rename
#it as _init_modeActions. Not doing that change in mmkit code cleanup
#commit(other modes still implement a method by same name)-ninad20070717
_superclass._createActions(self, parentWidget)
# The subControlActionGroup is the parent of all flyout QActions.
self.subControlActionGroup = QActionGroup(parentWidget)
# Shouldn't the parent be self.win?
# Isn't parentWidget = BuildAtomsPropertyManager.
# Ask Bruce about this. --Mark 2008-12-07.
self.subControlActionGroup.setExclusive(True)
#Following Actions are added in the Flyout toolbar.
#Defining them outside that method as those are being used
#by the subclasses of deposit mode (testmode.py as of 070410) -- ninad
self.atomsToolAction = \
NE1_QWidgetAction(self.subControlActionGroup, win = self.win)
self.atomsToolAction.setText("Atoms Tool")
self.atomsToolAction.setIcon(
geticon("ui/actions/Command Toolbar/BuildAtoms/AtomsTool.png"))
self.atomsToolAction.setCheckable(True)
self.atomsToolAction.setChecked(True)
self.atomsToolAction.setObjectName('ACTION_ATOMS_TOOL')
self.bondsToolAction = \
NE1_QWidgetAction(self.subControlActionGroup, win = self.win)
self.bondsToolAction.setText("Bonds Tool")
self.bondsToolAction.setIcon(
geticon("ui/actions/Command Toolbar/BuildAtoms/BondsTool.png"))
self.bondsToolAction.setCheckable(True)
self.bondsToolAction.setObjectName('ACTION_BOND_TOOL')
self.subControlActionGroup.addAction(self.atomsToolAction)
self.subControlActionGroup.addAction(self.bondsToolAction)
self.transmuteAtomsAction = \
NE1_QWidgetAction(self.subControlActionGroup, win = self.win)
self.transmuteAtomsAction.setText("Transmute Atoms")
self.transmuteAtomsAction.setIcon(
geticon(
"ui/actions/Command Toolbar/BuildAtoms/TransmuteAtoms.png"))
self.transmuteAtomsAction.setCheckable(False)
self._createBondToolActions(parentWidget)
def _createBondToolActions(self, parentWidget):
"""
Create the actions to be included in flyout toolbar , when the 'bonds
tool' is active. Note that the object names of these action will be
be used to find the Bond Tool type to which each action corresponds to.
@see: self._createActions() where this is called.
"""
self.bondToolsActionGroup = QActionGroup(parentWidget)
self.bondToolsActionGroup.setExclusive(True)
self.bond1Action = \
NE1_QWidgetAction(self.bondToolsActionGroup, win = self.win)
self.bond1Action.setText("Single")
self.bond1Action.setIcon(
geticon("ui/actions/Command Toolbar/BuildAtoms/SingleBond.png"))
self.bond1Action.setObjectName('ACTION_SINGLE_BOND_TOOL')
self.bond2Action = \
NE1_QWidgetAction(self.bondToolsActionGroup, win = self.win)
self.bond2Action.setText("Double")
self.bond2Action.setIcon(
geticon("ui/actions/Command Toolbar/BuildAtoms/DoubleBond.png"))
self.bond2Action.setObjectName('ACTION_DOUBLE_BOND_TOOL')
self.bond3Action = \
NE1_QWidgetAction(self.bondToolsActionGroup, win = self.win)
self.bond3Action.setText("Triple")
self.bond3Action.setIcon(
geticon("ui/actions/Command Toolbar/BuildAtoms/TripleBond.png"))
self.bond3Action.setObjectName('ACTION_TRIPLE_BOND_TOOL')
self.bondaAction = \
NE1_QWidgetAction(self.bondToolsActionGroup, win = self.win)
self.bondaAction.setText("Aromatic")
self.bondaAction.setIcon(
geticon("ui/actions/Command Toolbar/BuildAtoms/AromaticBond.png"))
self.bondaAction.setObjectName('ACTION_AROMATIC_BOND_TOOL')
self.bondgAction = \
NE1_QWidgetAction(self.bondToolsActionGroup, win = self.win)
self.bondgAction.setText("Graphitic")
self.bondgAction.setIcon(
geticon("ui/actions/Command Toolbar/BuildAtoms/GraphiticBond.png"))
self.bondgAction.setObjectName('ACTION_GRAPHITIC_BOND_TOOL')
self.cutBondsAction = \
NE1_QWidgetAction(self.bondToolsActionGroup, win = self.win)
self.cutBondsAction.setText("Cut Bonds")
self.cutBondsAction.setIcon(
geticon("ui/actions/Command Toolbar/BuildAtoms/CutBonds.png"))
self.cutBondsAction.setObjectName('ACTION_DELETE_BOND_TOOL')
for action in [
self.bond1Action, self.bond2Action, self.bond3Action,
self.bondaAction, self.bondgAction, self.cutBondsAction
]:
self.bondToolsActionGroup.addAction(action)
action.setCheckable(True)
return
def _addWhatsThisText(self):
"""
Add 'What's This' help text for all actions on toolbar.
"""
from ne1_ui.WhatsThisText_for_CommandToolbars import whatsThisTextForAtomsCommandToolbar
whatsThisTextForAtomsCommandToolbar(self)
return
def _addToolTipText(self):
"""
Add 'Tool tip' help text for all actions on toolbar.
"""
from ne1_ui.ToolTipText_for_CommandToolbars import toolTipTextForAtomsCommandToolbar
toolTipTextForAtomsCommandToolbar(self)
return
def connect_or_disconnect_signals(self, isConnect):
"""
Connect or disconnect widget signals sent to their slot methods.
This can be overridden in subclasses. By default it does nothing.
@param isConnect: If True the widget will send the signals to the slot
method.
@type isConnect: boolean
@see: self.activateFlyoutToolbar, self.deActivateFlyoutToolbar
"""
if isConnect:
change_connect = self.win.connect
else:
change_connect = self.win.disconnect
#Ui_AbstractFlyout connects the self.exitmodeAction, so call it first.
_superclass.connect_or_disconnect_signals(self, isConnect)
#Atom , Bond Tools Groupbox
change_connect(self.bondToolsActionGroup,
SIGNAL("triggered(QAction *)"),
self.command.changeBondTool)
change_connect(self.transmuteAtomsAction, SIGNAL("triggered()"),
self.command.transmutePressed)
change_connect(self.subControlActionGroup,
SIGNAL("triggered(QAction *)"),
self.updateCommandToolbar)
change_connect(self.bondsToolAction, SIGNAL("triggered()"),
self._activateBondsTool)
change_connect(self.atomsToolAction, SIGNAL("triggered()"),
self._activateAtomsTool)
def get_cursor_id_for_active_tool(self):
"""
Provides a cursor id (int) for updating cursor in graphics mode,
based on the checked action in its flyout toolbar. (i.e. based on the
active tool)
@see: BuildAtoms_GraphicsMode.update_cursor_for_no_MB_selection_filter_disabled
"""
if self.atomsToolAction.isChecked():
cursor_id = 0
elif self.bond1Action.isChecked():
cursor_id = 1
elif self.bond2Action.isChecked():
cursor_id = 2
elif self.bond3Action.isChecked():
cursor_id = 3
elif self.bondaAction.isChecked():
cursor_id = 4
elif self.bondgAction.isChecked():
cursor_id = 5
elif self.cutBondsAction.isChecked():
cursor_id = 6
else:
cursor_id = 0
return cursor_id
def _activateAtomsTool(self):
"""
Activate the atoms tool of the Build Atoms mode
hide only the Atoms Tools groupbox in the Build Atoms Property manager
and show all others the others.
"""
self.command.activateAtomsTool()
def _activateBondsTool(self):
"""
Activate the bond tool of the Build Atoms mode
Show only the Bond Tools groupbox in the Build Atoms Property manager
and hide the others.
@see:self._convert_bonds_bet_selected_atoms()
"""
self.command.activateBondsTool()
def getBondToolActions(self):
return self.bondToolsActionGroup.actions()
def getCheckedBondToolAction(self):
return self.bondToolsActionGroup.checkedAction()
def resetStateOfActions(self):
"""
Default implementation does nothing. Overridden in subclasses.
R Resets the state of actions in the flyout toolbar.
Generally, it unchecks all the actions except the ExitModeAction.
This is called while resuming a command.
Example: if exits is in Insert > Dna command,
the Build > Dna command is resumed. When this happens, program needs to
make sure that the Insert > dna button in the flyout is unchecked.
It is done by using this method.
@see: self.deActivateFlyoutToolbar()
@see: self.activateBreakStrands_Command()
@see: baseCommand.command_update_flyout() which calls this method.
@see: BuildAtoms_Command.command_update_state()
@see: BuildAtoms_Command.activateAtomsTool()
"""
self.atomsToolAction.setChecked(True)
##self.bondsToolAction.setChecked(False)
for action in self.bondToolsActionGroup.actions():
action.setChecked(False)
class CookieCtrlPanel(CookiePropertyManager):
"""
This class provides GUI controls to the cookie-cutter command.
"""
def __init__(self, parentMode):
"""
"""
self.w = parentMode.w
self.cookieCommand = parentMode
#bruce 071008, probably redundant with some other attribute
self.pw = None # pw is active part window
self._init_flyoutActions()
CookiePropertyManager.__init__(self, parentMode)
self._makeConnections()
def _init_flyoutActions(self):
"""
Define flyout toolbar actions for this command
"""
#Create an action group and add all the cookie selection shape buttons to it
self.cookieSelectionGroup = QActionGroup(self.w)
#Action List for subcontrol Area buttons.
#In cookie cutter, there is really no subcontrol area.
#We will treat subcontrol area same as 'command area'
#(subcontrol area buttons will have an empty list as their command area
#list). We will set the Comamnd Area palette background color to the
#subcontrol area. This list will be used in getFlyoutActionList
self.subControlAreaActionList = []
self.exitCrystalAction = NE1_QWidgetAction(self.w, win=self.w)
self.exitCrystalAction.setText("Exit Crystal")
self.exitCrystalAction.setCheckable(True)
self.exitCrystalAction.setChecked(True)
self.exitCrystalAction.setIcon(
geticon('ui/actions/Toolbars/Smart/Exit.png'))
self.subControlAreaActionList.append(self.exitCrystalAction)
separator = QtGui.QAction(self.w)
separator.setSeparator(True)
self.subControlAreaActionList.append(separator)
self.DefaultSelAction = NE1_QWidgetAction(self.w, win=self.w)
self.DefaultSelAction.setObjectName("DEFAULT")
self.DefaultSelAction.setText("Default")
self.subControlAreaActionList.append(self.DefaultSelAction)
self.DefaultSelAction.setToolTip("Default Selection (D)")
self.DefaultSelAction.setWhatsThis("""<b>Default </b>
<p>
Defines the crystal shape as a polygon with the user specifying the
sides
</p>""")
self.CircleSelAction = NE1_QWidgetAction(self.w, win=self.w)
self.CircleSelAction.setObjectName("CIRCLE")
self.CircleSelAction.setText("Circle")
self.subControlAreaActionList.append(self.CircleSelAction)
self.CircleSelAction.setToolTip("Circle (C)")
self.CircleSelAction.setWhatsThis("""<b>Circle </b>
<p>
Draws the crystal geometry as a circle
</p>""")
self.RectCtrSelAction = NE1_QWidgetAction(self.w, win=self.w)
self.RectCtrSelAction.setObjectName("RECTANGLE")
self.RectCtrSelAction.setText("RectCenter")
self.subControlAreaActionList.append(self.RectCtrSelAction)
self.RectCtrSelAction.setToolTip("Rectangular Center (R)")
self.RectCtrSelAction.setWhatsThis("""<b>Rectangle - Center Select</b>
<p>
Draws the crystal geometry as a rectangle with the cursor defining
the center of the rectangle
</p>""")
self.HexagonSelAction = NE1_QWidgetAction(self.w, win=self.w)
self.HexagonSelAction.setObjectName("HEXAGON")
self.HexagonSelAction.setText("Hexagon")
self.subControlAreaActionList.append(self.HexagonSelAction)
self.HexagonSelAction.setToolTip("Hexagon (H)")
self.HexagonSelAction.setWhatsThis("""<b>Hexagon </b>
<p>
Draws the crystal geometry as a hexagon
</p>""")
self.TriangleSelAction = NE1_QWidgetAction(self.w, win=self.w)
self.TriangleSelAction.setObjectName("TRIANGLE")
self.TriangleSelAction.setText("Triangle")
self.subControlAreaActionList.append(self.TriangleSelAction)
self.TriangleSelAction.setToolTip("Triangle (T)")
self.TriangleSelAction.setWhatsThis("""<b>Triangle </b>
<p>
Draws the crystal geometry as a triangle
</p>""")
self.RectCornerSelAction = NE1_QWidgetAction(self.w, win=self.w)
self.RectCornerSelAction.setObjectName("RECT_CORNER")
self.RectCornerSelAction.setText("RectCorners")
self.subControlAreaActionList.append(self.RectCornerSelAction)
self.RectCornerSelAction.setToolTip("Rectangular Corner (Shift+R)")
self.RectCornerSelAction.setWhatsThis(
"""<b>Rectangle - Corner Select</b>
<p>
Draws the crystal geometry as a rectangle with the cursor defining
the initial corner
</p>""")
self.LassoSelAction = NE1_QWidgetAction(self.w, win=self.w)
self.LassoSelAction.setObjectName("LASSO")
self.LassoSelAction.setText("Lasso")
self.subControlAreaActionList.append(self.LassoSelAction)
self.LassoSelAction.setToolTip("Lasso (L)")
self.LassoSelAction.setWhatsThis("""<b>Lasso</b>
<p>
Can be used to draw irregular crystal geometries
</p>""")
self.DiamondSelAction = NE1_QWidgetAction(self.w, win=self.w)
self.DiamondSelAction.setObjectName("DIAMOND")
self.DiamondSelAction.setText("Diamond")
self.subControlAreaActionList.append(self.DiamondSelAction)
self.DiamondSelAction.setToolTip("Diamond (D)")
self.DiamondSelAction.setWhatsThis("""<b>Diamond</b>
<p>
Draws the crystal geometry as a diamond
</p>""")
self.SquareSelAction = NE1_QWidgetAction(self.w, win=self.w)
self.SquareSelAction.setObjectName("SQUARE")
self.SquareSelAction.setText("Square")
self.subControlAreaActionList.append(self.SquareSelAction)
self.SquareSelAction.setToolTip("Square(S)")
self.SquareSelAction.setWhatsThis("""<b>Square</b>
<p>
Draws the crystal geometry as a square
</p>""")
for action in self.subControlAreaActionList[1:]:
if isinstance(action, QtGui.QWidgetAction):
action.setCheckable(True)
self.cookieSelectionGroup.addAction(action)
iconpath = "ui/actions/Toolbars/Smart/" + str(
action.text()) + ".png"
action.setIcon(geticon(iconpath))
if not self.cookieSelectionGroup.checkedAction():
self.DefaultSelAction.setChecked(True)
def getFlyoutActionList(self):
"""
Returns a tuple that contains mode-specific actionlists in the
added in the flyout toolbar of the mode.
CommandToolbar._createFlyoutToolBar method calls this
@return: params: A tuple that contains 3 lists:
(subControlAreaActionList, commandActionLists, allActionsList).
"""
#'allActionsList' returns all actions in the flyout toolbar
#including the subcontrolArea actions
allActionsList = []
#Action List for subcontrol Area buttons.
#In cookie cutter, there is really no subcontrol area.
#We will treat subcontrol area same as 'command area'
#(subcontrol area buttons will have an empty list as their command area
#list). We will set the Comamnd Area palette background color to the
#subcontrol area.
subControlAreaActionList = self.subControlAreaActionList
allActionsList.extend(subControlAreaActionList)
#Empty actionlist for the 'Command Area'
commandActionLists = []
#Append empty 'lists' in 'commandActionLists equal to the
#number of actions in subControlArea
for i in range(len(subControlAreaActionList)):
lst = []
commandActionLists.append(lst)
#The subcontrol area buuton and its command list form a 'key:value pair
#in a python dictionary object
flyoutDictionary = {}
counter = 0
for k in subControlAreaActionList:
# counter is used to sort the keys in the order in which they
#were added
key = (counter, k)
flyoutDictionary[key] = commandActionLists[counter]
#Also add command actions to the 'allActionsList'
allActionsList.extend(commandActionLists[counter])
counter += 1
params = (subControlAreaActionList, commandActionLists, allActionsList)
return params
def updateCommandToolbar(self, bool_entering=True):
"""
Update the Command Toolbar.
"""
if bool_entering:
action = self.w.toolsCookieCutAction
else:
action = None
# object that needs its own flyout toolbar. In this case it is just
# the mode itself.
# [later, bruce 071009: self is now a PM -- is that what this should
# now say, instead of 'mode'?]
obj = self
self.w.commandToolbar.updateCommandToolbar(action,
obj,
entering=bool_entering)
def _makeConnections(self):
"""Connect signal to slots """
self.connect(self.latticeCBox, SIGNAL("activated ( int )"),
self.changeLatticeType)
self.connect(self.orientButtonGroup, SIGNAL("buttonClicked(int)"),
self.changeGridOrientation)
self.connect(self.rotGridAntiClockwiseButton, SIGNAL("clicked()"),
self.antiRotateView)
self.connect(self.rotGridClockwiseButton, SIGNAL("clicked()"),
self.rotateView)
self.connect(self.addLayerButton, SIGNAL("clicked()"), self.addLayer)
self.connect(self.currentLayerComboBox, SIGNAL("activated(int)"),
self.changeLayer)
self.connect(self.layerCellsSpinBox, SIGNAL("valueChanged(int)"),
self.setThickness)
#self.connect(self.gridColorButton,SIGNAL("clicked()"),self.changeGridColor)
self.connect(self.gridLineCheckBox, SIGNAL("toggled(bool)"),
self.showGridLine)
self.connect(self.freeViewCheckBox, SIGNAL("toggled(bool)"),
self.setFreeView)
self.connect(self.fullModelCheckBox, SIGNAL("toggled(bool)"),
self.toggleFullModel)
self.connect(self.snapGridCheckBox, SIGNAL("toggled(bool)"),
self.setGridSnap)
self.connect(self.dispModeComboBox,
SIGNAL("activated(const QString &)"), self.changeDispMode)
self.connect(self.cookieSelectionGroup, SIGNAL("triggered(QAction *)"),
self.changeSelectionShape)
self.connect(self.exitCrystalAction, SIGNAL("triggered()"),
self.w.toolsDone)
def _setAutoShapeAcclKeys(self, on):
"""
If <on>, then set the acceleration keys for autoshape selection
in this command; otherwise, like when exit. set it to empty.
"""
if on:
self.DefaultSelAction.setShortcut('D')
self.CircleSelAction.setShortcut('C')
self.RectCtrSelAction.setShortcut('R')
self.HexagonSelAction.setShortcut('H')
self.TriangleSelAction.setShortcut('T')
self.RectCornerSelAction.setShortcut('SHIFT+R')
self.LassoSelAction.setShortcut('L')
self.DiamondSelAction.setShortcut('SHIFT+D')
self.SquareSelAction.setShortcut('S')
else:
for btn in self.cookieSelectionGroup.actions():
btn.setShortcut('')
def initGui(self):
"""
This is used to initialize GUI items which need
to change every time the command becomes active.
"""
# WARNING: the docstring said "every time when the mode is on"
# and I am only guessing that it meant "every time the command becomes active".
# [bruce 071009]
self.w.toolsCookieCutAction.setChecked(
1) # toggle on the Cookie Cutter icon
#always show Exit Crystal button checked. (this implementation may change in future --ninad 070131)
self.exitCrystalAction.setChecked(True)
self.show()
self.updateCommandToolbar(bool_entering=True)
self.latticeCBox.setEnabled(True)
#Set projection to ortho, display them
self.w.setViewOrthoAction.setChecked(True)
self.w.setViewOrthoAction.setEnabled(False)
self.w.setViewPerspecAction.setEnabled(False)
# Other things that have been lost at this point:
# self.layerThicknessLineEdit
self.layerCellsSpinBox.setValue(2)
self.rotateGridByAngleSpinBox.setValue(45)
self.currentLayerComboBox.clear()
self.currentLayerComboBox.addItem(
"1") #QString(str(len(self.layers[0])))) ? ? ?
self.addLayerButton.setEnabled(False)
# Disable some action items in the main window.
self.w.zoomToAreaAction.setEnabled(0) # Disable "Zoom to Area"
self.w.setViewZoomtoSelectionAction.setEnabled(
0) # Disable Zoom to Selection
self.w.viewOrientationAction.setEnabled(0) #Disable Orientation Window
# Temporarily disable the global display style combobox since this
# has its own special "crystal" display styles (i.e. spheres and tubes).
# I decided to leave them enabled since the user might want to see the
# entire model and change the global display style. --Mark 2008-03-16
#self.win.statusBar().dispbarLabel.setEnabled(False)
#self.win.statusBar().globalDisplayStylesComboBox.setEnabled(False)
# Disable these toolbars
self.w.buildToolsToolBar.setEnabled(False)
self.w.simulationToolBar.setEnabled(False)
# Set acceleration keys for auto-selection shape
self._setAutoShapeAcclKeys(True)
def restoreGui(self):
"""
Restore GUI items when exiting from the PM (command).
"""
self.updateCommandToolbar(bool_entering=False)
self.w.toolsCookieCutAction.setChecked(0) # Toggle cookie cutter icon
self.close()
self.w.zoomToAreaAction.setEnabled(1) # Enable "Zoom to Area"
self.w.setViewZoomtoSelectionAction.setEnabled(
1) # Enable Zoom to Selection
self.w.viewOrientationAction.setEnabled(1) #Enable Orientation Window
# See note above in initGui() about these. Mark 2008-01-30.
#self.w.panToolAction.setEnabled(1) # Enable "Pan Tool"
#self.w.rotateToolAction.setEnabled(1) # Enable "Rotate Tool"
# Enable these toolbars
self.w.buildToolsToolBar.setEnabled(True)
self.w.simulationToolBar.setEnabled(True)
# Enable all those view options
self.enableViewChanges(True)
# Restore global display style label and combobox.
# (see note above in initGui(). )
#self.win.statusBar().dispbarLabel.setEnabled(False)
#self.win.statusBar().globalDisplayStylesComboBox.setEnabled(False)
# Restore view projection, enable them.
self.w.setViewOrthoAction.setEnabled(True)
self.w.setViewPerspecAction.setEnabled(True)
# Turn off acclerating keys
self._setAutoShapeAcclKeys(False)
def enableViewChanges(self, enableFlag):
"""Turn on or off view changes depending on <param> 'enableFlag'.
Turn off view changes is needed during the cookie-cutting stage. """
for c in self.orientButtonGroup.buttons():
c.setEnabled(enableFlag)
self.rotateGridByAngleSpinBox.setEnabled(enableFlag)
self.rotGridAntiClockwiseButton.setEnabled(enableFlag)
self.rotGridClockwiseButton.setEnabled(enableFlag)
self.w.enableViews(enableFlag) # Mark 051122.
def changeSelectionShape(self, action):
"""Slot method that is called when user changes selection shape by GUI. """
cookieCommand = self.cookieCommand
if not cookieCommand.isCurrentCommand(): # [bruce 071008]
return
sShape = action.objectName()
cookieCommand.changeSelectionShape(sShape)
return
def getSelectionShape(self):
"""Return the current selection shape that is checked. """
selectionShape = self.cookieSelectionGroup.checkedAction().objectName()
return selectionShape
def setThickness(self, value):
self.cookieCommand.setThickness(value)
def addLayer(self):
self.addLayerButton.setEnabled(False)
layerId = self.cookieCommand.addLayer()
self.currentLayerComboBox.addItem(QString(str(layerId)))
self.currentLayerComboBox.setCurrentIndex(layerId - 1)
self.w.glpane.gl_update()
def changeLayer(self, value):
"""Change current layer to <value> layer """
self.cookieCommand.change2Layer(value)
def setFreeView(self, freeView):
"""Slot function to switch between free view/cookie selection states """
self.cookieCommand.setFreeView(freeView)
def toggleFullModel(self, showFullModel):
"""Slot function for the check box of 'Full Model' in cookie-cutter dashboard """
self.cookieCommand.toggleFullModel(showFullModel)
def showGridLine(self, show):
"""Slot function"""
self.cookieCommand.showGridLine(show)
def setGridSnap(self, snap):
"""Turn on/off the grid snap option """
self.cookieCommand.gridSnap = snap
pass
def changeGridColor(self):
"""Open the stand color chooser dialog to change grid line color """
c = QColorDialog.getColor(QColor(222, 148, 0), self)
if c.isValid():
self.gridColorLabel.setPaletteBackgroundColor(c)
self.cookieCommand.setGridLineColor(c)
def changeLatticeType(self, lType):
self.cookieCommand.changeLatticeType(lType)
if lType != 0: #Changes to other lattice type
#Disable the snap to grid feature
self.setGridSnap(False)
self.snapGridCheckBox.setEnabled(False)
else:
self.snapGridCheckBox.setEnabled(True)
def changeDispMode(self, display_style):
self.cookieCommand.changeDispMode(display_style)
def changeGridOrientation(self, value):
if value == 0: self._orient100()
elif value == 1: self._orient110()
elif value == 2: self._orient111()
def _rotView(self, direction):
"""Rotate the view anti-clockwise or clockWise.
If <direction> == True, anti-clockwise rotate, otherwise,
clockwise rotate"""
from math import pi
from geometry.VQT import Q, V
angle = self.rotateGridByAngleSpinBox.value()
if not direction: angle = -angle
angle = pi * angle / 180.0
glpane = self.w.glpane
glpane.quat += Q(V(0, 0, 1), angle)
glpane.gl_update()
def antiRotateView(self):
"""Anti-clockwise rotatation """
self._rotView(True)
def rotateView(self):
"""clock-wise rotation """
self._rotView(False)
def _orient100(self):
""" Along one axis """
self.cookieCommand.setOrientSurf(0)
self.w.glpane.snapquat100()
def _orient110(self):
"""halfway between two axes"""
self.cookieCommand.setOrientSurf(1)
self.w.glpane.snapquat110()
def _orient111(self):
"""equidistant from three axes """
self.cookieCommand.setOrientSurf(2)
self.w.glpane.snapquat111()