in data/gen_data_collisions.py [0:0]
def check_crossings(self):
for p in range(self.num_particles):
for q in range(p + 1, self.num_particles):
ploc = self.particle_grid_locations[p]
qloc = self.particle_grid_locations[q]
if ploc[0] == qloc[0] and ploc[1] == qloc[1]:
cp = self.particle_colors[p]
cq = self.particle_colors[q]
self.crossing_history[cp, cq, :2] = ploc
self.crossing_history[cp, cq, 2] = self.step
self.most_recent_crossing = (cp, cq)