Changeset 10846

Timestamp:

05/14/14 13:19:12 (10 years ago)

Author:

tsteinre

Message:

• TransformationToSymbolicTreeMapper implemented GenerateModel for
  • LinearTransformation

File:

• branches/DataPreprocessing/HeuristicLab.Problems.DataAnalysis.Symbolic/3.4/TransformationToSymbolicTreeMapper.cs (modified) (4 diffs)

branches/DataPreprocessing/HeuristicLab.Problems.DataAnalysis.Symbolic/3.4/TransformationToSymbolicTreeMapper.cs

@@ -20 +20 @@
 #endregion
 
+using System;
 using HeuristicLab.Encodings.SymbolicExpressionTreeEncoding;
 using HeuristicLab.Problems.DataAnalysis.Transformations;
@@ -25 +26 @@
 namespace HeuristicLab.Problems.DataAnalysis.Symbolic {
   public class TransformationToSymbolicTreeMapper : ITransformationMapper<ISymbolicExpressionTree> {
+    private ITransformation transformation;
+    private string column;
+    private ISymbolicExpressionTree tree;
+
     #region ITransformationMapper<ISymbolicExpressionTree> Members
 
     public ISymbolicExpressionTree GenerateModel(ITransformation transformation) {
-      var tree = CreateNewTree();
-      var column = transformation.Column;
+      InitComponents(transformation);
 
       if (transformation is LinearTransformation) {
-        var kValue = ((LinearTransformation)transformation).Multiplier;
-        var multiplicationNode = new Multiplication().CreateTreeNode();
-        var kNode = new Constant();
+        return GenerateModelForLinearTransformation();
       } else if (transformation is ExponentialTransformation) {
 
@@ -46 +48 @@
 
       }
-
-      //var variableNode = (VariableTreeNode)new Variable("dummy", "dummy description").CreateTreeNode();
-      //variableNode.VariableName = "dummy";
-      //tree.Root.AddSubtree(variableNode);
-
-      return tree;
+      throw new NotImplementedException();
     }
 
     public ISymbolicExpressionTree GenerateInverseModel(ITransformation transformation) {
-      var tree = CreateNewTree();
-      var column = transformation.Column;
+      InitComponents(transformation);
 
       if (transformation is LinearTransformation) {
@@ -72 +68 @@
       }
 
-      return tree;
+      throw new NotImplementedException();
     }
 
     #endregion
 
+    private ISymbolicExpressionTree GenerateModelForLinearTransformation() {
+      var linearTransformation = (LinearTransformation)transformation;
+      var tree = CreateNewTree();
+      string column = linearTransformation.Column;
+      var kValue = linearTransformation.Multiplier;
+      var dValue = linearTransformation.Addend;
+
+      // k * x
+      var multiplicationNode = new Multiplication().CreateTreeNode();
+      var kNode = new ConstantTreeNode(new Constant() { Name = "K" }) { Value = kValue };
+      var xNode = new Variable(column, "x").CreateTreeNode();
+      multiplicationNode.AddSubtree(kNode);
+      multiplicationNode.AddSubtree(xNode);
+
+      // ( k * x ) + d
+      var additionNode = new Addition().CreateTreeNode();
+      var dNode = new ConstantTreeNode(new Constant() { Name = "d" }) { Value = dValue };
+      additionNode.AddSubtree(multiplicationNode);
+      additionNode.AddSubtree(dNode);
+
+      tree.Root.AddSubtree(additionNode);
+      return tree;
+    }
+
     private ISymbolicExpressionTree CreateNewTree() {
       return new SymbolicExpressionTree(new ProgramRootSymbol().CreateTreeNode());
     }
+
+    private void InitComponents(ITransformation transformation) {
+      this.transformation = transformation;
+      column = transformation.Column;
+      tree = CreateNewTree();
+    }
   }
 }