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

Introduction

Here is the source code for com.bitfire.postprocessing.filters.Lens2.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 Lens2 extends Filter<Lens2> {
    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 Lens2(int width, int height) {
        super(ShaderLoader.fromFile("screenspace", "lensflare2"));
        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);
    }
}