libs/rs/java/Grass/src/com/android/grass/rs/GrassRS.java - platform/frameworks/base - Git at Google

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

 import android.content.res.Resources;
 import android.renderscript.Sampler;
 import static android.renderscript.ProgramFragment.EnvMode.*;
 import static android.renderscript.ProgramStore.DepthFunc.*;
 import static android.renderscript.ProgramStore.BlendSrcFunc;
 import static android.renderscript.ProgramStore.BlendDstFunc;
 import android.renderscript.RenderScript;
 import android.renderscript.ProgramFragment;
 import android.renderscript.ProgramStore;
 import android.renderscript.Allocation;
 import android.renderscript.ProgramVertexAlloc;
 import static android.renderscript.Element.*;
 import static android.util.MathUtils.*;
 import android.graphics.Bitmap;
 import android.graphics.BitmapFactory;
 import android.renderscript.ScriptC;
 import static android.renderscript.Sampler.Value.*;

 import java.util.TimeZone;

 class GrassRS {
     private static final int RSID_STATE = 0;
     private static final int RSID_STATE_FRAMECOUNT = 0;
     private static final int RSID_STATE_BLADES_COUNT = 1;
     private static final int RSID_STATE_WIDTH = 2;
     private static final int RSID_STATE_HEIGHT = 3;

     private static final int RSID_TEXTURES = 1;
     private static final int TEXTURES_COUNT = 5;

     private static final int RSID_BLADES = 2;
     private static final int BLADES_COUNT = 100;
     private static final int BLADE_STRUCT_FIELDS_COUNT = 12;
     private static final int BLADE_STRUCT_ANGLE = 0;
     private static final int BLADE_STRUCT_SIZE = 1;
     private static final int BLADE_STRUCT_XPOS = 2;
     private static final int BLADE_STRUCT_YPOS = 3;
     private static final int BLADE_STRUCT_OFFSET = 4;
     private static final int BLADE_STRUCT_SCALE = 5;
     private static final int BLADE_STRUCT_LENGTHX = 6;
     private static final int BLADE_STRUCT_LENGTHY = 7;
     private static final int BLADE_STRUCT_HARDNESS = 8;
     private static final int BLADE_STRUCT_H = 9;
     private static final int BLADE_STRUCT_S = 10;
     private static final int BLADE_STRUCT_B = 11;

     private Resources mResources;
     private RenderScript mRS;
     private final BitmapFactory.Options mBitmapOptions = new BitmapFactory.Options();

     private final int mWidth;
     private final int mHeight;

     private ScriptC mScript;
     private Sampler mSampler;
     private ProgramFragment mPfBackground;
     private ProgramStore mPfsBackground;
     private RenderScript.ProgramVertex mPvBackground;
     private ProgramVertexAlloc mPvOrthoAlloc;

     private Allocation mTexturesIDs;
     private Allocation[] mTextures;
     private int[] mTextureBufferIDs;

     private Allocation mState;
     private Allocation mBlades;

     private ProgramFragment mPfGrass;
     private ProgramStore mPfsGrass;

     public GrassRS(int width, int height) {
         mWidth = width;
         mHeight = height;
         mBitmapOptions.inScaled = false;
         mBitmapOptions.inPreferredConfig = Bitmap.Config.ARGB_8888;
     }

     public void init(RenderScript rs, Resources res) {
         mRS = rs;
         mResources = res;
         initRS();
     }

     public void destroy() {
         mScript.destroy();
         mSampler.destroy();
         mPfBackground.destroy();
         mPfsBackground.destroy();
         mPvBackground.destroy();
         mPvOrthoAlloc.mAlloc.destroy();
         mTexturesIDs.destroy();
         for (Allocation a : mTextures) {
             a.destroy();
         }
         mState.destroy();
         mBlades.destroy();
         mPfGrass.destroy();
         mPfsGrass.destroy();
         mTextureBufferIDs = null;
     }

     @Override
     protected void finalize() throws Throwable {
         try {
             destroy();
         } finally {
             super.finalize();
         }
     }

     private void initRS() {
         createProgramVertex();
         createProgramFragmentStore();
         createProgramFragment();
         createScriptStructures();

         ScriptC.Builder sb = new ScriptC.Builder(mRS);
         sb.setScript(mResources, R.raw.grass);
         sb.setRoot(true);
         mScript = sb.create();
         mScript.setClearColor(0.0f, 0.0f, 0.0f, 1.0f);
         mScript.setTimeZone(TimeZone.getDefault().getID());

         loadSkyTextures();
         mScript.bindAllocation(mState, RSID_STATE);
         mScript.bindAllocation(mTexturesIDs, RSID_TEXTURES);
         mScript.bindAllocation(mBlades, RSID_BLADES);

         mRS.contextBindRootScript(mScript);
     }

     private void createScriptStructures() {
         final int[] data = new int[4];
         mState = Allocation.createSized(mRS, USER_I32, data.length);
         data[RSID_STATE_FRAMECOUNT] = 0;
         data[RSID_STATE_BLADES_COUNT] = BLADES_COUNT;
         data[RSID_STATE_WIDTH] = mWidth;
         data[RSID_STATE_HEIGHT] = mHeight;
         mState.data(data);

         final float[] blades = new float[BLADES_COUNT * BLADE_STRUCT_FIELDS_COUNT];
         mBlades = Allocation.createSized(mRS, USER_FLOAT, blades.length);
         for (int i = 0; i < blades.length; i+= BLADE_STRUCT_FIELDS_COUNT) {
             createBlade(blades, i);
         }
         mBlades.data(blades);
     }

     private void createBlade(float[] blades, int index) {
         //noinspection PointlessArithmeticExpression
         blades[index + BLADE_STRUCT_ANGLE] = 0.0f;
         blades[index + BLADE_STRUCT_SIZE] = random(4.0f) + 4.0f;
         blades[index + BLADE_STRUCT_XPOS] = random(mWidth);
         blades[index + BLADE_STRUCT_YPOS] = mHeight;
         blades[index + BLADE_STRUCT_OFFSET] = random(0.2f) - 0.1f;
         blades[index + BLADE_STRUCT_SCALE] = random(0.6f) + 0.2f;
         blades[index + BLADE_STRUCT_LENGTHX] = random(4.5f) + 3.0f;
         blades[index + BLADE_STRUCT_LENGTHY] = random(5.5f) + 2.0f;
         blades[index + BLADE_STRUCT_HARDNESS] = random(1.0f) + 0.2f;
         blades[index + BLADE_STRUCT_H] = random(0.02f) + 0.2f;
         blades[index + BLADE_STRUCT_S] = random(0.22f) + 0.78f;
         blades[index + BLADE_STRUCT_B] = random(0.65f) + 0.35f;
     }

     private void loadSkyTextures() {
         mTextureBufferIDs = new int[TEXTURES_COUNT];
         mTextures = new Allocation[TEXTURES_COUNT];
         mTexturesIDs = Allocation.createSized(mRS, USER_FLOAT, TEXTURES_COUNT);

         final Allocation[] textures = mTextures;
         textures[0] = loadTexture(R.drawable.night, "night");
         textures[1] = loadTexture(R.drawable.sunrise, "sunrise");
         textures[2] = loadTexture(R.drawable.sky, "sky");
         textures[3] = loadTexture(R.drawable.sunset, "sunset");
         textures[4] = loadTextureARGB(R.drawable.aa, "aa");

         final int[] bufferIds = mTextureBufferIDs;
         final int count = textures.length;

         for (int i = 0; i < count; i++) {
             final Allocation texture = textures[i];
             texture.uploadToTexture(0);
             bufferIds[i] = texture.getID();
         }

         mTexturesIDs.data(bufferIds);
     }

     private Allocation loadTexture(int id, String name) {
         final Allocation allocation = Allocation.createFromBitmapResource(mRS, mResources,
                 id, RGB_565, false);
         allocation.setName(name);
         return allocation;
     }

     private Allocation loadTextureARGB(int id, String name) {
         // Forces ARGB 32 bits, because pngcrush sometimes optimize our PNGs to
         // indexed pictures, which are not well supported
         final Bitmap b = BitmapFactory.decodeResource(mResources, id, mBitmapOptions);
         final Allocation allocation = Allocation.createFromBitmap(mRS, b, RGBA_8888, false);
         allocation.setName(name);
         return allocation;
     }

     private void createProgramFragment() {
         Sampler.Builder bs = new Sampler.Builder(mRS);
         bs.setMin(LINEAR);
         bs.setMag(LINEAR);
         bs.setWrapS(CLAMP);
         bs.setWrapT(CLAMP);
         mSampler = bs.create();

         ProgramFragment.Builder b;
         b = new ProgramFragment.Builder(mRS, null, null);
         b.setTexEnable(true, 0);
         b.setTexEnvMode(REPLACE, 0);
         mPfBackground = b.create();
         mPfBackground.setName("PFBackground");
         mPfBackground.bindSampler(mSampler, 0);

         b.setTexEnvMode(MODULATE, 0);
         mPfGrass = b.create();
         mPfGrass.setName("PFGrass");
         mPfGrass.bindSampler(mSampler, 0);
     }

     private void createProgramFragmentStore() {
         ProgramStore.Builder b;
         b = new ProgramStore.Builder(mRS, null, null);

         b.setDepthFunc(ALWAYS);
         b.setBlendFunc(BlendSrcFunc.SRC_ALPHA, BlendDstFunc.ONE_MINUS_SRC_ALPHA);
         b.setDitherEnable(true);
         b.setDepthMask(false);
         mPfsBackground = b.create();
         mPfsBackground.setName("PFSBackground");

         b.setBlendFunc(BlendSrcFunc.ONE, BlendDstFunc.ONE_MINUS_SRC_ALPHA);
         mPfsGrass = b.create();
         mPfsGrass.setName("PFSGrass");
     }

     private void createProgramVertex() {
         mPvOrthoAlloc = new ProgramVertexAlloc(mRS);
         mPvOrthoAlloc.setupOrthoWindow(mWidth, mHeight);

         mRS.programVertexBegin(null, null);
         mRS.programVertexSetTextureMatrixEnable(true);
         mPvBackground = mRS.programVertexCreate();
         mPvBackground.bindAllocation(0, mPvOrthoAlloc.mAlloc);
         mPvBackground.setName("PVBackground");
     }
 }
	/*
	* Copyright (C) 2009 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.grass.rs;

	import android.content.res.Resources;
	import android.renderscript.Sampler;
	import static android.renderscript.ProgramFragment.EnvMode.*;
	import static android.renderscript.ProgramStore.DepthFunc.*;
	import static android.renderscript.ProgramStore.BlendSrcFunc;
	import static android.renderscript.ProgramStore.BlendDstFunc;
	import android.renderscript.RenderScript;
	import android.renderscript.ProgramFragment;
	import android.renderscript.ProgramStore;
	import android.renderscript.Allocation;
	import android.renderscript.ProgramVertexAlloc;
	import static android.renderscript.Element.*;
	import static android.util.MathUtils.*;
	import android.graphics.Bitmap;
	import android.graphics.BitmapFactory;
	import android.renderscript.ScriptC;
	import static android.renderscript.Sampler.Value.*;

	import java.util.TimeZone;

	class GrassRS {
	private static final int RSID_STATE = 0;
	private static final int RSID_STATE_FRAMECOUNT = 0;
	private static final int RSID_STATE_BLADES_COUNT = 1;
	private static final int RSID_STATE_WIDTH = 2;
	private static final int RSID_STATE_HEIGHT = 3;

	private static final int RSID_TEXTURES = 1;
	private static final int TEXTURES_COUNT = 5;

	private static final int RSID_BLADES = 2;
	private static final int BLADES_COUNT = 100;
	private static final int BLADE_STRUCT_FIELDS_COUNT = 12;
	private static final int BLADE_STRUCT_ANGLE = 0;
	private static final int BLADE_STRUCT_SIZE = 1;
	private static final int BLADE_STRUCT_XPOS = 2;
	private static final int BLADE_STRUCT_YPOS = 3;
	private static final int BLADE_STRUCT_OFFSET = 4;
	private static final int BLADE_STRUCT_SCALE = 5;
	private static final int BLADE_STRUCT_LENGTHX = 6;
	private static final int BLADE_STRUCT_LENGTHY = 7;
	private static final int BLADE_STRUCT_HARDNESS = 8;
	private static final int BLADE_STRUCT_H = 9;
	private static final int BLADE_STRUCT_S = 10;
	private static final int BLADE_STRUCT_B = 11;

	private Resources mResources;
	private RenderScript mRS;
	private final BitmapFactory.Options mBitmapOptions = new BitmapFactory.Options();

	private final int mWidth;
	private final int mHeight;

	private ScriptC mScript;
	private Sampler mSampler;
	private ProgramFragment mPfBackground;
	private ProgramStore mPfsBackground;
	private RenderScript.ProgramVertex mPvBackground;
	private ProgramVertexAlloc mPvOrthoAlloc;

	private Allocation mTexturesIDs;
	private Allocation[] mTextures;
	private int[] mTextureBufferIDs;

	private Allocation mState;
	private Allocation mBlades;

	private ProgramFragment mPfGrass;
	private ProgramStore mPfsGrass;

	public GrassRS(int width, int height) {
	mWidth = width;
	mHeight = height;
	mBitmapOptions.inScaled = false;
	mBitmapOptions.inPreferredConfig = Bitmap.Config.ARGB_8888;
	}

	public void init(RenderScript rs, Resources res) {
	mRS = rs;
	mResources = res;
	initRS();
	}

	public void destroy() {
	mScript.destroy();
	mSampler.destroy();
	mPfBackground.destroy();
	mPfsBackground.destroy();
	mPvBackground.destroy();
	mPvOrthoAlloc.mAlloc.destroy();
	mTexturesIDs.destroy();
	for (Allocation a : mTextures) {
	a.destroy();
	}
	mState.destroy();
	mBlades.destroy();
	mPfGrass.destroy();
	mPfsGrass.destroy();
	mTextureBufferIDs = null;
	}

	@Override
	protected void finalize() throws Throwable {
	try {
	destroy();
	} finally {
	super.finalize();
	}
	}

	private void initRS() {
	createProgramVertex();
	createProgramFragmentStore();
	createProgramFragment();
	createScriptStructures();

	ScriptC.Builder sb = new ScriptC.Builder(mRS);
	sb.setScript(mResources, R.raw.grass);
	sb.setRoot(true);
	mScript = sb.create();
	mScript.setClearColor(0.0f, 0.0f, 0.0f, 1.0f);
	mScript.setTimeZone(TimeZone.getDefault().getID());

	loadSkyTextures();
	mScript.bindAllocation(mState, RSID_STATE);
	mScript.bindAllocation(mTexturesIDs, RSID_TEXTURES);
	mScript.bindAllocation(mBlades, RSID_BLADES);

	mRS.contextBindRootScript(mScript);
	}

	private void createScriptStructures() {
	final int[] data = new int[4];
	mState = Allocation.createSized(mRS, USER_I32, data.length);
	data[RSID_STATE_FRAMECOUNT] = 0;
	data[RSID_STATE_BLADES_COUNT] = BLADES_COUNT;
	data[RSID_STATE_WIDTH] = mWidth;
	data[RSID_STATE_HEIGHT] = mHeight;
	mState.data(data);

	final float[] blades = new float[BLADES_COUNT * BLADE_STRUCT_FIELDS_COUNT];
	mBlades = Allocation.createSized(mRS, USER_FLOAT, blades.length);
	for (int i = 0; i < blades.length; i+= BLADE_STRUCT_FIELDS_COUNT) {
	createBlade(blades, i);
	}
	mBlades.data(blades);
	}

	private void createBlade(float[] blades, int index) {
	//noinspection PointlessArithmeticExpression
	blades[index + BLADE_STRUCT_ANGLE] = 0.0f;
	blades[index + BLADE_STRUCT_SIZE] = random(4.0f) + 4.0f;
	blades[index + BLADE_STRUCT_XPOS] = random(mWidth);
	blades[index + BLADE_STRUCT_YPOS] = mHeight;
	blades[index + BLADE_STRUCT_OFFSET] = random(0.2f) - 0.1f;
	blades[index + BLADE_STRUCT_SCALE] = random(0.6f) + 0.2f;
	blades[index + BLADE_STRUCT_LENGTHX] = random(4.5f) + 3.0f;
	blades[index + BLADE_STRUCT_LENGTHY] = random(5.5f) + 2.0f;
	blades[index + BLADE_STRUCT_HARDNESS] = random(1.0f) + 0.2f;
	blades[index + BLADE_STRUCT_H] = random(0.02f) + 0.2f;
	blades[index + BLADE_STRUCT_S] = random(0.22f) + 0.78f;
	blades[index + BLADE_STRUCT_B] = random(0.65f) + 0.35f;
	}

	private void loadSkyTextures() {
	mTextureBufferIDs = new int[TEXTURES_COUNT];
	mTextures = new Allocation[TEXTURES_COUNT];
	mTexturesIDs = Allocation.createSized(mRS, USER_FLOAT, TEXTURES_COUNT);

	final Allocation[] textures = mTextures;
	textures[0] = loadTexture(R.drawable.night, "night");
	textures[1] = loadTexture(R.drawable.sunrise, "sunrise");
	textures[2] = loadTexture(R.drawable.sky, "sky");
	textures[3] = loadTexture(R.drawable.sunset, "sunset");
	textures[4] = loadTextureARGB(R.drawable.aa, "aa");

	final int[] bufferIds = mTextureBufferIDs;
	final int count = textures.length;

	for (int i = 0; i < count; i++) {
	final Allocation texture = textures[i];
	texture.uploadToTexture(0);
	bufferIds[i] = texture.getID();
	}

	mTexturesIDs.data(bufferIds);
	}

	private Allocation loadTexture(int id, String name) {
	final Allocation allocation = Allocation.createFromBitmapResource(mRS, mResources,
	id, RGB_565, false);
	allocation.setName(name);
	return allocation;
	}

	private Allocation loadTextureARGB(int id, String name) {
	// Forces ARGB 32 bits, because pngcrush sometimes optimize our PNGs to
	// indexed pictures, which are not well supported
	final Bitmap b = BitmapFactory.decodeResource(mResources, id, mBitmapOptions);
	final Allocation allocation = Allocation.createFromBitmap(mRS, b, RGBA_8888, false);
	allocation.setName(name);
	return allocation;
	}

	private void createProgramFragment() {
	Sampler.Builder bs = new Sampler.Builder(mRS);
	bs.setMin(LINEAR);
	bs.setMag(LINEAR);
	bs.setWrapS(CLAMP);
	bs.setWrapT(CLAMP);
	mSampler = bs.create();

	ProgramFragment.Builder b;
	b = new ProgramFragment.Builder(mRS, null, null);
	b.setTexEnable(true, 0);
	b.setTexEnvMode(REPLACE, 0);
	mPfBackground = b.create();
	mPfBackground.setName("PFBackground");
	mPfBackground.bindSampler(mSampler, 0);

	b.setTexEnvMode(MODULATE, 0);
	mPfGrass = b.create();
	mPfGrass.setName("PFGrass");
	mPfGrass.bindSampler(mSampler, 0);
	}

	private void createProgramFragmentStore() {
	ProgramStore.Builder b;
	b = new ProgramStore.Builder(mRS, null, null);

	b.setDepthFunc(ALWAYS);
	b.setBlendFunc(BlendSrcFunc.SRC_ALPHA, BlendDstFunc.ONE_MINUS_SRC_ALPHA);
	b.setDitherEnable(true);
	b.setDepthMask(false);
	mPfsBackground = b.create();
	mPfsBackground.setName("PFSBackground");

	b.setBlendFunc(BlendSrcFunc.ONE, BlendDstFunc.ONE_MINUS_SRC_ALPHA);
	mPfsGrass = b.create();
	mPfsGrass.setName("PFSGrass");
	}

	private void createProgramVertex() {
	mPvOrthoAlloc = new ProgramVertexAlloc(mRS);
	mPvOrthoAlloc.setupOrthoWindow(mWidth, mHeight);

	mRS.programVertexBegin(null, null);
	mRS.programVertexSetTextureMatrixEnable(true);
	mPvBackground = mRS.programVertexCreate();
	mPvBackground.bindAllocation(0, mPvOrthoAlloc.mAlloc);
	mPvBackground.setName("PVBackground");
	}
	}