a J5d @sddlmZGdddZGdddeZGdddeZGdd d eZGd d d eZGd d d eZGdddeZGdddeZ GdddeZ GdddeZ GdddeZ eeeeee e e e d Z dS))utilsc@s:eZdZdZdddZddZddZd d Zd d Zd S)Matrixa*A square matrix created from a list of numbers. Elements are accessible using matrix notation. Negative indexing is not allowed. :param list numbers: the elements of the matrix :param int size: the width (also height) of the matrix :param int min_index: the minimum index cCst||_||_||_dSN)listnumberssize min_index)selfrrr r K/var/www/html/django/DPS/env/lib/python3.9/site-packages/tsplib95/matrix.py__init__s zMatrix.__init__cCs |j|Sr)value_at)r keyr r r __getitem__szMatrix.__getitem__cCsL||j8}||j8}|||s6td|d|d|||}|j|S)zGet the element at row *i* and column *j*. :param int i: row :param int j: column :return: value of element at (i,j) (z, z) is out of bonuds)r is_valid_row_column IndexError get_indexr)r ijindexr r r rs     zMatrix.value_atcCs4d|ko|jkno2d|ko.|jkSS)zReturn True if (i,j) is a row and column within the matrix. :param int i: row :param int j: column :return: whether (i,j) is within the bounds of the matrix :rtype: bool rrr rrr r r r&szMatrix.is_valid_row_columncCs tdS)zReturn the linear index for the element at (i,j). :param int i: row :param int j: column :return: linear index for element (i,j) :rtype: int N)NotImplementedErrorrr r r r0szMatrix.get_indexN)r) __name__ __module__ __qualname____doc__r rrrrr r r r rs   rc@seZdZdZddZdS) FullMatrixzA complete square matrix. :param list numbers: the elements of the matrix :param int size: the width (also height) of the matrix :param int min_index: the minimum index cCs||j|Srrrr r r rCszFullMatrix.get_indexN)rrrrrr r r r r;srcs$eZdZdZdZfddZZS) HalfMatrixzA triangular half-matrix. :param list numbers: the elements of the matrix :param int size: the width (also height) of the matrix :param int min_index: the minimum index Tcs0||kr|jsdS|||\}}t||S)Nr) has_diagonal _fix_indicessuperrr __class__r r rRszHalfMatrix.value_at)rrrrr!r __classcell__r r r$r r Gsr c@s$eZdZdZdZddZddZdS) UpperDiagRowzUpper-triangular matrix that includes the diagonal. :param list numbers: the elements of the matrix :param int size: the width (also height) of the matrix :param int min_index: the minimum index TcCs2||kr||fn||f\}}|js*|d8}||fSNrr!rr r r r"cszUpperDiagRow._fix_indicescCs*|jt|j }t|||||Sr)rintr!r integer_sum)r rrnr r r riszUpperDiagRow.get_indexNrrrrr!r"rr r r r r'Ysr'c@s$eZdZdZdZddZddZdS) LowerDiagRowzLower-triangular matrix that includes the diagonal. :param list numbers: the elements of the matrix :param int size: the width (also height) of the matrix :param int min_index: the minimum index TcCs2||kr||fn||f\}}|js*|d8}||fSr(r)rr r r r"xszLowerDiagRow._fix_indicescCst||Sr)rr+rr r r r~szLowerDiagRow.get_indexNr-r r r r r.nsr.c@seZdZdZdZdS)UpperRowzUpper-triangular matrix that does not include the diagonal. :param list numbers: the elements of the matrix :param int size: the width (also height) of the matrix :param int min_index: the minimum index FNrrrrr!r r r r r/sr/c@seZdZdZdZdS)LowerRowzLower-triangular matrix that does not include the diagonal. :param list numbers: the elements of the matrix :param int size: the width (also height) of the matrix :param int min_index: the minimum index FNr0r r r r r1sr1c@s eZdZdS)UpperColNrrrr r r r r2sr2c@s eZdZdS)LowerColNr3r r r r r4sr4c@s eZdZdS) UpperDiagColNr3r r r r r5sr5c@s eZdZdS) LowerDiagColNr3r r r r r6sr6) Z FULL_MATRIXZUPPER_DIAG_ROWZ UPPER_ROWZLOWER_DIAG_ROWZ LOWER_ROWZUPPER_DIAG_COLZ UPPER_COLZLOWER_DIAG_COLZ LOWER_COLN)rrrr r'r.r/r1r2r4r5r6TYPESr r r r s* 6