
    gg                     B    d dl Z d dlZ	 d Zd Zd Zd Zd Zd Zd Z	dS )	    Nc                     t          j        | d          }t          j        dk    r|                    d           |d         }|                    d|dz             d d dd f                                         S )	Nint32dtypebigTinplacer      )npfromfilesys	byteorderbyteswapreshapecopyfnameads      S/var/www/html/mpstechhub/venv/lib/python3.11/site-packages/faiss/contrib/vecs_io.py
ivecs_readr      su    
E)))A
}	

4
   	!A99RQ122&++---    c                 F    t          |                               d          S Nfloat32)r   viewr   s    r   
fvecs_readr          e!!),,,r   c                     t           j        dk    sJ t          j        | dd          }|d         }|                    d|dz             d d dd f         S )Nr   r   rr   moder   r
   r   )r   r   r   memmapr   r   s      r   
ivecs_mmapr&      s]    =E!!!!
	%wS111A	!A99RQ122&&r   c                 F    t          |                               d          S r   )r&   r   r   s    r   
fvecs_mmapr(   "   r    r   c                 b   t          j        | dd          }t          j        dk    rA|d d         d d d                                         }|                    d          d         }n#|d d                             d          d         }|                    d|dz             d d dd f         S )	Nuint8r"   r#   r      r
   r   r   )r   r%   r   r   r   r   r   )r   xdar   s       r   
bvecs_mmapr.   &   s    
	%wS111A
}rrU44R4[GGGQbqbEJJw"99RQ122&&r   c                     |j         \  }}t          j        ||dz   fd          }||d d df<   ||d d dd f<   t          j        dk    r|                    d           |                    |            d S )Nr   r   r   r   r   Tr   )shaper   emptyr   r   r   tofile)r   mnr   m1s        r   ivecs_writer6   0   s    7DAq	1a!e*G	,	,	,BBqqq!tHBqqq!""uI
}
D!!!IIer   c                 v    |                     d          }t          | |                    d                     d S )Nr   r   )astyper6   r   )r   r3   s     r   fvecs_writer9   :   s3    	Aqvvg'''''r   )
r   numpyr   r   r   r&   r(   r.   r6   r9    r   r   <module>r<      s    


    . . .- - -' ' '- - -' ' '  ( ( ( ( (r   