Whale Shark Population

globals [ max-whaleshark artificial-plankton artificial-plankton-threshold]

  ; whaleshark and fishingboats are both breeds of turtle.
breed [ whaleshark one-whaleshark ]
breed [ fishingboats fishingboat ]
breed [ touristboats touristboat ]

turtles-own [ energy      ; both fishingboats and whaleshark have energy
              eaten-artificial-plankton ; Added for whalesharks to indicate if they have eaten artificial plankton
              artificial-plankton-timer; Counter for ticks since eating artificial plankton
              time-on-artificial-plankton  ; New variable to track time on artificial plankton patch
              artificial-plankton-consumed  ; New variable to track amount of artificial plankton consumed
              ticks-since-last-reproduction
]

patches-own [ countdown ]

to setup
  clear-all
  ifelse netlogo-web? [set max-whaleshark 10000] [set max-whaleshark 20000]

  set artificial-plankton-threshold 50  ; Example value

  ask patches [
      set pcolor one-of [ cyan sky ]
      ifelse pcolor = cyan
        [ set countdown plankton-regrowth-time ]
      [ set countdown random plankton-regrowth-time ] ; initialize plankton regrowth clocks randomly for sky patches
    set artificial-plankton false  ; Initially, no patch has artificial plankton
    ]

  create-whaleshark initial-number-whaleshark  ; create the whaleshark, then initialize their variables
  [
    set shape  "whale shark"
    set color gray
    set size 2  ; easier to see
    set label-color sky - 2
    set energy random (2 * whaleshark-gain-from-food)
    setxy random-xcor random-ycor
    set eaten-artificial-plankton false
    set artificial-plankton-timer -1  ; Initialize with -1 to indicate not consumed artificial plankton
    set time-on-artificial-plankton 0
    set artificial-plankton-consumed 0
    set ticks-since-last-reproduction 50  ; Start with 50 so they can reproduce immediately
  ]

  create-fishingboats initial-number-fishingboats  ; create the fishingboats, then initialize their variables
  [
    set shape "boat"
    set color orange
    set size 3  ; easier to see
    set energy random (2 * fishingboat-gain-from-food)
    setxy random-xcor random-ycor
  ]

    ; Create a single touristboat only if model-version is "artificial plankton"
  if model-version = "artificial-plankton" [
    create-touristboats 1 [
      set shape "boat"
      set color red  ; or any color that distinguishes it
      set size 5
      setxy random-xcor random-ycor
    ]
  ]

  display-labels
  reset-ticks
end 

to go
  ; stop the simulation if no fishingboats or whaleshark
  if not any? turtles [ stop ]
  ; stop the model if there are no fishingboats and the number of whaleshark gets very large
  if count whaleshark > max-whaleshark [ user-message "The whale sharks have inherited the ocean." stop ]
  if not any? whaleshark [ user-message "Oh no! The whale sharks are extinct!" stop ]

  if count whaleshark > whaleshark-threshold and count fishingboats < max-fishingboats [
    create-fishingboats 1 [ initialize-fishingboat ]
    ]

   ; Whaleshark behavior
  ask whaleshark [
    if pcolor = yellow [  ; Check if whaleshark is on an artificial plankton patch
      set time-on-artificial-plankton time-on-artificial-plankton + 1
    ] if pcolor != yellow [
      set time-on-artificial-plankton 0  ; Reset the counter if not on artificial plankton
    ]
    move
    set energy energy - 1
    eat-plankton
    death
    set ticks-since-last-reproduction ticks-since-last-reproduction + 1  ; Increment reproduction timer
    reproduce-whaleshark
    flee-from-fishingboats
    move-towards-touristboats
    set eaten-artificial-plankton false  ; Reset at the end of each tick

    ]

  ask fishingboats [
    chase-whaleshark ;; fishingboats chase whalesharks
    eat-whaleshark ; fishingboats eat a whaleshark on their patch
    check-energy-and-leave ; fishingboats leave if out of energy
  ]

  ask touristboats [
    dispense-artificial-plankton  ; Ensure this is being called
    ; Other touristboat behaviors
  ]

  ; always grow plankton
  ask patches [ grow-plankton ]

  update-color-after-artificial-plankton  ; Include this call in your simulation loop

  tick
  display-labels
end 

to initialize-fishingboat  ; Initialize new fishingboat
  set shape "boat"
  set color orange
  set size 3  ; easier to see
  setxy random-xcor random-ycor
  set energy 10  ; Initialize energy to 0
end 

to dispense-artificial-plankton
  ask patches in-radius 2 [  ; Assuming a radius around the tourist boat
    set artificial-plankton true
    set pcolor yellow
  ]
end 

to chase-whaleshark
  let target one-of whaleshark in-cone 10 180  ; 10 is the radius, 180 is the angle
  if target != nobody [
    face target  ; Turn towards the whaleshark
    fd 1  ; Move forward towards the whaleshark
  ] if target = nobody [
    move  ; Call a procedure for random movement
  ]
  if energy <= 0 [ die ]  ; Check energy level
end 

to check-energy-and-leave  ; Check energy level of fishingboats
  if energy >= min-energy-to-leave [ die ]  ; Leave if minimum energy is reached
end 

to move  ; turtle procedure
  rt random 50
  lt random 50
  fd 1
end 

to eat-plankton  ; whaleshark procedure
  ; whaleshark eat plankton, turn the patch sky
  if pcolor = cyan [
    set pcolor sky
    set energy energy + whaleshark-gain-from-food  ; whaleshark gain energy by eating
  ]
  if pcolor = yellow [
    set energy energy + (whaleshark-gain-from-food * 0.5)
    set artificial-plankton-consumed artificial-plankton-consumed + 1
    set color yellow
    set pcolor sky
    set artificial-plankton-timer 0  ; Start the timer
  ]
end 

to reproduce-whaleshark
  let reproduction-chance whaleshark-reproduce
  if model-version = "artificial-plankton" [
  if artificial-plankton-timer >= 0 and artificial-plankton-timer < artificial-plankton-interval [
    set reproduction-chance reproduction-chance * 0.5  ; Halve the chance if within the effect window
  ]
]
  if ticks-since-last-reproduction >= reproduction-interval [  ; Check if it's time to reproduce
    if random-float 1 < reproduction-chance [  ; Chance to reproduce
     set energy (energy / 2)
      hatch 1 [
        set ticks-since-last-reproduction 0  ; Reset timer for offspring
        rt random-float 360
        fd 1
      ]
      set ticks-since-last-reproduction 0  ; Reset timer for parent
    ]
  ]
end 

to eat-whaleshark  ; fishingboat procedure
  let prey one-of whaleshark-here                    ; grab a random whaleshark
  if prey != nobody  [                          ; did we get one?  if so,
    ask prey [ die ]                            ; kill it, and...
    set energy energy + fishingboat-gain-from-food     ; get energy from eating
  ]
end 

to death  ; turtle procedure (i.e. both fishingboat nd whaleshark procedure)
  ; when energy dips below zero, die
  if energy < 0 [ die ]
end 

to grow-plankton  ; patch procedure
  ; countdown on sky patches: if reach 0, grow some plankton
  if pcolor = sky [
    ifelse countdown <= 0
      [ set pcolor cyan
        set countdown plankton-regrowth-time ]
      [ set countdown countdown - 1 ]
  ]
end 

to-report plankton
    report patches with [pcolor = cyan]
end 

to update-color-after-artificial-plankton
  ask whaleshark [
    if artificial-plankton-timer >= 0 [  ; Check if the timer is active
      if artificial-plankton-timer < artificial-plankton-interval [
        set artificial-plankton-timer artificial-plankton-timer + 1
      ]
      if artificial-plankton-timer = artificial-plankton-interval [
        set color gray  ; Revert color to gray
        set eaten-artificial-plankton false  ; Reset the effect on reproduction
        set artificial-plankton-timer -1  ; Reset the timer
      ]
    ]
  ]
end 

to flee-from-fishingboats
  let nearest-fishingboat min-one-of fishingboats [distance myself]
  if nearest-fishingboat != nobody [
    if (artificial-plankton-timer >= 0) and (artificial-plankton-timer < artificial-plankton-interval) [  ; Check if under the influence of artificial plankton
      face nearest-fishingboat  ; Move towards the fishing boat
      fd 1
    ]
    if (artificial-plankton-timer < 0) or (artificial-plankton-timer >= artificial-plankton-interval) [
       if random-float 1 < 0.2 [  ; flee-probability is the chance of fleeing
      face nearest-fishingboat
      rt 180
      fd 1
      ]
    ]
  ]
end 

to move-towards-touristboats
  if artificial-plankton-consumed < artificial-plankton-threshold [  ; Only move towards if under the threshold
    let nearest-touristboat min-one-of touristboats [distance myself]
    if nearest-touristboat != nobody and distance nearest-touristboat < attraction-distance [
      face nearest-touristboat
      fd 1
    ]
  ]
end 

to display-labels
  ask turtles [ set label "" ]
  if show-energy? [
    ask fishingboats [ set label round energy ]
    ask whaleshark [ set label round energy ]
  ]
end