feat: export headland to CSV

Author: Peter Vaiko

Exporter.lua

@@ -0,0 +1,22 @@
+-- Export the generated course or parts of it
+Exporter = CpObject()
+
+---@param fieldworkCourse CourseGenerator.FieldworkCourse
+function Exporter:init(fieldworkCourse)
+    self.fieldworkCourse = fieldworkCourse
+end
+
+--- Export a headland as a CSV file. Only includes the given headland, without the
+---@param headlandNumber number The headland to export, 1 ist the outermost
+---@param filename string The filename to save the CSV file as, under the export/ directory
+function Exporter:exportHeadlandAsCsv(headlandNumber, filename)
+    local file = io.open('export/' .. filename, 'w')
+    for _, v in self.fieldworkCourse:getHeadlandPath():vertices() do
+        if v:getAttributes():getHeadlandPassNumber() == headlandNumber and
+                not v:getAttributes():isHeadlandTransition() and
+                not v:getAttributes():isOnConnectingPath() then
+            file:write(string.format('%.2f,%.2f\n', v.x, v.y))
+        end
+    end
+    file:close()
+end

export/.gitignore

@@ -0,0 +1 @@
+!.gitignore

include.lua

@@ -9,4 +9,5 @@ package.path = package.path .. ";FS25_Courseplay/scripts/courseGenerator/Genetic
 dofile('FS25_Courseplay/scripts/courseGenerator/test/require.lua')
 require('AdjustableParameter')
 require('ToggleParameter')
-require('ListParameter')
+require('ListParameter')
+require('Exporter')

main.lua

@@ -10,21 +10,20 @@
 --
 --
 dofile('include.lua')
-require('pathfinder.pathfinder')
 
 local logger = Logger('main', Logger.level.debug)
 local parameters = {}
 -- working width of the equipment
-local workingWidth = AdjustableParameter(10, 'width', 'W', 'w', 0.1, 0, 100)
+local workingWidth = AdjustableParameter(5, 'width', 'W', 'w', 0.1, 0, 100)
 table.insert(parameters, workingWidth)
-local turningRadius = AdjustableParameter(7, 'radius', 'T', 't', 0.1, 0, 20)
+local turningRadius = AdjustableParameter(3, 'radius', 'T', 't', 0.1, 0, 20)
 table.insert(parameters, turningRadius)
 local fieldMargin = AdjustableParameter(0, 'margin', 'N', 'n', 0.1, -5, 5)
 table.insert(parameters, fieldMargin)
 -- number of headland passes around the field boundary
 local nHeadlandPasses = AdjustableParameter(1, 'headlands', 'P', 'p', 1, 0, 100)
 table.insert(parameters, nHeadlandPasses)
-local nHeadlandsWithRoundCorners = AdjustableParameter(2, 'headlands with round corners', 'R', 'r', 1, 0, 100)
+local nHeadlandsWithRoundCorners = AdjustableParameter(0, 'headlands with round corners', 'R', 'r', 1, 0, 100)
 table.insert(parameters, nHeadlandsWithRoundCorners)
 local headlandClockwise = ToggleParameter('headlands clockwise', true, 'c')
 table.insert(parameters, headlandClockwise)
@@ -83,7 +82,7 @@ local reverseCourse = ToggleParameter('reverse', false, 'v', true)
 table.insert(parameters, reverseCourse)
 local smallOverlaps = ToggleParameter('small overlaps', false, 'm', true)
 table.insert(parameters, smallOverlaps)
-local nVehicles = AdjustableParameter(2, 'number of vehicles', 'Y', 'y', 1, 1, 5)
+local nVehicles = AdjustableParameter(1, 'number of vehicles', 'Y', 'y', 1, 1, 5)
 table.insert(parameters, nVehicles)
 local useSameTurnWidth = ToggleParameter('use same turn width', false, 'u')
 table.insert(parameters, useSameTurnWidth)
@@ -229,6 +228,9 @@ local function generate()
         love.profiler.reset()
         love.profiler.stop()
     end
+    -- export the first headland as CSV
+    local exporter = Exporter(course)
+    exporter:exportHeadlandAsCsv(1, 'headland-1.csv')
     -- make sure all logs are now visible
     io.stdout:flush()
     errors = context:getErrors()