tests/java_api/RSUnitTests/src/com/android/rs/unittest/UT_single_source_alloc.java

/*
 * Copyright (C) 2017 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.rs.unittest;

import android.content.Context;
import android.renderscript.RenderScript;

public class UT_single_source_alloc extends UnitTest {
    private int dimX = 3;
    private int dimY = 4;
    private int dimZ = 5;
    private int start = 23;

    // rs_data_type for float, double, char, short, int, long, uchar, ushort, uint, ulong in that
    // order
    private int rsDataTypes[] = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11};

    public UT_single_source_alloc(Context ctx) {
        super("SingleSourceAllocation", ctx);
    }

    private void initializeGlobals(RenderScript RS, ScriptC_single_source_alloc s, int nDims) {
        s.set_gDimX(dimX);
        s.set_gDimY(nDims > 1 ? dimY : 0);
        s.set_gDimZ(nDims > 2 ? dimZ : 0);
        s.set_gStart(start);

        return;
    }

    public void run() {
        RenderScript pRS = RenderScript.create(mCtx);
        ScriptC_single_source_alloc s = new ScriptC_single_source_alloc(pRS);
        pRS.setMessageHandler(mRsMessage);

        // Test 1-D, 2-D and 3-D Allocations of basic RenderScript types by creating Allocations and
        // invoking a kernel on them.
        for (int dataType : rsDataTypes) {
            for (int vecSize = 1; vecSize <= 4; vecSize++) {
                for (int nDims = 1; nDims <= 3; nDims++) {
                    initializeGlobals(pRS, s, nDims);
                    s.invoke_CreateAndTestAlloc(dataType, vecSize);
                }
            }
        }

        // Exhaustively test valid and invalid calls to rs_* creation functions.  (These tests don't
        // walk the created allocations, though.)
        s.invoke_TestAllCases();

        s.invoke_single_source_alloc_test();
        pRS.finish();
        s.destroy();
        pRS.destroy();
    }
}