globals [current-color]
breed [individuals individual]
breed [goals goal]
individuals-own [
    pa-se
    ea-se
    vl-se
    sp-se
    pa-se-initial
    ea-se-initial
    vl-se-initial
    sp-se-initial
    efficacy
    efficacy-initial
    chosen-goal
    new-goal
    ;; setting efficacy levels
]
goals-own [
    name
]

to setup
    clear-all
    set-default-shape turtles "circle"
    set current-color 79
    ;; creating the four goals i the space
    create-goals 4 [
        set color blue
        setxy random-xcor random-ycor
        ;; I want to move the goals towards the center, so I make them face the center patch and move towards it a bit
        facexy 0 0
        forward 15
        set name ""
        set label-color white
        ask goal 0 [set name "GOAL 1"]
        ask goal 1 [set name "GOAL 2"]
        ask goal 2 [set name "GOAL 3"]
        ask goal 3 [set name "GOAL 4"]
        set label name
    ]
    reset-ticks
end

to go
    if ticks > 599 [stop]
    initialize-individual
    choose-goal
    layout
    tick
end

to initialize-individual
    create-individuals 1 [
        set color current-color
        ;; Assign variables randomly
        set pa-se random-float 1
        set ea-se random-float 1
        set vl-se random-float 1
        set sp-se random-float 1
        set chosen-goal ["GOAL 1"]
        ;;record the initial value of each component
        set pa-se-initial pa-se
        set ea-se-initial ea-se
        set vl-se-initial vl-se
        set sp-se-initial sp-se
        ;; the efficacy in each individual is the sum of the four components
        set efficacy ((pa-se + ea-se + vl-se + sp-se) / 4)
        set efficacy-initial efficacy
    ]
end

to choose-goal
    ;; set initial goal based on SE
    ask individuals [
        if (efficacy >= 0) and (efficacy <= 0.25) [
            set chosen-goal "GOAL 1"
            move-to goal 0
        ]   
        if (efficacy > 0.25) and (efficacy <= 0.5) [
            set chosen-goal "GOAL 2"
            move-to goal 1
        ]   
        if (efficacy > 0.5) and (efficacy <= 0.75) [
            set chosen-goal "GOAL 3"
            move-to goal 2
        ]  
        if (efficacy > 0.75) and (efficacy <= 1) [
            set chosen-goal "GOAL 4"
            move-to goal 3
        ]
        
        let goal-name chosen-goal 
        ;; update efficacy values 
        ask individuals [
            if chosen-goal != nobody [
                ;; Deduct from ea-se based on the selected goal and pa-se value
                if (chosen-goal = "GOAL 4") and (pa-se < 0.8) [
                    set ea-se (ea-se - (ea-se * (stringencyfactor / 100)))
                ]
                if (chosen-goal = "GOAL 3") and (pa-se < 0.6) [
                    set ea-se (ea-se - (ea-se * (stringencyfactor / 100)))
                ]
                if (chosen-goal ="GOAL 2") and (pa-se < 0.5) [
                    set ea-se (ea-se - (ea-se * (stringencyfactor / 100)))
                ]
                if ea-se < 0 [
                    set ea-se 0
                ]
            ]
        ]
        ;; set a new goal if conditions are met
        ask individuals [
            set new-goal chosen-goal
            if (chosen-goal =  "GOAL 1") and ((ea-se / (stringencyfactor / 100)) > 0.2) [
                set new-goal "GOAL 2"
                move-to one-of goals with [name = "GOAL 2"]
            ]
            if (chosen-goal = "GOAL 2") and ((ea-se / (stringencyfactor / 100)) > 0.2) [
                set new-goal "GOAL 3"
                move-to one-of goals with [name = "GOAL 3"]
            ]
            if (chosen-goal = "GOAL 3") and ((ea-se / (stringencyfactor / 100)) > 0.2) [
                set new-goal "GOAL 4"
                move-to one-of goals with [name = "GOAL 4"]
            ] 
            if new-goal != chosen-goal [
                set current-color current-color - 1
                set chosen-goal new-goal
                ;;to check final goal
            ]
        ]
    ]
end

to layout
    ;; the number 3 here is arbitrary; more repetitions slows down the
    ;; model, but too few gives poor layouts
    repeat 3 [
        ;; the more turtles we have to fit into the same amount of space,
        ;; the smaller the inputs to layout-spring we'll need to use
        let factor sqrt count turtles
        ;; numbers here are arbitrarily chosen for pleasing appearance
        layout-spring turtles links (1 / factor) (7 / factor) (1 / factor)
        display  ;; for smooth animation
    ]
    ;; don't bump the edges of the world
    let x-offset max [xcor] of turtles + min [xcor] of turtles
    let y-offset max [ycor] of turtles + min [ycor] of turtles
    ;; big jumps look funny, so only adjust a little each time
    set x-offset limit-magnitude x-offset 0.1
    set y-offset limit-magnitude y-offset 0.1
    ask goals [
        setxy (xcor - x-offset / 2) (ycor - y-offset / 2)
    ]
end

to-report limit-magnitude [number limit]
    if number > limit [
        report limit
    ]
    if number < (- limit) [
        report (- limit)
    ]
    report number
end