o º¼tf.Hã@söddlmZddlZddlZddlmZddlmZmZm Z ddl m Z ddl m Z ddlmZmZmZdd lmZdd lmZzdd lmZWn eySdZYnwdd d„ZGdd„deƒZGdd„deƒZGdd„deƒZGdd„deƒZdS)é)Ú b64decodeN)Úwarn)ÚQtCoreÚQtGuiÚ QtWidgets)ÚBool)Ú ClassNotFound)Úsave_svgÚsvg_to_clipboardÚ svg_to_imageé)Ú JupyterWidget)Ú get_colors)Ú latex_to_pngéíc Csrtj |¡s+z tj||dWdSty*}z|jtjkr‚WYd}~dSd}~wwtj |¡s7td|ƒ‚dS)zòensure that a directory exists If it doesn't exists, try to create it and protect against a race condition if another process is doing the same. The default permissions are 755, which differ from os.makedirs default of 777. )ÚmodeNz %r exists but is not a directory) ÚosÚpathÚexistsÚmakedirsÚOSErrorÚerrnoÚEEXISTÚisdirÚIOError)rrÚe©rúf/var/www/html/software/conda/envs/catlas/lib/python3.10/site-packages/qtconsole/rich_jupyter_widget.pyÚ_ensure_dir_existss  ÿ€ÿ  ÿrc@óeZdZdZdS)Ú LatexErrorzException for Latex errorsN©Ú__name__Ú __module__Ú __qualname__Ú__doc__rrrrr )ór c@r)ÚRichIPythonWidgetz4Dummy class for config inheritance. Destroyed below.Nr!rrrrr'-r&r'csòeZdZdZdZedƒZdZ‡fdd„Z‡fdd„Z ‡fdd „Z d d „Z ‡fd d „Z ‡fdd„Z dd„Zdd„Zd1dd„Zd1dd„Zd1dd„Zd2dd„Zdd„Zdd „Zd!d"„Zd3d$d%„Zd4d&d'„Zd4d(d)„Zd4d*d+„Zd,d-„Zd5d/d0„Z‡ZS)6ÚRichJupyterWidgetz² An JupyterWidget that supports rich text, including lists, images, and tables. Note that raw performance will be reduced compared to the plain text version. z0ipykernel.pylab.backend_payload.add_plot_payloadFcs>d|d<tƒj|i|¤Ž|j|j_i|_dtj ¡v|_ dS)z% Create a RichJupyterWidget. ÚrichÚkindZjpegN) ÚsuperÚ__init__Ú_get_image_tagZ_html_exporterZ image_tagÚ_name_to_svg_maprZ QImageReaderZsupportedImageFormatsÚ_jpg_supported)ÚselfÚargsÚkw©Ú __class__rrr,Cs  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)r0r3rrr6YszRichJupyterWidget.export_htmlcs¼ˆj |¡ ¡}| tjj¡‰ˆrVt ˆ¡}|  d‡‡fdd„¡|  d‡‡fdd„¡|  ¡ˆj   ˆd¡‰ˆdurT|  ¡|  d‡fdd„¡|  d ‡‡fd d„¡|St ƒ |¡}|S) zC Reimplemented to return a custom context menu for images. z Copy Imagecó ˆ ˆ¡S©N)Ú _copy_imager©Únamer0rrÚoó z6RichJupyterWidget._context_menu_make..zSave Image As...cr7r8)Ú _save_imagerr:rrr<pr=NzCopy SVGcstˆƒSr8)r r)Úsvgrrr<vszSave SVG As...cs tˆˆjƒSr8)r Ú_controlr)r0r?rrr<xs )r@ZcursorForPositionZ charFormatZstringPropertyrZ QTextFormatZ ImageNamerZQMenuZ addActionZ addSeparatorr.Úgetr+Ú_context_menu_make)r0ÚposÚformatZmenur3)r;r0r?rrBgs"  ÿ ÿ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)r0ÚmsgÚ prompt_numberrrrÚ_pre_image_append€sz#RichJupyterWidget._pre_image_appendcs–|j d| dd¡¡| |¡rÉ| ¡|d}| dd¡}|d}|dd}d|vrC| ||¡| |dd ¡| |jd ¡d Sd |vrk| ||¡t |d   d ¡ƒ}|j |d | d d ¡d | |jd ¡d Sd|vr–|j r–| ||¡t |d  d ¡ƒ}|j |d | dd ¡d | |jd ¡d Sd|vrÃ| ||¡z | |dd ¡Wnty¹tƒ |¡YSw| |jd ¡d Stƒ |¡Sd S)ú/Overridden to handle rich data types, like SVG.zexecute_result: %sÚcontentÚZexecution_countrÚdataÚmetadataú image/svg+xmlTú image/pngÚasciiN©rNú image/jpegú text/latex)ÚlogÚdebugrAÚinclude_outputÚflush_clearoutputrIÚ _append_svgrFZ output_sep2rÚencodeÚ _append_pngr/Ú _append_jpgÚ _append_latexr r+Ú_handle_display_dataÚ_handle_execute_result)r0rGrKrHrMrNÚpngÚjpgr3rrr_‰sF     ÿ ÿ  ÿ áz(RichJupyterWidget._handle_execute_resultcs:|j d| dd¡¡| |¡r›| ¡|dd}|dd}|j d| dd¡¡d|vr;|d}| |d¡d Sd |vrVt|d  d ¡ƒ}|j|d| d d ¡d d Sd |vrt|j rtt|d  d ¡ƒ}|j |d| d d ¡d d Sd|vr•t r•z |  |dd¡Wd St y”tƒ |¡YSwtƒ |¡Sd S)rJzdisplay_data: %srKrLrMrNz display: %srOTrPrQNrRrSrT)rUrVrArWrXrYrrZr[r/r\rr]r r+r^)r0rGrMrNr?r`rar3rrr^­s.     ÿ çz&RichJupyterWidget._handle_display_datac Csœddg}gd¢}dd„|Dƒ}|||}dd„|Dƒ}gd¢}||D]'\}} |t|ƒt| ƒ …} || vs;| | vr>dS| |¡rK| | ¡rKd Sq$dS) zs Determine if a Latex string is in math mode This is the only mode supported by qtconsole ÚmathZ displaymath)ZequationZeqnarraymultlineÚgatherÚalignZflalignZalignatcSsg|]}|d‘qS)Ú*r©Ú.0ÚenvrrrÚ ×sz4RichJupyterWidget._is_latex_math..cSsg|] }d |¡ ¡‘qS)z\begin{{{0}}} \end{{{0}}})rDÚsplitrfrrrriÚs))z\[z\])z\(z\))ú$$rk)ú$rlFT)ÚlenÚ startswithÚendswith) r0ÚlatexZ basic_envsZ starable_envsZ star_envsÚenvsZ env_syntaxZ math_syntaxÚstartÚendÚinnerrrrÚ_is_latex_mathÎs €z RichJupyterWidget._is_latex_mathcCs0zt|jƒ|WStytdƒ|YSw)z4Get color from the current syntax style if loadable.Údefault)rZ syntax_styler)r0ÚcolorrrrÚ _get_colorés  ýzRichJupyterWidget._get_colorNcCsˆd}| |¡rt|dd| d¡d}|dur7| d¡r7| d¡r7z t|dd| d¡d}Wn ty6Ynw|rB| |||¡dSt‚)z! Append latex data to the widget.NFZdvipngZfgcolor)ÚwrapÚbackendrwrlZ matplotlib)rurrxrnroÚ Exceptionr[r )r0rpÚ before_promptrNr`rrrr]òs  ÿ ÿ ÿzRichJupyterWidget._append_latexcCó|j|j|||ddS)z# Append raw JPG data to the widget.rRN)Ú_append_customÚ _insert_jpg)r0rar|rNrrrr\ szRichJupyterWidget._append_jpgcCr})z, Append raw PNG data to the widget. rRN)r~Ú _insert_png)r0r`r|rNrrrr[ szRichJupyterWidget._append_pngcCs| |j||¡dS)z, Append raw SVG data to the widget. N)r~Ú _insert_svg)r0r?r|rrrrYszRichJupyterWidget._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ÚQUrlZQTextImageFormatÚsetName)r0Úimager‚r;rDrrrÚ _add_images    ÿ zRichJupyterWidget._add_imagecCs| |¡}tj ¡ |¡dS)z@ Copies the ImageResource with 'name' to the clipboard. N)Ú _get_imagerZ QApplicationÚ clipboardZsetImage)r0r;rˆrrrr9#s zRichJupyterWidget._copy_imagecCs$|j ¡}| tjjt |¡¡}|S)zE Returns the QImage stored as the ImageResource with 'name'. )r@r‚Úresourcerr„r…rr†)r0r;r‚rˆrrrrŠ)s  ÿzRichJupyterWidget._get_imager`c Csj|dvrvz | | d¡¡}Wntyd| d¡YSw|durGt|ƒtj |¡}| d|| d¡|fd¡rEd|| d¡|fSdSt  ¡}t  |¡}|  tj j ¡| || ¡¡| ¡d |t d d t| ¡ ¡ ¡ƒ¡fS|d kr³z t|j| d¡ƒ}Wnty¡|jsžtj |d dtjj¡d|_YdSw| d¡} | dks­J‚|| d…SdS)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. )r`rar;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ÚKeyErrorrrrÚbasenameÚsaverZ QByteArrayZQBufferÚopenZ QIODeviceZ WriteOnlyÚupperÚcloseÚreÚsubrƒZtoBase64rMÚdecoder.r5rZ QMessageBoxÚwarningZOkÚfind) r0ÚmatchrrDrˆÚrelpathÚbaZbuffer_r?Úoffsetrrrr-1sP ÿ ÿ ÿ ÿ  úö   z RichJupyterWidget._get_image_tagcCó|j||d|ddS)z% Insert raw PNG data into the widget.rarRN©Ú _insert_img)r0ÚcursorrarNrrrrzszRichJupyterWidget._insert_jpgcCrŸ)z. Insert raw PNG data into the widget. r`rRNr )r0r¢r`rNrrrr€~szRichJupyterWidget._insert_pngc Csô|r| dd¡}| dd¡}nd}}z@t ¡}| || ¡¡|r5|r5| t|ƒt|ƒtjj tjj ¡}n|rD|sD|  t|ƒtjj ¡}n|rR|sR|  t|ƒtjj ¡}Wnt ye| |d|¡YdSw| |¡}| ¡| |¡| ¡dS)z insert a raw image, jpg or png ÚwidthNÚheightzReceived invalid %s data.)rArZQImageZ loadFromDatar”ZscaledÚintrZQtZIgnoreAspectRatioZSmoothTransformationZ scaledToWidthZscaledToHeightÚ ValueErrorÚ_insert_plain_textr‰Ú insertBlockÚ insertImage) r0r¢ZimgÚfmtrNr£r¤rˆrDrrrr¡ƒs0 þ€ ÿ   zRichJupyterWidget._insert_imgcCsdzt|ƒ}Wnty| |d¡YdSw| |¡}||j| ¡<| ¡| |¡| ¡dS)z- Insert raw SVG data into the widet. zReceived invalid SVG data.N)r r¦r§r‰r.r;r¨r©)r0r¢r?rˆrDrrrrs  ÿ   zRichJupyterWidget._insert_svgrcCsrt |jd¡}| tjj¡| | ¡¡| d|| ¡f¡| ¡r7|  ¡d}|  |¡}|  ||¡dSdS)z@ Shows a save dialog for the ImageResource with 'name'. z Save Imagez%s file (*.%s)rN) rZ QFileDialogr@Z setAcceptModeZ AcceptSaveZsetDefaultSuffixÚlowerZ setNameFilterÚexec_Z selectedFilesrŠr’)r0r;rDZdialogÚfilenamerˆrrrr>«s  ýzRichJupyterWidget._save_image)FN)F)Nr`r8)r)r"r#r$r%Z_payload_source_plotrr/r5r,r6rBrIr_r^rurxr]r\r[rYr‰r9rŠr-rr€r¡rr>Ú __classcell__rrr3rr(1s4    $!      I  r(cs eZdZdZ‡fdd„Z‡ZS)r'z(Deprecated class. Use RichJupyterWidget.cs tdtƒtƒj|i|¤ŽdS)Nz6RichIPythonWidget is deprecated, use RichJupyterWidget)rÚDeprecationWarningr+r,)r0Úar2r3rrr,¼sÿzRichIPythonWidget.__init__)r"r#r$r%r,r®rrr3rr'ºs)r)Úbase64rrr–ÚwarningsrZqtpyrrrZ traitletsrZ pygments.utilrZ qtconsole.svgr r r Zjupyter_widgetr ÚstylesrZIPython.lib.latextoolsrÚ ImportErrorrr{r r'r(rrrrÚs.       ÿ