Fix ThrottleInit (2)

main.lua

@@ -117,7 +117,24 @@ Config.Autopilot = {
     AutopilotDesiredHeading = nil
 }
 
+Throttles = {
+    ForeThrottle = {
+        name = nil,
+        side = nil
+    },
+    AftThrottle = {
+        name = nil,
+        side = nil
+    },
+    DownThrottle = {
+        name = nil,
+        side = nil
+    }
+}
+
 function ThrottleInit() 
+    Throttles = {}
+
     local sides = {
         "top",
         "bottom",
@@ -171,6 +188,46 @@ function ThrottleInit()
             end
         end
     end
+
+    print("\nIdentifying aft throttle")
+    print("\nPlease change the input signal for the throttle")
+
+    local initialStates = getRelayStates()
+
+    while true do
+        local currentStates = getRelayStates()
+
+        for pname, currentState in pairs(currentStates) do
+            local initialState = initialStates[pname]
+            for k, v in pairs(currentState) do
+                if initialState[k] ~= v then
+                    Throttles.AftThrottleThrottle.name = pname
+                    Throttles.AftThrottle.side = k
+                    return
+                end
+            end
+        end
+    end
+
+    print("\nIdentifying down throttle")
+    print("\nPlease change the input signal for the throttle")
+
+    local initialStates = getRelayStates()
+
+    while true do
+        local currentStates = getRelayStates()
+
+        for pname, currentState in pairs(currentStates) do
+            local initialState = initialStates[pname]
+            for k, v in pairs(currentState) do
+                if initialState[k] ~= v then
+                    Throttles.DownThrottle.name = pname
+                    Throttles.DownThrottle.side = k
+                    return
+                end
+            end
+        end
+    end
 end
 
 function PropellerInit()
@@ -679,35 +736,6 @@ function collectSensorData()
     return data
 end
 
-function Main()
-    Init()
-
-    while true do
-        local sensorData = collectSensorData()
-        
-        -- Update monitor displays
-        if Config.Monitors.InstrumentPanelMonitor then
-            displayInstrumentPanel(Config.Monitors.InstrumentPanelMonitor, sensorData)
-        end
-        
-        if Config.Monitors.AutopilotControlMonitor then
-            displayAutopilotControls(Config.Monitors.AutopilotControlMonitor, sensorData)
-        end
-
-        local unindexedThrusters = checkIfThrusterIsIndexed()
-        if unindexedThrusters ~= nil then
-            partiallyUpdateThrusters(unindexedThrusters)
-        end
-
-        PollSensors()
-        UpdateGlobalThrust()
-        UpdateStabilization()
-        
-        -- Add a small delay to avoid overloading the system
-        os.sleep(0.1)
-    end
-end
-
 function displayAutopilotControls(monitor)
     if not monitor or not monitor.peripheral then return end
     
@@ -759,4 +787,4 @@ function Main()
 end
 
 -- Run the main function when the script starts
-Main()
+Main()