BML Model with rerouting

globals [
  average-speed   ; average speed of all cars
  actual-density
  prop-orange
]

turtles-own [
  move-flag       ; auxiliary variable to track if a car should move
  speed           ; speed of the car
]

to setup
  clear-all
  set-default-shape turtles "default"  ; use the default arrow shape
  ;ask patches [  set pcolor gray]

  ; Set up the plot
  ;set-current-plot "Average Speed"
  ;set-plot-x-range 0 100  ; Adjust x-axis range as needed
  ;set-plot-y-range 0 1   ; Adjust y-axis range as needed

  ; Place cars randomly on the grid and assign a speed
  ask patches [
    if random-float 1 < density [ ; approximately 30% of the cells will have cars
      sprout 1 [
        ifelse random-float 1 < 0.5 [
          set heading 90   ; East (right)
          set color cyan
        ] [
          set heading 180  ; South (down)
          set color orange
        ]
        set speed 1  ; Speed can be adjusted if needed
        set move-flag false
      ]
    ]
  ]

  ; Initialize global variables
  set average-speed 1
  set actual-density count turtles / count patches
  set prop-orange 100 * count turtles with [ color = orange ] / count turtles

  reset-ticks
end 

to go
  ;; 180 Move Stage 1: Identify which cars should move
  ask turtles with [ heading = 180] [
    let next-patch patch-ahead 1  ; Get the patch directly ahead based on heading
    ifelse not any? turtles-on next-patch [  set move-flag true    ] [
      set move-flag false
      set speed 0  ; Cars that cannot move have their speed set to 0
    ]
  ]
  ;; Move Stage 2: Perform the moves based on `move-flag`
  ask turtles with [move-flag and heading = 180] [
    set speed 1  ; Cars that can move have their speed set to 1
    move-to patch-ahead 1
  ]
 ;; Re-route
  if prob_reroute > 0 [
    ask turtles with [not move-flag and heading = 180] [
      if random-float 1 < prob_reroute [
        set heading 90
        set color cyan
      ]
    ]
  ]

  ;; 90 Move Stage 1: Identify which cars should move
  ask turtles with [ heading = 90] [
    let next-patch patch-ahead 1  ; Get the patch directly ahead based on heading
    ifelse not any? turtles-on next-patch [  set move-flag true    ] [
      set move-flag false
      set speed 0  ; Cars that cannot move have their speed set to 0
    ]
  ]
  ;; Move Stage 2: Perform the moves based on `move-flag`
  ask turtles with [move-flag  and heading = 90] [
    set speed 1  ; Cars that can move have their speed set to 1
    move-to patch-ahead 1
  ]
 ;; Re-route
  if prob_reroute > 0 [
   ask turtles with [not move-flag and heading = 90] [
    if random-float 1 < prob_reroute [
      set heading 180
      set color orange
    ]
  ]]

  ; Calculate the average speed
  set average-speed mean [speed] of turtles
  ;if average-speed = 0 [ stop ]
  set prop-orange 100 * count turtles with [ color = orange ] / count turtles

  ; Clear the auxiliary variable for the next iteration
  ;ask turtles [ set move-flag false ]

  tick
end 

; Public Domain:
; To the extent possible under law, Jorge Laval has waived all
; copyright and related or neighboring rights to this code.