Attachment node component for auto generating attach nodes easier :3

cheese3660 · cheese3660 · commit f63b5b6bce12 · 2023-07-20T12:43:55.000-04:00
diff --git a/Assets/KSP2UnityTools/AttachmentNode.cs b/Assets/KSP2UnityTools/AttachmentNode.cs
@@ -0,0 +1,29 @@
+﻿using System;
+using KSP.Sim;
+using KSP.Sim.Definitions;
+using UnityEngine;
+using UnityEngine.Serialization;
+
+namespace KSP2UT.KSP2UnityTools
+{
+    public class AttachmentNode : MonoBehaviour
+    {
+        [Tooltip("Optional field that can be used to group nodes together, eg. 2 downward facing nodes grouped into a 'bottom' group. The group ID would be the same on both nodes. Empty means no group, which is default behavior.")]
+        public string nodeSymmetryGroupID;
+        public AttachNodeType nodeType;
+        public AttachNodeMethod attachMethod;
+        public bool isMultiJoint;
+        public int multiJointMaxJoint;
+        public float multiJointRadiusOffset;
+        public int size;
+        public float visualSize;
+        public bool isResourceCrossFeed;
+        public bool isRigid;
+        public float angularStrengthMultiplier;
+        public float contactArea;
+        public float overrideDragArea;
+        public bool isCompoundJoint;
+
+        
+    }
+}
diff --git a/Assets/KSP2UnityTools/AttachmentNode.cs.meta b/Assets/KSP2UnityTools/AttachmentNode.cs.meta
diff --git a/Assets/KSP2UnityTools/Editor/PartEditor.cs b/Assets/KSP2UnityTools/Editor/PartEditor.cs
@@ -1,20 +1,11 @@
-﻿
-using System;
-using System.Collections.Generic;
-using System.IO;
-using System.Linq;
+﻿using System.IO;
 using System.Reflection;
 using KSP;
 using UnityEditor;
-using Cheese.Extensions;
 using KSP.IO;
 using KSP.Modules;
 using KSP.Sim.Definitions;
-using KSP.Sim.impl;
-using Newtonsoft.Json;
-using Newtonsoft.Json.UnityConverters;
-using Newtonsoft.Json.UnityConverters.Configuration;
-using UnityEditor.VersionControl;
+using KSP2UT.KSP2UnityTools;
 using UnityEngine;
 
 [CustomEditor(typeof(CorePartData))]
@@ -51,6 +42,40 @@ private void OnSceneGUI()
     public override void OnInspectorGUI()
     {
         base.OnInspectorGUI();
+        GUILayout.Label("Attach Node Settings");
+        if (GUILayout.Button("Auto Generate AttachNodes"))
+        {
+            
+            TargetCore.data.attachNodes.Clear();
+            // Attach node naming scheme
+            foreach (var attachmentNode in TargetObject.GetComponentsInChildren<AttachmentNode>())
+            {
+                var obj = attachmentNode.gameObject;
+                var pos = TargetObject.transform.InverseTransformPoint(obj.transform.position);
+                var dir = Quaternion.Euler(TargetObject.transform.InverseTransformDirection(obj.transform.rotation.eulerAngles)) * Vector3.forward;
+                var newDefinition = new AttachNodeDefinition
+                {
+                    nodeID = obj.name,
+                    NodeSymmetryGroupID = attachmentNode.nodeSymmetryGroupID,
+                    nodeType = attachmentNode.nodeType,
+                    attachMethod = attachmentNode.attachMethod,
+                    IsMultiJoint = attachmentNode.isMultiJoint,
+                    MultiJointMaxJoint = attachmentNode.multiJointMaxJoint,
+                    MultiJointRadiusOffset = attachmentNode.multiJointRadiusOffset,
+                    position = pos,
+                    orientation = dir,
+                    size = attachmentNode.size,
+                    visualSize = attachmentNode.visualSize,
+                    angularStrengthMultiplier = attachmentNode.angularStrengthMultiplier,
+                    contactArea = attachmentNode.contactArea,
+                    overrideDragArea = attachmentNode.overrideDragArea,
+                    isCompoundJoint = attachmentNode.isCompoundJoint
+                };
+                TargetCore.data.attachNodes.Add(newDefinition);
+            }
+            EditorUtility.SetDirty(target);
+        }
+        
         GUILayout.Label("Gizmo Settings", EditorStyles.boldLabel);
         EditorGUI.BeginChangeCheck();
         _centerOfMassGizmos = EditorGUILayout.Toggle("CoM gizmos", _centerOfMassGizmos);
@@ -117,4 +142,14 @@ public static void DrawGizmoForPartCoreData(CorePartData data, GizmoType gizmoTy
             Gizmos.DrawSphere(pos,0.05f);
         }
     }
+
+    [DrawGizmo(GizmoType.Active | GizmoType.Selected)]
+    public static void DrawGizmoForAttachmentNode(AttachmentNode node, GizmoType gizmoType)
+    {
+        if (!_attachNodeGizmos) return;
+        Gizmos.color = new Color(Color.green.r, Color.green.g, Color.green.b, 0.5f);
+        var pos = node.transform.position;
+        Gizmos.DrawRay(pos, node.transform.rotation * Vector3.forward * 0.25f);
+        Gizmos.DrawSphere(pos,0.05f);
+    }
 }
diff --git a/Assets/KSP2UnityTools/game.asmdef b/Assets/KSP2UnityTools/game.asmdef
@@ -3,7 +3,8 @@
     "rootNamespace": "",
     "references": [],
     "includePlatforms": [
-        "Editor"
+        "Editor",
+        "WindowsStandalone64"
     ],
     "excludePlatforms": [],
     "allowUnsafeCode": false,
diff --git a/Assets/Scenes/SampleScene.unity b/Assets/Scenes/SampleScene.unity
@@ -349,6 +349,63 @@ Transform:
   m_Father: {fileID: 0}
   m_RootOrder: 1
   m_LocalEulerAnglesHint: {x: 50, y: -30, z: 0}
+--- !u!1 &759798002
+GameObject:
+  m_ObjectHideFlags: 0
+  m_CorrespondingSourceObject: {fileID: 0}
+  m_PrefabInstance: {fileID: 0}
+  m_PrefabAsset: {fileID: 0}
+  serializedVersion: 6
+  m_Component:
+  - component: {fileID: 759798003}
+  - component: {fileID: 759798004}
+  m_Layer: 0
+  m_Name: top
+  m_TagString: Untagged
+  m_Icon: {fileID: 0}
+  m_NavMeshLayer: 0
+  m_StaticEditorFlags: 0
+  m_IsActive: 1
+--- !u!4 &759798003
+Transform:
+  m_ObjectHideFlags: 0
+  m_CorrespondingSourceObject: {fileID: 0}
+  m_PrefabInstance: {fileID: 0}
+  m_PrefabAsset: {fileID: 0}
+  m_GameObject: {fileID: 759798002}
+  m_LocalRotation: {x: -0.7071068, y: 0, z: 0, w: 0.7071068}
+  m_LocalPosition: {x: 0, y: 0.212, z: 0}
+  m_LocalScale: {x: 1, y: 1, z: 1}
+  m_Children: []
+  m_Father: {fileID: 2045490028}
+  m_RootOrder: 0
+  m_LocalEulerAnglesHint: {x: -90, y: 0, z: 0}
+--- !u!114 &759798004
+MonoBehaviour:
+  m_ObjectHideFlags: 0
+  m_CorrespondingSourceObject: {fileID: 0}
+  m_PrefabInstance: {fileID: 0}
+  m_PrefabAsset: {fileID: 0}
+  m_GameObject: {fileID: 759798002}
+  m_Enabled: 1
+  m_EditorHideFlags: 0
+  m_Script: {fileID: 11500000, guid: bbdaa90d5d8f4373a3aea2f46737d90a, type: 3}
+  m_Name: 
+  m_EditorClassIdentifier: 
+  nodeSymmetryGroupID: 
+  nodeType: 0
+  attachMethod: 0
+  isMultiJoint: 0
+  multiJointMaxJoint: 0
+  multiJointRadiusOffset: 0
+  size: 0
+  visualSize: 0
+  isResourceCrossFeed: 1
+  isRigid: 0
+  angularStrengthMultiplier: 0
+  contactArea: 0
+  overrideDragArea: 0
+  isCompoundJoint: 0
 --- !u!1 &963194225
 GameObject:
   m_ObjectHideFlags: 0
@@ -531,6 +588,63 @@ Transform:
   m_Father: {fileID: 0}
   m_RootOrder: 2
   m_LocalEulerAnglesHint: {x: 0, y: 0, z: 0}
+--- !u!1 &1502141082
+GameObject:
+  m_ObjectHideFlags: 0
+  m_CorrespondingSourceObject: {fileID: 0}
+  m_PrefabInstance: {fileID: 0}
+  m_PrefabAsset: {fileID: 0}
+  serializedVersion: 6
+  m_Component:
+  - component: {fileID: 1502141083}
+  - component: {fileID: 1502141084}
+  m_Layer: 0
+  m_Name: bottom
+  m_TagString: Untagged
+  m_Icon: {fileID: 0}
+  m_NavMeshLayer: 0
+  m_StaticEditorFlags: 0
+  m_IsActive: 1
+--- !u!4 &1502141083
+Transform:
+  m_ObjectHideFlags: 0
+  m_CorrespondingSourceObject: {fileID: 0}
+  m_PrefabInstance: {fileID: 0}
+  m_PrefabAsset: {fileID: 0}
+  m_GameObject: {fileID: 1502141082}
+  m_LocalRotation: {x: 0.7071068, y: 0, z: 0, w: 0.7071068}
+  m_LocalPosition: {x: 0, y: -0.212, z: 0}
+  m_LocalScale: {x: 1, y: 1, z: 1}
+  m_Children: []
+  m_Father: {fileID: 2045490028}
+  m_RootOrder: 1
+  m_LocalEulerAnglesHint: {x: 90, y: 0, z: 0}
+--- !u!114 &1502141084
+MonoBehaviour:
+  m_ObjectHideFlags: 0
+  m_CorrespondingSourceObject: {fileID: 0}
+  m_PrefabInstance: {fileID: 0}
+  m_PrefabAsset: {fileID: 0}
+  m_GameObject: {fileID: 1502141082}
+  m_Enabled: 1
+  m_EditorHideFlags: 0
+  m_Script: {fileID: 11500000, guid: bbdaa90d5d8f4373a3aea2f46737d90a, type: 3}
+  m_Name: 
+  m_EditorClassIdentifier: 
+  nodeSymmetryGroupID: 
+  nodeType: 0
+  attachMethod: 0
+  isMultiJoint: 0
+  multiJointMaxJoint: 0
+  multiJointRadiusOffset: 0
+  size: 0
+  visualSize: 0
+  isResourceCrossFeed: 1
+  isRigid: 0
+  angularStrengthMultiplier: 0
+  contactArea: 0
+  overrideDragArea: 0
+  isCompoundJoint: 0
 --- !u!1 &1726311524
 GameObject:
   m_ObjectHideFlags: 0
@@ -659,7 +773,9 @@ Transform:
   m_LocalRotation: {x: 0, y: 0, z: 0, w: 1}
   m_LocalPosition: {x: 0, y: 0, z: 0}
   m_LocalScale: {x: 1, y: 1, z: 1}
-  m_Children: []
+  m_Children:
+  - {fileID: 759798003}
+  - {fileID: 1502141083}
   m_Father: {fileID: 0}
   m_RootOrder: 4
   m_LocalEulerAnglesHint: {x: 0, y: 0, z: 0}
@@ -1125,8 +1241,7 @@ MonoBehaviour:
   _engineModeChangeAnimationTriggerParamName: tChangeEngineMode
   _engineModeIndexAnimationParamName: iEngineMode
   FlameoutVFX: {fileID: 0}
-  ThrustTransforms:
-  - {fileID: 0}
+  ThrustTransforms: []
   ThrustTransformMultipliers: []
 --- !u!114 &2045490033
 MonoBehaviour:
@@ -1230,13 +1345,13 @@ MonoBehaviour:
           z: 0
         orientation:
           x: 0
-          y: 1
-          z: 0
+          y: 0.9999999403953552
+          z: 0.00000005960464477539063
         size: 0
         visualSize: 0
-        isResourceCrossfeed: 1
+        isResourceCrossfeed: 0
         isRigid: 0
-        angularStrengthMultiplier: 1
+        angularStrengthMultiplier: 0
         contactArea: 0
         overrideDragArea: 0
         isCompoundJoint: 0
@@ -1249,40 +1364,17 @@ MonoBehaviour:
         MultiJointRadiusOffset: 0
         position:
           x: 0
-          y: -0.18199999630451202
-          z: 0
-        orientation:
-          x: 0
-          y: -1
-          z: 0
-        size: 0
-        visualSize: 0
-        isResourceCrossfeed: 1
-        isRigid: 0
-        angularStrengthMultiplier: 1
-        contactArea: 0
-        overrideDragArea: 0
-        isCompoundJoint: 0
-      - nodeID: srfAttach
-        NodeSymmetryGroupID: 
-        nodeType: 1
-        attachMethod: 1
-        IsMultiJoint: 0
-        MultiJointMaxJoint: 0
-        MultiJointRadiusOffset: 0
-        position:
-          x: 0
-          y: 0.211999997496605
+          y: -0.21199999749660492
           z: 0
         orientation:
           x: 0
-          y: 1
-          z: 0
+          y: -0.9999999403953552
+          z: 0.00000005960464477539063
         size: 0
         visualSize: 0
-        isResourceCrossfeed: 1
+        isResourceCrossfeed: 0
         isRigid: 0
-        angularStrengthMultiplier: 1
+        angularStrengthMultiplier: 0
         contactArea: 0
         overrideDragArea: 0
         isCompoundJoint: 0
