,]c@scdZddlmZddlmZmZddlmZmZm Z m Z m Z m Z m Z ddlmZmZmZmZddlmZmZmZmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'ddl(Z)ddl*m+Z+dd l,m-Z-dd l.m/Z/m0Z0dd l1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8dd l9m:Z;dd l<m=Z=m>Z>m?Z?i(dd6dd6dd6dd6dd6dd6dd6dd6dd6dd6dd6dd6dd6dd6dd6dd6d d!6d"d#6d d$6d d%6d d&6d d'6d d(6d d)6d d*6d d+6d d,6d d-6d d.6d d/6d d06d d16d d26d d36d d46d d56d d66dd76dd86dd96Z@d:ZAd;ZBd<ZCd=ZDd>ZEd?ZFd@efdAYZGdBefdCYZHdDe%fdEYZIdFe'fdGYZJdHefdIYZKdS(Js' NumPy Array Editor Dialog based on Qt i(tprint_function(t from_qvariantt to_qvariant(tQAbstractTableModeltQItemSelectiontQLocaletQItemSelectionRanget QModelIndextQttSlot(tQColortQCursortQDoubleValidatort QKeySequence(tQAbstractItemDelegatet QApplicationt QCheckBoxt QComboBoxtQDialogtQDialogButtonBoxt QGridLayoutt QHBoxLayoutt QInputDialogt QItemDelegatetQLabelt QLineEdittQMenut QMessageBoxt QPushButtontQSpinBoxtQStackedWidgett QTableViewt QVBoxLayouttQWidgetN(t_(tDEFAULT_SMALL_DELTA(tget_fonttconfig_shortcut(tiotis_binary_stringt is_stringtis_text_stringtPY3tto_binary_stringtto_text_string(t icon_manager(t add_actionst create_actiont keybindings%.6gtsingletdoubletfloat_t longfloattfloat16tfloat32tfloat64tfloat96tfloat128s%rtcsingletcomplex_t clongfloatt complex64t complex128t complex192t complex256s%dtbytes%stbytes8tshorttintctint_tlonglongtintptint8tint16tint32tint64tubytetushorttuintctuintt ulonglongtuintptuint8tuint16tuint32tuint64tbool_tbool8tboolgAgj@i<cCsd|jkp|jdkS(s-Return True if datatype dtype is a float kindtfloatR1R2(R1R2(tname(tdtype((sJlib/python2.7/site-packages/spyder/widgets/variableexplorer/arrayeditor.pytis_float`scCs7t|p6d|jkp6d|jkp6d|jkS(s.Return True is datatype dtype is a number kindtinttlongRC(R\RZ(R[((sJlib/python2.7/site-packages/spyder/widgets/variableexplorer/arrayeditor.pyt is_numberes*cCsettg|D]}|j|jf^q \}}t|t|t|t|fS(s-Extract the boundaries from a list of indexes(tlisttziptrowtcolumntmintmax(t index_listtitrowstcols((sJlib/python2.7/site-packages/spyder/widgets/variableexplorer/arrayeditor.pyt get_idx_rectks=t ArrayModelcBseZdZdZdZdddeddZdZdZ dZ e dZ e d Z eed Zeed Zd Zd ZejdZejdZdZejdZdZRS(sArray Editor Table Modelii(s%.6gc CsLtj|||_i|_||_||_||_tjdgd|j |_ |j tj tj fkrtj |_n tj|_ddg}d|_d|_d|_||_||_|jjd|_|jjd|_|j|j}ytj|j||_tj|j||_|j|jkrZ|jd8_n|d|_|d|d|_t|_WnDt t!fk rd|_d|_d|_d|_t#|_nX|t$kr|j%|_&|j'|_(nT|jt)kr|j%|_&n |j|_&|jt*kr<|j'|_(n |j|_(dS( NiR[gQ?gGz?gffffff?g?g333333?i(+Rt__init__tdialogtchangestxlabelstylabelstreadonlytnptarrayR[t test_arrayR=R>tabst color_functrealtsattvaltalpt_datat_formattshapet total_rowst total_colstnanmintvmintnanmaxtvmaxthue0tdhuetTruetbgcolor_enabledt TypeErrort ValueErrortNonetFalset LARGE_SIZEt ROWS_TO_LOADt rows_loadedt COLS_TO_LOADt cols_loadedt LARGE_NROWSt LARGE_COLS( tselftdatatformatRoRpRqtparentthuerangetsize((sJlib/python2.7/site-packages/spyder/widgets/variableexplorer/arrayeditor.pyRlzsT                       cCs|jS(sReturn current format(R|(R((sJlib/python2.7/site-packages/spyder/widgets/variableexplorer/arrayeditor.pyt get_formatscCs|jS(s Return data(R{(R((sJlib/python2.7/site-packages/spyder/widgets/variableexplorer/arrayeditor.pytget_datascCs||_|jdS(sChange display formatN(R|treset(RR((sJlib/python2.7/site-packages/spyder/widgets/variableexplorer/arrayeditor.pyt set_formats cCs$|j|jkr|jS|jSdS(sArray column numberN(RR(Rtqindex((sJlib/python2.7/site-packages/spyder/widgets/variableexplorer/arrayeditor.pyt columnCountscCs$|j|jkr|jS|jSdS(sArray row numberN(R~R(RR((sJlib/python2.7/site-packages/spyder/widgets/variableexplorer/arrayeditor.pytrowCountscCsJ|r#|j|jkrtStSn|rF|j|jkr?tStSndS(N(R~RRRRR(RRhtcolumns((sJlib/python2.7/site-packages/spyder/widgets/variableexplorer/arrayeditor.pytcan_fetch_morescCs|jd|rt|j|j}t||j}|jt|j|j|d|j|7_|jn|jd|r|j|j }t||j }|j t|j |j |d|j |7_ |j ndS(NRhiR( RR~RRdRtbeginInsertRowsRt endInsertRowsRRRtbeginInsertColumnstendInsertColumns(RRhRtremindertitems_to_fetch((sJlib/python2.7/site-packages/spyder/widgets/variableexplorer/arrayeditor.pyt fetch_mores cCs|dk|_|jdS(sToggle backgroundcoloriN(RR(Rtstate((sJlib/python2.7/site-packages/spyder/widgets/variableexplorer/arrayeditor.pytbgcolorscCsl|j}|j}t|jjdkr@|j|}n|j||f}|jj||f|S(Ni(RbRctlenR{R}Rntget(RtindexRgtjtvalue((sJlib/python2.7/site-packages/spyder/widgets/variableexplorer/arrayeditor.pyt get_values   cCs|jstS|j|}t|rNyt|d}WqNqNXn|tjkr|tjj krsdSyt|j |SWqt k rt |_ t|SXn|tjkrtttjtjBS|tjkr|jr|tjj k ry|j|jt|j|j|t|j|j}ttj|}tj||j|j|j }t|SWqt k rtSXn"|tj!krtt"dt#StS(s Cell contenttutf8ttfont_size_delta($tisValidRRR'R,Rt DisplayRoleRrtmatmaskedR|RRRqtreprtTextAlignmentRoleR]t AlignCentert AlignVCentertBackgroundColorRoleRRRRYRRvRRuR tfromHsvFRxRyRztFontRoleR$R#(RRtroleRthuetcolor((sJlib/python2.7/site-packages/spyder/widgets/variableexplorer/arrayeditor.pyRs>    8!  c CsM|j s|jrtS|j}|j}t|t}|jjj }|dkryt t |}Wqt k r|j dk}qXn|jds|jdrt|d}n|jds|jdrt|}n|j jds"|j jdr&tSy%t|}|jsJ|j}nWn4t k r}tj|jd d t|tSXy||jd Rcsj|dtS(NRh(RR(Ry(R(sJlib/python2.7/site-packages/spyder/widgets/variableexplorer/arrayeditor.pyRR(RRltsetModeltsetItemDelegateRtranget columnWidthtviewporttresizeRdtheightR}t setup_menutmenuR%RthorizontalScrollBart valueChangedRtverticalScrollBar(RRRR[R}t total_widthtk((RsJlib/python2.7/site-packages/spyder/widgets/variableexplorer/arrayeditor.pyRls % cCs|jj}d}}|rb||jjkrb|jj}|jjd|n|r||jjkr|jj }|jjd|n|dk s|dk rt }x|D]}|j } |j } |dk r,| dkr,| |dkr,|jjd} n|j } |j} |dk r| dkr| |dkr|jj d} n|jj| | } |jj| | }t| |}|j|qW|jj||jjndS(NRhRii(tselectionModelt selectionRR tmaximumRRRRRRttoptbottomtlefttrightRRtappendtselecttClearAndSelect(RRRhRt old_selectiontold_rows_loadedtold_cols_loadedt new_selectiontpartRRRRttop_leftt bottom_right((sJlib/python2.7/site-packages/spyder/widgets/variableexplorer/arrayeditor.pyRs4         cCsNtjttj|j|jjdt|jtj dS(sResize cells to contentsRN( RtsetOverrideCursorR Rt WaitCursortresizeColumnsToContentsRRRtrestoreOverrideCursor(R((sJlib/python2.7/site-packages/spyder/widgets/variableexplorer/arrayeditor.pytresize_to_contentss   c Csht|tddtddtjdd|jdtj|_t |}t ||jg|S(sSetup context menutCopytshortcutticonteditcopyt triggeredR( R/R"R0timaR%RRtWidgetShortcutt copy_actionRR.(RR((sJlib/python2.7/site-packages/spyder/widgets/variableexplorer/arrayeditor.pyRs   cCs$|jj|j|jdS(sReimplement Qt methodN(Rtpopupt globalPostaccept(Rtevent((sJlib/python2.7/site-packages/spyder/widgets/variableexplorer/arrayeditor.pytcontextMenuEventscCs0|tjkr|jntj||dS(sReimplement Qt methodN(R R#RRt keyPressEvent(RR.((sJlib/python2.7/site-packages/spyder/widgets/variableexplorer/arrayeditor.pyR0s c Csb|s dSt|\}}}}|dkr]||jjdkr]|jjd}n|dkr||jjdkr|jjd}n|jj}trtj }n tj }yJt j ||||d||dfddd|jj Wn'tj|tdtddSX|jjd }|j|S( s)Copy an array portion to a unicode stringNiit delimiters tfmttWarnings1It was not possible to copy values for this arraysutf-8(RjRRRRR~RR*R&tBytesIOtStringIORrtsavetxtRRtwarningR"tgetvaluetdecodetclose( Rt cell_rangetrow_mintrow_maxtcol_mintcol_maxR{toutputtcontents((sJlib/python2.7/site-packages/spyder/widgets/variableexplorer/arrayeditor.pyt _sel_to_texts*%% -  cCs2|j|j}tj}|j|dS(sCopy text to clipboardN(RBtselectedIndexesRt clipboardR(RtcliptxtRD((sJlib/python2.7/site-packages/spyder/widgets/variableexplorer/arrayeditor.pyRs ( RRRRlRRR"RR/R0RBR R(((sJlib/python2.7/site-packages/spyder/widgets/variableexplorer/arrayeditor.pyRs #    tArrayEditorWidgetcBs5eZedddZdZdZdZRS(c Cs/tj||||_d|_t|jjdkrh|jj|_|jjddf|j_n6t|jjdkr|jj|_d |j_ntj|j j d}t |jd|d|d|d|d||_ t ||j |j |j|_t}|jtjttd }|jt|j |j||jj|jttd }|j||jj|jjttd } | j|j j| j|j j| jj|j j |j| t!} | j|j| j"||j#| dS( Niis%sRRoRpRqRtFormattResizesBackground color(ii($R!RlRRtold_data_shapeRR}tSUPPORTED_FORMATSRR[RZRkRRtviewRRRt AlignLeftRR"t setEnabledR\t addWidgettclickedRt change_formatR"Rt setCheckedRt stateChangedRR t addLayoutt setLayout( RRRRqRoRpRt btn_layouttbtnRtlayout((sJlib/python2.7/site-packages/spyder/widgets/variableexplorer/arrayeditor.pyRls>  !      cCsgx?t|jjjD]%\\}}}||j||f|sWt?t d|_@|j@jAtB|j@jCj%|jD| j|j@nt?t d|_E|jEjFtB|jEjGtB|jEjCj%|jH| j|jE|jjI| dd|jJd d!|jKtLjMtBS(#sb Setup ArrayEditor: return False if data is not supported, True otherwise is4Arrays with more than 3 dimensions are not supportedis=The 'xlabels' argument length do no match array column numberis:The 'ylabels' argument length do no match array row numberRRs %s arrayss%s are currently not supportedtarredits - s NumPy arrays Array editors (s read onlyt)iXisRecord array fields:is Masked datatDatatMasktkeyboardTrackingsAxis:sIndex:s.Warning: changes are applied separatelysmFor performance reasons, changes applied to masked array won't be reflected in array's data (and vice-versa).sSave and ClosetCloseii,N(NRRt writeableR[tnamesRt isinstanceRrRt MaskedArraytndimterrorR"RRR}RZRJRRRWRTt setWindowIconR(R%R,tsetWindowTitleRRRbRNRFtmaskt currentWidgetRaRRRtsave_and_close_enabletcurrentChangedtcurrent_widget_changedRRtfieldsR)RRRt index_spinR tchange_active_widgetRRRtaddItemstcurrentIndexChangedtcurrent_dim_changedt shape_labelt slicing_labelRftsetCurrentIndext setToolTipt addStretchRRct setDisabledRROR-RdtsetAutoDefaultt setDefaulttrejectRStsetMinimumSizetsetWindowFlagsRtWindow(RRttitleRqRoRptis_record_arraytis_masked_arraytdtntarrRZRURntfieldRRgtra_combotlabel((sJlib/python2.7/site-packages/spyder/widgets/variableexplorer/arrayeditor.pytsetup_and_checkes ((            !%           cCs@|jr<|jjt|jjt|jjtndS(sAHandle the data change event to enable the save and close button.N(RcRMRRR(Rtleft_topR((sJlib/python2.7/site-packages/spyder/widgets/variableexplorer/arrayeditor.pyRws cCs2|jj||_|jjjj|jdS(N(RbtwidgetRaRRRRw(RR((sJlib/python2.7/site-packages/spyder/widgets/variableexplorer/arrayeditor.pyRyscCsQdgd}d||j<|jjddj|d||dkre|jj|j|}n|}tdgd}|||j<|j|jj |}|dkr=|j j }y*|j j t ||jt|Wn-tk r|j j t ||jnX||j|j|<|j jn|j j|dS( s This is implemented for handling negative values in index for 3d arrays, to give the same behavior as slicing t:is%is Slicing: [s, t]iN(RfRRtjoinRR}tsliceRReRRbtcountRNRFttuplet IndexErrortupdateR(RRt string_indext data_indext slice_indext stack_index((sJlib/python2.7/site-packages/spyder/widgets/variableexplorer/arrayeditor.pyR|s(         cCs||_dgd}d||<|jjddj|d|jj|jjdkrs|jjdn |j d|jj |jj| |jj|dd S( s] This change the active axis the array editor is plotting over in 3D s%iis%isShape: (s, s) iiN( RfRRRRR}R{RtsetValueR|tsetRange(RRt string_size((sJlib/python2.7/site-packages/spyder/widgets/variableexplorer/arrayeditor.pyRs    cCsGx3t|jjD]}|jj|jqWtj|dS(sReimplement Qt methodN(RRbRRRYRR-(RR((sJlib/python2.7/site-packages/spyder/widgets/variableexplorer/arrayeditor.pyR--scCs|jS(s-Return modified array -- this is *not* a copy(R(R((sJlib/python2.7/site-packages/spyder/widgets/variableexplorer/arrayeditor.pyR4scCs7tj|td||jtj|jdS(s(An error occured, closing the dialog boxs Array editorN(RRR"R_RR`R(Rtmessage((sJlib/python2.7/site-packages/spyder/widgets/variableexplorer/arrayeditor.pyRr:scCsY|jdk rHx6t|jjD]}|jj|jq%Wntj|dS(sReimplement Qt methodN( RaRRRbRRRZRR(RR((sJlib/python2.7/site-packages/spyder/widgets/variableexplorer/arrayeditor.pyR@sN(RRRRRlRRR RRwRyR|RR-RRrR(((sJlib/python2.7/site-packages/spyder/widgets/variableexplorer/arrayeditor.pyR^Ps      (LRt __future__Rt qtpy.compatRRt qtpy.QtCoreRRRRRRR t qtpy.QtGuiR R R R tqtpy.QtWidgetsRRRRRRRRRRRRRRRRRRR R!tnumpyRrtspyder.config.baseR"tspyder.config.fontsR#tspyder.config.guiR$R%tspyder.py3compatR&R'R(R)R*R+R,t spyder.utilsR-R(tspyder.utils.qthelpersR.R/R0RJRRRR\R_RjRkRRRFR^(((sJlib/python2.7/site-packages/spyder/widgets/variableexplorer/arrayeditor.pyt s4" 4    +|D