added dropdown UIEditor to UIBinaries

mirik123 · Jul 23, 2016 · bd930b3 · bd930b3
1 parent 7343dd6
commit bd930b3
Show file tree

Hide file tree

Showing 3 changed files with 42 additions and 1 deletion.
diff --git a/QCStudioPlugin/Forms/OptionPageGrid.cs b/QCStudioPlugin/Forms/OptionPageGrid.cs
@@ -69,6 +69,7 @@ public string UIBinaries {
         [Category("UI")]
         [DefaultValue("QCTerminalControl.JSChartControl")]
         [DisplayName("Full class name for UI library")]
+        [Editor("QuantConnect.QCStudioPlugin.ClassListEditor", typeof(UITypeEditor))]
         public string UIClassName {
             get {
                 return _UIClassName;

diff --git a/QCStudioPlugin/Forms/VistaFolderBrowser.cs b/QCStudioPlugin/Forms/VistaFolderBrowser.cs
@@ -1,4 +1,5 @@
-using System;
+using Microsoft.VisualStudio.Shell;
+using System;
 using System.Collections.Generic;
 using System.ComponentModel;
 using System.Drawing.Design;
@@ -7,6 +8,7 @@
 using System.Text;
 using System.Threading.Tasks;
 using System.Windows.Forms;
+using System.Windows.Forms.Design;
 
 //http:// stackoverflow.com/questions/15368771/show-detailed-folder-browser-from-a-propertygrid
 //http:// stackoverflow.com/users/403671/simon-mourier
@@ -34,6 +36,33 @@ public override object EditValue(ITypeDescriptorContext context, IServiceProvide
         }
     }
 
+    public class ClassListEditor : UITypeEditor
+    {
+        public override UITypeEditorEditStyle GetEditStyle(ITypeDescriptorContext context)
+        {
+            return UITypeEditorEditStyle.DropDown;
+        }
+
+        public override object EditValue(ITypeDescriptorContext context, IServiceProvider provider, object value)
+        {
+            IWindowsFormsEditorService wfes = provider.GetService(typeof(IWindowsFormsEditorService)) as IWindowsFormsEditorService;
+            if (wfes != null)
+            {
+                var dte = (EnvDTE80.DTE2)ServiceProvider.GlobalProvider.GetService(typeof(EnvDTE.DTE));
+                string UIBinaries = (string)dte.Properties["QuantConnect Client", "General"].Item("UIBinaries").Value;
+                var items = QuantConnect.QCStudioPlugin.QCPluginUtilities.GetClassesList(UIBinaries, "QCInterfaces.ChartControl");
+
+                var lbc = new ListBox();
+                lbc.Items.AddRange(items);
+                lbc.SelectedIndexChanged += (sender, e) => wfes.CloseDropDown();
+
+                wfes.DropDownControl(lbc);
+                value = lbc.SelectedItem as string ?? value as string ?? string.Empty;
+            }
+            return value;
+        }
+    }
+
     public class VistaFolderBrowser
     {
         public string DirectoryPath { get; set; }

diff --git a/QCStudioPlugin/Utilities/QCPluginUtilities.cs b/QCStudioPlugin/Utilities/QCPluginUtilities.cs
@@ -125,6 +125,17 @@ public static string[] GetAlgorithmsList(string filePath, string classDll)
             return algorithms;
         }
 
+        public static string[] GetClassesList(string classDll, string parentClass)
+        {
+            var algoass = Assembly.LoadFrom(classDll);
+            var algorithms = algoass.GetTypes()
+                .Where(x => x.BaseType.FullName == parentClass)
+                .Select(x => x.FullName)
+                .ToArray();
+
+            return algorithms;
+        }
+
         private static string RetrieveAssemblyDirectory()
         {
             string codeBase = Assembly.GetExecutingAssembly().CodeBase;