U lu fLZ @sldZddlZddlZddlmZddlmZmZmZm Z ddl Z zddl m Z m Z mZdZWnek rtdZYnXz ddlZWn:ek rz ddlZWnek rdZYnXYnXdd d d gZd d ddddgddgdgddgdZdZdZdZGdddeZGdddeZGdd d ZGd!ddeZGd"d d eZGd#d d eZGd$d d eZdS)%z pygments.formatters.img ~~~~~~~~~~~~~~~~~~~~~~~ Formatter for Pixmap output. :copyright: Copyright 2006-2023 by the Pygments team, see AUTHORS. :license: BSD, see LICENSE for details. N) Formatter) get_bool_opt get_int_opt get_list_optget_choice_opt)Image ImageDraw ImageFontTFImageFormatterGifImageFormatterJpgImageFormatterBmpImageFormatterZRomanZBookZNormalZRegularZMediumZObliqueZItalicZBoldz Bold Obliquez Bold Italic)NORMALITALICBOLD BOLDITALICzDejaVu Sans Monoz Courier NewZMenloc@seZdZdZdS)PilNotAvailablez,When Python imaging library is not availableN__name__ __module__ __qualname____doc__rr;/tmp/pip-unpacked-wheel-7eq1ciah/pygments/formatters/img.pyr4src@seZdZdZdS) FontNotFoundz(When there are no usable fonts specifiedNrrrrrr8src@sleZdZdZdddZddZddZd d Zd d ZdddZ ddZ ddZ ddZ ddZ ddZdS) FontManagerz> Manages a set of fonts: normal, italic, bold, etc... cCs||_||_i|_d|_d|_t|ds4tj|r`t ||j}d|_t D]}||j|<qLdSt j dr|svt|_|n2t j dr|st|_|n|st|_|dS)NFreadTwindarwin) font_name font_sizefontsencodingvariablehasattrospathisfiler truetypeSTYLESsysplatform startswithDEFAULT_FONT_NAME_WIN _create_winDEFAULT_FONT_NAME_MAC _create_macDEFAULT_FONT_NAME_NIX _create_nix)selfr!r"fontstylerrr__init__As,     zFontManager.__init__c Csztjdd||fdgtjdd}|\}}|jdkrv|}|D].}|drRqB|d}|rB|SqBdSdS)Nzfc-listz %s:style=%sfile)stdoutstderrrsFontconfig warning::) subprocessPopenPIPE communicate returncode splitlinesr.decodestrip) r5namer7procr:_linesliner(rrr_get_nix_font_path\s    zFontManager._get_nix_font_pathcCstdD]2}||j|}|dk rt||j|jd<qJqtd|jdD]j}t|D]2}||j|}|dk rZt||j|j|<qNqZ|dkr|jd|j|<qN|jd|j|<qNdS)NrNo usable fonts named: "%s"rrrrr)r+rJr!r r*r"r#r)r5rEr(r7 stylenamerrrr4js"  zFontManager._create_nixcCs||d|S)N )getrDlower)r5font_maprEr7rrr_get_mac_font_pathszFontManager._get_mac_font_pathcsi}tjtddddfD]"|fddtDqtdD]4}|||j|}|dk rJt ||j |j d<qqJt d|jd D]l}t|D]4}|||j|}|dk rt ||j |j |<qq|d kr|j d |j |<q|j d|j |<qdS) NHOMEzLibrary/Fonts/z/Library/Fonts/z/System/Library/Fonts/c3s>|]6}|drtj|dtj|fVqdS))ZttfZttcrN)rPendswithr'r(splitextjoin).0fZfont_dirrr sz*FontManager._create_mac..rrKrLrr)r'r(rVgetenvupdatelistdirr+rRr!r r*r"r#r)r5rQrEr(r7rMrrYrr2s2   zFontManager._create_macFc Cs~dD]\}|D]R}z4d||o d||f}t||\}} |WStk r\Yq Yq Xq q|rztd||dfdS)N)rz (TrueType)z%s%s%srNz"Font %s (%s) not found in registryr)_winreg QueryValueExOSErrorr) r5keybasenamestylesZfailsuffixr7ZvalnamevalrGrrr _lookup_wins zFontManager._lookup_winc CsFd}tjdftjdftjdftjdfg}|D]}ztj|}zz|||jtdd}t ||j |j d<dD]\}|||jt|}|rt ||j |j |<qp|dkr|j d|j |<qp|j d|j |<qpWW4WdSt k r}z|}W5d}~XYnXW5t|XWq,t k r,Yq,Xq,|r:|t ddS) Nz2Software\Microsoft\Windows NT\CurrentVersion\Fontsz/Software\Microsoft\Windows\CurrentVersion\FontsrTrLrrz$Can't open Windows font registry key)r^HKEY_CURRENT_USERHKEY_LOCAL_MACHINEOpenKeyZCloseKeyrfr!r+r r*r"r#rr`)r5Z lookuperrorZkeynameskeynamerar(r7errrrrr0s8  zFontManager._create_wincCs |dS)z) Get the character size. M) get_text_sizer5rrr get_char_sizeszFontManager.get_char_sizecCs4|jd}t|dr&||ddS||SdS)z4 Get the text size (width, height). rgetbboxN)r#r&rpgetsize)r5textr6rrrrms  zFontManager.get_text_sizecCs||r"|r"|jr|dS|jdS|r@|jr6|dS|jdS|r^|jrT|dS|jdS|jrn|dS|jdSdS)z> Get the font based on bold and italic flags. rrrrN)r% get_styler#)r5boldZobliquerrrget_fonts       zFontManager.get_fontc Cs`|j|}t|D]H}z|||WStk r>Yqtk rX|YSXq|S)z} Get the specified style of the font if it is a variable font. If not found, return the normal font. )r#r+Zset_variation_by_name ValueErrorr`)r5r7r6Z style_namerrrrus    zFontManager.get_styleN)r)F)rrrrr8rJr4rRr2rfr0rormrwrurrrrr<s  ' rc@seZdZdZdZdddgZdgZdZdZddZ d-d d Z d d Z ddZ ddZ ddZddZddZddZddZddZddZd d!Zd"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. If a filename or a file-like object is specified, the user must provide different styles of the font. Default: "Courier New" on Windows, "Menlo" on Mac OS, and "DejaVu Sans Mono" 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 imgZIMGpngz*.pngFc Ksts tdtj|f|d|_t|j|_|jjdkr@d|_n |jj|_t |ddddd g|j d d |_ t |d d |_ t |dd|_t |dd}t|dd||_|j\|_|_|dd|_|dd|_t |dd|_t|dd|_t|dd|_t |dd|_t|dd |_t|dd |_t |d d!|_t |d"d!|_|jrd|j|j|jd|_ nd#|_ g|_!t"|d$g}|D]2}z|j!#t$|Wnt%k rYnXq|d%|jj&pd&|_'g|_(dS)'zE See the class docstring for explanation of options. z5Python Imaging Library is required for this formatterlatin1Nz#fff image_formatrzjpeggifbmpT)normcase image_pad line_padrqr"rr!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_availablerrr8r$dictr7rcbackground_colorrdefault_image_formatr|rrrrrOr#rofontwfonthrrrrrrrrrrrline_number_widthrrappendintrxZhighlight_colorr drawables)r5optionsZfontsizeZ hl_lines_strrIrrrr8s|        zImageFormatter.__init__rcCs tddS)NzWThe -S option is meaningless for the image formatter. Use -O style= instead.)NotImplementedError)r5argrrrget_style_defsszImageFormatter.get_style_defscCs |j|jS)z+ Get the height of a line. )rrrnrrr_get_line_heightszImageFormatter._get_line_heightcCs|||jS)z8 Get the Y coordinate of a line number. )rrr5linenorrr _get_line_yszImageFormatter._get_line_ycCs|jS)z/ Get the width of a character. )rrnrrr_get_char_widthszImageFormatter._get_char_widthcCs||j|jS)z? Get the X coordinate of a character position. )rr)r5 linelengthrrr _get_char_xszImageFormatter._get_char_xcCs||||fS)zL Get the actual position for a character and line position. )rr)r5rrrrr _get_text_posszImageFormatter._get_text_poscCs|j||fS)zI Get the actual position for the start of a line number. )rrrrrr_get_linenumber_possz"ImageFormatter._get_linenumber_poscCs"|ddk rd|d}nd}|S)zE Get the correct color for the token from the style. colorN#z#000r)r5r7fillrrr_get_text_colors zImageFormatter._get_text_colorcCs"|ddk rd|d}nd}|S)zP Get the correct background color for the token from the style. ZbgcolorNrr)r5r7Zbg_colorrrr_get_text_bg_colors z!ImageFormatter._get_text_bg_colorcCs|j|d|dS)z5 Get the correct font for the style. rvZitalic)r#rw)r5r7rrr_get_style_fontszImageFormatter._get_style_fontcCs$|||j||d|jfS)z. Get the required image size. r)rrr)r5 maxlinelength maxlinenorrr_get_image_sizeszImageFormatter._get_image_sizecCs:|j||t||j|j|j|j|j dddS)zA Remember a line number drawable to paint later. Nr6text_fgtext_bg) _draw_textrstrrjustrr#rwrrr)r5Zposnorrrr_draw_linenumber s zImageFormatter._draw_linenumbercCs|j|||||fdS)zB Remember a single drawable tuple to paint later. N)rr)r5posrtr6rrrrrrszImageFormatter._draw_textc Csd}}}d}}|D]\}}||jkr2|j}q |j|} |d}|d} t| D]\} } | d} | r|j|||| || | | | | d|j | \}}||7}t ||}|t| 7}t ||}| drXd}d}|d7}qXq||_||_||_dS)z9 Create drawables for the token content. rrrT rrN)rcparent expandtabsrB enumeraterstriprrrrrr#rmmaxlenrTr maxcharnor)r5 tokensourcerZcharnorrrttypevaluer7rHirItempZ temp_widthrGrrr_create_drawabless<             z ImageFormatter._create_drawablescCsB|js dSt|jD](}||j}||jdkr|||qdS)z8 Create drawables for the line numbers. Nr)rrangerrrr)r5pnrrr_draw_line_numbersDs  z!ImageFormatter._draw_line_numberscCs~|js dS|jdkrdSt|}|jd}|j|j|j}|jd||fg|j d|j rx|j |df||fg|jd~dS)z@ Paint the line number background on the image. N)rrrr) rrrDrawsizerrr rectanglerrrI)r5imdrawrecthrectwrrr_paint_line_number_bgOs   z$ImageFormatter._paint_line_number_bgcCs:|||td||j|j|j}||t |}|j r|j |j |jd}|}|jd|}|j D]6}||d} |j|| f||| |fg|jdqz|jD]j\} } } } }|r|j| | d}|j| d| d| d|d| d|dg|d|j| | | | dq|||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. ZRGBrrr)rtr6)r6rN)rrrnewrrrrrrrrrrrrrrrrrZtextsizertsaver|upper)r5routfilerrxrrZ linenumberyrrr6rrZ text_sizerrrformat`s0    6zImageFormatter.formatN)r)rrrrrEaliases filenamesZ unicodeoutputrr8rrrrrrrrrrrrrrrrrrrrrr s0o 8    ( 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 Zimg_gifr~z*.gifNrrrrrErrrrrrrr s 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 Zimg_jpgZjpgr}z*.jpgNrrrrrr s c@s&eZdZdZdZddgZdgZdZdS)r z Create a bitmap image from source code. This uses the Python Imaging Library to generate a pixmap from the source code. .. versionadded:: 1.0 Zimg_bmprZbitmapz*.bmpNrrrrrr s ) rr'r,Zpygments.formatterrZ pygments.utilrrrrr=ZPILrrr r ImportErrorr^winreg__all__r+r3r/r1r Exceptionrrr r r r rrrrsL     Uw