No Description
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

MemberInvocationInspector.cs 13KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419
  1. /*
  2. using System;
  3. using System.Collections.Generic;
  4. using System.Collections.ObjectModel;
  5. using System.Linq;
  6. using Unity.VisualScripting;
  7. using UnityEngine;
  8. using UnityObject = UnityEngine.Object;
  9. namespace Unity.VisualScripting
  10. {
  11. [Inspector(typeof(MemberInvocation))]
  12. public sealed class MemberInvocationInspector : InvocationInspector
  13. {
  14. public MemberInvocationInspector(Metadata metadata) : base(metadata) { }
  15. public override void Initialize()
  16. {
  17. base.Initialize();
  18. memberInspector = memberMetadata.Inspector<MemberManipulatorInspector>();
  19. memberInspector.direction = (ActionDirection)directionMetadata.value;
  20. referenceInspectors = new Dictionary<Metadata, Inspector>();
  21. memberFilter = metadata.GetAttribute<MemberFilter>() ?? MemberFilter.Any;
  22. memberTypeFilter = metadata.GetAttribute<TypeFilter>();
  23. memberMetadata.valueChanged += previousMember => ReflectMember(false);
  24. }
  25. private MemberManipulatorInspector memberInspector;
  26. private Dictionary<Metadata, Inspector> referenceInspectors;
  27. private IFuzzyOptionTree GetMemberOptions()
  28. {
  29. return new MemberOptionTree(typeSet, memberFilter, memberTypeFilter, (ActionDirection)directionMetadata.value);
  30. }
  31. #region Metadata
  32. private Metadata memberMetadata => metadata[nameof(MemberInvocation.member)];
  33. private Metadata directionMetadata => metadata[nameof(MemberInvocation.direction)];
  34. private Metadata targetMetadata => metadata[nameof(MemberInvocation.target)];
  35. #endregion
  36. #region Settings
  37. private ReadOnlyCollection<Type> _typeSet;
  38. private MemberFilter _memberFilter;
  39. private TypeFilter _memberTypeFilter;
  40. public override Type expectedType
  41. {
  42. get
  43. {
  44. return base.expectedType;
  45. }
  46. set
  47. {
  48. base.expectedType = value;
  49. memberTypeFilter = value != null ? new TypeFilter(TypesMatching.Any, value) : null;
  50. }
  51. }
  52. public ReadOnlyCollection<Type> typeSet
  53. {
  54. get
  55. {
  56. return _typeSet;
  57. }
  58. set
  59. {
  60. _typeSet = value;
  61. memberInspector.typeSet = value;
  62. }
  63. }
  64. public MemberFilter memberFilter
  65. {
  66. get
  67. {
  68. return _memberFilter;
  69. }
  70. set
  71. {
  72. _memberFilter = value;
  73. memberInspector.memberFilter = value;
  74. }
  75. }
  76. public TypeFilter memberTypeFilter
  77. {
  78. get
  79. {
  80. return _memberTypeFilter;
  81. }
  82. set
  83. {
  84. _memberTypeFilter = value;
  85. memberInspector.memberTypeFilter = value;
  86. }
  87. }
  88. #endregion
  89. #region Reflection
  90. private bool reflectionSucceeded;
  91. private void ReflectMember(bool reset)
  92. {
  93. memberMetadata.DisposeChildren();
  94. var member = (Member)memberMetadata.value;
  95. // Clear the metadata tree and optionally the arguments
  96. if (reset)
  97. {
  98. targetMetadata.value = null;
  99. argumentsMetadata.Clear();
  100. }
  101. referenceInspectors.Clear();
  102. // Attempt to reflect the member
  103. reflectionSucceeded = false;
  104. if (member != null)
  105. {
  106. try
  107. {
  108. member.EnsureReflected();
  109. reflectionSucceeded = true;
  110. }
  111. catch { }
  112. }
  113. if (!reflectionSucceeded)
  114. {
  115. return;
  116. }
  117. // Create the metadata tree and optionally assign default arguments
  118. if (reset)
  119. {
  120. IExpression target;
  121. if (member.requiresTarget)
  122. {
  123. if (ComponentHolderProtocol.IsComponentHolderType(member.targetType) && metadata.UnityObjectAncestor() != null)
  124. {
  125. target = new Self();
  126. }
  127. else
  128. {
  129. target = null;
  130. }
  131. }
  132. else
  133. {
  134. target = null;
  135. }
  136. targetMetadata.value = target;
  137. }
  138. targetMetadata.Inspector<IExpressionInspector>().expectedType = member.targetType;
  139. if (member.isInvocable)
  140. {
  141. var argumentIndex = 0;
  142. foreach (var parameterInfo in member.methodBase.GetParametersWithoutThis())
  143. {
  144. if (reset || argumentIndex >= argumentsMetadata.Count)
  145. {
  146. argumentsMetadata.Add(parameterInfo.DefaultInspectableExpression());
  147. }
  148. var argumentMetadata = argumentsMetadata[argumentIndex];
  149. PrepareParameterLabel(argumentMetadata, parameterInfo.HumanName(), member.methodBase.ParameterSummary(parameterInfo));
  150. if (parameterInfo.ParameterType.IsByRef)
  151. {
  152. var argument = argumentsMetadata[argumentIndex].value as Literal;
  153. if (argument == null || argument.type != typeof(IVariableReference))
  154. {
  155. argumentsMetadata[argumentIndex].value = new Literal(typeof(IVariableReference), null);
  156. }
  157. var referenceInspector = new IVariableReferenceInspector(argumentMetadata[nameof(ILiteral.value)]);
  158. referenceInspector.Initialize();
  159. referenceInspector.direction = ActionDirection.Set;
  160. referenceInspectors.Add(argumentMetadata, referenceInspector);
  161. PrepareParameterLabel(argumentMetadata[nameof(ILiteral.value)], parameterInfo.HumanName(), member.methodBase.ParameterSummary(parameterInfo));
  162. }
  163. else
  164. {
  165. PrepareParameterInspector(argumentMetadata, parameterInfo.ParameterType);
  166. }
  167. argumentIndex++;
  168. }
  169. }
  170. else if (member.isSettable && (ActionDirection)directionMetadata.value != ActionDirection.Get)
  171. {
  172. if (reset || argumentsMetadata.Count == 0)
  173. {
  174. argumentsMetadata.Add(member.type.ToInspectableExpression());
  175. }
  176. var argumentMetadata = argumentsMetadata[0];
  177. PrepareParameterLabel(argumentMetadata, member.info.HumanName(), member.info.Summary());
  178. PrepareParameterInspector(argumentMetadata, member.type);
  179. }
  180. SetHeightDirty();
  181. }
  182. public static bool WillFail(bool reflectionSucceeded, MemberInvocation invocation, UnityObject owner)
  183. {
  184. if (!reflectionSucceeded)
  185. {
  186. return true;
  187. }
  188. // We can only analyze if the member isn't null and
  189. // if we can infer the value of the target expression in edit mode
  190. var canAnalyze = invocation.member != null && (invocation.target is ILiteral || invocation.target is Self);
  191. if (!canAnalyze)
  192. {
  193. return false;
  194. }
  195. object target;
  196. if (invocation.target is ILiteral)
  197. {
  198. target = ((ILiteral)invocation.target).value;
  199. }
  200. else if (invocation.target is Self)
  201. {
  202. target = owner;
  203. }
  204. else
  205. {
  206. throw new NotSupportedException();
  207. }
  208. if (target == null)
  209. {
  210. return true;
  211. }
  212. var targetType = target.GetType();
  213. if (ComponentHolderProtocol.IsComponentHolderType(invocation.member.targetType))
  214. {
  215. if (!ComponentHolderProtocol.IsComponentHolderType(targetType))
  216. {
  217. return true;
  218. }
  219. // Don't fail true if the owner isn't a component holder
  220. // (e.g. it might be a graph asset)
  221. if (!owner.IsComponentHolder())
  222. {
  223. return false;
  224. }
  225. return invocation.member.targetType != typeof(GameObject) && !((UnityObject)target).GetComponents<Component>().Any(c => invocation.member.targetType.IsInstanceOfType(c));
  226. }
  227. else
  228. {
  229. return !invocation.member.targetType.IsAssignableFrom(targetType);
  230. }
  231. }
  232. private bool willFail => WillFail(reflectionSucceeded, (MemberInvocation)metadata.value, metadata.UnityObjectAncestor());
  233. #endregion
  234. #region Rendering
  235. protected override IEnumerable<GUIContent> compactLabels => base.compactLabels.Concat(targetMetadata.label.Yield());
  236. protected override float GetHeight(float width, GUIContent label)
  237. {
  238. var height = 0f;
  239. height += GetMemberHeight(width);
  240. using (LudiqGUIUtility.labelWidth.Override(GetCompactLabelsWidth(width)))
  241. {
  242. if (reflectionSucceeded)
  243. {
  244. height += Styles.spaceBetweenParameters;
  245. if (((Member)memberMetadata.value).requiresTarget)
  246. {
  247. height += GetTargetHeight(width);
  248. if (parameters.Count > 0)
  249. {
  250. height += Styles.spaceBetweenParameters;
  251. }
  252. }
  253. height += GetParametersHeight(width);
  254. }
  255. }
  256. height = HeightWithLabel(metadata, width, height, label);
  257. return height;
  258. }
  259. private float GetMemberHeight(float width)
  260. {
  261. return InspectorGUI.GetHeight(memberMetadata, width, GUIContent.none, this);
  262. }
  263. private float GetTargetHeight(float width)
  264. {
  265. return InspectorGUI.GetHeight(targetMetadata, width, targetMetadata.label, this);
  266. }
  267. protected override float GetParameterHeight(Metadata parameter, float width)
  268. {
  269. if (referenceInspectors.ContainsKey(parameter))
  270. {
  271. return referenceInspectors[parameter].GetHeight(width, GUIContent.none, this);
  272. }
  273. else
  274. {
  275. return InspectorGUI.GetHeight(parameter, width, GUIContent.none, this);
  276. }
  277. }
  278. protected override void OnGUI(Rect position, GUIContent label)
  279. {
  280. var memberPosition = position.VerticalSection(ref y, GetMemberHeight(position.width));
  281. memberPosition = PrefixLabel(metadata, memberPosition, label);
  282. OnMemberGUI(memberPosition);
  283. position = ReclaimImplementationSelector(position);
  284. if (reflectionSucceeded)
  285. {
  286. using (LudiqGUIUtility.labelWidth.Override(GetCompactLabelsWidth(position.width)))
  287. {
  288. y += Styles.spaceBetweenParameters;
  289. if (((Member)memberMetadata.value).requiresTarget)
  290. {
  291. var targetPosition = position.VerticalSection(ref y, GetTargetHeight(position.width));
  292. OnTargetGUI(targetPosition);
  293. if (parameters.Count > 0)
  294. {
  295. y += Styles.spaceBetweenParameters;
  296. }
  297. }
  298. OnParametersGUI(position);
  299. }
  300. }
  301. }
  302. private void OnMemberGUI(Rect memberPosition)
  303. {
  304. BeginBlock(memberMetadata, memberPosition, GUIContent.none);
  305. memberMetadata.Inspector<MemberManipulatorInspector>().willFail = willFail;
  306. InspectorGUI.Field(memberMetadata, memberPosition, GUIContent.none);
  307. if (EndBlock(memberMetadata))
  308. {
  309. ReflectMember(true);
  310. }
  311. }
  312. private void OnTargetGUI(Rect targetPosition)
  313. {
  314. InspectorGUI.Field(targetMetadata, targetPosition);
  315. }
  316. protected override void OnParameterGUI(Rect parameterPosition, Metadata parameter)
  317. {
  318. if (referenceInspectors.ContainsKey(parameter))
  319. {
  320. referenceInspectors[parameter].Field(parameterPosition);
  321. }
  322. else
  323. {
  324. base.OnParameterGUI(parameterPosition, parameter);
  325. }
  326. }
  327. #endregion
  328. }
  329. }
  330. */