1/*
2 * Copyright (C) 2015 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 *      http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17package com.android.rs.test;
18
19import android.content.Context;
20import android.content.res.Resources;
21import android.renderscript.*;
22
23public class UT_fp16 extends UnitTest {
24    private Resources mRes;
25    private int dimX = 7;
26    private int dimY = 5;
27    private int dimZ = 3;
28    private Allocation alloc;
29
30    protected UT_fp16(RSTestCore rstc, Resources res, Context ctx) {
31        super(rstc, "Fp16", ctx);
32        mRes = res;
33    }
34
35    private void initializeGlobals(RenderScript RS, ScriptC_fp16 s, Element e, int nDims) {
36        Type.Builder typeBuilder = new Type.Builder(RS, e);
37        switch(nDims) {
38            case 1:
39                typeBuilder.setX(dimX);
40                break;
41            case 2:
42                typeBuilder.setX(dimX).setY(dimY);
43                break;
44            case 3:
45                typeBuilder.setX(dimX).setY(dimY).setZ(dimZ);
46                break;
47        }
48        alloc = Allocation.createTyped(RS, typeBuilder.create());
49
50        s.set_gDimX(dimX);
51        s.set_gDimY(nDims > 1? dimY: 0);
52        s.set_gDimZ(nDims > 2? dimZ: 0);
53        s.set_gAlloc(alloc);
54    }
55
56    private void TestHalf(RenderScript RS, ScriptC_fp16 s, int nDims) {
57        initializeGlobals(RS, s, Element.F16(RS), nDims);
58
59        // set in kernel and test rsGetElementAt in invoke
60        s.forEach_set_kernel_half(alloc);
61        s.invoke_verify_half();
62
63        initializeGlobals(RS, s, Element.F16(RS), nDims);
64        // rsSetElementAt in invoke and verify in kernel
65        s.invoke_set_half();
66        s.forEach_verify_kernel_half(alloc);
67    }
68
69    private void TestHalf2(RenderScript RS, ScriptC_fp16 s, int nDims) {
70        initializeGlobals(RS, s, Element.F16_2(RS), nDims);
71
72        // set in kernel and test rsGetElementAt in invoke
73        s.forEach_set_kernel_half2(alloc);
74        s.invoke_verify_half2();
75
76        initializeGlobals(RS, s, Element.F16_2(RS), nDims);
77        // rsSetElementAt in invoke and verify in kernel
78        s.invoke_set_half2();
79        s.forEach_verify_kernel_half2(alloc);
80    }
81
82    private void TestHalf3(RenderScript RS, ScriptC_fp16 s, int nDims) {
83        initializeGlobals(RS, s, Element.F16_3(RS), nDims);
84
85        // set in kernel and test rsGetElementAt in invoke
86        s.forEach_set_kernel_half3(alloc);
87        s.invoke_verify_half3();
88
89        initializeGlobals(RS, s, Element.F16_3(RS), nDims);
90        // rsSetElementAt in invoke and verify in kernel
91        s.invoke_set_half3();
92        s.forEach_verify_kernel_half3(alloc);
93    }
94
95    private void TestHalf4(RenderScript RS, ScriptC_fp16 s, int nDims) {
96        initializeGlobals(RS, s, Element.F16_4(RS), nDims);
97
98        // set in kernel and test rsGetElementAt in invoke
99        s.forEach_set_kernel_half4(alloc);
100        s.invoke_verify_half4();
101
102        initializeGlobals(RS, s, Element.F16_4(RS), nDims);
103        // rsSetElementAt in invoke and verify in kernel
104        s.invoke_set_half4();
105        s.forEach_verify_kernel_half4(alloc);
106    }
107
108    public void run() {
109        Allocation A;
110        RenderScript pRS = RenderScript.create(mCtx);
111        pRS.setMessageHandler(mRsMessage);
112        ScriptC_fp16 s = new ScriptC_fp16(pRS);
113
114        for (int nDims = 1; nDims <= 3; nDims ++) {
115            TestHalf(pRS, s, nDims);
116            TestHalf2(pRS, s, nDims);
117            TestHalf3(pRS, s, nDims);
118            TestHalf4(pRS, s, nDims);
119        }
120
121        s.invoke_fp16_test();
122        pRS.finish();
123        waitForMessage();
124        pRS.destroy();
125    }
126}
127