com.gempukku.libgdx.graph.plugin.models.attribute.AttributeTangentShaderNodeBuilder Maven / Gradle / Ivy

Go to download
package com.gempukku.libgdx.graph.plugin.models.attribute;

import com.badlogic.gdx.graphics.VertexAttribute;
import com.badlogic.gdx.graphics.glutils.ShaderProgram;
import com.badlogic.gdx.utils.JsonValue;
import com.badlogic.gdx.utils.ObjectMap;
import com.badlogic.gdx.utils.ObjectSet;
import com.gempukku.libgdx.graph.plugin.models.ModelsUniformSetters;
import com.gempukku.libgdx.graph.plugin.models.config.attribute.AttributeTangentShaderNodeConfiguration;
import com.gempukku.libgdx.graph.shader.GraphShader;
import com.gempukku.libgdx.graph.shader.GraphShaderContext;
import com.gempukku.libgdx.graph.shader.ShaderFieldType;
import com.gempukku.libgdx.graph.shader.builder.FragmentShaderBuilder;
import com.gempukku.libgdx.graph.shader.builder.VertexShaderBuilder;
import com.gempukku.libgdx.graph.shader.node.ConfigurationShaderNodeBuilder;
import com.gempukku.libgdx.graph.shader.node.DefaultFieldOutput;
import com.gempukku.libgdx.graph.util.LibGDXCollections;

public class AttributeTangentShaderNodeBuilder extends ConfigurationShaderNodeBuilder {
    public AttributeTangentShaderNodeBuilder() {
        super(new AttributeTangentShaderNodeConfiguration());
    }

    @Override
    public ObjectMap buildVertexNodeSingleInputs(boolean designTime, String nodeId, JsonValue data, ObjectMap inputs, ObjectSet producedOutputs, VertexShaderBuilder vertexShaderBuilder, GraphShaderContext graphShaderContext, GraphShader graphShader) {
        vertexShaderBuilder.addAttributeVariable(VertexAttribute.Tangent(), ShaderProgram.TANGENT_ATTRIBUTE, "vec3");

        String coordinates = data.getString("coordinates");
        if (coordinates.equals("world")) {
            vertexShaderBuilder.addMainLine("// Attribute Tangent Node");
            vertexShaderBuilder.addUniformVariable("u_worldTrans", "mat4", false, ModelsUniformSetters.worldTrans);
            String name = "result_" + nodeId;
            vertexShaderBuilder.addMainLine("vec3" + " " + name + " = normalize((u_worldTrans * skinning * vec4(a_tangent, 0.0)).xyz);");

            return LibGDXCollections.singletonMap("tangent", new DefaultFieldOutput(ShaderFieldType.Vector3, name));
        } else if (coordinates.equals("object")) {
            String name = "result_" + nodeId;
            vertexShaderBuilder.addMainLine("vec3" + " " + name + " = normalize((skinning * vec4(a_tangent, 0.0)).xyz);");
            return LibGDXCollections.singletonMap("tangent", new DefaultFieldOutput(ShaderFieldType.Vector3, name));
        }
        throw new IllegalArgumentException();
    }

    @Override
    public ObjectMap buildFragmentNodeSingleInputs(boolean designTime, String nodeId, JsonValue
            data, ObjectMap inputs, ObjectSet producedOutputs, VertexShaderBuilder
                                                                                          vertexShaderBuilder, FragmentShaderBuilder fragmentShaderBuilder, GraphShaderContext
                                                                                          graphShaderContext, GraphShader graphShader) {
        vertexShaderBuilder.addAttributeVariable(VertexAttribute.Tangent(), ShaderProgram.TANGENT_ATTRIBUTE, "vec3");

        String coordinates = data.getString("coordinates");
        if (coordinates.equals("world")) {
            if (!vertexShaderBuilder.hasVaryingVariable("v_tangent_world")) {
                vertexShaderBuilder.addMainLine("// Attribute Tangent Node");
                vertexShaderBuilder.addUniformVariable("u_worldTrans", "mat4", false, ModelsUniformSetters.worldTrans);
                vertexShaderBuilder.addVaryingVariable("v_tangent_world", "vec3");
                vertexShaderBuilder.addMainLine("v_tangent_world = normalize((u_worldTrans * skinning * vec4(a_tangent, 0.0)).xyz);");

                fragmentShaderBuilder.addVaryingVariable("v_tangent_world", "vec3");
            }

            return LibGDXCollections.singletonMap("tangent", new DefaultFieldOutput(ShaderFieldType.Vector3, "v_tangent_world"));
        } else if (coordinates.equals("object")) {
            if (!vertexShaderBuilder.hasVaryingVariable("v_tangent_object")) {
                vertexShaderBuilder.addMainLine("// Attribute Tangent Node");
                vertexShaderBuilder.addVaryingVariable("v_tangent_object", "vec3");
                vertexShaderBuilder.addMainLine("v_tangent_object = normalize((skinning * vec4(a_tangent, 0.0)).xyz);");

                fragmentShaderBuilder.addVaryingVariable("v_tangent_object", "vec3");
            }

            return LibGDXCollections.singletonMap("tangent", new DefaultFieldOutput(ShaderFieldType.Vector3, "v_tangent_object"));
        }
        throw new IllegalArgumentException();
    }
}