Function collision_matrix

code/OO/collision_matrix.py:35–45

Use broadcasting to return a square matrix of Boolean values with True at position [i,j] and [j,i] if circles i and j intersect.

(x, y, R)

import time

def collision_matrix(x, y, R):
    """
    Use broadcasting to return a square matrix of Boolean
    values with True at position [i,j] and [j,i] if
    circles i and j intersect.
    """
    dx = x[:,np.newaxis] - x[np.newaxis,:]
    dy = y[:,np.newaxis] - y[np.newaxis,:]
    sep = np.sqrt( dx**2 + dy**2 )
    sum_r = R[:,np.newaxis] + R[np.newaxis,:]
    return sep < sum_r

def run_sim(N, n_iter):
  box_x, box_y = 10., 6.