Added pre and post compiler callbacks for compiling UdonSharpProgramA…

…sset.
MerlinVR · Nov 28, 2020 · 5bd9ca5 · 5bd9ca5
1 parent 79f95eb
commit 5bd9ca5
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/Assets/UdonSharp/Editor/UdonSharpCompiler.cs b/Assets/UdonSharp/Editor/UdonSharpCompiler.cs
@@ -42,6 +42,10 @@ public class CompileTaskResult
         private CompilationModule[] modules;
         private bool isEditorBuild = true;
 
+        public delegate void ProgramAssetCompileCallback(UdonSharpProgramAsset programAsset);
+        public static event ProgramAssetCompileCallback beforeProgramAssetCompile;
+        public static event ProgramAssetCompileCallback afterProgramAssetCompile;
+
         private static int initAssemblyCounter = 0;
 
         public UdonSharpCompiler(UdonSharpProgramAsset programAsset, bool editorBuild = true)
@@ -77,6 +81,9 @@ public void Compile()
                     if (programAsset == null || programAsset.sourceCsScript == null)
                         continue;
 
+                    if (beforeProgramAssetCompile != null)
+                        beforeProgramAssetCompile(programAsset);
+
                     programAssetsAndPaths.Add((programAsset, AssetDatabase.GetAssetPath(programAsset.sourceCsScript)));
                 }
 
@@ -242,6 +249,15 @@ public void Compile()
                         UdonSharpEditorCache.Instance.ClearSourceHash(module.programAsset);
                     }
                 }
+
+                foreach (UdonSharpProgramAsset programAsset in allPrograms)
+                {
+                    if (programAsset == null || programAsset.sourceCsScript == null)
+                        continue;
+
+                    if (afterProgramAssetCompile != null)
+                        afterProgramAssetCompile(programAsset);
+                }
             }
             finally
             {