U td6@sdZddlZddlmZddlmZddlmZddlm Z ddl m Z m Z ddl mZdd lmZd Ze ZGd d d ZdS) at Drawing and plotting routines for IGraph. IGraph has two plotting backends at the moment: Cairo and Matplotlib. The Cairo backend is dependent on the C{pycairo} or C{cairocffi} libraries that provide Python bindings to the popular U{Cairo library}. This means that if you don't have U{pycairo} or U{cairocffi} installed, you won't be able to use the Cairo plotting backend. Whenever the documentation refers to the C{pycairo} library, you can safely replace it with C{cairocffi} as the two are API-compatible. The Matplotlib backend uses the U{Matplotlib library}. You will need to install it from PyPI if you want to use the Matplotlib plotting backend. If you do not want to (or cannot) install any of the dependencies outlined above, you can still save the graph to an SVG file and view it from U{Mozilla Firefox} (free) or edit it in U{Inkscape} (free), U{Skencil} (formerly known as Sketch, also free) or Adobe Illustrator. N)BytesIO)warn) Configuration) find_cairo)Palettepalettes) BoundingBox)named_temporary_file) CairoPlotc@seZdZdZdddZdddZedd Zejd d Zd d d Z ddZ d!ddZ d"ddZ ddZ eddZeddZeddZeddZdS)#r aClass representing an arbitrary plot that uses the Cairo plotting backend. Objects that you can plot include graphs, matrices, palettes, clusterings, covers, and dendrograms. In Cairo, every plot has an associated surface object. The surface is an instance of C{cairo.Surface}, a member of the C{pycairo} library. The surface itself provides a unified API to various plotting targets like SVG files, X11 windows, PostScript files, PNG files and so on. C{igraph} does not usually know on which surface it is plotting at each point in time, since C{pycairo} takes care of the actual drawing. Everything that's supported by C{pycairo} should be supported by this class as well. Current Cairo surfaces include: - C{cairo.GlitzSurface} -- OpenGL accelerated surface for the X11 Window System. - C{cairo.ImageSurface} -- memory buffer surface. Can be written to a C{PNG} image file. - C{cairo.PDFSurface} -- PDF document surface. - C{cairo.PSSurface} -- PostScript document surface. - C{cairo.SVGSurface} -- SVG (Scalable Vector Graphics) document surface. - C{cairo.Win32Surface} -- Microsoft Windows screen rendering. - C{cairo.XlibSurface} -- X11 Window System screen rendering. If you create a C{Plot} object with a string given as the target surface, the string will be treated as a filename, and its extension will decide which surface class will be used. Please note that not all surfaces might be available, depending on your C{pycairo} installation. A C{Plot} has an assigned default palette (see L{igraph.drawing.colors.Palette}) which is used for plotting objects. A C{Plot} object also has a list of objects to be plotted with their respective bounding boxes, palettes and opacities. Palettes assigned to an object override the default palette of the plot. Objects can be added by the L{Plot.add} method and removed by the L{Plot.remove} method. NcCsd|_d|_|dkr"tdd|_n&t|ts6t|trBt||_n||_|dkr`t}|d}t|t srt |}||_ |dkrd|_t t jt|jjt|jj|_nt|t jr||_n||_tj|\}}|}|dkrt ||jj|jj|_n|dks|dkr,t ||jj|jj|_n^|d krZt t jt|jjt|jj|_n0|d kr~t ||jj|jj|_n td |t |j|_g|_d|_|dkrd }||_ dS) aCreates a new plot. @param target: the target surface to write to. It can be one of the following types: - C{None} -- a Cairo surface will be created and the object will be plotted there. - C{cairo.Surface} -- the given Cairo surface will be used. - C{string} -- a file with the given name will be created and an appropriate Cairo surface will be attached to it. @param bbox: the bounding box of the surface. It is interpreted differently with different surfaces: PDF and PS surfaces will treat it as points (1 point = 1/72 inch). Image surfaces will treat it as pixels. SVG surfaces will treat it as an abstract unit, but it will mostly be interpreted as pixels when viewing the SVG file in Firefox. @param palette: the palette primarily used on the plot if the added objects do not specify a private palette. Must be either an L{igraph.drawing.colors.Palette} object or a string referring to a valid key of C{igraph.drawing.colors.palettes} (see module L{igraph.drawing.colors}) or C{None}. In the latter case, the default palette given by the configuration key C{plotting.palette} is used. @param background: the background color. If C{None}, the background will be transparent. You can use any color specification here that is understood by L{igraph.drawing.colors.color_name_to_rgba}. NFiXzplotting.paletteTz.pdfz.psz.eps.pngz.svgz%image format not handled by Cairo: %swhite)! _filename _need_tmpfilerbbox isinstancetuplelistrinstancerr_palettecairo ImageSurfaceZ FORMAT_ARGB32intwidthheight_surfaceZSurfaceospathsplitextlowerZ PDFSurfaceZ PSSurface SVGSurface ValueErrorContext_ctx_objects _is_dirty background)selftargetrpaletter%config_extr,R/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/igraph/drawing/cairo/plot.py__init__[sp&         zCairoPlot.__init__?cOsd|dks|dkrtd|dkr&|j}|dk r@t|ts@t|}|j||||||f|dS)a?Adds an object to the plot. Arguments not specified here are stored and passed to the object's plotting function when necessary. Since you are most likely interested in the arguments acceptable by graphs, see L{Graph.__plot__} for more details. @param obj: the object to be added @param bbox: the bounding box of the object. If C{None}, the object will fill the entire area of the plot. @param palette: the color palette used for drawing the object. If the object tries to get a color assigned to a positive integer, it will use this palette. If C{None}, defaults to the global palette of the plot. @param opacity: the opacity of the object being plotted, in the range 0.0-1.0 @see: Graph.__plot__ gr/z#opacity must be between 0.0 and 1.0N)r rrrr#append mark_dirty)r&objrr(opacityargskwdsr,r,r-addsz CairoPlot.addcCs|jS)zbReturns the background color of the plot. C{None} means a transparent background. ) _backgroundr&r,r,r-r%szCairoPlot.backgroundcCs"|dkrd|_n|j||_dS)a Sets the background color of the plot. C{None} means a transparent background. You can use any color specification here that is understood by the C{get} method of the current palette or by L{igraph.drawing.colors.color_name_to_rgb}. N)r7rget)r&colorr,r,r-r%scCsvtt|jD]b}|j|dd\}}||kr|dks@||kr|d8}|dkrg|j||d<|dSqdS)aRemoves an object from the plot. If the object has been added multiple times and no bounding box was specified, it removes the instance which occurs M{idx}th in the list of identical instances of the object. @param obj: the object to be removed @param bbox: optional bounding box specification for the object. If given, only objects with exactly this bounding box will be considered. @param idx: if multiple objects match the specification given by M{obj} and M{bbox}, only the M{idx}th occurrence will be removed. @return: C{True} if the object has been removed successfully, C{False} if the object was not on the plot at all or M{idx} was larger than the count of occurrences rNr;TF)rangelenr#r1)r&r2ridxiZ current_objZ current_bboxr,r,r-removeszCairoPlot.removecCs d|_dS)z+Marks the plot as dirty (should be redrawn)TN)r$r8r,r,r-r1szCairoPlot.mark_dirtyc Cs|p|j}|jdk r@|j|j|dd|jj|jj||jD]\}}}}}}|dkrlt |d|j }t |dd} | dkrt d|fqF|dkr| n| | d|f|||d||dkr|||qF|qFd |_dS) zRedraws the plotNrZ_default_paletteZ__plot__z%s does not support plottingr/r)rr(F)r"r7Zset_source_rgbaZ rectanglerrrfillr#getattrrrZ push_groupsaveZpop_group_to_sourceZpaint_with_alpharestorer$) r&contextctxr2rr(r3r4r5Zplotterr,r,r-redraws(       zCairoPlot.redrawc Cs|jr|t|jtjr|dkrX|jrXtddd}|j|W5QRdSQRX|p`|j }|dkrrt d|jt |S|dk rt d|j |jdS)zSaves the plot. @param fname: the filename to save to. It is ignored if the surface of the plot is not an C{ImageSurface}. NZigraphr )prefixsuffixz4no file name is known for the surface and none givenz8filename is ignored for surfaces other than ImageSurface)r$rHrrrrrr Z write_to_pngr r strrr" show_pagefinish)r&fnamer,r,r-rD$s   zCairoPlot.savecCsZt}t||jj|jj}t|}|||| | d}|ddifS)zReturns an SVG representation of this plot as a string. This method is used by IPython to display this plot inline. zutf-8isolatedT) rrrrrrr!rHrLrMgetvaluedecode)r&iosurfacerFresultr,r,r- _repr_svg_@s  zCairoPlot._repr_svg_cCs t|jS)zPReturns the bounding box of the Cairo surface as a L{BoundingBox} object)rrr8r,r,r- bounding_boxSszCairoPlot.bounding_boxcCs|jjS)zJReturns the height of the Cairo surface on which the plot is drawn)rrr8r,r,r-rYszCairoPlot.heightcCs|jS)z4Returns the Cairo surface on which the plot is drawn)rr8r,r,r-rS_szCairoPlot.surfacecCs|jjS)zIReturns the width of the Cairo surface on which the plot is drawn)rrr8r,r,r-rdszCairoPlot.width)NNNN)NNr/)Nr;)N)N)__name__ __module__ __qualname____doc__r.r6propertyr%setterrAr1rHrDrUrVrrSrr,r,r,r-r ,s00 ^        r )rZrrRrwarningsrZigraph.configurationrZigraph.drawing.cairo.utilsrZigraph.drawing.colorsrrZigraph.drawing.utilsrZ igraph.utilsr __all__rr r,r,r,r-s