very poor

treemodel/main.lua

@@ -2,29 +2,36 @@ local screen_width, screen_height = 960, 540
 love.window.setMode(screen_width, screen_height)
 local game_time = os.time()
 love.math.setRandomSeed(game_time)
+game_time = 0 -- TEMP ?
+
+-- local core
+-- local generate_system
+-- generate_system = function(object)
+-- end
+-- core = generate_system()
 
 local star = {
-  x = 0, y = 0, orbital_radius = 0,
+  x = 0, y = 0, orbital_radius = 0, body_radius = 100^0.5,
   children = {
     {
-      orbital_radius = 100,
+      orbital_radius = 100, body_radius = 6^0.5,
       children = {
         {
-          orbital_radius = 10,
+          orbital_radius = 10, body_radius = 1.5^0.5,
         }
       }
     },
     {
-      orbital_radius = 50
+      orbital_radius = 50, body_radius = 3^0.5,
     },
     {
-      orbital_radius = 200,
+      orbital_radius = 200, body_radius = 12^0.5,
       children = {
         {
-          orbital_radius = 25,
+          orbital_radius = 25, body_radius = 3^0.5,
           children = {
             {
-              orbital_radius = 5
+              orbital_radius = 5, body_radius = 1^0.5,
             }
           }
         }
@@ -43,24 +50,87 @@ set_offsets = function(object)
 end
 set_offsets(star)
 
+-- max size star
+star = {
+  x = 0, y = 0, orbital_radius = 0, body_radius = 100^0.5, rotation_offset = 0,
+  children = {
+    {
+      orbital_radius = 950 / 2, body_radius = 10^0.5, rotation_offset = 0,
+    },
+    {
+      orbital_radius = 530 / 2, body_radius = 10^0.5, rotation_offset = math.pi / 2,
+    },
+    {
+      orbital_radius = 30 / 2, body_radius = 1^0.5, rotation_offset = 0,
+    }
+  }
+}
+
+-- min size star
+star = {
+  x = 0, y = 0, orbital_radius = 0, body_radius = 10^0.5, rotation_offset = 0,
+  children = {
+    {
+      orbital_radius = 950 / 2, body_radius = 2^0.5, rotation_offset = 0,
+    },
+    {
+      orbital_radius = 530 / 2, body_radius = 2^0.5, rotation_offset = math.pi / 2,
+    },
+    {
+      orbital_radius = 30 / 2, body_radius = 0.1^0.5, rotation_offset = 0,
+    }
+  }
+}
+
+local generate_bodies
+generate_bodies = function(parent)
+  local object
+  if parent then
+    object = {
+      rotation_offset = math.pi * 2 * love.math.random(),
+      body_radius = love.math.random() * math.sqrt(parent.body_radius / 2) + parent.body_radius * 0.05,
+    }
+    object.orbital_radius = object.body_radius * 1.1 + parent.body_radius * 1.05 + love.math.random() * parent.body_radius^2.5
+  else
+    object = {
+      x = 0, y = 0, orbital_radius = 0, rotation_offset = 0, body_radius = math.sqrt(love.math.random() * 200 + 20),
+    }
+  end
+  object.children = {}
+
+  local potential_child_count = math.floor(object.body_radius)
+  if love.math.random() > 0.5 then return object end
+  if potential_child_count >= 1 then
+    for i = 1, potential_child_count do
+      local child_object = generate_bodies(object)
+      if child_object then
+        table.insert(object.children, child_object)
+      end
+    end
+  end
+
+  return object
+end
+star = generate_bodies()
+
 function love.update(dt)
   game_time = game_time + dt
 
   local update_object_position
-  update_object_position = function(object, parent_x, parent_y)
-    local temporary_speed_up = 100
-    local argument = game_time / object.orbital_radius^1.337 * temporary_speed_up + object.rotation_offset
+  update_object_position = function(object, parent_x, parent_y, parent_mass)
+    local temporary_speed_up = 1
+    local argument = game_time / object.orbital_radius^1.337 * temporary_speed_up * parent_mass + object.rotation_offset
     if (argument == math.huge) or (argument ~= argument) then
       argument = 0
     end
     object.x, object.y = parent_x + object.orbital_radius * math.cos(argument), parent_y + object.orbital_radius * math.sin(argument)
     if object.children then
       for i = 1, #object.children do
-        update_object_position(object.children[i], object.x, object.y)
+        update_object_position(object.children[i], object.x, object.y, object.body_radius^2)
       end
     end
   end
-  update_object_position(star, 0, 0)
+  update_object_position(star, 0, 0, 0)
 end
 
 function love.draw()
@@ -69,8 +139,8 @@ function love.draw()
 
   local draw_object
   draw_object = function(object, parent_x, parent_y)
-    love.graphics.circle("fill", object.x, object.y, 3)
-    love.graphics.circle("line", parent_x, parent_y, object.orbital_radius)
+    love.graphics.circle("fill", object.x, object.y, math.max(object.body_radius, 0.8))
+    love.graphics.circle("line", parent_x, parent_y, object.orbital_radius) -- orbital path
     if object.children then
       for i = 1, #object.children do
         draw_object(object.children[i], object.x, object.y)
@@ -83,5 +153,7 @@ end
 function love.keypressed(key)
   if key == "escape" then
     love.event.quit()
+  elseif key == "r" then
+    star = generate_bodies()
   end
 end