mirror of
https://github.com/ARM-software/workload-automation.git
synced 2025-01-19 20:34:30 +00:00
5a67aabcbd
Port manual workload from WA2. This workload allows using WA to configure and instrument the target, while letting the user perform arbitrary actions in place of running a fixed workload.
96 lines
3.9 KiB
Python
96 lines
3.9 KiB
Python
# Copyright 2013-2015 ARM Limited
|
|
#
|
|
# 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.
|
|
#
|
|
# pylint: disable=E1101,W0201,E0203
|
|
import os
|
|
|
|
from wa import Workload, Parameter, ConfigError
|
|
from wa.utils.misc import getch
|
|
|
|
|
|
class ManualWorkload(Workload):
|
|
|
|
name = 'manual'
|
|
description = """
|
|
Yields control to the user, either for a fixed period or based on user
|
|
input, to perform custom operations on the device, which workload
|
|
automation does not know of.
|
|
|
|
"""
|
|
default_duration = 30
|
|
|
|
parameters = [
|
|
Parameter('duration', kind=int, default=None,
|
|
description="""
|
|
Control of the devices is yielded for the duration (in
|
|
seconds) specified. If not specified, ``user_triggered`` is
|
|
assumed.
|
|
"""),
|
|
Parameter('user_triggered', kind=bool, default=None,
|
|
description="""
|
|
If ``True``, WA will wait for user input after starting the
|
|
workload; otherwise fixed duration is expected. Defaults to
|
|
``True`` if ``duration`` is not specified, and ``False``
|
|
otherwise.
|
|
"""),
|
|
Parameter('view', default='SurfaceView',
|
|
description="""
|
|
Specifies the View of the workload. This enables instruments
|
|
that require a View to be specified, such as the ``fps``
|
|
instrument. This is required for using "SurfaceFlinger" to
|
|
collect FPS statistics and is primarily used on devices pre
|
|
API level 23.
|
|
"""),
|
|
Parameter('package',
|
|
description="""
|
|
Specifies the package name of the workload. This enables
|
|
instruments that require a Package to be specified, such as
|
|
the ``fps`` instrument. This allows for "gfxinfo" to be used
|
|
and is the preferred method of collection for FPS statistics
|
|
on devices API level 23+.
|
|
"""),
|
|
]
|
|
|
|
def validate(self):
|
|
if self.duration is None:
|
|
if self.user_triggered is None:
|
|
self.user_triggered = True
|
|
elif self.user_triggered is False:
|
|
self.duration = self.default_duration
|
|
if self.user_triggered and self.duration:
|
|
message = 'Manual Workload can either specify duration or be user triggered, but not both'
|
|
raise ConfigError(message)
|
|
if not self.user_triggered and not self.duration:
|
|
raise ConfigError('Either user_triggered must be ``True`` or duration must be > 0.')
|
|
|
|
def setup(self, context):
|
|
self.logger.info('Any setup required by your workload should be done now.')
|
|
self.logger.info('As soon as you are done hit any key and wait for the message')
|
|
self.logger.info('"START NOW!" to begin your manual workload.')
|
|
self.logger.info('')
|
|
self.logger.info('hit any key to finalize your setup...')
|
|
getch()
|
|
|
|
def run(self, context):
|
|
self.logger.info('START NOW!')
|
|
if self.duration:
|
|
self.device.sleep(self.duration)
|
|
elif self.user_triggered:
|
|
self.logger.info('')
|
|
self.logger.info('hit any key to end your workload execution...')
|
|
getch()
|
|
else:
|
|
raise ConfigError('Illegal parameters for manual workload')
|
|
self.logger.info('DONE!')
|