1 files changed, 10 insertions, 6 deletions
diff --git a/scripts/trace_filter.py b/scripts/trace_filter.py
index 5fbf51db4..3ad871515 100755
--- a/scripts/trace_filter.py
+++ b/scripts/trace_filter.py
@@ -2,6 +2,7 @@
 
 import argparse
 import re
+from pathlib import Path
 
 MARK_LIST = ['tracing_mark_write']
 
@@ -10,20 +11,23 @@ def get_arg():
     parser = argparse.ArgumentParser(description='Filter a log file to a trace file.')
     parser.add_argument('log_file', metavar='log_file', type=str,
                         help='The input log file to process.')
-    parser.add_argument('trace_file', metavar='trace_file', type=str, nargs='?', default='trace.systrace',
-                        help='The output trace file. If not provided, defaults to \'trace.systrace\'.')
+    parser.add_argument('trace_file', metavar='trace_file', type=str, nargs='?',
+                        help='The output trace file. If not provided, defaults to \'<log_file>.systrace\'.')
 
     args = parser.parse_args()
-
-    print('log_file: ' + args.log_file)
-    print('trace_file: ' + args.trace_file)
-
     return args
 
 
 if __name__ == '__main__':
     args = get_arg()
 
+    if not args.trace_file:
+        log_file = Path(args.log_file)
+        args.trace_file = log_file.with_suffix('.systrace').as_posix()
+
+    print('log_file  :', args.log_file)
+    print('trace_file:', args.trace_file)
+
     with open(args.log_file, 'r') as f:
         content = f.read()