ProgressControlsForDialogs.cs 4.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161
  1. using UnityEditor;
  2. using UnityEngine.UIElements;
  3. using PlasticGui;
  4. namespace Unity.PlasticSCM.Editor.UI.UIElements
  5. {
  6. class ProgressControlsForDialogs :
  7. VisualElement,
  8. IProgressControls
  9. {
  10. internal class Data
  11. {
  12. internal bool IsWaitingAsyncResult;
  13. internal float ProgressPercent;
  14. internal string ProgressMessage;
  15. internal MessageType StatusType;
  16. internal string StatusMessage;
  17. internal void CopyInto(Data other)
  18. {
  19. other.IsWaitingAsyncResult = IsWaitingAsyncResult;
  20. other.ProgressPercent = ProgressPercent;
  21. other.ProgressMessage = ProgressMessage;
  22. other.StatusType = StatusType;
  23. other.StatusMessage = StatusMessage;
  24. }
  25. }
  26. internal Data ProgressData { get { return mData; } }
  27. internal void ForcedUpdateProgress()
  28. {
  29. if (mData.IsWaitingAsyncResult)
  30. {
  31. mUndefinedProgress.RemoveFromClassList("display-none");
  32. mPercentageLabel.RemoveFromClassList("display-none");
  33. mLoadingSpinner.Start();
  34. EditorApplication.update += UpdatePercent;
  35. }
  36. else
  37. {
  38. mUndefinedProgress.AddToClassList("display-none");
  39. mPercentageLabel.AddToClassList("display-none");
  40. mLoadingSpinner.Stop();
  41. EditorApplication.update -= UpdatePercent;
  42. }
  43. mStatusLabel.text = mData.StatusMessage;
  44. mProgressLabel.text = mData.ProgressMessage;
  45. }
  46. internal void UpdatePercent()
  47. {
  48. if (mData.ProgressPercent >= 0)
  49. mPercentageLabel.text = string.Format("({0}%)", (int)(mData.ProgressPercent * 100));
  50. else
  51. mPercentageLabel.text = "";
  52. }
  53. internal ProgressControlsForDialogs(
  54. VisualElement[] actionControls)
  55. {
  56. mActionControls = actionControls;
  57. InitializeLayoutAndStyles();
  58. BuildComponents();
  59. }
  60. internal void EnableActionControls(bool enable)
  61. {
  62. if (mActionControls != null)
  63. foreach (var control in mActionControls)
  64. if (control != null)
  65. control.SetEnabled(enable);
  66. }
  67. void IProgressControls.HideProgress()
  68. {
  69. EnableActionControls(true);
  70. mData.IsWaitingAsyncResult = false;
  71. mData.ProgressMessage = string.Empty;
  72. ForcedUpdateProgress();
  73. }
  74. void IProgressControls.ShowProgress(string message)
  75. {
  76. EnableActionControls(false);
  77. CleanStatusMessage(mData);
  78. mData.IsWaitingAsyncResult = true;
  79. mData.ProgressPercent = -1f;
  80. mData.ProgressMessage = message;
  81. ForcedUpdateProgress();
  82. }
  83. void IProgressControls.ShowError(string message)
  84. {
  85. mData.StatusMessage = message;
  86. mData.StatusType = MessageType.Error;
  87. ForcedUpdateProgress();
  88. }
  89. void IProgressControls.ShowNotification(string message)
  90. {
  91. mData.StatusMessage = message;
  92. mData.StatusType = MessageType.Info;
  93. ForcedUpdateProgress();
  94. }
  95. void IProgressControls.ShowSuccess(string message)
  96. {
  97. mData.StatusMessage = message;
  98. mData.StatusType = MessageType.Info;
  99. ForcedUpdateProgress();
  100. }
  101. void IProgressControls.ShowWarning(string message)
  102. {
  103. mData.StatusMessage = message;
  104. mData.StatusType = MessageType.Warning;
  105. ForcedUpdateProgress();
  106. }
  107. void BuildComponents()
  108. {
  109. mUndefinedProgress = this.Q<VisualElement>("UndefinedProgress");
  110. mProgressLabel = this.Q<Label>("Progress");
  111. mStatusLabel = this.Q<Label>("Status");
  112. mPercentageLabel = this.Q<Label>("Percentage");
  113. mLoadingSpinner = new LoadingSpinner();
  114. mUndefinedProgress.Add(mLoadingSpinner);
  115. }
  116. void InitializeLayoutAndStyles()
  117. {
  118. this.LoadLayout(typeof(ProgressControlsForDialogs).Name);
  119. this.LoadStyle(typeof(ProgressControlsForDialogs).Name);
  120. }
  121. static void CleanStatusMessage(Data data)
  122. {
  123. data.StatusMessage = string.Empty;
  124. data.StatusType = MessageType.None;
  125. }
  126. Data mData = new Data();
  127. VisualElement mUndefinedProgress;
  128. Label mProgressLabel;
  129. Label mStatusLabel;
  130. Label mPercentageLabel;
  131. VisualElement[] mActionControls;
  132. LoadingSpinner mLoadingSpinner;
  133. }
  134. }