o 8Va@sddlmZddlmZddlmZddlmZGdddeZ e Z eddZd d d d ed fd d Z dddZ ddZ ddZd S))contextmanager)local expand_mul) dotprodsimpc@seZdZddZdS)DotProdSimpStatecCs d|_dSN)state)selfr :/usr/lib/python3/dist-packages/sympy/matrices/utilities.py__init__ s zDotProdSimpState.__init__N)__name__ __module__ __qualname__r r r r r rs rccs*tj}z |t_dVW|t_dS|t_wr_dotprodsimp_stater )xoldr r r rs rcCs|Srr rr r r srNcCs0|dus tjdur |S|dustjdur|S|S)aSupport function for controlling intermediate simplification. Returns a simplification function according to the global setting of dotprodsimp operation. ``deffunc`` - Function to be used by default. ``offfunc`` - Function to be used if dotprodsimp has been turned off. ``onfunc`` - Function to be used if dotprodsimp has been turned on. ``dotprodsimp`` - True, False or None. Will be overridden by global _dotprodsimp_state.state if that is not None. FTr)ZdeffuncZofffuncZonfuncrr r r _get_intermediate_simps  rFcCst|dd|S)zYSame as ``_get_intermediate_simp`` but returns bools instead of functions by default.FT)r)defaultrr r r _get_intermediate_simp_bool,srcCs t|ddS)zReturns True if x is zero.Zis_zeroN)getattrrr r r _iszero3s rcCs t|dkS)zNTests by expand_mul only, suitable for polynomials and rational functions.rrrr r r _is_zero_after_expand_mul8s r)FN) contextlibrZ threadingrZsympy.core.functionrZsympy.simplify.simplifyrZ _dotprodsimprrrrrrr r r r s