l jddlmZddlZddlmZddlmZmZmZddl Z ddgZ ddl m Z er ddlmZdd lmZGd d ZGd deZGd deZdS)) annotationsN)Iterable) TYPE_CHECKINGAnyoverloadLogBase LinearBase)Integer)Callable)VMobjectcdeZdZdZdddZedd Zedd Zdd Zdd ZddZdS) _ScaleBasezScale baseclass for graphing/functions. Parameters ---------- custom_labels Whether to create custom labels when plotted on a :class:`~.NumberLine`. F custom_labelsboolc||_dSN)r)selfrs W/home/agentuser/manim-venv/lib/python3.11/site-packages/manim/mobject/graphing/scale.py__init__z_ScaleBase.__init__s*valuefloatreturncdSrrrs rfunctionz_ScaleBase.functions/2sr np.ndarraycdSrrrs rrz_ScaleBase.function"s9<rct)aiThe function that will be used to scale the values. Parameters ---------- value The number/``np.ndarray`` to be scaled. Returns ------- float The value after it has undergone the scaling. Raises ------ NotImplementedError Must be subclassed. NotImplementedErrorrs rrz_ScaleBase.function%s $"!rct)zThe inverse of ``function``. Used for plotting on a particular axis. Raises ------ NotImplementedError Must be subclassed. r!rs rinverse_functionz_ScaleBase.inverse_function9s "!r val_rangeIterable[float]kw_argsrIterable[VMobject]c t)aCustom instructions for generating labels along an axis. Parameters ---------- val_range The position of labels. Also used for defining the content of the labels. Returns ------- Dict A list consisting of the labels. Can be passed to :meth:`~.NumberLine.add_labels() along with ``val_range``. Raises ------ NotImplementedError Can be subclassed, optional. r!)rr%r's rget_custom_labelsz_ScaleBase.get_custom_labelsCs ."!rN)F)rrrrrr)rrrr)r%r&r'rrr() __name__ __module__ __qualname____doc__rrrr$r*rrrrrs+++++222X2 <<<X<""""(""""""""""rrc2eZdZd d fd Zd dZd dZxZS) r ? scale_factorrcVt||_dS)zThe default scaling class. Parameters ---------- scale_factor The slope of the linear function, by default 1.0 N)superrr2)rr2 __class__s rrzLinearBase.__init__^s) (rrrc|j|zS)zMultiplies the value by the scale factor. Parameters ---------- value Value to be multiplied by the scale factor. r2rs rrzLinearBase.functionis 5((rc||jz S)zInverse of function. Divides the value by the scale factor. Parameters ---------- value value to be divided by the scale factor. r7rs rr$zLinearBase.inverse_functionsst(((r)r1)r2rr+)r,r-r.rrr$ __classcell__r5s@rr r ]sj ) ) ) ) ) ) )))))))))))))rc>eZdZddfd Zdd Zdd Z dddZxZS)r Tbaserrrcdt||_||_dS)aScale for logarithmic graphs/functions. Parameters ---------- base The base of the log, by default 10. custom_labels For use with :class:`~.Axes`: Whether or not to include ``LaTeX`` axis labels, by default True. Examples -------- .. code-block:: python func = ParametricFunction(lambda x: x, scaling=LogBase(base=2)) N)r4rr=r)rr=rr5s rrzLogBase.__init__s0$  *rrrc|j|z}|S)zMScales the value to fit it to a logarithmic scale.``self.function(5)==10**5``)r=)rr return_values rrzLogBase.functions"i. rct|tjr|dk}dd}n|dk}tj}|rt d|||j}|S) z9Inverse of ``function``. The value must be greater than 0rrrr=rcXtj|tj|z }|Sr)nplog)rr=r@s rfuncz&LogBase.inverse_function..funcs"&(fUmmbfTll&B ##rzIlog(0) is undefined. Make sure the value is in the domain of the function)rrr=rrr) isinstancerCndarrayanymathrD ValueErrorr=)rr conditionrEs rr$zLogBase.inverse_functions eRZ ( (  q(I $ $ $ $ $ I8D  [ UDI&& rrr%r&unit_decimal_placesint base_configr list[Integer]c ,fd|D}|S)aProduces custom :class:`~.Integer` labels in the form of ``10^2``. Parameters ---------- val_range The iterable of values used to create the labels. Determines the exponent. unit_decimal_places The number of decimal places to include in the exponent base_config Additional arguments to be passed to :class:`~.Integer`. c pg|]2}tjfdd|ddzi3S)unitz^{%s}.f)r r=r$).0irNrrLs r z-LogBase.get_custom_labels..sv% % %     4#8#8#;#; U>Q U U U UV   % % % rr)rr%rLrN tex_labelss` `` rr*zLogBase.get_custom_labelssA$% % % % % %  % % % r)r<T)r=rrrr+)r)r%r&rLrMrNrrrO)r,r-r.rrr$r*r9r:s@rrr~s+++++++, 0$%r) __future__rrIcollections.abcrtypingrrrnumpyrC__all__manim.mobject.text.numbersr r &manim.mobject.types.vectorized_mobjectr rr rrrrr`s?"""""" $$$$$$////////// l #......@((((((??????G"G"G"G"G"G"G"G"T))))))))BKKKKKjKKKKKr