1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
|
#!/usr/bin/python
# Copyright 2011 Google Inc. All Rights Reserved.
"""The driver script for running performance benchmarks on ChromeOS."""
import atexit
import optparse
import os
import sys
from experiment_runner import ExperimentRunner
from experiment_runner import MockExperimentRunner
from experiment_factory import ExperimentFactory
from experiment_file import ExperimentFile
from help import Help
from settings_factory import GlobalSettings
from utils import logger
import test_flag
class MyIndentedHelpFormatter(optparse.IndentedHelpFormatter):
def format_description(self, description):
return description
def SetupParserOptions(parser):
"""Add all options to the parser."""
parser.add_option("--dry_run",
dest="dry_run",
help=("Parse the experiment file and "
"show what will be done"),
action="store_true",
default=False)
# Allow each of the global fields to be overridden by passing in
# options. Add each global field as an option.
option_settings = GlobalSettings("")
for field_name in option_settings.fields:
field = option_settings.fields[field_name]
parser.add_option("--%s" % field.name,
dest=field.name,
help=field.description,
action="store")
def ConvertOptionsToSettings(options):
"""Convert options passed in into global settings."""
option_settings = GlobalSettings("option_settings")
for option_name in options.__dict__:
if (options.__dict__[option_name] is not None and
option_name in option_settings.fields):
option_settings.SetField(option_name, options.__dict__[option_name])
return option_settings
def Cleanup(experiment):
"""Handler function which is registered to the atexit handler."""
experiment.Cleanup()
def Main(argv):
parser = optparse.OptionParser(usage=Help().GetUsage(),
description=Help().GetHelp(),
formatter=MyIndentedHelpFormatter(),
version="%prog 0.1")
parser.add_option("-l", "--log_dir",
dest="log_dir",
default="",
help="The log_dir, default is under <crosperf_logs>/logs")
SetupParserOptions(parser)
options, args = parser.parse_args(argv)
# Convert the relevant options that are passed in into a settings
# object which will override settings in the experiment file.
option_settings = ConvertOptionsToSettings(options)
log_dir = os.path.abspath(os.path.expanduser(options.log_dir))
logger.GetLogger(log_dir)
if len(args) == 2:
experiment_filename = args[1]
else:
parser.error("Invalid number arguments.")
working_directory = os.getcwd()
if options.dry_run:
test_flag.SetTestMode(True)
experiment_file = ExperimentFile(open(experiment_filename, "rb"),
option_settings)
if not experiment_file.GetGlobalSettings().GetField("name"):
experiment_name = os.path.basename(experiment_filename)
experiment_file.GetGlobalSettings().SetField("name", experiment_name)
experiment = ExperimentFactory().GetExperiment(experiment_file,
working_directory,
log_dir)
atexit.register(Cleanup, experiment)
if options.dry_run:
runner = MockExperimentRunner(experiment)
else:
runner = ExperimentRunner(experiment)
runner.Run()
if __name__ == "__main__":
Main(sys.argv)
|
