Merge pull request #19 from BadiBasso/prt

PRT patch
boformer · Jul 23, 2018 · b9eb2ef · b9eb2ef
2 parents 9eba6a7 + 1f634ce
commit b9eb2ef
Showing 1 changed file with 18 additions and 1 deletion.
diff --git a/NetworkSkins/Net/NetManagerDetour.cs b/NetworkSkins/Net/NetManagerDetour.cs
@@ -71,12 +71,29 @@ public static void Revert()
 
             var caller = new System.Diagnostics.StackFrame(1).GetMethod().Name;
             //Debug.Log("CreateSegment (" + info.name + ") called by " + caller);
+            var caller2 = new System.Diagnostics.StackFrame(2).GetMethod().Name;
 
             switch (caller)
             {
+                case "CreateSegmentOriginal":
+                    if (caller2 == "CreateSegment")
+                    {
+                        var caller3Type = new System.Diagnostics.StackFrame(3).GetMethod().DeclaringType?.Name;
+                        if (caller3Type != null && caller3Type.StartsWith("NetTool", StringComparison.Ordinal))
+                        {
+                            if (success) EventSegmentCreate?.Invoke(segment);
+                            if (MoveMiddleNode_releasedSegment > 0) EventSegmentRelease?.Invoke(MoveMiddleNode_releasedSegment);
+                            if (SplitSegment_releasedSegment > 0) EventSegmentRelease?.Invoke(SplitSegment_releasedSegment);
+
+                            SplitSegment_releasedSegment = 0;
+                            MoveMiddleNode_releasedSegment = 0;
+                        }
+                    }
+                    break;
+
                 case "CreateNode":
 
-                    var caller2 = new System.Diagnostics.StackFrame(2).GetMethod().Name;
+                    //var caller2 = new System.Diagnostics.StackFrame(2).GetMethod().Name;
                     //Debug.Log("... called by " + caller2);
 
                     if (caller2 == "CreateNode") // check that caller was called by NetTool