com/android/startop/test/CPUIntensiveBenchmarks.java

/*
 * Copyright (C) 2019 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.
 */

/**
 *  A threaded CPU intensive class for use in benchmarks.
 */

package com.android.startop.test;

import android.app.Activity;

public class CPUIntensiveBenchmarks {
    public static final int ARRAY_SIZE = 30000;
    public static int[][] mArray;

    static class WorkerThread extends Thread {
        int mThreadNumber;
        WorkerThread(int number) {
            mThreadNumber = number;
        }
        public void run() {
            final int arrayLength = mArray[mThreadNumber].length;
            for (int i = 0; i < arrayLength; ++i) {
                mArray[mThreadNumber][i] = i * i;
            }
            for (int i = 0; i < arrayLength; ++i) {
                for (int j = 0; j < arrayLength; ++j) {
                    int swap = mArray[mThreadNumber][j];
                    mArray[mThreadNumber][j] = mArray[mThreadNumber][(j + i) % arrayLength];
                    mArray[mThreadNumber][(j + i) % arrayLength] = swap;
                }
            }
        }
    };

    static void doSomeWork(int threadCount) {
        mArray = new int[threadCount][ARRAY_SIZE];
        WorkerThread[] threads = new WorkerThread[threadCount];
        // Create the threads.
        for (int i = 0; i < threadCount; ++i) {
            threads[i] = new WorkerThread(i);
        }
        // Start the threads.
        for (int i = 0; i < threadCount; ++i) {
            threads[i].start();
        }
        // Join the threads.
        for (int i = 0; i < threadCount; ++i) {
            try {
                threads[i].join();
            } catch (Exception ex) {
            }
        }
    }

    // Time limit to run benchmarks in seconds
    public static final int TIME_LIMIT = 5;

    static void initializeBenchmarks(Activity parent, BenchmarkRunner benchmarks) {
        benchmarks.addBenchmark("Use 1 thread", () -> {
            doSomeWork(1);
        });
        benchmarks.addBenchmark("Use 2 threads", () -> {
            doSomeWork(2);
        });
        benchmarks.addBenchmark("Use 4 threads", () -> {
            doSomeWork(4);
        });
        benchmarks.addBenchmark("Use 8 threads", () -> {
            doSomeWork(8);
        });
        benchmarks.addBenchmark("Use 16 threads", () -> {
            doSomeWork(16);
        });
    }
}