xref: /frameworks/base/tests/RenderScriptTests/SceneGraph/src/com/android/scenegraph/ShaderParam.java

/*
 * Copyright (C) 2011 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.scenegraph;

import java.lang.Math;
import java.util.ArrayList;
import java.util.HashMap;

import com.android.scenegraph.SceneManager;
import com.android.scenegraph.Transform;

import android.renderscript.Element;
import android.renderscript.Matrix4f;
import android.renderscript.ProgramFragment;
import android.renderscript.ProgramStore;
import android.renderscript.ProgramVertex;
import android.renderscript.RenderScriptGL;
import android.util.Log;

/**
 * @hide
 */
public abstract class ShaderParam extends SceneGraphBase {

    static final String cameraPos        = "cameraPos";
    static final String cameraDir        = "cameraDir";

    static final String lightColor       = "lightColor";
    static final String lightPos         = "lightPos";
    static final String lightDir         = "lightDir";

    static final String view             = "view";
    static final String proj             = "proj";
    static final String viewProj         = "viewProj";
    static final String model            = "model";
    static final String modelView        = "modelView";
    static final String modelViewProj    = "modelViewProj";

    static final long sMaxTimeStamp = 0xffffffffL;

    ScriptField_ShaderParamData_s.Item mData;
    ScriptField_ShaderParamData_s mField;

    String mParamName;
    Camera mCamera;

    static ScriptField_ShaderParam_s fillInParams(Element constantElem,
                                                  HashMap<String, ShaderParam> sourceParams,
                                                  Transform transform) {
        RenderScriptGL rs = SceneManager.getRS();
        ArrayList<ScriptField_ShaderParam_s.Item> paramList;
        paramList = new ArrayList<ScriptField_ShaderParam_s.Item>();

        int subElemCount = constantElem.getSubElementCount();
        for (int i = 0; i < subElemCount; i ++) {
            String inputName = constantElem.getSubElementName(i);
            int offset = constantElem.getSubElementOffsetBytes(i);

            ShaderParam matchingParam = sourceParams.get(inputName);
            Element subElem = constantElem.getSubElement(i);
            // Make one if it's not there
            if (matchingParam == null) {
                if (subElem.getDataType() == Element.DataType.FLOAT_32) {
                    matchingParam = new Float4Param(inputName, 0.5f, 0.5f, 0.5f, 0.5f);
                } else if (subElem.getDataType() == Element.DataType.MATRIX_4X4) {
                    TransformParam trParam = new TransformParam(inputName);
                    trParam.setTransform(transform);
                    matchingParam = trParam;
                }
            }
            ScriptField_ShaderParam_s.Item paramRS = new ScriptField_ShaderParam_s.Item();
            paramRS.bufferOffset = offset;
            paramRS.transformTimestamp = 0;
            paramRS.dataTimestamp = 0;
            paramRS.data = matchingParam.getRSData().getAllocation();
            if (subElem.getDataType() == Element.DataType.FLOAT_32) {
                paramRS.float_vecSize = subElem.getVectorSize();
            }

            paramList.add(paramRS);
        }

        ScriptField_ShaderParam_s rsParams = null;
        int paramCount = paramList.size();
        if (paramCount != 0) {
            rsParams = new ScriptField_ShaderParam_s(rs, paramCount);
            for (int i = 0; i < paramCount; i++) {
                rsParams.set(paramList.get(i), i, false);
            }
            rsParams.copyAll();
        }
        return rsParams;
    }

    public ShaderParam(String name) {
        mParamName = name;
        mData = new ScriptField_ShaderParamData_s.Item();
    }

    public String getParamName() {
        return mParamName;
    }

    public void setCamera(Camera c) {
        mCamera = c;
        if (mField != null) {
            mData.camera = mCamera.getRSData().getAllocation();
            mField.set_camera(0, mData.camera, true);
        }
    }

    protected void incTimestamp() {
        if (mField != null) {
            mData.timestamp ++;
            mData.timestamp %= sMaxTimeStamp;
            mField.set_timestamp(0, mData.timestamp, true);
        }
    }

    abstract void initLocalData();

    public ScriptField_ShaderParamData_s getRSData() {
        if (mField != null) {
            return mField;
        }

        RenderScriptGL rs = SceneManager.getRS();
        mField = new ScriptField_ShaderParamData_s(rs, 1);

        if (mParamName != null) {
            mData.paramName = SceneManager.getCachedAlloc(mParamName);
            if (mData.paramName == null) {
                mData.paramName = SceneManager.getStringAsAllocation(rs, mParamName);
                SceneManager.cacheAlloc(mParamName, mData.paramName);
            }
        }
        initLocalData();
        mData.timestamp = 1;

        mField.set(mData, 0, true);
        return mField;
    }
}