diff options
Diffstat (limited to 'boltzgas/initial_condition.py')
-rw-r--r-- | boltzgas/initial_condition.py | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/boltzgas/initial_condition.py b/boltzgas/initial_condition.py new file mode 100644 index 0000000..4d66263 --- /dev/null +++ b/boltzgas/initial_condition.py @@ -0,0 +1,15 @@ +import numpy as np + +def grid_of_random_velocity_particles(width, radius, u_scale): + np_position = np.ndarray((width**2, 2)) + np_velocity = np.ndarray((width**2, 2)) + + grid = np.meshgrid(np.linspace(2*radius, 1-2*radius, width), + np.linspace(2*radius, 1-2*radius, width)) + np_position[:,0] = grid[0].flatten() + np_position[:,1] = grid[1].flatten() + + np_velocity[:,0] = u_scale*(-0.5 + np.random.random_sample((width**2,))) + np_velocity[:,1] = u_scale*(-0.5 + np.random.random_sample((width**2,))) + + return (np_position, np_velocity) |