o xiDM @s`dZddlZddlZddlmZddlmZmZmZm Z m Z ddl Z zddl m Z mZmZdZWn ey;dZYnwzddlZWney]zddlZWn eyZdZYnwYnwgdZgd d d gd gd dgdZdZdZdZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZ GdddeZ!dS) z pygments.formatters.img ~~~~~~~~~~~~~~~~~~~~~~~ Formatter for Pixmap output. :copyright: Copyright 2006-2017 by the Pygments team, see AUTHORS. :license: BSD, see LICENSE for details. N) Formatter) get_bool_opt get_int_opt get_list_optget_choice_optxrange)Image ImageDraw ImageFontTF)ImageFormatterGifImageFormatterJpgImageFormatterBmpImageFormatter)RomanBookNormalRegularMediumObliqueItalicBoldz Bold Obliquez Bold Italic)NORMALITALICBOLD BOLDITALICzBitstream Vera Sans Monoz Courier Newc@eZdZdZdS)PilNotAvailablez,When Python imaging library is not availableN__name__ __module__ __qualname____doc__r#r#X/home/ubuntu/.local/lib/python3.10/site-packages/wandb/vendor/pygments/formatters/img.pyr6rc@r) FontNotFoundz(When there are no usable fonts specifiedNrr#r#r#r$r&:r%r&c@s\eZdZdZdddZddZddZd d Zd d ZdddZ ddZ ddZ ddZ dS) FontManagerz> Manages a set of fonts: normal, italic, bold, etc... cCsr||_||_i|_d|_tjdr|st|_|dStjdr.|s(t |_| dS|s3t |_| dS)Nwindarwin) font_name font_sizefontsencodingsysplatform startswithDEFAULT_FONT_NAME_WIN _create_winDEFAULT_FONT_NAME_MAC _create_macDEFAULT_FONT_NAME_NIX _create_nix)selfr+r,r#r#r$__init__Cs     zFontManager.__init__c Csztjdd||fdgtjdd}|\}}|jdkr;|}|D]}|dr)q!|d}|r8|Sq!dSdS)Nzfc-listz %s:style=%sfile)stdoutstderrrsFontconfig warning::) subprocessPopenPIPE communicate returncode splitlinesr1decodestrip) r8namestyleprocr;_lineslinepathr#r#r$_get_nix_font_pathUs   zFontManager._get_nix_font_pathcCstdD]}||j|}|durt||j|jd<nqtd|jdD]5}t|D]}||j|}|durFt||j|j|<nq-|dkrT|jd|j|<q'|jd|j|<q'dS)NrNo usable fonts named: "%s"rrrrr)STYLESrMr+r truetyper,r-r&)r8rFrLrG stylenamer#r#r$r7cs*  zFontManager._create_nixcCs||d|S)N )getrElower)r8font_maprFrGr#r#r$_get_mac_font_pathxszFontManager._get_mac_font_pathcsi}tjtddddfD]|fddtDqtdD]}|||j|}|dur?t ||j |j d<nq%t d|jd D]6}t|D]}|||j|}|durit ||j |j |<nqO|d krw|j d |j |<qI|j d|j |<qIdS) NHOMEzLibrary/Fonts/z/Library/Fonts/z/System/Library/Fonts/c3s@|]}|drtj|dtj|fVqdS)ttfrN)rUendswithosrLsplitextjoin).0ffont_dirr#r$ s,z*FontManager._create_mac..rrNrOrr)r[rLr]getenvupdatelistdirrPrWr+r rQr,r-r&)r8rVrFrLrGrRr#r`r$r5{s<   zFontManager._create_macFc CsvdD]*}|D]%}zd||od||f}t||\}} |WSty+Yqwq|r9td||dfdS)N)rz (TrueType)z%s%s%srSz"Font %s (%s) not found in registryr)_winreg QueryValueExEnvironmentErrorr&) r8keybasenamestylesfailsuffixrGvalnamevalrIr#r#r$ _lookup_wins  zFontManager._lookup_winc Csz ttjd}Wnty'z ttjd}Wn ty$tdwYnwzN|||jtdd}t ||j |j d<dD].}|||jt|}|rYt ||j |j |<q@|dkrf|j d|j |<q@|j d|j |<q@Wt |dSt |w) Nz2Software\Microsoft\Windows NT\CurrentVersion\Fontsz/Software\Microsoft\Windows\CurrentVersion\Fontsz$Can't open Windows font registry keyrTrOrr) rfOpenKeyHKEY_LOCAL_MACHINErhr&rpr+rPr rQr,r-CloseKey)r8rirLrGr#r#r$r3s8   zFontManager._create_wincCs|jddS)z) Get the character size. rM)r-getsizer8r#r#r$ get_char_sizezFontManager.get_char_sizecCs8|r |r |jdS|r|jdS|r|jdS|jdS)z> Get the font based on bold and italic flags. rrrr)r-)r8boldobliquer#r#r$get_fonts    zFontManager.get_fontN)r()F) rr r!r"r9rMr7rWr5rpr3rwr{r#r#r#r$r'>s   r'c@seZdZdZdZgdZdgZdZdZddZ d+d d Z d d Z ddZ ddZ ddZddZddZddZddZddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd*S),r a Create a PNG image from source code. This uses the Python Imaging Library to generate a pixmap from the source code. .. versionadded:: 0.10 Additional options accepted: `image_format` An image format to output to that is recognised by PIL, these include: * "PNG" (default) * "JPEG" * "BMP" * "GIF" `line_pad` The extra spacing (in pixels) between each line of text. Default: 2 `font_name` The font name to be used as the base font from which others, such as bold and italic fonts will be generated. This really should be a monospace font to look sane. Default: "Bitstream Vera Sans Mono" on Windows, Courier New on \*nix `font_size` The font size in points to be used. Default: 14 `image_pad` The padding, in pixels to be used at each edge of the resulting image. Default: 10 `line_numbers` Whether line numbers should be shown: True/False Default: True `line_number_start` The line number of the first line. Default: 1 `line_number_step` The step used when printing line numbers. Default: 1 `line_number_bg` The background colour (in "#123456" format) of the line number bar, or None to use the style background color. Default: "#eed" `line_number_fg` The text color of the line numbers (in "#123456"-like format). Default: "#886" `line_number_chars` The number of columns of line numbers allowable in the line number margin. Default: 2 `line_number_bold` Whether line numbers will be bold: True/False Default: False `line_number_italic` Whether line numbers will be italicized: True/False Default: False `line_number_separator` Whether a line will be drawn between the line number area and the source code area: True/False Default: True `line_number_pad` The horizontal padding (in pixels) between the line number margin, and the source code area. Default: 6 `hl_lines` Specify a list of lines to be highlighted. .. versionadded:: 1.2 Default: empty list `hl_color` Specify the color for highlighting lines. .. versionadded:: 1.2 Default: highlight color of the selected style img)r|IMGpngz*.pngFr~c Kststdtj|fi|d|_t|j|_|jjdur"d|_n|jj|_t |dgd|j dd|_ t |d d |_ t |d d |_t |d d}t|dd||_|j\|_|_|dd|_|dd|_t |dd |_t|dd|_t|dd|_t |dd|_t|dd|_t|dd|_t |dd|_t |dd|_|jr|j|j|jd |_ nd |_ g|_!t"|d!g}|D]}z |j!#t$|Wqt%yYqw|d"|jj&pd#|_'g|_(dS)$zE See the class docstring for explanation of options. z5Python Imaging Library is required for this formatterlatin1Nz#fff image_format)r~jpeggifbmpT)normcase image_pad line_padr,r(r+rline_number_fgz#886line_number_bgz#eedline_number_charsline_number_boldFline_number_italicline_number_pad line_numbersline_number_separatorline_number_stepline_number_startrhl_lineshl_colorz#f90)) pil_availablerrr9r.dictrGrkbackground_colorrdefault_image_formatrrrrr'rTr-rwfontwfonthrrrrrrrrrrrline_number_widthrrappendint ValueErrorhighlight_colorr drawables)r8optionsfontsize hl_lines_strrKr#r#r$r9Jsp         zImageFormatter.__init__rcCstd)NzWThe -S option is meaningless for the image formatter. Use -O style= instead.)NotImplementedError)r8argr#r#r$get_style_defsszImageFormatter.get_style_defscCs |j|jS)z+ Get the height of a line. )rrrvr#r#r$_get_line_heights zImageFormatter._get_line_heightcCs|||jS)z8 Get the Y coordinate of a line number. )rrr8linenor#r#r$ _get_line_yszImageFormatter._get_line_ycCs|jS)z/ Get the width of a character. )rrvr#r#r$_get_char_widthszImageFormatter._get_char_widthcCs||j|j|jS)z? Get the X coordinate of a character position. )rrr)r8charnor#r#r$ _get_char_xzImageFormatter._get_char_xcCs||||fS)zL Get the actual position for a character and line position. )rr)r8rrr#r#r$ _get_text_posszImageFormatter._get_text_poscCs|j||fS)zI Get the actual position for the start of a line number. )rrrr#r#r$_get_linenumber_posrxz"ImageFormatter._get_linenumber_poscCs$|ddurd|d}|Sd}|S)zE Get the correct color for the token from the style. colorN#z#000r#)r8rGfillr#r#r$_get_text_colors  zImageFormatter._get_text_colorcCs|j|d|dS)z5 Get the correct font for the style. ryitalic)r-r{)r8rGr#r#r$_get_style_fontrzImageFormatter._get_style_fontcCs$|||j||d|jfS)z. Get the required image size. r)rrr)r8 maxcharno maxlinenor#r#r$_get_image_sizeszImageFormatter._get_image_sizecCs8|j||t||j|j|j|j|j ddS)zA Remember a line number drawable to paint later. fontrN) _draw_textrstrrjustrr-r{rrr)r8posnorr#r#r$_draw_linenumbers  zImageFormatter._draw_linenumbercKs|j||||fdS)zB Remember a single drawable tuple to paint later. N)rr)r8postextrkwr#r#r$rszImageFormatter._draw_textc Csd}}}|D]Y\}}||jvr|j}||jvs|j|}|d}|d}t|D]4\} } | d} | rU|j|||| ||| |d|t | 7}t ||}| dr`d}|d7}q,q||_ ||_dS)z9 Create drawables for the token content. rT rrN)rkparent expandtabsrC enumeraterstriprrrrlenmaxrZrr) r8 tokensourcerrrttypevaluerGrJirKtempr#r#r$_create_drawabless4            z ImageFormatter._create_drawablescCsB|jsdSt|jD]}||j}||jdkr|||q dS)z8 Create drawables for the line numbers. Nr)rrrrrr)r8pnr#r#r$_draw_line_numberss  z!ImageFormatter._draw_line_numberscCsx|jsdS|jdur dSt|}|jd}|j|j|j}|jd||fg|j d|j |df||fg|jd~dS)z@ Paint the line number background on the image. N)rrrr) rrr Drawsizerrr rectanglerrK)r8imdrawrecthrectwr#r#r$_paint_line_number_bgs   z$ImageFormatter._paint_line_number_bgcCs|||td||j|j|j}||t |}|j rY|j |j |jd}|}|jd|}|j D]}||d} |j|| f||| |fg|jdq=|jD]\} } } } |j| | fd| i| q\|||jdS)a  Format ``tokensource``, an iterable of ``(tokentype, tokenstring)`` tuples and write it into ``outfile``. This implementation calculates where it should draw each token on the pixmap, then calculates the required pixmap size and draws the items. RGBrrrrN)rrrnewrrrrrr rrrrrrrrrrrrsaverupper)r8routfilerrxrr linenumberyrrrrr#r#r$formats*    zImageFormatter.formatN)r)rr r!r"rFaliases filenames unicodeoutputrr9rrrrrrrrrrrrrrrrr#r#r#r$r s0l 8  ! r c@s$eZdZdZdZdgZdgZdZdS)r z Create a GIF image from source code. This uses the Python Imaging Library to generate a pixmap from the source code. .. versionadded:: 1.0 img_gifrz*.gifNrr r!r"rFrrrr#r#r#r$r 1s r c@s&eZdZdZdZddgZdgZdZdS)r z Create a JPEG image from source code. This uses the Python Imaging Library to generate a pixmap from the source code. .. versionadded:: 1.0 img_jpgjpgrz*.jpgNrr#r#r#r$r ? r c@s&eZdZdZdZddgZdgZdZdS)rz Create a bitmap image from source code. This uses the Python Imaging Library to generate a pixmap from the source code. .. versionadded:: 1.0 img_bmprbitmapz*.bmpNrr#r#r#r$rMrr)"r"r[r/pygments.formatterr pygments.utilrrrrrr>PILrr r r ImportErrorrfwinreg__all__rPr6r2r4r Exceptionr&objectr'r r r rr#r#r#r$sR       ^