Pivot baking example

Added example of baking pivot at rotating in a shader.. Added option to bake rotation of pivot into uvs Added option to bake local or world space pivot data (default is local)
WondermSwift · Aug 23, 2016 · c995861 · c995861
1 parent 14c12c2
commit c995861
Show file tree

Hide file tree

Showing 13 changed files with 193 additions and 5 deletions.
diff --git a/Editor/VertexPainterWindow_BakePivot.cs b/Editor/VertexPainterWindow_BakePivot.cs
@@ -22,6 +22,53 @@ enum PivotTarget
       }
 
       PivotTarget pivotTarget = PivotTarget.UV2;
+      bool bakePivotUseLocal = true;
+      void BakeRotation()
+      {
+         switch (pivotTarget)
+         {
+            case PivotTarget.UV0:
+               {
+                  InitBakeChannel(BakeChannel.UV0);
+                  foreach (PaintJob job in jobs)
+                  {
+                     Vector3 lp = bakePivotUseLocal ? job.meshFilter.transform.localRotation.eulerAngles : job.meshFilter.transform.rotation.eulerAngles;
+                     job.stream.SetUV0(new Vector4(lp.x, lp.y, lp.z, UnityEngine.Random.Range(0.0f, 1.0f)), job.verts.Length);
+                  }
+                  break;
+               }
+            case PivotTarget.UV1:
+               {
+                  InitBakeChannel(BakeChannel.UV1);
+                  foreach (PaintJob job in jobs)
+                  {
+                     Vector3 lp = bakePivotUseLocal ? job.meshFilter.transform.localRotation.eulerAngles : job.meshFilter.transform.rotation.eulerAngles;
+                     job.stream.SetUV1(new Vector4(lp.x, lp.y, lp.z, UnityEngine.Random.Range(0.0f, 1.0f)), job.verts.Length);
+                  }
+                  break;
+               }
+            case PivotTarget.UV2:
+               {
+                  InitBakeChannel(BakeChannel.UV2);
+                  foreach (PaintJob job in jobs)
+                  {
+                     Vector3 lp = bakePivotUseLocal ? job.meshFilter.transform.localRotation.eulerAngles : job.meshFilter.transform.rotation.eulerAngles;
+                     job.stream.SetUV2(new Vector4(lp.x, lp.y, lp.z, UnityEngine.Random.Range(0.0f, 1.0f)), job.verts.Length);
+                  }
+                  break;
+               }
+            case PivotTarget.UV3:
+               {
+                  InitBakeChannel(BakeChannel.UV3);
+                  foreach (PaintJob job in jobs)
+                  {
+                     Vector3 lp = bakePivotUseLocal ? job.meshFilter.transform.localRotation.eulerAngles : job.meshFilter.transform.rotation.eulerAngles;
+                     job.stream.SetUV3(new Vector4(lp.x, lp.y, lp.z, UnityEngine.Random.Range(0.0f, 1.0f)), job.verts.Length);
+                  }
+                  break;
+               }
+         }
+      }
 
       void BakePivot()
       {
@@ -32,7 +79,7 @@ void BakePivot()
                InitBakeChannel(BakeChannel.UV0);
                foreach (PaintJob job in jobs)
                {
-                  Vector3 lp = job.meshFilter.transform.localPosition;
+                  Vector3 lp = bakePivotUseLocal ? job.meshFilter.transform.localPosition : job.meshFilter.transform.position;
                   job.stream.SetUV0(new Vector4(lp.x, lp.y, lp.z, UnityEngine.Random.Range(0.0f, 1.0f)), job.verts.Length);
                }
                break;
@@ -42,7 +89,7 @@ void BakePivot()
                InitBakeChannel(BakeChannel.UV1);
                foreach (PaintJob job in jobs)
                {
-                  Vector3 lp = job.meshFilter.transform.localPosition;
+                  Vector3 lp = bakePivotUseLocal ? job.meshFilter.transform.localPosition : job.meshFilter.transform.position;
                   job.stream.SetUV1(new Vector4(lp.x, lp.y, lp.z, UnityEngine.Random.Range(0.0f, 1.0f)), job.verts.Length);
                }
                break;
@@ -52,7 +99,7 @@ void BakePivot()
                   InitBakeChannel(BakeChannel.UV2);
                   foreach (PaintJob job in jobs)
                   {
-                     Vector3 lp = job.meshFilter.transform.localPosition;
+                     Vector3 lp = bakePivotUseLocal ? job.meshFilter.transform.localPosition : job.meshFilter.transform.position;
                      job.stream.SetUV2(new Vector4(lp.x, lp.y, lp.z, UnityEngine.Random.Range(0.0f, 1.0f)), job.verts.Length);
                   }
                   break;
@@ -62,7 +109,7 @@ void BakePivot()
                InitBakeChannel(BakeChannel.UV3);
                foreach (PaintJob job in jobs)
                {
-                  Vector3 lp = job.meshFilter.transform.localPosition;
+                  Vector3 lp = bakePivotUseLocal ? job.meshFilter.transform.localPosition : job.meshFilter.transform.position;
                   job.stream.SetUV3(new Vector4(lp.x, lp.y, lp.z, UnityEngine.Random.Range(0.0f, 1.0f)), job.verts.Length);
                }
                break;

diff --git a/Editor/VertexPainterWindow_GUI.cs b/Editor/VertexPainterWindow_GUI.cs
@@ -560,13 +560,18 @@ void DrawBakeGUI()
          GUILayout.Space(10);
          GUILayout.Box("Bake Pivot", new GUILayoutOption[]{GUILayout.ExpandWidth(true), GUILayout.Height(20)});
          pivotTarget = (PivotTarget)EditorGUILayout.EnumPopup("Store in", pivotTarget);
+         bakePivotUseLocal = EditorGUILayout.Toggle("Use Local Space", bakePivotUseLocal);
 
          EditorGUILayout.BeginHorizontal();
          EditorGUILayout.Space();
-         if (GUILayout.Button("Bake"))
+         if (GUILayout.Button("Bake Pivot"))
          {
             BakePivot();
          }
+         if (GUILayout.Button("Bake Rotation"))
+         {
+            BakeRotation();
+         }
          EditorGUILayout.Space();
          EditorGUILayout.EndHorizontal();
 

diff --git a/Examples/PivotBaking.meta b/Examples/PivotBaking.meta
diff --git a/Examples/PivotBaking/pivot.mat b/Examples/PivotBaking/pivot.mat
diff --git a/Examples/PivotBaking/pivot.mat.meta b/Examples/PivotBaking/pivot.mat.meta
diff --git a/Examples/PivotBaking/pivot.shader b/Examples/PivotBaking/pivot.shader
@@ -0,0 +1,86 @@
+Shader "Unlit/pivot"
+{
+	Properties
+	{
+
+	}
+	SubShader
+	{
+		Tags { "RenderType"="Opaque" }
+		LOD 100
+
+		Pass
+		{
+			CGPROGRAM
+			#pragma vertex vert
+			#pragma fragment frag
+			// make fog work
+			#pragma multi_compile_fog
+
+			#include "UnityCG.cginc"
+
+			struct appdata
+			{
+				float4 vertex : POSITION;
+            float4 pivot : TEXCOORD2;
+			};
+
+			struct v2f
+			{
+				float4 vertex : SV_POSITION;
+            float4 color : TEXCOORD0;
+			};
+
+
+			v2f vert (appdata v)
+			{
+				v2f o;
+
+            // total rotation to apply in radians
+            float3 rotate = float3(0, _Time.y, 0);
+            float3 c = cos(rotate);
+            float3 s = sin(rotate);
+
+            // make a matrix for each axis
+            float4x4 rotateXMtx  = float4x4( 1, 0, 0, 0,
+                                       0, c.x, -s.x,   0,
+                                       0, s.x, c.x, 0,
+                                       0, 0, 0, 1);
+
+
+            float4x4 rotateYMtx  = float4x4( c.y, 0, s.y, 0,
+                                       0, 1, 0, 0,
+                                       -s.y,   0, c.y, 0,
+                                       0, 0, 0, 1);
+
+
+            float4x4 rotateZMtx  = float4x4( c.z, -s.z,   0, 0,
+                                       s.z, c.z, 0, 0,
+                                       0, 0, 1, 0,
+                                       0, 0, 0, 1);
+
+            // compute the vertex location as an offset from the pivot point
+            float4 pivot = v.pivot;
+            float4 offset = v.vertex - pivot;
+            // rotate around each axis
+            float4 rx = mul(offset, rotateXMtx);
+            float4 rxy = mul(rx, rotateYMtx);
+            float4 rxyz = mul(rxy, rotateZMtx);
+
+            // take the final rotation, and add the pivot back in to get our final position, then transform into MVP space
+            o.vertex = mul(UNITY_MATRIX_MVP, rxyz + pivot);
+            // the baker puts a random number into the W channel. This is VERY useful, for instance, if you want
+            // each object to rotate at a random speed or at a random angle, or for use in a random function as
+            // a seed.
+            o.color = v.pivot.wwww;
+				return o;
+			}
+
+			fixed4 frag (v2f i) : SV_Target
+			{
+				return i.color;
+         }
+			ENDCG
+		}
+	}
+}
diff --git a/Examples/PivotBaking/pivot.shader.meta b/Examples/PivotBaking/pivot.shader.meta
diff --git a/Examples/PivotBaking/pivot_combined.asset b/Examples/PivotBaking/pivot_combined.asset
diff --git a/Examples/PivotBaking/pivot_combined.asset.meta b/Examples/PivotBaking/pivot_combined.asset.meta
diff --git a/Examples/PivotBaking/pivot_complete.unity b/Examples/PivotBaking/pivot_complete.unity
diff --git a/Examples/PivotBaking/pivot_complete.unity.meta b/Examples/PivotBaking/pivot_complete.unity.meta
diff --git a/Examples/PivotBaking/pivot_start.unity b/Examples/PivotBaking/pivot_start.unity
diff --git a/Examples/PivotBaking/pivot_start.unity.meta b/Examples/PivotBaking/pivot_start.unity.meta