o Pf8 @sFdZddlZddlZddlZddlZddlZddlZddlZddlZddlm Z m Z m Z ddl m Z mZzddlZWn eyEdZYnwejrOddlmZejejeZddZd(d d Zd d Zd)ddZd*ddZ  d+de dede de dfde fddZ!ddZ"ddZ#ddZ$d d!Z%d"d#Z&d$d%Z'd&d'Z(dS),zA Utilities ------- Utility module for Folium helper functions. N)AnyCallableUnion Environment PackageLoaderColorMapcCsttdddS)zGet Jinja templates.branca templates)loaderrr r \/home/deployuser/azure_apps/autowriter/venv/lib/python3.10/site-packages/branca/utilities.py get_templates!sr$@cCsbt||kr |}|Sttt||}g}|dd|D]}||g7}|dg|d7}q|S)z Downsamples the number of legend values so that there isn't a collision of text on the legend colorbar (within reason). The colorbar seems to support ~10 entries as a maximum. N)lenintmathceil) legend_values max_labels legend_ticksspacerir r r legend_scaler&s  rc Csdd}g}t|dd|ddD]F\}}d}t|D];}||dd|dd||}||dd|dd||} ||dd |dd ||} |d d || | gqqg} t|D]} t| |d} t| t|d}| ||q`| S) a Given a list of hexcode values, will return a list of length nColors where the colors are linearly interpolated between the (r, g, b) tuples that are given. Examples -------- >>> linear_gradient([(0, 0, 0), (255, 0, 0), (255, 255, 0)], 100) cSsdd}t||d}t||t||}ttt||||dd}t|dkr0d|}|S)z Return the value correct value of a number that is in between start and finish, for use in a loop of length *length*. rx0)floatrhexsplitr)startfinishlengthrbasefractionrayngethexr r r_scaleDs$ zlinear_gradient.._scaleNrrir#)ziprangeappendjoinr"rr)hexListnColorsr, allColorsr%end nInterpolateindexrgbresultcounterr)r r rlinear_gradient8s  "  r@cCsd}d}t|tstdt|||krtd||kr!tdt|ts0tdt|d|dd d krL|d d}|d t|d }d }n|}|d t|d }d}ttj t d}t |}Wd n1svwYttj t d}t |} Wd n1swYttj t d}t |d} Wd n1swY|| vrt|dd } ||d urd} | s]| |dkr g} |D]} |d | vr| t| d dqtd|dtt| dtt| |}d}|D]-}d |vrq|d d|kr(qt|d d|kr?|}t|d d}q|sNt|||}|St||d d d|}|S|sh||d }|S||d d d d}|S)z Generate a colorbrewer color scheme of length 'len', type 'scheme. Live examples can be seen at http://colorbrewer2.org/ r-zn has to be an int, not a %szLThe maximum number of colors in a ColorBrewer sequential color series is 253zJThe minimum number of colors in a ColorBrewer sequential color series is 3z color should be a string, not a .N_r_r TFz _schemes.jsonzscheme_info.jsonzscheme_base_codes.jsoncodesz is not a valid ColorBrewer code QualitativerzgExpanded color support is not available for Qualitative schemes; restrict the number of colors for the z code to between z and rr) isinstancer TypeErrortype ValueErrorstrzfillopenospathr4rootpathjsonloadsreadgetr3r$minmaxkeysr@) color_coden maximum_n minimum_n base_codecore_color_code color_reversefschemes scheme_info core_schemesexplicit_schemematching_qualskeylongest_scheme_namelongest_scheme_nsn_name color_schemer r r color_brewerhs             rluppercCst|dr&t|dr|jdd}nd}d|t|d}n)t |t sGt |t sGt|drGt |||d }d t|d}nt t |}|d d S) asInfers the type of an image argument and transforms it into a URL. Parameters ---------- image: string, file or array-like object * If string, it will be written directly in the output file. * If file, it's content will be converted as embedded in the output file. * If array-like, it will be converted to PNG base64 string and embedded in the output. origin : ['upper' | 'lower'], optional, default 'upper' Place the [0, 0] index of the array in the upper left or lower left corner of the axes. colormap : callable, used only for `mono` image. Function of the form [x -> (r,g,b)] or [x -> (r,g,b,a)] for transforming a mono image into RGB. It must output iterables of length 3 or 4, with values between 0. and 1. Hint : you can use colormaps from `matplotlib.cm`. rUnamerCrpngzdata:image/{};base64,{}zutf-8__iter__)origincolormapzdata:image/png;base64,  )hasattrrnlowerr$formatbase64 b64encoderUdecoderIrMbytes write_pngrSrTdumpsreplace)imagerrrq fileformaturlror r r image_to_urls   rdatarqrrr returnc sddlm}tdurtdt||r|j}n t|r|}ndd}t|j\}}}|dvr5t dj|||fks?J|d krdt t t | jd }|d vr\t d |||fj|||fksnJ|d krtjt||d ffd dd}j|||fksJ|dksJjdkrtjddddjddddt<Wdn1swYd|dkrۈdddddddfdfddt|D}dd} dd| dtd||d d!ddd| d"t|d#| d$dgS)%a^ Transform an array of data into a PNG string. This can be written to disk using binary I/O, or encoded using base64 for an inline PNG like this: >>> png_str = write_png(array) >>> "data:image/png;base64," + png_str.encode("base64") Inspired from http://stackoverflow.com/questions/902761/saving-a-numpy-array-as-an-image Parameters ---------- data: numpy array or equivalent list-like object. Must be NxM (mono), NxMx3 (RGB) or NxMx4 (RGBA) origin : ['upper' | 'lower'], optional, default 'upper' Place the [0,0] index of the array in the upper left or lower left corner of the axes. colormap : ColorMap subclass or callable, optional Only needed to transform mono images into RGB. You have three options: - use a subclass of `ColorMap` like `LinearColorMap` - use a colormap from `matplotlib.cm` - use a custom function of the form [x -> (r,g,b)] or [x -> (r,g,b,a)]. It must output iterables of length 3 or 4 with values between 0 and 1. Returns ------- PNG formatted byte string rrNz4The NumPy package is required for this functionalitycSs |||dfS)Nrr rr r r"s zwrite_png..)rr-z5Data must be NxM (mono), NxMx3 (RGB), or NxMx4 (RGBA)r)r-rz9colormap must provide colors oflength 3 (RGB) or 4 (RGBA)r-r )axisruint8ignore)divideinvalidgo@)rr)rrrrvrcs*g|]}d|ddddfqS)N)tobytes).0rarrayr r Gs*zwrite_png..cSs2||}tdt||tddt|@S)Nz!Il)structpackrzlibcrc32)png_tagr chunk_headr r rpng_packIszwrite_png..png_packsPNG  sIHDRz!2I5BrAsIDAT sIEND)branca.colormapr np ImportErrorrIrgba_floats_tuplecallable atleast_3dshaperLrlistmapravelreshape concatenateonesdtypeerrstaterXisfiniteastyper4r2rrrcompress) rrqrrr colormap_callableheightwidthnblayersraw_datarr rrr|sX "       r|cs.dfddttDdddS)Nrcsg|]<\}}|tdkr |r |dr d|n|tdkr:|r:|dr:|dn|qS)rrF)risupperislowerrv)rrroutr rr_s   z_camelify..rF__)r4 enumeraterlstripr~rr rr _camelify[s  rcCst|ttfr t|dfSt|tr0td|}|r(t|d|dfStd|dt|t rOt|dttfrOt|dtrOt|d|dfSt d|d) Npxz([\d.]+)\s?(\w{1,5}|%)rr z Cannot parse z+, it should be a number followed by a unit.rzC, it should be a number or a string containing a number and a unit.) rIrr"rMre fullmatchstripgrouprLtuplerJ)valuematchr r r _parse_sizevs&     rcCs<t|drt|ddrttt|St|dddS|S)zMirrors the points in a list-of-list-of-...-of-list-of-points. For example: >>> _locations_mirror([[[1, 2], [3, 4]], [5, 6], [7, 8]]) [[[2, 1], [4, 3]], [6, 5], [8, 7]] rprNr)rurr_locations_mirrorrr r rrs rcCst|dr ttt|S|S)z?Transforms recursively a list of iterables into a list of list.rp)rurr_locations_tolistrr r rrs rcC"|dur|S|dur |St||SN)rWryr r rnone_min  rcCrr)rXrr r rnone_maxrrcCs`t|ttfr(t|r&t|dttfr#g}|D]}|t|7}q|S|gSgStd|d)zIterates over a list representing a feature, and returns a list of points, whatever the shape of the array (Point, MultiPolyline, etc). rzList/tuple type expected. Got rC)rIrrr iter_pointsrL)rrrr r rrsr)r)rA)Nrm)rmN))__doc__rxrSrrPrrtypingrrrrjinja2rrnumpyrr TYPE_CHECKINGrr rQabspathdirname__file__rRrrr@rlrrMr{r|rrrrrrrr r r rsT     0 a/  e