| /* |
| * Copyright (C) 2008 The Android Open Source Project |
| * |
| * 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. |
| */ |
| |
| package com.android.commands.monkey; |
| |
| import java.security.SecureRandom; |
| import java.util.ArrayList; |
| |
| /** |
| * Class for generating MonkeyEvents from multiple scripts. |
| */ |
| public class MonkeySourceRandomScript implements MonkeyEventSource { |
| /** The verbose level of the source (currently not used) */ |
| private int mVerbose = 0; |
| |
| /** The source for the setup script if it exists */ |
| private MonkeySourceScript mSetupSource = null; |
| |
| /** The list of MonkeySourceScript instances to be played in random order */ |
| private ArrayList<MonkeySourceScript> mScriptSources = new ArrayList<MonkeySourceScript>(); |
| |
| /** The current source, set to the setup source and then a random script */ |
| private MonkeySourceScript mCurrentSource = null; |
| |
| /** The random number generator */ |
| private SecureRandom mRandom; |
| |
| /** |
| * Creates a MonkeySourceRandomScript instance with an additional setup script. |
| * |
| * @param setupFileName The name of the setup script file on the device. |
| * @param scriptFileNames An ArrayList of the names of the script files to be run randomly. |
| * @param throttle The amount of time to sleep in ms between events. |
| * @param seed The seed of the random number generator. |
| */ |
| public MonkeySourceRandomScript(String setupFileName, ArrayList<String> scriptFileNames, |
| long throttle, long seed) { |
| if (setupFileName != null) { |
| mSetupSource = new MonkeySourceScript(setupFileName, throttle); |
| mCurrentSource = mSetupSource; |
| } |
| |
| for (String fileName: scriptFileNames) { |
| mScriptSources.add(new MonkeySourceScript(fileName, throttle)); |
| } |
| |
| mRandom = new SecureRandom(); |
| mRandom.setSeed((seed == 0) ? -1 : seed); |
| } |
| |
| /** |
| * Creates a MonkeySourceRandomScript instance without an additional setup script. |
| * |
| * @param scriptFileNames An ArrayList of the names of the script files to be run randomly. |
| * @param throttle The amount of time to sleep in ms between events. |
| * @param seed The seed of the random number generator. |
| */ |
| public MonkeySourceRandomScript(ArrayList<String> scriptFileNames, long throttle, long seed) { |
| this(null, scriptFileNames, throttle, seed); |
| } |
| |
| /** |
| * Gets the next event from the current event source. If the event source is null, a new |
| * script event source is chosen randomly from the list of script sources and the next event is |
| * chosen from that. |
| * |
| * @return The first event in the event queue or null if the end of the file |
| * is reached or if an error is encountered reading the file. |
| */ |
| public MonkeyEvent getNextEvent() { |
| if (mCurrentSource == null) { |
| int numSources = mScriptSources.size(); |
| if (numSources == 1) { |
| mCurrentSource = mScriptSources.get(0); |
| } else if (numSources > 1) { |
| mCurrentSource = mScriptSources.get(mRandom.nextInt(numSources)); |
| } |
| } |
| |
| if (mCurrentSource != null) { |
| MonkeyEvent nextEvent = mCurrentSource.getNextEvent(); |
| if (nextEvent == null) { |
| mCurrentSource = null; |
| } |
| return nextEvent; |
| } |
| return null; |
| } |
| |
| /** |
| * Sets the verbosity for the source as well as all sub event sources. |
| * |
| * @param verbose The verbose level. |
| */ |
| public void setVerbose(int verbose) { |
| mVerbose = verbose; |
| |
| if (mSetupSource != null) { |
| mSetupSource.setVerbose(verbose); |
| } |
| |
| for (MonkeySourceScript source: mScriptSources) { |
| source.setVerbose(verbose); |
| } |
| } |
| |
| /** |
| * Validates that all the underlying event sources are valid |
| * |
| * @return True if all the script files are valid. |
| * |
| * @see MonkeySourceScript#validate() |
| */ |
| public boolean validate() { |
| if (mSetupSource != null && !mSetupSource.validate()) { |
| return false; |
| } |
| |
| for (MonkeySourceScript source: mScriptSources) { |
| if (!source.validate()) { |
| return false; |
| } |
| } |
| |
| return true; |
| } |
| } |