summaryrefslogtreecommitdiff
path: root/simpleperf/scripts/api_profiler.py
diff options
context:
space:
mode:
Diffstat (limited to 'simpleperf/scripts/api_profiler.py')
-rwxr-xr-xsimpleperf/scripts/api_profiler.py13
1 files changed, 2 insertions, 11 deletions
diff --git a/simpleperf/scripts/api_profiler.py b/simpleperf/scripts/api_profiler.py
index d39b8e3d..424de698 100755
--- a/simpleperf/scripts/api_profiler.py
+++ b/simpleperf/scripts/api_profiler.py
@@ -1,4 +1,4 @@
-#!/usr/bin/env python3
+#!/usr/bin/env python
#
# Copyright (C) 2019 The Android Open Source Project
#
@@ -33,8 +33,7 @@ import os.path
import shutil
import zipfile
-from simpleperf_utils import AdbHelper, get_target_binary_path, log_exit, log_info, remove
-
+from utils import AdbHelper, get_target_binary_path, log_exit, log_info, remove
def prepare_recording(args):
adb = AdbHelper()
@@ -42,7 +41,6 @@ def prepare_recording(args):
upload_simpleperf_to_device(adb)
run_simpleperf_prepare_cmd(adb)
-
def enable_profiling_on_device(adb, args):
android_version = adb.get_android_version()
if android_version >= 10:
@@ -51,14 +49,12 @@ def enable_profiling_on_device(adb, args):
adb.set_property('debug.perf_event_mlock_kb', str(args.max_memory_in_kb[0]))
adb.set_property('security.perf_harden', '0')
-
def upload_simpleperf_to_device(adb):
device_arch = adb.get_device_arch()
simpleperf_binary = get_target_binary_path(device_arch, 'simpleperf')
adb.check_run(['push', simpleperf_binary, '/data/local/tmp'])
adb.check_run(['shell', 'chmod', 'a+x', '/data/local/tmp/simpleperf'])
-
def run_simpleperf_prepare_cmd(adb):
adb.check_run(['shell', '/data/local/tmp/simpleperf', 'api-prepare'])
@@ -70,7 +66,6 @@ def collect_data(args):
download_recording_data(adb, args)
unzip_recording_data(args)
-
def download_recording_data(adb, args):
""" download recording data to simpleperf_data.zip."""
upload_simpleperf_to_device(adb)
@@ -79,7 +74,6 @@ def download_recording_data(adb, args):
adb.check_run(['pull', '/data/local/tmp/simpleperf_data.zip', args.out_dir])
adb.check_run(['shell', 'rm', '-rf', '/data/local/tmp/simpleperf_data'])
-
def unzip_recording_data(args):
zip_file_path = os.path.join(args.out_dir, 'simpleperf_data.zip')
with zipfile.ZipFile(zip_file_path, 'r') as zip_fh:
@@ -90,12 +84,10 @@ def unzip_recording_data(args):
zip_fh.extract(name, args.out_dir)
remove(zip_file_path)
-
class ArgumentHelpFormatter(argparse.ArgumentDefaultsHelpFormatter,
argparse.RawDescriptionHelpFormatter):
pass
-
def main():
parser = argparse.ArgumentParser(description=__doc__,
formatter_class=ArgumentHelpFormatter)
@@ -121,6 +113,5 @@ def main():
args = parser.parse_args()
args.func(args)
-
if __name__ == '__main__':
main()