summaryrefslogtreecommitdiff
path: root/startop/scripts/iorap/compiler.py
diff options
context:
space:
mode:
authorYan Wang <yawanng@google.com>2019-07-26 10:53:45 -0700
committerYan Wang <yawanng@google.com>2019-07-31 14:38:40 -0700
commit6ddab32f683dcbec8c90c047c4a62f05ec5d2396 (patch)
tree68c25a0635c6b20dc54b6afebc270c646164fc50 /startop/scripts/iorap/compiler.py
parent59aab11f092b4f9d817c331a470fa8d2fad445ab (diff)
startop: Add support for trace duration.
Test: pytest app_startup/ Bug: 138233470 Change-Id: Ic8e99b369d3385015d7c86624a61bb6d5e8fdc70
Diffstat (limited to 'startop/scripts/iorap/compiler.py')
-rwxr-xr-xstartop/scripts/iorap/compiler.py8
1 files changed, 5 insertions, 3 deletions
diff --git a/startop/scripts/iorap/compiler.py b/startop/scripts/iorap/compiler.py
index a335b30da8a5..ef9b870b8113 100755
--- a/startop/scripts/iorap/compiler.py
+++ b/startop/scripts/iorap/compiler.py
@@ -31,8 +31,10 @@ import tempfile
from pathlib import Path
from typing import Iterable, Optional, List
-from generated.TraceFile_pb2 import *
-from lib.inode2filename import Inode2Filename
+DIR = os.path.abspath(os.path.dirname(__file__))
+sys.path.append(os.path.dirname(DIR))
+from iorap.generated.TraceFile_pb2 import *
+from iorap.lib.inode2filename import Inode2Filename
parent_dir_name = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))
sys.path.append(parent_dir_name)
@@ -305,7 +307,7 @@ def main(argv):
transform_perfetto_trace_to_systrace(options.perfetto_trace_file,
trace_file.name)
return run(sql_db_path,
- options.trace_file,
+ trace_file.name,
options.trace_duration,
options.output_file,
inode_table,