Project

General

Profile

Download (14.9 KB) Statistics
| Branch: | Revision:

library / src / main / java / org / distorted / library / effect / PostprocessEffectGlow.java @ 6770ef46

1 125cee3d Leszek Koltunski
///////////////////////////////////////////////////////////////////////////////////////////////////
2
// Copyright 2017 Leszek Koltunski                                                               //
3
//                                                                                               //
4
// This file is part of Distorted.                                                               //
5
//                                                                                               //
6
// Distorted is free software: you can redistribute it and/or modify                             //
7
// it under the terms of the GNU General Public License as published by                          //
8
// the Free Software Foundation, either version 2 of the License, or                             //
9
// (at your option) any later version.                                                           //
10
//                                                                                               //
11
// Distorted is distributed in the hope that it will be useful,                                  //
12
// but WITHOUT ANY WARRANTY; without even the implied warranty of                                //
13
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the                                 //
14
// GNU General Public License for more details.                                                  //
15
//                                                                                               //
16
// You should have received a copy of the GNU General Public License                             //
17
// along with Distorted.  If not, see <http://www.gnu.org/licenses/>.                            //
18
///////////////////////////////////////////////////////////////////////////////////////////////////
19
20
package org.distorted.library.effect;
21
22 e6519ac8 Leszek Koltunski
import android.opengl.GLES31;
23 1149be8f leszek
24 7266d8ef Leszek Koltunski
import org.distorted.library.main.DistortedEffects;
25 1dfc9074 leszek
import org.distorted.library.main.DistortedFramebuffer;
26 70b6a155 Leszek Koltunski
import org.distorted.library.main.DistortedOutputSurface;
27 7266d8ef Leszek Koltunski
import org.distorted.library.main.DistortedRenderState;
28 1149be8f leszek
import org.distorted.library.program.DistortedProgram;
29 125cee3d Leszek Koltunski
import org.distorted.library.type.Data1D;
30
import org.distorted.library.type.Data4D;
31
32
///////////////////////////////////////////////////////////////////////////////////////////////////
33 6b816678 Leszek Koltunski
/**
34
 * Add a (colored) glow about each object in the Framebuffer.
35
 */
36 125cee3d Leszek Koltunski
public class PostprocessEffectGlow extends PostprocessEffect
37
  {
38 1149be8f leszek
  private static final int MAX_HALO = 50;
39
40 0dd98279 Leszek Koltunski
  private Data1D mGlowRadius;
41
  private Data4D mColor;
42
43 1149be8f leszek
  private static final float GAUSSIAN[] =   // G(0.00), G(0.03), G(0.06), ..., G(3.00), 0
44
    {                                       // where G(x)= (1/(sqrt(2*PI))) * e^(-(x^2)/2). The last 0 terminates.
45
    0.398948f, 0.398769f, 0.398231f, 0.397336f, 0.396086f, 0.394485f, 0.392537f, 0.390247f, 0.387622f, 0.384668f,
46
    0.381393f, 0.377806f, 0.373916f, 0.369733f, 0.365268f, 0.360532f, 0.355538f, 0.350297f, 0.344823f, 0.339129f,
47
    0.333229f, 0.327138f, 0.320868f, 0.314436f, 0.307856f, 0.301142f, 0.294309f, 0.287373f, 0.280348f, 0.273248f,
48
    0.266089f, 0.258884f, 0.251648f, 0.244394f, 0.237135f, 0.229886f, 0.222657f, 0.215461f, 0.208311f, 0.201217f,
49
    0.194189f, 0.187238f, 0.180374f, 0.173605f, 0.166940f, 0.160386f, 0.153951f, 0.147641f, 0.141462f, 0.135420f,
50
    0.129520f, 0.123765f, 0.118159f, 0.112706f, 0.107408f, 0.102266f, 0.097284f, 0.092461f, 0.087797f, 0.083294f,
51
    0.078951f, 0.074767f, 0.070741f, 0.066872f, 0.063158f, 0.059596f, 0.056184f, 0.052920f, 0.049801f, 0.046823f,
52
    0.043984f, 0.041280f, 0.038707f, 0.036262f, 0.033941f, 0.031740f, 0.029655f, 0.027682f, 0.025817f, 0.024056f,
53
    0.022395f, 0.020830f, 0.019357f, 0.017971f, 0.016670f, 0.015450f, 0.014305f, 0.013234f, 0.012232f, 0.011295f,
54
    0.010421f, 0.009606f, 0.008847f, 0.008140f, 0.007483f, 0.006873f, 0.006307f, 0.005782f, 0.005296f, 0.004847f,
55
    0.004432f, 0.000000f
56
    };
57
  private static final int NUM_GAUSSIAN = GAUSSIAN.length-2;
58 7cd24173 leszek
59 1149be8f leszek
  // The (fixed-function-sampled) Gaussian Blur kernels are of the size k0=1, k1=2, k2=2, k3=3, k4=3, k5=4, k6=4,...
60
  // i.e. k(i)=floor((i+3)/2).  (the 'i' in k(i) means 'blur taking into account the present pixel and 'i' pixels
61
  // in all 4 directions)
62
  // We need room for MAX_BLUR of them, and sum(i=0...N, floor((i+3)/2)) = N + floor(N*N/4)
63
  private static float[] weightsCache = new float[MAX_HALO + MAX_HALO*MAX_HALO/4];
64
  private static float[] offsetsCache = new float[MAX_HALO + MAX_HALO*MAX_HALO/4];
65
  private static float[] mWeights = new float[MAX_HALO];
66
  private static float[] mOffsets = new float[MAX_HALO];
67 7cd24173 leszek
68 1149be8f leszek
  private static DistortedProgram mProgram1, mProgram2;
69
  private static int mIndex1, mIndex2;
70 7cd24173 leszek
71 125cee3d Leszek Koltunski
///////////////////////////////////////////////////////////////////////////////////////////////////
72 faa3ff56 Leszek Koltunski
// This implements the 'Better separable implementation using GPU fixed function sampling' from
73
// https://software.intel.com/en-us/blogs/2014/07/15/an-investigation-of-fast-real-time-gpu-based-image-blur-algorithms
74 1149be8f leszek
75 faa3ff56 Leszek Koltunski
  private void computeGaussianKernel(int radius)
76 125cee3d Leszek Koltunski
    {
77 faa3ff56 Leszek Koltunski
    int offset = radius + radius*radius/4;
78
79
    if( weightsCache[offset]==0.0f )
80
      {
81
      float z, x= 0.0f, P= (float)NUM_GAUSSIAN / (radius>3 ? radius:3);
82
      mWeights[0] = GAUSSIAN[0];
83
      float sum   = GAUSSIAN[0];
84
      int j;
85
86
      for(int i=1; i<=radius; i++)
87
        {
88
        x += P;
89
        j = (int)x;
90
        z = x-j;
91
92
        mWeights[i] = (1-z)*GAUSSIAN[j] + z*GAUSSIAN[j+1];
93
        sum += 2*mWeights[i];
94
        }
95
96
      for(int i=0; i<=radius; i++) mWeights[i] /= sum;
97
98
      int numloops = radius/2;
99
      weightsCache[offset] = mWeights[0];
100
      offsetsCache[offset] = 0.0f;
101
102
      for(int i=0; i<numloops; i++)
103
        {
104
        offsetsCache[offset+i+1] = mWeights[2*i+1]*(2*i+1) + mWeights[2*i+2]*(2*i+2);
105
        weightsCache[offset+i+1] = mWeights[2*i+1] + mWeights[2*i+2];
106
        offsetsCache[offset+i+1]/= weightsCache[offset+i+1];
107
        }
108
109
      if( radius%2 == 1 )
110
        {
111
        int index = offset + radius/2 +1;
112
        offsetsCache[index]=radius;
113
        weightsCache[index]=mWeights[radius];
114
        }
115
      }
116 125cee3d Leszek Koltunski
    }
117 15aa7d94 Leszek Koltunski
118
///////////////////////////////////////////////////////////////////////////////////////////////////
119 faa3ff56 Leszek Koltunski
/**
120
 * Only for use by the library itself.
121
 *
122
 * @y.exclude
123
 */
124 15aa7d94 Leszek Koltunski
  public boolean compute(float[] uniforms, int index, long currentDuration, long step )
125
    {
126 0dd98279 Leszek Koltunski
    mColor.get(uniforms,index+1,currentDuration,step);
127
    return mGlowRadius.get(uniforms,index,currentDuration,step);
128 15aa7d94 Leszek Koltunski
    }
129 1dfc9074 leszek
130
///////////////////////////////////////////////////////////////////////////////////////////////////
131 faa3ff56 Leszek Koltunski
/**
132
 * Only for use by the library itself.
133
 *
134
 * @y.exclude
135
 */
136 86d322b5 Leszek Koltunski
 public int getQuality()
137
   {
138
   return 0;
139
   }
140
141
///////////////////////////////////////////////////////////////////////////////////////////////////
142
/**
143
 * Only for use by the library itself.
144
 *
145
 * @y.exclude
146
 */
147 70b6a155 Leszek Koltunski
  public int apply(float[] uniforms, int index, DistortedOutputSurface[] buffers)
148 1dfc9074 leszek
    {
149 1149be8f leszek
    if( mProgram1 ==null)
150
      {
151 6770ef46 Leszek Koltunski
      try
152
        {
153
        mProgram1 = mPrograms.get(mIndex1);
154
        mProgram2 = mPrograms.get(mIndex2);
155
        }
156
      catch(Exception ex)
157
        {
158
        return 0;
159
        }
160 1149be8f leszek
      }
161
162 7266d8ef Leszek Koltunski
    DistortedRenderState.useStencilMark();
163
164 70b6a155 Leszek Koltunski
    DistortedFramebuffer  inBuffer = (DistortedFramebuffer)buffers[0];
165
    DistortedFramebuffer outBuffer = (DistortedFramebuffer)buffers[mQualityLevel];
166 4b9fe2e9 Leszek Koltunski
167 33b0fad7 Leszek Koltunski
    float w= outBuffer.getWidth();
168
    float h= outBuffer.getHeight();
169
    float n= 1.0f - outBuffer.getNear();
170 1149be8f leszek
171 ae2802b1 Leszek Koltunski
    float corrW = inBuffer.getWidthCorrection();
172
    float corrH = inBuffer.getHeightCorrection();
173
    float offsetCorrW = corrW/w;
174
    float offsetCorrH = corrH/h;
175
176 86d322b5 Leszek Koltunski
    int radius = (int)(uniforms[index]*mQualityScale);
177 1149be8f leszek
    if( radius>=MAX_HALO ) radius = MAX_HALO-1;
178
    computeGaussianKernel(radius);
179
180
    int offset = radius + radius*radius/4;
181
    radius = (radius+1)/2;
182 73208cab leszek
183 7266d8ef Leszek Koltunski
    outBuffer.setAsOutput();
184 e6519ac8 Leszek Koltunski
    GLES31.glViewport(0, 0, (int)w, (int)h);
185 1149be8f leszek
186 4b9fe2e9 Leszek Koltunski
    // horizontal glow
187 ae2802b1 Leszek Koltunski
    for(int i=0; i<=radius; i++) mOffsets[i] = offsetsCache[offset+i]*offsetCorrW;
188 1149be8f leszek
189
    mProgram1.useProgram();
190 7266d8ef Leszek Koltunski
    outBuffer.bindForOutput(1);
191 4b9fe2e9 Leszek Koltunski
    inBuffer.setAsInput(0);
192 1149be8f leszek
193 9455da17 Leszek Koltunski
    GLES31.glColorMask(true,true,true,true);
194
    GLES31.glClearColor(1.0f,1.0f,1.0f,0.0f);
195
    GLES31.glClear(GLES31.GL_COLOR_BUFFER_BIT);
196
197 e6519ac8 Leszek Koltunski
    GLES31.glUniform1f ( mProgram1.mUniform[0] , n );
198 ae2802b1 Leszek Koltunski
    GLES31.glUniform2f ( mProgram1.mUniform[1] , corrW, corrH );
199
    GLES31.glUniform1i ( mProgram1.mUniform[2] , 0 );
200
    GLES31.glUniform1fv( mProgram1.mUniform[3] , radius+1, mOffsets,0);
201
    GLES31.glUniform1fv( mProgram1.mUniform[4] , radius+1, weightsCache,offset);
202
    GLES31.glUniform1i ( mProgram1.mUniform[5] , radius);
203
    GLES31.glUniform4f ( mProgram1.mUniform[6] , uniforms[index+1], uniforms[index+2], uniforms[index+3], uniforms[index+4]);
204 e6519ac8 Leszek Koltunski
    GLES31.glVertexAttribPointer(mProgram1.mAttribute[0], POS_DATA_SIZE, GLES31.GL_FLOAT, false, 0, mQuadPositions);
205
    GLES31.glVertexAttribPointer(mProgram1.mAttribute[1], TEX_DATA_SIZE, GLES31.GL_FLOAT, false, 0, mQuadTexture);
206
    GLES31.glDrawArrays(GLES31.GL_TRIANGLE_STRIP, 0, 4);
207 1149be8f leszek
208 4b9fe2e9 Leszek Koltunski
    // vertical glow
209 ae2802b1 Leszek Koltunski
    for(int i=0; i<=radius; i++) mOffsets[i] = offsetsCache[offset+i]*offsetCorrH;
210 1149be8f leszek
211
    mProgram2.useProgram();
212 7266d8ef Leszek Koltunski
    outBuffer.bindForOutput(0);
213 4b9fe2e9 Leszek Koltunski
    outBuffer.setAsInput(1);
214 bed13bea leszek
215 e6519ac8 Leszek Koltunski
    GLES31.glUniform1f ( mProgram2.mUniform[0] , n );
216 ae2802b1 Leszek Koltunski
    GLES31.glUniform2f ( mProgram2.mUniform[1] , corrW, corrH );
217
    GLES31.glUniform1i ( mProgram2.mUniform[2] , 0 );
218
    GLES31.glUniform1fv( mProgram2.mUniform[3] , radius+1, mOffsets,0);
219
    GLES31.glUniform1fv( mProgram2.mUniform[4] , radius+1, weightsCache,offset);
220
    GLES31.glUniform1i ( mProgram2.mUniform[5] , radius);
221
    GLES31.glUniform4f ( mProgram1.mUniform[6] , uniforms[index+1], uniforms[index+2], uniforms[index+3], uniforms[index+4]);
222 e6519ac8 Leszek Koltunski
    GLES31.glVertexAttribPointer(mProgram2.mAttribute[0], POS_DATA_SIZE, GLES31.GL_FLOAT, false, 0, mQuadPositions);
223
    GLES31.glVertexAttribPointer(mProgram2.mAttribute[1], TEX_DATA_SIZE, GLES31.GL_FLOAT, false, 0, mQuadTexture);
224
    GLES31.glDrawArrays(GLES31.GL_TRIANGLE_STRIP, 0, 4);
225 7266d8ef Leszek Koltunski
226
    DistortedRenderState.unuseStencilMark();
227
228
    // blit back to inBuffer if we have to
229
    if( mQualityLevel>0 )
230
      {
231
      inBuffer.setAsOutput();
232
      outBuffer.setAsInput(0);
233 8dccc3c2 Leszek Koltunski
      GLES31.glEnable(GLES31.GL_BLEND);
234 7266d8ef Leszek Koltunski
      DistortedEffects.blitPriv(inBuffer);
235 8dccc3c2 Leszek Koltunski
      GLES31.glDisable(GLES31.GL_BLEND);
236 7266d8ef Leszek Koltunski
      return 3;
237
      }
238
239 bed13bea leszek
    return 2;
240 1149be8f leszek
    }
241
242
///////////////////////////////////////////////////////////////////////////////////////////////////
243 faa3ff56 Leszek Koltunski
// PUBLIC API
244
///////////////////////////////////////////////////////////////////////////////////////////////////
245
/**
246
 * Have to call this before the shaders get compiled (i.e before Distorted.onCreate()) for the Effect to work.
247
 */
248 1149be8f leszek
  public static void enable()
249
    {
250
    final String glowVertex =
251
252
      "#version 300 es        \n"+
253
      "precision lowp float;  \n"+
254
      "in vec2 a_Position;    \n"+
255
      "in vec2 a_TexCoord;    \n"+
256
      "out vec2 v_TexCoord;   \n"+
257
      "uniform float u_Depth; \n"+
258 ae2802b1 Leszek Koltunski
      "uniform vec2 u_TexCorr;\n"+
259 1149be8f leszek
260
      "void main()                                      \n"+
261
      "  {                                              \n"+
262 ae2802b1 Leszek Koltunski
      "  v_TexCoord = a_TexCoord * u_TexCorr;           \n"+
263 1149be8f leszek
      "  gl_Position= vec4(2.0*a_Position,u_Depth,1.0); \n"+
264
      "  }";
265
266
    final String glowFragment1 =
267
268
      "#version 300 es                    \n"+
269
      "#define MAX_BLUR "+MAX_HALO+      "\n"+
270
      "precision lowp float;              \n"+
271
      "in vec2 v_TexCoord;                \n"+
272
      "out vec4 fragColor;                \n"+
273
      "uniform sampler2D u_ColorTexture;  \n"+
274
      "uniform float u_Offsets[MAX_BLUR]; \n"+
275
      "uniform float u_Weights[MAX_BLUR]; \n"+
276
      "uniform int u_Radius;              \n"+
277 5b673c85 Leszek Koltunski
      "uniform vec4 u_Color;              \n"+
278
279 7ad20cce Leszek Koltunski
      "void main()                                                                                           \n"+
280
      "  {                                                                                                   \n"+
281
      "  vec4 pixel = texture(u_ColorTexture,v_TexCoord) * u_Weights[0];                                     \n"+
282
      "  for (int i=1; i<=u_Radius; i+=1)                                                                    \n"+
283
      "    {                                                                                                 \n"+
284
      "    pixel += ( texture(u_ColorTexture,vec2(v_TexCoord.x+u_Offsets[i],v_TexCoord.y)) +                 \n"+
285
      "               texture(u_ColorTexture,vec2(v_TexCoord.x-u_Offsets[i],v_TexCoord.y)) ) * u_Weights[i]; \n"+
286
      "    }                                                                                                 \n"+
287 bed13bea leszek
      "  fragColor = vec4( 0.5*(pixel.rgb+u_Color.rgb), sqrt(pixel.a*u_Color.a));                            \n"+
288 1149be8f leszek
      "  }";
289
290
    final String glowFragment2 =
291
292
      "#version 300 es                    \n"+
293
      "#define MAX_BLUR "+MAX_HALO+      "\n"+
294
      "precision lowp float;              \n"+
295
      "in vec2 v_TexCoord;                \n"+
296
      "out vec4 fragColor;                \n"+
297
      "uniform sampler2D u_ColorTexture;  \n"+
298
      "uniform float u_Offsets[MAX_BLUR]; \n"+
299
      "uniform float u_Weights[MAX_BLUR]; \n"+
300
      "uniform int u_Radius;              \n"+
301 5b673c85 Leszek Koltunski
      "uniform vec4 u_Color;              \n"+
302
303 7ad20cce Leszek Koltunski
      "void main()                                                                                           \n"+
304
      "  {                                                                                                   \n"+
305
      "  vec4 pixel = texture(u_ColorTexture,v_TexCoord) * u_Weights[0];                                     \n"+
306
      "  for (int i=1; i<=u_Radius; i+=1)                                                                    \n"+
307
      "    {                                                                                                 \n"+
308
      "    pixel += ( texture(u_ColorTexture,vec2(v_TexCoord.x,v_TexCoord.y+u_Offsets[i])) +                 \n"+
309
      "               texture(u_ColorTexture,vec2(v_TexCoord.x,v_TexCoord.y-u_Offsets[i])) ) * u_Weights[i]; \n"+
310
      "    }                                                                                                 \n"+
311 bed13bea leszek
      "  fragColor = vec4( 0.5*(pixel.rgb+u_Color.rgb), sqrt(pixel.a*u_Color.a));                            \n"+
312 1149be8f leszek
      "  }";
313
314 7ad20cce Leszek Koltunski
    mIndex1 = PostprocessEffect.register("GLOW1", glowVertex,glowFragment1);
315
    mIndex2 = PostprocessEffect.register("GLOW2", glowVertex,glowFragment2);
316 1149be8f leszek
    }
317
318
///////////////////////////////////////////////////////////////////////////////////////////////////
319
320 faa3ff56 Leszek Koltunski
/**
321
 * Make the object glow with a specific color and a halo of specific radius.
322
 *
323
 * @param glowRadius The 'strength' if the effect, in pixels. 0 = no halo, 10 = halo of roughly 10 pixels
324
 *                   around the whole object.
325
 * @param color      RGBA of the color with which to draw the glow.
326
 */
327
  public PostprocessEffectGlow(Data1D glowRadius, Data4D color)
328 1149be8f leszek
    {
329 faa3ff56 Leszek Koltunski
    super(EffectName.GLOW);
330
    mGlowRadius = glowRadius;
331
    mColor      = color;
332 1dfc9074 leszek
    }
333 125cee3d Leszek Koltunski
  }