Changeset 10883

Timestamp:

05/21/14 15:58:40 (11 years ago)

Author:

pfleck

Message:

• Implemented inverse backtransformation.

File:

• branches/DataPreprocessing/HeuristicLab.Problems.DataAnalysis.Symbolic/3.4/SymbolicExpressionTreeBacktransformator.cs (modified) (2 diffs)

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

@@ -43 +43 @@
 
     private void ApplyBacktransformation(ITransformation transformation, ISymbolicExpressionTree symbolicExpressionTree, string targetVariable) {
-      var variableNodes = symbolicExpressionTree.IterateNodesBreadth()
-        .OfType<VariableTreeNode>()
-        .Where(n => n.VariableName == transformation.Column);
+      if (transformation.Column != targetVariable) {
+        var variableNodes = symbolicExpressionTree.IterateNodesBreadth()
+          .OfType<VariableTreeNode>()
+          .Where(n => n.VariableName == transformation.Column);
+        ApplyRegularBacktransformation(transformation, variableNodes);
+      } else {
+        ApplyInverseBacktransformation(transformation, symbolicExpressionTree);
+      }
+    }
 
+    private void ApplyRegularBacktransformation(ITransformation transformation, IEnumerable<VariableTreeNode> variableNodes) {
       foreach (var variableNode in variableNodes) {
         // generate new subtrees because same subtree cannot be added more than once
         var transformationTree = transformationMapper.GenerateModel(transformation);
-        SwapTransformationWithVariable(transformationTree, variableNode);
+        SwapVariableWithTree(variableNode, transformationTree);
       }
     }
 
-    private void SwapTransformationWithVariable(ISymbolicExpressionTreeNode transformationTree, VariableTreeNode variableNode) {
+    private void ApplyInverseBacktransformation(ITransformation transformation, ISymbolicExpressionTree symbolicExpressionTree) {
+      var startSymbol = symbolicExpressionTree.Root.GetSubtree(0);
+      var modelTree = startSymbol.GetSubtree(0);
+      startSymbol.RemoveSubtree(0);
+
+      var transformationTree = transformationMapper.GenerateInverseModel(transformation);
+      var variableNode = transformationTree.IterateNodesBreadth()
+        .OfType<VariableTreeNode>()
+        .Single(n => n.VariableName == transformation.Column);
+
+      SwapVariableWithTree(variableNode, modelTree);
+
+      startSymbol.AddSubtree(transformationTree);
+    }
+
+    private void SwapVariableWithTree(VariableTreeNode variableNode, ISymbolicExpressionTreeNode treeNode) {
       var parent = variableNode.Parent;
       int index = parent.IndexOfSubtree(variableNode);
       parent.RemoveSubtree(index);
 
-      if (!variableNode.Weight.IsAlmost(1.0)) {
-        transformationTree = CreateNodeFromWeight(transformationTree, variableNode);
-      }
+      if (!variableNode.Weight.IsAlmost(1.0))
+        treeNode = CreateNodeFromWeight(treeNode, variableNode);
 
-      parent.InsertSubtree(index, transformationTree);
+      parent.InsertSubtree(index, treeNode);
     }
 
@@ -70 +91 @@
       multiplicationNode.AddSubtree(new ConstantTreeNode(new Constant()) { Value = variableNode.Weight });
       multiplicationNode.AddSubtree(transformationTree);
-      transformationTree = multiplicationNode;
-      return transformationTree;
+      return multiplicationNode;
     }
   }