def __init__(self, script="", parent=None):
super(QTextEditCS, self).__init__(parent)
self.resize(800, 600)
self.script = script
self.edit = QtGui.QTextEdit(self)
# met une police de caractère même largeur pour tous les caractères
font = QtGui.QFont()
font.setFamily(u"DejaVu Sans Mono") # police de Qt4
font.setStyleHint(QtGui.QFont.Courier) # si la police est indisponible
font.setPointSize(10)
self.edit.setFont(font)
# met en place la coloration syntaxique
self.colorSyntax = ColorSyntax(self.edit.document())
# affiche le script colorisé
self.setText(self.script)
# positionne le QTextEdit dans la fenêtre
layout = QHBoxLayout(self)
layout.setMargin(0)
layout.setSpacing(0)
self.number_bar = self.NumberBar()
self.number_bar.setTextEdit(self.edit)
layout.addWidget(self.number_bar)
layout.addWidget(self.edit)
self.setLayout(layout)
self.edit.installEventFilter(self)
self.edit.viewport().installEventFilter(self)
def __init__(self, *args):
QWidget.__init__(self, *args)
lo = QHBoxLayout(self)
lo.setContentsMargins(0, 0, 0, 0)
lo.setSpacing(5)
# type selector
self.wtypesel = QComboBox(self)
for i, tp in enumerate(self.ValueTypes):
self.wtypesel.addItem(tp.__name__)
QObject.connect(self.wtypesel, SIGNAL("activated(int)"), self._selectTypeNum)
typesel_lab = QLabel("&Type:", self)
typesel_lab.setBuddy(self.wtypesel)
lo.addWidget(typesel_lab, 0)
lo.addWidget(self.wtypesel, 0)
self.wvalue = QLineEdit(self)
self.wvalue_lab = QLabel("&Value:", self)
self.wvalue_lab.setBuddy(self.wvalue)
self.wbool = QComboBox(self)
self.wbool.addItems(["false", "true"])
self.wbool.setCurrentIndex(1)
lo.addWidget(self.wvalue_lab, 0)
lo.addWidget(self.wvalue, 1)
lo.addWidget(self.wbool, 1)
self.wvalue.hide()
# make input validators
self._validators = {int: QIntValidator(self), float: QDoubleValidator(self)}
# select bool type initially
self._selectTypeNum(0)
def __init__(self, parent, label, filename=None, dialog_label=None, file_types=None, default_suffix=None,
file_mode=QFileDialog.AnyFile):
QWidget.__init__(self, parent)
lo = QHBoxLayout(self)
lo.setContentsMargins(0, 0, 0, 0)
lo.setSpacing(5)
# label
lab = QLabel(label, self)
lo.addWidget(lab, 0)
# text field
self.wfname = QLineEdit(self)
self.wfname.setReadOnly(True)
self.setFilename(filename)
lo.addWidget(self.wfname, 1)
# selector
wsel = QToolButton(self)
wsel.setText("Choose...")
QObject.connect(wsel, SIGNAL("clicked()"), self._chooseFile)
lo.addWidget(wsel, 0)
# other init
self._file_dialog = None
self._dialog_label = dialog_label or label
self._file_types = file_types or "All files (*)"
self._file_mode = file_mode
self._default_suffix = default_suffix
self._dir = None
def __init__(self, *args):
QWidget.__init__(self, *args)
lo = QHBoxLayout(self)
lo.setContentsMargins(0, 0, 0, 0)
lo.setSpacing(5)
# type selector
self.wtypesel = QComboBox(self)
for i, tp in enumerate(self.ValueTypes):
self.wtypesel.addItem(tp.__name__)
QObject.connect(self.wtypesel, SIGNAL("activated(int)"),
self._selectTypeNum)
typesel_lab = QLabel("&Type:", self)
typesel_lab.setBuddy(self.wtypesel)
lo.addWidget(typesel_lab, 0)
lo.addWidget(self.wtypesel, 0)
self.wvalue = QLineEdit(self)
self.wvalue_lab = QLabel("&Value:", self)
self.wvalue_lab.setBuddy(self.wvalue)
self.wbool = QComboBox(self)
self.wbool.addItems(["false", "true"])
self.wbool.setCurrentIndex(1)
lo.addWidget(self.wvalue_lab, 0)
lo.addWidget(self.wvalue, 1)
lo.addWidget(self.wbool, 1)
self.wvalue.hide()
# make input validators
self._validators = {
int: QIntValidator(self),
float: QDoubleValidator(self)
}
# select bool type initially
self._selectTypeNum(0)
def __init__(self, script="", parent=None):
super(QTextEditCS, self).__init__(parent)
self.resize(800, 600)
self.script = script
self.edit = QtGui.QTextEdit(self)
# met une police de caractère même largeur pour tous les caractères
font = QtGui.QFont()
font.setFamily(u"DejaVu Sans Mono") # police de Qt4
font.setStyleHint(QtGui.QFont.Courier) # si la police est indisponible
font.setPointSize(10)
self.edit.setFont(font)
# met en place la coloration syntaxique
self.colorSyntax = ColorSyntax(self.edit.document())
# affiche le script colorisé
self.setText(self.script)
# positionne le QTextEdit dans la fenêtre
layout = QHBoxLayout(self)
layout.setMargin(0)
layout.setSpacing(0)
self.number_bar = self.NumberBar()
self.number_bar.setTextEdit(self.edit)
layout.addWidget(self.number_bar)
layout.addWidget(self.edit)
self.setLayout(layout)
self.edit.installEventFilter(self)
self.edit.viewport().installEventFilter(self)
def __init__(self,
parent,
label,
filename=None,
dialog_label=None,
file_types=None,
default_suffix=None,
file_mode=QFileDialog.AnyFile):
QWidget.__init__(self, parent)
lo = QHBoxLayout(self)
lo.setContentsMargins(0, 0, 0, 0)
lo.setSpacing(5)
# label
lab = QLabel(label, self)
lo.addWidget(lab, 0)
# text field
self.wfname = QLineEdit(self)
self.wfname.setReadOnly(True)
self.setFilename(filename)
lo.addWidget(self.wfname, 1)
# selector
wsel = QToolButton(self)
wsel.setText("Choose...")
QObject.connect(wsel, SIGNAL("clicked()"), self._chooseFile)
lo.addWidget(wsel, 0)
# other init
self._file_dialog = None
self._dialog_label = dialog_label or label
self._file_types = file_types or "All files (*)"
self._file_mode = file_mode
self._default_suffix = default_suffix
self._dir = None
class CapturePhotoCover(QDialog):
def __init__(self, gui, icon, do_user_config, qaction, us):
QDialog.__init__(self, gui)
self.main_gui = gui
self.do_user_config = do_user_config
self.qaction = qaction
self.us = us
#- main UI layout -----------------------------------------------------
self.ll = QVBoxLayout()
self.l = QHBoxLayout()
self.l.setSpacing(0)
#self.l.setMargin(0) # not in Qt5
self.w = QWidget()
self.w.setLayout(self.l)
self.setLayout(self.ll)
self.setWindowIcon(icon)
self.setSizePolicy(QSizePolicy.Preferred, QSizePolicy.Preferred)
#- webkit -------------------------------------------------------------
self.webview = QWebView()
self.webview.setMaximumWidth(680)
self.webview.setMaximumHeight(320)
self.webview.setSizePolicy(QSizePolicy.Expanding,
QSizePolicy.Expanding)
# self.webview.load(QUrl.fromLocalFile(
# os.path.join(self.us.portable_directory,
# "portable/kaliweb.html")))
self.loadUrl()
self.webview.page().networkAccessManager().sslErrors.connect(
self.sslErrorHandler)
#self.connect(self.webview.page().networkAccessManager(),
# SIGNAL("sslErrors (QNetworkReply *, \
# const QList<QSslError> &)"),
# self.sslErrorHandler)
logger.info("KALIWEB PATH: {}".format(
os.path.join(self.us.portable_directory, "portable/kaliweb.html")))
self.ll.addWidget(self.webview)
def loadUrl(self):
if self.us.initial:
self.webview.page().mainFrame().load(QUrl("https://google.com"))
self.us.initial = False
def keyPressEvent(self, event):
if event.key() == Qt.Key_Escape:
pass
def sslErrorHandler(self, reply, errorList):
reply.ignoreSslErrors()
logger.debug("SSL ERRORS: {}".format(errorList))
return
def closeEvent(self, e):
self.hide()
class CapturePhotoCover(QDialog):
def __init__(self, gui, icon, do_user_config, qaction, us):
QDialog.__init__(self, gui)
self.main_gui = gui
self.do_user_config = do_user_config
self.qaction = qaction
self.us = us
#- main UI layout -----------------------------------------------------
self.ll = QVBoxLayout()
self.l = QHBoxLayout()
self.l.setSpacing(0)
#self.l.setMargin(0) # not in Qt5
self.w = QWidget()
self.w.setLayout(self.l)
self.setLayout(self.ll)
self.setWindowIcon(icon)
self.setSizePolicy(QSizePolicy.Preferred, QSizePolicy.Preferred)
#- webkit -------------------------------------------------------------
self.webview = QWebView()
self.webview.setMaximumWidth(680)
self.webview.setMaximumHeight(320)
self.webview.setSizePolicy(QSizePolicy.Expanding,
QSizePolicy.Expanding)
# self.webview.load(QUrl.fromLocalFile(
# os.path.join(self.us.portable_directory,
# "portable/kaliweb.html")))
self.loadUrl()
self.webview.page().networkAccessManager().sslErrors.connect(self.sslErrorHandler)
#self.connect(self.webview.page().networkAccessManager(),
# SIGNAL("sslErrors (QNetworkReply *, \
# const QList<QSslError> &)"),
# self.sslErrorHandler)
logger.info("KALIWEB PATH: {}".format(
os.path.join(self.us.portable_directory,
"portable/kaliweb.html")))
self.ll.addWidget(self.webview)
def loadUrl(self):
if self.us.initial:
self.webview.page().mainFrame().load(QUrl("https://google.com"))
self.us.initial = False
def keyPressEvent(self, event):
if event.key() == Qt.Key_Escape:
pass
def sslErrorHandler(self, reply, errorList):
reply.ignoreSslErrors()
logger.debug("SSL ERRORS: {}".format(errorList))
return
def closeEvent(self, e):
self.hide()
def _createHeader(self, iconPath, title):
"""
Creates the Property Manager header, which contains an icon
(a QLabel with a pixmap) and white text (a QLabel with text).
@param iconPath: The relative path for the icon (PNG image) that
appears in the header.
@type iconPath: str
@param title: The title that appears in the header.
@type title: str
"""
# Heading frame (dark gray), which contains
# a pixmap and (white) heading text.
self.headerFrame = QFrame(self)
self.headerFrame.setFrameShape(QFrame.NoFrame)
self.headerFrame.setFrameShadow(QFrame.Plain)
self.headerFrame.setPalette(QPalette(pmHeaderFrameColor))
self.headerFrame.setAutoFillBackground(True)
# HBox layout for heading frame, containing the pixmap
# and label (title).
HeaderFrameHLayout = QHBoxLayout(self.headerFrame)
# 2 pixels around edges --
HeaderFrameHLayout.setMargin(PM_HEADER_FRAME_MARGIN)
# 5 pixel between pixmap and label. --
HeaderFrameHLayout.setSpacing(PM_HEADER_FRAME_SPACING)
# PropMgr icon. Set image by calling setHeaderIcon().
self.headerIcon = QLabel(self.headerFrame)
self.headerIcon.setSizePolicy(
QSizePolicy(QSizePolicy.Policy(QSizePolicy.Fixed),
QSizePolicy.Policy(QSizePolicy.Fixed)))
self.headerIcon.setScaledContents(True)
HeaderFrameHLayout.addWidget(self.headerIcon)
# PropMgr header title text (a QLabel).
self.headerTitle = QLabel(self.headerFrame)
headerTitlePalette = self._getHeaderTitlePalette()
self.headerTitle.setPalette(headerTitlePalette)
self.headerTitle.setAlignment(PM_LABEL_LEFT_ALIGNMENT)
# Assign header title font.
self.headerTitle.setFont(self._getHeaderFont())
HeaderFrameHLayout.addWidget(self.headerTitle)
self.vBoxLayout.addWidget(self.headerFrame)
# Set header icon and title text.
self.setHeaderIcon(iconPath)
self.setHeaderTitle(title)
def _createHeader(self, iconPath, title):
"""
Creates the Property Manager header, which contains an icon
(a QLabel with a pixmap) and white text (a QLabel with text).
@param iconPath: The relative path for the icon (PNG image) that
appears in the header.
@type iconPath: str
@param title: The title that appears in the header.
@type title: str
"""
# Heading frame (dark gray), which contains
# a pixmap and (white) heading text.
self.headerFrame = QFrame(self)
self.headerFrame.setFrameShape(QFrame.NoFrame)
self.headerFrame.setFrameShadow(QFrame.Plain)
self.headerFrame.setPalette(QPalette(pmHeaderFrameColor))
self.headerFrame.setAutoFillBackground(True)
# HBox layout for heading frame, containing the pixmap
# and label (title).
HeaderFrameHLayout = QHBoxLayout(self.headerFrame)
# 2 pixels around edges --
HeaderFrameHLayout.setMargin(PM_HEADER_FRAME_MARGIN)
# 5 pixel between pixmap and label. --
HeaderFrameHLayout.setSpacing(PM_HEADER_FRAME_SPACING)
# PropMgr icon. Set image by calling setHeaderIcon().
self.headerIcon = QLabel(self.headerFrame)
self.headerIcon.setSizePolicy(
QSizePolicy(QSizePolicy.Policy(QSizePolicy.Fixed),
QSizePolicy.Policy(QSizePolicy.Fixed)))
self.headerIcon.setScaledContents(True)
HeaderFrameHLayout.addWidget(self.headerIcon)
# PropMgr header title text (a QLabel).
self.headerTitle = QLabel(self.headerFrame)
headerTitlePalette = self._getHeaderTitlePalette()
self.headerTitle.setPalette(headerTitlePalette)
self.headerTitle.setAlignment(PM_LABEL_LEFT_ALIGNMENT)
# Assign header title font.
self.headerTitle.setFont(self._getHeaderFont())
HeaderFrameHLayout.addWidget(self.headerTitle)
self.vBoxLayout.addWidget(self.headerFrame)
# Set header icon and title text.
self.setHeaderIcon(iconPath)
self.setHeaderTitle(title)
def __init__(self, parent):
QWidget.__init__(self, parent)
layout = QHBoxLayout()
layout.setSpacing(5)
layout.setContentsMargins(0, 0, 0, 0)
self.tags_box = EditWithComplete(parent)
layout.addWidget(self.tags_box, stretch=1000)
self.editor_button = QToolButton(self)
self.editor_button.setToolTip(_('Open Item Editor'))
self.editor_button.setIcon(QIcon(I('chapters.png')))
layout.addWidget(self.editor_button)
self.setLayout(layout)
def __init__(self, parent):
QWidget.__init__(self, parent)
layout = QHBoxLayout()
layout.setSpacing(5)
layout.setContentsMargins(0, 0, 0, 0)
self.tags_box = EditWithComplete(parent)
layout.addWidget(self.tags_box, stretch=1000)
self.editor_button = QToolButton(self)
self.editor_button.setToolTip(_('Open Item Editor'))
self.editor_button.setIcon(QIcon(I('chapters.png')))
layout.addWidget(self.editor_button)
self.setLayout(layout)
def __init__(self, parent, values):
QWidget.__init__(self, parent)
layout = QHBoxLayout()
layout.setSpacing(5)
layout.setContentsMargins(0, 0, 0, 0)
self.tags_box = EditWithComplete(parent)
self.tags_box.update_items_cache(values)
layout.addWidget(self.tags_box, stretch=3)
self.checkbox = QCheckBox(_('Remove all tags'), parent)
layout.addWidget(self.checkbox)
layout.addStretch(1)
self.setLayout(layout)
self.connect(self.checkbox, SIGNAL('stateChanged(int)'), self.box_touched)
def __init__(self, parent, values):
QWidget.__init__(self, parent)
layout = QHBoxLayout()
layout.setSpacing(5)
layout.setContentsMargins(0, 0, 0, 0)
self.tags_box = EditWithComplete(parent)
self.tags_box.update_items_cache(values)
layout.addWidget(self.tags_box, stretch=3)
self.checkbox = QCheckBox(_('Remove all tags'), parent)
layout.addWidget(self.checkbox)
layout.addStretch(1)
self.setLayout(layout)
self.connect(self.checkbox, SIGNAL('stateChanged(int)'), self.box_touched)
def __init__(self, *args):
QtGui.QFrame.__init__(self, *args)
self.edit = Editor()
self.pyFile = ""
self.number_bar = NumberBar(self.edit)
hbox = QHBoxLayout(self) # Pack the number bar, then pack the editor
hbox.setSpacing(10)
hbox.setContentsMargins(10, 0, 0, 0)
hbox.addWidget(self.number_bar)
hbox.addWidget(self.edit)
self.edit.blockCountChanged.connect(self.number_bar.adjustWidth)
self.edit.updateRequest.connect(self.number_bar.updateContents)
def __init__(self, *args): QtGui.QFrame.__init__(self, *args) self.edit = Editor() self.pyFile = "" self.number_bar = NumberBar(self.edit) hbox = QHBoxLayout(self) # Pack the number bar, then pack the editor hbox.setSpacing(10) hbox.setContentsMargins(10,0,0,0) hbox.addWidget(self.number_bar) hbox.addWidget(self.edit) self.edit.blockCountChanged.connect(self.number_bar.adjustWidth) self.edit.updateRequest.connect(self.number_bar.updateContents)
class windowTitle(QFrame):
def __init__(self, parent, closeButton=True):
QFrame.__init__(self, parent)
self.setMaximumSize(QSize(9999999, 22))
self.setObjectName("windowTitle")
self.hboxlayout = QHBoxLayout(self)
self.hboxlayout.setSpacing(0)
self.hboxlayout.setContentsMargins(0, 0, 4, 0)
self.label = QLabel(self)
self.label.setObjectName("label")
self.label.setStyleSheet(
"padding-left:4px; font:bold 11px; color: #FFFFFF;")
self.hboxlayout.addWidget(self.label)
spacerItem = QSpacerItem(40, 20, QSizePolicy.Expanding,
QSizePolicy.Minimum)
self.hboxlayout.addItem(spacerItem)
if closeButton:
self.pushButton = QPushButton(self)
self.pushButton.setFocusPolicy(Qt.NoFocus)
self.pushButton.setObjectName("pushButton")
self.pushButton.setStyleSheet("font:bold;")
self.pushButton.setText("X")
self.hboxlayout.addWidget(self.pushButton)
self.dragPosition = None
self.mainwidget = self.parent()
self.setStyleSheet("""
QFrame#windowTitle {background-color:#222222;color:#FFF;}
""")
# Initial position to top left
self.dragPosition = self.mainwidget.frameGeometry().topLeft()
def mousePressEvent(self, event):
if event.button() == Qt.LeftButton:
self.dragPosition = event.globalPos(
) - self.mainwidget.frameGeometry().topLeft()
event.accept()
def mouseMoveEvent(self, event):
if event.buttons() & Qt.LeftButton:
self.mainwidget.move(event.globalPos() - self.dragPosition)
event.accept()
def __init__(self, *args):
QFrame.__init__(self, *args)
self.setFrameStyle(QFrame.StyledPanel | QFrame.Sunken)
self.edit = self.PlainTextEdit()
self.number_bar = self.NumberBar(self.edit)
hbox = QHBoxLayout(self)
hbox.setSpacing(0)
hbox.setMargin(0)
hbox.addWidget(self.number_bar)
hbox.addWidget(self.edit)
self.edit.blockCountChanged.connect(self.number_bar.adjustWidth)
self.edit.updateRequest.connect(self.number_bar.updateContents)
def __init__(self, *args):
QFrame.__init__(self, *args)
self.setFrameStyle(QFrame.StyledPanel | QFrame.Sunken)
self.edit = self.PlainTextEdit()
self.number_bar = self.NumberBar(self.edit)
hbox = QHBoxLayout(self)
hbox.setSpacing(0)
hbox.setMargin(0)
hbox.addWidget(self.number_bar)
hbox.addWidget(self.edit)
self.edit.blockCountChanged.connect(self.number_bar.adjustWidth)
self.edit.updateRequest.connect(self.number_bar.updateContents)
def __init__(self, parent, label, extra_widgets=[], style=QFrame.HLine + QFrame.Raised, offset=16):
QWidget.__init__(self, parent)
lo = QHBoxLayout(self)
lo.setContentsMargins(0, 0, 0, 0)
lo.setSpacing(4)
if offset:
frame = QFrame(self)
frame.setFrameStyle(style)
frame.setMinimumWidth(offset)
lo.addWidget(frame, 0)
lo.addWidget(QLabel(label, self), 0)
frame = QFrame(self)
frame.setFrameStyle(style)
lo.addWidget(frame, 1)
for w in extra_widgets:
lo.addWidget(w, 0)
class windowTitle(QFrame):
def __init__(self, parent, closeButton=True):
QFrame.__init__(self, parent)
self.setMaximumSize(QSize(9999999,22))
self.setObjectName("windowTitle")
self.hboxlayout = QHBoxLayout(self)
self.hboxlayout.setSpacing(0)
self.hboxlayout.setContentsMargins(0,0,4,0)
self.label = QLabel(self)
self.label.setObjectName("label")
self.label.setStyleSheet("padding-left:4px; font:bold 11px; color: #FFFFFF;")
self.hboxlayout.addWidget(self.label)
spacerItem = QSpacerItem(40,20,QSizePolicy.Expanding,QSizePolicy.Minimum)
self.hboxlayout.addItem(spacerItem)
if closeButton:
self.pushButton = QPushButton(self)
self.pushButton.setFocusPolicy(Qt.NoFocus)
self.pushButton.setObjectName("pushButton")
self.pushButton.setStyleSheet("font:bold;")
self.pushButton.setText("X")
self.hboxlayout.addWidget(self.pushButton)
self.dragPosition = None
self.mainwidget = self.parent()
self.setStyleSheet("""
QFrame#windowTitle {background-color:#222222;color:#FFF;}
""")
# Initial position to top left
self.dragPosition = self.mainwidget.frameGeometry().topLeft()
def mousePressEvent(self, event):
if event.button() == Qt.LeftButton:
self.dragPosition = event.globalPos() - self.mainwidget.frameGeometry().topLeft()
event.accept()
def mouseMoveEvent(self, event):
if event.buttons() & Qt.LeftButton:
self.mainwidget.move(event.globalPos() - self.dragPosition)
event.accept()
def pmAddHeader(propMgr):
"""Creates the Property Manager header, which contains
a pixmap and white text label.
"""
# Heading frame (dark gray), which contains
# a pixmap and (white) heading text.
propMgr.header_frame = QFrame(propMgr)
propMgr.header_frame.setFrameShape(QFrame.NoFrame)
propMgr.header_frame.setFrameShadow(QFrame.Plain)
header_frame_palette = propMgr.getPropMgrTitleFramePalette()
propMgr.header_frame.setPalette(header_frame_palette)
propMgr.header_frame.setAutoFillBackground(True)
# HBox layout for heading frame, containing the pixmap
# and label (title).
HeaderFrameHLayout = QHBoxLayout(propMgr.header_frame)
HeaderFrameHLayout.setMargin(pmHeaderFrameMargin) # 2 pixels around edges.
HeaderFrameHLayout.setSpacing(
pmHeaderFrameSpacing) # 5 pixel between pixmap and label.
# PropMgr icon. Set image by calling setPropMgrIcon() at any time.
propMgr.header_pixmap = QLabel(propMgr.header_frame)
propMgr.header_pixmap.setSizePolicy(
QSizePolicy(QSizePolicy.Policy(QSizePolicy.Fixed),
QSizePolicy.Policy(QSizePolicy.Fixed)))
propMgr.header_pixmap.setScaledContents(True)
HeaderFrameHLayout.addWidget(propMgr.header_pixmap)
# PropMgr title label
propMgr.header_label = QLabel(propMgr.header_frame)
header_label_palette = propMgr.getPropMgrTitleLabelPalette()
propMgr.header_label.setPalette(header_label_palette)
propMgr.header_label.setAlignment(pmLabelLeftAlignment)
# PropMgr heading font (for label).
propMgr.header_label.setFont(getHeaderFont())
HeaderFrameHLayout.addWidget(propMgr.header_label)
propMgr.pmVBoxLayout.addWidget(propMgr.header_frame)
def pmAddHeader(propMgr):
"""Creates the Property Manager header, which contains
a pixmap and white text label.
"""
# Heading frame (dark gray), which contains
# a pixmap and (white) heading text.
propMgr.header_frame = QFrame(propMgr)
propMgr.header_frame.setFrameShape(QFrame.NoFrame)
propMgr.header_frame.setFrameShadow(QFrame.Plain)
header_frame_palette = propMgr.getPropMgrTitleFramePalette()
propMgr.header_frame.setPalette(header_frame_palette)
propMgr.header_frame.setAutoFillBackground(True)
# HBox layout for heading frame, containing the pixmap
# and label (title).
HeaderFrameHLayout = QHBoxLayout(propMgr.header_frame)
HeaderFrameHLayout.setMargin(pmHeaderFrameMargin) # 2 pixels around edges.
HeaderFrameHLayout.setSpacing(pmHeaderFrameSpacing) # 5 pixel between pixmap and label.
# PropMgr icon. Set image by calling setPropMgrIcon() at any time.
propMgr.header_pixmap = QLabel(propMgr.header_frame)
propMgr.header_pixmap.setSizePolicy(
QSizePolicy(QSizePolicy.Policy(QSizePolicy.Fixed),
QSizePolicy.Policy(QSizePolicy.Fixed)))
propMgr.header_pixmap.setScaledContents(True)
HeaderFrameHLayout.addWidget(propMgr.header_pixmap)
# PropMgr title label
propMgr.header_label = QLabel(propMgr.header_frame)
header_label_palette = propMgr.getPropMgrTitleLabelPalette()
propMgr.header_label.setPalette(header_label_palette)
propMgr.header_label.setAlignment(pmLabelLeftAlignment)
# PropMgr heading font (for label).
propMgr.header_label.setFont(getHeaderFont())
HeaderFrameHLayout.addWidget(propMgr.header_label)
propMgr.pmVBoxLayout.addWidget(propMgr.header_frame)
def __init__(self, parent, modal=True, flags=Qt.WindowFlags()):
QDialog.__init__(self, parent, flags)
self.setModal(modal)
self.setWindowTitle("Add Tag")
lo = QVBoxLayout(self)
lo.setMargin(10)
lo.setSpacing(5)
# tag selector
lo1 = QHBoxLayout()
lo.addLayout(lo1)
lo1.setSpacing(5)
self.wtagsel = QComboBox(self)
self.wtagsel.setEditable(True)
wtagsel_lbl = QLabel("&Tag:", self)
wtagsel_lbl.setBuddy(self.wtagsel)
lo1.addWidget(wtagsel_lbl, 0)
lo1.addWidget(self.wtagsel, 1)
QObject.connect(self.wtagsel, SIGNAL("activated(int)"),
self._check_tag)
QObject.connect(self.wtagsel,
SIGNAL("editTextChanged(const QString &)"),
self._check_tag_text)
# value editor
self.valedit = ValueTypeEditor(self)
lo.addWidget(self.valedit)
# buttons
lo.addSpacing(10)
lo2 = QHBoxLayout()
lo.addLayout(lo2)
lo2.setContentsMargins(0, 0, 0, 0)
lo2.setMargin(5)
self.wokbtn = QPushButton("OK", self)
self.wokbtn.setMinimumWidth(128)
QObject.connect(self.wokbtn, SIGNAL("clicked()"), self.accept)
self.wokbtn.setEnabled(False)
cancelbtn = QPushButton("Cancel", self)
cancelbtn.setMinimumWidth(128)
QObject.connect(cancelbtn, SIGNAL("clicked()"), self.reject)
lo2.addWidget(self.wokbtn)
lo2.addStretch(1)
lo2.addWidget(cancelbtn)
self.setMinimumWidth(384)
def __init__(self, main, interpreterLocals, ventana_interprete, *args):
QFrame.__init__(self, *args)
self.setFrameStyle(QFrame.StyledPanel | QFrame.Sunken)
self.editor = WidgetEditor(self, interpreterLocals, ventana_interprete)
self.editor.setFrameStyle(QFrame.NoFrame)
self.editor.setAcceptRichText(False)
self.number_bar = self.NumberBar()
self.number_bar.setTextEdit(self.editor)
hbox = QHBoxLayout(self)
hbox.setSpacing(0)
hbox.setMargin(0)
hbox.addWidget(self.number_bar)
hbox.addWidget(self.editor)
self.editor.installEventFilter(self)
self.editor.viewport().installEventFilter(self)
def __init__(self, main, interpreterLocals, ventana_interprete, *args):
QFrame.__init__(self, *args)
self.setFrameStyle(QFrame.StyledPanel | QFrame.Sunken)
self.editor = WidgetEditor(self, interpreterLocals, ventana_interprete)
self.editor.setFrameStyle(QFrame.NoFrame)
self.editor.setAcceptRichText(False)
self.number_bar = self.NumberBar()
self.number_bar.setTextEdit(self.editor)
hbox = QHBoxLayout(self)
hbox.setSpacing(0)
hbox.setMargin(0)
hbox.addWidget(self.number_bar)
hbox.addWidget(self.editor)
self.editor.installEventFilter(self)
self.editor.viewport().installEventFilter(self)
def __init__(self, *args):
QFrame.__init__(self, *args)
self.setFrameStyle(QFrame.StyledPanel | QFrame.Sunken)
self.edit = QTextEdit()
self.edit.setFrameStyle(QFrame.NoFrame)
self.edit.setAcceptRichText(False)
self.number_bar = NumberBar()
self.number_bar.setTextEdit(self.edit)
hbox = QHBoxLayout(self)
hbox.setSpacing(0)
hbox.setMargin(0)
hbox.addWidget(self.number_bar)
hbox.addWidget(self.edit)
self.edit.installEventFilter(self)
self.edit.viewport().installEventFilter(self)
def __init__(self, *args):
QFrame.__init__(self, *args)
self.setFrameStyle(QFrame.StyledPanel | QFrame.Sunken)
self.edit = QTextEdit()
self.edit.setFrameStyle(QFrame.NoFrame)
self.edit.setAcceptRichText(False)
self.number_bar = NumberBar()
self.number_bar.setTextEdit(self.edit)
hbox = QHBoxLayout(self)
hbox.setSpacing(0)
hbox.setMargin(0)
hbox.addWidget(self.number_bar)
hbox.addWidget(self.edit)
self.edit.installEventFilter(self)
self.edit.viewport().installEventFilter(self)
def __init__(self, parent, modal=True, flags=Qt.WindowFlags()):
QDialog.__init__(self, parent, flags)
self.setModal(modal)
self.setWindowTitle("Add Tag")
lo = QVBoxLayout(self)
lo.setMargin(10)
lo.setSpacing(5)
# tag selector
lo1 = QHBoxLayout()
lo.addLayout(lo1)
lo1.setSpacing(5)
self.wtagsel = QComboBox(self)
self.wtagsel.setEditable(True)
wtagsel_lbl = QLabel("&Tag:", self)
wtagsel_lbl.setBuddy(self.wtagsel)
lo1.addWidget(wtagsel_lbl, 0)
lo1.addWidget(self.wtagsel, 1)
QObject.connect(self.wtagsel, SIGNAL("activated(int)"), self._check_tag)
QObject.connect(self.wtagsel, SIGNAL("editTextChanged(const QString &)"), self._check_tag_text)
# value editor
self.valedit = ValueTypeEditor(self)
lo.addWidget(self.valedit)
# buttons
lo.addSpacing(10)
lo2 = QHBoxLayout()
lo.addLayout(lo2)
lo2.setContentsMargins(0, 0, 0, 0)
lo2.setMargin(5)
self.wokbtn = QPushButton("OK", self)
self.wokbtn.setMinimumWidth(128)
QObject.connect(self.wokbtn, SIGNAL("clicked()"), self.accept)
self.wokbtn.setEnabled(False)
cancelbtn = QPushButton("Cancel", self)
cancelbtn.setMinimumWidth(128)
QObject.connect(cancelbtn, SIGNAL("clicked()"), self.reject)
lo2.addWidget(self.wokbtn)
lo2.addStretch(1)
lo2.addWidget(cancelbtn)
self.setMinimumWidth(384)
class PM_ColorChooser(QWidget):
"""
The PM_ColorChooser widget provides a color chooser widget for a
Property Manager group box. The PM_ColorChooser widget is a composite widget
made from 3 other Qt widgets:
- a QLabel
- a QFrame and
- a QToolButton (with a "..." text label).
IMAGE(http://www.nanoengineer-1.net/mediawiki/images/e/e2/PM_ColorChooser1.jpg)
The user can color using Qt's color (chooser) dialog by clicking the "..."
button. The selected color will be used as the color of the QFrame widget.
The parent must make the following signal-slot connection to be
notified when the user has selected a new color via the color chooser
dialog:
self.connect(pmColorChooser.colorFrame, SIGNAL("editingFinished()"), self.mySlotMethod)
@cvar setAsDefault: Determines whether to reset the value of the
color to I{defaultColor} when the user clicks
the "Restore Defaults" button.
@type setAsDefault: boolean
@cvar labelWidget: The Qt label widget of this PM widget.
@type labelWidget: U{B{QLabel}<http://doc.trolltech.com/4/qlabel.html>}
@cvar colorFrame: The Qt frame widget for this PM widget.
@type colorFrame: U{B{QFrame}<http://doc.trolltech.com/4/qframe.html>}
@cvar chooseButton: The Qt tool button widget for this PM widget.
@type chooseButton: U{B{QToolButton}<http://doc.trolltech.com/4/qtoolbutton.html>}
"""
defaultColor = None
setAsDefault = True
hidden = False
chooseButton = None
customColorCount = 0
standardColorList = [white, black]
def __init__(
self,
parentWidget,
label='Color:',
labelColumn=0,
color=white,
setAsDefault=True,
spanWidth=False,
):
"""
Appends a color chooser widget to <parentWidget>, a property manager
group box.
@param parentWidget: the parent group box containing this widget.
@type parentWidget: PM_GroupBox
@param label: The label that appears to the left or right of the
color frame (and "Browse" button).
If spanWidth is True, the label will be displayed on
its own row directly above the lineedit (and button).
To suppress the label, set I{label} to an
empty string.
@type label: str
@param labelColumn: The column number of the label in the group box
grid layout. The only valid values are 0 (left
column) and 1 (right column). The default is 0
(left column).
@type labelColumn: int
@param color: initial color. White is the default.
@type color: tuple of 3 floats (r, g, b)
@param setAsDefault: if True, will restore L{color} when the
"Restore Defaults" button is clicked.
@type setAsDefault: boolean
@param spanWidth: if True, the widget and its label will span the width
of the group box. Its label will appear directly above
the widget (unless the label is empty) and is left
justified.
@type spanWidth: boolean
@see: U{B{QColorDialog}<http://doc.trolltech.com/4/qcolordialog.html>}
"""
QWidget.__init__(self)
self.parentWidget = parentWidget
self.label = label
self.labelColumn = labelColumn
self.color = color
self.setAsDefault = setAsDefault
self.spanWidth = spanWidth
if label: # Create this widget's QLabel.
self.labelWidget = QLabel()
self.labelWidget.setText(label)
# Create the color frame (color swath) and "..." button.
self.colorFrame = QFrame()
self.colorFrame.setFrameShape(QFrame.Box)
self.colorFrame.setFrameShadow(QFrame.Plain)
# Set browse button text and make signal-slot connection.
self.chooseButton = QToolButton()
self.chooseButton.setText("...")
self.connect(self.chooseButton, SIGNAL("clicked()"),
self.openColorChooserDialog)
# Add a horizontal spacer to keep the colorFrame and "..." squeezed
# together, even when the PM width changes.
self.hSpacer = QSpacerItem(10, 10, QSizePolicy.MinimumExpanding,
QSizePolicy.Fixed)
# Create vertical box layout.
self.hBoxLayout = QHBoxLayout(self)
self.hBoxLayout.setMargin(0)
self.hBoxLayout.setSpacing(2)
self.hBoxLayout.insertWidget(-1, self.colorFrame)
self.hBoxLayout.insertWidget(-1, self.chooseButton)
# Set this to False to make the colorFrame an expandable rectangle.
COLORFRAME_IS_SQUARE = True
if COLORFRAME_IS_SQUARE:
squareSize = 20
self.colorFrame.setMinimumSize(QSize(squareSize, squareSize))
self.colorFrame.setMaximumSize(QSize(squareSize, squareSize))
self.hBoxLayout.addItem(self.hSpacer)
self.setColor(color, default=setAsDefault)
parentWidget.addPmWidget(self)
return
def setColor(self, color, default=False):
"""
Set the color.
@param color: The color.
@type color: tuple of 3 floats (r, g, b)
@param default: If True, make I{color} the default color. Default is
False.
@type default: boolean
"""
if default:
self.defaultColor = color
self.setAsDefault = default
self.color = color
self._updateColorFrame()
return
def getColor(self):
"""
Return the current color.
@return: The current r, g, b color.
@rtype: Tuple of 3 floats (r, g, b)
"""
return self.color
def getQColor(self):
"""
Return the current QColor.
@return: The current color.
@rtype: QColor
"""
return RGBf_to_QColor(self.color)
def _updateColorFrame(self):
"""
Updates the color frame with the current color.
"""
colorframe = self.colorFrame
try:
qcolor = self.getQColor()
palette = QPalette(
) # QPalette(qcolor) would have window color set from qcolor, but that doesn't help us here
qcolorrole = QPalette.Window
## http://doc.trolltech.com/4.2/qpalette.html#ColorRole-enum says:
## QPalette.Window 10 A general background color.
palette.setColor(QPalette.Active, qcolorrole,
qcolor) # used when window is in fg and has focus
palette.setColor(
QPalette.Inactive, qcolorrole,
qcolor) # used when window is in bg or does not have focus
palette.setColor(QPalette.Disabled, qcolorrole,
qcolor) # used when widget is disabled
colorframe.setPalette(palette)
colorframe.setAutoFillBackground(True)
except:
print "data for following exception: ",
print "colorframe %r has palette %r" % (colorframe,
colorframe.palette())
pass
def openColorChooserDialog(self):
"""
Prompts the user to choose a color and then updates colorFrame with
the selected color.
"""
qcolor = RGBf_to_QColor(self.color)
if not self.color in self.standardColorList:
QColorDialog.setCustomColor(self.customColorCount, qcolor.rgb())
self.customColorCount += 1
c = QColorDialog.getColor(qcolor, self)
if c.isValid():
self.setColor(QColor_to_RGBf(c))
self.colorFrame.emit(SIGNAL("editingFinished()"))
def restoreDefault(self):
"""
Restores the default value.
"""
if self.setAsDefault:
self.setColor(self.defaultColor)
return
def hide(self):
"""
Hides the lineedit and its label (if it has one).
@see: L{show}
"""
QWidget.hide(self)
if self.labelWidget:
self.labelWidget.hide()
return
def show(self):
"""
Unhides the lineedit and its label (if it has one).
@see: L{hide}
"""
QWidget.show(self)
if self.labelWidget:
self.labelWidget.show()
return
def _createTopRowBtns(self):
"""
Creates the Done, Cancel, Preview, Restore Defaults and What's This
buttons row at the top of the Property Manager.
"""
topBtnSize = QSize(22, 22) # button images should be 16 x 16, though.
# Main "button group" widget (but it is not a QButtonGroup).
self.pmTopRowBtns = QHBoxLayout()
# This QHBoxLayout is (probably) not necessary. Try using just the frame
# for the foundation. I think it should work. Mark 2007-05-30
# Horizontal spacer
horizontalSpacer = QSpacerItem(1, 1,
QSizePolicy.Expanding,
QSizePolicy.Minimum)
# Widget containing all the buttons.
self.topRowBtnsContainer = QWidget()
# Create Hbox layout for main frame.
topRowBtnsHLayout = QHBoxLayout(self.topRowBtnsContainer)
topRowBtnsHLayout.setMargin(PM_TOPROWBUTTONS_MARGIN)
topRowBtnsHLayout.setSpacing(PM_TOPROWBUTTONS_SPACING)
# Set to True to center align the buttons in the PM
if False: # Left aligns the buttons.
topRowBtnsHLayout.addItem(horizontalSpacer)
# Done (OK) button.
self.done_btn = QToolButton(self.topRowBtnsContainer)
self.done_btn.setIcon(
geticon("ui/actions/Properties Manager/Done_16x16.png"))
self.done_btn.setIconSize(topBtnSize)
self.done_btn.setAutoRaise(True)
self.connect(self.done_btn,
SIGNAL("clicked()"),
self.doneButtonClicked)
self.done_btn.setToolTip("Done")
topRowBtnsHLayout.addWidget(self.done_btn)
# Cancel (Abort) button.
self.cancel_btn = QToolButton(self.topRowBtnsContainer)
self.cancel_btn.setIcon(
geticon("ui/actions/Properties Manager/Abort_16x16.png"))
self.cancel_btn.setIconSize(topBtnSize)
self.cancel_btn.setAutoRaise(True)
self.connect(self.cancel_btn,
SIGNAL("clicked()"),
self.cancelButtonClicked)
self.cancel_btn.setToolTip("Cancel")
topRowBtnsHLayout.addWidget(self.cancel_btn)
#@ abort_btn deprecated. We still need it because modes use it.
self.abort_btn = self.cancel_btn
# Restore Defaults button.
self.restore_defaults_btn = QToolButton(self.topRowBtnsContainer)
self.restore_defaults_btn.setIcon(
geticon("ui/actions/Properties Manager/Restore_16x16.png"))
self.restore_defaults_btn.setIconSize(topBtnSize)
self.restore_defaults_btn.setAutoRaise(True)
self.connect(self.restore_defaults_btn,
SIGNAL("clicked()"),
self.restoreDefaultsButtonClicked)
self.restore_defaults_btn.setToolTip("Restore Defaults")
topRowBtnsHLayout.addWidget(self.restore_defaults_btn)
# Preview (glasses) button.
self.preview_btn = QToolButton(self.topRowBtnsContainer)
self.preview_btn.setIcon(
geticon("ui/actions/Properties Manager/Preview_16x16.png"))
self.preview_btn.setIconSize(topBtnSize)
self.preview_btn.setAutoRaise(True)
self.connect(self.preview_btn,
SIGNAL("clicked()"),
self.previewButtonClicked)
self.preview_btn.setToolTip("Preview")
topRowBtnsHLayout.addWidget(self.preview_btn)
# What's This (?) button.
self.whatsthis_btn = QToolButton(self.topRowBtnsContainer)
self.whatsthis_btn.setIcon(
geticon("ui/actions/Properties Manager/WhatsThis_16x16.png"))
self.whatsthis_btn.setIconSize(topBtnSize)
self.whatsthis_btn.setAutoRaise(True)
self.connect(self.whatsthis_btn,
SIGNAL("clicked()"),
self.whatsThisButtonClicked)
self.whatsthis_btn.setToolTip("Enter \"What's This\" help mode")
topRowBtnsHLayout.addWidget(self.whatsthis_btn)
topRowBtnsHLayout.addItem(horizontalSpacer)
# Create Button Row
self.pmTopRowBtns.addWidget(self.topRowBtnsContainer)
self.vBoxLayout.addLayout(self.pmTopRowBtns)
# Add What's This for buttons.
self.done_btn.setWhatsThis("""<b>Done</b>
<p>
<img source=\"ui/actions/Properties Manager/Done_16x16.png\"><br>
Completes and/or exits the current command.</p>""")
self.cancel_btn.setWhatsThis("""<b>Cancel</b>
<p>
<img source=\"ui/actions/Properties Manager/Abort_16x16.png\"><br>
Cancels the current command.</p>""")
self.restore_defaults_btn.setWhatsThis("""<b>Restore Defaults</b>
<p><img source=\"ui/actions/Properties Manager/Restore_16x16.png\"><br>
Restores the defaut values of the Property Manager.</p>""")
self.preview_btn.setWhatsThis("""<b>Preview</b>
<p>
<img source=\"ui/actions/Properties Manager/Preview_16x16.png\"><br>
Preview the structure based on current Property Manager settings.
</p>""")
self.whatsthis_btn.setWhatsThis("""<b>What's This</b>
<p>
<img source=\"ui/actions/Properties Manager/WhatsThis_16x16.png\"><br>
This invokes \"What's This?\" help mode which is part of
NanoEngineer-1's online help system, and provides users with
information about the functionality and usage of a particular
command button or widget.
</p>""")
return
def _createTopRowBtns(self):
"""
Creates the Done, Cancel, Preview, Restore Defaults and What's This
buttons row at the top of the Property Manager.
"""
# Main "button group" widget (but it is not a QButtonGroup).
self.pmTopRowBtns = QHBoxLayout()
# This QHBoxLayout is (probably) not necessary. Try using just the frame
# for the foundation. I think it should work. Mark 2007-05-30
# Horizontal spacer
horizontalSpacer = QSpacerItem(1, 1, QSizePolicy.Expanding,
QSizePolicy.Minimum)
# Frame containing all the buttons.
self.topRowBtnsFrame = QFrame()
self.topRowBtnsFrame.setFrameShape(QFrame.NoFrame)
self.topRowBtnsFrame.setFrameShadow(QFrame.Plain)
# Create Hbox layout for main frame.
topRowBtnsHLayout = QHBoxLayout(self.topRowBtnsFrame)
topRowBtnsHLayout.setMargin(PM_TOPROWBUTTONS_MARGIN)
topRowBtnsHLayout.setSpacing(PM_TOPROWBUTTONS_SPACING)
topRowBtnsHLayout.addItem(horizontalSpacer)
# Set button type.
if 1: # Mark 2007-05-30
# Needs to be QToolButton for MacOS. Fine for Windows, too.
buttonType = QToolButton
# May want to use QToolButton.setAutoRaise(1) below. Mark 2007-05-29
else:
buttonType = QPushButton # Do not use.
# Done (OK) button.
self.done_btn = buttonType(self.topRowBtnsFrame)
self.done_btn.setIcon(
geticon("ui/actions/Properties Manager/Done.png"))
self.done_btn.setIconSize(QSize(22, 22))
self.connect(self.done_btn, SIGNAL("clicked()"),
self.doneButtonClicked)
self.done_btn.setToolTip("Done")
topRowBtnsHLayout.addWidget(self.done_btn)
# Cancel (Abort) button.
self.cancel_btn = buttonType(self.topRowBtnsFrame)
self.cancel_btn.setIcon(
geticon("ui/actions/Properties Manager/Abort.png"))
self.cancel_btn.setIconSize(QSize(22, 22))
self.connect(self.cancel_btn, SIGNAL("clicked()"),
self.cancelButtonClicked)
self.cancel_btn.setToolTip("Cancel")
topRowBtnsHLayout.addWidget(self.cancel_btn)
#@ abort_btn deprecated. We still need it because modes use it.
self.abort_btn = self.cancel_btn
# Restore Defaults button.
self.restore_defaults_btn = buttonType(self.topRowBtnsFrame)
self.restore_defaults_btn.setIcon(
geticon("ui/actions/Properties Manager/Restore.png"))
self.restore_defaults_btn.setIconSize(QSize(22, 22))
self.connect(self.restore_defaults_btn, SIGNAL("clicked()"),
self.restoreDefaultsButtonClicked)
self.restore_defaults_btn.setToolTip("Restore Defaults")
topRowBtnsHLayout.addWidget(self.restore_defaults_btn)
# Preview (glasses) button.
self.preview_btn = buttonType(self.topRowBtnsFrame)
self.preview_btn.setIcon(
geticon("ui/actions/Properties Manager/Preview.png"))
self.preview_btn.setIconSize(QSize(22, 22))
self.connect(self.preview_btn, SIGNAL("clicked()"),
self.previewButtonClicked)
self.preview_btn.setToolTip("Preview")
topRowBtnsHLayout.addWidget(self.preview_btn)
# What's This (?) button.
self.whatsthis_btn = buttonType(self.topRowBtnsFrame)
self.whatsthis_btn.setIcon(
geticon("ui/actions/Properties Manager/WhatsThis.png"))
self.whatsthis_btn.setIconSize(QSize(22, 22))
self.connect(self.whatsthis_btn, SIGNAL("clicked()"),
self.whatsThisButtonClicked)
self.whatsthis_btn.setToolTip("Enter \"What's This\" help mode")
topRowBtnsHLayout.addWidget(self.whatsthis_btn)
topRowBtnsHLayout.addItem(horizontalSpacer)
# Create Button Row
self.pmTopRowBtns.addWidget(self.topRowBtnsFrame)
self.vBoxLayout.addLayout(self.pmTopRowBtns)
# Add What's This for buttons.
self.done_btn.setWhatsThis("""<b>Done</b>
<p>
<img source=\"ui/actions/Properties Manager/Done.png\"><br>
Completes and/or exits the current command.</p>""")
self.cancel_btn.setWhatsThis("""<b>Cancel</b>
<p>
<img source=\"ui/actions/Properties Manager/Abort.png\"><br>
Cancels the current command.</p>""")
self.restore_defaults_btn.setWhatsThis("""<b>Restore Defaults</b>
<p><img source=\"ui/actions/Properties Manager/Restore.png\"><br>
Restores the defaut values of the Property Manager.</p>""")
self.preview_btn.setWhatsThis("""<b>Preview</b>
<p>
<img source=\"ui/actions/Properties Manager/Preview.png\"><br>
Preview the structure based on current Property Manager settings.
</p>""")
self.whatsthis_btn.setWhatsThis("""<b>What's This</b>
<p>
<img source=\"ui/actions/Properties Manager/WhatsThis.png\"><br>
This invokes \"What's This?\" help mode which is part of
NanoEngineer-1's online help system, and provides users with
information about the functionality and usage of a particular
command button or widget.
</p>""")
return
def __init__(self):
QMainWindow.__init__(self)
self.setWindowTitle("My Main Window")
self.setMinimumWidth(MAIN_WINDOW_SIZE[0])
self.setMinimumHeight(MAIN_WINDOW_SIZE[1])
self.statusbar = QtGui.QStatusBar(self)
self.statusbar.showMessage("Status message")
self.setStatusBar(self.statusbar)
################################################
self.menubar = self.menuBar()
# Any menu action makes the status bar message disappear
fileMenu = QtGui.QMenu(self.menubar)
fileMenu.setTitle("File")
self.menubar.addAction(fileMenu.menuAction())
newAction = QtGui.QAction("New", self)
newAction.setIcon(QtGui.QtIcon(icons + '/GroupPropDialog_image0.png'))
fileMenu.addAction(newAction)
openAction = QtGui.QAction("Open", self)
openAction.setIcon(QtGui.QtIcon(icons + "/MainWindowUI_image1"))
fileMenu.addAction(openAction)
saveAction = QtGui.QAction("Save", self)
saveAction.setIcon(QtGui.QtIcon(icons + "/MainWindowUI_image2"))
fileMenu.addAction(saveAction)
self.connect(newAction,SIGNAL("activated()"),self.fileNew)
self.connect(openAction,SIGNAL("activated()"),self.fileOpen)
self.connect(saveAction,SIGNAL("activated()"),self.fileSave)
for otherMenuName in ('Edit', 'View', 'Display', 'Select', 'Modify', 'NanoHive-1'):
otherMenu = QtGui.QMenu(self.menubar)
otherMenu.setTitle(otherMenuName)
self.menubar.addAction(otherMenu.menuAction())
helpMenu = QtGui.QMenu(self.menubar)
helpMenu.setTitle("Help")
self.menubar.addAction(helpMenu.menuAction())
aboutAction = QtGui.QAction("About", self)
aboutAction.setIcon(QtGui.QtIcon(icons + '/MainWindowUI_image0.png'))
helpMenu.addAction(aboutAction)
self.connect(aboutAction,SIGNAL("activated()"),self.helpAbout)
##############################################
self.setMenuBar(self.menubar)
centralwidget = QWidget()
self.setCentralWidget(centralwidget)
layout = QVBoxLayout(centralwidget)
layout.setMargin(0)
layout.setSpacing(0)
middlewidget = QWidget()
self.bigButtons = QWidget()
bblo = QHBoxLayout(self.bigButtons)
bblo.setMargin(0)
bblo.setSpacing(0)
self.bigButtons.setMinimumHeight(50)
self.bigButtons.setSizePolicy(QSizePolicy.Fixed, QSizePolicy.Fixed)
for name in ('Features', 'Sketch', 'Build', 'Dimension', 'Simulator'):
btn = QPushButton(self.bigButtons)
btn.setMaximumWidth(80)
btn.setMinimumHeight(50)
btn.setText(name)
self.bigButtons.layout().addWidget(btn)
self.bigButtons.hide()
layout.addWidget(self.bigButtons)
self.littleIcons = QWidget()
self.littleIcons.setMinimumHeight(30)
self.littleIcons.setMaximumHeight(30)
lilo = QHBoxLayout(self.littleIcons)
lilo.setMargin(0)
lilo.setSpacing(0)
pb = QPushButton(self.littleIcons)
pb.setIcon(QIcon(icons + '/GroupPropDialog_image0.png'))
self.connect(pb,SIGNAL("clicked()"),self.fileNew)
lilo.addWidget(pb)
for x in "1 2 4 5 6 7 8 18 42 10 43 150 93 94 97 137".split():
pb = QPushButton(self.littleIcons)
pb.setIcon(QIcon(icons + '/MainWindowUI_image' + x + '.png'))
lilo.addWidget(pb)
layout.addWidget(self.littleIcons)
layout.addWidget(middlewidget)
self.layout = QGridLayout(middlewidget)
self.layout.setMargin(0)
self.layout.setSpacing(2)
self.gridPosition = GridPosition()
self.numParts = 0
self.show()
explainWindow = AboutWindow("Select <b>Help->About</b>"
" for instructions...",
200, 3)
class LetsShareBooksDialog(QDialog):
def __init__(self, gui, icon, do_user_config, qaction, us):
QDialog.__init__(self, gui)
self.gui = gui
self.do_user_config = do_user_config
self.qaction = qaction
self.us = us
self.clip = QApplication.clipboard()
self.main_gui = calibre_main()
self.urllib_thread = UrlLibThread(self.us)
self.kill_servers_thread = KillServersThread(self.us)
self.us.check_finished = True
self.pxmp = QPixmap()
self.pxmp.load('images/icon_connected.png')
self.icon_connected = QIcon(self.pxmp)
self.setStyleSheet("""
QDialog {
background-color: white;
}
QPushButton {
font-size: 16px;
border-style: solid;
border-color: red;
font-family:'BitstreamVeraSansMono',Consolas,monospace;
text-transform: uppercase;
}
QPushButton#arrow {
border-width: 16px;
border-right-color:white;
padding: -10px;
color:red;
}
QPushButton#url {
background-color: red;
min-width: 460px;
color: white;
text-align: left;
}
QPushButton#url:hover {
background-color: white;
color: red;
}
QPushButton#share {
background-color: red;
color: white;
margin-right: 10px;
}
QPushButton#share:hover {
background-color: white;
color: red;
}
QPushButton#url2 {
color: #222;
text-align: left;
}
QPushButton#url2:hover {
color: red;
}
""")
self.ll = QVBoxLayout()
#self.ll.setSpacing(1)
self.l = QHBoxLayout()
self.l.setSpacing(0)
self.l.setMargin(0)
#self.l.setContentsMargins(0,0,0,0)
self.w = QWidget()
self.w.setLayout(self.l)
self.setLayout(self.ll)
self.setWindowIcon(icon)
self.lets_share_button = QPushButton()
self.lets_share_button.setSizePolicy(QSizePolicy.MinimumExpanding, QSizePolicy.MinimumExpanding)
self.lets_share_button.setObjectName("share")
self.lets_share_button.clicked.connect(self.lets_share)
self.stop_share_button = QPushButton()
self.stop_share_button.setSizePolicy(QSizePolicy.MinimumExpanding, QSizePolicy.MinimumExpanding)
self.stop_share_button.setObjectName("share")
self.stop_share_button.clicked.connect(self.stop_share)
self.l.addWidget(self.lets_share_button)
self.l.addWidget(self.stop_share_button)
if self.us.button_state == "start":
self.lets_share_button.show()
self.stop_share_button.hide()
self.lets_share_button.setText(self.us.share_button_text)
else:
self.lets_share_button.hide()
self.stop_share_button.show()
self.stop_share_button.setText(self.us.share_button_text)
self.url_label = QPushButton()
self.url_label.setSizePolicy(QSizePolicy.MinimumExpanding, QSizePolicy.MinimumExpanding)
self.url_label.setObjectName("url")
self.url_label.clicked.connect(self.open_url)
self.l.addWidget(self.url_label)
self.arrow_button = QPushButton("_____")
self.arrow_button.setObjectName("arrow")
self.l.addWidget(self.arrow_button)
self.ll.addWidget(self.w)
self.ll.addSpacing(10)
self.chat_button = QPushButton("Chat room: https://chat.memoryoftheworld.org")
#self.chat_button.hovered.connect(self.setCursorToHand)
self.chat_button.setObjectName("url2")
self.chat_button.setToolTip('Meetings every thursday at 23:59 (central eruopean time)')
self.chat_button.clicked.connect(functools.partial(self.open_url2, "https://chat.memoryoftheworld.org"))
self.ll.addWidget(self.chat_button)
self.about_project_button = QPushButton('Public Library: http://www.memoryoftheworld.org')
self.about_project_button.setObjectName("url2")
self.about_project_button.setToolTip('When everyone is librarian, library is everywhere.')
self.about_project_button.clicked.connect(functools.partial(self.open_url2, "http://www.memoryoftheworld.org"))
self.ll.addWidget(self.about_project_button)
#self.debug_log = QListWidget()
#self.ll.addWidget(self.debug_log)
#self.debug_log.addItem("Initiatied!")
self.upgrade_button = QPushButton('Please download and upgrade from {0} to {1} version of plugin.'.format(self.us.running_version, self.us.latest_version))
self.upgrade_button.setObjectName("url2")
self.upgrade_button.setToolTip('Running latest version you make developers happy')
self.upgrade_button.clicked.connect(functools.partial(self.open_url2, self.us.plugin_url))
version_list = [self.us.running_version, self.us.latest_version]
version_list.sort(key=lambda s: map(int, s.split('.')))
if self.us.running_version != self.us.latest_version:
if self.us.running_version == version_list[0]:
self.ll.addSpacing(20)
self.ll.addWidget(self.upgrade_button)
self.resize(self.sizeHint())
self.se = open("lsb.log", "w+b")
self.so = self.se
sys.stdout = os.fdopen(sys.stdout.fileno(), 'w', 0)
os.dup2(self.so.fileno(), sys.stdout.fileno())
os.dup2(self.se.fileno(), sys.stderr.fileno())
self.timer = QTimer()
self.timer.timeout.connect(self.check_and_render)
self.timer_period = 300
self.timer.start(self.timer_period)
self.error_log = ""
def lets_share(self):
self.lets_share_button.setEnabled(False)
self.timer.stop()
self.us.share_button_text = "Connecting..."
#self.debug_log.addItem("Let's share!")
self.us.counter = 0
self.us.lost_connection = False
if not self.us.ssh_proc:
self.main_gui.start_content_server()
opts, args = server_config().option_parser().parse_args(['calibre-server'])
self.calibre_server_port = opts.port
if sys.platform == "win32":
self.win_reg = subprocess.Popen("regedit /s .hosts.reg")
self.us.win_port = int(random.random()*40000+10000)
self.us.ssh_proc = subprocess.Popen("lsbtunnel.exe -N -T tunnel@{2} -R {0}:localhost:{1} -P 722".format(self.us.win_port, self.calibre_server_port, prefs['lsb_server']), shell=True)
self.us.lsb_url = "https://www{0}.{1}".format(self.us.win_port, prefs['lsb_server'])
#_dev_self.us.lsb_url = "http://www{0}.{1}".format(self.us.win_port, prefs['lsb_server'])
self.us.lsb_url_text = "Go to: {0}".format(self.us.lsb_url)
self.us.found_url = True
else:
self.us.ssh_proc = subprocess.Popen(['ssh', '-T', '-N', '-g', '-o', 'UserKnownHostsFile=.userknownhostsfile', '-o', 'TCPKeepAlive=yes', '-o', 'ServerAliveINterval=60', prefs['lsb_server'], '-l', 'tunnel', '-R', '0:localhost:{0}'.format(self.calibre_server_port), '-p', '722'])
self.us.found_url = None
self.qaction.setIcon(get_icon('images/icon_connected.png'))
self.us.connecting = True
self.us.connecting_now = datetime.datetime.now()
self.timer.start(self.timer_period)
def stop_share(self):
self.stop_share_button.setEnabled(False)
#self.debug_log.addItem("Stop Share!")
self.timer.stop()
self.us.lsb_url = 'nourl'
self.us.urllib_result = ''
self.us.disconnecting = True
self.qaction.setIcon(get_icon('images/icon.png'))
self.kill_servers_thread.start()
self.timer.start(self.timer_period)
def check_and_render(self):
#self.show_debug()
if self.us.button_state == "start":
self.stop_share_button.hide()
self.lets_share_button.show()
self.lets_share_button.setText(self.us.share_button_text)
else:
self.lets_share_button.hide()
self.stop_share_button.show()
self.stop_share_button.setText(self.us.share_button_text)
if self.us.disconnecting:
self.us.share_button_text = "Disconnecting..."
if self.us.lost_connection:
self.us.lsb_url_text = 'Lost connection. Please start sharing again.'
self.us.url_label_tooltip = '<<<< Click on Start sharing button again.'
else:
self.us.lsb_url_text = 'Be a librarian. Share your library.'
self.us.url_label_tooltip = '<<<< Be a librarian. Click on Start sharing button.<<<<'
if self.us.kill_finished:
#self.debug_log.addItem("Let's share connect!")
self.us.button_state = "start"
self.us.share_button_text = "Start sharing"
self.us.disconnecting = False
self.us.kill_finished = False
self.lets_share_button.setEnabled(True)
elif self.us.connecting:
if self.us.connecting_now:
if (datetime.datetime.now() - self.us.connecting_now) > datetime.timedelta(seconds=10):
#self.debug_log.addItem("Timeout!")
self.us.http_error = None
self.us.lost_connection = True
self.us.connecting = False
self.us.connecting_now = None
self.stop_share()
elif self.us.found_url:
self.us.check_finished = False
self.urllib_thread.start()
if self.us.lsb_url == "nourl" and self.us.ssh_proc and sys.platform != "win32":
#self.debug_log.addItem("Wait for Allocated port!")
self.se.seek(0)
result = self.se.readlines()
for line in result:
m = re.match("^Allocated port (.*) for .*", line)
try:
#self.debug_log.addItem(self.us.lsb_url)
self.us.lsb_url = 'https://www{0}.{1}'.format(m.groups()[0], prefs['lsb_server'])
#_dev_self.us.lsb_url = 'http://www{0}.{1}'.format(m.groups()[0], prefs['lsb_server'])
self.us.lsb_url_text = "Go to: {0}".format(self.us.lsb_url)
self.us.url_label_tooltip = 'Copy URL to clipboard and check it out in a browser!'
self.us.http_error = None
self.us.found_url = True
except:
pass
elif self.us.urllib_result == 200:
#self.debug_log.addItem("Finish Connecting State!")
self.se.seek(0)
self.se.truncate()
self.us.share_button_text = "Stop sharing"
self.us.button_state = "stop"
self.stop_share_button.setEnabled(True)
self.us.connecting = False
self.us.connecting_now = None
self.us.found_url = None
elif self.us.http_error and self.us.button_state == "stop":
#self.debug_log.addItem("Error!")
self.us.http_error = None
self.us.lost_connection = True
self.stop_share()
elif self.us.check_finished:
#if self.debug_log.item(self.debug_log.count()-1).text()[:10] == "Finally Ca":
# self.us.debug_counter = self.us.debug_counter + 1
#else:
# self.debug_log.addItem("Finally Called Thread!({0})".format(self.us.debug_counter))
# self.us.debug_counter = 1
self.us.check_finished = False
self.urllib_thread.start()
if self.us.urllib_result == 200 and self.us.button_state == "stop":
self.stop_share_button.setEnabled(True)
if self.us.lsb_url == 'nourl' and self.us.button_state == "start":
self.lets_share_button.setEnabled(True)
self.setWindowTitle("{0} - {1}".format(self.us.window_title, self.us.lsb_url))
self.url_label.setToolTip(self.us.url_label_tooltip)
self.url_label.setText(self.us.lsb_url_text)
def open_url(self):
if self.us.lsb_url == "nourl" and not self.us.http_error:
self.us.url_label_tooltip = '<<<< Be a librarian. Click on Start sharing button.'
self.us.lsb_url_text = '<<<< Be a librarian. Click on Start sharing button.'
else:
self.clip.setText(self.us.lsb_url)
webbrowser.open(str(self.us.lsb_url))
if self.us.lsb_url != "nourl":
self.us.lsb_url_text = "Library at: {0}".format(self.us.lsb_url)
def open_url2(self, url):
self.clip.setText(url)
webbrowser.open(url)
def show_debug(self):
if self.us.debug_item:
self.debug_log.addItem(str(self.us.debug_item))
self.us.debug_item = None
self.debug_log.scrollToBottom()
self.debug_log.repaint()
def closeEvent(self, e):
self.hide()
#self.urllib_thread.stop()
#self.kill_servers_thread.stop()
def config(self):
self.do_user_config(parent=self)
self.label.setText(prefs['lsb_server'])
class MainWidget(QWidget):
def __init__(self):
QWidget.__init__(self)
# define periodic table widget for element selection
self.periodicTableWidget = widgets.PeriodicTableDialog()
# initial molecule Zmatrix (can be empty)
# self.inp = []
self.inp = [['H'],
['O', 1, 0.9],
['O', 2, 1.4, 1, 105.],
['H', 3, 0.9, 2, 105., 1, 120.]]
self.atomList = []
self.highList = []
self.labelList = []
self.fast = False
# define & initialize ZMatModel that will contain Zmatrix data
self.ZMatModel = QStandardItemModel(len(self.inp), 7, self)
self.ZMatTable = QTableView(self)
self.ZMatTable.setModel(self.ZMatModel)
self.ZMatTable.setFixedWidth(325)
#self.ZMatTable.installEventFilter(self)
#self.ZMatModel.installEventFilter(self)
self.ZMatModel.setHorizontalHeaderLabels(['atom','','bond','','angle','','dihedral'])
for j, width in enumerate([40, 22, 65, 22, 65, 22, 65]):
self.ZMatTable.setColumnWidth(j, width)
# populate the ZMatModel
self.populateZMatModel()
#define Menu bar menus and their actions
self.menuBar = QMenuBar(self)
fileMenu = self.menuBar.addMenu('&File')
editMenu = self.menuBar.addMenu('&Edit')
viewMenu = self.menuBar.addMenu('&View')
measureMenu = self.menuBar.addMenu('&Measure')
helpMenu = self.menuBar.addMenu('&Help')
readZmatAction = QAction('&Read &ZMat', self)
readZmatAction.setShortcut('Ctrl+O')
readZmatAction.setStatusTip('Read Zmat from file')
readZmatAction.triggered.connect(self.readZmat)
fileMenu.addAction(readZmatAction)
readXYZAction = QAction('&Read &XYZ', self)
readXYZAction.setShortcut('Ctrl+Shift+O')
readXYZAction.setStatusTip('Read XYZ from file')
readXYZAction.triggered.connect(self.readXYZ)
fileMenu.addAction(readXYZAction)
readGaussianAction = QAction('&Read &Gaussian log', self)
readGaussianAction.setShortcut('Ctrl+G')
readGaussianAction.setStatusTip('Read Gaussian log file')
readGaussianAction.triggered.connect(self.readGaussian)
fileMenu.addAction(readGaussianAction)
writeZmatAction = QAction('&Write &ZMat', self)
writeZmatAction.setShortcut('Ctrl+S')
writeZmatAction.setStatusTip('Write Zmat to file')
writeZmatAction.triggered.connect(self.writeZmat)
fileMenu.addAction(writeZmatAction)
writeXYZAction = QAction('&Write &XYZ', self)
writeXYZAction.setShortcut('Ctrl+Shift+S')
writeXYZAction.setStatusTip('Write XYZ from file')
writeXYZAction.triggered.connect(self.writeXYZ)
fileMenu.addAction(writeXYZAction)
exitAction = QAction('&Exit', self)
exitAction.setShortcut('Ctrl+Q')
exitAction.setStatusTip('Exit application')
exitAction.triggered.connect(qApp.quit)
fileMenu.addAction(exitAction)
addRowAction = QAction('&Add &row', self)
addRowAction.setShortcut('Ctrl+R')
addRowAction.setStatusTip('Add row to ZMatrix')
addRowAction.triggered.connect(self.addRow)
editMenu.addAction(addRowAction)
deleteRowAction = QAction('&Delete &row', self)
deleteRowAction.setShortcut('Ctrl+Shift+R')
deleteRowAction.setStatusTip('Delete row from ZMatrix')
deleteRowAction.triggered.connect(self.deleteRow)
editMenu.addAction(deleteRowAction)
addAtomAction = QAction('&Add &atom', self)
addAtomAction.setShortcut('Ctrl+A')
addAtomAction.setStatusTip('Add atom to ZMatrix')
addAtomAction.triggered.connect(self.buildB)
editMenu.addAction(addAtomAction)
drawModeMenu = QMenu('Draw mode', self)
viewMenu.addMenu(drawModeMenu)
fastDrawAction = QAction('&Fast draw', self)
fastDrawAction.triggered.connect(self.fastDraw)
normalDrawAction = QAction('&Normal draw', self)
normalDrawAction.triggered.connect(self.normalDraw)
drawModeMenu.addAction(normalDrawAction)
drawModeMenu.addAction(fastDrawAction)
clearHighlightsAction = QAction('&Clear selection', self)
clearHighlightsAction.setShortcut('Ctrl+C')
clearHighlightsAction.setStatusTip('Clear highlighted atoms')
clearHighlightsAction.triggered.connect(self.clearHighlights)
viewMenu.addAction(clearHighlightsAction)
clearLabelsAction = QAction('&Clear labels', self)
clearLabelsAction.setShortcut('Ctrl+Alt+C')
clearLabelsAction.setStatusTip('Clear labels')
clearLabelsAction.triggered.connect(self.clearLabels)
viewMenu.addAction(clearLabelsAction)
clearUpdateViewAction = QAction('&Clear selection and labels', self)
clearUpdateViewAction.setShortcut('Ctrl+Shift+C')
clearUpdateViewAction.setStatusTip('Clear highlighted atoms and labels')
clearUpdateViewAction.triggered.connect(self.clearUpdateView)
viewMenu.addAction(clearUpdateViewAction)
self.showGaussAction = QAction('Show &Gaussian geometry optimization', self)
self.showGaussAction.setShortcut('Ctrl+G')
self.showGaussAction.setStatusTip('Show Gaussian geometry optimization plots for energy, force and displacement.')
self.showGaussAction.setEnabled(False)
self.showGaussAction.triggered.connect(self.showGauss)
viewMenu.addAction(self.showGaussAction)
self.showFreqAction = QAction('Show &IR frequency plot', self)
self.showFreqAction.setShortcut('Ctrl+I')
self.showFreqAction.setStatusTip('Show Gaussian calculated IR frequency plot.')
self.showFreqAction.setEnabled(False)
self.showFreqAction.triggered.connect(self.showFreq)
viewMenu.addAction(self.showFreqAction)
measureDistanceAction = QAction('&Measure &distance', self)
measureDistanceAction.setShortcut('Ctrl+D')
measureDistanceAction.setStatusTip('Measure distance between two atoms')
measureDistanceAction.triggered.connect(self.measureDistanceB)
measureMenu.addAction(measureDistanceAction)
measureAngleAction = QAction('&Measure &angle', self)
measureAngleAction.setShortcut('Ctrl+Shift+D')
measureAngleAction.setStatusTip('Measure angle between three atoms')
measureAngleAction.triggered.connect(self.measureAngleB)
measureMenu.addAction(measureAngleAction)
aboutAction = QAction('&About', self)
aboutAction.setStatusTip('About this program...')
aboutAction.triggered.connect(self.about)
helpMenu.addAction(aboutAction)
aboutQtAction = QAction('&About Qt', self)
aboutQtAction.setStatusTip('About Qt...')
aboutQtAction.triggered.connect(self.aboutQt)
helpMenu.addAction(aboutQtAction)
# define GL widget that displays the 3D molecule model
self.window = widgets.MyGLView()
self.window.installEventFilter(self)
self.window.setMinimumSize(500, 500)
#self.window.setBackgroundColor((50, 0, 10))
self.updateView()
self.gaussianPlot = GraphicsLayoutWidget()
self.gaussianPlot.resize(750, 250)
self.gaussianPlot.setWindowTitle('Gaussian geometry optimization')
#self.gaussianPlot.setAspectLocked(True)
#self.gaussianPlot.addLayout(rowspan=3, colspan=1)
self.FreqModel = QStandardItemModel(1, 3, self)
self.freqTable = QTableView(self)
self.freqTable.setModel(self.FreqModel)
self.freqTable.setMinimumWidth(240)
self.freqTable.installEventFilter(self)
self.FreqModel.installEventFilter(self)
self.FreqModel.setHorizontalHeaderLabels(['Frequency','IR Intensity','Raman Intensity'])
for j, width in enumerate([80, 80, 80]):
self.freqTable.setColumnWidth(j, width)
self.freqWidget = QWidget()
self.freqWidget.setWindowTitle('IR frequency plot & table')
self.freqWidget.resize(800, 400)
self.freqWidget.layout = QHBoxLayout(self.freqWidget)
self.freqWidget.layout.setSpacing(1)
self.freqWidget.layout.setContentsMargins(1, 1, 1, 1)
self.freqPlot = GraphicsLayoutWidget()
self.freqWidget.layout.addWidget(self.freqPlot)
self.freqWidget.layout.addWidget(self.freqTable)
self.freqTable.clicked.connect(self.freqCellClicked)
# define other application parts
self.statusBar = QStatusBar(self)
self.fileDialog = QFileDialog(self)
# define application layout
self.layout = QVBoxLayout(self)
self.layout.setSpacing(1)
self.layout.setContentsMargins(1, 1, 1, 1)
self.layout1 = QHBoxLayout()
self.layout1.setSpacing(1)
self.layout1.addWidget(self.ZMatTable)
self.layout1.addWidget(self.window)
self.layout.addWidget(self.menuBar)
self.layout.addLayout(self.layout1)
self.layout.addWidget(self.statusBar)
self.adjustSize()
self.setWindowTitle('Moldy')
iconPath = 'icon.png'
icon = QIcon(iconPath)
icon.addFile(iconPath, QSize(16, 16))
icon.addFile(iconPath, QSize(24, 24))
icon.addFile(iconPath, QSize(32, 32))
icon.addFile(iconPath, QSize(48, 48))
icon.addFile(iconPath, QSize(256, 256))
self.setWindowIcon(icon)
# start monitoring changes in the ZMatModel
self.ZMatModel.dataChanged.connect(self.clearUpdateView)
# run and show the application
def run(self):
self.show()
self.ZMatTable.clicked.connect(self.ZMatCellClicked)
qt_app.instance().aboutToQuit.connect(self.deleteGLwidget)
qt_app.exec_()
# fill the ZMatModel with initial data from 'self.inp'
def populateZMatModel(self):
self.ZMatModel.removeRows(0, self.ZMatModel.rowCount())
for i, row in enumerate(self.inp):
for j, cell in enumerate(row):
item = QStandardItem(str(cell))
self.ZMatModel.setItem(i, j, item)
# some cells should not be editable, they are disabled
for i in range(min(len(self.inp), 3)):
for j in range(2*i+1, 7):
self.ZMatModel.setItem(i, j, QStandardItem())
self.ZMatModel.item(i, j).setBackground(QColor(150,150,150))
self.ZMatModel.item(i, j).setFlags(Qt.ItemIsEnabled)
def populateFreqModel(self):
self.FreqModel.removeRows(0, self.FreqModel.rowCount())
for i, row in enumerate(zip(self.vibfreqs, self.vibirs, self.vibramans)):
for j, cell in enumerate(row):
item = QStandardItem(str(cell))
self.FreqModel.setItem(i, j, item)
# add a row to the bottom of the ZMatModel
def addRow(self):
# temporarily stop updating the GL window
self.ZMatModel.dataChanged.disconnect(self.clearUpdateView)
row = self.ZMatModel.rowCount()
self.ZMatModel.insertRow(row)
# some cells should not be editable
if row < 3:
for j in range(2*row+1, 7):
self.ZMatModel.setItem(row, j, QStandardItem())
self.ZMatModel.item(row, j).setBackground(QColor(150,150,150))
self.ZMatModel.item(row, j).setFlags(Qt.ItemIsEnabled)
# restart GL window updating
self.ZMatModel.dataChanged.connect(self.clearUpdateView)
self.statusBar.clearMessage()
self.statusBar.showMessage('Added 1 row.', 3000)
# delete the last row of the ZMatModel
def deleteRow(self):
xyz = [list(vi) for vi in list(v)]
atoms = [str(elements[e]) for e in elems]
oldLen = self.ZMatModel.rowCount()
idxs = sorted(set(idx.row() for idx in self.ZMatTable.selectedIndexes()), reverse=True)
newLen = oldLen - len(idxs)
if newLen == oldLen:
self.ZMatModel.removeRow(self.ZMatModel.rowCount()-1)
else:
self.ZMatModel.dataChanged.disconnect(self.clearUpdateView)
for idx in idxs:
self.ZMatModel.removeRow(idx)
if idx < 3:
for i in range(idx, min(3, newLen)):
for j in range(2*i+1, 7):
self.ZMatModel.setItem(i, j, QStandardItem())
self.ZMatModel.item(i, j).setBackground(QColor(150,150,150))
self.ZMatModel.item(i, j).setFlags(Qt.ItemIsEnabled)
if len(xyz) > idx:
xyz.pop(idx)
atoms.pop(idx)
self.inp = xyz2zmat(xyz, atoms)
self.populateZMatModel()
for i in reversed(self.highList):
self.window.removeItem(i[1])
self.highList = []
self.ZMatModel.dataChanged.connect(self.clearUpdateView)
self.updateView()
self.statusBar.clearMessage()
if idxs:
self.statusBar.showMessage('Deleted row(s): '+str([i+1 for i in idxs]), 3000)
else:
self.statusBar.showMessage('Deleted last row.', 3000)
# show the periodic table widget
def periodicTable(self):
self.statusBar.clearMessage()
self.statusBar.showMessage('Select element from periodic table.')
self.periodicTableWidget.exec_()
selection = self.periodicTableWidget.selection()
return selection
# import molecule with zmatrix coordinates
def readZmat(self):
self.ZMatModel.dataChanged.disconnect(self.clearUpdateView)
filename = self.fileDialog.getOpenFileName(self, 'Open file', expanduser('~'), '*.zmat;;*.*')
self.inp = []
self.populateZMatModel()
if filename:
with open(filename, 'r') as f:
next(f)
next(f)
for row in f:
self.inp.append(row.split())
f.close()
self.populateZMatModel()
self.ZMatModel.dataChanged.connect(self.clearUpdateView)
self.updateView()
self.statusBar.clearMessage()
self.statusBar.showMessage('Read molecule from '+filename+'.', 5000)
self.showGaussAction.setEnabled(False)
self.showFreqAction.setEnabled(False)
# import molecule with xyz coordinates
def readXYZ(self):
self.ZMatModel.dataChanged.disconnect(self.clearUpdateView)
filename = self.fileDialog.getOpenFileName(self, 'Open file', expanduser('~'), '*.xyz;;*.*')
xyz = []
elems = []
self.inp = []
self.populateZMatModel()
if filename:
with open(filename, 'r') as f:
next(f)
next(f)
for row in f:
rs = row.split()
if len(rs) == 4:
elems.append(rs[0])
xyz.append([float(f) for f in rs[1:]])
f.close()
self.inp = xyz2zmat(xyz, elems)
self.populateZMatModel()
#print(elems)
self.ZMatModel.dataChanged.connect(self.clearUpdateView)
self.updateView()
self.statusBar.clearMessage()
self.statusBar.showMessage('Read molecule from '+filename+'.', 5000)
self.showGaussAction.setEnabled(False)
self.showFreqAction.setEnabled(False)
# import Gaussian log file
def readGaussian(self):
global vsShifted
self.ZMatModel.dataChanged.disconnect(self.clearUpdateView)
filename = self.fileDialog.getOpenFileName(self, 'Open file', expanduser('~'), '*.log;;*.*')
if filename:
self.gaussianPlot.clear()
self.inp = []
self.populateZMatModel()
file = ccopen(filename)
data = file.parse().getattributes()
self.natom = data['natom']
self.atomnos = data['atomnos'].tolist()
self.atomsymbols = [ str(elements[e]) for e in self.atomnos ]
self.atomcoords = data['atomcoords'].tolist()
self.scfenergies = data['scfenergies'].tolist()
self.geovalues = data['geovalues'].T.tolist()
self.geotargets = data['geotargets'].tolist()
if 'vibfreqs' in data.keys():
self.vibfreqs = data['vibfreqs']
#print(self.vibfreqs)
self.vibirs = data['vibirs']
#print(self.vibirs)
#print(data.keys())
if 'vibramans' in data.keys():
self.vibramans = data['vibramans']
else:
self.vibramans = [''] * len(self.vibirs)
self.vibdisps = data['vibdisps']
#print(self.vibdisps)
self.inp = xyz2zmat(self.atomcoords[0], self.atomsymbols)
self.populateZMatModel()
titles = ['SCF Energies', 'RMS & Max Forces', 'RMS & Max Displacements']
for i in range(3):
self.gaussianPlot.addPlot(row=1, col=i+1)
plot = self.gaussianPlot.getItem(1, i+1)
plot.setTitle(title=titles[i])
if i == 0:
c = ['c']
x = [0]
y = [self.scfenergies]
else:
c = ['r', 'y']
x = [0, 0]
y = [self.geovalues[2*i-2], self.geovalues[2*i-1]]
targety = [self.geotargets[2*i-2], self.geotargets[2*i-1]]
plot.clear()
plot.maxData = plot.plot(y[0], symbol='o', symbolPen=c[0], symbolBrush=c[0], pen=c[0], symbolSize=5, pxMode=True, antialias=True, autoDownsample=False)
plot.highlight=plot.plot(x, [ yy[0] for yy in y ], symbol='o', symbolPen='w', symbolBrush=None, pen=None, symbolSize=15, pxMode=True, antialias=True, autoDownsample=False)
plot.maxData.sigPointsClicked.connect(self.gausclicked)
if i > 0:
for j in range(2):
plot.addLine(y=np.log10(targety[j]), pen=mkPen((255, 255*j, 0, int(255/2)), width=1))
plot.RMSData=plot.plot(y[1], symbol='o', symbolPen=c[1], symbolBrush=c[1], pen=c[1], symbolSize=5, pxMode=True, antialias=True, autoDownsample=False)
plot.RMSData.sigPointsClicked.connect(self.gausclicked)
plot.setLogMode(y=True)
self.showGauss()
self.updateView()
self.statusBar.clearMessage()
self.statusBar.showMessage('Read molecule from '+filename+'.', 5000)
self.ZMatModel.dataChanged.connect(self.clearUpdateView)
if self.natom:
self.showGaussAction.setEnabled(True)
if 'vibfreqs' in data.keys():
self.showFreqAction.setEnabled(True)
# populate the FreqModel
self.populateFreqModel()
self.freqPlot.clear()
irPlot = self.freqPlot.addPlot(row=1, col=1)
irPlot.clear()
minFreq = np.min(self.vibfreqs)
maxFreq = np.max(self.vibfreqs)
maxInt = np.max(self.vibirs)
x = np.sort(np.concatenate([np.linspace(minFreq-100, maxFreq+100, num=1000), self.vibfreqs]))
y = x*0
for f,i in zip(self.vibfreqs, self.vibirs):
y += lorentzv(x, f, 2*np.pi, i)
#xy = np.array([np.concatenate([x, np.array(self.vibfreqs)]), np.concatenate([y, np.array(self.vibirs)])]).T
#xysort = xy[xy[:,0].argsort()]
irPlot.maxData = irPlot.plot(x, y, antialias=True)
markers = ErrorBarItem(x=self.vibfreqs, y=self.vibirs, top=maxInt/30, bottom=None, pen='r')
irPlot.addItem(markers)
self.showFreq()
#self.vibdisps = np.append(self.vibdisps, [np.mean(self.vibdisps, axis=0)], axis=0)
maxt = 100
vsShifted = np.array([ [ vs + self.vibdisps[i]*np.sin(t*2*np.pi/maxt)/3 for t in range(maxt) ] for i in range(len(self.vibfreqs)) ])
else:
self.showFreqAction.setEnabled(False)
self.freqWidget.hide()
def showGauss(self):
self.gaussianPlot.show()
def showFreq(self):
self.freqWidget.show()
# export Zmatrix to csv
def writeZmat(self):
zm = model2list(self.ZMatModel)
filename = self.fileDialog.getSaveFileName(self, 'Save file', expanduser('~')+'/'+getFormula(list(list(zip(*zm))[0]))+'.zmat', '*.zmat;;*.*')
try:
filename
except NameError:
pass
else:
if filename:
writeOutput(zm, filename)
self.statusBar.clearMessage()
self.statusBar.showMessage('Wrote molecule to '+filename+'.', 5000)
# export XYZ coordinates to csv
def writeXYZ(self):
xyz = []
zm = model2list(self.ZMatModel)
for i in range(len(v)):
xyz.append(np.round(v[i], 7).tolist())
xyz[i][:0] = zm[i][0]
if len(v) > 0:
formula = getFormula(list(list(zip(*xyz))[0]))
else:
formula = 'moldy_output'
filename = self.fileDialog.getSaveFileName(self, 'Save file', expanduser('~')+'/'+formula+'.xyz', '*.xyz;;*.*')
try:
filename
except NameError:
pass
else:
if filename:
writeOutput(xyz, filename)
self.statusBar.clearMessage()
self.statusBar.showMessage('Wrote molecule to '+filename+'.', 5000)
# redraw the 3D molecule in GL widget
def updateView(self):
global r
global c
global v
global vs
global elems
global nelems
data = model2list(self.ZMatModel)
try:
# create a list with element coordinates
v = zmat2xyz(data)
except (AssertionError, IndexError, ZMError):
pass
else:
# clear the screen before redraw
for item in reversed(self.window.items):
self.window.removeItem(item)
# create a second coordinate list 'vs' that is centered in the GL view
self.atomList = []
if len(v) > 0:
shift = np.mean(v, axis=0)
vs = np.add(v, -shift)
elems = [ 1 + next((i for i, sublist in enumerate(colors) if row[0] in sublist), -1) for row in data ]
nelems = len(elems)
# define molecule radii and colors
r = []
c = []
for i in elems:
r.append(elements[i].covalent_radius)
c.append(colors[i-1][-1])
# draw atoms
for i in range(nelems):
addAtom(self.window, i, r, vs, c, fast=self.fast)
self.atomList.append([i, self.window.items[-1]])
#print(self.atomList)
# draw bonds where appropriate
combs = list(itertools.combinations(range(nelems), 2))
bonds = []
for i in combs:
bonds.append(addBond(self.window, i[0], i[1], r, vs, c, fast=self.fast))
if self.fast:
bondedAtoms = set(filter((None).__ne__, flatten(bonds)))
for i in set(range(nelems)) - bondedAtoms:
addUnbonded(self.window, i, vs, c)
self.atomList[i][1]=self.window.items[-1]
#print(self.atomList)
for i in self.highList:
self.window.addItem(i[1])
for i in self.labelList:
self.window.addItem(i)
if len(v) > 1:
maxDim = float('-inf')
for dim in v.T:
span = max(dim)-min(dim)
if span > maxDim:
maxDim = span
else: maxDim = 2
self.window.setCameraPosition(distance=maxDim*1.5+1)
global index
index = 0
def updateFreq(self):
global vsShifted, index, r, c
index += 1
index = index % len(vsShifted[0])
#print(index)
#print(vsShifted[index])
for item in reversed(self.window.items):
self.window.removeItem(item)
for i in range(nelems):
addAtom(self.window, i, r, vsShifted[self.freqIndex, index], c, fast=self.fast)
self.atomList.append([i, self.window.items[-1]])
combs = itertools.combinations(range(nelems), 2)
bonds = []
for i in combs:
bonds.append(addBond(self.window, i[0], i[1], r, vsShifted[self.freqIndex, index], c, fast=self.fast))
if self.fast:
bondedAtoms = set(filter((None).__ne__, flatten(bonds)))
for i in set(range(nelems)) - bondedAtoms:
addUnbonded(self.window, i, vsShifted[self.freqIndex, index], c)
self.atomList[i][1]=self.window.items[-1]
# detect mouse clicks in GL window and process them
def eventFilter(self, obj, event):
if obj == self.window:
if event.type() == event.MouseButtonPress:
itms = obj.itemsAt((event.pos().x()-2, event.pos().y()-2, 4, 4))
if len(itms):
self.highlight(obj, [itms[0]])
elif len(self.atomList) == 0:
self.build()
# also do the default click action
return super(MainWidget, self).eventFilter(obj, event)
def ZMatCellClicked(self):
idxs = sorted(set(idx.row() for idx in self.ZMatTable.selectedIndexes()), reverse=True)
itms = []
if self.highList:
highIdx = list(np.array(self.highList).T[0])
for idx in idxs:
if self.highList and idx in highIdx:
itms.append(self.highList[highIdx.index(idx)][1])
elif len(self.atomList) > idx:
itms.append(self.atomList[idx][1])
self.highlight(self.window, itms)
def freqCellClicked(self):
global vsShifted
self.timer = QTimer()
self.timer.setInterval(30)
self.timer.timeout.connect(self.updateFreq)
idxs = [ idx.row() for idx in self.freqTable.selectedIndexes() ]
if len(idxs) == 1:
self.freqIndex = idxs[0]
self.timer.stop()
self.timer.timeout.connect(self.updateFreq)
try:
self.ZMatModel.dataChanged.disconnect(self.clearUpdateView)
except TypeError:
pass
self.timer.start()
if len(idxs) != 1:
self.timer.stop()
self.freqTable.clearSelection()
self.timer.timeout.disconnect(self.updateFreq)
self.ZMatModel.dataChanged.connect(self.clearUpdateView)
self.clearUpdateView()
def gausclicked(self, item, point):
itemdata = item.scatter.data
points = [ row[7] for row in itemdata ]
idx = points.index(point[0])
for i in range(3):
if i == 0:
x = [idx]
y = [self.scfenergies[idx]]
else:
x = [idx, idx]
y = [self.geovalues[2*i-2][idx], self.geovalues[2*i-1][idx]]
plot = self.gaussianPlot.getItem(1, i+1)
plot.removeItem(plot.highlight)
plot.highlight=plot.plot(x, y, symbol='o', symbolPen='w', symbolBrush=None, pen=None, symbolSize=15, pxMode=True, antialias=True, autoDownsample=False)
self.ZMatModel.dataChanged.disconnect(self.clearUpdateView)
self.inp = []
self.populateZMatModel()
self.inp = xyz2zmat(self.atomcoords[min(idx, len(self.atomcoords)-1)], self.atomsymbols)
self.populateZMatModel()
self.ZMatModel.dataChanged.connect(self.clearUpdateView)
self.updateView()
def highlight(self, obj, itms):
for itm in itms:
idx = next((i for i, sublist in enumerate(self.atomList) if itm in sublist), -1)
#print(idx)
if idx != -1:
addAtom(obj, idx, r, vs, c, opt='highlight', fast=self.fast)
self.highList.append([idx, obj.items[-1]])
self.ZMatTable.selectRow(idx)
idx = next((i for i, sublist in enumerate(self.highList) if itm in sublist), -1)
if idx != -1:
obj.removeItem(self.highList[idx][1])
self.highList.pop(idx)
self.ZMatTable.clearSelection()
self.statusBar.clearMessage()
if len(self.highList) > 0:
idxs = np.asarray(self.highList).T[0]
selected = []
for i in idxs:
selected.append(str(i+1)+str(elements[elems[i]]))
self.statusBar.showMessage('Selected atoms: '+str(selected), 5000)
def buildB(self):
try:
nelems
except NameError:
self.build()
else:
if len(self.highList) <= min(nelems, 3):
diff = min(nelems, 3) - len(self.highList)
if diff != 0:
self.statusBar.clearMessage()
self.statusBar.showMessage('Please select '+str(diff)+' more atom(s).')
else:
self.build()
else:
self.statusBar.clearMessage()
self.statusBar.showMessage('Too many atoms selected.')
def build(self):
selection = self.periodicTable()
row = self.ZMatModel.rowCount()
self.addRow()
self.ZMatModel.dataChanged.disconnect(self.clearUpdateView)
newSymbol = selection[1]
newData = [newSymbol]
if len(self.highList) >= 1:
newBond = round(2.1*gmean([ elements[e].covalent_radius for e in [selection[0], elems[self.highList[0][0]]] ]), 4)
newData.append(self.highList[0][0]+1)
newData.append(newBond)
if len(self.highList) >= 2:
newAngle = 109.4712
newData.append(self.highList[1][0]+1)
newData.append(newAngle)
if len(self.highList) == 3:
newDihedral = 120.
newData.append(self.highList[2][0]+1)
newData.append(newDihedral)
for j, cell in enumerate(newData):
item = QStandardItem(str(cell))
self.ZMatModel.setItem(row, j, item)
self.highList = []
self.ZMatModel.dataChanged.connect(self.clearUpdateView)
self.updateView()
def measureDistanceB(self):
sel = len(self.highList)
if sel <= 2:
if sel < 2:
self.statusBar.clearMessage()
self.statusBar.showMessage('Please select '+str(2-sel)+' more atom(s).')
else:
self.measureDistance()
else:
self.statusBar.clearMessage()
self.statusBar.showMessage('Too many atoms selected.')
def measureDistance(self):
pts = []
for pt in self.highList:
pts.append(vs[pt[0]])
pts = np.array(pts)
self.clearHighlights()
line = gl.GLLinePlotItem(pos=pts, color=(0., 1., 0., 1.), width=3)
self.window.addItem(line)
self.labelList.append(line)
q = pts[1]-pts[0]
dist = round(np.sqrt(np.dot(q, q)), 4)
self.window.labelPos.append(np.mean(pts[0:2], axis=0))
self.window.labelText.append(str(dist))
self.statusBar.clearMessage()
self.statusBar.showMessage('Measured distance: '+str(dist)+' A.', 3000)
def measureAngleB(self):
sel = len(self.highList)
if sel <= 3:
if sel < 3:
self.statusBar.clearMessage()
self.statusBar.showMessage('Please select '+str(3-sel)+' more atom(s).')
else:
self.measureAngle()
else:
self.statusBar.clearMessage()
self.statusBar.showMessage('Too many atoms selected.')
def measureAngle(self):
pts = []
for pt in self.highList:
pts.append(vs[pt[0]])
pts = np.array(pts)
q = pts[1]-pts[0]
r = pts[2]-pts[0]
q_u = q / np.sqrt(np.dot(q, q))
r_u = r / np.sqrt(np.dot(r, r))
angle = round(degrees(acos(np.dot(q_u, r_u))), 1)
srange = np.array([slerp(q, r, t) for t in np.arange(0.0, 13/12, 1/12)])
self.clearHighlights()
for i in range(12):
mesh = gl.MeshData(np.array([[0,0,0],srange[i],srange[i+1]]))
tri = gl.GLMeshItem(meshdata=mesh, smooth=False, computeNormals=False, color=(0.3, 1., 0.3, 0.5), glOptions=('translucent'))
tri.translate(pts[0][0], pts[0][1], pts[0][2])
self.window.addItem(tri)
self.labelList.append(tri)
self.window.labelPos.append(slerp(q, r, 0.5)+pts[0])
self.window.labelText.append(str(angle))
self.statusBar.clearMessage()
self.statusBar.showMessage('Measured angle: '+str(angle)+'°', 3000)
def clearLabels(self):
self.window.labelPos = []
self.window.labelText = []
self.labelList = []
self.updateView()
def clearHighlights(self):
for item in reversed(self.highList):
self.window.removeItem(item[1])
self.highList = []
self.updateView()
def clearUpdateView(self):
self.window.labelPos = []
self.window.labelText = []
self.labelList = []
for item in reversed(self.highList):
self.window.removeItem(item[1])
self.highList = []
self.updateView()
#print(self.highList)
def fastDraw(self):
if not self.fast:
self.fast = True
self.updateView()
def normalDraw(self):
if self.fast:
self.fast = False
self.updateView()
def about(self):
QMessageBox.about(self, 'About moldy', 'moldy beta 15. 9. 2015')
def aboutQt(self):
QMessageBox.aboutQt(self, 'About Qt')
def deleteGLwidget(self):
self.window.setParent(None)
del self.window
class InformationWindow(QWidget):
def __init__(self):
QWidget.__init__(self, ctx.mainScreen)
self.setObjectName("InformationWindow")
self.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
self.setFixedHeight(50)
self.setMaximumWidth(800)
self.setStyleSheet("""
QFrame#frame { border: 1px solid rgba(255,255,255,30);
/*border-radius: 4px;*/
background-color: rgba(0,0,0,100);}
QLabel { border:none;
color:#FFFFFF;}
QProgressBar { border: 1px solid white;}
QProgressBar::chunk { background-color: #F1610D;
width: 0.5px;}
""")
self.gridlayout = QGridLayout(self)
self.frame = QFrame(self)
self.frame.setObjectName("frame")
self.horizontalLayout = QHBoxLayout(self.frame)
self.horizontalLayout.setContentsMargins(10, 0, 10, 0)
# Spinner
self.spinner = QLabel(self.frame)
self.spinner.setMinimumSize(QSize(16, 16))
self.spinner.setMaximumSize(QSize(16, 16))
self.spinner.setIndent(6)
self.movie = QMovie(':/images/working.mng')
self.spinner.setMovie(self.movie)
self.movie.start()
self.horizontalLayout.addWidget(self.spinner)
# Message
self.label = QLabel(self.frame)
self.label.setAlignment(Qt.AlignCenter)
self.label.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
self.icon = QLabel(self.frame)
self.icon.setFixedWidth(16)
self.icon.setFixedHeight(16)
self.horizontalLayout.setSpacing(10)
self.horizontalLayout.addWidget(self.icon)
self.horizontalLayout.addWidget(self.label)
self.gridlayout.addWidget(self.frame,0,0,1,1)
def update(self, message, type=None, spinner=False):
fontMetric = self.label.fontMetrics()
textWidth = fontMetric.width(message)
if type:
self.icon.show()
if type == "error":
self.icon.setPixmap(QPixmap(":/gui/pics/dialog-error.png"))
self.setStyleSheet(" QFrame#frame {background-color: rgba(255,0,0,100);} ")
elif type == "warning":
self.icon.setPixmap(QPixmap(":/gui/pics/dialog-warning.png"))
self.setStyleSheet(" QFrame#frame {background-color: rgba(0,0,0,100);} ")
self.setFixedWidth(textWidth + self.icon.width() + 50)
self.label.setText(message)
else:
self.icon.hide()
self.setStyleSheet(" QFrame#frame {background-color: rgba(0,0,0,100);} ")
self.setFixedWidth(textWidth + self.icon.width() + 100)
self.label.setText(message)
self.spinner.setVisible(spinner)
self.move(ctx.mainScreen.width()/2 - self.width()/2,
ctx.mainScreen.height() - self.height()/2 - 50)
self.show()
def refresh(self):
ctx.mainScreen.processEvents()
def show(self):
QWidget.show(self)
self.refresh()
def hide(self):
QWidget.hide(self)
self.refresh()
class LetsShareBooksDialog(QDialog):
def __init__(self, gui, icon, do_user_config, qaction, us):
QDialog.__init__(self, gui)
self.gui = gui
self.do_user_config = do_user_config
self.qaction = qaction
self.us = us
self.clip = QApplication.clipboard()
self.main_gui = calibre_main()
self.urllib_thread = UrlLibThread(self.us)
self.kill_servers_thread = KillServersThread(self.us)
self.us.check_finished = True
self.pxmp = QPixmap()
self.pxmp.load('images/icon_connected.png')
self.icon_connected = QIcon(self.pxmp)
self.setStyleSheet("""
QDialog {
background-color: white;
}
QPushButton {
font-size: 16px;
border-style: solid;
border-color: red;
font-family:'BitstreamVeraSansMono',Consolas,monospace;
text-transform: uppercase;
}
QPushButton#arrow {
border-width: 16px;
border-right-color:white;
padding: -10px;
color:red;
}
QPushButton#url {
background-color: red;
min-width: 460px;
color: white;
text-align: left;
}
QPushButton#url:hover {
background-color: white;
color: red;
}
QPushButton#share {
background-color: red;
color: white;
margin-right: 10px;
}
QPushButton#share:hover {
background-color: white;
color: red;
}
QPushButton#url2 {
color: #222;
text-align: left;
}
QPushButton#url2:hover {
color: red;
}
""")
self.ll = QVBoxLayout()
#self.ll.setSpacing(1)
self.l = QHBoxLayout()
self.l.setSpacing(0)
self.l.setMargin(0)
#self.l.setContentsMargins(0,0,0,0)
self.w = QWidget()
self.w.setLayout(self.l)
self.setLayout(self.ll)
self.setWindowIcon(icon)
self.lets_share_button = QPushButton()
self.lets_share_button.setSizePolicy(QSizePolicy.MinimumExpanding, QSizePolicy.MinimumExpanding)
self.lets_share_button.setObjectName("share")
self.lets_share_button.clicked.connect(self.lets_share)
self.stop_share_button = QPushButton()
self.stop_share_button.setSizePolicy(QSizePolicy.MinimumExpanding, QSizePolicy.MinimumExpanding)
self.stop_share_button.setObjectName("share")
self.stop_share_button.clicked.connect(self.stop_share)
self.l.addWidget(self.lets_share_button)
self.l.addWidget(self.stop_share_button)
if self.us.button_state == "start":
self.lets_share_button.show()
self.stop_share_button.hide()
self.lets_share_button.setText(self.us.share_button_text)
else:
self.lets_share_button.hide()
self.stop_share_button.show()
self.stop_share_button.setText(self.us.share_button_text)
self.url_label = QPushButton()
self.url_label.setSizePolicy(QSizePolicy.MinimumExpanding, QSizePolicy.MinimumExpanding)
self.url_label.setObjectName("url")
self.url_label.clicked.connect(self.open_url)
self.l.addWidget(self.url_label)
self.arrow_button = QPushButton("_____")
self.arrow_button.setObjectName("arrow")
self.l.addWidget(self.arrow_button)
self.ll.addWidget(self.w)
self.ll.addSpacing(10)
self.chat_button = QPushButton("Chat room: https://chat.memoryoftheworld.org")
#self.chat_button.hovered.connect(self.setCursorToHand)
self.chat_button.setObjectName("url2")
self.chat_button.setToolTip('Meetings every thursday at 23:59 (central eruopean time)')
self.chat_button.clicked.connect(functools.partial(self.open_url2, "https://chat.memoryoftheworld.org"))
self.ll.addWidget(self.chat_button)
self.about_project_button = QPushButton('Public Library: http://www.memoryoftheworld.org')
self.about_project_button.setObjectName("url2")
self.about_project_button.setToolTip('When everyone is librarian, library is everywhere.')
self.about_project_button.clicked.connect(functools.partial(self.open_url2, "http://www.memoryoftheworld.org"))
self.ll.addWidget(self.about_project_button)
self.debug_log = QListWidget()
self.ll.addWidget(self.debug_log)
self.debug_log.addItem("Initiatied!")
self.metadata_thread = MetadataLibThread(self.debug_log)
self.metadata_button = QPushButton("Get library metadata!")
self.metadata_button.setObjectName("url2")
self.metadata_button.setToolTip('Get library metadata!')
self.metadata_button.clicked.connect(self.get_metadata)
self.ll.addWidget(self.metadata_button)
self.upgrade_button = QPushButton('Please download and upgrade from {0} to {1} version of plugin.'.format(self.us.running_version, self.us.latest_version))
self.upgrade_button.setObjectName("url2")
self.upgrade_button.setToolTip('Running latest version you make developers happy')
self.upgrade_button.clicked.connect(functools.partial(self.open_url2, self.us.plugin_url))
version_list = [self.us.running_version, self.us.latest_version]
version_list.sort(key=lambda s: map(int, s.split('.')))
if self.us.running_version != self.us.latest_version:
if self.us.running_version == version_list[0]:
self.ll.addSpacing(20)
self.ll.addWidget(self.upgrade_button)
self.resize(self.sizeHint())
self.se = open("lsb.log", "w+b")
self.so = self.se
sys.stdout = os.fdopen(sys.stdout.fileno(), 'w', 0)
os.dup2(self.so.fileno(), sys.stdout.fileno())
os.dup2(self.se.fileno(), sys.stderr.fileno())
self.timer = QTimer()
self.timer.timeout.connect(self.check_and_render)
self.timer_period = 300
self.timer.start(self.timer_period)
self.error_log = ""
def lets_share(self):
self.lets_share_button.setEnabled(False)
self.timer.stop()
self.us.share_button_text = "Connecting..."
#self.debug_log.addItem("Let's share!")
self.us.counter = 0
self.us.lost_connection = False
if not self.us.ssh_proc:
self.main_gui.start_content_server()
opts, args = server_config().option_parser().parse_args(['calibre-server'])
self.calibre_server_port = opts.port
if sys.platform == "win32":
self.win_reg = subprocess.Popen("regedit /s .hosts.reg")
self.us.win_port = int(random.random()*40000+10000)
self.us.ssh_proc = subprocess.Popen("lsbtunnel.exe -N -T tunnel@{2} -R {0}:localhost:{1} -P 722".format(self.us.win_port, self.calibre_server_port, prefs['lsb_server']), shell=True)
self.us.lsb_url = "https://www{0}.{1}".format(self.us.win_port, prefs['lsb_server'])
#_dev_self.us.lsb_url = "http://www{0}.{1}".format(self.us.win_port, prefs['lsb_server'])
self.us.lsb_url_text = "Go to: {0}".format(self.us.lsb_url)
self.us.found_url = True
else:
self.us.ssh_proc = subprocess.Popen(['ssh', '-T', '-N', '-g', '-o', 'UserKnownHostsFile=.userknownhostsfile', '-o', 'TCPKeepAlive=yes', '-o', 'ServerAliveINterval=60', prefs['lsb_server'], '-l', 'tunnel', '-R', '0:localhost:{0}'.format(self.calibre_server_port), '-p', '722'])
self.us.found_url = None
self.qaction.setIcon(get_icon('images/icon_connected.png'))
self.us.connecting = True
self.us.connecting_now = datetime.datetime.now()
self.timer.start(self.timer_period)
def stop_share(self):
self.stop_share_button.setEnabled(False)
#self.debug_log.addItem("Stop Share!")
self.timer.stop()
self.us.lsb_url = 'nourl'
self.us.urllib_result = ''
self.us.disconnecting = True
self.qaction.setIcon(get_icon('images/icon.png'))
self.kill_servers_thread.start()
self.timer.start(self.timer_period)
def check_and_render(self):
#self.show_debug()
if self.us.button_state == "start":
self.stop_share_button.hide()
self.lets_share_button.show()
self.lets_share_button.setText(self.us.share_button_text)
else:
self.lets_share_button.hide()
self.stop_share_button.show()
self.stop_share_button.setText(self.us.share_button_text)
if self.us.disconnecting:
self.us.share_button_text = "Disconnecting..."
if self.us.lost_connection:
self.us.lsb_url_text = 'Lost connection. Please start sharing again.'
self.us.url_label_tooltip = '<<<< Click on Start sharing button again.'
else:
self.us.lsb_url_text = 'Be a librarian. Share your library.'
self.us.url_label_tooltip = '<<<< Be a librarian. Click on Start sharing button.<<<<'
if self.us.kill_finished:
#self.debug_log.addItem("Let's share connect!")
self.us.button_state = "start"
self.us.share_button_text = "Start sharing"
self.us.disconnecting = False
self.us.kill_finished = False
self.lets_share_button.setEnabled(True)
elif self.us.connecting:
if self.us.connecting_now:
if (datetime.datetime.now() - self.us.connecting_now) > datetime.timedelta(seconds=10):
#self.debug_log.addItem("Timeout!")
self.us.http_error = None
self.us.lost_connection = True
self.us.connecting = False
self.us.connecting_now = None
self.stop_share()
elif self.us.found_url:
self.us.check_finished = False
self.urllib_thread.start()
if self.us.lsb_url == "nourl" and self.us.ssh_proc and sys.platform != "win32":
#self.debug_log.addItem("Wait for Allocated port!")
self.se.seek(0)
result = self.se.readlines()
for line in result:
m = re.match("^Allocated port (.*) for .*", line)
try:
#self.debug_log.addItem(self.us.lsb_url)
self.us.lsb_url = 'https://www{0}.{1}'.format(m.groups()[0], prefs['lsb_server'])
#_dev_self.us.lsb_url = 'http://www{0}.{1}'.format(m.groups()[0], prefs['lsb_server'])
self.us.lsb_url_text = "Go to: {0}".format(self.us.lsb_url)
self.us.url_label_tooltip = 'Copy URL to clipboard and check it out in a browser!'
self.us.http_error = None
self.us.found_url = True
except:
pass
elif self.us.urllib_result == 200:
#self.debug_log.addItem("Finish Connecting State!")
self.se.seek(0)
self.se.truncate()
self.us.share_button_text = "Stop sharing"
self.us.button_state = "stop"
self.stop_share_button.setEnabled(True)
self.us.connecting = False
self.us.connecting_now = None
self.us.found_url = None
elif self.us.http_error and self.us.button_state == "stop":
#self.debug_log.addItem("Error!")
self.us.http_error = None
self.us.lost_connection = True
self.stop_share()
elif self.us.check_finished:
#if self.debug_log.item(self.debug_log.count()-1).text()[:10] == "Finally Ca":
# self.us.debug_counter = self.us.debug_counter + 1
#else:
# self.debug_log.addItem("Finally Called Thread!({0})".format(self.us.debug_counter))
# self.us.debug_counter = 1
self.us.check_finished = False
self.urllib_thread.start()
if self.us.urllib_result == 200 and self.us.button_state == "stop":
self.stop_share_button.setEnabled(True)
if self.us.lsb_url == 'nourl' and self.us.button_state == "start":
self.lets_share_button.setEnabled(True)
self.setWindowTitle("{0} - {1}".format(self.us.window_title, self.us.lsb_url))
self.url_label.setToolTip(self.us.url_label_tooltip)
self.url_label.setText(self.us.lsb_url_text)
def open_url(self):
if self.us.lsb_url == "nourl" and not self.us.http_error:
self.us.url_label_tooltip = '<<<< Be a librarian. Click on Start sharing button.'
self.us.lsb_url_text = '<<<< Be a librarian. Click on Start sharing button.'
else:
self.clip.setText(self.us.lsb_url)
webbrowser.open(str(self.us.lsb_url))
if self.us.lsb_url != "nourl":
self.us.lsb_url_text = "Library at: {0}".format(self.us.lsb_url)
def open_url2(self, url):
self.clip.setText(url)
webbrowser.open(url)
def get_metadata(self):
self.metadata_thread.start()
def show_debug(self):
if self.us.debug_item:
self.debug_log.addItem(str(self.us.debug_item))
self.us.debug_item = None
self.debug_log.scrollToBottom()
self.debug_log.repaint()
def closeEvent(self, e):
self.hide()
#self.urllib_thread.stop()
#self.kill_servers_thread.stop()
def config(self):
self.do_user_config(parent=self)
self.label.setText(prefs['lsb_server'])
class InformationWindow(QWidget):
def __init__(self):
QWidget.__init__(self, ctx.mainScreen)
self.setObjectName("InformationWindow")
self.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
self.setFixedHeight(50)
self.setMaximumWidth(800)
self.setStyleSheet("""
QFrame#frame { border: 1px solid rgba(255,255,255,30);
/*border-radius: 4px;*/
background-color: rgba(0,0,0,100);}
QLabel { border:none;
color:#FFFFFF;}
QProgressBar { border: 1px solid white;}
QProgressBar::chunk { background-color: #F1610D;
width: 0.5px;}
""")
self.gridlayout = QGridLayout(self)
self.frame = QFrame(self)
self.frame.setObjectName("frame")
self.horizontalLayout = QHBoxLayout(self.frame)
self.horizontalLayout.setContentsMargins(10, 0, 10, 0)
# Spinner
self.spinner = QLabel(self.frame)
self.spinner.setMinimumSize(QSize(16, 16))
self.spinner.setMaximumSize(QSize(16, 16))
self.spinner.setIndent(6)
self.movie = QMovie(':/images/working.mng')
self.spinner.setMovie(self.movie)
self.movie.start()
self.horizontalLayout.addWidget(self.spinner)
# Message
self.label = QLabel(self.frame)
self.label.setAlignment(Qt.AlignCenter)
self.label.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
self.icon = QLabel(self.frame)
self.icon.setFixedWidth(16)
self.icon.setFixedHeight(16)
self.horizontalLayout.setSpacing(10)
self.horizontalLayout.addWidget(self.icon)
self.horizontalLayout.addWidget(self.label)
self.gridlayout.addWidget(self.frame, 0, 0, 1, 1)
def update(self, message, type=None, spinner=False):
fontMetric = self.label.fontMetrics()
textWidth = fontMetric.width(message)
if type:
self.icon.show()
if type == "error":
self.icon.setPixmap(QPixmap(":/gui/pics/dialog-error.png"))
self.setStyleSheet(
" QFrame#frame {background-color: rgba(255,0,0,100);} ")
elif type == "warning":
self.icon.setPixmap(QPixmap(":/gui/pics/dialog-warning.png"))
self.setStyleSheet(
" QFrame#frame {background-color: rgba(0,0,0,100);} ")
self.setFixedWidth(textWidth + self.icon.width() + 50)
self.label.setText(message)
else:
self.icon.hide()
self.setStyleSheet(
" QFrame#frame {background-color: rgba(0,0,0,100);} ")
self.setFixedWidth(textWidth + self.icon.width() + 100)
self.label.setText(message)
self.spinner.setVisible(spinner)
self.move(ctx.mainScreen.width() / 2 - self.width() / 2,
ctx.mainScreen.height() - self.height() / 2 - 50)
self.show()
def refresh(self):
ctx.mainScreen.processEvents()
def show(self):
QWidget.show(self)
self.refresh()
def hide(self):
QWidget.hide(self)
self.refresh()
def setupUi(self):
"""
Setup the UI for the command toolbar.
"""
#ninad 070123 : It's important to set the Vertical size policy of the
# cmd toolbar widget. otherwise the flyout QToolbar messes up the
#layout (makes the command toolbar twice as big)
#I have set the vertical policy as fixed. Works fine. There are some
# MainWindow resizing problems for but those are not due to this
#size policy AFAIK
self.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Fixed)
layout_cmdtoolbar = QHBoxLayout(self)
layout_cmdtoolbar.setMargin(2)
layout_cmdtoolbar.setSpacing(2)
#See comment at the top for details about this flag
if DEFINE_CONTROL_AREA_AS_A_QWIDGET:
self.cmdToolbarControlArea = QWidget(self)
else:
self.cmdToolbarControlArea = QToolBar_WikiHelp(self)
self.cmdToolbarControlArea.setAutoFillBackground(True)
self.ctrlAreaPalette = self.getCmdMgrCtrlAreaPalette()
self.cmdToolbarControlArea.setPalette(self.ctrlAreaPalette)
self.cmdToolbarControlArea.setMinimumHeight(62)
self.cmdToolbarControlArea.setMinimumWidth(380)
self.cmdToolbarControlArea.setSizePolicy(QSizePolicy.Fixed,
QSizePolicy.Fixed)
#See comment at the top for details about this flag
if DEFINE_CONTROL_AREA_AS_A_QWIDGET:
layout_controlArea = QHBoxLayout(self.cmdToolbarControlArea)
layout_controlArea.setMargin(0)
layout_controlArea.setSpacing(0)
self.cmdButtonGroup = QButtonGroup()
btn_index = 0
for name in ('Build', 'Insert', 'Tools', 'Move', 'Simulation'):
btn = QToolButton(self.cmdToolbarControlArea)
btn.setObjectName(name)
btn.setMinimumWidth(75)
btn.setMaximumWidth(75)
btn.setMinimumHeight(62)
btn.setAutoRaise(True)
btn.setCheckable(True)
btn.setAutoExclusive(True)
iconpath = "ui/actions/Command Toolbar/ControlArea/" + name + ".png"
btn.setIcon(geticon(iconpath))
btn.setIconSize(QSize(22, 22))
btn.setText(name)
btn.setToolButtonStyle(Qt.ToolButtonTextUnderIcon)
btn.setPalette(self.ctrlAreaPalette)
self.cmdButtonGroup.addButton(btn, btn_index)
btn_index += 1
#See comment at the top for details about this flag
if DEFINE_CONTROL_AREA_AS_A_QWIDGET:
layout_controlArea.addWidget(btn)
else:
self.cmdToolbarControlArea.layout().addWidget(btn)
#following has issues. so not adding widget directly to the
#toolbar. (instead adding it in its layout)-- ninad 070124
##self.cmdToolbarControlArea.addWidget(btn)
layout_cmdtoolbar.addWidget(self.cmdToolbarControlArea)
#Flyout Toolbar in the command toolbar
self.flyoutToolBar = FlyoutToolBar(self)
layout_cmdtoolbar.addWidget(self.flyoutToolBar)
#ninad 070116: Define a spacer item. It will have the exact geometry
# as that of the flyout toolbar. it is added to the command toolbar
# layout only when the Flyout Toolbar is hidden. It is required
# to keep the 'Control Area' widget fixed in its place (otherwise,
#after hiding the flyout toolbar, the layout adjusts the position of
#remaining widget items)
self.spacerItem = QSpacerItem(0,
0,
QtGui.QSizePolicy.Expanding,
QtGui.QSizePolicy.Minimum)
self.spacerItem.setGeometry = self.flyoutToolBar.geometry()
for btn in self.cmdButtonGroup.buttons():
if str(btn.objectName()) == 'Build':
btn.setMenu(self.win.buildStructuresMenu)
btn.setPopupMode(QToolButton.MenuButtonPopup)
btn.setToolTip("Build Commands")
whatsThisTextForCommandToolbarBuildButton(btn)
if str(btn.objectName()) == 'Insert':
btn.setMenu(self.win.insertMenu)
btn.setPopupMode(QToolButton.MenuButtonPopup)
btn.setToolTip("Insert Commands")
whatsThisTextForCommandToolbarInsertButton(btn)
if str(btn.objectName()) == 'Tools':
#fyi: cmd stands for 'command toolbar' - ninad070406
self.win.cmdToolsMenu = QtGui.QMenu(self.win)
self.win.cmdToolsMenu.addAction(self.win.toolsExtrudeAction)
self.win.cmdToolsMenu.addAction(self.win.toolsFuseChunksAction)
self.win.cmdToolsMenu.addSeparator()
self.win.cmdToolsMenu.addAction(self.win.modifyMergeAction)
self.win.cmdToolsMenu.addAction(self.win.modifyMirrorAction)
self.win.cmdToolsMenu.addAction(self.win.modifyInvertAction)
self.win.cmdToolsMenu.addAction(self.win.modifyStretchAction)
btn.setMenu(self.win.cmdToolsMenu)
btn.setPopupMode(QToolButton.MenuButtonPopup)
btn.setToolTip("Tools")
whatsThisTextForCommandToolbarToolsButton(btn)
if str(btn.objectName()) == 'Move':
self.win.moveMenu = QtGui.QMenu(self.win)
self.win.moveMenu.addAction(self.win.toolsMoveMoleculeAction)
self.win.moveMenu.addAction(self.win.rotateComponentsAction)
self.win.moveMenu.addSeparator()
self.win.moveMenu.addAction(
self.win.modifyAlignCommonAxisAction)
##self.win.moveMenu.addAction(\
## self.win.modifyCenterCommonAxisAction)
btn.setMenu(self.win.moveMenu)
btn.setPopupMode(QToolButton.MenuButtonPopup)
btn.setToolTip("Move Commands")
whatsThisTextForCommandToolbarMoveButton(btn)
if str(btn.objectName()) == 'Dimension':
btn.setMenu(self.win.dimensionsMenu)
btn.setPopupMode(QToolButton.MenuButtonPopup)
btn.setToolTip("Dimensioning Commands")
if str(btn.objectName()) == 'Simulation':
btn.setMenu(self.win.simulationMenu)
btn.setPopupMode(QToolButton.MenuButtonPopup)
btn.setToolTip("Simulation Commands")
whatsThisTextForCommandToolbarSimulationButton(btn)
# Convert all "img" tags in the button's "What's This" text
# into abs paths (from their original rel paths).
# Partially fixes bug 2943. --mark 2008-12-07
# [bruce 081209 revised this -- removed mac = False]
fix_QAction_whatsthis(btn)
return
def pmAddTopRowButtons(propMgr, showFlags=pmAllButtons):
"""Creates the OK, Cancel, Preview, and What's This
buttons row at the top of the Property Manager <propMgr>.
<showFlags> is an enum that can be used to show only certain
Property Manager buttons, where:
pmDoneButton = 1
pmCancelButton = 2
pmRestoreDefaultsButton = 4
pmPreviewButton = 8
pmWhatsThisButton = 16
pmAllButtons = 31
These flags are defined in PropMgr_Constants.py.
This subroutine is used by the following Property Managers
(which are still not using the PropMgrBaseClass):
- Build Atoms
- Build Crystal
- Extrude
- Move
- Movie Player
- Fuse Chunks
Note: This subrouting is temporary. It will be removed after the
PropMgrs in this list are converted to the PropMgrBaseClass.
Mark 2007-06-24
"""
# The Top Buttons Row includes the following widgets:
#
# - propMgr.pmTopRowBtns (Hbox Layout containing everything:)
#
# - frame
# - hbox layout "frameHboxLO" (margin=2, spacing=2)
# - Done (OK) button
# - Abort (Cancel) button
# - Restore Defaults button
# - Preview button
# - What's This button
# - right spacer (10x10)
# Main "button group" widget (but it is not a QButtonGroup).
propMgr.pmTopRowBtns = QHBoxLayout()
# Horizontal spacer
HSpacer = QSpacerItem(1, 1,
QSizePolicy.Expanding,
QSizePolicy.Minimum)
# Frame containing all the buttons.
propMgr.TopRowBtnsFrame = QFrame()
propMgr.TopRowBtnsFrame.setFrameShape(QFrame.NoFrame)
propMgr.TopRowBtnsFrame.setFrameShadow(QFrame.Plain)
# Create Hbox layout for main frame.
TopRowBtnsHLayout = QHBoxLayout(propMgr.TopRowBtnsFrame)
TopRowBtnsHLayout.setMargin(pmTopRowBtnsMargin)
TopRowBtnsHLayout.setSpacing(pmTopRowBtnsSpacing)
TopRowBtnsHLayout.addItem(HSpacer)
# Set button type.
buttonType = QToolButton
# May want to use QToolButton.setAutoRaise(1) below. Mark 2007-05-29
# OK (Done) button.
propMgr.done_btn = buttonType(propMgr.TopRowBtnsFrame)
propMgr.done_btn.setIcon(
geticon("ui/actions/Properties Manager/Done.png"))
propMgr.done_btn.setIconSize(QSize(22,22))
propMgr.connect(propMgr.done_btn,
SIGNAL("clicked()"),
propMgr.ok_btn_clicked)
propMgr.done_btn.setToolTip("Done")
TopRowBtnsHLayout.addWidget(propMgr.done_btn)
# Cancel (Abort) button.
propMgr.abort_btn = buttonType(propMgr.TopRowBtnsFrame)
propMgr.abort_btn.setIcon(
geticon("ui/actions/Properties Manager/Abort.png"))
propMgr.abort_btn.setIconSize(QSize(22,22))
propMgr.connect(propMgr.abort_btn,
SIGNAL("clicked()"),
propMgr.abort_btn_clicked)
propMgr.abort_btn.setToolTip("Cancel")
TopRowBtnsHLayout.addWidget(propMgr.abort_btn)
# Restore Defaults button.
propMgr.restore_defaults_btn = buttonType(propMgr.TopRowBtnsFrame)
propMgr.restore_defaults_btn.setIcon(
geticon("ui/actions/Properties Manager/Restore.png"))
propMgr.restore_defaults_btn.setIconSize(QSize(22,22))
propMgr.connect(propMgr.restore_defaults_btn,
SIGNAL("clicked()"),
propMgr.restore_defaults_btn_clicked)
propMgr.restore_defaults_btn.setToolTip("Restore Defaults")
TopRowBtnsHLayout.addWidget(propMgr.restore_defaults_btn)
# Preview (glasses) button.
propMgr.preview_btn = buttonType(propMgr.TopRowBtnsFrame)
propMgr.preview_btn.setIcon(
geticon("ui/actions/Properties Manager/Preview.png"))
propMgr.preview_btn.setIconSize(QSize(22,22))
propMgr.connect(propMgr.preview_btn,
SIGNAL("clicked()"),
propMgr.preview_btn_clicked)
propMgr.preview_btn.setToolTip("Preview")
TopRowBtnsHLayout.addWidget(propMgr.preview_btn)
# What's This (?) button.
propMgr.whatsthis_btn = buttonType(propMgr.TopRowBtnsFrame)
propMgr.whatsthis_btn.setIcon(
geticon("ui/actions/Properties Manager/WhatsThis.png"))
propMgr.whatsthis_btn.setIconSize(QSize(22,22))
propMgr.connect(propMgr.whatsthis_btn,
SIGNAL("clicked()"),
QWhatsThis.enterWhatsThisMode)
propMgr.whatsthis_btn.setToolTip("What\'s This Help")
TopRowBtnsHLayout.addWidget(propMgr.whatsthis_btn)
TopRowBtnsHLayout.addItem(HSpacer)
# Create Button Row
propMgr.pmTopRowBtns.addWidget(propMgr.TopRowBtnsFrame)
propMgr.pmVBoxLayout.addLayout(propMgr.pmTopRowBtns)
# Add What's This for buttons.
propMgr.done_btn.setWhatsThis("""<b>Done</b>
<p><img source=\"ui/actions/Properties Manager/Done.png\"><br>
Completes and/or exits the current command.</p>""")
propMgr.abort_btn.setWhatsThis("""<b>Cancel</b>
<p><img source=\"ui/actions/Properties Manager/Abort.png\"><br>
Cancels the current command.</p>""")
propMgr.restore_defaults_btn.setWhatsThis("""<b>Restore Defaults</b>
<p><img source=\"ui/actions/Properties Manager/Restore.png\"><br>
Restores the defaut values of the Property Manager.</p>""")
propMgr.preview_btn.setWhatsThis("""<b>Preview</b>
<p><img source=\"ui/actions/Properties Manager/Preview.png\"><br>
Preview the structure based on current Property Manager settings.</p>""")
propMgr.whatsthis_btn.setWhatsThis("""<b>What's This</b>
<p><img source=\"ui/actions/Properties Manager/WhatsThis.png\"><br>
Click this option to invoke a small question mark that is attached to the mouse pointer,
then click on an object which you would like more information about.
A pop-up box appears with information about the object you selected.</p>""")
# Hide the buttons that shouldn't be displayed base on <showFlags>.
if not showFlags & pmDoneButton:
propMgr.done_btn.hide()
if not showFlags & pmCancelButton:
propMgr.abort_btn.hide()
if not showFlags & pmRestoreDefaultsButton:
propMgr.restore_defaults_btn.hide()
if not showFlags & pmPreviewButton:
propMgr.preview_btn.hide()
if not showFlags & pmWhatsThisButton:
propMgr.whatsthis_btn.hide()
return
def _updateModel(self, what=SkyModel.UpdateAll, origin=None):
if origin is self or not what & (SkyModel.UpdateTags
| SkyModel.UpdateGroupStyle):
return
model = self.model
self._setting_model = True
# to ignore cellChanged() signals (in valueChanged())
# _item_cb is a dict (with row,col keys) containing the widgets (CheckBoxes ComboBoxes) per each cell
self._item_cb = {}
# lists of "list" and "plot" checkboxes per each grouping (excepting the default grouping); each entry is an (row,col,item) tuple.
# used as argument to self._showControls()
self._list_controls = []
self._plot_controls = []
# list of selection callbacks (to which signals are connected)
self._callbacks = []
# set requisite number of rows,and start filling
self.table.setRowCount(len(model.groupings))
for irow, group in enumerate(model.groupings):
self.table.setItem(irow, 0, QTableWidgetItem(group.name))
if group is model.selgroup:
self._irow_selgroup = irow
# total # source in group: skip for "current"
if group is not model.curgroup:
self.table.setItem(irow, 1, QTableWidgetItem(str(group.total)))
# selection controls: skip for current and selection
if group not in (model.curgroup, model.selgroup):
btns = QWidget()
lo = QHBoxLayout(btns)
lo.setContentsMargins(0, 0, 0, 0)
lo.setSpacing(0)
# make selector buttons (depending on which group we're in)
if group is model.defgroup:
Buttons = (("+", lambda src, grp=group: True,
"select all sources"),
("-", lambda src, grp=group: False,
"unselect all sources"))
else:
Buttons = (
("=", lambda src, grp=group: grp.func(src),
"select only this grouping"),
("+",
lambda src, grp=group: src.selected or grp.func(src),
"add grouping to selection"),
("-", lambda src, grp=group: src.selected and not grp.
func(src), "remove grouping from selection"),
("&&",
lambda src, grp=group: src.selected and grp.func(src),
"intersect selection with grouping"))
lo.addStretch(1)
for label, predicate, tooltip in Buttons:
btn = QToolButton(btns)
btn.setText(label)
btn.setMinimumWidth(24)
btn.setMaximumWidth(24)
btn.setToolTip(tooltip)
lo.addWidget(btn)
# add callback
QObject.connect(
btn, SIGNAL("clicked()"),
self._currier.curry(self.selectSources, predicate))
lo.addStretch(1)
self.table.setCellWidget(irow, 2, btns)
# "list" checkbox (not for current and selected groupings: these are always listed)
if group not in (model.curgroup, model.selgroup):
item = self._makeCheckItem("", group, "show_list")
self.table.setItem(irow, self.ColList, item)
item.setToolTip(
"""<P>If checked, sources in this grouping will be listed in the source table. If un-checked, sources will be
excluded from the table. If partially checked, then the default list/no list setting of "all sources" will be in effect.
</P>""")
# "plot" checkbox (not for the current grouping, since that's always plotted)
if group is not model.curgroup:
item = self._makeCheckItem("", group, "show_plot")
self.table.setItem(irow, self.ColPlot, item)
item.setToolTip(
"""<P>If checked, sources in this grouping will be included in the plot. If un-checked, sources will be
excluded from the plot. If partially checked, then the default plot/no plot setting of "all sources" will be in effect.
</P>""")
# custom style control
# for default, current and selected, this is just a text label
if group is model.defgroup:
item = QTableWidgetItem("default:")
item.setTextAlignment(Qt.AlignRight | Qt.AlignVCenter)
item.setToolTip(
"""<P>This is the default plot style used for all sources for which a custom grouping style is not selected.</P>"""
)
self.table.setItem(irow, self.ColApply, item)
elif group is model.curgroup:
item = QTableWidgetItem("")
item.setTextAlignment(Qt.AlignRight | Qt.AlignVCenter)
item.setToolTip(
"""<P>This is the plot style used for the highlighted source, if any.</P>"""
)
self.table.setItem(irow, self.ColApply, item)
elif group is model.selgroup:
item = QTableWidgetItem("")
item.setTextAlignment(Qt.AlignRight | Qt.AlignVCenter)
item.setToolTip(
"""<P>This is the plot style used for the currently selected sources.</P>"""
)
self.table.setItem(irow, self.ColApply, item)
# for the rest, a combobox with custom priorities
else:
cb = QComboBox()
cb.addItems(["default"] +
["custom %d" % p for p in range(1, 10)])
index = max(0, min(group.style.apply, 9))
# dprint(0,group.name,"apply",index)
cb.setCurrentIndex(index)
QObject.connect(
cb, SIGNAL("activated(int)"),
self._currier.xcurry(self._valueChanged,
(irow, self.ColApply)))
self.table.setCellWidget(irow, self.ColApply, cb)
cb.setToolTip(
"""<P>This controls whether sources within this group are plotted with a customized
plot style. Customized styles have numeric priority; if a source belongs to multiple groups, then
the style with the lowest priority takes precedence.<P>""")
# attribute comboboxes
for icol, attr in self.AttrByCol.items():
# get list of options for this style attribute. If dealing with first grouping (i==0), which is
# the "all sources" grouping, then remove the "default" option (which is always first in the list)
options = PlotStyles.StyleAttributeOptions[attr]
if irow == 0:
options = options[1:]
# make combobox
cb = QComboBox()
cb.addItems(list(map(str, options)))
# the "label" option is also editable
if attr == "label":
cb.setEditable(True)
try:
index = options.index(getattr(group.style, attr))
cb.setCurrentIndex(index)
except ValueError:
cb.setEditText(str(getattr(group.style, attr)))
slot = self._currier.xcurry(self._valueChanged, (irow, icol))
QObject.connect(cb, SIGNAL("activated(int)"), slot)
QObject.connect(cb, SIGNAL("editTextChanged(const QString &)"),
slot)
cb.setEnabled(group is model.defgroup or group.style.apply)
self.table.setCellWidget(irow, icol, cb)
label = attr
if irow:
cb.setToolTip(
"""<P>This is the %s used to plot sources in this group, when a "custom" style for the group
is enabled via the style control.<P>""" % label)
else:
cb.setToolTip(
"<P>This is the default %s used for all sources for which a custom style is not specified below.<P>"
% label)
self.table.resizeColumnsToContents()
# re-enable processing of cellChanged() signals
self._setting_model = False
class PM_FileChooser(QWidget):
"""
The PM_FileChooser widget provides a file chooser widget for a
Property Manager group box. The PM_FileChooser widget is a composite widget
made from 3 other Qt widgets:
- a QLabel
- a QLineEdit and
- a QToolButton (with a "..." text label).
IMAGE(http://www.nanoengineer-1.net/mediawiki/images/e/e2/PM_FileChooser1.jpg)
The user can type the path name of a file into the line edit widget or
select a file using Qt's file (chooser) dialog by clicking the "..."
button. The path name of the selected file will be inserted into the
line edit widget.
The parent must make the following signal-slot connection to be
notified when the user has selected a new file via the file chooser dialog:
self.connect(pmFileChooser.lineEdit, SIGNAL("editingFinished()"), self.mySlotMethod)
@cvar defaultText: The default text (path) of the line edit widget.
@type defaultText: string
@cvar setAsDefault: Determines whether to reset the value of the
lineedit to I{defaultText} when the user clicks
the "Restore Defaults" button.
@type setAsDefault: boolean
@cvar labelWidget: The Qt label widget of this PM widget.
@type labelWidget: U{B{QLabel}<http://doc.trolltech.com/4/qlabel.html>}
@cvar lineEdit: The Qt line edit widget for this PM widget.
@type lineEdit: U{B{QLineEdit}<http://doc.trolltech.com/4/qlineedit.html>}
@cvar browseButton: The Qt tool button widget for this PM widget.
@type browseButton: U{B{QToolButton}<http://doc.trolltech.com/4/qtoolbutton.html>}
"""
defaultText = ""
setAsDefault = True
hidden = False
lineEdit = None
browseButton = None
def __init__(self,
parentWidget,
label='',
labelColumn=0,
text='',
setAsDefault=True,
spanWidth=False,
caption="Choose file",
directory='',
filter="All Files (*.*)"):
"""
Appends a file chooser widget to <parentWidget>, a property manager
group box.
@param parentWidget: the parent group box containing this widget.
@type parentWidget: PM_GroupBox
@param label: The label that appears to the left or right of the
file chooser lineedit (and "Browse" button).
If spanWidth is True, the label will be displayed on
its own row directly above the lineedit (and button).
To suppress the label, set I{label} to an
empty string.
@type label: str
@param labelColumn: The column number of the label in the group box
grid layout. The only valid values are 0 (left
column) and 1 (right column). The default is 0
(left column).
@type labelColumn: int
@param text: initial value of LineEdit widget.
@type text: string
@param setAsDefault: if True, will restore <val> when the
"Restore Defaults" button is clicked.
@type setAsDefault: boolean
@param spanWidth: if True, the widget and its label will span the width
of the group box. Its label will appear directly above
the widget (unless the label is empty) and is left justified.
@type spanWidth: boolean
@param caption: The caption used as the title of the file chooser
dialog. "Choose file" is the default.
@type caption: string
@param directory: The directory that the file chooser dialog should
open in when the "..." button is clicked.
If blank or if directory does not exist,
the current working directory is used.
@type directory: string
@param filter: The file type filters to use for the file chooser dialog.
@type filter: string (a semicolon-separated list of file types)
@see: U{B{QLineEdit}<http://doc.trolltech.com/4/qlineedit.html>}
"""
QWidget.__init__(self)
self.parentWidget = parentWidget
self.label = label
self.labelColumn = labelColumn
self.text = text
self.setAsDefault = setAsDefault
self.spanWidth = spanWidth
self.caption = caption
self.directory = directory
self.filter = filter
if label: # Create this widget's QLabel.
self.labelWidget = QLabel()
self.labelWidget.setText(label)
else:
# Create a dummy attribute for PM_GroupBox to see. This might have
# needed to be fixed in PM_GroupBox, but it was done here to try to
# avoid causing errors in other PM widgets. -Derrick 20080916
self.labelWidget = None
self.lineEdit = QLineEdit()
self.browseButton = QToolButton()
# Create vertical box layout.
self.hBoxLayout = QHBoxLayout(self)
self.hBoxLayout.setMargin(0)
self.hBoxLayout.setSpacing(2)
self.hBoxLayout.insertWidget(-1, self.lineEdit)
self.hBoxLayout.insertWidget(-1, self.browseButton)
# Set (QLineEdit) text
self.setText(text)
# Set browse button text and make signal-slot connection.
self.browseButton.setText("...")
self.connect(self.browseButton, SIGNAL("clicked()"),
self.openFileChooserDialog)
# Set default value
self.defaultText = text
self.setAsDefault = setAsDefault
parentWidget.addPmWidget(self)
return
def setText(self, text):
"""
Set the line edit text.
@param text: The text.
@type text: string
"""
self.lineEdit.setText(text)
self.text = text
return
def openFileChooserDialog(self):
"""
Prompts the user to choose a file from disk and inserts the full path
into the lineEdit widget.
"""
_dir = getDefaultWorkingDirectory()
if self.directory:
if os.path.isdir(self.directory):
_dir = self.directory
fname = QFileDialog.getOpenFileName(self, self.caption, _dir,
self.filter)
if fname:
self.setText(fname)
self.lineEdit.emit(SIGNAL("editingFinished()"))
return
def restoreDefault(self):
"""
Restores the default value.
"""
if self.setAsDefault:
self.setText(self.defaultText)
return
def hide(self):
"""
Hides the lineedit and its label (if it has one).
@see: L{show}
"""
QWidget.hide(self)
if self.labelWidget:
self.labelWidget.hide()
return
def show(self):
"""
Unhides the lineedit and its label (if it has one).
@see: L{hide}
"""
QWidget.show(self)
if self.labelWidget:
self.labelWidget.show()
return
# End of PM_FileChooser ############################
def _updateModel(self, what=SkyModel.UpdateAll, origin=None):
if origin is self or not what & (SkyModel.UpdateTags | SkyModel.UpdateGroupStyle):
return
model = self.model
self._setting_model = True; # to ignore cellChanged() signals (in valueChanged())
# _item_cb is a dict (with row,col keys) containing the widgets (CheckBoxes ComboBoxes) per each cell
self._item_cb = {}
# lists of "list" and "plot" checkboxes per each grouping (excepting the default grouping); each entry is an (row,col,item) tuple.
# used as argument to self._showControls()
self._list_controls = []
self._plot_controls = []
# list of selection callbacks (to which signals are connected)
self._callbacks = []
# set requisite number of rows,and start filling
self.table.setRowCount(len(model.groupings))
for irow, group in enumerate(model.groupings):
self.table.setItem(irow, 0, QTableWidgetItem(group.name))
if group is model.selgroup:
self._irow_selgroup = irow
# total # source in group: skip for "current"
if group is not model.curgroup:
self.table.setItem(irow, 1, QTableWidgetItem(str(group.total)))
# selection controls: skip for current and selection
if group not in (model.curgroup, model.selgroup):
btns = QWidget()
lo = QHBoxLayout(btns)
lo.setContentsMargins(0, 0, 0, 0)
lo.setSpacing(0)
# make selector buttons (depending on which group we're in)
if group is model.defgroup:
Buttons = (
("+", lambda src, grp=group: True, "select all sources"),
("-", lambda src, grp=group: False, "unselect all sources"))
else:
Buttons = (
("=", lambda src, grp=group: grp.func(src), "select only this grouping"),
("+", lambda src, grp=group: src.selected or grp.func(src), "add grouping to selection"),
("-", lambda src, grp=group: src.selected and not grp.func(src),
"remove grouping from selection"),
("&&", lambda src, grp=group: src.selected and grp.func(src),
"intersect selection with grouping"))
lo.addStretch(1)
for label, predicate, tooltip in Buttons:
btn = QToolButton(btns)
btn.setText(label)
btn.setMinimumWidth(24)
btn.setMaximumWidth(24)
btn.setToolTip(tooltip)
lo.addWidget(btn)
# add callback
QObject.connect(btn, SIGNAL("clicked()"), self._currier.curry(self.selectSources, predicate))
lo.addStretch(1)
self.table.setCellWidget(irow, 2, btns)
# "list" checkbox (not for current and selected groupings: these are always listed)
if group not in (model.curgroup, model.selgroup):
item = self._makeCheckItem("", group, "show_list")
self.table.setItem(irow, self.ColList, item)
item.setToolTip("""<P>If checked, sources in this grouping will be listed in the source table. If un-checked, sources will be
excluded from the table. If partially checked, then the default list/no list setting of "all sources" will be in effect.
</P>""")
# "plot" checkbox (not for the current grouping, since that's always plotted)
if group is not model.curgroup:
item = self._makeCheckItem("", group, "show_plot")
self.table.setItem(irow, self.ColPlot, item)
item.setToolTip("""<P>If checked, sources in this grouping will be included in the plot. If un-checked, sources will be
excluded from the plot. If partially checked, then the default plot/no plot setting of "all sources" will be in effect.
</P>""")
# custom style control
# for default, current and selected, this is just a text label
if group is model.defgroup:
item = QTableWidgetItem("default:")
item.setTextAlignment(Qt.AlignRight | Qt.AlignVCenter)
item.setToolTip(
"""<P>This is the default plot style used for all sources for which a custom grouping style is not selected.</P>""")
self.table.setItem(irow, self.ColApply, item)
elif group is model.curgroup:
item = QTableWidgetItem("")
item.setTextAlignment(Qt.AlignRight | Qt.AlignVCenter)
item.setToolTip("""<P>This is the plot style used for the highlighted source, if any.</P>""")
self.table.setItem(irow, self.ColApply, item)
elif group is model.selgroup:
item = QTableWidgetItem("")
item.setTextAlignment(Qt.AlignRight | Qt.AlignVCenter)
item.setToolTip("""<P>This is the plot style used for the currently selected sources.</P>""")
self.table.setItem(irow, self.ColApply, item)
# for the rest, a combobox with custom priorities
else:
cb = QComboBox()
cb.addItems(["default"] + ["custom %d" % p for p in range(1, 10)])
index = max(0, min(group.style.apply, 9))
# dprint(0,group.name,"apply",index)
cb.setCurrentIndex(index)
QObject.connect(cb, SIGNAL("activated(int)"),
self._currier.xcurry(self._valueChanged, (irow, self.ColApply)))
self.table.setCellWidget(irow, self.ColApply, cb)
cb.setToolTip("""<P>This controls whether sources within this group are plotted with a customized
plot style. Customized styles have numeric priority; if a source belongs to multiple groups, then
the style with the lowest priority takes precedence.<P>""")
# attribute comboboxes
for icol, attr in self.AttrByCol.items():
# get list of options for this style attribute. If dealing with first grouping (i==0), which is
# the "all sources" grouping, then remove the "default" option (which is always first in the list)
options = PlotStyles.StyleAttributeOptions[attr]
if irow == 0:
options = options[1:]
# make combobox
cb = QComboBox()
cb.addItems(list(map(str, options)))
# the "label" option is also editable
if attr == "label":
cb.setEditable(True)
try:
index = options.index(getattr(group.style, attr))
cb.setCurrentIndex(index)
except ValueError:
cb.setEditText(str(getattr(group.style, attr)))
slot = self._currier.xcurry(self._valueChanged, (irow, icol))
QObject.connect(cb, SIGNAL("activated(int)"), slot)
QObject.connect(cb, SIGNAL("editTextChanged(const QString &)"), slot)
cb.setEnabled(group is model.defgroup or group.style.apply)
self.table.setCellWidget(irow, icol, cb)
label = attr
if irow:
cb.setToolTip("""<P>This is the %s used to plot sources in this group, when a "custom" style for the group
is enabled via the style control.<P>""" % label)
else:
cb.setToolTip(
"<P>This is the default %s used for all sources for which a custom style is not specified below.<P>" % label)
self.table.resizeColumnsToContents()
# re-enable processing of cellChanged() signals
self._setting_model = False
def pmAddTopRowButtons(propMgr, showFlags=pmAllButtons):
"""Creates the OK, Cancel, Preview, and What's This
buttons row at the top of the Property Manager <propMgr>.
<showFlags> is an enum that can be used to show only certain
Property Manager buttons, where:
pmDoneButton = 1
pmCancelButton = 2
pmRestoreDefaultsButton = 4
pmPreviewButton = 8
pmWhatsThisButton = 16
pmAllButtons = 31
These flags are defined in PropMgr_Constants.py.
This subroutine is used by the following Property Managers
(which are still not using the PropMgrBaseClass):
- Build Atoms
- Build Crystal
- Extrude
- Move
- Movie Player
- Fuse Chunks
Note: This subrouting is temporary. It will be removed after the
PropMgrs in this list are converted to the PropMgrBaseClass.
Mark 2007-06-24
"""
# The Top Buttons Row includes the following widgets:
#
# - propMgr.pmTopRowBtns (Hbox Layout containing everything:)
#
# - frame
# - hbox layout "frameHboxLO" (margin=2, spacing=2)
# - Done (OK) button
# - Abort (Cancel) button
# - Restore Defaults button
# - Preview button
# - What's This button
# - right spacer (10x10)
# Main "button group" widget (but it is not a QButtonGroup).
propMgr.pmTopRowBtns = QHBoxLayout()
# Horizontal spacer
HSpacer = QSpacerItem(1, 1, QSizePolicy.Expanding, QSizePolicy.Minimum)
# Frame containing all the buttons.
propMgr.TopRowBtnsFrame = QFrame()
propMgr.TopRowBtnsFrame.setFrameShape(QFrame.NoFrame)
propMgr.TopRowBtnsFrame.setFrameShadow(QFrame.Plain)
# Create Hbox layout for main frame.
TopRowBtnsHLayout = QHBoxLayout(propMgr.TopRowBtnsFrame)
TopRowBtnsHLayout.setMargin(pmTopRowBtnsMargin)
TopRowBtnsHLayout.setSpacing(pmTopRowBtnsSpacing)
TopRowBtnsHLayout.addItem(HSpacer)
# Set button type.
buttonType = QToolButton
# May want to use QToolButton.setAutoRaise(1) below. Mark 2007-05-29
# OK (Done) button.
propMgr.done_btn = buttonType(propMgr.TopRowBtnsFrame)
propMgr.done_btn.setIcon(geticon("ui/actions/Properties Manager/Done.png"))
propMgr.done_btn.setIconSize(QSize(22, 22))
propMgr.connect(propMgr.done_btn, SIGNAL("clicked()"),
propMgr.ok_btn_clicked)
propMgr.done_btn.setToolTip("Done")
TopRowBtnsHLayout.addWidget(propMgr.done_btn)
# Cancel (Abort) button.
propMgr.abort_btn = buttonType(propMgr.TopRowBtnsFrame)
propMgr.abort_btn.setIcon(
geticon("ui/actions/Properties Manager/Abort.png"))
propMgr.abort_btn.setIconSize(QSize(22, 22))
propMgr.connect(propMgr.abort_btn, SIGNAL("clicked()"),
propMgr.abort_btn_clicked)
propMgr.abort_btn.setToolTip("Cancel")
TopRowBtnsHLayout.addWidget(propMgr.abort_btn)
# Restore Defaults button.
propMgr.restore_defaults_btn = buttonType(propMgr.TopRowBtnsFrame)
propMgr.restore_defaults_btn.setIcon(
geticon("ui/actions/Properties Manager/Restore.png"))
propMgr.restore_defaults_btn.setIconSize(QSize(22, 22))
propMgr.connect(propMgr.restore_defaults_btn, SIGNAL("clicked()"),
propMgr.restore_defaults_btn_clicked)
propMgr.restore_defaults_btn.setToolTip("Restore Defaults")
TopRowBtnsHLayout.addWidget(propMgr.restore_defaults_btn)
# Preview (glasses) button.
propMgr.preview_btn = buttonType(propMgr.TopRowBtnsFrame)
propMgr.preview_btn.setIcon(
geticon("ui/actions/Properties Manager/Preview.png"))
propMgr.preview_btn.setIconSize(QSize(22, 22))
propMgr.connect(propMgr.preview_btn, SIGNAL("clicked()"),
propMgr.preview_btn_clicked)
propMgr.preview_btn.setToolTip("Preview")
TopRowBtnsHLayout.addWidget(propMgr.preview_btn)
# What's This (?) button.
propMgr.whatsthis_btn = buttonType(propMgr.TopRowBtnsFrame)
propMgr.whatsthis_btn.setIcon(
geticon("ui/actions/Properties Manager/WhatsThis.png"))
propMgr.whatsthis_btn.setIconSize(QSize(22, 22))
propMgr.connect(propMgr.whatsthis_btn, SIGNAL("clicked()"),
QWhatsThis.enterWhatsThisMode)
propMgr.whatsthis_btn.setToolTip("What\'s This Help")
TopRowBtnsHLayout.addWidget(propMgr.whatsthis_btn)
TopRowBtnsHLayout.addItem(HSpacer)
# Create Button Row
propMgr.pmTopRowBtns.addWidget(propMgr.TopRowBtnsFrame)
propMgr.pmVBoxLayout.addLayout(propMgr.pmTopRowBtns)
# Add What's This for buttons.
propMgr.done_btn.setWhatsThis("""<b>Done</b>
<p><img source=\"ui/actions/Properties Manager/Done.png\"><br>
Completes and/or exits the current command.</p>""")
propMgr.abort_btn.setWhatsThis("""<b>Cancel</b>
<p><img source=\"ui/actions/Properties Manager/Abort.png\"><br>
Cancels the current command.</p>""")
propMgr.restore_defaults_btn.setWhatsThis("""<b>Restore Defaults</b>
<p><img source=\"ui/actions/Properties Manager/Restore.png\"><br>
Restores the defaut values of the Property Manager.</p>""")
propMgr.preview_btn.setWhatsThis("""<b>Preview</b>
<p><img source=\"ui/actions/Properties Manager/Preview.png\"><br>
Preview the structure based on current Property Manager settings.</p>"""
)
propMgr.whatsthis_btn.setWhatsThis("""<b>What's This</b>
<p><img source=\"ui/actions/Properties Manager/WhatsThis.png\"><br>
Click this option to invoke a small question mark that is attached to the mouse pointer,
then click on an object which you would like more information about.
A pop-up box appears with information about the object you selected.</p>"""
)
# Hide the buttons that shouldn't be displayed base on <showFlags>.
if not showFlags & pmDoneButton:
propMgr.done_btn.hide()
if not showFlags & pmCancelButton:
propMgr.abort_btn.hide()
if not showFlags & pmRestoreDefaultsButton:
propMgr.restore_defaults_btn.hide()
if not showFlags & pmPreviewButton:
propMgr.preview_btn.hide()
if not showFlags & pmWhatsThisButton:
propMgr.whatsthis_btn.hide()
return
def __init__(self, parent, rc, imgman):
"""An ImageControlDialog is initialized with a parent widget, a RenderControl object,
and an ImageManager object"""
QDialog.__init__(self, parent)
image = rc.image
self.setWindowTitle("%s: Colour Controls" % image.name)
self.setWindowIcon(pixmaps.colours.icon())
self.setModal(False)
self.image = image
self._rc = rc
self._imgman = imgman
self._currier = PersistentCurrier()
# init internal state
self._prev_range = self._display_range = None, None
self._hist = None
self._geometry = None
# create layouts
lo0 = QVBoxLayout(self)
# lo0.setContentsMargins(0,0,0,0)
# histogram plot
whide = self.makeButton("Hide", self.hide, width=128)
whide.setShortcut(Qt.Key_F9)
lo0.addWidget(Separator(self, "Histogram and ITF", extra_widgets=[whide]))
lo1 = QHBoxLayout()
lo1.setContentsMargins(0, 0, 0, 0)
self._histplot = QwtPlot(self)
self._histplot.setAutoDelete(False)
lo1.addWidget(self._histplot, 1)
lo2 = QHBoxLayout()
lo2.setContentsMargins(0, 0, 0, 0)
lo2.setSpacing(2)
lo0.addLayout(lo2)
lo0.addLayout(lo1)
self._wautozoom = QCheckBox("autozoom", self)
self._wautozoom.setChecked(True)
self._wautozoom.setToolTip("""<P>If checked, then the histrogram plot will zoom in automatically when
you narrow the current intensity range.</P>""")
self._wlogy = QCheckBox("log Y", self)
self._wlogy.setChecked(True)
self._ylogscale = True
self._wlogy.setToolTip(
"""<P>If checked, a log-scale Y axis is used for the histogram plot instead of a linear one.""")
QObject.connect(self._wlogy, SIGNAL("toggled(bool)"), self._setHistLogScale)
self._whistunzoom = self.makeButton("", self._unzoomHistogram, icon=pixmaps.full_range.icon())
self._whistzoomout = self.makeButton("-", self._currier.curry(self._zoomHistogramByFactor, math.sqrt(.1)))
self._whistzoomin = self.makeButton("+", self._currier.curry(self._zoomHistogramByFactor, math.sqrt(10)))
self._whistzoomin.setToolTip("""<P>Click to zoom into the histogram plot by one step. This does not
change the current intensity range.</P>""")
self._whistzoomout.setToolTip("""<P>Click to zoom out of the histogram plot by one step. This does not
change the current intensity range.</P>""")
self._whistunzoom.setToolTip("""<P>Click to reset the histogram plot back to its full extent.
This does not change the current intensity range.</P>""")
self._whistzoom = QwtWheel(self)
self._whistzoom.setOrientation(Qt.Horizontal)
self._whistzoom.setMaximumWidth(80)
self._whistzoom.setRange(10, 0)
self._whistzoom.setStep(0.1)
self._whistzoom.setTickCnt(30)
self._whistzoom.setTracking(False)
QObject.connect(self._whistzoom, SIGNAL("valueChanged(double)"), self._zoomHistogramFinalize)
QObject.connect(self._whistzoom, SIGNAL("sliderMoved(double)"), self._zoomHistogramPreview)
self._whistzoom.setToolTip("""<P>Use this wheel control to zoom in/out of the histogram plot.
This does not change the current intensity range.
Note that the zoom wheel should also respond to your mouse wheel, if you have one.</P>""")
# This works around a stupid bug in QwtSliders -- when using the mousewheel, only sliderMoved() signals are emitted,
# with no final valueChanged(). If we want to do a fast preview of something on sliderMoved(), and a "slow" final
# step on valueChanged(), we're in trouble. So we start a timer on sliderMoved(), and if the timer expires without
# anything else happening, do a valueChanged().
# Here we use a timer to call zoomHistogramFinalize() w/o an argument.
self._whistzoom_timer = QTimer(self)
self._whistzoom_timer.setSingleShot(True)
self._whistzoom_timer.setInterval(500)
QObject.connect(self._whistzoom_timer, SIGNAL("timeout()"), self._zoomHistogramFinalize)
# set same size for all buttons and controls
width = 24
for w in self._whistunzoom, self._whistzoomin, self._whistzoomout:
w.setMinimumSize(width, width)
w.setMaximumSize(width, width)
self._whistzoom.setMinimumSize(80, width)
self._wlab_histpos_text = "(hover here for help)"
self._wlab_histpos = QLabel(self._wlab_histpos_text, self)
self._wlab_histpos.setToolTip("""
<P>The plot shows a histogram of either the full image or its selected subset
(as per the "Data subset" section below).</P>
<P>The current intensity range is indicated by the grey box
in the plot.</P>
<P>Use the left mouse button to change the low intensity limit, and the right
button (on Macs, use Ctrl-click) to change the high limit.</P>
<P>Use Shift with the left mouse button to zoom into an area of the histogram,
or else use the "zoom wheel" control or the plus/minus toolbuttons above the histogram to zoom in or out.
To zoom back out to the full extent of the histogram, click on the rightmost button above the histogram.</P>
""")
lo2.addWidget(self._wlab_histpos, 1)
lo2.addWidget(self._wautozoom)
lo2.addWidget(self._wlogy, 0)
lo2.addWidget(self._whistzoomin, 0)
lo2.addWidget(self._whistzoom, 0)
lo2.addWidget(self._whistzoomout, 0)
lo2.addWidget(self._whistunzoom, 0)
self._zooming_histogram = False
sliced_axes = rc.slicedAxes()
dprint(1, "sliced axes are", sliced_axes)
self._stokes_axis = None
# subset indication
lo0.addWidget(Separator(self, "Data subset"))
# sliced axis selectors
self._wslicers = []
if sliced_axes:
lo1 = QHBoxLayout()
lo1.setContentsMargins(0, 0, 0, 0)
lo1.setSpacing(2)
lo0.addLayout(lo1)
lo1.addWidget(QLabel("Current slice: ", self))
for i, (iextra, name, labels) in enumerate(sliced_axes):
lo1.addWidget(QLabel("%s:" % name, self))
if name == "STOKES":
self._stokes_axis = iextra
# add controls
wslicer = QComboBox(self)
self._wslicers.append(wslicer)
wslicer.addItems(labels)
wslicer.setToolTip("""<P>Selects current slice along the %s axis.</P>""" % name)
wslicer.setCurrentIndex(self._rc.currentSlice()[iextra])
QObject.connect(wslicer, SIGNAL("activated(int)"), self._currier.curry(self._rc.changeSlice, iextra))
lo2 = QVBoxLayout()
lo1.addLayout(lo2)
lo2.setContentsMargins(0, 0, 0, 0)
lo2.setSpacing(0)
wminus = QToolButton(self)
wminus.setArrowType(Qt.UpArrow)
QObject.connect(wminus, SIGNAL("clicked()"), self._currier.curry(self._rc.incrementSlice, iextra, 1))
if i == 0:
wminus.setShortcut(Qt.SHIFT + Qt.Key_F7)
elif i == 1:
wminus.setShortcut(Qt.SHIFT + Qt.Key_F8)
wplus = QToolButton(self)
wplus.setArrowType(Qt.DownArrow)
QObject.connect(wplus, SIGNAL("clicked()"), self._currier.curry(self._rc.incrementSlice, iextra, -1))
if i == 0:
wplus.setShortcut(Qt.Key_F7)
elif i == 1:
wplus.setShortcut(Qt.Key_F8)
wminus.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Fixed)
wplus.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Fixed)
sz = QSize(12, 8)
wminus.setMinimumSize(sz)
wplus.setMinimumSize(sz)
wminus.resize(sz)
wplus.resize(sz)
lo2.addWidget(wminus)
lo2.addWidget(wplus)
lo1.addWidget(wslicer)
lo1.addSpacing(5)
lo1.addStretch(1)
# subset indicator
lo1 = QHBoxLayout()
lo1.setContentsMargins(0, 0, 0, 0)
lo1.setSpacing(2)
lo0.addLayout(lo1)
self._wlab_subset = QLabel("Subset: xxx", self)
self._wlab_subset.setToolTip("""<P>This indicates the current data subset to which the histogram
and the stats given here apply. Use the "Reset to" control on the right to change the
current subset and recompute the histogram and stats.</P>""")
lo1.addWidget(self._wlab_subset, 1)
self._wreset_full = self.makeButton("\u2192 full", self._rc.setFullSubset)
lo1.addWidget(self._wreset_full)
if sliced_axes:
# if self._stokes_axis is not None and len(sliced_axes)>1:
# self._wreset_stokes = self.makeButton(u"\u21920Stokes",self._rc.setFullSubset)
self._wreset_slice = self.makeButton("\u2192 slice", self._rc.setSliceSubset)
lo1.addWidget(self._wreset_slice)
else:
self._wreset_slice = None
# min/max controls
lo1 = QHBoxLayout()
lo1.setContentsMargins(0, 0, 0, 0)
lo0.addLayout(lo1, 0)
self._wlab_stats = QLabel(self)
lo1.addWidget(self._wlab_stats, 0)
self._wmore_stats = self.makeButton("more...", self._showMeanStd)
self._wlab_stats.setMinimumHeight(self._wmore_stats.height())
lo1.addWidget(self._wmore_stats, 0)
lo1.addStretch(1)
# intensity controls
lo0.addWidget(Separator(self, "Intensity mapping"))
lo1 = QHBoxLayout()
lo1.setContentsMargins(0, 0, 0, 0)
lo1.setSpacing(2)
lo0.addLayout(lo1, 0)
self._range_validator = FloatValidator(self)
self._wrange = QLineEdit(self), QLineEdit(self)
self._wrange[0].setToolTip("""<P>This is the low end of the intensity range.</P>""")
self._wrange[1].setToolTip("""<P>This is the high end of the intensity range.</P>""")
for w in self._wrange:
w.setValidator(self._range_validator)
QObject.connect(w, SIGNAL("editingFinished()"), self._changeDisplayRange)
lo1.addWidget(QLabel("low:", self), 0)
lo1.addWidget(self._wrange[0], 1)
self._wrangeleft0 = self.makeButton("\u21920", self._setZeroLeftLimit, width=32)
self._wrangeleft0.setToolTip("""<P>Click this to set the low end of the intensity range to 0.</P>""")
lo1.addWidget(self._wrangeleft0, 0)
lo1.addSpacing(8)
lo1.addWidget(QLabel("high:", self), 0)
lo1.addWidget(self._wrange[1], 1)
lo1.addSpacing(8)
self._wrange_full = self.makeButton(None, self._setHistDisplayRange, icon=pixmaps.intensity_graph.icon())
lo1.addWidget(self._wrange_full)
self._wrange_full.setToolTip(
"""<P>Click this to reset the intensity range to the current extent of the histogram plot.</P>""")
# add menu for display range
range_menu = QMenu(self)
wrange_menu = QToolButton(self)
wrange_menu.setText("Reset to")
wrange_menu.setToolTip("""<P>Use this to reset the intensity range to various pre-defined settings.</P>""")
lo1.addWidget(wrange_menu)
self._qa_range_full = range_menu.addAction(pixmaps.full_range.icon(), "Full subset",
self._rc.resetSubsetDisplayRange)
self._qa_range_hist = range_menu.addAction(pixmaps.intensity_graph.icon(), "Current histogram limits",
self._setHistDisplayRange)
for percent in (99.99, 99.9, 99.5, 99, 98, 95):
range_menu.addAction("%g%%" % percent, self._currier.curry(self._changeDisplayRangeToPercent, percent))
wrange_menu.setMenu(range_menu)
wrange_menu.setPopupMode(QToolButton.InstantPopup)
lo1 = QGridLayout()
lo1.setContentsMargins(0, 0, 0, 0)
lo0.addLayout(lo1, 0)
self._wimap = QComboBox(self)
lo1.addWidget(QLabel("Intensity policy:", self), 0, 0)
lo1.addWidget(self._wimap, 1, 0)
self._wimap.addItems(rc.getIntensityMapNames())
QObject.connect(self._wimap, SIGNAL("currentIndexChanged(int)"), self._rc.setIntensityMapNumber)
self._wimap.setToolTip("""<P>Use this to change the type of the intensity transfer function (ITF).</P>""")
# log cycles control
lo1.setColumnStretch(1, 1)
self._wlogcycles_label = QLabel("Log cycles: ", self)
lo1.addWidget(self._wlogcycles_label, 0, 1)
# self._wlogcycles = QwtWheel(self)
# self._wlogcycles.setTotalAngle(360)
self._wlogcycles = QwtSlider(self)
self._wlogcycles.setToolTip(
"""<P>Use this to change the log-base for the logarithmic intensity transfer function (ITF).</P>""")
# This works around a stupid bug in QwtSliders -- see comments on histogram zoom wheel above
self._wlogcycles_timer = QTimer(self)
self._wlogcycles_timer.setSingleShot(True)
self._wlogcycles_timer.setInterval(500)
QObject.connect(self._wlogcycles_timer, SIGNAL("timeout()"), self._setIntensityLogCycles)
lo1.addWidget(self._wlogcycles, 1, 1)
self._wlogcycles.setRange(1., 10)
self._wlogcycles.setStep(0.1)
self._wlogcycles.setTracking(False)
QObject.connect(self._wlogcycles, SIGNAL("valueChanged(double)"), self._setIntensityLogCycles)
QObject.connect(self._wlogcycles, SIGNAL("sliderMoved(double)"), self._previewIntensityLogCycles)
self._updating_imap = False
# lock intensity map
lo1 = QHBoxLayout()
lo1.setContentsMargins(0, 0, 0, 0)
lo0.addLayout(lo1, 0)
# lo1.addWidget(QLabel("Lock range accross",self))
wlock = QCheckBox("Lock display range", self)
wlock.setToolTip("""<P>If checked, then the intensity range will be locked. The ranges of all locked images
change simultaneously.</P>""")
lo1.addWidget(wlock)
wlockall = QToolButton(self)
wlockall.setIcon(pixmaps.locked.icon())
wlockall.setText("Lock all to this")
wlockall.setToolButtonStyle(Qt.ToolButtonTextBesideIcon)
wlockall.setAutoRaise(True)
wlockall.setToolTip("""<P>Click this to lock together the intensity ranges of all images.</P>""")
lo1.addWidget(wlockall)
wunlockall = QToolButton(self)
wunlockall.setIcon(pixmaps.unlocked.icon())
wunlockall.setText("Unlock all")
wunlockall.setToolButtonStyle(Qt.ToolButtonTextBesideIcon)
wunlockall.setAutoRaise(True)
wunlockall.setToolTip("""<P>Click this to unlock the intensity ranges of all images.</P>""")
lo1.addWidget(wunlockall)
wlock.setChecked(self._rc.isDisplayRangeLocked())
QObject.connect(wlock, SIGNAL("clicked(bool)"), self._rc.lockDisplayRange)
QObject.connect(wlockall, SIGNAL("clicked()"),
self._currier.curry(self._imgman.lockAllDisplayRanges, self._rc))
QObject.connect(wunlockall, SIGNAL("clicked()"), self._imgman.unlockAllDisplayRanges)
QObject.connect(self._rc, SIGNAL("displayRangeLocked"), wlock.setChecked)
# self._wlock_imap_axis = [ QCheckBox(name,self) for iaxis,name,labels in sliced_axes ]
# for iw,w in enumerate(self._wlock_imap_axis):
# QObject.connect(w,SIGNAL("toggled(bool)"),self._currier.curry(self._rc.lockDisplayRangeForAxis,iw))
# lo1.addWidget(w,0)
lo1.addStretch(1)
# lo0.addWidget(Separator(self,"Colourmap"))
# color bar
self._colorbar = QwtPlot(self)
lo0.addWidget(self._colorbar)
self._colorbar.setAutoDelete(False)
self._colorbar.setMinimumHeight(32)
self._colorbar.enableAxis(QwtPlot.yLeft, False)
self._colorbar.enableAxis(QwtPlot.xBottom, False)
# color plot
self._colorplot = QwtPlot(self)
lo0.addWidget(self._colorplot)
self._colorplot.setAutoDelete(False)
self._colorplot.setMinimumHeight(64)
self._colorplot.enableAxis(QwtPlot.yLeft, False)
self._colorplot.enableAxis(QwtPlot.xBottom, False)
# self._colorplot.setSizePolicy(QSizePolicy.Expanding,QSizePolicy.Preferred)
self._colorbar.hide()
self._colorplot.hide()
# color controls
lo1 = QHBoxLayout()
lo1.setContentsMargins(0, 0, 0, 0)
lo0.addLayout(lo1, 1)
lo1.addWidget(QLabel("Colourmap:", self))
# colormap list
### NB: use setIconSize() and icons in QComboBox!!!
self._wcolmaps = QComboBox(self)
self._wcolmaps.setIconSize(QSize(128, 16))
self._wcolmaps.setToolTip("""<P>Use this to select a different colourmap.</P>""")
for cmap in self._rc.getColormapList():
self._wcolmaps.addItem(QIcon(cmap.makeQPixmap(128, 16)), cmap.name)
lo1.addWidget(self._wcolmaps)
QObject.connect(self._wcolmaps, SIGNAL("activated(int)"), self._rc.setColorMapNumber)
# add widgetstack for colormap controls
self._wcolmap_control_stack = QStackedWidget(self)
self._wcolmap_control_blank = QWidget(self._wcolmap_control_stack)
self._wcolmap_control_stack.addWidget(self._wcolmap_control_blank)
lo0.addWidget(self._wcolmap_control_stack)
self._colmap_controls = []
# add controls to stack
for index, cmap in enumerate(self._rc.getColormapList()):
if isinstance(cmap, Colormaps.ColormapWithControls):
controls = cmap.makeControlWidgets(self._wcolmap_control_stack)
self._wcolmap_control_stack.addWidget(controls)
QObject.connect(cmap, SIGNAL("colormapChanged"),
self._currier.curry(self._previewColormapParameters, index, cmap))
QObject.connect(cmap, SIGNAL("colormapPreviewed"),
self._currier.curry(self._previewColormapParameters, index, cmap))
self._colmap_controls.append(controls)
else:
self._colmap_controls.append(self._wcolmap_control_blank)
# connect updates from renderControl and image
self.image.connect(SIGNAL("slice"), self._updateImageSlice)
QObject.connect(self._rc, SIGNAL("intensityMapChanged"), self._updateIntensityMap)
QObject.connect(self._rc, SIGNAL("colorMapChanged"), self._updateColorMap)
QObject.connect(self._rc, SIGNAL("dataSubsetChanged"), self._updateDataSubset)
QObject.connect(self._rc, SIGNAL("displayRangeChanged"), self._updateDisplayRange)
# update widgets
self._setupHistogramPlot()
self._updateDataSubset(*self._rc.currentSubset())
self._updateColorMap(image.colorMap())
self._updateIntensityMap(rc.currentIntensityMap(), rc.currentIntensityMapNumber())
self._updateDisplayRange(*self._rc.displayRange())
class PM_FileChooser( QWidget ):
"""
The PM_FileChooser widget provides a file chooser widget for a
Property Manager group box. The PM_FileChooser widget is a composite widget
made from 3 other Qt widgets:
- a QLabel
- a QLineEdit and
- a QToolButton (with a "..." text label).
IMAGE(http://www.nanoengineer-1.net/mediawiki/images/e/e2/PM_FileChooser1.jpg)
The user can type the path name of a file into the line edit widget or
select a file using Qt's file (chooser) dialog by clicking the "..."
button. The path name of the selected file will be inserted into the
line edit widget.
The parent must make the following signal-slot connection to be
notified when the user has selected a new file via the file chooser dialog:
self.connect(pmFileChooser.lineEdit, SIGNAL("editingFinished()"), self.mySlotMethod)
@cvar defaultText: The default text (path) of the line edit widget.
@type defaultText: string
@cvar setAsDefault: Determines whether to reset the value of the
lineedit to I{defaultText} when the user clicks
the "Restore Defaults" button.
@type setAsDefault: boolean
@cvar labelWidget: The Qt label widget of this PM widget.
@type labelWidget: U{B{QLabel}<http://doc.trolltech.com/4/qlabel.html>}
@cvar lineEdit: The Qt line edit widget for this PM widget.
@type lineEdit: U{B{QLineEdit}<http://doc.trolltech.com/4/qlineedit.html>}
@cvar browseButton: The Qt tool button widget for this PM widget.
@type browseButton: U{B{QToolButton}<http://doc.trolltech.com/4/qtoolbutton.html>}
"""
defaultText = ""
setAsDefault = True
hidden = False
lineEdit = None
browseButton = None
def __init__(self,
parentWidget,
label = '',
labelColumn = 0,
text = '',
setAsDefault = True,
spanWidth = False,
caption = "Choose file",
directory = '',
filter = "All Files (*.*)"
):
"""
Appends a file chooser widget to <parentWidget>, a property manager
group box.
@param parentWidget: the parent group box containing this widget.
@type parentWidget: PM_GroupBox
@param label: The label that appears to the left or right of the
file chooser lineedit (and "Browse" button).
If spanWidth is True, the label will be displayed on
its own row directly above the lineedit (and button).
To suppress the label, set I{label} to an
empty string.
@type label: str
@param labelColumn: The column number of the label in the group box
grid layout. The only valid values are 0 (left
column) and 1 (right column). The default is 0
(left column).
@type labelColumn: int
@param text: initial value of LineEdit widget.
@type text: string
@param setAsDefault: if True, will restore <val> when the
"Restore Defaults" button is clicked.
@type setAsDefault: boolean
@param spanWidth: if True, the widget and its label will span the width
of the group box. Its label will appear directly above
the widget (unless the label is empty) and is left justified.
@type spanWidth: boolean
@param caption: The caption used as the title of the file chooser
dialog. "Choose file" is the default.
@type caption: string
@param directory: The directory that the file chooser dialog should
open in when the "..." button is clicked.
If blank or if directory does not exist,
the current working directory is used.
@type directory: string
@param filter: The file type filters to use for the file chooser dialog.
@type filter: string (a semicolon-separated list of file types)
@see: U{B{QLineEdit}<http://doc.trolltech.com/4/qlineedit.html>}
"""
QWidget.__init__(self)
self.parentWidget = parentWidget
self.label = label
self.labelColumn = labelColumn
self.text = text
self.setAsDefault = setAsDefault
self.spanWidth = spanWidth
self.caption = caption
self.directory = directory
self.filter = filter
if label: # Create this widget's QLabel.
self.labelWidget = QLabel()
self.labelWidget.setText(label)
self.lineEdit = QLineEdit()
self.browseButton = QToolButton()
# Create vertical box layout.
self.hBoxLayout = QHBoxLayout(self)
self.hBoxLayout.setMargin(0)
self.hBoxLayout.setSpacing(2)
self.hBoxLayout.insertWidget(-1, self.lineEdit)
self.hBoxLayout.insertWidget(-1, self.browseButton)
# Set (QLineEdit) text
self.setText(text)
# Set browse button text and make signal-slot connection.
self.browseButton.setText("...")
self.connect(self.browseButton, SIGNAL("clicked()"), self.openFileChooserDialog)
# Set default value
self.defaultText = text
self.setAsDefault = setAsDefault
parentWidget.addPmWidget(self)
return
def setText(self, text):
"""
Set the line edit text.
@param text: The text.
@type text: string
"""
self.lineEdit.setText(text)
self.text = text
return
def openFileChooserDialog(self):
"""
Prompts the user to choose a file from disk and inserts the full path
into the lineEdit widget.
"""
_dir = getDefaultWorkingDirectory()
if self.directory:
if os.path.isdir(self.directory):
_dir = self.directory
fname = QFileDialog.getOpenFileName(self,
self.caption,
_dir,
self.filter)
if fname:
self.setText(fname)
self.lineEdit.emit(SIGNAL("editingFinished()"))
return
def restoreDefault(self):
"""
Restores the default value.
"""
if self.setAsDefault:
self.setText(self.defaultText)
return
def hide(self):
"""
Hides the lineedit and its label (if it has one).
@see: L{show}
"""
QWidget.hide(self)
if self.labelWidget:
self.labelWidget.hide()
return
def show(self):
"""
Unhides the lineedit and its label (if it has one).
@see: L{hide}
"""
QWidget.show(self)
if self.labelWidget:
self.labelWidget.show()
return
# End of PM_FileChooser ############################
class PM_ColorChooser( QWidget ):
"""
The PM_ColorChooser widget provides a color chooser widget for a
Property Manager group box. The PM_ColorChooser widget is a composite widget
made from 3 other Qt widgets:
- a QLabel
- a QFrame and
- a QToolButton (with a "..." text label).
IMAGE(http://www.nanoengineer-1.net/mediawiki/images/e/e2/PM_ColorChooser1.jpg)
The user can color using Qt's color (chooser) dialog by clicking the "..."
button. The selected color will be used as the color of the QFrame widget.
The parent must make the following signal-slot connection to be
notified when the user has selected a new color via the color chooser
dialog:
self.connect(pmColorChooser.colorFrame, SIGNAL("editingFinished()"), self.mySlotMethod)
@cvar setAsDefault: Determines whether to reset the value of the
color to I{defaultColor} when the user clicks
the "Restore Defaults" button.
@type setAsDefault: boolean
@cvar labelWidget: The Qt label widget of this PM widget.
@type labelWidget: U{B{QLabel}<http://doc.trolltech.com/4/qlabel.html>}
@cvar colorFrame: The Qt frame widget for this PM widget.
@type colorFrame: U{B{QFrame}<http://doc.trolltech.com/4/qframe.html>}
@cvar chooseButton: The Qt tool button widget for this PM widget.
@type chooseButton: U{B{QToolButton}<http://doc.trolltech.com/4/qtoolbutton.html>}
"""
defaultColor = None
setAsDefault = True
hidden = False
chooseButton = None
customColorCount = 0
standardColorList = [white, black]
def __init__(self,
parentWidget,
label = 'Color:',
labelColumn = 0,
color = white,
setAsDefault = True,
spanWidth = False,
):
"""
Appends a color chooser widget to <parentWidget>, a property manager
group box.
@param parentWidget: the parent group box containing this widget.
@type parentWidget: PM_GroupBox
@param label: The label that appears to the left or right of the
color frame (and "Browse" button).
If spanWidth is True, the label will be displayed on
its own row directly above the lineedit (and button).
To suppress the label, set I{label} to an
empty string.
@type label: str
@param labelColumn: The column number of the label in the group box
grid layout. The only valid values are 0 (left
column) and 1 (right column). The default is 0
(left column).
@type labelColumn: int
@param color: initial color. White is the default.
@type color: tuple of 3 floats (r, g, b)
@param setAsDefault: if True, will restore L{color} when the
"Restore Defaults" button is clicked.
@type setAsDefault: boolean
@param spanWidth: if True, the widget and its label will span the width
of the group box. Its label will appear directly above
the widget (unless the label is empty) and is left
justified.
@type spanWidth: boolean
@see: U{B{QColorDialog}<http://doc.trolltech.com/4/qcolordialog.html>}
"""
QWidget.__init__(self)
self.parentWidget = parentWidget
self.label = label
self.labelColumn = labelColumn
self.color = color
self.setAsDefault = setAsDefault
self.spanWidth = spanWidth
if label: # Create this widget's QLabel.
self.labelWidget = QLabel()
self.labelWidget.setText(label)
# Create the color frame (color swath) and "..." button.
self.colorFrame = QFrame()
self.colorFrame.setFrameShape(QFrame.Box)
self.colorFrame.setFrameShadow(QFrame.Plain)
# Set browse button text and make signal-slot connection.
self.chooseButton = QToolButton()
self.chooseButton.setText("...")
self.connect(self.chooseButton, SIGNAL("clicked()"), self.openColorChooserDialog)
# Add a horizontal spacer to keep the colorFrame and "..." squeezed
# together, even when the PM width changes.
self.hSpacer = QSpacerItem(10, 10,
QSizePolicy.MinimumExpanding,
QSizePolicy.Fixed)
# Create vertical box layout.
self.hBoxLayout = QHBoxLayout(self)
self.hBoxLayout.setMargin(0)
self.hBoxLayout.setSpacing(2)
self.hBoxLayout.insertWidget(-1, self.colorFrame)
self.hBoxLayout.insertWidget(-1, self.chooseButton)
# Set this to False to make the colorFrame an expandable rectangle.
COLORFRAME_IS_SQUARE = True
if COLORFRAME_IS_SQUARE:
squareSize = 20
self.colorFrame.setMinimumSize(QSize(squareSize, squareSize))
self.colorFrame.setMaximumSize(QSize(squareSize, squareSize))
self.hBoxLayout.addItem(self.hSpacer)
self.setColor(color, default = setAsDefault)
parentWidget.addPmWidget(self)
return
def setColor(self, color, default = False):
"""
Set the color.
@param color: The color.
@type color: tuple of 3 floats (r, g, b)
@param default: If True, make I{color} the default color. Default is
False.
@type default: boolean
"""
if default:
self.defaultColor = color
self.setAsDefault = default
self.color = color
self._updateColorFrame()
return
def getColor(self):
"""
Return the current color.
@return: The current r, g, b color.
@rtype: Tuple of 3 floats (r, g, b)
"""
return self.color
def getQColor(self):
"""
Return the current QColor.
@return: The current color.
@rtype: QColor
"""
return RGBf_to_QColor(self.color)
def _updateColorFrame(self):
"""
Updates the color frame with the current color.
"""
colorframe = self.colorFrame
try:
qcolor = self.getQColor()
palette = QPalette() # QPalette(qcolor) would have window color set from qcolor, but that doesn't help us here
qcolorrole = QPalette.Window
## http://doc.trolltech.com/4.2/qpalette.html#ColorRole-enum says:
## QPalette.Window 10 A general background color.
palette.setColor(QPalette.Active, qcolorrole, qcolor) # used when window is in fg and has focus
palette.setColor(QPalette.Inactive, qcolorrole, qcolor) # used when window is in bg or does not have focus
palette.setColor(QPalette.Disabled, qcolorrole, qcolor) # used when widget is disabled
colorframe.setPalette(palette)
colorframe.setAutoFillBackground(True)
except:
print "data for following exception: ",
print "colorframe %r has palette %r" % (colorframe, colorframe.palette())
pass
def openColorChooserDialog(self):
"""
Prompts the user to choose a color and then updates colorFrame with
the selected color.
"""
qcolor = RGBf_to_QColor(self.color)
if not self.color in self.standardColorList:
QColorDialog.setCustomColor(self.customColorCount, qcolor.rgb())
self.customColorCount += 1
c = QColorDialog.getColor(qcolor, self)
if c.isValid():
self.setColor(QColor_to_RGBf(c))
self.colorFrame.emit(SIGNAL("editingFinished()"))
def restoreDefault(self):
"""
Restores the default value.
"""
if self.setAsDefault:
self.setColor(self.defaultColor)
return
def hide(self):
"""
Hides the lineedit and its label (if it has one).
@see: L{show}
"""
QWidget.hide(self)
if self.labelWidget:
self.labelWidget.hide()
return
def show(self):
"""
Unhides the lineedit and its label (if it has one).
@see: L{hide}
"""
QWidget.show(self)
if self.labelWidget:
self.labelWidget.show()
return
class LetsShareBooksDialog(QDialog):
started_calibre_web_server = QtCore.pyqtSignal()
calibre_didnt_start = QtCore.pyqtSignal()
established_ssh_tunnel = QtCore.pyqtSignal()
def __init__(self, gui, icon, do_user_config, qaction, us):
QDialog.__init__(self, gui)
self.main_gui = gui
self.do_user_config = do_user_config
self.qaction = qaction
self.us = us
self.initial = True
self.lsb_url_text = 'Be a librarian. Share your library.'
self.url_label_tooltip = '<<<< Be a librarian. Click on Start sharing button.<<<<'
self.lsb_url = 'nourl'
if prefs['librarian'] == 'l' or prefs['librarian'] == '':
self.librarian = get_libranon()
else:
self.librarian = prefs['librarian']
self.metadata_thread = MetadataLibThread(self.librarian)
self.check_connection = ConnectionCheck()
self.clip = QApplication.clipboard()
self.pxmp = QPixmap()
self.pxmp.load('images/icon_connected.png')
self.icon_connected = QIcon(self.pxmp)
self.setStyleSheet("""
QDialog {
background-color: white;
}
QPushButton {
font-size: 16px;
border-style: solid;
border-color: red;
font-family:'BitstreamVeraSansMono',Consolas,monospace;
text-transform: uppercase;
}
QPushButton#arrow {
border-width: 16px;
border-right-color:white;
padding: -10px;
color:red;
}
QPushButton#url {
background-color: red;
min-width: 460px;
color: white;
text-align: left;
}
QPushButton#url:hover {
background-color: white;
color: red;
}
QPushButton#share {
background-color: red;
color: white;
margin-right: 10px;
}
QPushButton#share:hover {
background-color: white;
color: red;
}
QPushButton#url2 {
color: #222;
text-align: left;
}
QPushButton#url2:hover {
color: red;
}
QLineEdit#edit {
background-color: white;
color: black;
font-size: 16px;
border-style: solid;
border-color: red;
font-family:'BitstreamVeraSansMono',Consolas,monospace;
text-transform: uppercase;
}
""")
self.ll = QVBoxLayout()
#self.ll.setSpacing(1)
self.l = QHBoxLayout()
self.l.setSpacing(0)
self.l.setMargin(0)
#self.l.setContentsMargins(0,0,0,0)
self.w = QWidget()
self.w.setLayout(self.l)
self.setLayout(self.ll)
self.setWindowIcon(icon)
self.debug_label = QLabel()
self.ll.addWidget(self.debug_label)
self.debug_label.show()
self.lets_share_button = QPushButton()
self.lets_share_button.setSizePolicy(QSizePolicy.MinimumExpanding, QSizePolicy.MinimumExpanding)
self.lets_share_button.setObjectName("share")
#self.lets_share_button.clicked.connect(self.lets_share)
self.l.addWidget(self.lets_share_button)
self.url_label = QPushButton()
self.url_label.setSizePolicy(QSizePolicy.MinimumExpanding, QSizePolicy.MinimumExpanding)
self.url_label.setObjectName("url")
#self.url_label.clicked.connect(self.open_url)
self.l.addWidget(self.url_label)
self.arrow_button = QPushButton("_____")
self.arrow_button.setObjectName("arrow")
self.l.addWidget(self.arrow_button)
self.ll.addWidget(self.w)
self.ll.addSpacing(5)
self.libranon_layout = QHBoxLayout()
self.libranon_layout.setSpacing(0)
self.libranon_layout.setMargin(0)
#self.l.setContentsMargins(0,0,0,0)
self.libranon_container = QWidget()
self.libranon_container.setLayout(self.libranon_layout)
self.edit = QLineEdit()
self.edit.setObjectName("edit")
self.edit.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
self.edit.setToolTip("Change your librarian name")
self.edit.setText(self.librarian)
#self.edit.textChanged.connect(self.handle_text_changed)
self.save_libranon = QPushButton("librarian:")
self.save_libranon.setSizePolicy(QSizePolicy.Maximum, QSizePolicy.Maximum)
self.save_libranon.setObjectName("share")
self.save_libranon.setToolTip("Save your librarian name")
self.libranon_layout.addWidget(self.save_libranon)
self.libranon_layout.addWidget(self.edit)
self.save_libranon.clicked.connect(self.save_librarian)
self.ll.addWidget(self.libranon_container)
self.ll.addSpacing(10)
self.chat_button = QPushButton("Chat room: https://chat.memoryoftheworld.org")
#self.chat_button.hovered.connect(self.setCursorToHand)
self.chat_button.setObjectName("url2")
self.chat_button.setToolTip('Meetings every thursday at 23:59 (central eruopean time)')
self.chat_button.clicked.connect(functools.partial(self.open_url, "https://chat.memoryoftheworld.org/?nick={}".format(self.librarian.lower().replace(" ", "_"))))
self.ll.addWidget(self.chat_button)
self.about_project_button = QPushButton('Public Library: http://www.memoryoftheworld.org')
self.about_project_button.setObjectName("url2")
self.about_project_button.setToolTip('When everyone is librarian, library is everywhere.')
self.metadata_thread.uploaded.connect(lambda: self.render_library_button("{}://library.{}".format(prefs['server_prefix'], prefs['lsb_server']), "Building together real-time p2p library infrastructure."))
self.ll.addWidget(self.about_project_button)
self.debug_log = QListWidget()
self.ll.addWidget(self.debug_log)
self.debug_log.addItem("Initiatied!")
self.debug_log.hide()
from PyQt4 import QtWebKit
self.webview = QtWebKit.QWebView()
self.webview.setMaximumWidth(680)
self.webview.setMaximumHeight(400)
self.webview.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
self.webview.setUrl(QtCore.QUrl( "https://chat.memoryoftheworld.org/?nick={}".format(self.librarian.lower().replace(" ", "_"))))
self.ll.addWidget(self.webview)
#self.webview.hide()
#- check if there is a new version of plugin ----------------------------------------------
self.plugin_url = "https://github.com/marcellmars/letssharebooks/raw/master/calibreletssharebooks/letssharebooks_calibre.zip"
self.running_version = ".".join(map(str, lsb.version))
try:
r = requests.get('https://raw.github.com/marcellmars/letssharebooks/master/calibreletssharebooks/_version', timeout=3)
self.latest_version = r.text[-1]
except:
self.latest_version = "0.0.0"
self.upgrade_button = QPushButton('Please download and upgrade from {0} to {1} version of plugin.'.format(self.us.running_version, self.latest_version))
self.upgrade_button.setObjectName("url2")
self.upgrade_button.setToolTip('Running latest version you make developers happy')
self.upgrade_button.clicked.connect(functools.partial(self.open_url, self.plugin_url))
version_list = [self.us.running_version, self.us.latest_version]
version_list.sort(key=lambda s: map(int, s.split('.')))
if self.us.running_version != self.us.latest_version:
if self.us.running_version == version_list[0]:
self.ll.addSpacing(20)
self.ll.addWidget(self.upgrade_button)
#------------------------------------------------------------------------------------------
self.resize(self.sizeHint())
#- parsing/tee log file -------------------------------------------------------------------
self.se = open("/tmp/lsb.log", "w+b")
#self.se = tempfile.NamedTemporaryFile()
self.so = self.se
sys.stdout = os.fdopen(sys.stdout.fileno(), 'w', 0)
os.dup2(self.so.fileno(), sys.stdout.fileno())
os.dup2(self.se.fileno(), sys.stderr.fileno())
#- state machine --------------------------------------------------------------------------
self.machine = QtCore.QStateMachine()
self.on = QtCore.QState()
self.on.setObjectName("on")
self.on.entered.connect(lambda: self.render_lsb_button("Start sharing", self.lsb_url_text))
self.calibre_web_server = QtCore.QState()
self.calibre_web_server.setObjectName("calibre_web_server")
self.calibre_web_server.entered.connect(self.start_calibre_server)
self.calibre_web_server.entered.connect(lambda: self.render_lsb_button("Stop sharing", self.lsb_url_text))
self.calibre_web_server.assignProperty(self.debug_label, 'text', 'Starting Calibre web server...')
self.ssh_server = QtCore.QState()
self.ssh_server.setObjectName("ssh_server")
self.ssh_server.entered.connect(lambda: self.render_lsb_button("Stop sharing", "Connecting..."))
self.ssh_server.entered.connect(self.establish_ssh_server)
self.ssh_server.assignProperty(self.debug_label, 'text', 'Establishing SSH tunnel...')
self.ssh_server_established = QtCore.QState()
self.ssh_server_established.setObjectName("ssh_server_established")
self.ssh_server_established.entered.connect(lambda: self.render_lsb_button("Stop sharing", self.lsb_url_text))
self.ssh_server_established.entered.connect(self.check_connections)
self.ssh_server_established.assignProperty(self.debug_label, 'text', 'Established SSH tunnel...')
self.url_label_clicked = QtCore.QState()
self.url_label_clicked.setObjectName("url_label_clicked")
self.url_label_clicked.entered.connect(lambda: self.open_url(self.lsb_url))
self.url_label_clicked.assignProperty(self.debug_label, 'text', 'URL label clicked!')
self.about_project_clicked = QtCore.QState()
self.about_project_clicked.setObjectName("about_project_clicked")
self.about_project_clicked.entered.connect(lambda: self.open_url("{}://library.{}".format(prefs['server_prefix'], prefs['lsb_server'])))
self.about_project_clicked.assignProperty(self.debug_label, 'text', 'about_project_button clicked!')
self.library_state_changed = QtCore.QState()
self.library_state_changed.entered.connect(lambda: self.render_library_button("Uploading library metadata...", "Sharing with the others who share their libraries now..."))
self.library_state_changed.setObjectName("library_state_changed")
self.library_state_changed.entered.connect(self.sync_metadata)
self.off = QtCore.QState()
self.off.setObjectName("off")
self.off.entered.connect(lambda: self.disconnect_all())
self.off.assignProperty(self.debug_label, 'text', 'Start again...')
self.on.addTransition(self.lets_share_button.clicked, self.calibre_web_server)
self.calibre_web_server.addTransition(self.lets_share_button.clicked, self.off)
self.calibre_web_server.addTransition(self.calibre_didnt_start, self.off)
self.calibre_web_server.addTransition(self.started_calibre_web_server, self.ssh_server)
self.ssh_server.addTransition(self.lets_share_button.clicked, self.off)
self.ssh_server.addTransition(self.check_connection.lost_connection, self.off)
self.ssh_server.addTransition(self.established_ssh_tunnel, self.ssh_server_established)
self.ssh_server_established.addTransition(self.lets_share_button.clicked, self.off)
self.ssh_server_established.addTransition(self.url_label.clicked, self.url_label_clicked)
self.ssh_server_established.addTransition(self.about_project_button.clicked, self.about_project_clicked)
self.ssh_server_established.addTransition(self.check_connection.lost_connection, self.off)
self.ssh_server_established.addTransition(self.us.library_changed, self.library_state_changed)
self.url_label_clicked.addTransition(self.ssh_server_established)
self.about_project_clicked.addTransition(self.ssh_server_established)
self.library_state_changed.addTransition(self.metadata_thread.uploaded, self.ssh_server_established)
self.library_state_changed.addTransition(self.metadata_thread.upload_error, self.off)
self.library_state_changed.addTransition(self.lets_share_button.clicked, self.off)
self.library_state_changed.addTransition(self.url_label.clicked, self.url_label_clicked)
self.off.addTransition(self.on)
self.machine.addState(self.on)
self.machine.addState(self.calibre_web_server)
self.machine.addState(self.ssh_server)
self.machine.addState(self.ssh_server_established)
self.machine.addState(self.url_label_clicked)
self.machine.addState(self.about_project_clicked)
self.machine.addState(self.library_state_changed)
self.machine.addState(self.off)
self.machine.setInitialState(self.on)
self.machine.start()
#------------------------------------------------------------------------------------------
def sql_db_changed(self, event, ids):
# this could be used for better update on added/removed books
if not self.metadata_thread.isRunning():
self.sync_metadata()
else:
print("metadata_thread is running! no sync!")
def sync_metadata(self):
from calibre.gui2.ui import get_gui
try:
del self.sql_db
except:
pass
self.sql_db = get_gui().current_db
self.sql_db.add_listener(self.sql_db_changed)
self.metadata_thread.sql_db = self.sql_db
self.metadata_thread.port = self.port
self.metadata_thread.uploaded.connect(lambda: self.debug_log.addItem("uploaded!"))
self.metadata_thread.upload_error.connect(lambda: self.debug_log.addItem("upload_ERROR!"))
self.metadata_thread.start()
def check_connections(self):
#self.webview.show()
if self.initial:
print("initial!")
self.us.library_changed_emit()
self.initial = False
self.qaction.setIcon(get_icon('images/icon_connected.png'))
self.check_connection.add_urls(["http://*****:*****@{2} -R {0}:localhost:{1} -P 722".format(self.port, self.calibre_server_port, prefs['lsb_server']), shell=True)
self.lsb_url = "{}://www{}.{}".format(prefs['server_prefix'], self.port, prefs['lsb_server'])
self.lsb_url_text = "Go to: {}".format(self.lsb_url)
QTimer.singleShot(3000, self.established_ssh_tunnel.emit)
else:
self.ssh_proc = subprocess.Popen(['ssh', '-T', '-N', '-g', '-o', 'TCPKeepAlive=yes', '-o', 'UserKnownHostsFile=/dev/null', '-o', 'StrictHostKeyChecking=no','-o', 'ServerAliveINterval=60', prefs['lsb_server'], '-l', 'tunnel', '-R', '0:localhost:{0}'.format(self.calibre_server_port), '-p', '722'])
if self.ssh_proc:
def parse_log():
gotcha = False
try:
self.se.seek(0)
result = self.se.readlines()
self.se.seek(0)
self.se.truncate()
for line in result:
m = re.match("^Allocated port (.*) for .*", line)
try:
self.port = m.groups()[0]
self.lsb_url = '{}://www{}.{}'.format(prefs['server_prefix'], self.port, prefs['lsb_server'])
self.lsb_url_text = "Go to: {0}".format(self.lsb_url)
self.url_label_tooltip = 'Copy URL to clipboard and check it out in a browser!'
self.established_ssh_tunnel.emit()
gotcha = True
except:
pass
finally:
if not gotcha:
QTimer.singleShot(500, parse_log)
parse_log()
def start_calibre_server(self):
if self.main_gui.content_server is None:
self.main_gui.start_content_server()
opts, args = server_config().option_parser().parse_args(['calibre-server'])
self.calibre_server_port = opts.port
self.started_calibre_web_server.emit()
else:
self.calibre_didnt_start.emit()
def config(self):
self.do_user_config(parent=self)
self.label.setText(prefs['lsb_server'])
def save_librarian(self):
print('librarian {} saved!'.format(str(self.edit.text())))
if self.edit.text() != "" and self.edit.text() != "l":
prefs['librarian'] = str(self.edit.text())
else:
prefs['librarian'] = get_libranon()
self.edit.setText(prefs['librarian'])
def open_url(self, url):
self.clip.setText(url)
webbrowser.open(url)
def keyPressEvent(self, event):
if event.key() == QtCore.Qt.Key_Escape:
pass
def closeEvent(self, e):
print("close popup!")
self.hide()
def __init__(self):
QMainWindow.__init__(self)
self.setWindowTitle("My Main Window")
self.setMinimumWidth(MAIN_WINDOW_SIZE[0])
self.setMinimumHeight(MAIN_WINDOW_SIZE[1])
self.statusbar = QtGui.QStatusBar(self)
self.statusbar.showMessage("Status message")
self.setStatusBar(self.statusbar)
################################################
self.menubar = self.menuBar()
# Any menu action makes the status bar message disappear
fileMenu = QtGui.QMenu(self.menubar)
fileMenu.setTitle("File")
self.menubar.addAction(fileMenu.menuAction())
newAction = QtGui.QAction("New", self)
newAction.setIcon(QtGui.QtIcon(icons + '/GroupPropDialog_image0.png'))
fileMenu.addAction(newAction)
openAction = QtGui.QAction("Open", self)
openAction.setIcon(QtGui.QtIcon(icons + "/MainWindowUI_image1"))
fileMenu.addAction(openAction)
saveAction = QtGui.QAction("Save", self)
saveAction.setIcon(QtGui.QtIcon(icons + "/MainWindowUI_image2"))
fileMenu.addAction(saveAction)
self.connect(newAction, SIGNAL("activated()"), self.fileNew)
self.connect(openAction, SIGNAL("activated()"), self.fileOpen)
self.connect(saveAction, SIGNAL("activated()"), self.fileSave)
for otherMenuName in ('Edit', 'View', 'Display', 'Select', 'Modify',
'NanoHive-1'):
otherMenu = QtGui.QMenu(self.menubar)
otherMenu.setTitle(otherMenuName)
self.menubar.addAction(otherMenu.menuAction())
helpMenu = QtGui.QMenu(self.menubar)
helpMenu.setTitle("Help")
self.menubar.addAction(helpMenu.menuAction())
aboutAction = QtGui.QAction("About", self)
aboutAction.setIcon(QtGui.QtIcon(icons + '/MainWindowUI_image0.png'))
helpMenu.addAction(aboutAction)
self.connect(aboutAction, SIGNAL("activated()"), self.helpAbout)
##############################################
self.setMenuBar(self.menubar)
centralwidget = QWidget()
self.setCentralWidget(centralwidget)
layout = QVBoxLayout(centralwidget)
layout.setMargin(0)
layout.setSpacing(0)
middlewidget = QWidget()
self.bigButtons = QWidget()
bblo = QHBoxLayout(self.bigButtons)
bblo.setMargin(0)
bblo.setSpacing(0)
self.bigButtons.setMinimumHeight(50)
self.bigButtons.setSizePolicy(QSizePolicy.Fixed, QSizePolicy.Fixed)
for name in ('Features', 'Sketch', 'Build', 'Dimension', 'Simulator'):
btn = QPushButton(self.bigButtons)
btn.setMaximumWidth(80)
btn.setMinimumHeight(50)
btn.setText(name)
self.bigButtons.layout().addWidget(btn)
self.bigButtons.hide()
layout.addWidget(self.bigButtons)
self.littleIcons = QWidget()
self.littleIcons.setMinimumHeight(30)
self.littleIcons.setMaximumHeight(30)
lilo = QHBoxLayout(self.littleIcons)
lilo.setMargin(0)
lilo.setSpacing(0)
pb = QPushButton(self.littleIcons)
pb.setIcon(QIcon(icons + '/GroupPropDialog_image0.png'))
self.connect(pb, SIGNAL("clicked()"), self.fileNew)
lilo.addWidget(pb)
for x in "1 2 4 5 6 7 8 18 42 10 43 150 93 94 97 137".split():
pb = QPushButton(self.littleIcons)
pb.setIcon(QIcon(icons + '/MainWindowUI_image' + x + '.png'))
lilo.addWidget(pb)
layout.addWidget(self.littleIcons)
layout.addWidget(middlewidget)
self.layout = QGridLayout(middlewidget)
self.layout.setMargin(0)
self.layout.setSpacing(2)
self.gridPosition = GridPosition()
self.numParts = 0
self.show()
explainWindow = AboutWindow(
"Select <b>Help->About</b>"
" for instructions...", 200, 3)
