Examples with WhitePixel com.gempukku.libgdx.graph.util.WhitePixel used on opensource projects

Example 1 with WhitePixel

use of com.gempukku.libgdx.graph.util.WhitePixel in project gdx-graph by MarcinSc.

the class ModelShaderRendererPipelineNodeProducer method createNodeForSingleInputs.

@Override
public PipelineNode createNodeForSingleInputs(JsonValue data, ObjectMap<String, String> inputTypes, ObjectMap<String, String> outputTypes) {
    final ShaderContextImpl shaderContext = new ShaderContextImpl(pluginPrivateDataSource);
    final ObjectMap<String, ShaderGroup> shaderGroups = new ObjectMap<>();
    final Array<String> allShaderTags = new Array<>();
    final Array<String> depthDrawingShaderTags = new Array<>();
    final JsonValue shaderDefinitions = data.get("shaders");
    RenderOrder renderOrder = RenderOrder.valueOf(data.getString("renderOrder", "Shader_Unordered"));
    final ModelRenderingStrategy renderingStrategy = createRenderingStrategy(renderOrder);
    final RenderingStrategyCallback colorStrategyCallback = new RenderingStrategyCallback(shaderContext, new Function<String, GraphShader>() {

        @Override
        public GraphShader apply(String s) {
            return shaderGroups.get(s).getColorShader();
        }
    });
    final RenderingStrategyCallback depthStrategyCallback = new RenderingStrategyCallback(shaderContext, new Function<String, GraphShader>() {

        @Override
        public GraphShader apply(String s) {
            return shaderGroups.get(s).getDepthShader();
        }
    });
    final ObjectMap<String, PipelineNode.FieldOutput<?>> result = new ObjectMap<>();
    final DefaultFieldOutput<RenderPipeline> output = new DefaultFieldOutput<>(PipelineFieldType.RenderPipeline);
    result.put("output", output);
    return new SingleInputsPipelineNode(result) {

        private FullScreenRender fullScreenRender;

        private TimeProvider timeProvider;

        private GraphModelsImpl models;

        private WhitePixel whitePixel;

        @Override
        public void initializePipeline(PipelineDataProvider pipelineDataProvider) {
            fullScreenRender = pipelineDataProvider.getFullScreenRender();
            timeProvider = pipelineDataProvider.getTimeProvider();
            models = pipelineDataProvider.getPrivatePluginData(GraphModelsImpl.class);
            whitePixel = pipelineDataProvider.getWhitePixel();
            for (JsonValue shaderDefinition : shaderDefinitions) {
                ShaderGroup shaderGroup = new ShaderGroup(shaderDefinition);
                shaderGroup.initialize(whitePixel);
                allShaderTags.add(shaderGroup.getTag());
                shaderGroups.put(shaderGroup.getTag(), shaderGroup);
                if (shaderGroup.getColorShader().isDepthWriting())
                    depthDrawingShaderTags.add(shaderGroup.getTag());
            }
            for (ShaderGroup shaderGroup : shaderGroups.values()) {
                GraphShader shader = shaderGroup.getColorShader();
                models.registerTag(shader.getTag(), shader);
            }
        }

        private void initializeDepthShaders() {
            for (String depthDrawingShaderTag : depthDrawingShaderTags) {
                shaderGroups.get(depthDrawingShaderTag).initializeDepthShader(whitePixel);
            }
        }

        private boolean needsDepth() {
            for (ShaderGroup shaderGroup : shaderGroups.values()) {
                GraphShader colorShader = shaderGroup.getColorShader();
                if (colorShader.isUsingDepthTexture() && models.hasModelWithTag(colorShader.getTag()))
                    return true;
            }
            return false;
        }

        private boolean isRequiringSceneColor() {
            for (ShaderGroup shaderGroup : shaderGroups.values()) {
                GraphShader colorShader = shaderGroup.getColorShader();
                if (colorShader.isUsingColorTexture() && models.hasModelWithTag(colorShader.getTag()))
                    return true;
            }
            return false;
        }

        @Override
        public void processPipelineRequirements(PipelineRequirements pipelineRequirements) {
            if (needsDepth())
                pipelineRequirements.setRequiringDepthTexture();
        }

        @Override
        public void executeNode(PipelineRenderingContext pipelineRenderingContext, PipelineRequirementsCallback pipelineRequirementsCallback) {
            final PipelineNode.FieldOutput<Boolean> processorEnabled = (PipelineNode.FieldOutput<Boolean>) inputs.get("enabled");
            final PipelineNode.FieldOutput<Camera> cameraInput = (PipelineNode.FieldOutput<Camera>) inputs.get("camera");
            final PipelineNode.FieldOutput<RenderPipeline> renderPipelineInput = (PipelineNode.FieldOutput<RenderPipeline>) inputs.get("input");
            RenderPipeline renderPipeline = renderPipelineInput.getValue();
            boolean enabled = processorEnabled == null || processorEnabled.getValue();
            if (enabled) {
                boolean needsToDrawDepth = pipelineRequirementsCallback.getPipelineRequirements().isRequiringDepthTexture();
                if (needsToDrawDepth)
                    initializeDepthShaders();
                boolean usesDepth = needsDepth();
                boolean needsSceneColor = isRequiringSceneColor();
                RenderPipelineBuffer currentBuffer = renderPipeline.getDefaultBuffer();
                Camera camera = cameraInput.getValue();
                shaderContext.setCamera(camera);
                shaderContext.setTimeProvider(timeProvider);
                shaderContext.setRenderWidth(currentBuffer.getWidth());
                shaderContext.setRenderHeight(currentBuffer.getHeight());
                if (needsToDrawDepth || usesDepth) {
                    renderPipeline.enrichWithDepthBuffer(currentBuffer);
                    shaderContext.setDepthTexture(currentBuffer.getDepthBufferTexture());
                }
                RenderPipelineBuffer sceneColorBuffer = null;
                if (needsSceneColor)
                    sceneColorBuffer = setupColorTexture(renderPipeline, currentBuffer, pipelineRenderingContext);
                // Drawing models on color buffer
                colorStrategyCallback.prepare(pipelineRenderingContext, models);
                currentBuffer.beginColor();
                renderingStrategy.processModels(models, allShaderTags, camera, colorStrategyCallback);
                currentBuffer.endColor();
                if (needsToDrawDepth) {
                    // Drawing models on depth buffer
                    depthStrategyCallback.prepare(pipelineRenderingContext, models);
                    currentBuffer.beginDepth();
                    renderingStrategy.processModels(models, depthDrawingShaderTags, camera, depthStrategyCallback);
                    currentBuffer.endDepth();
                }
                if (sceneColorBuffer != null)
                    renderPipeline.returnFrameBuffer(sceneColorBuffer);
            }
            output.setValue(renderPipeline);
        }

        private RenderPipelineBuffer setupColorTexture(final RenderPipeline renderPipeline, final RenderPipelineBuffer currentBuffer, PipelineRenderingContext pipelineRenderingContext) {
            RenderPipelineBuffer sceneColorBuffer = renderPipeline.getNewFrameBuffer(currentBuffer, Color.BLACK);
            shaderContext.setColorTexture(sceneColorBuffer.getColorBufferTexture());
            renderPipeline.drawTexture(currentBuffer, sceneColorBuffer, pipelineRenderingContext, fullScreenRender);
            return sceneColorBuffer;
        }

        @Override
        public void dispose() {
            for (ShaderGroup shaderGroup : shaderGroups.values()) {
                shaderGroup.dispose();
            }
        }
    };
}