14
14
import argparse
15
15
import subprocess
16
16
import time
17
+ import shutil
17
18
import string
18
19
import pathlib
19
20
@@ -50,10 +51,18 @@ def vehicleinfo_key(self):
50
51
return "ArduCopter"
51
52
raise ValueError ("Can't determine vehicleinfo_key from binary path" )
52
53
54
+ def model (self ):
55
+ path = self .binary .lower ()
56
+ if "plane" in path :
57
+ return "quadplane"
58
+ if "copter" in path :
59
+ return "X"
60
+ raise ValueError ("Can't determine vehicleinfo_key from binary path" )
61
+
53
62
def run (self ):
54
63
self .start_SITL (
55
64
binary = self .binary ,
56
- model = "novehicle" ,
65
+ model = self . model () ,
57
66
wipe = True ,
58
67
sitl_home = "1,1,1,1" ,
59
68
)
@@ -90,10 +99,18 @@ def vehicleinfo_key(self):
90
99
return "ArduCopter"
91
100
raise ValueError ("Can't determine vehicleinfo_key from binary path" )
92
101
102
+ def model (self ):
103
+ path = self .binary .lower ()
104
+ if "plane" in path :
105
+ return "quadplane"
106
+ if "copter" in path :
107
+ return "X"
108
+ raise ValueError ("Can't determine vehicleinfo_key from binary path" )
109
+
93
110
def run (self ):
94
111
self .start_SITL (
95
112
binary = self .binary ,
96
- model = "novehicle" ,
113
+ model = self . model () ,
97
114
sitl_home = "1,1,1,1" ,
98
115
wipe = False ,
99
116
)
@@ -243,6 +260,7 @@ def run(self):
243
260
244
261
self .run_git (["checkout" , master_commit ], show_output = False )
245
262
self .run_git (["submodule" , "update" , "--recursive" ], show_output = False )
263
+ shutil .rmtree ("build" , ignore_errors = True )
246
264
board = "sitl"
247
265
if "AP_Periph" in self .vehicle :
248
266
board = "sitl_periph_universal"
@@ -257,6 +275,7 @@ def run(self):
257
275
258
276
self .run_git (["checkout" , branch ], show_output = False )
259
277
self .run_git (["submodule" , "update" , "--recursive" ], show_output = False )
278
+ shutil .rmtree ("build" , ignore_errors = True )
260
279
util .build_SITL (
261
280
self .build_target_name (self .vehicle ),
262
281
board = board ,
@@ -281,12 +300,14 @@ def __init__(self,
281
300
vehicles = None ,
282
301
run_eedump_before = False ,
283
302
run_eedump_after = False ,
303
+ master_branch = "master" ,
284
304
):
285
305
self .vehicles = vehicles
286
306
self .param_changes = param_changes
287
307
self .vehicles = vehicles
288
308
self .run_eedump_before = run_eedump_before
289
309
self .run_eedump_after = run_eedump_after
310
+ self .master_branch = master_branch
290
311
291
312
if self .vehicles is None :
292
313
self .vehicles = self .all_vehicles ()
@@ -310,6 +331,7 @@ def run(self):
310
331
self .param_changes ,
311
332
run_eedump_before = self .run_eedump_before ,
312
333
run_eedump_after = self .run_eedump_after ,
334
+ master_branch = self .master_branch ,
313
335
)
314
336
s .run ()
315
337
@@ -346,7 +368,12 @@ def run(self):
346
368
default = False ,
347
369
help = "run the (already-compiled) eedump tool on eeprom.bin after doing conversion" ,
348
370
)
349
-
371
+ parser .add_argument (
372
+ "--master-branch" ,
373
+ type = str ,
374
+ default = "master" ,
375
+ help = "master branch to use" ,
376
+ )
350
377
args = parser .parse_args ()
351
378
352
379
param_changes = []
@@ -374,5 +401,6 @@ def run(self):
374
401
vehicles = vehicles ,
375
402
run_eedump_before = args .run_eedump_before ,
376
403
run_eedump_after = args .run_eedump_after ,
404
+ master_branch = args .master_branch ,
377
405
)
378
406
x .run ()
0 commit comments