blob: 6662f802456f17714800e83c0838dcabe66792cc [file] [log] [blame]
/*
* Copyright (C) 2013 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.uiautomator.platform;
import android.os.Bundle;
import android.os.Environment;
import android.util.Log;
import com.android.uiautomator.core.UiDevice;
import com.android.uiautomator.testrunner.UiAutomatorTestCase;
import java.io.BufferedOutputStream;
import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.FileWriter;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.List;
import java.util.Properties;
/**
* Base class for jank test.
* All jank test needs to extend JankTestBase
*/
public class JankTestBase extends UiAutomatorTestCase {
private static final String TAG = JankTestBase.class.getSimpleName();
protected UiDevice mDevice;
protected TestWatchers mTestWatchers = null;
protected BufferedWriter mWriter = null;
protected BufferedWriter mStatusWriter = null;
protected int mIteration = 20; // default iteration is set 20
/* can be used to enable/disable systrace in the test */
protected int mTraceTime = 0;
protected Bundle mParams;
protected String mTestCaseName;
protected int mSuccessTestRuns = 0;
protected Thread mThread = null;
// holds all params for the derived tests
private static final String PROPERTY_FILE_NAME = "UiJankinessTests.conf";
private static final String PARAM_CONFIG = "conf";
private static final String LOCAL_TMP_DIR = "/data/local/tmp/";
// File that hold the test results
private static String OUTPUT_FILE_NAME = LOCAL_TMP_DIR + "UiJankinessTestsOutput.txt";
// File that hold test status, e.g successful test iterations
private static String STATUS_FILE_NAME = LOCAL_TMP_DIR + "UiJankinessTestsStatus.txt";
private static final String RAW_DATA_DIR = LOCAL_TMP_DIR + "UiJankinessRawData";
private static int SUCCESS_THRESHOLD = 80;
private static boolean DEBUG = false;
/* default animation time is set to 2 seconds */
protected static final long DEFAULT_ANIMATION_TIME = 2 * 1000;
/* default swipe steps for fling animation */
protected static final int DEFAULT_FLING_STEPS = 8;
/* Array to record jankiness data in each test iteration */
private int[] jankinessArray;
/* Array to record frame rate in each test iteration */
private double[] frameRateArray;
/* Array to save max accumulated frame number in each test iteration */
private int[] maxDeltaVsyncArray;
/* Default file to store the systrace */
private static final File SYSTRACE_DIR = new File(LOCAL_TMP_DIR, "systrace");
/* Default trace file name */
private static final String TRACE_FILE_NAME = "trace.txt";
/* Default tracing time is 5 seconds */
private static final int DEFAULT_TRACE_TIME = 5; // 5 seconds
// Command to dump compressed trace data
private static final String ATRACE_COMMAND = "atrace -z -t %d gfx input view sched freq";
/**
* Thread to capture systrace log from the test
*/
public class SystraceTracker implements Runnable {
File mFile = new File(SYSTRACE_DIR, TRACE_FILE_NAME);
int mTime = DEFAULT_TRACE_TIME;
public SystraceTracker(int traceTime, String fileName) {
try {
if (!SYSTRACE_DIR.exists()) {
if (!SYSTRACE_DIR.mkdir()) {
log(String.format("create directory %s failed, you can manually create "
+ "it and start the test again", SYSTRACE_DIR.getAbsolutePath()));
return;
}
}
} catch (SecurityException e) {
Log.e(TAG, "creating directory failed?", e);
}
if (traceTime > 0) {
mTime = traceTime;
}
if (fileName != null) {
mFile = new File(SYSTRACE_DIR, fileName);
}
}
@Override
public void run() {
String command = String.format(ATRACE_COMMAND, mTime);
Log.v(TAG, "command: " + command);
Process p = null;
InputStream in = null;
BufferedOutputStream out = null;
try {
p = Runtime.getRuntime().exec(command);
Log.v(TAG, "write systrace into file: " + mFile.getAbsolutePath());
// read bytes from the process output stream as the output is compressed
byte[] buffer = new byte[1024];
in = p.getInputStream();
out = new BufferedOutputStream(new FileOutputStream(mFile));
int n;
while ((n = in.read(buffer)) != -1) {
out.write(buffer, 0, n);
out.flush();
}
in.close();
out.close();
// read error message
BufferedReader br = new BufferedReader(new InputStreamReader(p.getErrorStream()));
String line;
while ((line = br.readLine()) != null) {
Log.e(TAG, "Command return errors: " + line);
}
br.close();
// Due to limited buffer size for standard input and output stream,
// promptly reading from the input stream or output stream to avoid block
int status = p.waitFor();
if (status != 0) {
Log.e(TAG, String.format("Run shell command: %s, status: %s",
command, status));
}
} catch (InterruptedException e) {
Log.e(TAG, "Exception from command " + command + ":");
Log.e(TAG, "Thread interrupted? ", e);
} catch (IOException e) {
Log.e(TAG, "Open file error: ", e);
} catch (IllegalThreadStateException e) {
Log.e(TAG, "the process has not exit yet ", e);
}
}
}
@Override
protected void setUp() throws Exception {
super.setUp();
mDevice = UiDevice.getInstance();
mTestWatchers = new TestWatchers(); // extends the common class UiWatchers
mTestWatchers.registerAnrAndCrashWatchers();
mWriter = new BufferedWriter(new FileWriter(new File(OUTPUT_FILE_NAME), true));
mStatusWriter = new BufferedWriter(new FileWriter(new File(STATUS_FILE_NAME), true));
mParams = getParams();
if (mParams != null && !mParams.isEmpty()) {
log("mParams is not empty, get properties.");
String mIterationStr = getPropertyString(mParams, "iteration");
if (mIterationStr != null) {
mIteration = Integer.valueOf(mIterationStr);
}
String mTraceTimeStr = getPropertyString(mParams, "tracetime");
if (mTraceTimeStr != null) {
mTraceTime = Integer.valueOf(mTraceTimeStr);
}
}
jankinessArray = new int[mIteration];
frameRateArray = new double[mIteration];
maxDeltaVsyncArray = new int[mIteration];
mTestCaseName = this.getName();
mSuccessTestRuns = 0;
mDevice.pressHome();
}
/**
* Create a new thread for systrace and start the thread
*
* @param testCaseName
* @param iteration
*/
protected void startTrace(String testCaseName, int iteration) {
if (mTraceTime > 0) {
String outputFile = String.format("%s_%d_trace", mTestCaseName, iteration);
mThread = new Thread(new SystraceTracker(mTraceTime, outputFile));
mThread.start();
}
}
/**
* Wait for the tracing thread to exit
*/
protected void endTrace() {
if (mThread != null) {
try {
mThread.join();
} catch (InterruptedException e) {
Log.e(TAG, "wait for the trace thread to exit exception:", e);
}
}
}
/**
* Expects a file from the command line via conf param or default following format each on its
* own line. <code>
* key=Value
* Browser_URL1=cnn.com
* Browser_URL2=google.com
* Camera_ShutterDelay=1000
* etc...
* </code>
* @param Bundle params
* @param key
* @return the value of the property else defaultValue
* @throws FileNotFoundException
* @throws IOException
*/
protected String getPropertyString(Bundle params, String key)
throws FileNotFoundException, IOException {
Properties prop = new Properties();
prop.load(new FileInputStream(new File(LOCAL_TMP_DIR,
params.getString(PARAM_CONFIG, PROPERTY_FILE_NAME))));
String value = prop.getProperty(key);
if (value != null && !value.isEmpty())
return value;
return null;
}
/**
* Expects a file from the command line via conf param or default following format each on its
* own line. <code>
* key=Value
* Browser_URL1=cnn.com
* Browser_URL2=google.com
* Camera_ShutterDelay=1000
* etc...
* </code>
* @param Bundle params
* @param key
* @return the value of the property else defaultValue
* @throws FileNotFoundException
* @throws IOException
*/
protected long getPropertyLong(Bundle params, String key)
throws FileNotFoundException, IOException {
Properties prop = new Properties();
prop.load(new FileInputStream(new File(LOCAL_TMP_DIR,
params.getString(PARAM_CONFIG, PROPERTY_FILE_NAME))));
String value = prop.getProperty(key);
if (value != null && !value.trim().isEmpty())
return Long.valueOf(value.trim());
return 0;
}
/**
* Verify the test result by comparing data sample size with expected value
* @param expectedDataSize the expected data size
*/
protected boolean validateResults(int expectedDataSize) {
int receivedDataSize = SurfaceFlingerHelper.getDataSampleSize();
return ((expectedDataSize > 0) && (receivedDataSize >= expectedDataSize));
}
/**
* Process the raw data, calculate jankiness, frame rate and max accumulated frames number
* @param testCaseName
* @param iteration
*/
protected void recordResults(String testCaseName, int iteration) {
long refreshPeriod = SurfaceFlingerHelper.getRefreshPeriod();
// if the raw directory doesn't exit, create the directory
File rawDataDir = new File(RAW_DATA_DIR);
try {
if (!rawDataDir.exists()) {
if (!rawDataDir.mkdir()) {
log(String.format("create directory %s failed, you can manually create " +
"it and start the test again", rawDataDir));
}
}
} catch (SecurityException e) {
Log.e(TAG, "create directory failed: ", e);
}
String rawFileName = String.format("%s/%s_%d.txt", RAW_DATA_DIR, testCaseName, iteration);
// write results into a file
BufferedWriter fw = null;
try {
fw = new BufferedWriter(new FileWriter(new File(rawFileName), false));
fw.write(SurfaceFlingerHelper.getFrameBufferData());
} catch (IOException e) {
Log.e(TAG, "failed to write to file", e);
return;
} finally {
try {
if (fw != null) {
fw.close();
}
}
catch (IOException e) {
Log.e(TAG, "close file failed.", e);
}
}
// get jankiness count
int jankinessCount = SurfaceFlingerHelper.getVsyncJankiness();
// get frame rate
double frameRate = SurfaceFlingerHelper.getFrameRate();
// get max accumulated frames
int maxDeltaVsync = SurfaceFlingerHelper.getMaxDeltaVsync();
// only record data when they are valid
if (jankinessCount >=0 && frameRate > 0) {
jankinessArray[iteration] = jankinessCount;
frameRateArray[iteration] = frameRate;
maxDeltaVsyncArray[iteration] = maxDeltaVsync;
mSuccessTestRuns++;
}
String msg = String.format("%s, iteration %d\n" +
"refresh period: %d\n" +
"jankiness count: %d\n" +
"frame rate: %f\n" +
"max accumulated frames: %d\n",
testCaseName, iteration, refreshPeriod,
jankinessCount, frameRate, maxDeltaVsync);
log(msg);
if (DEBUG) {
SurfaceFlingerHelper.printData(testCaseName, iteration);
}
}
/**
* Process data from all test iterations, and save to disk
* @param testCaseName
*/
protected void saveResults(String testCaseName) {
// write test status into status file
try {
mStatusWriter.write(String.format("%s: %d success runs out of %d iterations\n",
testCaseName, mSuccessTestRuns, mIteration));
} catch (IOException e) {
log("failed to write output for test case " + testCaseName);
}
// if successful test runs is less than the threshold, no results will be saved.
if (mSuccessTestRuns * 100 / mIteration < SUCCESS_THRESHOLD) {
log(String.format("In %s, # of successful test runs out of %s iterations: %d ",
testCaseName, mIteration, mSuccessTestRuns));
log(String.format("threshold is %d%%", SUCCESS_THRESHOLD));
return;
}
if (DEBUG) {
print(jankinessArray, "jankiness array");
print(frameRateArray, "frame rate array");
print(maxDeltaVsyncArray, "max delta vsync array");
}
double avgJankinessCount = getAverage(jankinessArray);
int maxJankinessCount = getMaxValue(jankinessArray);
double avgFrameRate = getAverage(frameRateArray);
double avgMaxDeltaVsync = getAverage(maxDeltaVsyncArray);
String avgMsg = String.format("%s\n" +
"average number of jankiness: %f\n" +
"max number of jankiness: %d\n" +
"average frame rate: %f\n" +
"average of max accumulated frames: %f\n",
testCaseName, avgJankinessCount, maxJankinessCount, avgFrameRate, avgMaxDeltaVsync);
log(avgMsg);
try {
mWriter.write(avgMsg);
} catch (IOException e) {
log("failed to write output for test case " + testCaseName);
}
}
// return the max value in an integer array
private int getMaxValue(int[] intArray) {
int index = 0;
int max = intArray[index];
for (int i = 1; i < intArray.length; i++) {
if (max < intArray[i]) {
max = intArray[i];
}
}
return max;
}
private double getAverage(int[] intArray) {
int mean = 0;
int numberTests = 0;
for (int i = 0; i < intArray.length; i++) {
// in case in some iteration, test fails, no data points is collected
if (intArray[i] >= 0) {
mean += intArray[i];
++numberTests;
}
}
return (double)mean/numberTests;
}
private double getAverage(double[] doubleArray) {
double mean = 0;
int numberTests = 0;
for (int i = 0; i < doubleArray.length; i++) {
// in case in some iteration, test fails, no data points is collected
if (doubleArray[i] >= 0) {
mean += doubleArray[i];
++numberTests;
}
}
return mean/numberTests;
}
private void print(int[] intArray, String arrayName) {
log("start to print array for " + arrayName);
for (int i = 0; i < intArray.length; i++) {
log(String.format("%d: %d", i, intArray[i]));
}
}
private void print(double[] doubleArray, String arrayName) {
log("start to print array for " + arrayName);
for (int i = 0; i < doubleArray.length; i++) {
log(String.format("%d: %f", i, doubleArray[i]));
}
}
@Override
protected void tearDown() throws Exception {
super.tearDown();
if (mWriter != null) {
mWriter.close();
}
if (mStatusWriter != null) {
mStatusWriter.close();
}
}
private void log(String message) {
Log.v(TAG, message);
}
/**
* Set the total number of test iteration
* @param iteration
*/
protected void setIteration(int iteration){
mIteration = iteration;
}
/**
* Get the total number of test iteration
* @return iteration
*/
protected int getIteration(){
return mIteration;
}
}