Three Body Problem

{"name":"Three Body Problem","description":"Try to achieve stability in a three-body orbit.","photo":"https://jippity.s3.amazonaws.com/Screenshot 2024-05-30 at 11.40.51 AM.png","view_count":0,"fork_count":0,"expires_at":null,"created_at":"2024-10-01T13:58:06.424296Z","updated_at":"2024-10-01T13:58:06.424298Z","user_id":5,"status":"PRIVATE","parent_id":1685,"id":3414,"user":{"photo":null,"email":null,"recess_id":"b5574579-4904-b775-ad8e-eff5c085f819","created_at":"2023-12-12T12:00:51.087995Z","reset_token":null,"stripe_customer_id":null,"seats":null,"username":"theop","id":5,"password_hash":null,"updated_at":"2023-12-12T12:00:51.087997Z","reset_token_expiry":null,"subscription_status":null},"code":{"script.js":"let bodies = [];\nlet stars = []; // Array to store star positions\nlet stabilityRating = 100; // Initial stability rating\nlet distanceFluctuations = []; // To store distance changes\nlet initialEnergy = 0; // Initial total energy of the system\nlet initialAngularMomentum = 0; // Initial total angular momentum of the system\nlet timeSteps = 0; // Counter to track simulation steps\nlet stabilityRatings = []; // Array to store recent stability ratings\nconst MAX_RATINGS = 50;\nlet panX = 0;\nlet panY = 0;\n\n// UI elements\nlet inputs = [];\nlet simulationStarted = false;\n\nfunction setup() {\n createCanvas(800, 600).position(0, 100);\n\n // Generate star positions\n for (let i = 0; i < 200; i++) {\n stars.push({ x: random(width), y: random(height) });\n }\n\n // Initial positions and velocities for more circular orbits\n let initialConditions = [\n { x: 300, y: 300, mass: 20, vx: 0, vy: 1.5 },\n { x: 500, y: 300, mass: 10, vx: 0, vy: -1.5 },\n { x: 400, y: 400, mass: 15, vx: 1.5, vy: 0 }\n ];\n\n // Create input elements for initial conditions with labels\n let initialConditionsDiv = select('#initial-conditions');\n for (let i = 0; i < initialConditions.length; i++) {\n let body = initialConditions[i];\n initialConditionsDiv.child(createP(`BODY ${i + 1}`).style('color', '#fff')); // Label for each body\n let xInput = createInput(body.x.toString()).size(50).parent(initialConditionsDiv);\n let yInput = createInput(body.y.toString()).size(50).parent(initialConditionsDiv);\n let massInput = createInput(body.mass.toString()).size(50).parent(initialConditionsDiv);\n let vxInput = createInput(body.vx.toString()).size(50).parent(initialConditionsDiv);\n let vyInput = createInput(body.vy.toString()).size(50).parent(initialConditionsDiv);\n inputs.push({ xInput, yInput, massInput, vxInput, vyInput });\n }\n\n select('#start-simulation').mousePressed(startSimulation);\n\n let toggleButton = select('#toggle-panel');\n toggleButton.mousePressed(() => {\n let panel = select('#initial-conditions');\n if (panel.style('display') === 'none') {\n panel.style('display', 'block');\n toggleButton.html('Hide Initial Conditions');\n } else {\n panel.style('display', 'none');\n toggleButton.html('Show Initial Conditions');\n }\n });\n\n background(0); // Dark background for a space-like appearance\n}\n\nfunction startSimulation() {\n panX = 0;\n panY = 0;\n simulationStarted = true;\n\n // Clear previous bodies and trails\n bodies = [];\n\n for (let input of inputs) {\n let x = float(input.xInput.value());\n let y = float(input.yInput.value());\n let mass = float(input.massInput.value());\n let vx = float(input.vxInput.value());\n let vy = float(input.vyInput.value());\n bodies.push({ x, y, mass, vx, vy, trail: [] });\n }\n\n // Calculate initial energy and angular momentum\n initialEnergy = calculateTotalEnergy();\n initialAngularMomentum = calculateTotalAngularMomentum();\n\n timeSteps = 0; // Reset time steps\n distanceFluctuations = []; // Reset distance fluctuations\n loop(); // Restart the draw loop\n\n // Hide the initial conditions panel but keep the control panel visible\n select('#initial-conditions').style('display', 'none');\n select('#toggle-panel').html('Show Initial Conditions');\n}\n\nfunction draw() {\n // Clear the background\n background(0);\n\n // Draw the starry background\n drawStars();\n\n if (!simulationStarted) {\n displayInitialPositions();\n } else {\n // Handle panning with WASD keys\n handlePan();\n\n // Calculate center of mass and translate canvas\n let centerOfMass = calculateCenterOfMass();\n translate(width / 2 - centerOfMass.x + panX, height / 2 - centerOfMass.y + panY);\n\n // Update bodies and draw everything\n updateBodies();\n drawGravitationalFields();\n drawBodies();\n drawTrails();\n calculateStability();\n updateStabilityDisplay();\n\n // Increment time steps\n timeSteps++;\n }\n}\n\nfunction handlePan() {\n let speed = keyIsDown(SHIFT) ? 50 : 5; // Set speed to 50 if Shift is held, otherwise 5\n\n if (keyIsDown(87)) { // W key\n panY += speed;\n }\n if (keyIsDown(83)) { // S key\n panY -= speed;\n }\n if (keyIsDown(65)) { // A key\n panX += speed;\n }\n if (keyIsDown(68)) { // D key\n panX -= speed;\n }\n}\n\nfunction drawStars() {\n noStroke();\n fill(255); // White color for stars\n for (let star of stars) {\n ellipse(star.x, star.y, 2, 2); // Draw each star as a small circle\n }\n}\n\nfunction displayInitialPositions() {\n noStroke();\n for (let input of inputs) {\n let x = float(input.xInput.value());\n let y = float(input.yInput.value());\n let mass = float(input.massInput.value());\n let vx = float(input.vxInput.value());\n let vy = float(input.vyInput.value());\n\n // Draw celestial body\n fill(255, 204, 0); // Yellow color\n ellipse(x, y, mass * 2, mass * 2); // Draw bodies as circles proportional to their mass\n\n // Draw velocity vector as an arrow\n stroke(255, 0, 0); // Red color for velocity vector\n line(x, y, x + vx * 20, y + vy * 20); // Arrow shaft\n drawArrowhead(x, y, vx, vy); // Arrowhead\n }\n}\n\nfunction drawArrowhead(x, y, vx, vy) {\n let angle = atan2(vy, vx);\n let arrowSize = 7;\n fill(255, 0, 0); // Red color for arrowhead\n push();\n translate(x + vx * 20, y + vy * 20);\n rotate(angle);\n beginShape();\n vertex(0, 0);\n vertex(-arrowSize, arrowSize / 2);\n vertex(-arrowSize, -arrowSize / 2);\n endShape(CLOSE);\n pop();\n}\n\nfunction drawBodies() {\n noStroke();\n for (let body of bodies) {\n fill(255, 204, 0); // Yellow color for celestial bodies\n ellipse(body.x, body.y, body.mass * 2, body.mass * 2); // Draw bodies as circles proportional to their mass\n }\n}\n\nfunction drawGravitationalFields() {\n stroke(0, 255, 0, 100); // Green color with transparency for gravitational field lines\n for (let i = 0; i < bodies.length; i++) {\n for (let j = i + 1; j < bodies.length; j++) {\n let bodyA = bodies[i];\n let bodyB = bodies[j];\n\n // Draw a line between bodyA and bodyB\n line(bodyA.x, bodyA.y, bodyB.x, bodyB.y);\n }\n }\n}\n\nfunction drawTrails() {\n stroke(255, 255, 255, 100); // White color with transparency for trails\n noFill();\n for (let body of bodies) {\n beginShape();\n for (let pos of body.trail) {\n vertex(pos.x, pos.y);\n }\n endShape();\n }\n}\n\nfunction updateBodies() {\n let G = 18; // Gravitational constant for simplicity\n let speedMultiplier = keyIsDown(SHIFT) ? 10 : 1; // Speed multiplier\n\n for (let i = 0; i < bodies.length; i++) {\n let ax = 0;\n let ay = 0;\n\n for (let j = 0; j < bodies.length; j++) {\n if (i != j) {\n let bodyA = bodies[i];\n let bodyB = bodies[j];\n\n let dx = bodyB.x - bodyA.x;\n let dy = bodyB.y - bodyA.y;\n let distance = sqrt(dx * dx + dy * dy);\n let force = (G * bodyA.mass * bodyB.mass) / (distance * distance);\n let angle = atan2(dy, dx);\n\n ax += cos(angle) * force / bodyA.mass;\n ay += sin(angle) * force / bodyA.mass;\n }\n }\n\n bodies[i].vx += ax * speedMultiplier;\n bodies[i].vy += ay * speedMultiplier;\n }\n\n for (let body of bodies) {\n body.x += body.vx * speedMultiplier;\n body.y += body.vy * speedMultiplier;\n\n // Update the trail\n body.trail.push({ x: body.x, y: body.y });\n if (body.trail.length > 100) { // Limit trail length to 100 points\n body.trail.shift();\n }\n }\n}\n\nfunction calculateCenterOfMass() {\n let totalMass = 0;\n let centerX = 0;\n let centerY = 0;\n\n for (let body of bodies) {\n totalMass += body.mass;\n centerX += body.x * body.mass;\n centerY += body.y * body.mass;\n }\n\n return { x: centerX / totalMass, y: centerY / totalMass };\n}\n\nfunction calculateTotalEnergy() {\n let totalEnergy = 0;\n let G = 18; // Gravitational constant\n\n for (let i = 0; i < bodies.length; i++) {\n let kineticEnergy = 0.5 * bodies[i].mass * (bodies[i].vx * bodies[i].vx + bodies[i].vy * bodies[i].vy);\n let potentialEnergy = 0;\n\n for (let j = 0; j < bodies.length; j++) {\n if (i != j) {\n let dx = bodies[j].x - bodies[i].x;\n let dy = bodies[j].y - bodies[i].y;\n let distance = sqrt(dx * dx + dy * dy);\n potentialEnergy -= (G * bodies[i].mass * bodies[j].mass) / distance;\n }\n }\n\n totalEnergy += kineticEnergy + potentialEnergy / 2; // Potential energy is shared between pairs\n }\n\n return totalEnergy;\n}\n\nfunction calculateTotalAngularMomentum() {\n let totalAngularMomentum = 0;\n\n for (let body of bodies) {\n totalAngularMomentum += body.mass * (body.x * body.vy - body.y * body.vx);\n }\n\n return totalAngularMomentum;\n}\n\nfunction calculateStability() {\n let currentEnergy = calculateTotalEnergy();\n let currentAngularMomentum = calculateTotalAngularMomentum();\n\n // Calculate energy and angular momentum differences\n let energyDifference = abs(currentEnergy - initialEnergy);\n let angularMomentumDifference = abs(currentAngularMomentum - initialAngularMomentum);\n\n // Calculate distance fluctuations and check for overlaps\n let distanceFluctuation = 0;\n let overlapDetected = false;\n for (let i = 0; i < bodies.length; i++) {\n for (let j = i + 1; j < bodies.length; j++) {\n let bodyA = bodies[i];\n let bodyB = bodies[j];\n\n let dx = bodyB.x - bodyA.x;\n let dy = bodyB.y - bodyA.y;\n let distance = sqrt(dx * dx + dy * dy);\n\n // Check for overlap\n if (distance < (bodyA.mass + bodyB.mass)) {\n overlapDetected = true;\n }\n\n if (!distanceFluctuations[i]) distanceFluctuations[i] = [];\n if (!distanceFluctuations[i][j]) distanceFluctuations[i][j] = [];\n\n distanceFluctuations[i][j].push(distance);\n if (distanceFluctuations[i][j].length > 100) { // Limit number of stored distances\n distanceFluctuations[i][j].shift();\n }\n\n let maxDistance = Math.max(...distanceFluctuations[i][j]);\n let minDistance = Math.min(...distanceFluctuations[i][j]);\n distanceFluctuation += maxDistance - minDistance;\n }\n }\n\n // Calculate stability rating based on energy, angular momentum, distance fluctuations, and overlap\n let stabilityRating = 0;\n if (!overlapDetected) {\n stabilityRating = max(0, 100 - (energyDifference * 0.1 + angularMomentumDifference * 0.1 + distanceFluctuation * 0.1));\n }\n\n // Add the current stability rating to the array\n stabilityRatings.push(stabilityRating);\n if (stabilityRatings.length > MAX_RATINGS) {\n stabilityRatings.shift(); // Remove the oldest rating if we exceed the max number\n }\n}\n\nfunction updateStabilityDisplay() {\n // Calculate the moving average of the stability ratings\n let averageStability = stabilityRatings.reduce((sum, rating) => sum + rating, 0) / stabilityRatings.length;\n select('#stability').html(averageStability.toFixed(2));\n}","style.css":"body {\n margin: 0;\n display: flex;\n flex-direction: column;\n align-items: center;\n justify-content: flex-start;\n height: 100vh;\n background-color: #000;\n /* Black background to match canvas */\n}\n\ncanvas {\n border: 1px solid #fff;\n /* White border for the canvas */\n}\n\np {\n color: #fff;\n /* White text color */\n margin: 5px;\n}\n\n#control-panel {\n position: absolute;\n top: 10px;\n left: 10px;\n background-color: rgba(0, 0, 0, 0.8);\n padding: 10px;\n border: 1px solid #fff;\n z-index: 10;\n /* Ensures the control panel is rendered above the canvas */\n}\n\n#stability-panel {\n position: absolute;\n top: 10px;\n left: 350px;\n background-color: rgba(0, 0, 0, 0.8);\n padding: 10px;\n border: 1px solid #fff;\n z-index: 10;\n /* Ensures the stability panel is rendered above the canvas */\n}\n\nbutton {\n margin: 5px;\n}\n\n#toggle-panel {\n display: block;\n margin-bottom: 10px;\n}\n\n#initial-conditions input {\n margin: 5px;\n}\n\n#instructions-box {\n position: absolute;\n top: 10px;\n left: 500px;\n /* Adjust as necessary to position correctly */\n background-color: rgba(0, 0, 0, 0.8);\n padding: 10px;\n border: 1px solid #fff;\n z-index: 10;\n /* Ensures the instructions box is rendered above the canvas */\n color: #fff;\n /* White text color */\n}","index.html":"\n\n\n\n \n \n Three Body Problem\n \n\n\n\n

\n \n

Stability Rating:

100

WASD to pan around

\n \n \n\n\n"},"chat_history":[{"project_id":3414,"sender":"human","message":"Let's get started!","deleted":0,"created_at":"2024-10-01T13:58:08.967323Z","output_tokens":null,"code_id":158961,"input_tokens":null,"id":77669},{"project_id":3414,"sender":"ai","message":"","deleted":0,"created_at":"2024-10-01T13:58:08.984807Z","output_tokens":null,"code_id":158961,"input_tokens":null,"id":77670},{"project_id":3414,"sender":"human","message":"lets make the shift key multiply speedd by 10 while its held","deleted":0,"created_at":"2024-10-01T13:58:24.818718Z","output_tokens":null,"code_id":158962,"input_tokens":null,"id":77671},{"project_id":3414,"sender":"ai","message":"

\n# Reasoning\nThe user has requested a feature to multiply the panning speed by 10 when the Shift key is held. This is feasible within the scope of the project and can be implemented by modifying the `handlePan` function.\n

\n\n

\n# Answer: \n

\n\nWe can definitely add this feature. \n\nLet's update the `handlePan` function to accommodate this change.\n\n```javascript\nfunction handlePan() {\n let speed = keyIsDown(SHIFT) ? 50 : 5; // Set speed to 50 if Shift is held, otherwise 5\n\n if (keyIsDown(87)) { // W key\n panY += speed;\n }\n if (keyIsDown(83)) { // S key\n panY -= speed;\n }\n if (keyIsDown(65)) { // A key\n panX += speed;\n }\n if (keyIsDown(68)) { // D key\n panX -= speed;\n }\n}\n```\n\n- The `speed` variable is set to 50 if the Shift key is held, otherwise it is set to 5.\n- The panning values for `panY` and `panX` are adjusted based on the `speed` variable.\n\nWould you like to add any other features or modifications?","deleted":0,"created_at":"2024-10-01T13:58:24.825538Z","output_tokens":null,"code_id":158962,"input_tokens":null,"id":77672},{"project_id":3414,"sender":"human","message":"NOT PAN I MEAN THE SIMULATION","deleted":0,"created_at":"2024-10-01T13:59:27.351477Z","output_tokens":null,"code_id":158964,"input_tokens":null,"id":77673},{"project_id":3414,"sender":"ai","message":"Got it! Let's modify the simulation speed instead.\n\nWe'll adjust the velocity of the bodies based on whether the Shift key is held or not. To do this, we'll modify the `updateBodies` function.\n\nLet's add a check to see if the Shift key is pressed and adjust the velocities accordingly.\n\n```javascript\nfunction updateBodies() {\n let G = 18; // Gravitational constant for simplicity\n let speedMultiplier = keyIsDown(SHIFT) ? 10 : 1; // Speed multiplier\n\n for (let i = 0; i < bodies.length; i++) {\n let ax = 0;\n let ay = 0;\n\n for (let j = 0; j < bodies.length; j++) {\n if (i != j) {\n let bodyA = bodies[i];\n let bodyB = bodies[j];\n\n let dx = bodyB.x - bodyA.x;\n let dy = bodyB.y - bodyA.y;\n let distance = sqrt(dx * dx + dy * dy);\n let force = (G * bodyA.mass * bodyB.mass) / (distance * distance);\n let angle = atan2(dy, dx);\n\n ax += cos(angle) * force / bodyA.mass;\n ay += sin(angle) * force / bodyA.mass;\n }\n }\n\n bodies[i].vx += ax * speedMultiplier;\n bodies[i].vy += ay * speedMultiplier;\n }\n\n for (let body of bodies) {\n body.x += body.vx * speedMultiplier;\n body.y += body.vy * speedMultiplier;\n\n // Update the trail\n body.trail.push({ x: body.x, y: body.y });\n if (body.trail.length > 100) { // Limit trail length to 100 points\n body.trail.shift();\n }\n }\n}\n```\n\n- The `speedMultiplier` variable is set to 10 if the Shift key is held, otherwise it is set to 1.\n- The accelerations and positions of the bodies are adjusted based on the `speedMultiplier`.\n\nWould you like to make any other changes or add more features?","deleted":0,"created_at":"2024-10-01T13:59:27.356347Z","output_tokens":null,"code_id":158964,"input_tokens":null,"id":77674}],"assets":[],"messages_sent":3,"comments":[],"steps":[{"instruction":"Add visual lines to represent the gravitational fields between the celestial bodies.","title":"Visualize Gravitational Fields","created_at":"2024-05-30T15:58:34.626383","project_id":3414,"complete":false,"step_number":0,"updated_at":"2024-10-01T13:58:06.428607","id":10726},{"instruction":"Allow users to add and interact with celestial bodies in the simulation.","title":"Interactive Celestial Bodies","created_at":"2024-05-30T15:58:34.626455","project_id":3414,"complete":false,"step_number":1,"updated_at":"2024-10-01T13:58:06.428675","id":10727},{"instruction":"Enable users to customize parameters such as mass, initial velocity, and gravitational constant.","title":"Customizable Parameters","created_at":"2024-05-30T15:58:34.626547","project_id":3414,"complete":true,"step_number":3,"updated_at":"2024-10-01T13:58:06.428717","id":10728},{"instruction":"Show relevant information about the celestial bodies and their interactions, such as velocities, accelerations, and forces.","title":"Information Display","created_at":"2024-05-30T15:58:34.626591","project_id":3414,"complete":false,"step_number":4,"updated_at":"2024-10-01T13:58:06.428757","id":10729},{"instruction":"Provide visual representations of the data generated during the simulation, such as graphs showing the trajectories and energies of the bodies.","title":"Data Visualization","created_at":"2024-05-30T15:58:34.626636","project_id":3414,"complete":false,"step_number":5,"updated_at":"2024-10-01T13:58:06.428795","id":10730},{"instruction":"Include links to educational materials about gravitational physics and celestial mechanics to enhance the learning experience.","title":"Educational Resources","created_at":"2024-05-30T15:58:34.626687","project_id":3414,"complete":false,"step_number":6,"updated_at":"2024-10-01T13:58:06.428833","id":10731}]}