Free cookie consent management tool by TermsFeed Policy Generator

source: branches/GP-Refactoring-713/sources/HeuristicLab.GP.StructureIdentification/3.3/FunctionLibraryInjector.cs @ 2202

Last change on this file since 2202 was 2202, checked in by gkronber, 15 years ago

Created a branch for #713
File size: 12.4 KB
Line 
1#region License Information
2/* HeuristicLab
3 * Copyright (C) 2002-2008 Heuristic and Evolutionary Algorithms Laboratory (HEAL)
4 *
5 * This file is part of HeuristicLab.
6 *
7 * HeuristicLab is free software: you can redistribute it and/or modify
8 * it under the terms of the GNU General Public License as published by
9 * the Free Software Foundation, either version 3 of the License, or
10 * (at your option) any later version.
11 *
12 * HeuristicLab is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15 * GNU General Public License for more details.
16 *
17 * You should have received a copy of the GNU General Public License
18 * along with HeuristicLab. If not, see <http://www.gnu.org/licenses/>.
19 */
20#endregion
21
22using System;
23using System.Collections.Generic;
24using System.Text;
25using System.Xml;
26using HeuristicLab.Core;
27using HeuristicLab.Data;
28using HeuristicLab.DataAnalysis;
29using HeuristicLab.Constraints;
30using StructId = HeuristicLab.GP.StructureIdentification;
31
32namespace HeuristicLab.GP.StructureIdentification {
33  public class FunctionLibraryInjector : OperatorBase {
34    private const string FUNCTIONLIBRARY = "FunctionLibrary";
35    private const string TARGETVARIABLE = "TargetVariable";
36    private const string MINTIMEOFFSET = "MinTimeOffset";
37    private const string MAXTIMEOFFSET = "MaxTimeOffset";
38
39    private const string DIFFERENTIALS_ALLOWED = "Differentials";
40    private const string VARIABLES_ALLOWED = "Variables";
41    private const string CONSTANTS_ALLOWED = "Constants";
42    private const string ADDITION_ALLOWED = "Addition";
43    private const string AVERAGE_ALLOWED = "Average";
44    private const string AND_ALLOWED = "And";
45    private const string COSINUS_ALLOWED = "Cosinus";
46    private const string DIVISION_ALLOWED = "Division";
47    private const string EQUAL_ALLOWED = "Equal";
48    private const string EXPONENTIAL_ALLOWED = "Exponential";
49    private const string GREATERTHAN_ALLOWED = "GreaterThan";
50    private const string IFTHENELSE_ALLOWED = "IfThenElse";
51    private const string LESSTHAN_ALLOWED = "LessThan";
52    private const string LOGARTIHM_ALLOWED = "Logarithm";
53    private const string MULTIPLICATION_ALLOWED = "Multiplication";
54    private const string NOT_ALLOWED = "Not";
55    private const string POWER_ALLOWED = "Power";
56    private const string OR_ALLOWED = "Or";
57    private const string SIGNUM_ALLOWED = "Signum";
58    private const string SINUS_ALLOWED = "Sinus";
59    private const string SQRT_ALLOWED = "SquareRoot";
60    private const string SUBTRACTION_ALLOWED = "Subtraction";
61    private const string TANGENS_ALLOWED = "Tangens";
62    private const string XOR_ALLOWED = "Xor";
63
64
65    public override string Description {
66      get { return @"Injects a configurable function library for regression and classification problems."; }
67    }
68
69    public FunctionLibraryInjector()
70      : base() {
71      AddVariableInfo(new VariableInfo(TARGETVARIABLE, "The target variable", typeof(IntData), VariableKind.In));
72      AddVariableInfo(new VariableInfo(MINTIMEOFFSET, "Minimal time offset for all features", typeof(IntData), VariableKind.In));
73      AddVariableInfo(new VariableInfo(MAXTIMEOFFSET, "Maximal time offset for all feature", typeof(IntData), VariableKind.In));
74      AddVariableInfo(new VariableInfo(FUNCTIONLIBRARY, "Preconfigured default function library", typeof(FunctionLibrary), VariableKind.New));
75
76      AddVariable(DIFFERENTIALS_ALLOWED, false);
77      AddVariable(VARIABLES_ALLOWED, true);
78      AddVariable(CONSTANTS_ALLOWED, true);
79      AddVariable(ADDITION_ALLOWED, true);
80      AddVariable(AVERAGE_ALLOWED, false);
81      AddVariable(AND_ALLOWED, true);
82      AddVariable(COSINUS_ALLOWED, true);
83      AddVariable(DIVISION_ALLOWED, true);
84      AddVariable(EQUAL_ALLOWED, true);
85      AddVariable(EXPONENTIAL_ALLOWED, true);
86      AddVariable(GREATERTHAN_ALLOWED, true);
87      AddVariable(IFTHENELSE_ALLOWED, true);
88      AddVariable(LESSTHAN_ALLOWED, true);
89      AddVariable(LOGARTIHM_ALLOWED, true);
90      AddVariable(MULTIPLICATION_ALLOWED, true);
91      AddVariable(NOT_ALLOWED, true);
92      AddVariable(POWER_ALLOWED, true);
93      AddVariable(OR_ALLOWED, true);
94      AddVariable(SIGNUM_ALLOWED, true);
95      AddVariable(SINUS_ALLOWED, true);
96      AddVariable(SQRT_ALLOWED, true);
97      AddVariable(SUBTRACTION_ALLOWED, true);
98      AddVariable(TANGENS_ALLOWED, true);
99      AddVariable(XOR_ALLOWED, false);
100    }
101
102    private void AddVariable(string name, bool allowed) {
103      AddVariableInfo(new VariableInfo(name, name + " allowed", typeof(BoolData), Core.VariableKind.New));
104      GetVariableInfo(name).Local = true;
105      AddVariable(new Core.Variable(name, new BoolData(allowed)));
106    }
107
108    public override IOperation Apply(IScope scope) {
109      FunctionLibrary functionLibrary = new FunctionLibrary();
110      int targetVariable = GetVariableValue<IntData>(TARGETVARIABLE, scope, true).Data;
111
112      // try to get minTimeOffset (use 0 as default if not available)
113      IItem minTimeOffsetItem = GetVariableValue(MINTIMEOFFSET, scope, true, false);
114      int minTimeOffset = minTimeOffsetItem == null ? 0 : ((IntData)minTimeOffsetItem).Data;
115      // try to get maxTimeOffset (use 0 as default if not available)
116      IItem maxTimeOffsetItem = GetVariableValue(MAXTIMEOFFSET, scope, true, false);
117      int maxTimeOffset = maxTimeOffsetItem == null ? 0 : ((IntData)maxTimeOffsetItem).Data;
118
119      Variable variable = new Variable();
120      Constant constant = new Constant();
121      Differential differential = new Differential();
122      Addition addition = new Addition();
123      And and = new And();
124      Average average = new Average();
125      Cosinus cosinus = new Cosinus();
126      Division division = new Division();
127      Equal equal = new Equal();
128      Exponential exponential = new Exponential();
129      GreaterThan greaterThan = new GreaterThan();
130      IfThenElse ifThenElse = new IfThenElse();
131      LessThan lessThan = new LessThan();
132      Logarithm logarithm = new Logarithm();
133      Multiplication multiplication = new Multiplication();
134      Not not = new Not();
135      Or or = new Or();
136      Power power = new Power();
137      Signum signum = new Signum();
138      Sinus sinus = new Sinus();
139      Sqrt sqrt = new Sqrt();
140      Subtraction subtraction = new Subtraction();
141      Tangens tangens = new Tangens();
142      Xor xor = new Xor();
143
144
145      List<IFunction> booleanFunctions = new List<IFunction>();
146      ConditionalAddFunction(AND_ALLOWED, and, booleanFunctions);
147      ConditionalAddFunction(EQUAL_ALLOWED, equal, booleanFunctions);
148      ConditionalAddFunction(GREATERTHAN_ALLOWED, greaterThan, booleanFunctions);
149      ConditionalAddFunction(LESSTHAN_ALLOWED, lessThan, booleanFunctions);
150      ConditionalAddFunction(NOT_ALLOWED, not, booleanFunctions);
151      ConditionalAddFunction(OR_ALLOWED, or, booleanFunctions);
152      ConditionalAddFunction(XOR_ALLOWED, xor, booleanFunctions);
153
154      List<IFunction> doubleFunctions = new List<IFunction>();
155      ConditionalAddFunction(DIFFERENTIALS_ALLOWED, differential, doubleFunctions);
156      ConditionalAddFunction(VARIABLES_ALLOWED, variable, doubleFunctions);
157      ConditionalAddFunction(CONSTANTS_ALLOWED, constant, doubleFunctions);
158      ConditionalAddFunction(ADDITION_ALLOWED, addition, doubleFunctions);
159      ConditionalAddFunction(AVERAGE_ALLOWED, average, doubleFunctions);
160      ConditionalAddFunction(COSINUS_ALLOWED, cosinus, doubleFunctions);
161      ConditionalAddFunction(DIVISION_ALLOWED, division, doubleFunctions);
162      ConditionalAddFunction(EXPONENTIAL_ALLOWED, exponential, doubleFunctions);
163      ConditionalAddFunction(IFTHENELSE_ALLOWED, ifThenElse, doubleFunctions);
164      ConditionalAddFunction(LOGARTIHM_ALLOWED, logarithm, doubleFunctions);
165      ConditionalAddFunction(MULTIPLICATION_ALLOWED, multiplication, doubleFunctions);
166      ConditionalAddFunction(POWER_ALLOWED, power, doubleFunctions);
167      ConditionalAddFunction(SIGNUM_ALLOWED, signum, doubleFunctions);
168      ConditionalAddFunction(SINUS_ALLOWED, sinus, doubleFunctions);
169      ConditionalAddFunction(SQRT_ALLOWED, sqrt, doubleFunctions);
170      ConditionalAddFunction(SUBTRACTION_ALLOWED, subtraction, doubleFunctions);
171      ConditionalAddFunction(TANGENS_ALLOWED, tangens, doubleFunctions);
172
173      SetAllowedSubOperators(and, booleanFunctions);
174      SetAllowedSubOperators(equal, doubleFunctions);
175      SetAllowedSubOperators(greaterThan, doubleFunctions);
176      SetAllowedSubOperators(lessThan, doubleFunctions);
177      SetAllowedSubOperators(not, booleanFunctions);
178      SetAllowedSubOperators(or, booleanFunctions);
179      SetAllowedSubOperators(xor, booleanFunctions);
180      SetAllowedSubOperators(addition, doubleFunctions);
181      SetAllowedSubOperators(average, doubleFunctions);
182      SetAllowedSubOperators(cosinus, doubleFunctions);
183      SetAllowedSubOperators(division, doubleFunctions);
184      SetAllowedSubOperators(exponential, doubleFunctions);
185      SetAllowedSubOperators(ifThenElse, 0, booleanFunctions);
186      SetAllowedSubOperators(ifThenElse, 1, doubleFunctions);
187      SetAllowedSubOperators(ifThenElse, 2, doubleFunctions);
188      SetAllowedSubOperators(logarithm, doubleFunctions);
189      SetAllowedSubOperators(multiplication, doubleFunctions);
190      SetAllowedSubOperators(power, doubleFunctions);
191      SetAllowedSubOperators(signum, doubleFunctions);
192      SetAllowedSubOperators(sinus, doubleFunctions);
193      SetAllowedSubOperators(sqrt, doubleFunctions);
194      SetAllowedSubOperators(subtraction, doubleFunctions);
195      SetAllowedSubOperators(tangens, doubleFunctions);
196
197      ConditionalAddOperator(DIFFERENTIALS_ALLOWED, functionLibrary, differential);
198      ConditionalAddOperator(VARIABLES_ALLOWED, functionLibrary, variable);
199      ConditionalAddOperator(CONSTANTS_ALLOWED, functionLibrary, constant);
200      ConditionalAddOperator(ADDITION_ALLOWED, functionLibrary, addition);
201      ConditionalAddOperator(AVERAGE_ALLOWED, functionLibrary, average);
202      ConditionalAddOperator(AND_ALLOWED, functionLibrary, and);
203      ConditionalAddOperator(COSINUS_ALLOWED, functionLibrary, cosinus);
204      ConditionalAddOperator(DIVISION_ALLOWED, functionLibrary, division);
205      ConditionalAddOperator(EQUAL_ALLOWED, functionLibrary, equal);
206      ConditionalAddOperator(EXPONENTIAL_ALLOWED, functionLibrary, exponential);
207      ConditionalAddOperator(GREATERTHAN_ALLOWED, functionLibrary, greaterThan);
208      ConditionalAddOperator(IFTHENELSE_ALLOWED, functionLibrary, ifThenElse);
209      ConditionalAddOperator(LESSTHAN_ALLOWED, functionLibrary, lessThan);
210      ConditionalAddOperator(LOGARTIHM_ALLOWED, functionLibrary, logarithm);
211      ConditionalAddOperator(MULTIPLICATION_ALLOWED, functionLibrary, multiplication);
212      ConditionalAddOperator(NOT_ALLOWED, functionLibrary, not);
213      ConditionalAddOperator(POWER_ALLOWED, functionLibrary, power);
214      ConditionalAddOperator(OR_ALLOWED, functionLibrary, or);
215      ConditionalAddOperator(SIGNUM_ALLOWED, functionLibrary, signum);
216      ConditionalAddOperator(SINUS_ALLOWED, functionLibrary, sinus);
217      ConditionalAddOperator(SQRT_ALLOWED, functionLibrary, sqrt);
218      ConditionalAddOperator(SUBTRACTION_ALLOWED, functionLibrary, subtraction);
219      ConditionalAddOperator(TANGENS_ALLOWED, functionLibrary, tangens);
220      ConditionalAddOperator(XOR_ALLOWED, functionLibrary, xor);
221
222      variable.SetConstraints(minTimeOffset, maxTimeOffset);
223      differential.SetConstraints(minTimeOffset, maxTimeOffset);
224
225      scope.AddVariable(new HeuristicLab.Core.Variable(scope.TranslateName(FUNCTIONLIBRARY), functionLibrary));
226
227      return null;
228    }
229
230    private void ConditionalAddFunction(string condName, IFunction fun, List<IFunction> list) {
231      if (GetVariableValue<BoolData>(condName, null, false).Data) list.Add(fun);
232    }
233
234    private void ConditionalAddOperator(string condName, FunctionLibrary functionLib, IFunction op) {
235      if (GetVariableValue<BoolData>(condName, null, false).Data) functionLib.AddFunction(op);
236    }
237
238    private void SetAllowedSubOperators(IFunction f, List<IFunction> gs) {
239      for (int i = 0; i < f.MaxArity; i++) {
240        SetAllowedSubOperators(f, i, gs);
241      }
242    }
243
244    private void SetAllowedSubOperators(IFunction f, int i, List<IFunction> gs) {
245      gs.ForEach((g) => f.AddAllowedSubFunction(g, i));
246    }
247  }
248}
Note: See TracBrowser for help on using the repository browser.