{Tc @ s d Z d d l Z d d l m Z m Z d d l Td d l Z d d l Z d d l Z d e j f d YZ
d e j f d YZ d e j
f d
YZ d S( s
qtgui/pictureWizardGui.py: CCP4MG Molecular Graphics Program
Copyright (C) 2001-2008 University of York, CCLRC
Copyright (C) 2009-2011 University of York
Copyright (C) 2012 STFC
This library is free software: you can redistribute it and/or
modify it under the terms of the GNU Lesser General Public License
version 3, modified in accordance with the provisions of the
license to address the requirements of UK law.
You should have received a copy of the modified GNU Lesser General
Public License along with this library. If not, copies may be
downloaded from http://www.ccp4.ac.uk/ccp4license.php
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU Lesser General Public License for more details.
iN( t QtGuit QtCore( t *t drawingStyleGuic B sb e Z d d d e d Z d d Z d Z d d Z d Z d Z d Z
d
Z RS( id i c C sC t j j | | | | _ | | _ | j d | | j d S( Nt description( R t QWidgett __init__t disable_query_scenest picSizet drawt show( t selft parentt heightR R ( ( sC /opt/xtal/ccp4-6.5-linux32/lib/py2/ccp4mg/qtgui/pictureWizardGui.pyR $ s
c C s d } t j } d } | j | | | | | j | | j | _ | j | j | j | j | j t j
d | j | j | j t j
d | j | j | j t j
d | j t j
| _ t j d } | j j | | j | j | ryt j | | _ | j j d | j j | t j } | j | | j | j | j | n d | _ | j | j | | j | j d S( NiP i s itemSelectionChanged()s! itemEntered(QTreeWidgetItem*,int)s' itemDoubleClicked(QTreeWidgetItem*,int)s ccp4mg_icon.pngi ( R t QHBoxLayoutt setContentsMarginst
setSpacingt drawTreet treet setStylet addWidgett connectR t SIGNALt
changePicturet handleDoubleClickt QLabelt wiz_labelt QPixmapt setPixmapt QTextEditt descriptionWidgett setReadOnlyt setFixedHeightt QVBoxLayoutt addLayoutt setLayoutt Nonet
addStretch( R R t description_heightt layoutt margint
wiz_pixmapt layout0( ( sC /opt/xtal/ccp4-6.5-linux32/lib/py2/ccp4mg/qtgui/pictureWizardGui.pyR , s8
"""
c
C s d d l } t } | j d g } t j } | j t j j | j d x | D]} t j | j
| d d d g } | j t j
j | j | x | d D] \ } } | r| j rt j | j
| d d g } | rl| j t j
j | j | qlq t j | j
| d d g } | r | j t j
j t j
j B | j | q q WqZ W| j j d | S( Nit styles* Select a drawing style to apply to a modeli t _t i ( t stringt
PICTUREWIZARDt define_model_style_menuR t QTreeWidgett setSelectionModet QAbstractItemViewt SingleSelectiont
setToolTipt QTreeWidgetItemt replacet setFlagsR t Qtt
ItemIsEnabledt addTopLevelItemR t NoItemFlagst addChildt ItemIsSelectablet
headerItemt setHidden(
R R. t picwizt menu_listR t groupt itemt filet queryt item0( ( sC /opt/xtal/ccp4-6.5-linux32/lib/py2/ccp4mg/qtgui/pictureWizardGui.pyR W s,
%
!! t c C s d d l } | r` d d l } t | | j sT | j | j | d d d } n | | _ n* | j | j t j d d d | _ | j j
t j | j d t j
j } d } d } x | d k r| t | k r| | j } d } x{ | d k ru| | k rut | | j | j d | j d k rh| } | j j | | j | n | d } q W| d } q Wd S( NiR, R- t :i i ( R. t typest
isinstancet ListTypet splitR7 R+ R/ t default_drawing_styleR t findItemsR t QStringR9 t MatchExactlyt lent
childCountt strt childt textt setCurrentItem( R R+ R. RJ t itemst hitt iit nChildrent nc( ( sC /opt/xtal/ccp4-6.5-linux32/lib/py2/ccp4mg/qtgui/pictureWizardGui.pyR w s( $ $+!/ c
C s& d d l } d d l } | j j } | r"| j sL | j | j q"| j t | j j d d d } | j t | j d d d } | | g | _ t
j d | d | d d } y> t j
| } | j | j t j j } | j j | Wn d G| GHn Xt
j d | d | d d
} t
j d | d d
\ } }
d } | s| j j |
d |
d } n | j r| j j t
j d | d d \ } } | s| j j | | j d d qn t
j | d | | j t j d n d S( Nii R- R, t
drawing_styleRI t modet pngsnapshott Not mgpict filenamet sectiont TITLERH s s
t NOTESs
s
t changed( R. t osR t currentItemR R R+ R7 RT RV R/ t
snapshot_fileR R t scaledToHeightR R R9 t SmoothTransformationR R t read_mgpic_file_sectionR5 R t cleart setTextt set_default_drawing_stylet emitR (
R R. Rg t currentt subdirR+ t
style_fileR) t pic_filet rvt titlet desc_textt desc( ( sC /opt/xtal/ccp4-6.5-linux32/lib/py2/ccp4mg/qtgui/pictureWizardGui.pyR s8 *$#
#!
! &c C s d d l } | j j } | r | j r | j t | j j d d d } | j t | j d d d } | d | Sd Sd S( Nii R- R, RI RH ( R. R Rh R R7 RT RV ( R R. Rq Rr R+ ( ( sC /opt/xtal/ccp4-6.5-linux32/lib/py2/ccp4mg/qtgui/pictureWizardGui.pyt getStyle s *$ c C s d G| G| GHd S( NR ( ( R RD t col( ( sC /opt/xtal/ccp4-6.5-linux32/lib/py2/ccp4mg/qtgui/pictureWizardGui.pyR s c C s d G| G| GHd S( Nt handleItemEntered( ( R RD Rz ( ( sC /opt/xtal/ccp4-6.5-linux32/lib/py2/ccp4mg/qtgui/pictureWizardGui.pyR{ s N( t __name__t
__module__R$ t FalseR R R R R Ry R R{ ( ( ( sC /opt/xtal/ccp4-6.5-linux32/lib/py2/ccp4mg/qtgui/pictureWizardGui.pyR s + ' t drawingStyleComboc B s, e Z d g d Z g d Z d Z RS( c C sD t t | j | | j d | j d | j d | d S( Ni i
R+ ( t superR R t setEditablet setMaxVisibleItemst populateMenu( R R R+ ( ( sC /opt/xtal/ccp4-6.5-linux32/lib/py2/ccp4mg/qtgui/pictureWizardGui.pyR s
c C s t } | j d | \ } } x> t d t | D]' } | j | | t j | | q7 W| j | j r | j | j
| j n d S( NR+ i ( R/ R0 t rangeRR t addItemR t QVariantt countRN t setCurrentIndext index( R R+ RA RB t
alias_listRZ ( ( sC /opt/xtal/ccp4-6.5-linux32/lib/py2/ccp4mg/qtgui/pictureWizardGui.pyR s %c C s1 t | j | j j } | t _ | S( N( RT t itemDatat currentIndext toStringR/ RN ( R t alias( ( sC /opt/xtal/ccp4-6.5-linux32/lib/py2/ccp4mg/qtgui/pictureWizardGui.pyRy s ! N( R| R} R$ R R Ry ( ( ( sC /opt/xtal/ccp4-6.5-linux32/lib/py2/ccp4mg/qtgui/pictureWizardGui.pyR s
t pictureWizardGuic B sb e Z d
d d e d Z d d Z d d Z d Z d Z d Z d Z
d Z RS( RH c C s t j j | | | | _ d | _ g | _ | | _ | d k rc | j d d | j d n | j | j d | | j j
| | j | | j | j t
j d | j | j d S( NRH t ALLR i s- Picture wizard to apply to ALL VISIBLE modelss Picture wizard: Rf ( R t QDialogR t recentreIfOnlyOneR] t choicest moldata_nameR t setWindowTitlet drawingStyleWidgetR t updateDrawingStyleR R R R
( R R R] R R ( ( sC /opt/xtal/ccp4-6.5-linux32/lib/py2/ccp4mg/qtgui/pictureWizardGui.pyR s
"i c C sr t j } t j d | | _ | j j d t j | j | j t | d d d d d | | _ | j | j t j
| | _ | j | j t j | } | j
t j j x d t j j | j g d t j j | j g d t j j | j g g D]J \ } } } | j | | } | j d
| j | t j d | qW| j | | j | d S( Ns" Delete any existing display objecti R R
i R s Create picturet Cancelt Helpi s clicked()( R R! t QCheckBoxt deleteWidgett
setCheckedR/ t keep_display_objectsR R R t QFramet definitionFramet QDialogButtonBoxt setOrientationR R9 t
Horizontalt
ActionRolet Applyt
RejectRolet hidet HelpRoleR t addButtont setAutoDefaultR R R# ( R R R' t
button_boxt labelt rolet slott button( ( sC /opt/xtal/ccp4-6.5-linux32/lib/py2/ccp4mg/qtgui/pictureWizardGui.pyR s$ !(
c C s | s | j j } n | | j k r+ d S| | _ t j d | j d d | _ t j d | } | d d k r t j j | d | | d n
| d | _
| j d S( NR] R^ Ra i s Error reading drawing style i ( R Ry R] R/ Ri Rb t read_mgpic_file_choicesR t QMessageBoxt warningR t drawDefinitionFrame( R R] Ru ( ( sC /opt/xtal/ccp4-6.5-linux32/lib/py2/ccp4mg/qtgui/pictureWizardGui.pyR , s !!
c
C s | j j t j } | j } x | D] } | j q( W| j j } | d k r t j } | j d | j
d d d d | j j | n | j d k rpt
d d } d } g } x | D] } t | d r| d k rt j | j } t j | j } n t | d r | d k r x1 | D]% } | | j k r5| j | q5q5Wq q W| j d d d }
xY| j D] } | j d d } d
g j | r|
d }
t j | j d d | }
| j |
|
d d d | j d d } | d
k r-t j | } | j | n | j | | j t j j | j | |
d d d qqWngd }
x^| j D]S} | j d d } d
d
g j | r|
d }
t j | j d d | }
| j |
|
d d d | j d d } | d
k rYt j! | d | j d d d d } | j" | r| j# d | | qnz | d
k rd d d d d g } t j$ | d t% | j j& d | d d d g } | j" | r| j' | | qn | j | | j t j j | j | |
d d d q| d k r|
d }
t j( | j d d | } | j d d } | j | | j" | r| j) | | n | j) t* | j d d | j t j j | j | |
d d d qqWd S( Ni i R t object_typet MolDatat chainsRH it typet chainR t namet monomert MolData_nameR^ R t blankt atomNamet monomerst
saccharidet solutet peptidet nucleict
abstractModelt modelGroupst selectionWidgetst namedSelectiont logicalt initiali (+ R t findChildrenR R t getParamst closeR' R$ t QGridLayoutR R R# R t get_dataobjt hasattrt copyt deepcopyR t removet insertR t getR R R t QComboBoxt addItemst
setObjectNamet setAttributeR R9 t WA_DeleteOnCloset mgWidgetst mgAtomPickert has_keyt setSelectiont mgSelectionCombot datat treeModelRn R R t int( R t togot paramsRD R' t dataobjst firstChainst allChainst ot cht nnt defnt defn_typeR R t widgetR ( ( sC /opt/xtal/ccp4-6.5-linux32/lib/py2/ccp4mg/qtgui/pictureWizardGui.pyR A s
#
$ 3
c
C s | j | j d k r1 t d d } t } n t d | j g } t } x^ | D]V } | j rS | j d | j j d | j
d d t | j j
d | qS qS Wd
d l } | j d | d | t } | r | j t j d
n x6 | D]. } t j | j } | r | j q q W| rO| j t j d n | j rt t d d d k rt j d d n | j d S( NR R R R R] t chosenR i t force_recolour_bymolit targett controllingWidgett applyWizardt finishedApplyWizardt glidei ( R R R t TrueR R~ t visiblet applyDrawingStyleR Ry R R R t isCheckedt rebuildt
UpdateDisplayt DISPLAYTABLERp R R t
getDataObjR t
updateDispObjR RR t
MAINWINDOWt handleRecentreR ( R t target_listR R R t dtt obj( ( sC /opt/xtal/ccp4-6.5-linux32/lib/py2/ccp4mg/qtgui/pictureWizardGui.pyR s2
'
$c C s d S( N( ( R ( ( sC /opt/xtal/ccp4-6.5-linux32/lib/py2/ccp4mg/qtgui/pictureWizardGui.pyt undo s c C s i } x|| j D]q} | j j t j | j d d } | r d g j | j d d r{ t | j | | d s
"