update_ini: copy values from TRAJ to DISPLAY

Author: hansu

src/emc/ini/update_ini.py

@@ -28,6 +28,17 @@ def writeifexists(file, section, src_item, dest_item = "None"):
     if dest_item == 'None': dest_item = src_item
     val = ini.find(section, src_item)
     if val: file.write("%s = %s\n" % (dest_item, val))
+    
+def insert_after(key, new_line, section):
+    match = re.search(key, section)
+    if match:
+        # Find the end of the line that contains the match
+        line_end = section.find("\n", match.end())
+        if line_end == -1: line_end = len(section) # match is on the last line
+    else:
+        line_end = len(section) # place it at the end
+        new_line += "\n"
+    return section[:line_end + 1] + new_line + section[line_end + 1:]
 
 force = 0
 dialogs = 0
@@ -452,6 +463,22 @@ def ini_preamble():
             section = re.sub("MAX_SPINDLE_SPEED", "MAX_SPINDLE_0_SPEED", section)
         if re.search("MIN_VELOCITY", section):
             section = re.sub("MIN_VELOCITY", "MIN_LINEAR_VELOCITY", section)
+        
+        
+        # Copy values from TRAJ
+        if not re.search("DEFAULT_LINEAR_VELOCITY", section):
+            val = ini.find("TRAJ", "DEFAULT_LINEAR_VELOCITY")
+            if val:
+                section = insert_after("MAX_LINEAR_VELOCITY", f"DEFAULT_LINEAR_VELOCITY = {val}\n" , section)
+        if not re.search("MIN_LINEAR_VELOCITY", section):
+            val = ini.find("TRAJ", "MIN_LINEAR_VELOCITY")
+            if val:
+                section = insert_after("MAX_LINEAR_VELOCITY", f"MIN_LINEAR_VELOCITY = {val}\n" , section)
+        if not re.search("MAX_LINEAR_VELOCITY", section):
+            val = ini.find("TRAJ", "MAX_LINEAR_VELOCITY")
+            if val:
+                section = insert_after("MIN_LINEAR_VELOCITY", f"MAX_LINEAR_VELOCITY = {val}\n" , section)
+
         newini.write(section)
 
     # TODO update-ini 1.1 --> 1.2:
@@ -465,7 +492,7 @@ def ini_preamble():
     #
     # copy [TRAJ]DEFAULT_LINEAR_VELOCITY -> [DISPLAY]DEFAULT_LINEAR_VELOCITY
     # move [TRAJ]MIN_LINEAR_VELOCITY -> [DISPLAY]MIN_LINEAR_VELOCITY
-    # rename [TRAJ, DISPLAY]MIN_VELOCITY --> MIN_LINEAR_VELOCITY
+    # ~rename [TRAJ, DISPLAY]MIN_VELOCITY --> MIN_LINEAR_VELOCITY~
     # copy [TRAJ]MAX_LINEAR_VELOCITY -> [DISPLAY]MAX_LINEAR_VELOCITY
 
     #These sections don't need any work.