B [@sddlmZmZddlmZmZmZmZmZm Z ddl m Z ddl mZddlmZddlmZmZddlmZdd Zee e<Gd d d eeZeje_eZGd d d eeZdS))print_functiondivision)BasicIntegerTupleDictSsympify) converter) MatrixBase) DenseMatrix) SparseMatrixMutableSparseMatrix) MatrixExprcCs|S)N)Z as_immutable)argr7lib/python3.7/site-packages/sympy/matrices/immutable.pysympify_matrix src@seZdZdZdZdZdZddZej Z e ddZ e d d Z d d Zd dZddZddZe ddZe ddZe ddZdS)ImmutableDenseMatrixarCreate an immutable version of a matrix. Examples ======== >>> from sympy import eye >>> from sympy.matrices import ImmutableMatrix >>> ImmutableMatrix(eye(3)) Matrix([ [1, 0, 0], [0, 1, 0], [0, 0, 1]]) >>> _[0, 0] = 42 Traceback (most recent call last): ... TypeError: Cannot set values of ImmutableDenseMatrix Tgn$@cOs |j||S)N)_new)clsargskwargsrrr__new__)szImmutableDenseMatrix.__new__cOst|dkr"t|dtr"|dS|dddkrRt|dkrFtd|\}}}n|j||\}}}t|}t|}t|}t|tst|}t ||||S)NrcopyTFzA'copy=False' requires a matrix be initialized as rows,cols,[list]) len isinstancerget TypeErrorZ_handle_creation_inputslistrrrr)rrrrowscolsZ flat_listrrrr.s   zImmutableDenseMatrix._newcCs |jdjS)N)r)selfrrr_mat@szImmutableDenseMatrix._matcCst|||fS)N)r __getitem__)r&ijrrr_entryGszImmutableDenseMatrix._entrycGstd|jdS)NzCannot set values of {})r!format __class__)r&rrrr __setitem__Jsz ImmutableDenseMatrix.__setitem__cCsFt|dr|j|jkrtjSt|tr4t|ts4dS||}t|jS)aHelper method for Equality with matrices. Relational automatically converts matrices to ImmutableDenseMatrix instances, so this method only applies here. Returns True if the matrices are definitively the same, False if they are definitively different, and None if undetermined (e.g. if they contain Symbols). Returning None triggers default handling of Equalities. shapeN) hasattrr/rZfalserrrr is_zero)r&otherZdiffrrr_eval_EqMs  zImmutableDenseMatrix._eval_EqcsR|j|jfdd|D}|jt|ttfdd|DddiddS)Nc3s$|]}D]}||Vq qdS)Nr).0r)r*)r$colsListrr dsz5ImmutableDenseMatrix._eval_extract..c3s|]}|VqdS)Nr)r4r))matrrr6fsr F)r)r'r$rrr)r&ZrowsListr5indicesr)r$r5r7r _eval_extract_s z"ImmutableDenseMatrix._eval_extractcCst|jdS)Nr)intr)r&rrrr$hszImmutableDenseMatrix.colscCst|jdS)Nr)r:r)r&rrrr#lszImmutableDenseMatrix.rowscCstdd|jddDS)Ncss|]}t|VqdS)N)r:)r4r)rrrr6rsz-ImmutableDenseMatrix.shape..r%)tupler)r&rrrr/pszImmutableDenseMatrix.shapeN)__name__ __module__ __qualname____doc__Z _iterable_class_priorityZ _op_priorityrr__hash__ classmethodrpropertyr'r+r.r3r9r$r#r/rrrrrs    rc@sBeZdZdZdZdZeddZddZdd Z d d Z e j Z d S) ImmutableSparseMatrixaCreate an immutable version of a sparse matrix. Examples ======== >>> from sympy import eye >>> from sympy.matrices.immutable import ImmutableSparseMatrix >>> ImmutableSparseMatrix(1, 1, {}) Matrix([[0]]) >>> ImmutableSparseMatrix(eye(3)) Matrix([ [1, 0, 0], [0, 1, 0], [0, 0, 1]]) >>> _[0, 0] = 42 Traceback (most recent call last): ... TypeError: Cannot set values of ImmutableSparseMatrix >>> _.shape (3, 3) T cOsRt|}t|j}t|j}t|j}t||||}|j|_|j|_|j|_|S)N)rrr#r$r_smatrr)rrrsr#r$r7objrrrrs   zImmutableSparseMatrix._newcOs |j||S)N)r)rrrrrrrszImmutableSparseMatrix.__new__cGs tddS)Nz*Cannot set values of ImmutableSparseMatrix)r!)r&rrrrr.sz!ImmutableSparseMatrix.__setitem__cCs tt|jf|jt|jfS)N)hashtyper<r/r;rF)r&rrrrAszImmutableSparseMatrix.__hash__N) r<r=r>r?Z is_Matrixr@rBrrr.rArr3rrrrrD~s rDN)Z __future__rrZ sympy.corerrrrrr Zsympy.core.sympifyr Zsympify_converterZsympy.matrices.matricesr Zsympy.matrices.denser Zsympy.matrices.sparser rZsympy.matrices.expressionsrrrr1ZImmutableMatrixrDrrrrs     h