java/tests/VrDemo/src/com/example/android/rs/vr/loaders/Mandelbulb.java

/*
 * Copyright (C) 2015 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.example.android.rs.vr.loaders;

import android.renderscript.Allocation;
import android.renderscript.RenderScript;
import android.renderscript.Type;
import android.util.Log;

import com.example.android.rs.vr.engine.ScriptC_mandelbulb;
import com.example.android.rs.vr.engine.Volume;

/**
 * Provides a simple an example of a computed data set and allows the application to
 * be run without any data sets.
 */
public class Mandelbulb {
    private static final String LOGTAG = "RawLoader";
    private static final String simpleLook = "green";
    private static final int[][] simpleOpacity = {{120, 0x0}, {140, 0xFF}};
    private static final int[][] simpleColor = {
            {144, 0xA4C639, 10, 80, 0},
            {155, 0xA4C639, 10, 60, 0},
            {300, 0xAA5555, 40, 60, 0},
            {255, 0xAAAAAA, 10, 80, 0}};


    private static final String tranlLook = "purple";
    private static final int[][] tranOpacity = {{110, 0x0},{140, 0x13},{143, 0x0}, {400, 0xFF}};
    private static final int[][] tranColor = {
            {144, 0xA4C639, 70, 30, 0},
            {230, 0xAA44AA, 70, 30, 0},
            {300, 0xAA5555, 70, 30, 20},
            {400, 0xAAAAAA, 70, 30, 20}};

    private static final int SIZE = 256;
    public static final String NAME = "A Mandelbulb";

    public static Volume buildRSVolume(RenderScript rs,
                                       final VolumeLoader.ProgressListener listener) {
        ScriptC_mandelbulb scriptC_mandelbulb = new ScriptC_mandelbulb(rs);

        Volume v = new Volume();
        v.mDimx = v.mDimy = v.mDimz = SIZE;
        v.mVoxelDim[0] = v.mVoxelDim[1] = v.mVoxelDim[2] = 1.f;

        v.addLook(simpleLook, simpleColor, simpleOpacity);
        v.addLook(tranlLook, tranColor, tranOpacity);

        Type.Builder b = new Type.Builder(rs, android.renderscript.Element.I16(rs));
        b.setX(v.mDimx).setY(v.mDimy);
        Allocation tmp = Allocation.createTyped(rs, b.create(), Allocation.USAGE_SCRIPT);
        b.setZ(v.mDimz);
        b.setX(v.mDimx).setY(v.mDimy).setZ(v.mDimz);
        v.mVolumeAllocation = Allocation.createTyped(rs, b.create(), Allocation.USAGE_SCRIPT);

        scriptC_mandelbulb.set_volume(v.mVolumeAllocation);
        scriptC_mandelbulb.set_size(SIZE);
        long time = System.nanoTime();
        for (int z = 0; z < v.mDimz; z++) {
            scriptC_mandelbulb.set_z(z);
            scriptC_mandelbulb.forEach_mandelbulb(tmp);
            scriptC_mandelbulb.forEach_copy(tmp);
            rs.finish();
            listener.progress(z, v.mDimz);
        }

        Log.v(LOGTAG, "compute Mandelbulb in" + ((System.nanoTime() - time) / 1E9f) + "seconds");
        tmp.destroy();
        scriptC_mandelbulb.destroy();
        return v;
    }

}