2 files changed, 22 insertions, 22 deletions
diff --git a/Software/Visual_Studio/MachineStudio/Tango.MachineStudio.Common/ExtensionMethods/IStudioMessageExtensions.cs b/Software/Visual_Studio/MachineStudio/Tango.MachineStudio.Common/ExtensionMethods/IStudioMessageExtensions.cs
deleted file mode 100644
index 7ae259e04..000000000
--- a/Software/Visual_Studio/MachineStudio/Tango.MachineStudio.Common/ExtensionMethods/IStudioMessageExtensions.cs
+++ /dev/null
@@ -1,22 +0,0 @@
-using GalaSoft.MvvmLight.Messaging;
-using System;
-using System.Collections.Generic;
-using System.Linq;
-using System.Text;
-using System.Threading.Tasks;
-using Tango.MachineStudio.Common.Messages;
-
-/// <summary>
-/// Contains <see cref="IStudioMessage"/> extension methods.
-/// </summary>
-public static class IStudioMessageExtensions
-{
-    /// <summary>
-    /// Sends the message.
-    /// </summary>
-    /// <param name="message">The message.</param>
-    public static void Send(this IStudioMessage message)
-    {
-        Messenger.Default.Send(message);
-    }
-}
diff --git a/Software/Visual_Studio/MachineStudio/Tango.MachineStudio.Common/ExtensionMethods/TangoIOCExtensions.cs b/Software/Visual_Studio/MachineStudio/Tango.MachineStudio.Common/ExtensionMethods/TangoIOCExtensions.cs
new file mode 100644
index 000000000..827650f59
--- /dev/null
+++ b/Software/Visual_Studio/MachineStudio/Tango.MachineStudio.Common/ExtensionMethods/TangoIOCExtensions.cs
@@ -0,0 +1,22 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+using Tango.Core.DI;
+using Tango.MachineStudio.Common;
+
+public static class TangoIOCExtensions
+{
+    /// <summary>
+    /// Gets the specified module <see cref="IStudioViewModel"/> view models.
+    /// </summary>
+    /// <param name="tangoIOC">The Tango IOC.</param>
+    /// <param name="module">The module.</param>
+    /// <returns></returns>
+    public static IEnumerable<IStudioViewModel> GetModuleViewModels(this TangoIOC tangoIOC, IStudioModule module)
+    {
+        var viewModels = tangoIOC.GetAllInstancesByBase<IStudioViewModel>().Where(x => x != null && x.GetType().BaseType.GetGenericArguments().Length > 0 && x.GetType().BaseType.GetGenericArguments()[0] == module.GetType());
+        return viewModels;
+    }
+}