Infectious Disease Outbreak-Variants

;;
;; This model is developed by Dr. Lin Xiang at the University of Kentucky. Contact: lin.xiang@uky.edu
;;
;; If you see this page rather than a "download" button when downloading the model, click the "download" icon
;; in your browser to download this model file.
;;
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;


turtles-own [ days-after-exposure  days-of-sick labeled]
Patches-own [ ]
Globals [ watching  max-daily-cases-a max-daily-cases-b death-a death-b]

to setup
  ca
  setup-turtles

   set watching false
  set death-a 0
  set death-b 0
  plot-levels
  reset-ticks
end 

to setup-turtles
  create-turtles population-size
  [set color 68
    set size 1.25
    set labeled 0
    set shape "person"
    setxy random-xcor random-ycor
  ]
end 

to go

  if ticks >= days [stop]
  move
  transmission-A
  transmission-B
  incubation-A
  incubation-B
  sickness-A
  sickness-B
  find-max-daily-cases-A
  find-max-daily-cases-B
  plot-levels
  tick
end 

to move
  ask turtles with [shape = "person"]
  [right random 360 forward mobility]
end 

to add-a-carrier-A
  ask one-of turtles                     ;choose a random person
  [set color orange                      ;set the color as orange (variant-A carrier color)
  ]
end 

to add-a-carrier-B
  ask one-of turtles                     ;choose a random person
  [set color brown                      ;set the color as brown (variant-B carrier  color)
  ]
end 

to transmission-A
  ask turtles with [color = orange]                                              ;Hi, carriers
  [let susceptible-person one-of other turtles in-radius 1.5 with [color = 68]   ;Are there any susceptible people near you (within a radius of 1.5)?
    if susceptible-person != nobody                                              ;If there is at least one
    [ask susceptible-person                                                      ;ask the suscetiple person
      [ if random 100 < VA-transmission-rate                                        ;at the defined transmission rate
        [set color orange]                                                       ;to become a carrier (orange color)
      ]
    ]
  ]
end 

to transmission-B
  ask turtles with [color = brown]                                              ;Hi, carriers
  [let susceptible-person one-of other turtles in-radius 1.5 with [color = 68]   ;Are there any susceptible people near you (within a radius of 1.5)?
    if susceptible-person != nobody                                              ;If there is at least one
    [ask susceptible-person                                                      ;ask the suscetiple person
      [ if random 100 < VB-transmission-rate                                        ;at the defined transmission rate
        [set color brown]                                                       ;to become a carrier (orange color)
      ]
    ]
  ]
end 

to incubation-A
  ask turtles with [color = orange]                    ;Hi carriers
   [ifelse days-after-exposure < VA-incubation-period     ;If the number of days after you get infected is smaller than the defined incubation period
    [set days-after-exposure days-after-exposure + 1]  ;increase the number of days after exposure by 1
    [set color red set shape "house" ]                   ;otherwise, become infectious (red color)
   ]
end 

to incubation-B
  ask turtles with [color = brown]                    ;Hi carriers
   [ifelse days-after-exposure < VB-incubation-period     ;If the number of days after you get infected is smaller than the defined incubation period
    [set days-after-exposure days-after-exposure + 1]  ;increase the number of days after exposure by 1
      [set color magenta set shape "house"          ]         ;otherwise, become infectious (red color)
   ]
end 

to sickness-A
  ask turtles with [color = red]          ;Hi sick people,
  [ifelse days-of-sick <= VA-illness-period             ;if you have not been sick for 15 days
    [set days-of-sick days-of-sick + 1]   ;remain sick and increase the number of sick days by 1
    [ifelse random 100 < VA-mortality        ;otherwise, at the define mortality rate
      [set death-a death-a + 1 die]                               ;die
      [set color blue set shape "person"] ;or recover
    ]
   ]
end 

to sickness-B
  ask turtles with [color = magenta]          ;Hi sick people,
  [ifelse days-of-sick <= VB-illness-period             ;if you have not been sick for 15 days
    [set days-of-sick days-of-sick + 1]   ;remain sick and increase the number of sick days by 1
    [ifelse random 100 < VB-mortality        ;otherwise, at the define mortality rate
      [set death-b death-b + 1 die]                               ;die
      [set color sky set shape "person"] ;or recover
    ]
   ]
end 

to find-max-daily-cases-A
  if count turtles with [color = red ] > max-daily-cases-a        ;Count the infectious.If it is greater than the current record of max daily cases
  [set max-daily-cases-a count turtles with [color = red ]]       ;update the max daily case
end 

to find-max-daily-cases-B
  if count turtles with [color = magenta ] > max-daily-cases-b        ;Count the infectious.If it is greater than the current record of max daily cases
  [set max-daily-cases-b count turtles with [color = magenta ]]       ;update the max daily case
end 

to watch-an-infected-person
  watch one-of turtles with [color = red]
end 

to label-a-person

  if any? turtles = false [user-message "Nobody is here." stop]


  if mouse-inside? [
    if mouse-down? [
      let labeled-person min-one-of turtles [distancexy mouse-xcor mouse-ycor]
      if labeled-person != nobody [
        ask labeled-person [set label "me" set labeled 1]      ;label the 1st selected person as me
        watch labeled-person]
  ]]


 if count turtles with [labeled = 1] >= 1 [stop]   ;disable the button when one person is labelled.
end 

to unselect

  ask turtles
  [set labeled 0
    set label ""]
  rp
end 

to plot-levels ;; this creates creates the bar graph
    set-current-plot "Max daily cases and deaths"
    clear-plot
    plot-pen-down
    set-current-plot-pen "VA max-daily-cases"
    plotxy 1 max-daily-cases-a
    set-current-plot-pen "VB max-daily-cases"
    plotxy 2 max-daily-cases-b
    set-current-plot-pen "VA Deaths"
    plotxy 4 death-a
    set-current-plot-pen "VB Deaths"
    plotxy 5 death-b
end