/*
 * Copyright (C) 2012 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 android.renderscript.cts;

import android.renderscript.Allocation;
import android.renderscript.Element;
import android.renderscript.Type;
import android.renderscript.Sampler;
import android.renderscript.Script;
import android.renderscript.RSRuntimeException;

public class ClearObjectTest extends RSBaseCompute {
    int ObjectNum = 1;
    static final int TEST_ID_ELEMENT = 0;
    static final int TEST_ID_TYPE = 1;
    static final int TEST_ID_ALLOCATION = 2;
    static final int TEST_ID_SAMPLER = 3;
    static final int TEST_ID_SCRIPT = 4;
    private ScriptC_clear_object ms_clear;

    @Override
    protected void setUp() throws Exception {
        super.setUp();
        ms_clear = new ScriptC_clear_object(mRS);
    }

    @Override
    public void forEach(int testId, Allocation mIn, Allocation mOut) throws RSRuntimeException {
        switch (testId) {
        case TEST_ID_ELEMENT:
            ms_clear.forEach_clear_element(mOut);
            break;
        case TEST_ID_TYPE:
            ms_clear.forEach_clear_type(mOut);
            break;
        case TEST_ID_ALLOCATION:
            ms_clear.forEach_clear_allocation(mOut);
            break;
        case TEST_ID_SAMPLER:
            ms_clear.forEach_clear_sampler(mOut);
            break;
        case TEST_ID_SCRIPT:
            ms_clear.forEach_clear_script(mOut);
            break;
        }
    }

    public void testClearObjectElement() {
        Element element = Element.BOOLEAN(mRS);
        Allocation mIn = Allocation.createSized(mRS, Element.I32(mRS), ObjectNum);
        Allocation mOut = Allocation.createSized(mRS, Element.I32(mRS), ObjectNum);
        ms_clear.set_element(element);
        try {
            forEach(TEST_ID_ELEMENT, mIn, mOut);
        } catch (RSRuntimeException e) {
        }
        int[] tmpArray = new int[ObjectNum];
        mOut.copyTo(tmpArray);

        for(int i = 0; i < ObjectNum; i++)
            assertEquals(tmpArray[i], 1);
    }

    public void testclearObjectType() {
        Type type= new Type.Builder(mRS, Element.I8(mRS)).setX(1).create();
        Allocation mOut = Allocation.createSized(mRS, Element.I32(mRS), ObjectNum);
        Allocation mIn = Allocation.createSized(mRS, Element.I32(mRS), ObjectNum);
        ms_clear.set_type(type);

        try {
            forEach(TEST_ID_TYPE, mIn, mOut);
        } catch (RSRuntimeException e) {
        }
        int[] tmpArray = new int[ObjectNum];
        mOut.copyTo(tmpArray);

        for(int i = 0; i < ObjectNum; i++)
            assertEquals(tmpArray[i], 1);
    }

    public void testclearObjectAllocation() {
        Allocation mOut = Allocation.createSized(mRS, Element.I32(mRS), ObjectNum);
        Allocation mIn = Allocation.createSized(mRS, Element.I32(mRS), ObjectNum);
        Allocation allocation = Allocation.createTyped(mRS, mIn.getType());
        ms_clear.set_allocation(allocation);
        try {
            forEach(TEST_ID_ALLOCATION, mIn, mOut);
        } catch (RSRuntimeException e) {
        }
        int[] tmpArray = new int[ObjectNum];
        mOut.copyTo(tmpArray);

        for(int i = 0; i < ObjectNum; i++)
            assertEquals(tmpArray[i], 1);
    }

    public void testclearObjectSampler() {
        Sampler sampler = new Sampler.Builder(mRS).create();
        Allocation mOut = Allocation.createSized(mRS, Element.I32(mRS), ObjectNum);
        Allocation mIn = Allocation.createSized(mRS, Element.I32(mRS), ObjectNum);
        ms_clear.set_sampler(sampler);
        try {
            forEach(TEST_ID_SAMPLER, mIn, mOut);
        } catch (RSRuntimeException e) {
        }
        int[] tmpArray = new int[ObjectNum];
        mOut.copyTo(tmpArray);

        for(int i = 0; i < ObjectNum; i++)
            assertEquals(tmpArray[i], 1);
    }

    public void testclearObjectScript() {
        Script script = new ScriptC_clear_object(mRS);
        Allocation mIn = Allocation.createSized(mRS, Element.I32(mRS), ObjectNum);
        Allocation mOut = Allocation.createSized(mRS, Element.I32(mRS), ObjectNum);
        ms_clear.set_script(script);
        try {
            forEach(TEST_ID_SCRIPT, mIn, mOut);
        } catch (RSRuntimeException e) {
        }
        int[] tmpArray = new int[ObjectNum];
        mOut.copyTo(tmpArray);

        for(int i = 0; i < ObjectNum; i++)
            assertEquals(tmpArray[i], 1);
    }
}