Priante/boids_algorithm_app
0
1B
2 �(�cU � @ sP d Z ddlZddlmZmZ ddlZdd� Zdd� Zdd � Z d3d � Z4d d
� Z dS )z95Created on Sat Aug 27 17:53:41 20226 7@author: francesca8� N)�pdist�9squareformc C s� t tt�| |g���}||k}t�|�}t�|�}t�|�} t�|�}10xztd|d �D ]h} t�| || f �|| < t�||| f �|| < | | || | | | < || || | |11| < qVW | |12fS )Nr � )r r �np� transpose�zeros�range�mean) �x0�y0�neighbors_dist�n�R�m�idxZcenter_xZcenter_y�vx�vy�i� r �E/Users/francesca/Desktop/coding_shit/boids_algorithm_app/functions.py� moveToCenter s 13141516 r c 17 C s� t tt�| |g���}||k |dk@ }t�t�t�|���}t�|�}t�|�} | |d d �df | |d d �df | ||d d �df < ||d d �df ||d d �df | | |d d �df < || fS )Nr r )r r r r �array�wherer )18r19 r r
�privacy�bird_repulsion�distZidxmatr r r r r r � avoidOthers+ s 2021<<r c C s� t tt�| |g���}||k}t�|�} t�|�}22xbtd|d �D ]P} t�||| f | || f �| | < t�||| f ||| f �|23| < qBW | | |24| fS )Nr r )r r r r r r r ) �x_prev�y_prevr25 r r
r �match_velocityr r Zvmean_xZvmean_yr r r r �matchVelocities@ s 2627&*r c C s� t | ||||�\}}t| ||||�\}}t||| |||| �\}}| | | | | }|| | | | }t�d|f�}t�||g�}tjj|dd�}|d d �|dkf ||dk |
|d d �|dkf< |dd d �f }|dd d �f }|28| }| |29 }|30| }||31 } ||| k | ||| k < ||| k | ||| k < ||| k | ||| k < || | k | || | k < | | } || } t�| �} t�| �} | | fS )N� r )�axisr ) r r r r r r �linalg�norm�round) r32 r r r r
r r r r r �L�margin�wall_repulsion� max_speedZvx1Zvy1Zvx2Zvy2Zvx3Zvy3r r Zv_normZv_vectorr$ Zright_border_distZleft_border_distZupper_border_distZbottom_border_dist�x1�y1r r r �moveV s0 43334r, c C s< | |g}t jt �|�| dd�t jj|dd� }t �|�}||dk ||dk < |dk|tjk @ } t �| �d } d| | dk< |}35| tjd | | } t � |�} |||k | ||k< |||k | ||k |36||k | ||k< t �37| �t jj|dd� }
t � | �t jj|dd� }t � |
| d�}t � || d�}||fS )Nr )r" r �����r! � )
r �sumr r# r$ �arccos�math�pi�asarrayr �cos�sinr% )Zvv_xZvv_yZ
perpendicular�dr' r
�vvZ scalar_prodZalpha0Zincrease_angleZ intercept�slope� new_angleZnew_vvxZnew_vvyZdev_vvxZdev_vvyr r r �deviate_from_walls� s" (38 39,r: ) �__doc__�numpyr Zscipy.spatial.distancer r r1 r r r r, r: r r r r �<module> s E