mirror of
https://gitee.com/openharmony/arkui_ace_engine
synced 2024-10-07 16:33:54 +00:00
89 lines
2.8 KiB
Python
89 lines
2.8 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
|
|
#
|
|
# Copyright (c) 2024 Huawei Device Co., Ltd.
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
#
|
|
|
|
|
|
import argparse
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
|
|
from src.content_parser import ContentParser
|
|
from src.graph.graph_converter import generate_all_graphs
|
|
from src.pre_process import handle_file_preprocess
|
|
from src.utils.log_wrapper import log_info, log_error, enable_debug
|
|
|
|
|
|
def usage():
|
|
print('python main.py -i input.txt')
|
|
print('\n Usage: main.py <cmd> <input>')
|
|
print(' <cmd>: TODO')
|
|
print(' <input>: input dump file')
|
|
return
|
|
|
|
|
|
def parse_args():
|
|
parser = argparse.ArgumentParser(description='')
|
|
parser.add_argument('-d', action='store_true', default=False, help='enable debug info')
|
|
parser.add_argument('-r', action='store_true', default=False, help='dump event tree with device')
|
|
parser.add_argument('-i',
|
|
type=str, default='./resources/dumpfile/input.txt', help='input the dump source file')
|
|
parser.add_argument('-m', action='store_true', default=False, help='add details info')
|
|
argument = parser.parse_args()
|
|
argument.input_file = argument.i
|
|
argument.detailed = argument.m
|
|
argument.dump_from_device = argument.r
|
|
argument.debug = argument.d
|
|
return argument
|
|
|
|
|
|
def dump_from_device():
|
|
bat_file_path = r'src\bats\dump_event.bat'
|
|
try:
|
|
subprocess.call([bat_file_path])
|
|
print('capture event tree done.')
|
|
except FileNotFoundError:
|
|
print(f'file not found: {bat_file_path}')
|
|
except Exception as e:
|
|
print(f'exception: {e}')
|
|
|
|
|
|
# python main.py -i input.txt
|
|
if __name__ == '__main__':
|
|
# parse the args
|
|
args = parse_args()
|
|
# config log model
|
|
if args.debug:
|
|
enable_debug(True)
|
|
# dump trace from device if needed
|
|
if args.dump_from_device:
|
|
if os.name == 'nt':
|
|
dump_from_device()
|
|
else:
|
|
log_error('only support dump from device on windows')
|
|
sys.exit(1)
|
|
# pre process
|
|
handle_file_preprocess(args.input_file, 'dump_temp.txt')
|
|
# read the dump file and parse
|
|
dump_result = ContentParser('dump_temp.txt').do_parse()
|
|
if dump_result.is_succeed():
|
|
log_info('parse done')
|
|
dump_result.dump()
|
|
else:
|
|
log_error('parse failed')
|
|
generate_all_graphs(dump_result, args.detailed)
|