B #õå[QDã@sÞddlmZddlZddlZddlmZddlmZmZddl m Z ddl m Z ddl mZmZmZdd lmZydd lmZWnek r˜dZYnXGd d „d eƒZGd d„deƒZGdd„deƒZGdd„deƒZdS)é)Ú b64decodeN)Úwarn)ÚQtCoreÚQtGui)Úensure_dir_exists)ÚBool)Úsave_svgÚsvg_to_clipboardÚ svg_to_imageé)Ú JupyterWidget)Ú latex_to_pngc@seZdZdZdS)Ú LatexErrorzException for Latex errorsN)Ú__name__Ú __module__Ú __qualname__Ú__doc__©rrúd|d<tt|ƒj||Ž|j|j_i|_dtj  ¡k|_ dS)z% Create a RichJupyterWidget. ZrichZkindZjpegN) ÚsuperrÚ__init__Ú_get_image_tagZ_html_exporterZ image_tagÚ_name_to_svg_maprZ QImageReaderZsupportedImageFormatsÚ_jpg_supported)ÚselfÚargsÚkw)Ú __class__rrr1s  zRichJupyterWidget.__init__csd|_tt|ƒ ¡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_displayedrrÚ export_html)r)rrrr!GszRichJupyterWidget.export_htmlcs¼ˆj |¡ ¡}| tjj¡‰ˆr¨t ¡}| d‡‡fdd„¡| d‡‡fdd„¡|  ¡ˆj   ˆd¡‰ˆdk r¸|  ¡| d‡fdd„¡| d ‡‡fd d„¡nt t ˆƒ |¡}|S) zC Reimplemented to return a custom context menu for images. z Copy Imagecs ˆ ˆ¡S)N)Ú _copy_imager)ÚnamerrrÚ]sz6RichJupyterWidget._context_menu_make..zSave Image As...cs ˆ ˆ¡S)N)Ú _save_imager)r#rrrr$^sNzCopy SVGcstˆƒS)N)r r)Úsvgrrr$dszSave SVG As...cs tˆˆjƒS)N)rÚ_controlr)rr&rrr$fs)r'ZcursorForPositionZ charFormatZstringPropertyrZ QTextFormatZ ImageNameZQMenuZ addActionZ addSeparatorrÚgetrrÚ_context_menu_make)rÚposÚformatZmenu)r)r#rr&rr)Usz$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_appendnsz#RichJupyterWidget._pre_image_appendcs |j d| dd¡¡| |¡rœ| ¡|d}| dd¡}|d}|dd}d|krˆ| ||¡| |dd ¡| |jd ¡nd |krÖ| ||¡t |d   d ¡ƒ}|j |d | d d ¡d | |jd ¡nÆd|kr.|j r.| ||¡t |d  d ¡ƒ}|j |d | dd ¡d | |jd ¡nnd|krŒ| ||¡y| |dd ¡Wn"tk rztt|ƒ |¡SX| |jd ¡ntt|ƒ |¡Sd S)z/Overridden to handle rich data types, like SVG.zexecute_result: %sÚcontentÚZexecution_countrÚdataÚmetadataz image/svg+xmlTz image/pngÚasciiN)r4z image/jpegz text/latex)ÚlogÚdebugr(Úinclude_outputÚflush_clearoutputr0Ú _append_svgr-Z output_sep2rÚencodeÚ _append_pngrÚ _append_jpgÚ _append_latexrrrÚ_handle_display_dataÚ_handle_execute_result)rr.r1r/r3r4ÚpngÚjpg)rrrr@ws>          z(RichJupyterWidget._handle_execute_resultcsB|j d| dd¡¡| |¡r>| ¡|dd}|dd}|j d| dd¡¡d|krv|d}| |d¡nÈd |krªt|d  d ¡ƒ}|j|d| d d ¡d n”d |krä|j rät|d  d ¡ƒ}|j |d| d d ¡d nZd|kr.t r.y|  |dd¡Wn"t k r*tt|ƒ |¡SXntt|ƒ |¡Sd S)z/Overridden to handle rich data types, like SVG.zdisplay_data: %sr1r2r3r4z display: %sz image/svg+xmlTz image/pngr5N)r4z image/jpegz text/latex)r6r7r(r8r9r:rr;r<rr=r r>rrrr?)rr.r3r4r&rArB)rrrr?›s*   z&RichJupyterWidget._handle_display_datac Cs¨ddg}ddddddg}d d „|Dƒ}|||}d d „|Dƒ}d d ddg}xV||D]J\}} |t|ƒt| ƒ …} || ks„| | krˆdS| |¡rV| | ¡rVdSqVWdS)zs Determine if a Latex string is in math mode This is the only mode supported by qtconsole ZmathZ displaymathZequationZeqnarraymultlineZgatherZalignZflalignZalignatcSsg|] }|d‘qS)Ú*r)Ú.0Úenvrrrú Åsz4RichJupyterWidget._is_latex_math..cSsg|]}d |¡ ¡‘qS)z\begin{{{0}}} \end{{{0}}})r+Úsplit)rDrErrrrFÈs)z\[z\])z\(z\))z$$z$$)ú$rHFT)Ú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_mathNcCs|d}| |¡rt|ddd}|dkr`| d¡r`| d¡r`yt|ddd}Wntk r^YnX|rt| |||¡nt‚dS)z! Append latex data to the widget.NFZdvipng)ZwrapZbackendrHZ matplotlib)rPr rJrKÚ Exceptionr<r)rrLÚ before_promptr4rArrrr>×s zRichJupyterWidget._append_latexcCs|j|j|||ddS)z# Append raw JPG data to the widget.)r4N)Ú_append_customÚ _insert_jpg)rrBrRr4rrrr=ìszRichJupyterWidget._append_jpgcCs|j|j|||ddS)z, Append raw PNG data to the widget. )r4N)rSÚ _insert_png)rrArRr4rrrr<ðszRichJupyterWidget._append_pngcCs| |j||¡dS)z, Append raw SVG data to the widget. N)rSÚ _insert_svg)rr&rRrrrr:õszRichJupyterWidget._append_svgcCsD|j ¡}t| ¡ƒ}| tjjt  |¡|¡t  ¡}|  |¡|S)zr Adds the specified QImage to the document and returns a QTextImageFormat that references it. ) r'ÚdocumentÚstrZcacheKeyZ addResourcerÚ QTextDocumentÚ ImageResourcerÚQUrlZQTextImageFormatZsetName)rÚimagerWr#r+rrrÚ _add_imageús    zRichJupyterWidget._add_imagecCs| |¡}tj ¡ |¡dS)z@ Copies the ImageResource with 'name' to the clipboard. N)Ú _get_imagerZ QApplicationZ clipboardZsetImage)rr#r\rrrr"s zRichJupyterWidget._copy_imagecCs$|j ¡}| tjjt |¡¡}|S)zE Returns the QImage stored as the ImageResource with 'name'. )r'rWÚresourcerrYrZrr[)rr#rWr\rrrr^ s   zRichJupyterWidget._get_imagerAc Csx|dkrîy| | d¡¡}Wntk r:d| d¡SX|dk rŽt|ƒtj |¡}| d|| d¡|fd¡rˆd|| d¡|fSdSn^t  ¡}t  |¡}|  tj j ¡| || ¡¡| ¡d |t d d t| ¡ ¡ ¡ƒ¡fSn†|d krpyt|j| d¡ƒ}Wn:tk rJ|jsFtj |d dtjj¡d|_dSX| d¡} | dksdt‚|| 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. )rArBr#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)r^ÚgroupÚKeyErrorrÚosÚpathÚbasenameÚsaverZ QByteArrayZQBufferÚopenZ QIODeviceZ WriteOnlyÚupperÚcloseÚreÚsubrXZtoBase64r3Údecoderr rZ QMessageBoxZwarningZOkÚfindÚAssertionError) rÚmatchrer+r\ÚrelpathZbaZbuffer_r&ÚoffsetrrrrsD  &     z RichJupyterWidget._get_image_tagcCs|j||d|ddS)z% Insert raw PNG data into the widget.rB)r4N)Ú _insert_img)rÚcursorrBr4rrrrT]szRichJupyterWidget._insert_jpgcCs|j||d|ddS)z. Insert raw PNG data into the widget. rA)r4N)rs)rrtrAr4rrrrUaszRichJupyterWidget._insert_pngc Csä|r| dd¡}| dd¡}nd}}ypt ¡}| || ¡¡|rb|rb| ||tjjtjj ¡}n2|r||s||  |tjj ¡}n|r”|s”|  |tjj ¡}Wn$t k rº|  |d|¡Yn&X| |¡}| ¡| |¡| ¡dS)z insert a raw image, jpg or png ÚwidthNÚheightzReceived invalid %s data.)r(rZQImageZ loadFromDatariZscaledrZQtZIgnoreAspectRatioZSmoothTransformationZ scaledToWidthZscaledToHeightÚ ValueErrorÚ_insert_plain_textr]Ú insertBlockÚ insertImage) rrtZimgZfmtr4rurvr\r+rrrrsfs*    zRichJupyterWidget._insert_imgcCsdy t|ƒ}Wn tk r,| |d¡Yn4X| |¡}||j| ¡<| ¡| |¡| ¡dS)z- Insert raw SVG data into the widet. zReceived invalid SVG data.N)r rwrxr]rr#ryrz)rrtr&r\r+rrrrV€s   zRichJupyterWidget._insert_svgr`cCsnt |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 QFileDialogr'Z setAcceptModeZ AcceptSaveZsetDefaultSuffixÚlowerZ setNameFilterZexec_Z selectedFilesr^rg)rr#r+ZdialogÚfilenamer\rrrr%Žs  zRichJupyterWidget._save_image)FN)FN)FN)F)NrA)N)N)N)r`)rrrrZ_payload_source_plotrrr rr!r)r0r@r?rPr>r=r<r:r]r"r^rrTrUrsrVr%Ú __classcell__rr)rrrs0    $ !      I   rcs eZdZdZ‡fdd„Z‡ZS)rz(Deprecated class. Use RichJupyterWidget.cs tdtƒtt|ƒj||ŽdS)Nz6RichIPythonWidget is deprecated, use RichJupyterWidget)rÚDeprecationWarningrrr)rÚar)rrrrŸszRichIPythonWidget.__init__)rrrrrr}rr)rrrs)Úbase64rrdrkÚwarningsrZ qtconsole.qtrrZipython_genutils.pathrZ traitletsrZ qtconsole.svgrr r Zjupyter_widgetr ZIPython.lib.latextoolsr Ú ImportErrorrQrrrrrrrÚs&