模仿Flickr 使用GPUImage的实时相机滤镜(iOS源代码)

from: http://code1.okbase.net/codefile/GPUImagePoissonBlendFilter.m_2016011433675_220.htm

1. #import "GPUImagePoissonBlendFilter.h"
2. 
   
3. #if TARGET_IPHONE_SIMULATOR || TARGET_OS_IPHONE
4. NSString *const kGPUImagePoissonBlendFragmentShaderString = SHADER_STRING
5. (
6. precision mediump float;
7. 
   
8. varying vec2 textureCoordinate;
9. varying vec2 leftTextureCoordinate;
10. varying vec2 rightTextureCoordinate;
11. varying vec2 topTextureCoordinate;
12. varying vec2 bottomTextureCoordinate;
13. 
    
14. varying vec2 textureCoordinate2;
15. varying vec2 leftTextureCoordinate2;
16. varying vec2 rightTextureCoordinate2;
17. varying vec2 topTextureCoordinate2;
18. varying vec2 bottomTextureCoordinate2;
19. 
    
20. uniform sampler2D inputImageTexture;
21. uniform sampler2D inputImageTexture2;
22. 
    
23. uniform lowp float mixturePercent;
24. 
    
25. void main()
26. {
27. vec4 centerColor = texture2D(inputImageTexture, textureCoordinate);
28. vec3 bottomColor = texture2D(inputImageTexture, bottomTextureCoordinate).rgb;
29. vec3 leftColor = texture2D(inputImageTexture, leftTextureCoordinate).rgb;
30. vec3 rightColor = texture2D(inputImageTexture, rightTextureCoordinate).rgb;
31. vec3 topColor = texture2D(inputImageTexture, topTextureCoordinate).rgb;
32. 
    
33. vec4 centerColor2 = texture2D(inputImageTexture2, textureCoordinate2);
34. vec3 bottomColor2 = texture2D(inputImageTexture2, bottomTextureCoordinate2).rgb;
35. vec3 leftColor2 = texture2D(inputImageTexture2, leftTextureCoordinate2).rgb;
36. vec3 rightColor2 = texture2D(inputImageTexture2, rightTextureCoordinate2).rgb;
37. vec3 topColor2 = texture2D(inputImageTexture2, topTextureCoordinate2).rgb;
38. 
    
39. vec3 meanColor = (bottomColor + leftColor + rightColor + topColor) / 4.0;
40. vec3 diffColor = centerColor.rgb - meanColor;
41. 
    
42. vec3 meanColor2 = (bottomColor2 + leftColor2 + rightColor2 + topColor2) / 4.0;
43. vec3 diffColor2 = centerColor2.rgb - meanColor2;
44. 
    
45. vec3 gradColor = (meanColor + diffColor2);
46. 
    
47. gl_FragColor = vec4(mix(centerColor.rgb, gradColor, centerColor2.a * mixturePercent), centerColor.a);
48. }
49. );
50. #else
51. NSString *const kGPUImagePoissonBlendFragmentShaderString = SHADER_STRING
52. (
53. varying vec2 textureCoordinate;
54. varying vec2 leftTextureCoordinate;
55. varying vec2 rightTextureCoordinate;
56. varying vec2 topTextureCoordinate;
57. varying vec2 bottomTextureCoordinate;
58. 
    
59. varying vec2 textureCoordinate2;
60. varying vec2 leftTextureCoordinate2;
61. varying vec2 rightTextureCoordinate2;
62. varying vec2 topTextureCoordinate2;
63. varying vec2 bottomTextureCoordinate2;
64. 
    
65. uniform sampler2D inputImageTexture;
66. uniform sampler2D inputImageTexture2;
67. 
    
68. uniform float mixturePercent;
69. 
    
70. void main()
71. {
72. vec4 centerColor = texture2D(inputImageTexture, textureCoordinate);
73. vec3 bottomColor = texture2D(inputImageTexture, bottomTextureCoordinate).rgb;
74. vec3 leftColor = texture2D(inputImageTexture, leftTextureCoordinate).rgb;
75. vec3 rightColor = texture2D(inputImageTexture, rightTextureCoordinate).rgb;
76. vec3 topColor = texture2D(inputImageTexture, topTextureCoordinate).rgb;
77. 
    
78. vec4 centerColor2 = texture2D(inputImageTexture2, textureCoordinate2);
79. vec3 bottomColor2 = texture2D(inputImageTexture2, bottomTextureCoordinate2).rgb;
80. vec3 leftColor2 = texture2D(inputImageTexture2, leftTextureCoordinate2).rgb;
81. vec3 rightColor2 = texture2D(inputImageTexture2, rightTextureCoordinate2).rgb;
82. vec3 topColor2 = texture2D(inputImageTexture2, topTextureCoordinate2).rgb;
83. 
    
84. vec3 meanColor = (bottomColor + leftColor + rightColor + topColor) / 4.0;
85. vec3 diffColor = centerColor.rgb - meanColor;
86. 
    
87. vec3 meanColor2 = (bottomColor2 + leftColor2 + rightColor2 + topColor2) / 4.0;
88. vec3 diffColor2 = centerColor2.rgb - meanColor2;
89. 
    
90. vec3 gradColor = (meanColor + diffColor2);
91. 
    
92. gl_FragColor = vec4(mix(centerColor.rgb, gradColor, centerColor2.a * mixturePercent), centerColor.a);
93. }
94. );
95. #endif
96. 
    
97. @implementation GPUImagePoissonBlendFilter
98. 
    
99. @synthesize mix = _mix;
100. @synthesize numIterations = _numIterations;
101. 
     
102. - (id)init;
103. {
104. if (!(self = [super initWithFragmentShaderFromString:kGPUImagePoissonBlendFragmentShaderString]))
105. {
106. return nil;
107. }
108. 
     
109. mixUniform = [filterProgram uniformIndex:@"mixturePercent"];
110. self.mix = 0.5;
111. 
     
112. self.numIterations = 10;
113. 
     
114. return self;
115. }
116. 
     
117. - (void)setMix:(CGFloat)newValue;
118. {
119. _mix = newValue;
120. 
     
121. [self setFloat:_mix forUniform:mixUniform program:filterProgram];
122. }
123. 
     
124. //- (void)setOutputFBO;
125. //{
126. // if (self.numIterations % 2 == 1) {
127. // [self setSecondFilterFBO];
128. // } else {
129. // [self setFilterFBO];
130. // }
131. //}
132. 
     
133. - (void)renderToTextureWithVertices:(const GLfloat *)vertices textureCoordinates:(const GLfloat *)textureCoordinates;
134. {
135. // Run the first stage of the two-pass filter
136. [GPUImageContext setActiveShaderProgram:filterProgram];
137. 
     
138. [super renderToTextureWithVertices:vertices textureCoordinates:textureCoordinates];
139. 
     
140. for (int pass = 1; pass < self.numIterations; pass++) {
141. 
     
142. if (pass % 2 == 0) {
143. 
     
144. [GPUImageContext setActiveShaderProgram:filterProgram];
145. 
     
146. // TODO: This will over-unlock the incoming framebuffer
147. [super renderToTextureWithVertices:vertices textureCoordinates:[[self class] textureCoordinatesForRotation:kGPUImageNoRotation]];
148. } else {
149. // Run the second stage of the two-pass filter
150. secondOutputFramebuffer = [[GPUImageContext sharedFramebufferCache] fetchFramebufferForSize:[self sizeOfFBO] textureOptions:self.outputTextureOptions onlyTexture:NO];
151. [secondOutputFramebuffer activateFramebuffer];
152. 
     
153. [GPUImageContext setActiveShaderProgram:filterProgram];
154. 
     
155. glClearColor(backgroundColorRed, backgroundColorGreen, backgroundColorBlue, backgroundColorAlpha);
156. glClear(GL_COLOR_BUFFER_BIT);
157. 
     
158. glActiveTexture(GL_TEXTURE2);
159. glBindTexture(GL_TEXTURE_2D, [outputFramebuffer texture]);
160. glUniform1i(filterInputTextureUniform, 2);
161. 
     
162. glActiveTexture(GL_TEXTURE3);
163. glBindTexture(GL_TEXTURE_2D, [secondInputFramebuffer texture]);
164. glUniform1i(filterInputTextureUniform2, 3);
165. 
     
166. glVertexAttribPointer(filterPositionAttribute, 2, GL_FLOAT, 0, 0, vertices);
167. glVertexAttribPointer(filterTextureCoordinateAttribute, 2, GL_FLOAT, 0, 0, [[self class] textureCoordinatesForRotation:kGPUImageNoRotation]);
168. glVertexAttribPointer(filterSecondTextureCoordinateAttribute, 2, GL_FLOAT, 0, 0, [[self class] textureCoordinatesForRotation:inputRotation2]);
169. 
     
170. glDrawArrays(GL_TRIANGLE_STRIP, 0, 4);
171. }
172. }
173. }
174. 
     
175. @end// 鐗堟潈灞炰簬鍘熶綔鑰?// http://code4app.com (cn) http://code4app.net (en)
176. // 鍙戝竷浠ｇ爜浜庢渶涓撲笟鐨勬簮鐮佸垎浜綉绔? Code4App.com