U Hv fJ@sdZddlZddlZddlmZddlmZmZm Z dddd d d gZ Gd dde Z Gd dde ZddZGdd d ZdddZddZddZGdd d ZGdd d ZddZdddZdS)a Implementation of Harwell-Boeing read/write. At the moment not the full Harwell-Boeing format is supported. Supported features are: - assembled, non-symmetric, real matrices - integer for pointer/indices - exponential format for float values, and int format N) csc_matrix)FortranFormatParser IntFormat ExpFormatMalformedHeaderhb_readhb_writeHBInfoHBFile HBMatrixTypec@s eZdZdS)rN__name__ __module__ __qualname__rr?/tmp/pip-unpacked-wheel-96ln3f52/scipy/io/_harwell_boeing/hb.pyrsc@s eZdZdS) LineOverflowNr rrrrr#srcCs|j|jd|dS)zcReturn the number of bytes to read to get every full lines for the given parsed fortran format.r)repeatwidth)fmtnlinesrrr _nbytes_full'src@s8eZdZed ddZeddZdd d Zd d ZdS)r Default title0NcCs|jdd}|j}|j}|j}|j\} } |j} |dkrtt |d} tt |d} |j j tj dkrt t t| }q|j j tj dkrtt t| }qtd|j j ntd|dkr._nlines)tocscindptrindicesdatashapeZnnzrZ from_numbernpmaxdtypekindZ typecodesrabsNotImplementedErrorZ isrealobj ValueErrorr r!fortran_format)clsmtitlekeymxtyperpointerr%valuesnrowsncols nnon_zerosZ pointer_fmtZ indices_fmtZ values_fmttpr"pointer_nlinesindices_nlines values_nlines total_nlinesrrr from_data.sZ        zHBInfo.from_datacCsN|d}t|dks&td||dd}|dd}|d}t|dkshtd|t|dd}t|dd}t|dd }t|d d}|dd} | d krd } nt| } | d kstd |d}t|d kstd||dd} t| dks,tdt| } | j dkrNtd| | j dksftd| | j dksztd|dddkstd|t|dd} t|dd } t|d d}t|dd }|d kstd||d}| }t|dkstd||||||||| | | ||d |d|d| |S)a6Create a HBInfo instance from a file object containing a matrix in the HB format. Parameters ---------- fid : file-like matrix File or file-like object containing a matrix in the HB format. Returns ------- hb_info : HBInfo instance  Hz8Expected at least 72 characters for first line, got: %sN8z9Expected at least 56 characters for second line, got: %s*rz5Only files without right hand side supported for now.Fz6Expected at least 72 character for third line, got: %sz'mxtype expected to be 3 characters long)rrz=Only real or integer matrices supported for now (detected %s)rz9Only unsymmetric matrices supported for now (detected %s)r z)Only assembled matrices supported for nowz z!Malformed data for third line: %sz4Unexpected value %d for nltvl (last entry of line 3)zExpected 3 formats, got %sr) readlinestriplenr.rstrip _expect_intupperr from_fortran value_type structurestoragesplit)r0fidliner2r3r>r;r<r=Z rhs_nlinesZmxtype_sr4r7r8r9 nelementalsctrrr from_file{s          zHBInfo.from_filercCs||_||_|dkrd}t|dkr,td|dkr8d}t|dkrTtd|t||_||_||_ ||_ t }| | }t |tstd|| | }t |tstd || | }t |tr|jd krtd ||ftj}n 72 charactersz|No Keyz!key is > 8 characters (key is %s)z.Expected int format for pointer format, got %sz.Expected int format for indices format, got %s)rcomplexz6Inconsistency between matrix type %s and value type %s)rz Unsupported format for values %r)$r2r3rLr.warningswarnrr>r;r<r=rparse isinstancerrrQr(Zfloat64intpointer_formatindices_format values_formatZint32Z pointer_dtypeZ indices_dtype values_dtyperpointer_nbytes_fullindices_nbytes_fullvalues_nbytes_fullr7r8r9rWr4)selfr2r3r>r;r<r=r4r7r8r9Zpointer_format_strZindices_format_strZvalues_format_strZright_hand_sides_nlinesrWparserrarbrcrdrrr__init__sp              zHBInfo.__init__cCs|jd|jdg}|d|j|j|j|jf|d|jj d|j |j |j df|j j }|jj }|jj }|d|d|d|d fd |S) zr;r<r=r4r/r7r8r9rarbrcjoin)rhheaderZpffmtZiffmtZvffmtrrrdumps*z HBInfo.dump)rrNN)rr)rrr classmethodr?rYrjrqrrrrr -s L X Hc CsLz t|WStk rF}z|dkr(d}t|||W5d}~XYnXdS)NzExpected an int, got %s)r`r.)valuemsgerrrrN+s  rNc Csd||j|g}tj|tdd}d||j|g}tj|tdd}d||j|g}tj||j dd}z$t ||d|df|j |j fdWSt k r}z|W5d}~XYnXdS)NrF )r*sepr)r')roreadrerJr( fromstringr`rfrgrdrr7r8r.) contentrpZ ptr_stringptrZ ind_stringindZ val_stringvalrurrr _read_hb_data4s6 r~cCst|jdd}dd}|||d|||jd|j|j|||jd|j|j|||j |j |j dS)NFrc Ss|j}||j}|d|d|j}||d|jfD]}||t|dq:|j|j}|dkr|||t||j|dddS)Nrr@r)Z python_formatrZreshapewritetupler!) farrrZpyfmtZ pyfmt_fullfullrowZnremainrrr write_arrayOs  z _write_data..write_arrayr@r) r#rrqr$r;rar%r<rbr&r=rc)r1rUrprrrr _write_dataLs   rc@seZdZdZdddddZddd d dd Zd d dZeddeDZ eddeDZ eddeDZ e ddZ dddZeddZddZdS)r zClass to hold the matrix type.RCPI)rr[patternrSUHZ)Z symmetricrZ hermitianZ skewsymmetricZ rectangularAE)r Z elementalcCsg|]\}}||fqSrr.0ijrrr ~szHBMatrixType.cCsg|]\}}||fqSrrrrrrrscCsg|]\}}||fqSrrrrrrrsc Cst|dkstdz8|j|d}|j|d}|j|d}||||WStk r|}ztd||W5d}~XYnXdS)NrHz:Fortran format for matrix type should be 3 characters longrrrIzUnrecognized format %s)rLr. _f2q_type_f2q_structure _f2q_storageKeyError)r0rrQrRrSrurrrrPs zHBMatrixType.from_fortranr cCsX||_||_||_||jkr(td|||jkr>td|||jkrTtd|dS)NzUnrecognized type %szUnrecognized structure %szUnrecognized storage %s)rQrRrS _q2f_typer._q2f_structure _q2f_storage)rhrQrRrSrrrrjs     zHBMatrixType.__init__cCs$|j|j|j|j|j|jSN)rrQrrRrrSrhrrrr/s    zHBMatrixType.fortran_formatcCsd|j|j|jfS)NzHBMatrixType(%s, %s, %s))rQrRrSrrrr__repr__szHBMatrixType.__repr__N)r )rrr__doc__rrrdictitemsrrrrrrPrjpropertyr/rrrrrr is0  c@sbeZdZdddZeddZeddZedd Zed d Zed d Z ddZ ddZ dS)r NcCs&||_|dkrt||_n||_dS)aCreate a HBFile instance. Parameters ---------- file : file-object StringIO work as well hb_info : HBInfo, optional Should be given as an argument for writing, in which case the file should be writable. N)_fidr rY_hb_info)rhfilehb_inforrrrjs zHBFile.__init__cCs|jjSr)rr2rrrrr2sz HBFile.titlecCs|jjSr)rr3rrrrr3sz HBFile.keycCs |jjjSr)rr4rQrrrrtypesz HBFile.typecCs |jjjSr)rr4rRrrrrrRszHBFile.structurecCs |jjjSr)rr4rSrrrrrSszHBFile.storagecCst|j|jSr)r~rrrrrr read_matrixszHBFile.read_matrixcCst||j|jSr)rrr)rhr1rrr write_matrixszHBFile.write_matrix)N) rrrrjrr2r3rrRrSrrrrrrr s      c CsBdd}t|dr||St|}||W5QRSQRXdS)aRead HB-format file. Parameters ---------- path_or_open_file : path-like or file-like If a file-like object, it is used as-is. Otherwise, it is opened before reading. Returns ------- data : scipy.sparse.csc_matrix instance The data read from the HB file as a sparse matrix. Notes ----- At the moment not the full Harwell-Boeing format is supported. Supported features are: - assembled, non-symmetric, real matrices - integer for pointer/indices - exponential format for float values, and int format Examples -------- We can read and write a harwell-boeing format file: >>> from scipy.io import hb_read, hb_write >>> from scipy.sparse import csr_matrix, eye >>> data = csr_matrix(eye(3)) # create a sparse matrix >>> hb_write("data.hb", data) # write a hb file >>> print(hb_read("data.hb")) # read a hb file (0, 0) 1.0 (1, 1) 1.0 (2, 2) 1.0 cSst|}|Sr)r rrUZhbrrr _get_matrixszhb_read.._get_matrixrxN)hasattropen)path_or_open_filerrrrrrs %  c shjdddkrtfdd}t|dr>||St|d}||W5QRSQRXdS)aWrite HB-format file. Parameters ---------- path_or_open_file : path-like or file-like If a file-like object, it is used as-is. Otherwise, it is opened before writing. m : sparse-matrix the sparse matrix to write hb_info : HBInfo contains the meta-data for write Returns ------- None Notes ----- At the moment not the full Harwell-Boeing format is supported. Supported features are: - assembled, non-symmetric, real matrices - integer for pointer/indices - exponential format for float values, and int format Examples -------- We can read and write a harwell-boeing format file: >>> from scipy.io import hb_read, hb_write >>> from scipy.sparse import csr_matrix, eye >>> data = csr_matrix(eye(3)) # create a sparse matrix >>> hb_write("data.hb", data) # write a hb file >>> print(hb_read("data.hb")) # read a hb file (0, 0) 1.0 (1, 1) 1.0 (2, 2) 1.0 FrNcst|}|Sr)r rrrr1rr _set_matrix2s zhb_write.._set_matrixrw)r#r r?rr)rr1rrrrrrr s(    )N)N)rr\Znumpyr(Z scipy.sparserZ_fortran_format_parserrrr__all__ ExceptionrWarningrrr rNr~rr r rr rrrrs(   =/0