Get streamlines by building trajectory set.
(self)
| 291 | self.trajectories.append(t) |
| 292 | |
| 293 | def get_streamlines(self): |
| 294 | """ |
| 295 | Get streamlines by building trajectory set. |
| 296 | """ |
| 297 | for indent in range(self.density // 2): |
| 298 | for xi in range(self.density - 2 * indent): |
| 299 | self.traj(xi + indent, indent) |
| 300 | self.traj(xi + indent, self.density - 1 - indent) |
| 301 | self.traj(indent, xi + indent) |
| 302 | self.traj(self.density - 1 - indent, xi + indent) |
| 303 | |
| 304 | self.st_x = [ |
| 305 | np.array(t[0]) * self.delta_x + self.x[0] for t in self.trajectories |
| 306 | ] |
| 307 | self.st_y = [ |
| 308 | np.array(t[1]) * self.delta_y + self.y[0] for t in self.trajectories |
| 309 | ] |
| 310 | |
| 311 | for index in range(len(self.st_x)): |
| 312 | self.st_x[index] = self.st_x[index].tolist() |
| 313 | self.st_x[index].append(np.nan) |
| 314 | |
| 315 | for index in range(len(self.st_y)): |
| 316 | self.st_y[index] = self.st_y[index].tolist() |
| 317 | self.st_y[index].append(np.nan) |
| 318 | |
| 319 | def get_streamline_arrows(self): |
| 320 | """ |