o *ήc0@sddlmZddlmZmZddlmZddlmZm Z m Z m Z ddl m Z GdddZiZGd d d Zed \ZZZZZZeeeeed <eed dddefed<eeddefgded<eeddefgded<dS))Basic)Symbolsymbols)lambdify) interpolate rinterpolate create_bounds update_boundssiftc@sHeZdZgdgdfZdZddZddZdd Zd d Zd d Z dS) ColorGradient)皙?rr)?rr)?cstdkrt|_ddg|_n2tdkrCtddkr#tdfddtdtdD|_fd dtdtdD|_t|jt|jksOJdS) Nrrrzlen(args) should be evencg|]}|qSr.0iargsrM/tmp/pip-target-vg8gfxp4/lib/python/sympy/plotting/pygletplot/color_scheme.py z*ColorGradient.__init__..rcrrrrrrrrr)lenlistcolors intervals ValueErrorrange)selfrrrr__init__ s      zColorGradient.__init__cCs,t}dd|jD|_|jdd|_|S)NcSsg|]}|ddqSNr)rerrrrz&ColorGradient.copy..)r rr )r#crrrcopyszColorGradient.copycCsNt|j}d}||dkr%|j||kr%|d7}||dkr%|j||ks|S)Nrr)rr )r#vmrrrr_find_intervals zColorGradient._find_intervalcCsJ||}t|j|d|j||}t|j|d||j|||S)Nr)r,rr rr)r#axisr*rrrr_interpolate_axis$s $zColorGradient._interpolate_axiscCs"|j}|d||d||d|fS)Nrrr)r.)r#rgbr(rrr__call__)szColorGradient.__call__N) __name__ __module__ __qualname__rr r$r)r,r.r2rrrrr s  r c@sheZdZddZddZddZddZd d Zd d Zd dZ dddZ dddZ ddZ ddZ dS) ColorSchemecOs||_dt|_|_t|dkr$t|dts$t|dr$|d|_n7t|dkrRt|dtrR|dt vrIt |d}|j|j |_|_nt d|d|_n | |\|_|_| t|jtsitddS)Nrr x,y,z,u,vzHColor gradient not properly initialized. (Not a ColorGradient instance.))rr fgradientr isinstancercallablestrdefault_color_schemesr)r_interpret_args_test_color_functionr!)r#rkwargscsrrrr$2s&    zColorScheme.__init__c Cs d|j}}||\}}||}||}ddd|D}dd|D}ddd|D}td||f} t|dkrX|d} z t|| | | g}Wn,tyW| wt|d krw|\} } } z t|| | | g}Wn tyv| wtd t|dkrg}ndt|dkr|d}nYt|d krz |\\}}}\}}}Wn tytd w|}n8t|d krz |\\}}\}}\}}Wn t ytd w|||g|||gg}ntdddd|D|r z t |}W||fSt y }z td|t |fd}~ww||fS)N, cs|]}t|VqdSr%r<)rfarrr Lz.ColorScheme._interpret_args..csrCr%rDrsarrrrFMrGcss"|] }|ddkr|VqdS)unboundrN)findrHrrrrFNs z4Could not interpret arguments %s as functions of %s.rrzDA ColorScheme must provide 1 or 3 functions in x, y, z, u, and/or v.rz^If two color arguments are given, they must be given in the format (r1, g1, b1), (r2, g2, b2).zIf three color arguments are given, they must be given in the format (r1, r2), (g1, g2), (b1, b2). To create a multi-step gradient, use the syntax [0, colorStart, step1, color1, ..., 1, colorEnd].z3Don't know what to do with collection arguments %s.csrCr%rD)rlrrrrFrGzFCould not initialize a gradient with arguments %s. Inner exception: %s) r9 _sort_args_pop_symbol_list _fill_in_varsjoinr!rr TypeError Exceptionr r<)r#rr8r9atomslistssf_strs_strf_errorfvfrfgfbgargsr1g1b1r2g2b2exrrrr>Esr                 zColorScheme._interpret_argscCszg}|D]"}d}|D]}|durt|tsd}nq |r&||||qt|dkr1|dSt|dkr9gStd)NTFrrz9Only one list of Symbols can be given for a color scheme.)r:rremoveappendrr!)r#rU symbol_listsrMmarkrVrrrrOs"    zColorScheme._pop_symbol_listcCsZtd}td}t|dkr|St|ttfs|t|dkr!|S|D] }|dur0t|ts0|q#ddtddD}t|dkrI|d|d<|St|d kri|ddur[|d|d<|ddurg|d|d <|St|dkr|ddur{|d|d<|ddur|d|d<|d dur|d |d <t|d kr|d|d<t|d kr|d |d <|S) Nr7zCould not find what to plot.rcSsg|]}td|qS)z unbound%i)rrrrrrr'z-ColorScheme._fill_in_vars..rrLr)rr!rr:tuplerrr")r#rdefaultsv_errorrVvarsrrrrPsD                  zColorScheme._fill_in_varscCst|dddd\}}||fS)NcSst|ttfSr%)r:rmr)arrrsz(ColorScheme._sort_args..T)binaryr )r#rrUrTrrrrNs zColorScheme._sort_argscCs|t|js tdz|ddddd}t|dkrtdWdSty*tdty4tdty=YdSw)NzColor function is not callable.rrLzlength should be equal to 3zZColor function needs to accept x,y,z,u,v, as arguments even if it doesn't use all of them.z-Color function needs to return 3-tuple r,g,b.)r;r8r!rrRAssertionErrorrS)r#resultrrrr?s     z ColorScheme._test_color_functioncCs*z ||||||WStyYdSwr%)r8rS)r#xyzur*rrrr2s  zColorScheme.__call__NcCs(t}t}t|r|t|dtt|D]=}||dur&|dn(||\}} } ||d} } ||| | | | } | durIt| } t|| || t|rU|qtt|D]5}||durtdD]}t||d||d||||||<qh|j||||<t|r|q\|S)zt Apply this color scheme to a set of vertices over a single independent variable u. rNrLrr r rr;rr"rgr rr9)r#vertsu_setset_leninc_posboundscverts_urvrwrxryr*r(_crrrapply_to_curves8       zColorScheme.apply_to_curvecCst}t}t|r|t|t|dtt|D]T}t} tt|D]C} ||| dur7| dn,||| \} } } |||| }}|| | | ||}|dur^t|}t||| |t|rj|q'|| qtt|D]H}tt|D]?} ||| durtdD]}t||d||d||| |||| |<q|j||| ||| <t|r|qqw|S)zv Apply this color scheme to a set of vertices over two independent variables u and v. rNrLrrrz)r#r{r|v_setr}r~rrrcolumn_vrvrwrxryr*r(rrrrapply_to_surface sB       zColorScheme.apply_to_surfacecCsddd|jDS)NrBcsrCr%rD)rrqrrrrF8rGz'ColorScheme.str_base..)rQrr#rrrstr_base7szColorScheme.str_basecCs d|S)Nz%s)rrrrr__repr__:s zColorScheme.__repr__)NN)r3r4r5r$r>rOrPrNr?r2rrrrrrrrr60sJ+  &* r6z x,y,z,t,u,vrainbow)rr ףp= ?)rrrNzfade) r)皙?rrgffffff?)r皙?rg?)333333?rr?)rrrr)rrrzfade3) r)rrrr)rrrg?)ffffff?rrr)rrrg333333?)rgffffff?rr)rrrzfade4)sympy.core.basicrsympy.core.symbolrrsympy.utilities.lambdifyrutilrrr r sympy.utilities.iterablesr r r=r6rvrwrxtryr*rrrrs*   %