Project

General

Profile

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

library / src / main / java / org / distorted / library / main / EffectQueuePostprocess.java @ 6c00149d

1 4c1dd6e9 Leszek Koltunski
///////////////////////////////////////////////////////////////////////////////////////////////////
2
// Copyright 2016 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 fe82a979 Leszek Koltunski
package org.distorted.library.main;
21 4c1dd6e9 Leszek Koltunski
22 a13dde77 Leszek Koltunski
import android.content.res.Resources;
23 8dccc3c2 Leszek Koltunski
import android.opengl.GLES31;
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 6c00149d Leszek Koltunski
import org.distorted.library.mesh.MeshObject;
31 4c1dd6e9 Leszek Koltunski
import org.distorted.library.message.EffectMessage;
32 a13dde77 Leszek Koltunski
import org.distorted.library.program.DistortedProgram;
33
34
import java.io.InputStream;
35 8fa96e69 Leszek Koltunski
36 4c1dd6e9 Leszek Koltunski
///////////////////////////////////////////////////////////////////////////////////////////////////
37
38
class EffectQueuePostprocess extends EffectQueue
39
  {
40 15aa7d94 Leszek Koltunski
  private static final int NUM_UNIFORMS = PostprocessEffect.NUM_UNIFORMS;
41 da9b3f07 Leszek Koltunski
  private static final int INDEX = EffectType.POSTPROCESS.ordinal();
42 4c1dd6e9 Leszek Koltunski
43 fff1110e leszek
  private int mHalo;
44 9e771d06 Leszek Koltunski
  private float mR, mG, mB, mA;
45 a31dbc5c Leszek Koltunski
46 247d8225 Leszek Koltunski
  private static DistortedProgram mPreProgram;
47
  private static int mPreColorH;
48 9e771d06 Leszek Koltunski
  private static int mPreTextureH;
49 a13dde77 Leszek Koltunski
50 4c1dd6e9 Leszek Koltunski
///////////////////////////////////////////////////////////////////////////////////////////////////
51
52
  EffectQueuePostprocess(long id)
53
    { 
54 15aa7d94 Leszek Koltunski
    super(id,NUM_UNIFORMS,INDEX );
55 4c1dd6e9 Leszek Koltunski
    }
56
57 c90b9e01 Leszek Koltunski
///////////////////////////////////////////////////////////////////////////////////////////////////
58
59 1149be8f leszek
  void compute(long currTime)
60 4c1dd6e9 Leszek Koltunski
    {
61 15aa7d94 Leszek Koltunski
    if( currTime==mTime ) return;
62 4c1dd6e9 Leszek Koltunski
    if( mTime==0 ) mTime = currTime;
63
    long step = (currTime-mTime);
64 e02264ff leszek
65 9e771d06 Leszek Koltunski
    mR = mG = mB = mA = 0.0f;
66 fff1110e leszek
    mHalo = 0;
67
    int halo;
68
69 4c1dd6e9 Leszek Koltunski
    for(int i=0; i<mNumEffects; i++)
70
      {
71 3a70bd6d leszek
      mCurrentDuration[i] += step;
72
73 15aa7d94 Leszek Koltunski
      if( mEffects[i].compute(mUniforms, NUM_UNIFORMS*i, mCurrentDuration[i], step) )
74 4c1dd6e9 Leszek Koltunski
        {
75
        for(int j=0; j<mNumListeners; j++)
76 26a4e5f6 leszek
          EffectMessageSender.newMessage( mListeners.get(j), EffectMessage.EFFECT_FINISHED, mEffects[i].getID(), mDistortedEffectsID);
77 4c1dd6e9 Leszek Koltunski
78 da9b3f07 Leszek Koltunski
        if( mEffects[i].isUnity( mUniforms, NUM_UNIFORMS*i) )
79 4c1dd6e9 Leszek Koltunski
          {
80 15aa7d94 Leszek Koltunski
          remove(i--);
81 6b0b4f60 leszek
          mNumEffectsToBe--;
82 1149be8f leszek
          regenerateIDandSort();
83 4c1dd6e9 Leszek Koltunski
          continue;
84
          }
85
        }
86 a31dbc5c Leszek Koltunski
87 fff1110e leszek
      halo = (int)mUniforms[NUM_UNIFORMS*i];
88
      if( halo>mHalo ) mHalo = halo;
89 4c1dd6e9 Leszek Koltunski
      }
90 e02264ff leszek
91 9e771d06 Leszek Koltunski
    // TODO  (now only really works in case of 1 effect!)
92
    if( mNumEffects>0 )
93
      {
94
      mR = mUniforms[1];
95
      mG = mUniforms[2];
96
      mB = mUniforms[3];
97
      mA = mUniforms[4];
98
      }
99
100 95c441a2 leszek
    mTime = currTime;
101 4c1dd6e9 Leszek Koltunski
    }
102
103 984dc935 Leszek Koltunski
///////////////////////////////////////////////////////////////////////////////////////////////////
104 247d8225 Leszek Koltunski
105
  static void createPrograms(Resources resources)
106 a13dde77 Leszek Koltunski
    {
107
    final InputStream mainVertStream = resources.openRawResource(R.raw.main_vertex_shader);
108
    final InputStream mainFragStream = resources.openRawResource(R.raw.preprocess_fragment_shader);
109
110
    int numV = VertexEffect.getNumEnabled();
111
112
    String mainVertHeader= Distorted.GLSL_VERSION + ("#define NUM_VERTEX "   + ( numV>0 ? DistortedEffects.getMax(EffectType.VERTEX  ) : 0 ) + "\n");
113
    String mainFragHeader= Distorted.GLSL_VERSION + "\n";
114
115
    String enabledEffectV= VertexEffect.getGLSL();
116
117
    try
118
      {
119
      mPreProgram = new DistortedProgram(mainVertStream, mainFragStream, mainVertHeader, mainFragHeader,
120
                                         enabledEffectV, null, Distorted.GLSL, null);
121
      }
122
    catch(Exception e)
123
      {
124
      Log.e("POSTPROCESS", e.getClass().getSimpleName()+" trying to compile PRE program: "+e.getMessage());
125
      throw new RuntimeException(e.getMessage());
126
      }
127
128
    int preProgramH = mPreProgram.getProgramHandle();
129
    EffectQueueVertex.getUniforms( preProgramH,2 );
130
    EffectQueueMatrix.getUniforms( preProgramH,2 );
131 9e771d06 Leszek Koltunski
    mPreColorH  = GLES31.glGetUniformLocation( preProgramH, "u_Color"  );
132
    mPreTextureH= GLES31.glGetUniformLocation( preProgramH, "u_Texture");
133 86d322b5 Leszek Koltunski
    }
134
135 7266d8ef Leszek Koltunski
///////////////////////////////////////////////////////////////////////////////////////////////////
136 2386a081 Leszek Koltunski
// TODO  (now only really works in case of 1 effect!)
137 7266d8ef Leszek Koltunski
138 9e771d06 Leszek Koltunski
  int getQuality()
139 7266d8ef Leszek Koltunski
    {
140 9e771d06 Leszek Koltunski
    return mNumEffects>0 ? ((PostprocessEffect)mEffects[0]).getQuality() : 0;
141 7266d8ef Leszek Koltunski
    }
142
143 2386a081 Leszek Koltunski
///////////////////////////////////////////////////////////////////////////////////////////////////
144 a13dde77 Leszek Koltunski
// TODO  (now only really works in case of 1 effect!)
145 2386a081 Leszek Koltunski
146 9e771d06 Leszek Koltunski
  boolean getRender()
147 a13dde77 Leszek Koltunski
    {
148 9e771d06 Leszek Koltunski
    return mNumEffects>0 ? ((PostprocessEffect)mEffects[0]).getRender() : false;
149 a13dde77 Leszek Koltunski
    }
150
151
///////////////////////////////////////////////////////////////////////////////////////////////////
152
153 9e771d06 Leszek Koltunski
  int preprocess(DistortedOutputSurface buffer, DistortedNode node)
154 2386a081 Leszek Koltunski
    {
155 9e771d06 Leszek Koltunski
    buffer.setAsOutput();
156 a13dde77 Leszek Koltunski
    DistortedSurface input = node.getInternalSurface();
157 2386a081 Leszek Koltunski
158 a13dde77 Leszek Koltunski
    if( input.setAsInput() )
159
      {
160 247d8225 Leszek Koltunski
      MeshObject mesh = node.getMesh();
161 9e771d06 Leszek Koltunski
      float    margin = mHalo*buffer.mMipmap;
162 2386a081 Leszek Koltunski
163 a13dde77 Leszek Koltunski
      float halfW = input.getWidth() / 2.0f;
164
      float halfH = input.getHeight()/ 2.0f;
165 6c00149d Leszek Koltunski
      float halfZ = halfW*mesh.getZFactor();
166 2386a081 Leszek Koltunski
167 9e771d06 Leszek Koltunski
      DistortedRenderState.setUpStencilMark(mA!=0.0f);
168 a13dde77 Leszek Koltunski
169 9e771d06 Leszek Koltunski
      GLES31.glViewport(0, 0, buffer.mWidth, buffer.mHeight );
170 a13dde77 Leszek Koltunski
171
      mPreProgram.useProgram();
172
173 466450b5 Leszek Koltunski
      GLES31.glBindBuffer(GLES31.GL_ARRAY_BUFFER, mesh.getVBO() );
174 a13dde77 Leszek Koltunski
      GLES31.glVertexAttribPointer(mPreProgram.mAttribute[0], MeshObject.POS_DATA_SIZE, GLES31.GL_FLOAT, false, MeshObject.VERTSIZE, MeshObject.OFFSET0);
175
      GLES31.glVertexAttribPointer(mPreProgram.mAttribute[1], MeshObject.NOR_DATA_SIZE, GLES31.GL_FLOAT, false, MeshObject.VERTSIZE, MeshObject.OFFSET1);
176
      GLES31.glVertexAttribPointer(mPreProgram.mAttribute[2], MeshObject.TEX_DATA_SIZE, GLES31.GL_FLOAT, false, MeshObject.VERTSIZE, MeshObject.OFFSET2);
177
      GLES31.glBindBuffer(GLES31.GL_ARRAY_BUFFER, 0);
178
179 9e771d06 Leszek Koltunski
      node.getEffects().send(halfW, halfH, halfZ, margin, buffer, 2);
180 247d8225 Leszek Koltunski
181 9e771d06 Leszek Koltunski
      if( mA!=0.0f )
182 247d8225 Leszek Koltunski
        {
183 9e771d06 Leszek Koltunski
        GLES31.glUniform4f(mPreColorH, mR, mG, mB, mA);
184
        GLES31.glUniform1i(mPreTextureH, 0);
185 247d8225 Leszek Koltunski
        }
186
187 6c00149d Leszek Koltunski
      GLES31.glDrawArrays(GLES31.GL_TRIANGLE_STRIP, 0, mesh.getNumVertices() );
188 a13dde77 Leszek Koltunski
189
      DistortedRenderState.unsetUpStencilMark();
190
191
      return 1;
192
      }
193
    return 0;
194 2386a081 Leszek Koltunski
    }
195
196 4b9fe2e9 Leszek Koltunski
///////////////////////////////////////////////////////////////////////////////////////////////////
197 cf7394cc leszek
198 9e771d06 Leszek Koltunski
  int postprocess(DistortedFramebuffer buffer)
199 4b9fe2e9 Leszek Koltunski
    {
200 bed13bea leszek
    int numRenders = 0;
201
202 8dccc3c2 Leszek Koltunski
    GLES31.glDisable(GLES31.GL_BLEND);
203
204 1149be8f leszek
    for(int i=0; i<mNumEffects; i++)
205 95c441a2 leszek
      {
206 9e771d06 Leszek Koltunski
      numRenders += ((PostprocessEffect)mEffects[i]).apply(mUniforms,NUM_UNIFORMS*i, buffer);
207 95c441a2 leszek
      }
208 4b9fe2e9 Leszek Koltunski
209 8dccc3c2 Leszek Koltunski
    GLES31.glEnable(GLES31.GL_BLEND);
210
211 bed13bea leszek
    return numRenders;
212 cf7394cc leszek
    }
213 4c1dd6e9 Leszek Koltunski
  }