Project

General

Profile

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

library / src / main / java / org / distorted / library / effectqueue / EffectQueuePostprocess.java @ b7074bc6

1 4c1dd6e9 Leszek Koltunski
///////////////////////////////////////////////////////////////////////////////////////////////////
2
// Copyright 2016 Leszek Koltunski                                                               //
3
//                                                                                               //
4 46b572b5 Leszek Koltunski
// This file is part of Distorted.                                                               //
5 4c1dd6e9 Leszek Koltunski
//                                                                                               //
6 46b572b5 Leszek Koltunski
// Distorted is free software: you can redistribute it and/or modify                             //
7 4c1dd6e9 Leszek Koltunski
// 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 46b572b5 Leszek Koltunski
// Distorted is distributed in the hope that it will be useful,                                  //
12 4c1dd6e9 Leszek Koltunski
// 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 46b572b5 Leszek Koltunski
// along with Distorted.  If not, see <http://www.gnu.org/licenses/>.                            //
18 4c1dd6e9 Leszek Koltunski
///////////////////////////////////////////////////////////////////////////////////////////////////
19
20 809dcae3 Leszek Koltunski
package org.distorted.library.effectqueue;
21 4c1dd6e9 Leszek Koltunski
22 a13dde77 Leszek Koltunski
import android.content.res.Resources;
23 b7074bc6 Leszek Koltunski
import android.opengl.GLES30;
24 a13dde77 Leszek Koltunski
import android.util.Log;
25 8dccc3c2 Leszek Koltunski
26 a13dde77 Leszek Koltunski
import org.distorted.library.R;
27 da9b3f07 Leszek Koltunski
import org.distorted.library.effect.EffectType;
28 fe82a979 Leszek Koltunski
import org.distorted.library.effect.PostprocessEffect;
29 a13dde77 Leszek Koltunski
import org.distorted.library.effect.VertexEffect;
30 c90aca24 Leszek Koltunski
import org.distorted.library.main.DistortedEffects;
31 7602a827 Leszek Koltunski
import org.distorted.library.main.DistortedLibrary;
32 809dcae3 Leszek Koltunski
import org.distorted.library.main.DistortedFramebuffer;
33
import org.distorted.library.main.DistortedNode;
34 7602a827 Leszek Koltunski
import org.distorted.library.main.InternalOutputSurface;
35
import org.distorted.library.main.InternalRenderState;
36 715e7726 Leszek Koltunski
import org.distorted.library.mesh.MeshBase;
37 ed06301f Leszek Koltunski
import org.distorted.library.message.EffectMessageSender;
38 a13dde77 Leszek Koltunski
import org.distorted.library.program.DistortedProgram;
39
40
import java.io.InputStream;
41 8fa96e69 Leszek Koltunski
42 4c1dd6e9 Leszek Koltunski
///////////////////////////////////////////////////////////////////////////////////////////////////
43 809dcae3 Leszek Koltunski
/**
44
 * Not part of public API, do not document
45
 *
46
 * @y.exclude
47
 */
48
public class EffectQueuePostprocess extends EffectQueue
49 4c1dd6e9 Leszek Koltunski
  {
50 15aa7d94 Leszek Koltunski
  private static final int NUM_UNIFORMS = PostprocessEffect.NUM_UNIFORMS;
51 da9b3f07 Leszek Koltunski
  private static final int INDEX = EffectType.POSTPROCESS.ordinal();
52 4c1dd6e9 Leszek Koltunski
53 fff1110e leszek
  private int mHalo;
54 9e771d06 Leszek Koltunski
  private float mR, mG, mB, mA;
55 a31dbc5c Leszek Koltunski
56 247d8225 Leszek Koltunski
  private static DistortedProgram mPreProgram;
57
  private static int mPreColorH;
58 9e771d06 Leszek Koltunski
  private static int mPreTextureH;
59 a13dde77 Leszek Koltunski
60 4c1dd6e9 Leszek Koltunski
///////////////////////////////////////////////////////////////////////////////////////////////////
61
62 20dbec0e Leszek Koltunski
  EffectQueuePostprocess()
63 4c1dd6e9 Leszek Koltunski
    { 
64 20dbec0e Leszek Koltunski
    super(NUM_UNIFORMS,INDEX );
65 4c1dd6e9 Leszek Koltunski
    }
66
67 c90b9e01 Leszek Koltunski
///////////////////////////////////////////////////////////////////////////////////////////////////
68
69 1149be8f leszek
  void compute(long currTime)
70 4c1dd6e9 Leszek Koltunski
    {
71 15aa7d94 Leszek Koltunski
    if( currTime==mTime ) return;
72 4c1dd6e9 Leszek Koltunski
    if( mTime==0 ) mTime = currTime;
73
    long step = (currTime-mTime);
74 e02264ff leszek
75 9e771d06 Leszek Koltunski
    mR = mG = mB = mA = 0.0f;
76 fff1110e leszek
    mHalo = 0;
77
    int halo;
78
79 4c1dd6e9 Leszek Koltunski
    for(int i=0; i<mNumEffects; i++)
80
      {
81 3a70bd6d leszek
      mCurrentDuration[i] += step;
82
83 e49b26ba Leszek Koltunski
      // first zero out the 'alpha' because BLUR effect will not overwrite this (it is a 1D effect)
84
      // and if previously there was a GLOW effect here then mA would be non-zero and we don't want
85
      // that (see preprocess())
86
      mUniforms[NUM_UNIFORMS*i+4]=0.0f;
87
88 15aa7d94 Leszek Koltunski
      if( mEffects[i].compute(mUniforms, NUM_UNIFORMS*i, mCurrentDuration[i], step) )
89 4c1dd6e9 Leszek Koltunski
        {
90 20dbec0e Leszek Koltunski
        EffectMessageSender.newMessage(mEffects[i]);
91 4c1dd6e9 Leszek Koltunski
        }
92 a31dbc5c Leszek Koltunski
93 fff1110e leszek
      halo = (int)mUniforms[NUM_UNIFORMS*i];
94
      if( halo>mHalo ) mHalo = halo;
95 4c1dd6e9 Leszek Koltunski
      }
96 e02264ff leszek
97 9e771d06 Leszek Koltunski
    // TODO  (now only really works in case of 1 effect!)
98
    if( mNumEffects>0 )
99
      {
100
      mR = mUniforms[1];
101
      mG = mUniforms[2];
102
      mB = mUniforms[3];
103
      mA = mUniforms[4];
104
      }
105
106 95c441a2 leszek
    mTime = currTime;
107 4c1dd6e9 Leszek Koltunski
    }
108
109 984dc935 Leszek Koltunski
///////////////////////////////////////////////////////////////////////////////////////////////////
110 247d8225 Leszek Koltunski
111 b7074bc6 Leszek Koltunski
  public static void createPrograms(Resources resources, int GLSL)
112 a13dde77 Leszek Koltunski
    {
113
    final InputStream mainVertStream = resources.openRawResource(R.raw.main_vertex_shader);
114
    final InputStream mainFragStream = resources.openRawResource(R.raw.preprocess_fragment_shader);
115
116
    int numV = VertexEffect.getNumEnabled();
117
118 b7074bc6 Leszek Koltunski
    String version = "#version "+GLSL+" es\n";
119
    String mainVertHeader= version + ("#define NUM_VERTEX "   + ( numV>0 ? DistortedLibrary.getMax(EffectType.VERTEX  ) : 0 ) + "\n");
120
    String mainFragHeader= version + "\n";
121 a13dde77 Leszek Koltunski
122
    String enabledEffectV= VertexEffect.getGLSL();
123
124
    try
125
      {
126
      mPreProgram = new DistortedProgram(mainVertStream, mainFragStream, mainVertHeader, mainFragHeader,
127 b7074bc6 Leszek Koltunski
                                         enabledEffectV, null, GLSL, null);
128 a13dde77 Leszek Koltunski
      }
129
    catch(Exception e)
130
      {
131
      Log.e("POSTPROCESS", e.getClass().getSimpleName()+" trying to compile PRE program: "+e.getMessage());
132
      throw new RuntimeException(e.getMessage());
133
      }
134
135
    int preProgramH = mPreProgram.getProgramHandle();
136
    EffectQueueVertex.getUniforms( preProgramH,2 );
137
    EffectQueueMatrix.getUniforms( preProgramH,2 );
138 b7074bc6 Leszek Koltunski
    mPreColorH  = GLES30.glGetUniformLocation( preProgramH, "u_Color"  );
139
    mPreTextureH= GLES30.glGetUniformLocation( preProgramH, "u_Texture");
140 86d322b5 Leszek Koltunski
    }
141
142 7266d8ef Leszek Koltunski
///////////////////////////////////////////////////////////////////////////////////////////////////
143 2386a081 Leszek Koltunski
// TODO  (now only really works in case of 1 effect!)
144 7266d8ef Leszek Koltunski
145 809dcae3 Leszek Koltunski
  public int getQuality()
146 7266d8ef Leszek Koltunski
    {
147 9e771d06 Leszek Koltunski
    return mNumEffects>0 ? ((PostprocessEffect)mEffects[0]).getQuality() : 0;
148 7266d8ef Leszek Koltunski
    }
149
150 2386a081 Leszek Koltunski
///////////////////////////////////////////////////////////////////////////////////////////////////
151 a13dde77 Leszek Koltunski
// TODO  (now only really works in case of 1 effect!)
152 2386a081 Leszek Koltunski
153 809dcae3 Leszek Koltunski
  public boolean getRender()
154 a13dde77 Leszek Koltunski
    {
155 9e771d06 Leszek Koltunski
    return mNumEffects>0 ? ((PostprocessEffect)mEffects[0]).getRender() : false;
156 a13dde77 Leszek Koltunski
    }
157
158
///////////////////////////////////////////////////////////////////////////////////////////////////
159
160 7602a827 Leszek Koltunski
  public int preprocess(InternalOutputSurface buffer, DistortedNode node, float distance, float mipmap, float[] projection)
161 2386a081 Leszek Koltunski
    {
162 4bb94a7d Leszek Koltunski
    MeshBase mesh = node.getMesh();
163
    DistortedEffects effects = node.getEffects();
164 2386a081 Leszek Koltunski
165 4bb94a7d Leszek Koltunski
    int width   = buffer.getWidth();
166
    int height  = buffer.getHeight();
167 809dcae3 Leszek Koltunski
168 4bb94a7d Leszek Koltunski
    InternalRenderState.setUpStencilMark(mA!=0.0f);
169
    InternalRenderState.disableBlending();
170 a13dde77 Leszek Koltunski
171 b7074bc6 Leszek Koltunski
    GLES30.glViewport(0, 0, width, height );
172 a13dde77 Leszek Koltunski
173 4bb94a7d Leszek Koltunski
    mPreProgram.useProgram();
174 a13dde77 Leszek Koltunski
175 4bb94a7d Leszek Koltunski
    mesh.bindVertexAttribs(mPreProgram);
176 a13dde77 Leszek Koltunski
177 4bb94a7d Leszek Koltunski
    EffectQueue[] queues = effects.getQueues();
178
    EffectQueueMatrix matrix = (EffectQueueMatrix)queues[0];
179
    EffectQueueVertex vertex = (EffectQueueVertex)queues[1];
180 84d51487 Leszek Koltunski
181 4bb94a7d Leszek Koltunski
    float inflate=0.0f;
182 84d51487 Leszek Koltunski
183 23b733db Leszek Koltunski
    matrix.send(distance, mipmap, projection, mesh, 2);
184 84d51487 Leszek Koltunski
185 4bb94a7d Leszek Koltunski
    if( mHalo!=0.0f )
186
      {
187 23b733db Leszek Koltunski
      inflate = matrix.magnify(projection, width, height, mipmap, mesh, mHalo);
188 4bb94a7d Leszek Koltunski
      }
189 84d51487 Leszek Koltunski
190 4bb94a7d Leszek Koltunski
    vertex.send(inflate,2);
191 247d8225 Leszek Koltunski
192 4bb94a7d Leszek Koltunski
    if( mA!=0.0f )
193
      {
194 b7074bc6 Leszek Koltunski
      GLES30.glUniform4f(mPreColorH, mR, mG, mB, mA);
195
      GLES30.glUniform1i(mPreTextureH, 0);
196 4bb94a7d Leszek Koltunski
      }
197 247d8225 Leszek Koltunski
198 b7074bc6 Leszek Koltunski
    GLES30.glDrawArrays(GLES30.GL_TRIANGLE_STRIP, 0, mesh.getNumVertices() );
199 a13dde77 Leszek Koltunski
200 4bb94a7d Leszek Koltunski
    InternalRenderState.restoreBlending();
201
    InternalRenderState.unsetUpStencilMark();
202 a13dde77 Leszek Koltunski
203 4bb94a7d Leszek Koltunski
    return 1;
204 2386a081 Leszek Koltunski
    }
205
206 4b9fe2e9 Leszek Koltunski
///////////////////////////////////////////////////////////////////////////////////////////////////
207 cf7394cc leszek
208 809dcae3 Leszek Koltunski
  public int postprocess(DistortedFramebuffer buffer)
209 4b9fe2e9 Leszek Koltunski
    {
210 bed13bea leszek
    int numRenders = 0;
211
212 b7074bc6 Leszek Koltunski
    GLES30.glDisable(GLES30.GL_BLEND);
213 8dccc3c2 Leszek Koltunski
214 1149be8f leszek
    for(int i=0; i<mNumEffects; i++)
215 95c441a2 leszek
      {
216 9e771d06 Leszek Koltunski
      numRenders += ((PostprocessEffect)mEffects[i]).apply(mUniforms,NUM_UNIFORMS*i, buffer);
217 95c441a2 leszek
      }
218 4b9fe2e9 Leszek Koltunski
219 b7074bc6 Leszek Koltunski
    GLES30.glEnable(GLES30.GL_BLEND);
220 8dccc3c2 Leszek Koltunski
221 bed13bea leszek
    return numRenders;
222 cf7394cc leszek
    }
223 4c1dd6e9 Leszek Koltunski
  }