From 235123e65b5edb3d5f0c8f14b05e2d2e56f9e40a Mon Sep 17 00:00:00 2001
From: Adrian Kummerlaender
Date: Tue, 24 Mar 2020 20:51:06 +0100
Subject: Separate classes into modules

---
 boltzgas/initial_condition.py | 15 +++++++++++++++
 1 file changed, 15 insertions(+)
 create mode 100644 boltzgas/initial_condition.py

(limited to 'boltzgas/initial_condition.py')

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)
-- 
cgit v1.2.3