^MhDdZddgZddlmZddlmZddlmZd dZd Z d S) z+ Functions that operate on sparse matrices count_blocksestimate_blocksize)issparse) csr_array)csr_count_blocksffffff?ct|r |jdvst|}|jdkrdSd|cxkrdksnt dd|zdz }t |j}|j\}}|dzdkr |dzdkr|dt|d zz }nd }|d zdkr |d zdkr|d t|d zz }nd }||kr&||kr |dt|dzz }||krdSd S|dzdkr |dzdkr|dt|dzz } nd } | |krdS||krd S||krd SdS)zAttempt to determine the blocksize of a sparse matrix Returns a blocksize=(r,c) such that - A.nnz / A.tobsr( (r,c) ).nnz > efficiency )csccsr)rrg?z.efficiency must satisfy 0.0 < efficiency < 1.0g@)r r g )rr$)r)rr)rformatrnnz ValueErrorfloatshaper) A efficiencyhigh_efficiencyrMNe22e33e66e44s U/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/scipy/sparse/_spfuncs.pyrr s QKKAH66 aLLuzzu z    C    IJJJZ'3.O ,,C 'CAa1uzza!eqjjQa.../1uzza!eqjjQa.../ _!6!6R,q///0   55 q5A::!a%1**l1U3334CCC   5 :  5 :  55cD|\}}|dks|dkrtdt|rU|jdkr(|j\}}t |||||j|jS|jdkrt|j||fStt||S)z]For a given blocksize=(r,c) count the number of occupied blocks in a sparse matrix A rzr and c must be positiver r ) rrrrrindptrindicesrTr)r blocksizercrrs r"rr>s CAa1uuA3444{{+ 8u  'CAa#Aa!AHQY?? ? X  QqE** *  ! Y / //r#N)r) __doc____all___baser_csrr _sparsetoolsrrrr#r"r1s . /******0000f00000r#