add flag to process all tracks of a particular location

2021-07-20 05:46:43 -07:00
parent 793bb64af4
commit 2e29f42d75
1 changed files with 7 additions and 1 deletions
--- a/src/expert_data.py
+++ b/src/expert_data.py
@@ -91,5 +91,11 @@ if __name__ == '__main__':
                       help='location (default 0)')
    parser.add_argument('--track', default=0, type=int,
                       help='track number (default 0)')
+    parser.add_argument('--all-tracks', action='store_true',
+                       help='whether to process all tracks at location')
    args = parser.parse_args()
-    generate_expert_data(loc=args.loc,track=args.track)
+    if args.all_tracks:
+        for i in range(intersim.MAX_TRACKS):
+            generate_expert_data(loc=args.loc, track=i)
+    else:
+        generate_expert_data(loc=args.loc,track=args.track)