build lst_sq_computer 4 and 5

selfdrive/locationd/kalman/SConscript

@@ -6,6 +6,7 @@ ekf_sym = "helpers/ekf_sym.py"
 
 to_build = {
     'pos_computer_4': 'helpers/lst_sq_computer.py',
+    'pos_computer_5': 'helpers/lst_sq_computer.py',
     'feature_handler_5': 'helpers/feature_handler.py',
     'gnss': 'models/gnss_kf.py',
     'loc_4': 'models/loc_kf.py',
@@ -24,7 +25,7 @@ for target, command in found.items():
     command_file = File(command)
     env.Command(target_files,
                 [templates, command_file, sympy_helpers, ekf_sym],
-                command_file.get_abspath()
+                command_file.get_abspath()+" "+target
     )
 
     env.SharedLibrary('generated/' + target, target_files[0])

selfdrive/locationd/kalman/helpers/lst_sq_computer.py

@@ -1,6 +1,6 @@
 #!/usr/bin/env python3
-
 import os
+import sys
 
 import numpy as np
 import sympy as sp
@@ -174,5 +174,6 @@ def project(poses, ecef_pos):
 
 
 if __name__ == "__main__":
-  # TODO: get K from argparse
-  LstSqComputer.generate_code()
+  K = int(sys.argv[1].split("_")[-1])
+  LstSqComputer.generate_code(K=K)
+