1 using System;
2 using
UnityEngine;
3
4 namespace
UnityStandardAssets.ImageEffects
5 {
6     
[ExecuteInEditMode]
7     
[AddComponentMenu ("Image Effects/Color Adjustments/Color Correction (Curves, Saturation)")]
8     
public class ColorCorrectionCurves : PostEffectsBase
9     {
10         
public enum ColorCorrectionMode
11         {
12             Simple =
0,
13             Advanced =
1
14         }
15
16         
public AnimationCurve redChannel = new AnimationCurve(new Keyframe(0f,0f), new Keyframe(1f,1f));
17         
public AnimationCurve greenChannel = new AnimationCurve(new Keyframe(0f,0f), new Keyframe(1f,1f));
18         
public AnimationCurve blueChannel = new AnimationCurve(new Keyframe(0f,0f), new Keyframe(1f,1f));
19
20         
public bool useDepthCorrection = false;
21
22         
public AnimationCurve zCurve = new AnimationCurve(new Keyframe(0f,0f), new Keyframe(1f,1f));
23         
public AnimationCurve depthRedChannel = new AnimationCurve(new Keyframe(0f,0f), new Keyframe(1f,1f));
24         
public AnimationCurve depthGreenChannel = new AnimationCurve(new Keyframe(0f,0f), new Keyframe(1f,1f));
25         
public AnimationCurve depthBlueChannel = new AnimationCurve(new Keyframe(0f,0f), new Keyframe(1f,1f));
26
27         
private Material ccMaterial;
28         
private Material ccDepthMaterial;
29         
private Material selectiveCcMaterial;
30
31         
private Texture2D rgbChannelTex;
32         
private Texture2D rgbDepthChannelTex;
33         
private Texture2D zCurveTex;
34
35         
public float saturation = 1.0f;
36
37         
public bool selectiveCc = false;
38
39         
public Color selectiveFromColor = Color.white;
40         
public Color selectiveToColor = Color.white;
41
42         
public ColorCorrectionMode mode;
43
44         
public bool updateTextures = true;
45
46         
public Shader colorCorrectionCurvesShader = null;
47         
public Shader simpleColorCorrectionCurvesShader = null;
48         
public Shader colorCorrectionSelectiveShader = null;
49
50         
private bool updateTexturesOnStartup = true;
51
52
53         
new void Start ()
54         {
55             
base.Start ();
56             updateTexturesOnStartup =
true;
57         }
58
59         
void Awake () { }
60
61
62         
public override bool CheckResources ()
63         {
64             CheckSupport (mode == ColorCorrectionMode.Advanced);
65
66             ccMaterial = CheckShaderAndCreateMaterial (simpleColorCorrectionCurvesShader, ccMaterial);
67             ccDepthMaterial = CheckShaderAndCreateMaterial (colorCorrectionCurvesShader, ccDepthMaterial);
68             selectiveCcMaterial = CheckShaderAndCreateMaterial (colorCorrectionSelectiveShader, selectiveCcMaterial);
69
70             
if (!rgbChannelTex)
71                 rgbChannelTex =
new Texture2D (256, 4, TextureFormat.ARGB32, false, true);
72             
if (!rgbDepthChannelTex)
73                 rgbDepthChannelTex =
new Texture2D (256, 4, TextureFormat.ARGB32, false, true);
74             
if (!zCurveTex)
75                 zCurveTex =
new Texture2D (256, 1, TextureFormat.ARGB32, false, true);
76
77             rgbChannelTex.hideFlags = HideFlags.DontSave;
78             rgbDepthChannelTex.hideFlags = HideFlags.DontSave;
79             zCurveTex.hideFlags = HideFlags.DontSave;
80
81             rgbChannelTex.wrapMode = TextureWrapMode.Clamp;
82             rgbDepthChannelTex.wrapMode = TextureWrapMode.Clamp;
83             zCurveTex.wrapMode = TextureWrapMode.Clamp;
84
85             
if (!isSupported)
86                 ReportAutoDisable ();
87             
return isSupported;
88         }
89
90         
public void UpdateParameters ()
91         {
92             CheckResources();
// textures might not be created if we're tweaking UI while disabled
93
94             
if (redChannel != null && greenChannel != null && blueChannel != null)
95             {
96                 
for (float i = 0.0f; i <= 1.0f; i += 1.0f / 255.0f)
97                 {
98                     
float rCh = Mathf.Clamp (redChannel.Evaluate(i), 0.0f, 1.0f);
99                     
float gCh = Mathf.Clamp (greenChannel.Evaluate(i), 0.0f, 1.0f);
100                     
float bCh = Mathf.Clamp (blueChannel.Evaluate(i), 0.0f, 1.0f);
101
102                     rgbChannelTex.SetPixel ((
int) Mathf.Floor(i*255.0f), 0, new Color(rCh,rCh,rCh) );
103                     rgbChannelTex.SetPixel ((
int) Mathf.Floor(i*255.0f), 1, new Color(gCh,gCh,gCh) );
104                     rgbChannelTex.SetPixel ((
int) Mathf.Floor(i*255.0f), 2, new Color(bCh,bCh,bCh) );
105
106                     
float zC = Mathf.Clamp (zCurve.Evaluate(i), 0.0f,1.0f);
107
108                     zCurveTex.SetPixel ((
int) Mathf.Floor(i*255.0f), 0, new Color(zC,zC,zC) );
109
110                     rCh = Mathf.Clamp (depthRedChannel.Evaluate(i),
0.0f,1.0f);
111                     gCh = Mathf.Clamp (depthGreenChannel.Evaluate(i),
0.0f,1.0f);
112                     bCh = Mathf.Clamp (depthBlueChannel.Evaluate(i),
0.0f,1.0f);
113
114                     rgbDepthChannelTex.SetPixel ((
int) Mathf.Floor(i*255.0f), 0, new Color(rCh,rCh,rCh) );
115                     rgbDepthChannelTex.SetPixel ((
int) Mathf.Floor(i*255.0f), 1, new Color(gCh,gCh,gCh) );
116                     rgbDepthChannelTex.SetPixel ((
int) Mathf.Floor(i*255.0f), 2, new Color(bCh,bCh,bCh) );
117                 }
118
119                 rgbChannelTex.Apply ();
120                 rgbDepthChannelTex.Apply ();
121                 zCurveTex.Apply ();
122             }
123         }
124
125         
void UpdateTextures ()
126         {
127             UpdateParameters ();
128         }
129
130         
void OnRenderImage (RenderTexture source, RenderTexture destination)
131         {
132             
if (CheckResources()==false)
133             {
134                 Graphics.Blit (source, destination);
135                 
return;
136             }
137
138             
if (updateTexturesOnStartup)
139             {
140                 UpdateParameters ();
141                 updateTexturesOnStartup =
false;
142             }
143
144             
if (useDepthCorrection)
145                 GetComponent<Camera>().depthTextureMode |= DepthTextureMode.Depth;
146
147             RenderTexture renderTarget2Use = destination;
148
149             
if (selectiveCc)
150             {
151                 renderTarget2Use = RenderTexture.GetTemporary (source.width, source.height);
152             }
153
154             
if (useDepthCorrection)
155             {
156                 ccDepthMaterial.SetTexture (
"_RgbTex", rgbChannelTex);
157                 ccDepthMaterial.SetTexture (
"_ZCurve", zCurveTex);
158                 ccDepthMaterial.SetTexture (
"_RgbDepthTex", rgbDepthChannelTex);
159                 ccDepthMaterial.SetFloat (
"_Saturation", saturation);
160
161                 Graphics.Blit (source, renderTarget2Use, ccDepthMaterial);
162             }
163             
else
164             {
165                 ccMaterial.SetTexture (
"_RgbTex", rgbChannelTex);
166                 ccMaterial.SetFloat (
"_Saturation", saturation);
167
168                 Graphics.Blit (source, renderTarget2Use, ccMaterial);
169             }
170
171             
if (selectiveCc)
172             {
173                 selectiveCcMaterial.SetColor (
"selColor", selectiveFromColor);
174                 selectiveCcMaterial.SetColor (
"targetColor", selectiveToColor);
175                 Graphics.Blit (renderTarget2Use, destination, selectiveCcMaterial);
176
177                 RenderTexture.ReleaseTemporary (renderTarget2Use);
178             }
179         }
180     }
181 }


Tìm kết quả nhanh