o o h/@sddlmZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z ddlmZdd lmZdd lmZdd lmZd d ZdddZddZdddZddZddZddZdS)Tuple)Basic)Expr) AppliedUndef) Relational)Dummy)sympify)BooleanFunction)ImageSet) FiniteSet)IndexedcCsrt|tttfs |g}tdd|DrtStjdd|D}|jdd|D}|p8tjdd|DS)aReturns the free symbols of a symbolic expression. If the expression contains any of these elements, assume that they are the "free symbols" of the expression: * indexed objects * applied undefined function (useful for sympy.physics.mechanics module) css|]}t|VqdSNcallable.0erh/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/sympy/plotting/utils.py sz$_get_free_symbols..cSg|]}|tqSr)atomsr rrrr z%_get_free_symbols..cSrr)rrrrrrrrcSg|]}|jqSr free_symbolsrrrrr) isinstancelisttuplesetallunion)exprsfreerrr_get_free_symbolss r' csH|t}|D]}t|tfddtd|D}|||}q|S)aExtract numerical solutions from a set solution (computed by solveset, linsolve, nonlinsolve). Often, it is not trivial do get something useful out of them. Parameters ========== n : int, optional In order to replace ImageSet with FiniteSet, an iterator is created for each ImageSet contained in `set_sol`, starting from 0 up to `n`. Default value: 10. csg|]}tqSr)next)rnitrrr1z$extract_solution..r)findr iterr rangesubs)set_solr*imagesimsrr+rextract_solution!s r6cCst|tr|St|}t|D]3\}}t|ttfr&tt|ddi||<qt|ttfsBt |sB|j j dkrls z _create_ranges..NzToo many free symbols. zExpected {} free symbols. zReceived {}: {}z)Too many ranges. Received %s, expected %scSg|]}|dqSrrrrrrrr~r-z"_create_ranges..z$Multiple ranges with the same symbolcSrDrErrFrrrrr-rz>Incompatible free symbols of the expressions with the ranges. z$Free symbols in the expressions: {} zFree symbols in the ranges: {}) r' differencekeyslen ValueErrorformatr"r$appendr0r) r%rangesnparlabelparamsget_default_rangerrfssymbolsr5r?rrr_create_rangesUsJ      rUcCsRt|to(t|dko(t|jdt o(|jdjo(t|jdt o(|jdjS)zUA range is defined as (symbol, start, end). start and end should be numbers. )rrrJr>r: is_number)rGrrr _is_ranges    rZcGsxdd|D}dd|D}|sdn|d}dd|D}|s!dn|d}dd|D}ddt||D}||||fS) aGiven a list/tuple of arguments previously processed by _plot_sympify() and/or _check_arguments(), separates and returns its components: expressions, ranges, label and rendering keywords. Examples ======== >>> from sympy import cos, sin, symbols >>> from sympy.plotting.utils import _plot_sympify, _unpack_args >>> x, y = symbols('x, y') >>> args = (sin(x), (x, -10, 10), "f1") >>> args = _plot_sympify(args) >>> _unpack_args(*args) ([sin(x)], [(x, -10, 10)], 'f1', None) >>> args = (sin(x**2 + y**2), (x, -2, 2), (y, -3, 3), "f2") >>> args = _plot_sympify(args) >>> _unpack_args(*args) ([sin(x**2 + y**2)], [(x, -2, 2), (y, -3, 3)], 'f2', None) >>> args = (sin(x + y), cos(x - y), x + y, (x, -2, 2), (y, -3, 3), "f3") >>> args = _plot_sympify(args) >>> _unpack_args(*args) ([sin(x + y), cos(x - y), x + y], [(x, -2, 2), (y, -3, 3)], 'f3', None) cSg|]}t|r|qSrrZrtrrrrz _unpack_args..cSg|] }t|tr|qSrrr:r]rrrrNrcSr`rrr;r]rrrrrbcSs,g|]}t|pt|ttfp|du qSr)rZrr:r;rr@rrrrs,cSsg|]\}}|r|qSrr)rr@brrrrr_)zip)r>rNlabelsrP rendering_kwresultsr%rrr _unpack_argss rjc s6|sgSg}|dd}tdd|d|Drgt|\}}}} tjdd|D} t|||||}|dkrCt||krCt|f}|D]} t| t t t f} | rT| fn| } | g| ||| RqE|St|\}}}} |rt|gng}t|t|| durt| nd}|dkr|d| n|}t|dt ttfs|g}|D]td dD}|s|}d dD}|s|}d dD}t|dkr| n|d}fd dt|Dt} td dDr| jddD} t||krt||d|}|sdn|d}| g|||Rq|S)a,Checks the arguments and converts into tuples of the form (exprs, ranges, label, rendering_kw). Parameters ========== args The arguments provided to the plot functions nexpr The number of sub-expression forming an expression to be plotted. For example: nexpr=1 for plot. nexpr=2 for plot_parametric: a curve is represented by a tuple of two elements. nexpr=1 for plot3d. nexpr=3 for plot3d_parametric_line: a curve is represented by a tuple of three elements. npar The number of free symbols required by the plot functions. For example, npar=1 for plot, npar=2 for plot3d, ... **kwargs : keyword arguments passed to the plotting function. It will be used to verify if ``params`` has ben provided. Examples ======== .. plot:: :context: reset :format: doctest :include-source: True >>> from sympy import cos, sin, symbols >>> from sympy.plotting.plot import _check_arguments >>> x = symbols('x') >>> _check_arguments([cos(x), sin(x)], 2, 1) [(cos(x), sin(x), (x, -10, 10), None, None)] >>> _check_arguments([cos(x), sin(x), "test"], 2, 1) [(cos(x), sin(x), (x, -10, 10), 'test', None)] >>> _check_arguments([cos(x), sin(x), "test", {"a": 0, "b": 1}], 2, 1) [(cos(x), sin(x), (x, -10, 10), 'test', {'a': 0, 'b': 1})] >>> _check_arguments([x, x**2], 1, 1) [(x, (x, -10, 10), None, None), (x**2, (x, -10, 10), None, None)] rQNcss |] }t|tttfVqdSr)rrrr rdrrrrsz#_check_arguments..cSrrrrrrrr rz$_check_arguments..rWrcSr`rrardrrrr/rbcSr[rr\rdrrrr2r_cSr`rrcrdrrrr5rbcsg|]}|qSrr)rr?argrrr:r-css|]}t| VqdSrrrdrrrr<scSrrrrdrrrr=rrA)getr#rjr"r$rUrJr!rrrr rMr rcopyr0)r>nexprrOkwargsoutputrQr%rNrPrhrexpris_exprr_rgr*new_argslrGrend_kwrrkr_check_argumentssT0   -rx)r()rAN)sympy.core.containersrsympy.core.basicrsympy.core.exprrsympy.core.functionrsympy.core.relationalrsympy.core.symbolrsympy.core.sympifyr sympy.logic.boolalgr sympy.sets.fancysetsr sympy.sets.setsr sympy.tensor.indexedr r'r6r9rUrZrjrxrrrrs$            F &