API cleanup for renderscript. This will be a multiproject commit.
Change-Id: Ida62d3a155285a20725be9daa62217faef1c3734
This commit is contained in:
@ -0,0 +1,237 @@
|
||||
/*
|
||||
* Copyright (C) 2008 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;
|
||||
|
||||
|
||||
import android.util.Config;
|
||||
import android.util.Log;
|
||||
|
||||
|
||||
/**
|
||||
* @hide
|
||||
*
|
||||
**/
|
||||
public class ProgramFragmentFixedFunction extends ProgramFragment {
|
||||
ProgramFragmentFixedFunction(int id, RenderScript rs) {
|
||||
super(id, rs);
|
||||
}
|
||||
|
||||
static class InternalBuilder extends BaseProgramBuilder {
|
||||
public InternalBuilder(RenderScript rs) {
|
||||
super(rs);
|
||||
}
|
||||
|
||||
public ProgramFragmentFixedFunction create() {
|
||||
mRS.validate();
|
||||
int[] tmp = new int[(mInputCount + mOutputCount + mConstantCount + mTextureCount) * 2];
|
||||
int idx = 0;
|
||||
|
||||
for (int i=0; i < mInputCount; i++) {
|
||||
tmp[idx++] = ProgramParam.INPUT.mID;
|
||||
tmp[idx++] = mInputs[i].getID();
|
||||
}
|
||||
for (int i=0; i < mOutputCount; i++) {
|
||||
tmp[idx++] = ProgramParam.OUTPUT.mID;
|
||||
tmp[idx++] = mOutputs[i].getID();
|
||||
}
|
||||
for (int i=0; i < mConstantCount; i++) {
|
||||
tmp[idx++] = ProgramParam.CONSTANT.mID;
|
||||
tmp[idx++] = mConstants[i].getID();
|
||||
}
|
||||
for (int i=0; i < mTextureCount; i++) {
|
||||
tmp[idx++] = ProgramParam.TEXTURE_TYPE.mID;
|
||||
tmp[idx++] = mTextureTypes[i].mID;
|
||||
}
|
||||
|
||||
int id = mRS.nProgramFragmentCreate(mShader, tmp);
|
||||
ProgramFragmentFixedFunction pf = new ProgramFragmentFixedFunction(id, mRS);
|
||||
initProgram(pf);
|
||||
return pf;
|
||||
}
|
||||
}
|
||||
|
||||
public static class Builder {
|
||||
public static final int MAX_TEXTURE = 2;
|
||||
int mNumTextures;
|
||||
boolean mPointSpriteEnable;
|
||||
boolean mVaryingColorEnable;
|
||||
String mShader;
|
||||
RenderScript mRS;
|
||||
|
||||
public enum EnvMode {
|
||||
REPLACE (1),
|
||||
MODULATE (2),
|
||||
DECAL (3);
|
||||
|
||||
int mID;
|
||||
EnvMode(int id) {
|
||||
mID = id;
|
||||
}
|
||||
}
|
||||
|
||||
public enum Format {
|
||||
ALPHA (1),
|
||||
LUMINANCE_ALPHA (2),
|
||||
RGB (3),
|
||||
RGBA (4);
|
||||
|
||||
int mID;
|
||||
Format(int id) {
|
||||
mID = id;
|
||||
}
|
||||
}
|
||||
|
||||
private class Slot {
|
||||
EnvMode env;
|
||||
Format format;
|
||||
Slot(EnvMode _env, Format _fmt) {
|
||||
env = _env;
|
||||
format = _fmt;
|
||||
}
|
||||
}
|
||||
Slot[] mSlots;
|
||||
|
||||
private void buildShaderString() {
|
||||
mShader = "//rs_shader_internal\n";
|
||||
mShader += "varying lowp vec4 varColor;\n";
|
||||
mShader += "varying vec2 varTex0;\n";
|
||||
|
||||
mShader += "void main() {\n";
|
||||
if (mVaryingColorEnable) {
|
||||
mShader += " lowp vec4 col = varColor;\n";
|
||||
} else {
|
||||
mShader += " lowp vec4 col = UNI_Color;\n";
|
||||
}
|
||||
|
||||
if (mNumTextures != 0) {
|
||||
if (mPointSpriteEnable) {
|
||||
mShader += " vec2 t0 = gl_PointCoord;\n";
|
||||
} else {
|
||||
mShader += " vec2 t0 = varTex0.xy;\n";
|
||||
}
|
||||
}
|
||||
|
||||
for(int i = 0; i < mNumTextures; i ++) {
|
||||
switch(mSlots[i].env) {
|
||||
case REPLACE:
|
||||
switch (mSlots[i].format) {
|
||||
case ALPHA:
|
||||
mShader += " col.a = texture2D(UNI_Tex0, t0).a;\n";
|
||||
break;
|
||||
case LUMINANCE_ALPHA:
|
||||
mShader += " col.rgba = texture2D(UNI_Tex0, t0).rgba;\n";
|
||||
break;
|
||||
case RGB:
|
||||
mShader += " col.rgb = texture2D(UNI_Tex0, t0).rgb;\n";
|
||||
break;
|
||||
case RGBA:
|
||||
mShader += " col.rgba = texture2D(UNI_Tex0, t0).rgba;\n";
|
||||
break;
|
||||
}
|
||||
break;
|
||||
case MODULATE:
|
||||
switch (mSlots[i].format) {
|
||||
case ALPHA:
|
||||
mShader += " col.a *= texture2D(UNI_Tex0, t0).a;\n";
|
||||
break;
|
||||
case LUMINANCE_ALPHA:
|
||||
mShader += " col.rgba *= texture2D(UNI_Tex0, t0).rgba;\n";
|
||||
break;
|
||||
case RGB:
|
||||
mShader += " col.rgb *= texture2D(UNI_Tex0, t0).rgb;\n";
|
||||
break;
|
||||
case RGBA:
|
||||
mShader += " col.rgba *= texture2D(UNI_Tex0, t0).rgba;\n";
|
||||
break;
|
||||
}
|
||||
break;
|
||||
case DECAL:
|
||||
mShader += " col = texture2D(UNI_Tex0, t0);\n";
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
mShader += " gl_FragColor = col;\n";
|
||||
mShader += "}\n";
|
||||
}
|
||||
|
||||
public Builder(RenderScript rs) {
|
||||
mRS = rs;
|
||||
mSlots = new Slot[MAX_TEXTURE];
|
||||
mPointSpriteEnable = false;
|
||||
}
|
||||
|
||||
public Builder setTexture(EnvMode env, Format fmt, int slot)
|
||||
throws IllegalArgumentException {
|
||||
if((slot < 0) || (slot >= MAX_TEXTURE)) {
|
||||
throw new IllegalArgumentException("MAX_TEXTURE exceeded.");
|
||||
}
|
||||
mSlots[slot] = new Slot(env, fmt);
|
||||
return this;
|
||||
}
|
||||
|
||||
public Builder setPointSpriteTexCoordinateReplacement(boolean enable) {
|
||||
mPointSpriteEnable = enable;
|
||||
return this;
|
||||
}
|
||||
|
||||
public Builder setVaryingColor(boolean enable) {
|
||||
mVaryingColorEnable = enable;
|
||||
return this;
|
||||
}
|
||||
|
||||
public ProgramFragmentFixedFunction create() {
|
||||
InternalBuilder sb = new InternalBuilder(mRS);
|
||||
mNumTextures = 0;
|
||||
for(int i = 0; i < MAX_TEXTURE; i ++) {
|
||||
if(mSlots[i] != null) {
|
||||
mNumTextures ++;
|
||||
}
|
||||
}
|
||||
buildShaderString();
|
||||
sb.setShader(mShader);
|
||||
|
||||
Type constType = null;
|
||||
if (!mVaryingColorEnable) {
|
||||
Element.Builder b = new Element.Builder(mRS);
|
||||
b.add(Element.F32_4(mRS), "Color");
|
||||
Type.Builder typeBuilder = new Type.Builder(mRS, b.create());
|
||||
typeBuilder.setX(1);
|
||||
constType = typeBuilder.create();
|
||||
sb.addConstant(constType);
|
||||
}
|
||||
for (int i = 0; i < mNumTextures; i ++) {
|
||||
sb.addTexture(TextureType.TEXTURE_2D);
|
||||
}
|
||||
|
||||
ProgramFragmentFixedFunction pf = sb.create();
|
||||
pf.mTextureCount = MAX_TEXTURE;
|
||||
if (!mVaryingColorEnable) {
|
||||
Allocation constantData = Allocation.createTyped(mRS,constType);
|
||||
float[] data = new float[4];
|
||||
data[0] = data[1] = data[2] = data[3] = 1.0f;
|
||||
constantData.copyFrom(data);
|
||||
pf.bindConstants(constantData, 0);
|
||||
}
|
||||
return pf;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
Reference in New Issue
Block a user