FormulaT - Graphing

globals [
  last-turtle
  velocity-data
  old-vel
  old-acc
]
turtles-own [
  mark
  vel-data
  acc-data
]

to setup
  clear-all

;; calculates for every other column and every other row
;; and then sets their color to light brown

  ask patches with
    [(floor((pxcor + max-pxcor - floor(2)) mod 2) = 0) or  
    (floor((pycor + max-pycor) mod 2) = 0)] [              
    set pcolor grey - 4                                    
    ]

;; creates x and y axis lines

  ask patches with [ pxcor = 0 ] [                   
    set pcolor brown
  ]
  ask patches with [ pycor = 0 ] [                   
    set pcolor brown
  ]

;; sets the labels for every other line to be half their
;; coordinate values on the axis lines

  ask patches with [ (pxcor = 0) and (floor((pycor + max-pycor) mod 2) = 0) ] [   
    set plabel ( pycor / 2)
  ]
  ask patches with [ (pycor = 0) and (floor((pxcor + max-pxcor) mod 2) = 0) ] [
    set plabel ( pxcor / 2)
  ]

;; Creates a hidden turtle that stands in for the starting line data.
;; This is also important so the graph "starts" at zero  

  create-turtles 1 [
    set color yellow
    set shape "circle"
    setxy 0 0
    hide-turtle
  ]
  
  set last-turtle -1  ;; presets last-turtle variable for use in new-point code
end 

;; To make points the user must press the Interact with Graph" button
;; and then click on intersection locations in the view which sprouts a new turtle.
;; Users cannot click on non-intersection locations and may not
;; create a point behind points they've already made.
;; When a new turtle is made, a link is made to the previous turtle
;; and it's x & y data (divided by two) is set as it's mark and
;; velocity and the "rise over run" is calculated as it's acceleration.
;; This data is "set" (so it can be written easily into a file) and then
;; it is written into the output area.
;; Users can also edit points by clicking on previously made turtles and moving
;; them to a new location.

to new-point-check 
  if mouse-down? [
    let active-patch patch mouse-xcor mouse-ycor
    ifelse not any? turtles-on active-patch 
      [new-point]
      [edit-point]
  ]
end 

to new-point
  ask patch mouse-xcor mouse-ycor [
    if not any? turtles-here [
      if (floor((pycor + max-pycor) mod 2) = 0) and (floor((pxcor + max-pxcor) mod 2) = 0) [  ;; click must be in intersection point
        if not any? turtles with [ pxcor >= round (mouse-xcor) ] [ 
          if abs(pycor - [pycor] of turtle (last-turtle + 1)) <= 4 [
            sprout 1 [  ;; make a new turtle at the clicking location
              set color yellow
              set shape "circle"
              set last-turtle who - 1  ;; sets last-turtle to previously made turtle
              set mark (pxcor / 2)     ;; things are divided by two since the graph is twice as big as it needs to be so that it looks better
              set vel-data (pycor / 2)
              set acc-data (precision( (pycor - ([pycor] of turtle (last-turtle))) / (pxcor - ([pxcor] of turtle (last-turtle))) )2)
              create-link-with turtle (last-turtle)  ;; link to last turtle so a nice line graph is made
              output-type "Marker " output-type mark output-type "  "  ;; write data in output area
              output-type "Velocity " output-type vel-data output-type "  "
              output-type "Acceleration " output-print acc-data
            ]
          ]
        ] 
      ]
    ]
  ]
end 

to edit-point
  let candidate min-one-of turtles [distancexy mouse-xcor mouse-ycor]  ;; looks for turtle under mouse
  set old-vel ([vel-data] of candidate)                                ;; saves old vel and acc data into a variable
  set old-acc ([acc-data] of candidate)
  if [distancexy mouse-xcor mouse-ycor] of candidate < 1 [
    watch candidate                                                    ;; halo turtle
    while [mouse-down?] [
      if (floor((round(mouse-ycor) + max-pycor) mod 2) = 0) and        
      (floor((round(mouse-xcor) + max-pxcor) mod 2) = 0) [             ;; turtle can only move to intersection points
        ask subject [
          ifelse turtle (who + 1) = nobody [                           ;; still building graph..
            if abs(mouse-ycor - [ycor] of turtle (who - 1)) <= 4 [     ;; |acceleration| < 2
              set ycor round(mouse-ycor)                               ;; move and update info
              set mark (pxcor / 2)
              set vel-data (pycor / 2)
              set acc-data (precision( (pycor - ([pycor] of turtle (who - 1))) / (pxcor - ([pxcor] of turtle (who - 1))) )2)
            ]
          ] [
            if abs(mouse-ycor - [ycor] of turtle (who + 1)) <= 4 and   ;; same stuff as above, but for editing already made graphs
            abs(mouse-ycor - [ycor] of turtle (who - 1)) <= 4 [
              set ycor round(mouse-ycor)
              set mark (pxcor / 2)
              set vel-data (pycor / 2)
              set acc-data (precision( (pycor - ([pycor] of turtle (who - 1))) / (pxcor - ([pxcor] of turtle (who - 1))) )2)
              ask turtle ([who] of subject + 1) [
                set acc-data (precision( (pycor - ([pycor] of turtle (who - 1))) / (pxcor - ([pxcor] of turtle (who - 1))) )2)
              ]
            ]
          ]
        ] 
      ]
    ]
    ask subject [
      if vel-data != old-vel [                                        ;; checks to make sure you actually changed stuff
        output-type "Marker " output-print mark                       ;; if you did, update output area
        output-type "Velocity changed from " output-type old-vel
        output-type " to " output-print vel-data
        output-type "Acceleration changed from " output-type old-acc
        output-type " to " output-print acc-data
      ]
    ]
    reset-perspective                                                 ;; turns off halo
  ]
end 

;; Saves the turtle data into a text file that can be
;; read by the FormulaT-pitboss model

to save-graph-data
  let file user-new-file

  if ( file != false )
  [
    if file-exists? file [  ;; overwrites files of the same name
      file-delete file
    ]
    file-open file
    ask turtles
    [
      file-write mark
      file-write vel-data
      file-write acc-data
    ]
    file-close
  ]
end 

to save-graph-image
  let file user-new-file

  if ( file != false )
  [
    if file-exists? file [  ;; overwrites files of the same name
      file-delete file
    ]
    export-view file
  ]
end 

;; Loads previously made graph points.  Note: The links and the output
;; area are not recreated, this just recreates the turtles and moves
;; them to the correct spot.

to load-data
  create-turtles 20 [
    set shape "circle"
    set color yellow
    ]
  let file user-file

  if ( file != false )
  [
    set velocity-data []
    file-open file

    while [ not file-at-end? ]
      [ set velocity-data sentence velocity-data (list (list file-read file-read file-read)) ]

    user-message "File loading complete!"
    file-close
  ]
end 

;; Moves the turtles to the correct location outlined in the loaded file.

to train-turtles
  ifelse ( is-list? velocity-data )
    [ 
      foreach velocity-data [ ask turtle first ? [ setxy (first ? * 2) (item 1 ? * 2)
        set mark first ? set vel-data item 1 ? set acc-data last ? ] ] ]
    [ user-message "You need to load in patch data first!" ]
  ask turtles [
    if who != 0 [
      create-link-with turtle (who - 1)
    ]
  ]
end 

to next-race
  run "__magic-open \\"FormulaT-pitboss\\""              ;; should open the next race...
end