COVID-19 Outbreak and Policies

breed[healthys healthy]
;;healthys-own [color shape]
breed[sicks sick]
sicks-own [recovery]
turtles-own [susceptibility job PPE speed] ;;Probability of contracting the disease

to setup
  clear-all
  reset-ticks
  create-healthys Population [
    setxy random-xcor random-ycor
    set susceptibility random-float 1
    set shape "person"
    set color green
    set breed healthys
    set job "Citizen"
    set heading random 360
    set speed 0.1
  ]
end 

to infect ;; creates an infection
 ;;create-sicks 1 [
 ;;   setxy random-xcor random-ycor
 ;;   set heading random 360
 ;;   set shape "person"
 ;;   set breed sicks
 ;;   set color yellow
 ;;   set job "citizen"
 ;;   set recovery random-float .5
 ;; ]
 ask one-of healthys [
    set breed sicks
    set color yellow
    set shape "person"
    set recovery random-float .5
    set speed 0.1
  ]
end 

to go
  ask turtles [
    let current-turtle self
    forward speed display
    if distance current-turtle < 1 + (count sicks) [
      set heading heading + (random-float 5 - random-float 5)]
  ]
  stay-away
  epidemic
  get-well
  quarantine
  mass-test
  lockdown
  tick
end 

to epidemic
  ask sicks [
    let current-sick self
    ask healthys with [distance current-sick < 1] [
      set breed sicks
      set color yellow
      set shape "person"
      set recovery random-float 1
      set speed 0.1
    ]
  ]
end 

to get-well
  ask sicks [
    set recovery recovery + (random-float .5) - (random-float .5)
    if recovery > .8 [
      set breed healthys
      set shape "person"
      set color green
      set speed 0.1
    ]
  ]
end 

to plot-agents
  plot count sicks
  plot count healthys
end 

to stay-away  ;;Ask healthy people to stay away from sicks who have been identified through mass-testing
  ask sicks with [color = red] [
    let current-sick self
    ask healthys with [distance current-sick < (count sicks with [color = red])] [
      set heading towards current-sick  ;; to move away, first the healthy faces the sick then makes 180 degree turn.
      set heading heading + 180
    ]
  ]
end 

to quarantine ;; Ask confirmed sick people to limit their movement.
ifelse Quarantine-sick?
  [ask sicks with [color = red] [set speed 0.00001]]
  [ask sicks with [color = yellow] [set speed 0.1]]
end 

to mass-test ;; Ask all sicks to disclose their sickness
  ifelse Mass-test?
  [ask sicks [set color red]]
  [ask sicks [set color yellow]]
end 

to lockdown
  ifelse Lockdown?
  [ask turtles [set speed 0]]
  [ask healthys [set speed 0.1]
    ask sicks with [color = red] [ifelse Quarantine-sick?
      [set speed 0.00001]
      [set speed 0.1]]
    ask sicks with [color = yellow] [set speed 0.1]
  ]
end