a
    ^gg                     @   sL   d dl Z d dlZdd Zdd Zdd Zdd	 Zd
d Zdd Zdd Z	dS )    Nc                 C   sP   t j| dd}tjdkr$|jdd |d }|d|d d d dd f  S )	Nint32dtypebigTZinplacer      )npfromfilesys	byteorderbyteswapreshapecopyfnamead r   c/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/faiss/contrib/vecs_io.py
ivecs_read   s
    
r   c                 C   s   t | dS Nfloat32)r   viewr   r   r   r   
fvecs_read   s    r   c                 C   sF   t jdksJ tj| ddd}|d }|d|d d d dd f S )Nr   r   rr   moder   r   r   )r   r   r	   memmapr   r   r   r   r   
ivecs_mmap   s    r    c                 C   s   t | dS r   )r    r   r   r   r   r   
fvecs_mmap"   s    r!   c                 C   sz   t j| ddd}tjdkrD|d d d d d  }|dd }n|d d dd }|d|d d d dd f S )	NZuint8r   r   r      r   r   r   )r	   r   r   r   r   r   r   )r   xdar   r   r   r   
bvecs_mmap&   s    
r%   c                 C   sh   |j \}}tj||d fdd}||d d df< ||d d dd f< tjdkrZ|jdd ||  d S )Nr   r   r   r   r   Tr   )shaper	   emptyr   r   r   tofile)r   mnr   m1r   r   r   ivecs_write0   s    

r,   c                 C   s   | d}t| |d d S )Nr   r   )Zastyper,   r   )r   r)   r   r   r   fvecs_write:   s    
r-   )
r   numpyr	   r   r   r    r!   r%   r,   r-   r   r   r   r   <module>   s   

