>qjPdZddlZddlZddlZddlmZddlmZm Z ddl m Z m Z m Z ddgZGd d eZGd d eZd ZGddZddZdZdZGddZGddZeddZddZy)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)_NoValue) csc_array csc_matrix)FortranFormatParser IntFormat ExpFormathb_readhb_writec eZdZy)MalformedHeaderN__name__ __module__ __qualname__Z/opt/rentech/trading_bot/.venv/lib/python3.12/site-packages/scipy/io/_harwell_boeing/hb.pyr r rr c eZdZy) LineOverflowNrrrrrr$rrrcF|j|jzdz|dz zS)zcReturn the number of bytes to read to get every full lines for the given parsed fortran format.r)repeatwidth)fmtnliness r _nbytes_fullr(s% JJ "Q &6A: 66rc>eZdZeddZedZ ddZdZy)HBInfoNc|jd}|j}|j}|j}|j\} } |j } |bt jtj|dzd} t jtj|dzd} |jjtjdvrRtjtjtj|tjd }n|jjtjdvr?t jtjtj|d }n0d|jjd}t!|t!d |tj"|s t%d |jjtjdvrd }nE|jjtjdvrd }nt!d|jd t'|dd}n t%dd}|| |j(}|| |j(}|||j(}||z|z}||||||||| | | | j*| j*|j* S)aCreate a HBInfo instance from an existing sparse matrix. Parameters ---------- m : sparse array or matrix the HBInfo instance will derive its parameters from m title : str Title to put in the HB header key : str Key mxtype : HBMatrixType type of the input matrix fmt : dict not implemented Returns ------- hb_info : HBInfo instance Fcopyr)initialAllFloat AllIntegerztype z not implemented yetzfmt argument not supported yet.z Complex values not supported yetintegerrealz for values not implemented unsymmetric assembledz mxtype argument not handled yet.cR||jz}||jz|k7r|dz }|S)Nr)r)rsizers r_nlinesz!HBInfo.from_data.._nlinesns/SZZ'F "d*! Mr)tocscindptrindicesdatashapennzr from_numbernpmaxdtypekind typecodesr absfloat64NotImplementedError isrealobj ValueError HBMatrixTyper+fortran_format)clsmtitlekeymxtyperpointerr/valuesnrowsncols nnon_zeros pointer_fmt indices_fmt values_fmtmessagetpr,pointer_nlinesindices_nlines values_nlines total_nliness r from_datazHBInfo.from_data/sh* GGG (())ww uUU ;$//w{A0NOK#//w{A0NOK||  BLL$<<&22VVBFF6NBJJqMBBD ""bll<&@@&22VVBFF6NA668 "&,,"3"3!44HI)'22%&GH H ><<' !CDD||  BLL$>>""bll:&>>)FLL>)DEGG!"m[AF?@ @  !gll; gll; FKK8 %6F 5# ..- E5*  & & (B(B  % % ' 'rc4|jjd}t|dkDstd||dd}|dd}|jjd}t|j dk\std|t |dd}t |dd}t |dd }t |d d}|ddj} | d k(rd } n t | } | d k(s td |jjd}t|d k\std||ddj } t| dk(s tdtj| } | jdvrtd| d| jdk(std| d| jdk(s td|dddk(std|t |dd} t |dd } t |d d}t |dd }|d k(std|d|jjd}|j}t|dk(std||||||||| | | ||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  Hz6Expected at least 72 characters for first line, got: N8z7Expected at least 56 characters for second line, got: *rz5Only files without right hand side supported for now.Fz4Expected at least 72 character for third line, got: z'mxtype expected to be 3 characters long)r'r&z:Only real or integer matrices supported for now (detected )r(z6Only unsymmetric matrices supported for now (detected r)z)Only assembled matrices supported for nowz zMalformed data for third line: zUnexpected value z! for nltvl (last entry of line 3)zExpected 3 formats, got r) readlinestriplenr=rstrip _expect_intupperr> from_fortran value_type structurestoragesplit)r@fidlinerBrCrRrOrPrQ rhs_nlinesmxtype_srDrGrHrI nelementalscts r from_filezHBInfo.from_files ||~##D)4y2~''+f./ /Sb 23i||~##D)4;;=!R'''+f./ /"49- $T"R[1$T"R[1#DBK0 "R[&&(  J$Z0JQ$% %||~##D)4yB &&*V-. .8>>#8}!FG G**84   $7 7..4XQ89 9=0..4XQ89 9~~,HI IAbzX%>tfEF FDBK(DBK( b- !$r"+. a#K=0QR  ||~##D) ZZ\2w!|7t<= =5#5%a5"Q%A{ , ,rcR|d}t|dkDr td|d}t|dkDr tjd|dtd ||_||_||_||_||_ ||_ t}|j| }t|tstd ||j| }t|tstd ||j| }t|tr0|j d vrtd|d|t"j$}nDt|tr&|j dvrtd|d|t&}ntd|||_||_||_t"j.|_t"j.|_||_||_t7|||_||_ t7|||_||_ t7|||_||_| |_ | |_!||_"||_#y)z@Do not use this directly, but the class ctrs (from_* functions).NzNo TitlerVztitle cannot be > 72 charactersz|No Keyzkey is > 8 characters (key is r^r]) stacklevelz,Expected int format for pointer format, got z,Expected int format for indices format, got )r'complexz"Inconsistency between matrix type z and value type )r&zUnsupported format for values )$rbr=warningswarnrrBrCrRrOrPrQrparse isinstancerr rgr4r:intpointer_formatindices_format values_formatint32 pointer_dtype indices_dtype values_dtyperpointer_nbytes_fullindices_nbytes_fullvalues_nbytes_fullrGrHrIrorD)selfrBrCrRrOrPrQrDrGrHrIpointer_format_strindices_format_strvalues_format_strright_hand_sides_nlinesroparserr{r|r}rs r__init__zHBInfo.__init__sJ =E u:?>? ? ;C s8a< MM:3%qA&1 6 (,,*$&&89.)4K ./12 2 &89.)4K ./12 2 %67 mY /  (;; #EfXN//.write_arrayMs!!SZZ' ,FQJ#**,-<<3:: 67 4C GGJs+d2 3 4''DII% Q; GGUW_b71B1C.D(EEL M rrUr) r-rrr.rOr{r/rPr|r0rQr})rArkrrs r _write_datarJs UA NIIfkkmIIdOQXXaZ!6!6%%'QYYq[&"7"7%%'QVVV11$$&rcteZdZdZdddddZddd d dd Zd d dZejDcic]\}}|| c}}}ZejDcic]\}}|| c}}}Z ejDcic]\}}|| c}}}Z e dZ ddZ edZdZycc}}}wcc}}}wcc}}}w)r>zClass to hold the matrix type.RCPI)r'rupatternr&SUHZ) symmetricr( hermitian skewsymmetric rectangularAE)r) elementalct|dk(s td |j|d}|j|d}|j|d}||||S#t $r}td||d}~wwxYw)Nr]z:Fortran format for matrix type should be 3 characters longrrr_zUnrecognized format )rbr= _f2q_type_f2q_structure _f2q_storageKeyError)r@rrgrhrirs rrfzHBMatrixType.from_fortrans3x1}/0 0 Bs1v.J**3q62I&&s1v.Gz9g6 6 B3C59: A Bs?A A8$A33A8c||_||_||_||jvrt d|||j vrt d|||j vrt d|y)NzUnrecognized type zUnrecognized structure zUnrecognized storage )rgrhri _q2f_typer=_q2f_structure _q2f_storage)rrgrhris rrzHBMatrixType.__init__s}$" T^^ +1*>? ? D// /6ykBC C $++ +4WI>? ? ,rc|j|j|j|jz|j|j zSN)rrgrrhrrirs rr?zHBMatrixType.fortran_formatsE~~doo.""4>>23  ./ /rcVd|jd|jd|jdS)Nz HBMatrixType(z, r^)rgrhrirs r__repr__zHBMatrixType.__repr__s,t/r$..1ADLL>QRSSrN)r))rrr__doc__rrritemsrrrrrfrpropertyr?r).0ijs000rr>r>gs( I  NL #,//"344$!QA4I'5';';'=>>tq!ad>N%1%7%7%9::TQAqD:L B B @// TG5>:s B% B,; B3r>cpeZdZd dZedZedZedZedZedZ dZ d Z y) HBFileNcZ||_|tj||_y||_y)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)_fidrrq_hb_info)rfilehb_infos rrzHBFile.__init__s+ ?",,T2DM$DMrc.|jjSr)rrBrs rrBz HBFile.titles}}"""rc.|jjSr)rrCrs rrCz HBFile.keys}}   rcB|jjjSr)rrDrgrs rtypez HBFile.types}}##...rcB|jjjSr)rrDrhrs rrhzHBFile.structures}}##---rcB|jjjSr)rrDrirs rrizHBFile.storages}}##+++rcBt|j|jSr)rrrrs r read_matrixzHBFile.read_matrixsTYY 66rcDt||j|jSr)rrr)rrAs r write_matrixzHBFile.write_matrixs1dii77rr) rrrrrrBrCrrhrirrrrrrrsu$&##!!//..,,78rr)spmatrixc8d}t|dr ||}nt|5}||}ddd|turDd}tjj t f}tj|t|d}|r tSS#1swYdxYw)a}Read 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. spmatrix : bool, optional (default: True) If ``True``, return sparse matrix. Otherwise return sparse array. .. deprecated:: 1.18.0 The default value for `spmatrix` is changing to False in v1.20. That means the default return value will be a sparse array. Unless you use * instead of @, ** for matrix power, or you depend on 2D shapes from e.g. ``A.sum(axis=0)``, it may not matter to you. See :ref:`Migration from spmatrix to sparray `. Returns ------- data : csc_array or csc_matrix The data read from the HB file as a sparse array. 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_array, eye >>> data = csr_array(eye(3)) # create a sparse array >>> hb_write("data.hb", data) # write a hb file >>> print(hb_read("data.hb", spmatrix=False)) # read a hb file Coords Values (0, 0) 1.0 (1, 1) 1.0 (2, 2) 1.0 c8t|}|jSr)rr)rkhbs r _get_matrixzhb_read.._get_matrixs C[~~rrNaThe default value for `spmatrix` is changing to `False` in v1.20. That means the default return type will be a sparse array. Unless you use * instead of @, ** for matrix power, or you depend on 2D shapes from e.g. `A.sum(axis=0)` it may not matter to you. See the spmatrix to sparray migration guide for details. https://docs.scipy.org/doc/scipy/reference/sparse.migration_to_sparray.html )skip_file_prefixesT) hasattropenrospathdirname__file__rvrwDeprecationWarningr)path_or_open_filerrr0rrprefixess rr r s`  &),- # $ "q>D "8GGOOH-/ c-(K$ K# " "s BBcjdtjfd}t|dr ||yt |d5}||dddy#1swYyxYw)akWrite 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 array or matrix the sparse array to write hb_info : HBInfo contains the meta-data for write 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_array, eye >>> data = csr_array(eye(3)) # create a sparse array >>> hb_write("data.hb", data) # write a hb file >>> print(hb_read("data.hb", spmatrix=False)) # read a hb file Coords Values (0, 0) 1.0 (1, 1) 1.0 (2, 2) 1.0 Fr!Nc@t|}|jyr)rr)rkrrrAs r _set_matrixzhb_write.._set_matrixHs C ! rrw)r-rrSrr)rrArrrs `` rr r soL UA""1% '*%& #S ) Q N   s  A%%A.r)rrrvnumpyr4scipy._lib.deprecationr scipy.sparserr_fortran_format_parserrrr __all__ Exceptionr Warningrrrrdrrr>rr r rrrr s ( +.MM j ! i  7 7 |!|!~-N(&:9T9Tx,8,8^,4HV3r