From 3c1383a6fd509621752e5ca8063cfaeb1ec09f9b Mon Sep 17 00:00:00 2001
From: kiritow <1362050620@qq.com>
Date: Tue, 17 Oct 2017 22:12:08 +0800
Subject: [PATCH] Update Station 2/3 Program

---
 station_2_3.lua | 98 ++++++++++++++++++++++++++++++++++++++++++++++---
 1 file changed, 92 insertions(+), 6 deletions(-)

diff --git a/station_2_3.lua b/station_2_3.lua
index abb43fd..0df007f 100644
--- a/station_2_3.lua
+++ b/station_2_3.lua
@@ -9,8 +9,8 @@ local event=require("event")
 Redstone Signals
 Direction_IO_Description : number
 ]]
-local ab_in_OutsideRoute,ab_in_EnterSensor,ab_in_ExitSensor
-local ba_in_OutsideRoute,ba_in_EnterSensor,ba_in_ExitSensor
+local ab_in_OutsideRoute,ab_in_OutsideSensor,ab_in_EnterSensor,ab_in_ExitSensor
+local ba_in_OutsideRoute,ba_in_OutsideSensor,ba_in_EnterSensor,ba_in_ExitSensor
 local mid_in_lamp
 
 local ab_out_Outside,ab_out_Motor,ab_out_Stop=0,0,0
@@ -35,21 +35,55 @@ end
 Network
 Schedule program must connect to MC.TrainNet to send and recv train status.
 Config your connect-card here
+NOTICE: You must give different value to station_id (>=0)
 ]]
 local trainnet=proxy("tunnel")
+local station_id=0
+local station_a_id=-1
+local station_b_id=-1
 
 -- Internal Schedule Status (Notice: Program must start without any trains in station)
 local ab_station_free=true
 local ba_station_free=true
 local mid_free=true
+local mid_direction="ab"
 local ab_next_free=true
 local ba_next_free=true
 
--- Internal Functions
+--[[
+    Internal Functions
+]]
+
+local netpack_head=string.pack("iii",6,1,1)
+
 local function doInit()
-    -- TODO
-    event.listen("redstone_change")
-    event.listen("modem_message")
+    -- TODO: You must change this function to fix your station
+    event.listen("redstone_change",
+        function(_event,_addr,_side,_old,_new)
+            -- [NOTICE] Change information here!!
+            -- Here: your must send an arrive message to last station
+            
+        end
+    )
+    event.listen("modem_message",
+        function(_event,_recevier,_sender,_port,_distance,...)
+            local apack={...}
+            -- netpack_head FromStationID ToStationID Message(1: Leave 2: Arrive)
+            if(apack[1]==netpack_head) then
+                if(apack[3]==station_id) then -- Train will come to this station
+                    -- Just do nothing (We don't worry about it)
+                elseif(apack[2]==station_id) then -- Information about train left from this station
+                    if(apack[4]==2) then -- Train arrives information
+                        if(apack[3]==station_a_id) then -- Arrive at A
+                            ba_next_free=true
+                        elseif(apack[3]==station_b_id) then -- Arrive at B
+                            ab_next_free=true
+                        end
+                    end -- We don't care about leave information
+                end
+            end
+        end
+    )
 
     -- Flash output to zero.
     updateRedstoneOutput()
@@ -58,4 +92,56 @@ end
 -- Main Program
 local function main()
     doInit()
+    while true do
+        -- Flush input
+        updateRedstoneInput()
+        -- Update status
+
+        -- Judge Incoming bus.
+        if(ab_in_OutsideSensor>0) then -- New incoming bus from A to B
+            if(ab_in_OutsideRoute>0) then -- This bus want to stop
+                if(ab_station_free) then -- If AB Station is free
+                    ab_station_free=false -- Mark station busy.
+                    ab_out_Outside=15 -- enable to allow incoming
+                    ab_out_Motor=0 -- disable to allow incoming to station
+                else -- AB Station is not free
+                    -- This train should wait outside the station
+                end
+            else -- This bus want to pass by
+                if(mid_free) then -- Mid is free can pass
+                    mid_free=false -- Mark mid busy.
+                    ab_out_Outside=15 -- enable to allow incoming
+                    ab_out_Motor=15 -- enable motor to let it pass.
+                else -- Mid is busy
+                    -- This train should wait outside the station
+                end
+            end
+        end
+
+        if(ba_in_OutsideSensor>0) then -- New incoming bus from B to A
+            if(ba_in_OutsideRoute>0) then -- This bus want to stop
+                if(ba_station_free) then -- If BA Station is free
+                    ba_station_free=false -- Mark station busy.
+                    ba_out_Outside=15 -- enable to allow incoming
+                    ba_out_Motor=0 -- disable to allow incoming to station
+                else -- BA Station is not free
+                    -- This train should wait outside the station
+                end
+            else -- This bus want to pass by
+                if(mid_free) then -- Mid is free can pass
+                    mid_free=false -- Mark mid busy.
+                    ba_out_Outside=15 -- enable to allow incoming
+                    ba_out_Motor=15 -- enable motor to let it pass.
+                else -- Mid is busy
+                    -- This train should wait outside the station
+                end
+            end
+        end
+
+        -- Set output
+        updateRedstoneOutput()
+
+        -- Sleep for next loop
+        os.sleep(0.5)
+    end
 end
\ No newline at end of file