#region License Information /* HeuristicLab * Copyright (C) Heuristic and Evolutionary Algorithms Laboratory (HEAL) * * This file is part of HeuristicLab. * * HeuristicLab is free software: you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation, either version 3 of the License, or * (at your option) any later version. * * HeuristicLab is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with HeuristicLab. If not, see . */ #endregion using System; using System.CodeDom; using System.CodeDom.Compiler; using System.Collections.Generic; using System.IO; using System.Linq; using System.Reflection; using System.Text; using System.Text.RegularExpressions; using HeuristicLab.Collections; using HeuristicLab.Common; using HeuristicLab.Common.Resources; using HeuristicLab.Core; using HeuristicLab.Persistence.Auxiliary; using HEAL.Attic; using HeuristicLab.PluginInfrastructure; using Microsoft.CSharp; namespace HeuristicLab.Operators.Programmable { [Item("ProgrammableOperator", "An operator that can be programmed for arbitrary needs.")] [StorableType("741279E4-4C38-4D66-B9F1-03E0F8B47A78")] public class ProgrammableOperator : Operator, IParameterizedNamedItem, IStorableContent, IProgrammableItem { #region Fields & Properties public string Filename { get; set; } public new ParameterCollection Parameters { get { return base.Parameters; } } public static new System.Drawing.Image StaticItemImage { get { return VSImageLibrary.Script; } } private MethodInfo executeMethod; public CompilerErrorCollection CompileErrors { get; private set; } public string CompilationUnitCode { get; private set; } [Storable] private string code; public string Code { get { return code; } set { if (value != code) { code = value; executeMethod = null; OnCodeChanged(); } } } private object syncRoot = new object(); private static object initLock = new object(); private static Dictionary> defaultPluginDict; private static Dictionary defaultAssemblyDict; public readonly Dictionary> Plugins; protected Dictionary Assemblies; [Storable(Name = "SelectedAssemblies")] private List _selectedAssemblyNames_persistence { get { return Assemblies.Where(a => a.Value).Select(a => a.Key.FullName).ToList(); } set { var selectedAssemblyNames = new HashSet(value.Select(n => new AssemblyName(n).Name)); foreach (var a in Assemblies.Keys.ToList()) { Assemblies[a] = selectedAssemblyNames.Contains(new AssemblyName(a.FullName).Name); } } } public IEnumerable AvailableAssemblies { get { return Assemblies.Keys; } } public IEnumerable SelectedAssemblies { get { return Assemblies.Where(kvp => kvp.Value).Select(kvp => kvp.Key); } } [Storable] private HashSet namespaces; public IEnumerable Namespaces { get { return namespaces; } } public override bool CanChangeDescription { get { return true; } } #endregion #region Extended Accessors public void SelectAssembly(Assembly a) { if (a != null && Assemblies.ContainsKey(a) && !Assemblies[a]) { Assemblies[a] = true; } } public void UnselectAssembly(Assembly a) { if (a != null && Assemblies.ContainsKey(a) && Assemblies[a]) { Assemblies[a] = false; } } public void SelectNamespace(string ns) { namespaces.Add(ns); } public void UnselectNamespace(string ns) { namespaces.Remove(ns); } public IEnumerable GetAllNamespaces(bool selectedAssembliesOnly) { var namespaces = new HashSet(); foreach (var a in Assemblies) { if (!selectedAssembliesOnly || a.Value) { foreach (var t in a.Key.GetTypes()) { if (t.IsPublic) { foreach (string ns in GetNamespaceHierachy(t.Namespace)) { namespaces.Add(ns); } } } } } return namespaces; } private IEnumerable GetNamespaceHierachy(string ns) { if (ns != null) { for (int i = ns.Length; i != -1; i = ns.LastIndexOf('.', i - 1)) { yield return ns.Substring(0, i); } } } #endregion #region Construction & Initialization [StorableConstructor] protected ProgrammableOperator(StorableConstructorFlag _) : base(_) { ProgrammableOperator.StaticInitialize(); Assemblies = defaultAssemblyDict.ToDictionary(kvp => kvp.Key, kvp => kvp.Value); Plugins = defaultPluginDict.ToDictionary(kvp => kvp.Key, kvp => kvp.Value.ToList()); } protected ProgrammableOperator(ProgrammableOperator original, Cloner cloner) : base(original, cloner) { code = original.Code; executeMethod = original.executeMethod; Assemblies = original.Assemblies.ToDictionary(kvp => kvp.Key, kvp => kvp.Value); Plugins = original.Plugins.ToDictionary(kvp => kvp.Key, kvp => kvp.Value); namespaces = new HashSet(original.namespaces); CompilationUnitCode = original.CompilationUnitCode; if (original.CompileErrors != null) CompileErrors = new CompilerErrorCollection(original.CompileErrors); RegisterEvents(); } public ProgrammableOperator() { code = ""; executeMethod = null; ProgrammableOperator.StaticInitialize(); Assemblies = defaultAssemblyDict.ToDictionary(kvp => kvp.Key, kvp => kvp.Value); Plugins = defaultPluginDict.ToDictionary(kvp => kvp.Key, kvp => kvp.Value.ToList()); namespaces = new HashSet(DiscoverNamespaces()); RegisterEvents(); } [StorableHook(HookType.AfterDeserialization)] private void AfterDeserialization() { // ensure default namespaces and assemblies are present if deserializing old operators namespaces.Add("HeuristicLab.Operators.Programmable"); Assemblies[typeof(HeuristicLab.Operators.Operator).Assembly] = true; Assemblies[typeof(HeuristicLab.Operators.Programmable.ProgrammableOperator).Assembly] = true; Assemblies[typeof(System.Threading.Barrier).Assembly] = true; // ensure new System assembly is selected (4.0.0.0) RegisterEvents(); } public override IDeepCloneable Clone(Cloner cloner) { return new ProgrammableOperator(this, cloner); } private void RegisterEvents() { Parameters.ItemsAdded += Parameters_Changed; Parameters.ItemsRemoved += Parameters_Changed; Parameters.ItemsReplaced += Parameters_Changed; Parameters.CollectionReset += Parameters_Changed; } private void Parameters_Changed(object sender, CollectionItemsChangedEventArgs args) { OnSignatureChanged(); } protected void OnSignatureChanged() { executeMethod = null; EventHandler handler = SignatureChanged; if (handler != null) handler(this, EventArgs.Empty); } private static void StaticInitialize() { lock (initLock) { if (defaultPluginDict != null && defaultAssemblyDict != null) return; defaultAssemblyDict = DiscoverAssemblies(); defaultPluginDict = GroupAssemblies(defaultAssemblyDict.Keys); } } private static Dictionary> GroupAssemblies(IEnumerable assemblies) { var plugins = new Dictionary>(); var locationTable = assemblies.ToDictionary(a => a.Location, a => a); foreach (var plugin in ApplicationManager.Manager.Plugins) { var aList = new List(); foreach (var aName in from file in plugin.Files where file.Type == PluginFileType.Assembly select file.Name) { Assembly a; locationTable.TryGetValue(aName, out a); if (a != null) { aList.Add(a); locationTable.Remove(aName); } } plugins[plugin.Name] = aList; } plugins["other"] = locationTable.Values.ToList(); return plugins; } protected static List defaultAssemblies = new List() { // mscorlib automatically included (would cause duplicate) typeof(System.ComponentModel.INotifyPropertyChanged).Assembly, // System.dll typeof(System.Linq.Enumerable).Assembly, // System.Core.dll typeof(System.Data.Linq.DataContext).Assembly, // System.Data.Linq.dll typeof(HeuristicLab.Common.IDeepCloneable).Assembly, typeof(HeuristicLab.Core.Item).Assembly, typeof(HeuristicLab.Data.IntValue).Assembly, typeof(HeuristicLab.Parameters.ValueParameter).Assembly, typeof(HeuristicLab.Collections.ObservableList).Assembly, typeof(HeuristicLab.Operators.Operator).Assembly, typeof(HeuristicLab.Operators.Programmable.ProgrammableOperator).Assembly, }; protected static Dictionary DiscoverAssemblies() { var assemblies = new Dictionary(); foreach (var a in AppDomain.CurrentDomain.GetAssemblies()) { try { if (File.Exists(a.Location)) { assemblies.Add(a, false); } } catch (NotSupportedException) { // NotSupportedException is thrown while accessing // the Location property of the anonymously hosted // dynamic methods assembly, which is related to // LINQ queries } } foreach (var a in defaultAssemblies) { if (assemblies.ContainsKey(a)) { assemblies[a] = true; } else { assemblies.Add(a, true); } } return assemblies; } protected static List DiscoverNamespaces() { return new List() { "HeuristicLab.Common", "HeuristicLab.Core", "HeuristicLab.Data", "HeuristicLab.Parameters", "HeuristicLab.Operators.Programmable", "System", "System.Collections.Generic", "System.Text", "System.Linq", "System.Data.Linq", }; } #endregion #region Compilation private static CSharpCodeProvider codeProvider = new CSharpCodeProvider( new Dictionary() { { "CompilerVersion", "v4.0" }, // support C# 4.0 syntax }); private CompilerResults DoCompile() { CompilerParameters parameters = new CompilerParameters(); parameters.GenerateExecutable = false; parameters.GenerateInMemory = true; parameters.IncludeDebugInformation = false; parameters.ReferencedAssemblies.AddRange(SelectedAssemblies.Select(a => a.Location).ToArray()); var unit = CreateCompilationUnit(); var writer = new StringWriter(); codeProvider.GenerateCodeFromCompileUnit( unit, writer, new CodeGeneratorOptions() { BracingStyle = "C", ElseOnClosing = true, IndentString = " ", }); CompilationUnitCode = writer.ToString(); return codeProvider.CompileAssemblyFromDom(parameters, unit); } public virtual void Compile() { var results = DoCompile(); executeMethod = null; CompileErrors = results.Errors; if (results.Errors.HasErrors) { StringBuilder sb = new StringBuilder(); foreach (CompilerError error in results.Errors) { sb.Append(error.Line).Append(':') .Append(error.Column).Append(": ") .AppendLine(error.ErrorText); } throw new Exception(string.Format( "Compilation of \"{0}\" failed:{1}{2}", Name, Environment.NewLine, sb.ToString())); } else { Assembly assembly = results.CompiledAssembly; Type[] types = assembly.GetTypes(); executeMethod = types[0].GetMethod("Execute"); } } private CodeCompileUnit CreateCompilationUnit() { CodeNamespace ns = new CodeNamespace("HeuristicLab.Operators.Programmable.CustomOperators"); ns.Types.Add(CreateType()); ns.Imports.AddRange( GetSelectedAndValidNamespaces() .Select(n => new CodeNamespaceImport(n)) .ToArray()); CodeCompileUnit unit = new CodeCompileUnit(); unit.Namespaces.Add(ns); return unit; } public IEnumerable GetSelectedAndValidNamespaces() { var possibleNamespaces = new HashSet(GetAllNamespaces(true)); foreach (var ns in Namespaces) if (possibleNamespaces.Contains(ns)) yield return ns; } public static readonly Regex SafeTypeNameCharRegex = new Regex("[_a-zA-Z0-9]+"); public static readonly Regex SafeTypeNameRegex = new Regex("[_a-zA-Z][_a-zA-Z0-9]*"); public string CompiledTypeName { get { var sb = new StringBuilder(); foreach (string s in SafeTypeNameCharRegex.Matches(Name).Cast().Select(m => m.Value)) { sb.Append(s); } return SafeTypeNameRegex.Match(sb.ToString()).Value; } } private CodeTypeDeclaration CreateType() { CodeTypeDeclaration typeDecl = new CodeTypeDeclaration(CompiledTypeName) { IsClass = true, TypeAttributes = TypeAttributes.Public, }; typeDecl.Members.Add(CreateMethod()); return typeDecl; } public string Signature { get { var sb = new StringBuilder() .Append("public static IOperation Execute(") .Append(TypeNameParser.Parse(GetType().FullName).GetTypeNameInCode(namespaces)) .Append(" op, ") .Append(TypeNameParser.Parse(typeof(IExecutionContext).FullName).GetTypeNameInCode(namespaces)) .Append(" context"); foreach (IParameter param in Parameters) { sb.Append(String.Format(", {0} {1}", TypeNameParser.Parse(param.GetType().FullName).GetTypeNameInCode(namespaces), param.Name)); } return sb.Append(")").ToString(); } } public event EventHandler SignatureChanged; private static Regex lineSplitter = new Regex(@"\r\n|\r|\n"); private CodeMemberMethod CreateMethod() { CodeMemberMethod method = new CodeMemberMethod(); method.Name = "Execute"; method.ReturnType = new CodeTypeReference(typeof(IOperation)); method.Attributes = MemberAttributes.Public | MemberAttributes.Static; method.Parameters.Add(new CodeParameterDeclarationExpression(GetType(), "op")); method.Parameters.Add(new CodeParameterDeclarationExpression(typeof(IExecutionContext), "context")); foreach (var param in Parameters) method.Parameters.Add(new CodeParameterDeclarationExpression(param.GetType(), param.Name)); string[] codeLines = lineSplitter.Split(code); for (int i = 0; i < codeLines.Length; i++) { codeLines[i] = string.Format("#line {0} \"ProgrammableOperator\"{1}{2}", i + 1, "\r\n", codeLines[i]); } method.Statements.Add(new CodeSnippetStatement(string.Join("\r\n", codeLines) + MethodSuffix)); return method; } public virtual string MethodSuffix { get { return "return null;"; } } #endregion #region HeuristicLab interfaces public override IOperation Apply() { lock (syncRoot) { if (executeMethod == null) { Compile(); } } var parameters = new List() { this, ExecutionContext }; parameters.AddRange(Parameters.Select(p => (object)p)); return (IOperation)executeMethod.Invoke(null, parameters.ToArray()); } public event EventHandler CodeChanged; protected virtual void OnCodeChanged() { EventHandler handler = CodeChanged; if (handler != null) handler(this, EventArgs.Empty); } #endregion } }