gravity visible

src/anybody.js

@@ -429,7 +429,9 @@ export class Anybody extends EventEmitter {
     // )
     // this.bodies = bodies
     // this.missiles = missiles || []
-    this.bodies = this.forceAccumulator(this.bodies)
+    const forceResults = this.forceAccumulator(this.bodies)
+    this.bodies = forceResults.bodies
+    this.forces = forceResults.forces
     var results = this.detectCollision(this.bodies, this.missiles)
     this.bodies = results.bodies
     this.missiles = results.missiles || []

src/calculations.js

@@ -4,9 +4,11 @@
 export const Calculations = {
   forceAccumulator(bodies = this.bodies) {
     bodies = this.convertBodiesToBigInts(bodies)
-    bodies = this.forceAccumulatorBigInts(bodies)
+    const results = this.forceAccumulatorBigInts(bodies)
+    bodies = results.bodies
+    const forces = results.forces
     bodies = this.convertBigIntsToBodies(bodies)
-    return bodies
+    return { bodies, forces }
   },
 
   // async circomStep(bodies, missiles) {
@@ -94,12 +96,14 @@ export const Calculations = {
     for (let i = 0; i < bodies.length; i++) {
       accumulativeForces.push([0n, 0n])
     }
+    const forces = {}
     const time = BigInt(this.speedFactor)
     for (let i = 0; i < bodies.length; i++) {
       const body = bodies[i]
       for (let j = i + 1; j < bodies.length; j++) {
         const otherBody = bodies[j]
         const force = this.calculateForceBigInt(body, otherBody)
+        forces[i + '-' + j] = force
         // const bodyVelocity = [
         //   body.radius == 0n
         //     ? 0n
@@ -185,7 +189,7 @@ export const Calculations = {
         body.position.y = scaledWindowWidth
       }
     }
-    return bodies
+    return { bodies, forces }
   },
 
   calculateBodyFinal() {

src/nft.js

@@ -8,6 +8,7 @@ window.anybody
 q5.setup = () => {
   window.anybody = new Anybody(q5, {
     mode: 'game',
+    level: 2,
     // showLevels: true,
     // target: 'inside',
     // globalStyle: 'psycho',

src/visuals.js

@@ -140,6 +140,10 @@ export const Visuals = {
     //   }
     // }
 
+    if (this.forces) {
+      this.drawForces()
+    }
+
     if (!this.firstFrame) {
       this.drawBodies()
     }
@@ -215,6 +219,29 @@ export const Visuals = {
     }
     this.firstFrame = false
   },
+  drawForces() {
+    const { forces } = this
+    this.p.stroke('rgba(255,255,255,0.2)')
+    this.p.strokeCap(this.p.PROJECT)
+
+    for (const [key, value] of Object.entries(forces)) {
+      // Your code here
+      console.log({ key })
+      console.log({ value })
+      const bodyIndexes = key.split('-')
+      const body1 = this.bodies[bodyIndexes[0]]
+      const body2 = this.bodies[bodyIndexes[1]]
+      const force = Math.abs(parseInt(value[0] + value[1]))
+      console.log({ force })
+      this.p.strokeWeight(force)
+      this.p.line(
+        body1.position.x,
+        body1.position.y,
+        body2.position.x,
+        body2.position.y
+      )
+    }
+  },
   drawPause() {
     if (this.paused) {
       this.p.fill('white')

test/forceAccumulator.test.js

@@ -38,7 +38,8 @@ describe('forceAccumulatorMain circuit', () => {
     )
     // console.dir({ bodies }, { depth: null })
 
-    const out_bodies = anybody
+    const results = anybody
+    const out_bodies = results.bodies
       .forceAccumulatorBigInts(bodies)
       .map(anybody.convertScaledBigIntBodyToArray.bind(anybody))
     // console.log({ out_bodies })

test/nftTest.test.js

@@ -50,7 +50,8 @@ describe.skip('nft circuit', () => {
     )
     // console.dir({ bodies }, { depth: null })
     for (let i = 0; i < steps; i++) {
-      bodies = anybody.forceAccumulatorBigInts(bodies)
+      const results = anybody.forceAccumulatorBigInts(bodies)
+      bodies = results.bodies
     }
     const out_bodies = bodies.map(
       anybody.convertScaledBigIntBodyToArray.bind(anybody)