IMHOTEP Framework
 All Classes Namespaces Functions Variables Enumerations Enumerator Properties Pages
Bloom.cs
1 using System;
2 using UnityEngine;
3 
4 namespace UnityStandardAssets.ImageEffects
5 {
6  [ExecuteInEditMode]
7  [RequireComponent (typeof(Camera))]
8  [AddComponentMenu ("Image Effects/Bloom and Glow/Bloom")]
9  public class Bloom : PostEffectsBase
10  {
11  public enum LensFlareStyle
12  {
13  Ghosting = 0,
14  Anamorphic = 1,
15  Combined = 2,
16  }
17 
18  public enum TweakMode
19  {
20  Basic = 0,
21  Complex = 1,
22  }
23 
24  public enum HDRBloomMode
25  {
26  Auto = 0,
27  On = 1,
28  Off = 2,
29  }
30 
31  public enum BloomScreenBlendMode
32  {
33  Screen = 0,
34  Add = 1,
35  }
36 
37  public enum BloomQuality
38  {
39  Cheap = 0,
40  High = 1,
41  }
42 
43  public TweakMode tweakMode = 0;
44  public BloomScreenBlendMode screenBlendMode = BloomScreenBlendMode.Add;
45 
46  public HDRBloomMode hdr = HDRBloomMode.Auto;
47  private bool doHdr = false;
48  public float sepBlurSpread = 2.5f;
49 
50  public BloomQuality quality = BloomQuality.High;
51 
52  public float bloomIntensity = 0.5f;
53  public float bloomThreshold = 0.5f;
54  public Color bloomThresholdColor = Color.white;
55  public int bloomBlurIterations = 2;
56 
57  public int hollywoodFlareBlurIterations = 2;
58  public float flareRotation = 0.0f;
59  public LensFlareStyle lensflareMode = (LensFlareStyle) 1;
60  public float hollyStretchWidth = 2.5f;
61  public float lensflareIntensity = 0.0f;
62  public float lensflareThreshold = 0.3f;
63  public float lensFlareSaturation = 0.75f;
64  public Color flareColorA = new Color (0.4f, 0.4f, 0.8f, 0.75f);
65  public Color flareColorB = new Color (0.4f, 0.8f, 0.8f, 0.75f);
66  public Color flareColorC = new Color (0.8f, 0.4f, 0.8f, 0.75f);
67  public Color flareColorD = new Color (0.8f, 0.4f, 0.0f, 0.75f);
68  public Texture2D lensFlareVignetteMask;
69 
70  public Shader lensFlareShader;
71  private Material lensFlareMaterial;
72 
73  public Shader screenBlendShader;
74  private Material screenBlend;
75 
76  public Shader blurAndFlaresShader;
77  private Material blurAndFlaresMaterial;
78 
79  public Shader brightPassFilterShader;
80  private Material brightPassFilterMaterial;
81 
82 
83  public override bool CheckResources ()
84  {
85  CheckSupport (false);
86 
87  screenBlend = CheckShaderAndCreateMaterial (screenBlendShader, screenBlend);
88  lensFlareMaterial = CheckShaderAndCreateMaterial(lensFlareShader,lensFlareMaterial);
89  blurAndFlaresMaterial = CheckShaderAndCreateMaterial (blurAndFlaresShader, blurAndFlaresMaterial);
90  brightPassFilterMaterial = CheckShaderAndCreateMaterial(brightPassFilterShader, brightPassFilterMaterial);
91 
92  if (!isSupported)
93  ReportAutoDisable ();
94  return isSupported;
95  }
96 
97  public void OnRenderImage (RenderTexture source, RenderTexture destination)
98  {
99  if (CheckResources()==false)
100  {
101  Graphics.Blit (source, destination);
102  return;
103  }
104 
105  // screen blend is not supported when HDR is enabled (will cap values)
106 
107  doHdr = false;
108  if (hdr == HDRBloomMode.Auto)
109  doHdr = source.format == RenderTextureFormat.ARGBHalf && GetComponent<Camera>().hdr;
110  else {
111  doHdr = hdr == HDRBloomMode.On;
112  }
113 
114  doHdr = doHdr && supportHDRTextures;
115 
116  BloomScreenBlendMode realBlendMode = screenBlendMode;
117  if (doHdr)
118  realBlendMode = BloomScreenBlendMode.Add;
119 
120  var rtFormat= (doHdr) ? RenderTextureFormat.ARGBHalf : RenderTextureFormat.Default;
121  var rtW2= source.width/2;
122  var rtH2= source.height/2;
123  var rtW4= source.width/4;
124  var rtH4= source.height/4;
125 
126  float widthOverHeight = (1.0f * source.width) / (1.0f * source.height);
127  float oneOverBaseSize = 1.0f / 512.0f;
128 
129  // downsample
130  RenderTexture quarterRezColor = RenderTexture.GetTemporary (rtW4, rtH4, 0, rtFormat);
131  RenderTexture halfRezColorDown = RenderTexture.GetTemporary (rtW2, rtH2, 0, rtFormat);
132  if (quality > BloomQuality.Cheap) {
133  Graphics.Blit (source, halfRezColorDown, screenBlend, 2);
134  RenderTexture rtDown4 = RenderTexture.GetTemporary (rtW4, rtH4, 0, rtFormat);
135  Graphics.Blit (halfRezColorDown, rtDown4, screenBlend, 2);
136  Graphics.Blit (rtDown4, quarterRezColor, screenBlend, 6);
137  RenderTexture.ReleaseTemporary(rtDown4);
138  }
139  else {
140  Graphics.Blit (source, halfRezColorDown);
141  Graphics.Blit (halfRezColorDown, quarterRezColor, screenBlend, 6);
142  }
143  RenderTexture.ReleaseTemporary (halfRezColorDown);
144 
145  // cut colors (thresholding)
146  RenderTexture secondQuarterRezColor = RenderTexture.GetTemporary (rtW4, rtH4, 0, rtFormat);
147  BrightFilter (bloomThreshold * bloomThresholdColor, quarterRezColor, secondQuarterRezColor);
148 
149  // blurring
150 
151  if (bloomBlurIterations < 1) bloomBlurIterations = 1;
152  else if (bloomBlurIterations > 10) bloomBlurIterations = 10;
153 
154  for (int iter = 0; iter < bloomBlurIterations; iter++)
155  {
156  float spreadForPass = (1.0f + (iter * 0.25f)) * sepBlurSpread;
157 
158  // vertical blur
159  RenderTexture blur4 = RenderTexture.GetTemporary (rtW4, rtH4, 0, rtFormat);
160  blurAndFlaresMaterial.SetVector ("_Offsets", new Vector4 (0.0f, spreadForPass * oneOverBaseSize, 0.0f, 0.0f));
161  Graphics.Blit (secondQuarterRezColor, blur4, blurAndFlaresMaterial, 4);
162  RenderTexture.ReleaseTemporary(secondQuarterRezColor);
163  secondQuarterRezColor = blur4;
164 
165  // horizontal blur
166  blur4 = RenderTexture.GetTemporary (rtW4, rtH4, 0, rtFormat);
167  blurAndFlaresMaterial.SetVector ("_Offsets", new Vector4 ((spreadForPass / widthOverHeight) * oneOverBaseSize, 0.0f, 0.0f, 0.0f));
168  Graphics.Blit (secondQuarterRezColor, blur4, blurAndFlaresMaterial, 4);
169  RenderTexture.ReleaseTemporary (secondQuarterRezColor);
170  secondQuarterRezColor = blur4;
171 
172  if (quality > BloomQuality.Cheap)
173  {
174  if (iter == 0)
175  {
176  Graphics.SetRenderTarget(quarterRezColor);
177  GL.Clear(false, true, Color.black); // Clear to avoid RT restore
178  Graphics.Blit (secondQuarterRezColor, quarterRezColor);
179  }
180  else
181  {
182  quarterRezColor.MarkRestoreExpected(); // using max blending, RT restore expected
183  Graphics.Blit (secondQuarterRezColor, quarterRezColor, screenBlend, 10);
184  }
185  }
186  }
187 
188  if (quality > BloomQuality.Cheap)
189  {
190  Graphics.SetRenderTarget(secondQuarterRezColor);
191  GL.Clear(false, true, Color.black); // Clear to avoid RT restore
192  Graphics.Blit (quarterRezColor, secondQuarterRezColor, screenBlend, 6);
193  }
194 
195  // lens flares: ghosting, anamorphic or both (ghosted anamorphic flares)
196 
197  if (lensflareIntensity > Mathf.Epsilon)
198  {
199 
200  RenderTexture rtFlares4 = RenderTexture.GetTemporary (rtW4, rtH4, 0, rtFormat);
201 
202  if (lensflareMode == 0)
203  {
204  // ghosting only
205 
206  BrightFilter (lensflareThreshold, secondQuarterRezColor, rtFlares4);
207 
208  if (quality > BloomQuality.Cheap)
209  {
210  // smooth a little
211  blurAndFlaresMaterial.SetVector ("_Offsets", new Vector4 (0.0f, (1.5f) / (1.0f * quarterRezColor.height), 0.0f, 0.0f));
212  Graphics.SetRenderTarget(quarterRezColor);
213  GL.Clear(false, true, Color.black); // Clear to avoid RT restore
214  Graphics.Blit (rtFlares4, quarterRezColor, blurAndFlaresMaterial, 4);
215 
216  blurAndFlaresMaterial.SetVector ("_Offsets", new Vector4 ((1.5f) / (1.0f * quarterRezColor.width), 0.0f, 0.0f, 0.0f));
217  Graphics.SetRenderTarget(rtFlares4);
218  GL.Clear(false, true, Color.black); // Clear to avoid RT restore
219  Graphics.Blit (quarterRezColor, rtFlares4, blurAndFlaresMaterial, 4);
220  }
221 
222  // no ugly edges!
223  Vignette (0.975f, rtFlares4, rtFlares4);
224  BlendFlares (rtFlares4, secondQuarterRezColor);
225  }
226  else
227  {
228 
229  //Vignette (0.975ff, rtFlares4, rtFlares4);
230  //DrawBorder(rtFlares4, screenBlend, 8);
231 
232  float flareXRot = 1.0f * Mathf.Cos(flareRotation);
233  float flareyRot = 1.0f * Mathf.Sin(flareRotation);
234 
235  float stretchWidth = (hollyStretchWidth * 1.0f / widthOverHeight) * oneOverBaseSize;
236 
237  blurAndFlaresMaterial.SetVector ("_Offsets", new Vector4 (flareXRot, flareyRot, 0.0f, 0.0f));
238  blurAndFlaresMaterial.SetVector ("_Threshhold", new Vector4 (lensflareThreshold, 1.0f, 0.0f, 0.0f));
239  blurAndFlaresMaterial.SetVector ("_TintColor", new Vector4 (flareColorA.r, flareColorA.g, flareColorA.b, flareColorA.a) * flareColorA.a * lensflareIntensity);
240  blurAndFlaresMaterial.SetFloat ("_Saturation", lensFlareSaturation);
241 
242  // "pre and cut"
243  quarterRezColor.DiscardContents();
244  Graphics.Blit (rtFlares4, quarterRezColor, blurAndFlaresMaterial, 2);
245  // "post"
246  rtFlares4.DiscardContents();
247  Graphics.Blit (quarterRezColor, rtFlares4, blurAndFlaresMaterial, 3);
248 
249  blurAndFlaresMaterial.SetVector ("_Offsets", new Vector4 (flareXRot * stretchWidth, flareyRot * stretchWidth, 0.0f, 0.0f));
250  // stretch 1st
251  blurAndFlaresMaterial.SetFloat ("_StretchWidth", hollyStretchWidth);
252  quarterRezColor.DiscardContents();
253  Graphics.Blit (rtFlares4, quarterRezColor, blurAndFlaresMaterial, 1);
254  // stretch 2nd
255  blurAndFlaresMaterial.SetFloat ("_StretchWidth", hollyStretchWidth * 2.0f);
256  rtFlares4.DiscardContents();
257  Graphics.Blit (quarterRezColor, rtFlares4, blurAndFlaresMaterial, 1);
258  // stretch 3rd
259  blurAndFlaresMaterial.SetFloat ("_StretchWidth", hollyStretchWidth * 4.0f);
260  quarterRezColor.DiscardContents();
261  Graphics.Blit (rtFlares4, quarterRezColor, blurAndFlaresMaterial, 1);
262 
263  // additional blur passes
264  for (int iter = 0; iter < hollywoodFlareBlurIterations; iter++)
265  {
266  stretchWidth = (hollyStretchWidth * 2.0f / widthOverHeight) * oneOverBaseSize;
267 
268  blurAndFlaresMaterial.SetVector ("_Offsets", new Vector4 (stretchWidth * flareXRot, stretchWidth * flareyRot, 0.0f, 0.0f));
269  rtFlares4.DiscardContents();
270  Graphics.Blit (quarterRezColor, rtFlares4, blurAndFlaresMaterial, 4);
271 
272  blurAndFlaresMaterial.SetVector ("_Offsets", new Vector4 (stretchWidth * flareXRot, stretchWidth * flareyRot, 0.0f, 0.0f));
273  quarterRezColor.DiscardContents();
274  Graphics.Blit (rtFlares4, quarterRezColor, blurAndFlaresMaterial, 4);
275  }
276 
277  if (lensflareMode == (LensFlareStyle) 1)
278  // anamorphic lens flares
279  AddTo (1.0f, quarterRezColor, secondQuarterRezColor);
280  else
281  {
282  // "combined" lens flares
283 
284  Vignette (1.0f, quarterRezColor, rtFlares4);
285  BlendFlares (rtFlares4, quarterRezColor);
286  AddTo (1.0f, quarterRezColor, secondQuarterRezColor);
287  }
288  }
289  RenderTexture.ReleaseTemporary (rtFlares4);
290  }
291 
292  int blendPass = (int) realBlendMode;
293  //if (Mathf.Abs(chromaticBloom) < Mathf.Epsilon)
294  // blendPass += 4;
295 
296  screenBlend.SetFloat ("_Intensity", bloomIntensity);
297  screenBlend.SetTexture ("_ColorBuffer", source);
298 
299  if (quality > BloomQuality.Cheap)
300  {
301  RenderTexture halfRezColorUp = RenderTexture.GetTemporary (rtW2, rtH2, 0, rtFormat);
302  Graphics.Blit (secondQuarterRezColor, halfRezColorUp);
303  Graphics.Blit (halfRezColorUp, destination, screenBlend, blendPass);
304  RenderTexture.ReleaseTemporary (halfRezColorUp);
305  }
306  else
307  Graphics.Blit (secondQuarterRezColor, destination, screenBlend, blendPass);
308 
309  RenderTexture.ReleaseTemporary (quarterRezColor);
310  RenderTexture.ReleaseTemporary (secondQuarterRezColor);
311  }
312 
313  private void AddTo (float intensity_, RenderTexture from, RenderTexture to)
314  {
315  screenBlend.SetFloat ("_Intensity", intensity_);
316  to.MarkRestoreExpected(); // additive blending, RT restore expected
317  Graphics.Blit (from, to, screenBlend, 9);
318  }
319 
320  private void BlendFlares (RenderTexture from, RenderTexture to)
321  {
322  lensFlareMaterial.SetVector ("colorA", new Vector4 (flareColorA.r, flareColorA.g, flareColorA.b, flareColorA.a) * lensflareIntensity);
323  lensFlareMaterial.SetVector ("colorB", new Vector4 (flareColorB.r, flareColorB.g, flareColorB.b, flareColorB.a) * lensflareIntensity);
324  lensFlareMaterial.SetVector ("colorC", new Vector4 (flareColorC.r, flareColorC.g, flareColorC.b, flareColorC.a) * lensflareIntensity);
325  lensFlareMaterial.SetVector ("colorD", new Vector4 (flareColorD.r, flareColorD.g, flareColorD.b, flareColorD.a) * lensflareIntensity);
326  to.MarkRestoreExpected(); // additive blending, RT restore expected
327  Graphics.Blit (from, to, lensFlareMaterial);
328  }
329 
330  private void BrightFilter (float thresh, RenderTexture from, RenderTexture to)
331  {
332  brightPassFilterMaterial.SetVector ("_Threshhold", new Vector4 (thresh, thresh, thresh, thresh));
333  Graphics.Blit (from, to, brightPassFilterMaterial, 0);
334  }
335 
336  private void BrightFilter (Color threshColor, RenderTexture from, RenderTexture to)
337  {
338  brightPassFilterMaterial.SetVector ("_Threshhold", threshColor);
339  Graphics.Blit (from, to, brightPassFilterMaterial, 1);
340  }
341 
342  private void Vignette (float amount, RenderTexture from, RenderTexture to)
343  {
344  if (lensFlareVignetteMask)
345  {
346  screenBlend.SetTexture ("_ColorBuffer", lensFlareVignetteMask);
347  to.MarkRestoreExpected(); // using blending, RT restore expected
348  Graphics.Blit (from == to ? null : from, to, screenBlend, from == to ? 7 : 3);
349  }
350  else if (from != to)
351  {
352  Graphics.SetRenderTarget (to);
353  GL.Clear(false, true, Color.black); // clear destination to avoid RT restore
354  Graphics.Blit (from, to);
355  }
356  }
357  }
358 }
Generated by   doxygen 1.8.6