Fix issue visualization of ModifyBones

Author: pafuhana1213

Plugins/KawaiiPhysics/Source/KawaiiPhysicsEd/Private/AnimGraphNode_KawaiiPhysics.cpp

@@ -65,58 +65,6 @@ FEditorModeID UAnimGraphNode_KawaiiPhysics::GetEditorMode() const
 	return "AnimGraph.SkeletalControl.KawaiiPhysics";
 }
 
-void UAnimGraphNode_KawaiiPhysics::Draw(FPrimitiveDrawInterface* PDI, USkeletalMeshComponent* SkelMeshComp) const
-{
-	if (!SkelMeshComp)
-	{
-		return;
-	}
-
-	FAnimNode_KawaiiPhysics* ActiveNode = GetActiveInstanceNode<FAnimNode_KawaiiPhysics>(SkelMeshComp->GetAnimInstance());
-	if (!ActiveNode)
-	{
-		return;
-	}
-
-	if (bEnableDebugDrawBone)
-	{
-		for (auto& Bone : ActiveNode->ModifyBones)
-		{
-			PDI->DrawPoint(Bone.Location, FLinearColor::White, 5.0f, SDPG_Foreground);
-
-			if (Bone.PhysicsSettings.Radius > 0)
-			{
-				auto Color = Bone.bDummy ? FColor::Red : FColor::Yellow;
-				DrawWireSphere(PDI, Bone.Location, Color, Bone.PhysicsSettings.Radius, 16, SDPG_Foreground);
-			}
-
-			for (const int32 ChildIndex : Bone.ChildIndexs)
-			{
-				DrawDashedLine(PDI, Bone.Location, ActiveNode->ModifyBones[ChildIndex].Location,
-				               FLinearColor::White, 1, SDPG_Foreground);
-			}
-		}
-	}
-
-	if (bEnableDebugDrawAngleLimit)
-	{
-		for (auto& Bone : ActiveNode->ModifyBones)
-		{
-			if (Bone.ParentIndex < 0 || Bone.PhysicsSettings.LimitAngle <= 0)
-			{
-				continue;
-			}
-
-			auto& ParentBone = ActiveNode->ModifyBones[Bone.ParentIndex];
-			FTransform ParentBoneTransform =
-				FTransform(FQuat::FindBetween(FVector::ForwardVector, Bone.PoseLocation - ParentBone.PoseLocation), ParentBone.Location);
-			TArray<FVector> Verts;
-			DrawWireCone(PDI, Verts, ParentBoneTransform, (Bone.PoseLocation - ParentBone.PoseLocation).Size(),
-			             Bone.PhysicsSettings.LimitAngle, 16, FColor::Green, SDPG_World);
-		}
-	}
-}
-
 void UAnimGraphNode_KawaiiPhysics::ValidateAnimNodePostCompile(FCompilerResultsLog& MessageLog, UAnimBlueprintGeneratedClass* CompiledClass, int32 CompiledNodeIndex)
 {
 	UAnimGraphNode_SkeletalControlBase::ValidateAnimNodePostCompile(MessageLog, CompiledClass, CompiledNodeIndex);

Plugins/KawaiiPhysics/Source/KawaiiPhysicsEd/Private/KawaiiPhysicsEditMode.cpp

@@ -91,6 +91,7 @@ void FKawaiiPhysicsEditMode::Render(const FSceneView* View, FViewport* Viewport,
 	if (SkelMeshComp && SkelMeshComp->SkeletalMesh && SkelMeshComp->SkeletalMesh->GetSkeleton())
 #endif
 	{
+		RenderModifyBones(PDI);
 		RenderSphericalLimits(PDI);
 		RenderCapsuleLimit(PDI);
 		RenderPlanerLimit(PDI);
@@ -121,6 +122,29 @@ void FKawaiiPhysicsEditMode::Render(const FSceneView* View, FViewport* Viewport,
 
 }
 
+void FKawaiiPhysicsEditMode::RenderModifyBones(FPrimitiveDrawInterface* PDI) const
+{
+	if (GraphNode->bEnableDebugDrawBone)
+	{
+		for (auto& Bone : RuntimeNode->ModifyBones)
+		{
+			PDI->DrawPoint(Bone.Location, FLinearColor::White, 5.0f, SDPG_Foreground);
+
+			if (Bone.PhysicsSettings.Radius > 0)
+			{
+				auto Color = Bone.bDummy ? FColor::Red : FColor::Yellow;
+				DrawWireSphere(PDI, Bone.Location, Color, Bone.PhysicsSettings.Radius, 16, SDPG_Foreground);
+			}
+
+			for (const int32 ChildIndex : Bone.ChildIndexs)
+			{
+				DrawDashedLine(PDI, Bone.Location, RuntimeNode->ModifyBones[ChildIndex].Location,
+						   FLinearColor::White, 1, SDPG_Foreground);
+			}
+		}
+	}
+}
+
 void FKawaiiPhysicsEditMode::RenderSphericalLimits(FPrimitiveDrawInterface* PDI) const
 {
 	if (GraphNode->bEnableDebugDrawSphereLimit)

Plugins/KawaiiPhysics/Source/KawaiiPhysicsEd/Public/AnimGraphNode_KawaiiPhysics.h

@@ -35,7 +35,6 @@ class UAnimGraphNode_KawaiiPhysics : public UAnimGraphNode_SkeletalControlBase
 	//virtual FText GetControllerDescription() const override;
 	virtual FText GetControllerDescription() const override;
 	virtual const FAnimNode_SkeletalControlBase* GetNode() const override { return &Node; }
-	virtual void Draw(FPrimitiveDrawInterface* PDI, USkeletalMeshComponent* SkelMeshComp) const override;
 	// End of UAnimGraphNode_SkeletalControlBase interface
 
 	// UObject interface

Plugins/KawaiiPhysics/Source/KawaiiPhysicsEd/Public/KawaiiPhysicsEditMode.h

@@ -46,6 +46,9 @@ class FKawaiiPhysicsEditMode : public FAnimNodeEditMode
 
 private:
 
+	
+	void RenderModifyBones(FPrimitiveDrawInterface* PDI) const;
+
 	/** Render each collisions */
 	void RenderSphericalLimits(FPrimitiveDrawInterface* PDI) const;
 	void RenderCapsuleLimit(FPrimitiveDrawInterface* PDI) const;