Adding particle on mouse click

gravis3d/body.py

@@ -6,6 +6,7 @@ class Body():
     '''
     A class representing all the bodies
     in the simulation.
+    contants are added
     '''
     def __init__(self, radius = None, pos = None, velocity = None, colour = None,density = None):
         self.radius = radius or 6.9634*1e9

gravis3d/generator.py

@@ -17,10 +17,17 @@ def __init__(self, N = None):
             #newbody = Body()
             self.bodies.append(newbody)
 
+    def add_particle(self,position):
+        r = abs(normal(1e9,1e8,1))
+        vx,vy,vz = normal(0,1e7,1), normal(0,1e7,1), normal(0,1e7,1)
+        add = Body(radius = r[0], pos = vector(position), velocity = vector(vx[0],vy[0],vz[0]))
+        self.bodies.append(add)    
+
+
     def update(self):
         for i in range(self.N):
             for j in range(self.N):
                 if i!=j:
                     self.bodies[i].attract(self.bodies[j])
                     #self.bodies[i].move()
-
+

gravis3d/play.py

@@ -1,7 +1,8 @@
 from generator import Nbodies
 from time import sleep
-from vpython import sphere, vector, rate, scene, color,mag2
 from constants import R, wait, no_of_bodies
+from vpython import *
+
 
 scene.caption = """
 To rotate "camera", drag with right button or Ctrl-drag.
@@ -14,4 +15,9 @@
 sleep(wait)
 while(1):
     rate = R
+    def Get_pos(evt):
+        loc = evt.pos
+        universe.add_particle(loc)
+
+    scene.bind('click', Get_pos)
     universe.update()