|Le9pdZddlZddlZddlZddlZddlZ ddl Z e j dZ e j Z ejejedZde e jddDZejdddd Zejd d d d Zd Zed fedffZdZdZdZefdZdZdZdZ dZ!dZ"dZ#dZ$dZ%dZ&dZ'dZ(dZ)dZ*d Z+e,d!kr=e(Z-e.e/e-0e)e-dSdS)"z0Tests for antialiased line drawing in DatashaderNz matplotlib.cmdatacjg|]0}tdt|ddD1S)c3 K|] }|dzV dS)go@N).0vs ?lib/python3.11/site-packages/datashader/tests/test_antialias.py z.s&//1$//////N)tuple)rrgbs r rsM 0 0 0 //c 2A2/// / / 0 0 0r 2)r1) plot_width plot_heightx_rangey_range)"))T antialias)Fnoaanormalclippedc(tj|d|dgtj|d|dg}}tj||dd}||ddt jd|S) aJDraw a single line. Parameters ---------- cvs: canvas A Datashader canvas p1: tuple The first vertex of the line p2: tuple The second vertex of the line antialias: boolean To anti-alias or not is the question Returns ------- agg: A Datashader aggregator (xarray) rr@xyvalr!r"r#aggr)nparraypd DataFramelineds reductionsmax)cvsp1p2rxsyspointss r draw_liner4s&Xr!uben % %rxA1'?'?B \C88 9 9F 88FC"-*;*;E*B*B'  ) ))r c g}|D]4\\}}\}}|t|||f||f|5tj|ddS)aDraw multiple line. Parameters ---------- cvs: canvas A Datashader canvas points: list of tuple of tuple The lines to render as a list of tuples, where each tuple represents a line consisting of two tuples each containing two scalars describing the two vertices of the line. antialias: boolean To anti-alias or not is the question Returns ------- agg: A Datashader aggregator (xarray) stackdim)appendr4xrconcatsum)r.r3raggsx1y1x2y2s r draw_linesrBsu$ D &CC"b8B IcB8b"XyAABBBB 9T7 # # ' 'G ' 4 44r cgg}}|D]/\}}||||0tj|tj|}}tj||dd}||ddt jd|} tj | gd dS) atDraw multi-line segment line. Parameters ---------- cvs: canvas A Datashader canvas points: list of tuples List of tuples of two scalars that represent each of the vertices in the multi-segment line. antialias: boolean To anti-alias or not is the question Returns ------- agg: A Datashader aggregator (xarray) rr r!r"r#r$r6r7) r9r&r'r(r)r*r+r,r-r:r;r<) r.r3rr!r"r>r?r1r2r%s r draw_multi_segment_linerDs" rqAR    Xa[["(1++B \C88 9 9F ((63):):5)A)A'  ) )C 9cUG $ $ ( (W ( 5 55r ctj||}tj|d}|S)zShade/render the aggregator. Parameters ---------- aggregators: xarray The aggregator(s) to shade cmap: color map The colormap to use Returns ------- img: xarray The shaded image. cmapz#ffffff)r+transfer_functionsshadeset_background) aggregatorsrGimgs r rIrIs<  % %k % = =C  . .sI > >C Jr ctjt|dz}t d|z||dS)zSave a shaded image as PNG file. Parameters ---------- img: xarray The image to save filename: unicode The name of the file to save to, 'png' extension will be appended. z.pngSaving: N)ospathjoindatadirprintto_pilsaverLfilenames r save_to_imagerXsSw||GX%677H *x    JJLLhr ctjt|dz}t d|z||dS)zSave a shaded image as NetCDF file. Parameters ---------- img: xarray The image to save filename: unicode The name of the file to save to, 'nc' extension will be appended. .ncrNN)rOrPrQrRrS to_netcdfrVs r save_to_netcdfr\ sIw||GX%566H *x    MM(r cztjt|dz}t j|S)zLoad a shaded image from NetCDF file. Parameters ---------- filename: unicode The name of the file to load from. Returns ------- img: xarray The loaded image. rZ)rOrPrQrRr:open_dataarray)rWs r load_from_netcdfr_s/w||GX%566H X & &&r chg}tdddD]}|dd|ff|dfS)Nr7rrrtest_001ranger9r3as r generate_test_001ri.sI F 1b!__)) vAw'(((( : r chg}tdddD]}|d|dff|dfS)Nrrarbrcrtest_002rergs r generate_test_002rl4sI F 1b!__)) v2w'(((( : r chg}tdddD]}|dd|ff|dfS)Nrrarbrrtest_003rergs r generate_test_003rp:I F 1b!__** x!Q()))) : r chg}tdddD]}|d|dff|dfS)Nrrarbrntest_004rergs r generate_test_004rt@rqr cgd}|dfS)N) )rcrr)rvr)rwrxr)ryr)rzr{r)r|r)r}r~r)rr)rrr)r r)rrr)rrr)rrntest_005rr3s r generate_test_005rFs F : r cgd}|dfS)N)rcrvrwryrzr|r}rrrrrrrntest_006rrs r generate_test_006rXs F : r cgd}|dfS)N))?r)?@H@)r)rr)rr)G@r)r)rrtest_007rrs r generate_test_007rks F : r ci}tD]\}}tD]\}}tttt t tfD]R}|\}}t|||}t|t} d |||} | || <StfD]R}|\}}t|||} t| t} d |||} | || <Sڌ|S)zGenerate all test images. Returns ------- results: dict A dictionary mapping test case name to xarray images. rFz{}_{}_{})antialias_optionscanvas_optionsrirlrprtrrrBrIcmap01formatrrD) resultsr aa_descriptorcanvascanvas_descriptorfuncr3namerKrL description aggregators r generate_test_imagesrts2G%6++ =)7 + + %F%******  + + $tvv (CC Kf555(//-):<< '* $$*- + +#tvv 4VVYOO JV444(//-):<< '* $$  + +, Nr cz|D]%\}}t||t||&dS)zSave all images as PNG and NetCDF files Parameters ---------- images: dict A dictionary mapping test case names to xarray images. N)itemsrXr\)imagesrrLs r save_test_imagesrsM#LLNN)) Sc;'''sK(((())r cbi}|D]\}}t|||<|S)zLoad all images from NetCDF files Returns ------- loaded: dict A dictionary mapping test case names to xarray images. )rr_)rloadedr_s r load_test_imagesrs>F ,,..<< Q.{;;{ Mr ct}t|}tt||D](}||||ksJ)dS)zTest case for all images. Will generate all test cases, then load all test cases from disk and compare them with each other N)rrrSlistkeysall)rrrs r test_antialiasingrs#$$F f % %F $v{{}}  {{}}BB {#vk'::??AAAAAABBr __main__)1__doc__rOxarrayr: datashaderr+pandasr(numpyr&pytest importorskipcmbinaryrPrQdirname__file__rRlinspacerCanvas regular_cvs reduced_cvsrrr4rBrDrIrXr\r_rirlrprtrrrrrrr__name__rrSrrrr r rs66t  V))  ',,rwx00& 9 9 0 0VKBK1--.. 0 0 0bi22 '::: bi22 ((<<< ;)K+CD)))25550666:#&        '''&    $&   D ) ) )    B B B z""$$F E$$v{{}}  V r