G@d:dZddlZddlmZddlZddlmZmZddl m Z ddl m Z m Z ddlmZmZmZmZmZmZmZmZddlmZmZddlmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$dd l%m&Z&dd l'm(Z(dd l)m*Z*dd l+m,Z,d Z-ddZ.Gdde$e(Z/Gddee(Z0GddeZ1Gdde$Z2GddeZ3dS)zI Figure browser widget This is the main widget used in the Plots plugin N)svg_to_clipboard svg_to_image)PYQT5)getexistingdirectorygetsavefilename)QEventQPointQRectQSizeQtQTimerSignalSlot)QPainterQPixmap) QApplicationQFrame QGridLayout QHBoxLayout QScrollArea QScrollBar QSplitterQStyle QVBoxLayoutQWidget)_)SpyderWidgetMixin)getcwd_or_home) QStylePalettecttj|\}}|dkr,|dkr&t|}||dS|dkr*t |t r|d}t|d5}||ddddS#1swxYwYdS)z1Save fig to fname in the format specified by fmt..png image/svg+xmlzutf-8wbN) ospsplitextrsave isinstancestrencodeopenwrite)figfmtfnamerootextqimgfs Jlib/python3.11/site-packages/spyder/plugins/plots/widgets/figurebrowser.pysave_figure_tofiler4's U##ID# f}}//C   % / ! !jc&:&: !**W%%C %   ! GGCLLL                  s B--B14B1Fcd}d||}|rd}d|||} tjtj||rd|||}|dz }ntj||SZ)zd Append a number to "root" to form a filename that does not already exist in "dirname". z{}{}rz {} ({}){})formatr$existsjoin)dirnamer/r0 start_at_zeroifignames r3get_unique_fignamer>5s AmmD#&&G3 $$T1c22. :chw00 1 1 .!((q#66G FAA8GW-- - .ceZdZdZeZ eeZ eeZ ee Z eee Z eeZ dfd ZdZdZdZdZdZd Zd Zd Zd Zd ZdZdZdZdZdZdZ xZ!S) FigureBrowserzt Widget to browse the figures that were sent by the kernel to the IPython console to be plotted inline. Nctr$t||n+tj||t j||d|_d|_d|_d|_||_ d|_ d|_ t||j |_|jj |j|jj|j|jj|j|jj|jt'|j||j |_|jj |j|jj|j|jj|jt1|x|_}||j||j|t9|ddddt?|}||| ||dddd|!d|dddddS)N)parent class_parentrDTFrCbackground_color)rCr)"rsuper__init__rr shellwidget is_visible figviewersetup_in_progressrGmute_inline_plottingzoom_disp_value FigureViewersig_context_menu_requestedconnectsig_figure_menu_requestedsig_figure_loadedsig_zoom_changed_update_zoom_valueThumbnailScrollBar thumbnails_sbsig_thumbnail_menu_requestedsig_save_dir_changedsig_redirect_stdio_requestedrsplitter addWidget setFrameStyler frameStylesetContentsMarginsr setLayout setSpacing)selfrCrGr\layout __class__s r3rIzFigureBrowser.__init__s  B GG  F  @ @ @ @  T6 * * *  &t& A A A A!& 0$(!#&T7;7LNNN 199  * , , , (001GHHH '//0EFFF '//0GHHH0 N!2    5==  - / / / /77  % ' ' ' 7??  - / / /$-D#9#9#99 4>***4-...{}}7799:::##Aq!Q///T""""" v!!!Q1---! 1a+++++r?c||_dS)z" Used in testing. N)rOrcvalues r3rVz FigureBrowser._update_zoom_values %r?c$|jdddd|D]]\}}|dkr||!|dkr||_/|dkr||K|dkr ||j_^dS)z/Setup the figure browser with provided options.rauto_fit_plottingrNshow_plot_outlinesave_dirN)r\r`itemschange_auto_fit_plottingrNshow_fig_outline_in_viewerrXrl)rcoptionsoptionrhs r3setupzFigureBrowser.setups ((Aq!444$]]__ 4 4MFE,,,--e4444111,1))...//6666:%%.3"+ 4 4r?cr|jj}||z dkr!|j||z |gdSdS)z Update the widths to provide the scrollbar with a fixed minimum width. Parameters ---------- base_width: int The available splitter width. rN)rX_min_scrollbar_widthr\setSizes)rc base_width min_sb_widths r3update_splitter_widthsz$FigureBrowser.update_splitter_widthssM)>  $q ( ( M " "J$=|#L M M M M M ) (r?c|dur.|jjdtjzdS|jjddS)z7Draw a frame around the figure viewer if state is True.T#FigureCanvas{border: 2px solid %s;}zFigureCanvas{border: 0px;}N)rL figcanvas setStyleSheetrCOLOR_BACKGROUND_4rcstates r3roz(FigureBrowser.show_fig_outline_in_viewersl D== N $ 2 2501      N $ 2 2, . . . . .r?c||j_dS)z5Change the auto_fit_plotting option and scale images.N)rLrjr~s r3rnz&FigureBrowser.change_auto_fit_plottings+0(((r?c|||_|||j|jdS)z3Bind the shellwidget instance to the figure browserN)rJset_figurebrowsersig_new_inline_figurerR_handle_new_figure)rcrJs r3set_shellwidgetzFigureBrowser.set_shellwidgets>&%%d+++)11$2IJJJJJr?c<|j||dS)z` Handle when a new figure is sent to the IPython console by the kernel. N)rX add_thumbnailrcr,r-s r3rz FigureBrowser._handle_new_figures# ((c22222r?c8|jdS)z9Zoom the figure in by a single step in the figure viewer.N)rLzoom_inrcs r3rzFigureBrowser.zoom_ins      r?c8|jdS)z:Zoom the figure out by a single step in the figure viewer.N)rLzoom_outrs r3rzFigureBrowser.zoom_outs !!!!!r?c8|jdS)zq Select the thumbnail previous to the currently selected one in the thumbnail scrollbar. N)rXgo_previous_thumbnailrs r3rz#FigureBrowser.go_previous_thumbnails 0022222r?c8|jdS)zm Select the thumbnail next to the currently selected one in the thumbnail scrollbar. N)rXgo_next_thumbnailrs r3rzFigureBrowser.go_next_thumbnails ,,.....r?c8|jdS)z>Save the currently selected figure in the thumbnail scrollbar.N)rXsave_current_figure_asrs r3 save_figurezFigureBrowser.save_figures 1133333r?c4|jS)z-Save all the figures in a selected directory.)rXsave_all_figures_asrs r3save_all_figureszFigureBrowser.save_all_figuress!55777r?c8|jdS)z?Close the currently selected figure in the thumbnail scrollbar.N)rXremove_current_thumbnailrs r3 close_figurezFigureBrowser.close_figures 3355555r?c8|jdS)z1Close all the figures in the thumbnail scrollbar.N)rXremove_all_thumbnailsrs r3close_all_figureszFigureBrowser.close_all_figuress 0022222r?cz|jr1|jjjr"|jjdSdSdS)z(Copy figure from figviewer to clipboard.N)rLr{r, copy_figurers r3rzFigureBrowser.copy_figuresL > 3dn6: 3 N $ 0 0 2 2 2 2 2 3 3 3 3r?NN)"__name__ __module__ __qualname____doc__rrTr rSboolr[r(rZobjectrYintrUrIrVrrrxrornrrrrrrrrrrr __classcell__res@r3rArAHs = &v$*6$<< "6#;;$*6& #9  vc{{1,1,1,1,1,1,f%%% 4 4 4 N N N . . .111KKK 333!!!"""333///4448886663333333333r?rAceZdZdZeeZ eeZ eZ dfd Z e dZ e j dZ dZdZdZd Zd Zd Zd Zd ZdZdZxZS)rPz A scrollarea that displays a single FigureCanvas with zooming and panning capability with CTRL + Mouse_wheel and Left-press mouse button event. Nchtr$t||n+tj||t j|||t j| d| dt|zdz| d||_ d|_d|_d|_d|_|d |_d |_dS) NrE figviewportz#figviewport {background-color:}rg333333? iF)rrHrIrr setAlignmentr AlignCenterviewport setObjectNamer|r(r^rG _scalefactor _scalestep_sfmax_sfminsetup_figcanvasrj _ispanningrcrCrGres r3rIzFigureViewer.__init__As   B GG  V&  9 9 9 9  v . . .  &t& A A A A ".))) %%m444 %% -4D0E0E E K M M M 1 0   !& r?c|jS)zW Return whether to automatically fit the plot to the scroll area size. )_auto_fit_plottingrs r3rjzFigureViewer.auto_fit_plottingZs &&r?c:||_|r?|tj|tjn>|tj|tj|dS)zT Set whether to automatically fit the plot to the scroll area size. N)rsetVerticalScrollBarPolicyr ScrollBarAlwaysOffsetHorizontalScrollBarPolicyScrollBarAsNeeded scale_imagergs r3rjzFigureViewer.auto_fit_plottingas #(  D  + +B,A B B B  - -b.C D D D D  + +B,@ A A A  - -b.B C C C r?ct||j|_|j||jj|j||jdS)zSetup the FigureCanvas.rFN) FigureCanvasrGr{installEventFiltercustomContextMenuRequestedrRshow_context_menu setWidgetrs r3rzFigureViewer.setup_figcanvasosv%T7;7LNNN ))$/// 199  " $ $ $ t~&&&&&r?c|jrB|jj8|j|}|j|dSdSdS)z3Only emit context menu signal if there is a figure.N)r{r, mapToGlobalrQemit)rcqpointpoints r3rzFigureViewer.show_context_menuxsX > 8dn0<N..v66E  + 0 0 7 7 7 7 7 8 8<!>JJ!$VJ #FW$4v$= > >I$ % % %''77,0KK,H,H)))LMMOOO" $  % >   E)Z$@$@ @ @ N ' ' : > > >  ! & &t'7'7'9'9 : : : : : A @s AF!!A*H Hc|j}|jj}|dkrt||z dzSdS)z1Get the current scaling of the figure in percent.rd)r{rrround)rcrrs r3rzFigureViewer.get_scalingsD$$&&& Q;;#-.. .3r?c<d|_|dS)z%Reset the image to its original size.rN)rrrs r3reset_original_imagez!FigureViewer.reset_original_images" r?c|}|t||z|dz |zdz z|}|t||z|dz |zdz zdS)zg Adjust the scrollbar position to take into account the zooming of the figure. r6N)rrrrhpageStepr)rcr2hbvbs r3rzFigureViewer._adjust_scrollbars  % % ' ' CBHHJJ1q5BKKMM*A!*CDEEFFF # # % % CBHHJJ1q5BKKMM*A!*CDEEFFFFFr?r)rrrrrrrUr rQrTrIpropertyrjsetterrrrrrrrrrrrrs@r3rPrP$s[ vc{{"(=      2''X'    '''888!!!080808f444666';';';R G G G G G G Gr?rPceZdZdZdZeeZ eeZ ee e Z d fd Z dZdZdZfdZd Zd Zd Zd Zd ZdZdZdZdZdZdZdZdZdZdZ dZ!dZ"dZ#dZ$dZ%dZ&dZ'dZ(xZ)S)!rWz A widget that manages the display of the FigureThumbnails that are created when a figure is sent to the IPython console by the kernel and that controls what is displayed in the FigureViewer. r Ncft|g|_||_t |_d|_|||d|_ |j j |jdS)NF)rHrI _thumbnailsrGrrlcurrent_thumbnailset_figureviewer setup_gui_new_thumbnail_added scrollarear rangeChangedrR_scroll_to_newest_item)rc figure_viewerrCrGres r3rIzThumbnailScrollBar.__init__5s     0&(( !% m,,, %*! ))++8@@  ' ) ) ) ) )r?ct|}|dddd|d||dSz$Setup the main layout of the widget.rN)rr`rbr]setup_scrollarearcrds r3rzThumbnailScrollBar.setup_guiKscT""!!!Q1---!..0011111r?ct|_t|j|_|jdddd|jdt |_|j|j|j d|j d|j dddd|j tj|j|j|jt%|j||dddd|d|jS)zr4append) rcr:fignames figname_root thumbnailr,r-fextr=s r3r6z)ThumbnailScrollBar.save_all_figures_todirs! )--//889JKKL ) % %I"&C"&C!'"(%+---02D),7;===G sC 1 1 1 OOG $ $ $ $r?c|j6||jjj|jjjdSdSz#Save the currently selected figure.N)rsave_figure_asrBr,r-rs r3rz)ThumbnailScrollBar.save_current_figure_assQ  ! -    6 = A $ 6 = A C C C C C . -r?cZ||jj|jjdSrI)rJrBr,r-rcrFs r3save_thumbnail_figure_asz+ThumbnailScrollBar.save_thumbnail_figure_ass* I,0)2B2FGGGGGr?cdddd|\}}t|jdtjdz|}|jdt|d||d d \}}|jd |r?|j tj |t|||d Sd S) zSave the figure to a file.)r!z PNG (*.png))r:z JPEG (*.jpg;*.jpeg;*.jpe;*.jfif))r;zSVG (*.svg);;PNG (*.png)r<r8r9Fz Save FigurerN)rCcaptionbasedirfiltersselectedfilterrpT) r>rlr?r@rAr[rrrCrZr$r:r4)rcr,r-rGffiltr=r.s r3rJz!ThumbnailScrollBar.save_figure_ass1FACCDGH e % M )--//889JKK K  )..u555%;;==-Ut--- t )..t444  0  % * *3;u+=+= > > > sC / / / / / 0 0r?ctjdkrdnd}|jd|zz |jz |jz |z |j z }|dz }|S)zX Calculate the width the thumbnails need to have to fit the scrollarea. darwinrrr) sysplatformrr lineWidthviewportMarginsleftrightrsizeHint)rc extra_paddingfigure_canvas_widths r3_calculate_figure_canvas_widthz1ThumbnailScrollBar._calculate_figure_canvas_widths"lh66A O ! ! # #    ! O + + - - 2 2 4 4 5 O + + - - 3 3 5 5 6    O - - / / 8 8 : : @ @ B B  C 2A5""r?cX|}||dS)zl Scale the thumbnail's canvas size so that it fits the thumbnail scrollbar's width. N)r`scale_canvas_size)rcrFmax_canvas_sizes r3_setup_thumbnail_sizez(ThumbnailScrollBar._setup_thumbnail_sizes/ ==??##O44444r?c|j|jD]}|||jdS)zd Update the thumbnails size so that their width fit that of the scrollarea. N)r'hiderrdshowrLs r3r4z)ThumbnailScrollBar._update_thumbnail_sizesW ) 2 2I  & &y 1 1 1 1 r?cp|j|}|j||dS)zV Emit global positioned point and thumbnail for context menu request. N)rBrrQr)rcrrFs r3rz$ThumbnailScrollBar.show_context_menus9 ,,U33 ',,UI>>>>>r?cZtjj||jjjjj j j fdj d_jjdz djjdz djjddS)zB Add a new thumbnail to that thumbnail scrollbar. rFc0|SN)r)rrcrFs r3z2ThumbnailScrollBar.add_thumbnail..s$00 BBr?Tr6rr N)FigureThumbnailrGrBrsig_canvas_clickedrRset_current_thumbnailsig_remove_figure_requestedremove_thumbnailsig_save_figure_requestedrJrQrrCrr( setRowStretchrowCountr]rgrd)rcr,r-rFs` @r3rz ThumbnailScrollBar.add_thumbnails$$*?AAA $$S#...$,,T-GHHH-55d6KLLL+33D4GHHH,44 B B B B B D D D  ***$(!   !4!4!6!6!:A>>> Y (;(;(=(=(A1EEE   !4!4!6!6<<< ""9--- ""9-----r?cL|j||jdSdS)z(Remove the currently selected thumbnail.N)rrqrs r3rz+ThumbnailScrollBar.remove_current_thumbnails0  ! -  ! !$"8 9 9 9 9 9 . -r?c|jD]}|j|j|j|||d|| g|_d|_ |j j dS)zRemove all thumbnails.N)rrn disconnectrprrrd removeWidget setParentrfcloserr r{ clear_canvasrLs r3rz(ThumbnailScrollBar.remove_all_thumbnailss)  I  ( 3 3 5 5 5  1 < < > > >  / : : < < < KKMM & &y 1 1 1    % % % NN    OO    !% $1133333r?c"jvrj} jjjn#t $rYnwxYwjvrjjkrvtjdkr9 t|tjdz n%j j d_t'jdfddS)zRemove thumbnail.rr6Nc.Srk)_remove_thumbnail_parentrLsr3rlz5ThumbnailScrollBar.remove_thumbnail..Ds66yAAr?)rindexrnrwrprr TypeErrorremoverlenset_current_indexminr r{r{rdrxrfrzr singleShot)rcrFrs`` r3rqz#ThumbnailScrollBar.remove_thumbnail#s ( ( ($**955E   ( 3 3 5 5 5  1 < < > > >  / : : < < < <    D  ( ( (   # #I . . . . . .4#$$q((&&s4#344q899",99;;;)-& ""9---  AAAAA C C C C CsA A33 B?BcT |ddS#t$rYdSwxYwrk)ry RuntimeErrorrLs r3rz+ThumbnailScrollBar._remove_thumbnail_parentFsC     % % % % %    DD s  ''cF||j|dS)z2Set the currently selected thumbnail by its index.N)rorrcrs r3rz$ThumbnailScrollBar.set_current_indexMs$ ""4#3E#:;;;;;r?cd |j|jS#t$rYdSwxYw)z5Return the index of the currently selected thumbnail.)rrr ValueErrorrs r3get_current_indexz$ThumbnailScrollBar.get_current_indexQsC #))$*@AA A   22 s ! //c|j|krdS|j|jd||_|j|jj|jj|jddS)z%Set the currently selected thumbnail.NFT)rhighlight_canvasr rrBr,r-rLs r3roz(ThumbnailScrollBar.set_current_thumbnailXs  !Y . . F  ! -  " 3 3E : : :!* &&   )"2"6 8 8 8 //55555r?c|jm|j|jdz }|dkr|nt|jdz }||||dSdS)z