a œòJbFã@søddlmZddlZddlZddlmZddlmZmZm Z ddl m Z ddl m Z ddlmZddlmZmZmZd d lmZd d lmZzdd lmZWney²dZYn0Gd d„deƒZGdd„deƒZGdd„deƒZGdd„deƒZdS)é)Ú b64decodeN)Úwarn)ÚQtCoreÚQtGuiÚ QtWidgets)Úensure_dir_exists)ÚBool)Ú ClassNotFound)Úsave_svgÚsvg_to_clipboardÚ svg_to_imageé)Ú JupyterWidget)Ú get_colors)Ú latex_to_pngc@seZdZdZdS)Ú LatexErrorzException for Latex errorsN©Ú__name__Ú __module__Ú __qualname__Ú__doc__©rrúd|d<tƒj|i|¤Ž|j|j_i|_dtj ¡v|_ dS)z% Create a RichJupyterWidget. ZrichÚkindZjpegN) ÚsuperÚ__init__Ú_get_image_tagZ_html_exporterZ image_tagÚ_name_to_svg_maprZ QImageReaderZsupportedImageFormatsÚ_jpg_supported)ÚselfÚargsÚkw©Ú __class__rrr3s  zRichJupyterWidget.__init__csd|_tƒ ¡dS)zª Shows a dialog to export HTML/XML in various formats. Overridden in order to reset the _svg_warning_displayed flag prior to the export running. FN)Ú_svg_warning_displayedrÚ export_html)r!r$rrr'IszRichJupyterWidget.export_htmlcsºˆj |¡ ¡}| tjj¡‰ˆrªt ˆ¡}|  d‡‡fdd„¡|  d‡‡fdd„¡|  ¡ˆj   ˆd¡‰ˆdur¶|  ¡|  d‡fdd„¡|  d ‡‡fd d„¡n t ƒ |¡}|S) zC Reimplemented to return a custom context menu for images. z Copy Imagecs ˆ ˆ¡S©N)Ú _copy_imager©Únamer!rrÚ_óz6RichJupyterWidget._context_menu_make..zSave Image As...cs ˆ ˆ¡Sr()Ú _save_imagerr*rrr,`r-NzCopy SVGcstˆƒSr()r r)Úsvgrrr,fr-zSave SVG As...cs tˆˆjƒSr()r Ú_controlr)r!r/rrr,hr-)r0ZcursorForPositionZ charFormatZstringPropertyrZ QTextFormatZ ImageNamerZQMenuZ addActionZ addSeparatorrÚgetrÚ_context_menu_make)r!ÚposÚformatZmenur$)r+r!r/rr2Ws   ÿ z$RichJupyterWidget._context_menu_makecCs0| |jd¡| | |¡d¡| dd¡dS)ztAppend the Out[] prompt and make the output nicer Shared code for some the following if statement TÚ N)Z_append_plain_textZ output_sepÚ _append_htmlZ_make_out_prompt)r!ÚmsgÚ prompt_numberrrrÚ_pre_image_appendpsz#RichJupyterWidget._pre_image_appendcsš|j d| dd¡¡| |¡r–| ¡|d}| dd¡}|d}|dd}d|vrˆ| ||¡| |dd ¡| |jd ¡nd |vrÖ| ||¡t |d   d ¡ƒ}|j |d | d d ¡d | |jd ¡nÀd|vr.|j r.| ||¡t |d  d ¡ƒ}|j |d | dd ¡d | |jd ¡nhd|vrŠ| ||¡z| |dd ¡Wn tyxtƒ |¡YS0| |jd ¡n tƒ |¡Sd S)ú/Overridden to handle rich data types, like SVG.zexecute_result: %sÚcontentÚZexecution_countrÚdataÚmetadataú image/svg+xmlTú image/pngÚasciiN©r>ú image/jpegú text/latex)ÚlogÚdebugr1Úinclude_outputÚflush_clearoutputr9Ú _append_svgr6Z output_sep2rÚencodeÚ _append_pngr Ú _append_jpgÚ _append_latexrrÚ_handle_display_dataÚ_handle_execute_result)r!r7r;r8r=r>ÚpngÚjpgr$rrrOysB     ÿ ÿ  z(RichJupyterWidget._handle_execute_resultcs<|j d| dd¡¡| |¡r8| ¡|dd}|dd}|j d| dd¡¡d|vrv|d}| |d¡nÂd |vrªt|d  d ¡ƒ}|j|d| d d ¡d nŽd |vrä|j rät|d  d ¡ƒ}|j |d| d d ¡d nTd|vr,t r,z|  |dd¡Wn t y(tƒ |¡YS0n tƒ |¡Sd S)r:zdisplay_data: %sr;r<r=r>z display: %sr?Tr@rANrBrCrD)rErFr1rGrHrIrrJrKr rLrrMrrrN)r!r7r=r>r/rPrQr$rrrNs*   z&RichJupyterWidget._handle_display_datac Csœddg}gd¢}dd„|Dƒ}|||}dd„|Dƒ}gd¢}||D]N\}} |t|ƒt| ƒ …} || vsv| | vr|dS| |¡rH| | ¡rHd SqHdS) zs Determine if a Latex string is in math mode This is the only mode supported by qtconsole ZmathZ displaymath)ZequationZeqnarraymultlineZgatherZalignZflalignZalignatcSsg|] }|d‘qS)Ú*r©Ú.0ÚenvrrrÚ Çr-z4RichJupyterWidget._is_latex_math..cSsg|]}d |¡ ¡‘qS)z\begin{{{0}}} \end{{{0}}})r4ÚsplitrSrrrrVÊr-))z\[z\])z\(z\))ú$$rX)ú$rYFT)ÚlenÚ startswithÚendswith) r!ÚlatexZ basic_envsZ starable_envsZ star_envsZenvsZ env_syntaxZ math_syntaxÚstartÚendÚinnerrrrÚ_is_latex_math¾s z RichJupyterWidget._is_latex_mathcCs4zt|jƒ|WSty.tdƒ|YS0dS)z4Get color from the current syntax style if loadable.ÚdefaultN)rZ syntax_styler )r!ÚcolorrrrÚ _get_colorÙs zRichJupyterWidget._get_colorNcCsŠd}| |¡r$t|dd| d¡d}|durn| d¡rn| d¡rnzt|dd| d¡d}WntylYn0|r‚| |||¡nt‚dS)z! Append latex data to the widget.NFZdvipngZfgcolor)ÚwrapZbackendrcrYZ matplotlib)rarrdr[r\Ú ExceptionrKr)r!r]Ú before_promptr>rPrrrrMâs ÿÿ  zRichJupyterWidget._append_latexcCs|j|j|||ddS)z# Append raw JPG data to the widget.rBN)Ú_append_customÚ _insert_jpg)r!rQrgr>rrrrLùszRichJupyterWidget._append_jpgcCs|j|j|||ddS)z, Append raw PNG data to the widget. rBN)rhÚ _insert_png)r!rPrgr>rrrrKýszRichJupyterWidget._append_pngcCs| |j||¡dS)z, Append raw SVG data to the widget. N)rhÚ _insert_svg)r!r/rgrrrrIszRichJupyterWidget._append_svgcCsD|j ¡}t| ¡ƒ}| tjjt  |¡|¡t  ¡}|  |¡|S)zr Adds the specified QImage to the document and returns a QTextImageFormat that references it. ) r0ÚdocumentÚstrZcacheKeyZ addResourcerÚ QTextDocumentÚ ImageResourcerÚQUrlZQTextImageFormatÚsetName)r!Úimagerlr+r4rrrÚ _add_images    ÿ zRichJupyterWidget._add_imagecCs| |¡}tj ¡ |¡dS)z@ Copies the ImageResource with 'name' to the clipboard. N)Ú _get_imagerZ QApplicationZ clipboardZsetImage)r!r+rrrrrr)s zRichJupyterWidget._copy_imagecCs$|j ¡}| tjjt |¡¡}|S)zE Returns the QImage stored as the ImageResource with 'name'. )r0rlÚresourcerrnrorrp)r!r+rlrrrrrrts   ÿzRichJupyterWidget._get_imagerPc Csz|dvrðz| | d¡¡}Wn ty<d| d¡YS0|durt|ƒtj |¡}| d|| d¡|fd¡rŠd|| d¡|fSdSn^t  ¡}t  |¡}|  tj j ¡| || ¡¡| ¡d |t d d t| ¡ ¡ ¡ƒ¡fSn†|d krrzt|j| d¡ƒ}Wn:tyL|jsFtj |d dtjj¡d|_YdS0| d¡} | dksfJ‚|| d…SdSdS)ag Return (X)HTML mark-up for the image-tag given by match. Parameters ---------- match : re.SRE_Match A match to an HTML image tag as exported by Qt, with match.group("Name") containing the matched image ID. path : string|None, optional [default None] If not None, specifies a path to which supporting files may be written (e.g., for linked images). If None, all images are to be included inline. format : "png"|"svg"|"jpg", optional [default "png"] Format for returned or referenced images. )rPrQr+zCouldn't find image %sNz%s/qt_img%s.%sÚPNGzzCouldn't save image!z'z(.{60})z\1\nr/zError converting PNG to SVG.zÔCannot convert PNG images to SVG, export with PNG figures instead. If you want to export matplotlib figures as SVG, add to your ipython config: c.InlineBackend.figure_format = 'svg' And regenerate the figures.TzëCannot convert PNG images to SVG. You must export this session with PNG images. If you want to export matplotlib figures as SVG, add to your config c.InlineBackend.figure_format = 'svg' and regenerate the figures.zUnrecognized image format)rtÚgroupÚKeyErrorrÚosÚpathÚbasenameÚsaverZ QByteArrayZQBufferÚopenZ QIODeviceZ WriteOnlyÚupperÚcloseÚreÚsubrmZtoBase64r=Údecoderr&rZ QMessageBoxZwarningZOkÚfind) r!Úmatchr{r4rrÚrelpathZbaZbuffer_r/Úoffsetrrrr!sL  ÿ ÿ ÿ  ú  z RichJupyterWidget._get_image_tagcCs|j||d|ddS)z% Insert raw PNG data into the widget.rQrBN©Ú _insert_img)r!ÚcursorrQr>rrrrijszRichJupyterWidget._insert_jpgcCs|j||d|ddS)z. Insert raw PNG data into the widget. rPrBNrˆ)r!rŠrPr>rrrrjnszRichJupyterWidget._insert_pngc Csâ|r| dd¡}| dd¡}nd}}zpt ¡}| || ¡¡|rb|rb| ||tjjtjj ¡}n2|r||s||  |tjj ¡}n|r”|s”|  |tjj ¡}Wn"t y¸|  |d|¡Yn&0| |¡}| ¡| |¡| ¡dS)z insert a raw image, jpg or png ÚwidthNÚheightzReceived invalid %s data.)r1rZQImageZ loadFromDatarZscaledrZQtZIgnoreAspectRatioZSmoothTransformationZ scaledToWidthZscaledToHeightÚ ValueErrorÚ_insert_plain_textrsÚ insertBlockÚ insertImage) r!rŠZimgZfmtr>r‹rŒrrr4rrrr‰ss, þ   zRichJupyterWidget._insert_imgcCsbz t|ƒ}Wnty*| |d¡Yn40| |¡}||j| ¡<| ¡| |¡| ¡dS)z- Insert raw SVG data into the widet. zReceived invalid SVG data.N)r rrŽrsrr+rr)r!rŠr/rrr4rrrrks    zRichJupyterWidget._insert_svgrvcCsnt |jd¡}| tjj¡| | ¡¡| d|| ¡f¡| ¡rj|  ¡d}|  |¡}|  ||¡dS)z@ Shows a save dialog for the ImageResource with 'name'. z Save Imagez%s file (*.%s)rN) rZ QFileDialogr0Z setAcceptModeZ AcceptSaveZsetDefaultSuffixÚlowerZ setNameFilterZexec_Z selectedFilesrtr})r!r+r4ZdialogÚfilenamerrrrrr.›s  zRichJupyterWidget._save_image)FN)FN)FN)F)NrP)N)N)N)rv)rrrrZ_payload_source_plotrr r&rr'r2r9rOrNrardrMrLrKrIrsr)rtrrirjr‰rkr.Ú __classcell__rrr$rr!s2    $ !      I   rcs eZdZdZ‡fdd„Z‡ZS)rz(Deprecated class. Use RichJupyterWidget.cs tdtƒtƒj|i|¤ŽdS)Nz6RichIPythonWidget is deprecated, use RichJupyterWidget)rÚDeprecationWarningrr)r!Úar#r$rrr¬sÿzRichIPythonWidget.__init__)rrrrrr“rrr$rrªs)Úbase64rrzrÚwarningsrZqtpyrrrZipython_genutils.pathrZ traitletsrZ pygments.utilr Z qtconsole.svgr r r Zjupyter_widgetrZstylesrZIPython.lib.latextoolsrÚ ImportErrorrfrrrrrrrÚs*