Ants and Termites

patches-own [
  chemicalTermite      ;; amount of chemical from termite on this patch
  chemicalAnt          ;; amount of chemical from ant on this match
  food                 ;; amount of food on this patch (0, 1, or 2)
  nest?                ;; if it is a nest, whether it is a Termite or ant nest, it will be true
  nestTermite?         ;; true on termite nest patches, false elsewhere
  nestAnt?             ;; true on ant nest patches, false elsewhere
  nest-scent-termite   ;; number that is higher closer to the termite nest
  nest-scent-ant       ;; number that is higher closer to the ant nest
  foodSource?          ;;
]
breed [ants ant]
breed [termites termite]

;;;;;;;;;;;;;;;;;;;;;;;;
;;; Setup procedures ;;;
;;;;;;;;;;;;;;;;;;;;;;;;

to setup
  clear-all
  set-default-shape turtles "bug"
  create-ants population / 2
  [ set size 2
    set color green ;; green = not carryiing food
    setxy 75 -25]

  create-termites population / 2
  [ set size 4
    set color red ;; red = not carryiing food
    setxy 75 25]
  setup-patches
  reset-ticks
end 

to setup-patches
  ask patches
  [ setup-nest
    setup-food  ;; if the statement is : if distancexy 0 0 [blah blah blah], it is from the point of view of the center 0 0
    recolor-patch ]
end 

to setup-nest  ;; patch procedure
  ;; set nest? variable to true inside the nest, false elsewhere
  set nestTermite? false
  set nestAnt? false
  set nest? false
  set foodSource? false
  if (distancexy 75 25) < 5 [set nestTermite? true
                              set nest? true]  ;; for Termites
  if (distancexy 75 -25) < 5 [set nestAnt? true
                             set nest? true]   ;; for Ants
  ;; spread a nest-scent over the whole world -- stronger near the nest
  set nest-scent-termite 200 - distancexy 75 25
  set nest-scent-ant 200 - distancexy 75 -25
end 

to setup-food  ;; patch procedure

  if (distancexy (-0.6 * max-pxcor) (0.8 * max-pycor)) < 5 [set foodSource? true]
  if (distancexy (-0.6 * max-pxcor) (0.4 * max-pycor)) < 5 [set foodSource? true]
  if (distancexy (-0.6 * max-pxcor) (0.0 * max-pycor)) < 5 [set foodSource? true]
  if (distancexy (-0.6 * max-pxcor) (-0.4 * max-pycor)) < 5 [set foodSource? true]
  if (distancexy (-0.6 * max-pxcor) (-0.8 * max-pycor)) < 5 [set foodSource? true]

  ;; set "food" at sources to either 1 or 2, randomly
  if foodSource? = true
  [ set food one-of [1 2 3 4] ]
end 

to set-foodSource
  let x random 195 - 95
  let y random 195 - 95
  if (distancexy x y ) < 5 [set foodSource? true]
end 

to recolor-patch  ;; patch procedure
  ;; give color to nest and food sources
  ifelse nest?
  [ set pcolor violet ]
  [ ifelse food > 0   ;; should be ifelse rather than if, when pcolor scale-color green chemical 0.1 5 is not blocked
    [ if foodSource? [ set pcolor cyan ]
    ]
    ;; scale color to show chemical concentration

    [if chemicalAnt > 0.01 [set pcolor scale-color lime chemicalAnt 0.1 5]
     if chemicalTermite > 0.01 [set pcolor scale-color magenta chemicalTermite  0.1 5]
    ]
  ]
end 

;;;;;;;;;;;;;;;;;;;;;
;;; Go procedures ;;;
;;;;;;;;;;;;;;;;;;;;;

to go  ;; forever button
  ask termites
  [ if who mod 2 = 0 [ stop ] ;; delay initial departure
    if chemicalTermite > 0.2 and chemicalTermite < 0.3 [stop]
    ifelse color = red
    [ look-for-food-termite  ]       ;; not carrying food? look for it
    [ return-to-nestTermite ]       ;; carrying food? take it back to nest
    wiggle
    fd 1 ]
  diffuse chemicalTermite (diffusion-rate / 100)
  ask patches
  [ set chemicalTermite chemicalTermite * (100 - evaporation-rate) / 100  ;; slowly evaporate chemical
    recolor-patch ]

   ask ants
  [ if who mod 2 = 1 [ stop ] ;; delay initial departure
    if chemicalAnt > 0.2 and chemicalAnt < 0.3 [stop]
    ifelse color = green
    [ look-for-food-Ant  ]       ;; not carrying food? look for it
    [ return-to-nestAnt ]       ;; carrying food? take it back to nest
    wiggle
    fd 1 ]
  diffuse chemicalAnt (diffusion-rate / 100)
  ask patches
  [ set chemicalAnt chemicalAnt * (100 - evaporation-rate) / 100  ;; slowly evaporate chemical
    recolor-patch ]
  tick
end 

to return-to-nestTermite  ;; turtle procedure
  ifelse nest?
  [ ;; drop food and head out again
    set color red
    rt 180 ]
  [ set chemicalTermite chemicalTermite + 60  ;; drop some chemical
    uphill-nest-scentTermite ]         ;; head toward the greatest value of nest-scent
end 

to return-to-nestAnt  ;; turtle procedure
  ifelse nest?
  [ ;; drop food and head out again
    set color green
    rt 180 ]
  [ set chemicalAnt chemicalAnt + 60  ;; drop some chemical
    uphill-nest-scentAnt ]         ;; head toward the greatest value of nest-scent
end 

to look-for-food-termite  ;; termite procedure
  if food > 0
  [ set color orange + 1     ;; pick up food
    set food food - 1        ;; and reduce the food source
    rt 180                   ;; and turn around
    stop ]
  ;; go in the direction where the chemical smell is strongest
  if (chemicalTermite >= 0.05) and (chemicalTermite < 2)
  [ uphill-chemicalTermite ]
end 

to look-for-food-ant  ;; ant procedure
   if food > 0
  [ set color blue + 1     ;; pick up food
    set food food - 1        ;; and reduce the food source
    rt 180                   ;; and turn around
    stop ]
  ;; go in the direction where the chemical smell is strongest
  if (chemicalAnt >= 0.05) and (chemicalAnt < 2)
  [ uphill-chemicalAnt ]
end 

;; sniff left and right, and go where the strongest smell is

to uphill-chemicalTermite  ;; Termite procedure
  let scent-ahead chemical-scentTermite-at-angle   0
  let scent-right chemical-scentTermite-at-angle  45
  let scent-left  chemical-scentTermite-at-angle -45
  if (scent-right > scent-ahead) or (scent-left > scent-ahead)
  [ ifelse scent-right > scent-left
    [ rt 45 ]
    [ lt 45 ] ]
end 

;; sniff left and right, and go where the strongest smell is

to uphill-chemicalAnt  ;; Termite procedure
  let scent-ahead chemical-scentAnt-at-angle   0
  let scent-right chemical-scentAnt-at-angle  45
  let scent-left  chemical-scentAnt-at-angle -45
  if (scent-right > scent-ahead) or (scent-left > scent-ahead)
  [ ifelse scent-right > scent-left
    [ rt 45 ]
    [ lt 45 ] ]
end 

;; sniff left and right, and go where the strongest smell is for termite

to uphill-nest-scentTermite  ;; turtle procedure
  let scent-ahead nest-scentTermite-at-angle   0
  let scent-right nest-scentTermite-at-angle  45
  let scent-left  nest-scentTermite-at-angle -45
  if (scent-right > scent-ahead) or (scent-left > scent-ahead)
  [ ifelse scent-right > scent-left
    [ rt 45 ]
    [ lt 45 ] ]
end 

;; sniff left and right, and go where the strongest smell is for ant

to uphill-nest-scentAnt  ;; turtle procedure
  let scent-ahead nest-scentAnt-at-angle   0
  let scent-right nest-scentAnt-at-angle  45
  let scent-left  nest-scentAnt-at-angle -45
  if (scent-right > scent-ahead) or (scent-left > scent-ahead)
  [ ifelse scent-right > scent-left
    [ rt 45 ]
    [ lt 45 ] ]
end 

to wiggle  ;; turtle procedure
  rt random 40
  lt random 40
  if not can-move? 1 [ rt 180 ]
end 

to-report nest-scentTermite-at-angle [angle]
  let p patch-right-and-ahead angle 1
  if p = nobody [ report 0 ]
  report [nest-scent-termite] of p
end 

to-report nest-scentAnt-at-angle [angle]
  let p patch-right-and-ahead angle 1
  if p = nobody [ report 0 ]
  report [nest-scent-Ant] of p
end 

to-report chemical-scentTermite-at-angle [angle]
  let p patch-right-and-ahead angle 1
  if p = nobody [ report 0 ]
  report [chemicalTermite] of p
end 

to-report chemical-scentAnt-at-angle [angle]
  let p patch-right-and-ahead angle 1
  if p = nobody [ report 0 ]
  report [chemicalAnt] of p
end 


; Copyright 1997 Uri Wilensky.
; See Info tab for full copyright and license.