com.bitfire.postprocessing.filters.Flare.java Source code

Introduction

Here is the source code for com.bitfire.postprocessing.filters.Flare.java

Source

/*******************************************************************************
 * 
 * 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.bitfire.postprocessing.filters;

import com.badlogic.gdx.graphics.Texture;
import com.badlogic.gdx.math.Vector2;
import com.bitfire.utils.ShaderLoader;

/**
 * Lens flare effect as described in John Chapman's article (without lens dirt
 * or diffraction starburst). Lens color image (lenscolor.png) is located in
 * src/main/resources/ folder.
 * 
 * @see <a href=
 *      "http://john-chapman-graphics.blogspot.co.uk/2013/02/pseudo-lens-flare.html">http://john-chapman-graphics.blogspot.co.uk/2013/02/pseudo-lens-flare.html</a>
 * @author Toni Sagrista
 **/
public final class Flare extends Filter<Flare> {
    private Vector2 viewportInverse;
    private int ghosts;
    private float haloWidth;
    private Texture lensColorTexture;

    public enum Param implements Parameter {
        // @formatter:off
        Texture("u_texture0", 0), LensColor("u_texture1", 0), ViewportInverse("u_viewportInverse",
                2), Ghosts("u_ghosts", 0), HaloWidth("u_haloWidth", 0);
        // @formatter:on

        private String mnemonic;
        private int elementSize;

        private Param(String mnemonic, int arrayElementSize) {
            this.mnemonic = mnemonic;
            this.elementSize = arrayElementSize;
        }

        @Override
        public String mnemonic() {
            return this.mnemonic;
        }

        @Override
        public int arrayElementSize() {
            return this.elementSize;
        }
    }

    public Flare(int width, int height) {
        super(ShaderLoader.fromFile("screenspace", "flare"));
        viewportInverse = new Vector2(1f / width, 1f / height);
        rebind();
    }

    public void setViewportSize(float width, float height) {
        this.viewportInverse.set(1f / width, 1f / height);
        setParam(Param.ViewportInverse, this.viewportInverse);
    }

    public int getGhosts() {
        return ghosts;
    }

    public void setGhosts(int ghosts) {
        this.ghosts = ghosts;
        setParam(Param.Ghosts, ghosts);
    }

    public float getHaloWidth() {
        return haloWidth;
    }

    public void setHaloWidth(float haloWidth) {
        this.haloWidth = haloWidth;
        setParam(Param.HaloWidth, haloWidth);
    }

    public void setLensColorTexture(Texture tex) {
        this.lensColorTexture = tex;
        setParam(Param.LensColor, u_texture1);
    }

    @Override
    public void rebind() {
        // Re-implement super to batch every parameter
        setParams(Param.Texture, u_texture0);
        setParams(Param.LensColor, u_texture1);
        setParams(Param.ViewportInverse, viewportInverse);
        setParams(Param.Ghosts, ghosts);
        setParams(Param.HaloWidth, haloWidth);
        endParams();
    }

    @Override
    protected void onBeforeRender() {
        inputTexture.bind(u_texture0);
        lensColorTexture.bind(u_texture1);
    }
}