def main():
# parse command line
option = parse_cmdline()
filtered_devices_id = option.devices
is_list_devices = int(option.list_devices)
filtered_cases_id = option.cases
is_list_cases = int(option.list_cases)
case_file = option.case_file
executable = option.executable
# check command line
if is_list_devices:
list_devices(is_list_devices)
return
if is_list_cases:
list_cases(is_list_cases)
return
if not os.path.isfile(executable):
print "Error: " + executable + " doesn't exist."
return
# parse cases
all_cases = CaseSet()
if (case_file != ""):
all_cases.parse_xml(os.path.join(os.path.dirname(__file__), case_file))
else:
all_cases.parse_xml_dir(os.path.join(os.path.dirname(__file__), "..", "cases"))
filtered_case_set = {}
if filtered_cases_id:
for c in filtered_cases_id:
if c in all_cases.keys():
filtered_case_set[c] = all_cases[c]
else:
print "Warning: " + c + " is not an legal case ID."
else:
filtered_case_set = all_cases
# parse devices
all_devices = DeviceSet()
all_devices.parse_xml(os.path.join(os.path.dirname(__file__), "..", "devices", "devices.xml"))
filtered_device_set = {}
for d in filtered_devices_id:
if d in all_devices.keys():
filtered_device_set[d] = all_devices[d]
else:
print "Warning: " + d + " is not an legal device ID."
# execute profiling
e = Executor(executable, filtered_device_set, filtered_case_set, duplicate=10)
try:
logfiles = e.run()
except (KeyboardInterrupt):
print "KeyboardInterrupt in run.main()"
sys.exit()
# analyze log
for case_id in logfiles.keys():
for device_id in logfiles[case_id].keys():
case = all_cases[case_id]
root_node = case.get_root_node()
coeff = case.get_mcps_norm_coeff()
interest_nodes = option.interest_nodes
ana = Analyzer(root_node, coeff, interest_nodes)
if len(logfiles[case_id][device_id]) > 0:
ana.run(logfiles[case_id][device_id])
print "Finished! \nSee profile logs here: "
for dir in e.get_reports_dir():
print "\t./" + dir
