Starting gravity 1 body around a body. Not yet finished.

Added todo list for project. Started 2 body problem also.

gravity_simulation/gravity.py

@@ -0,0 +1,25 @@
+import numpy as np
+import matplotlib.pyplot as plt
+
+time = 0
+
+
+class Body:
+
+    def __init__(self, position, mass=1):
+        # Initial parameters
+        self.mass = mass
+        self.position = np.array(position)
+
+
+def plot_t(positions):
+    fig, ax = plt.subplots()
+    ax.scatter(positions[:, 0], positions[:, 1])
+    plt.show()
+
+
+if __name__ == "__main__":
+    body1 = Body([1, 1])
+    body2 = Body([-1, -1])
+    positions = np.vstack([body1.position, body2.position])
+    plot_t(positions)

gravity_simulation/gravity1body.py

@@ -0,0 +1,20 @@
+import numpy as np
+import matplotlib.pyplot as plt
+
+time = 0
+# Initial conditions
+mass_b1 = 1
+position
+
+
+def plot_t(positions):
+    fig, ax = plt.subplots()
+    ax.scatter(positions[:, 0], positions[:, 1])
+    plt.show()
+
+
+if __name__ == "__main__":
+    body1 = Body([1, 1])
+    body2 = Body([-1, -1])
+    positions = np.vstack([body1.position, body2.position])
+    plot_t(positions)

gravity_simulation/todo.md

@@ -0,0 +1,16 @@
+# Lista con los objetivos del proyecto
+## Descripcion
+El objetivo del proyecto es de hacer un programa de simulacion gravitacional de N-cuerpos en 2D y 3D.
+
+## TO-DO
+- [ ] Simulacion en 2D
+  - [ ] Simulacion de 2 cuerpos
+  - [ ] Visualizacion de cuerpos (matplotlib)
+  - [ ] Simulacion de 3 cuerpos
+  - [ ] Simulacion de 10 cuerpos 
+  - [ ] Simulacion de 100 cuerpos
+  - [ ] Simulacion de 1000 cuerpos
+  - [ ] Simulacion de 1 000 000 cuerpos (possible con mi cpu?)
+- [ ] Simulacion en 3D
+- [ ] GUI para elegir 2D/3D, numero de cuerpos
+- [ ] Agregar configuraciones conocidas (sistema solar, sol tierra y luna)