Changeset 5431 for trunk/sources/HeuristicLab.Problems.DataAnalysis/3.3/Symbolic/Formatters

Timestamp:

02/04/11 12:38:34 (14 years ago)

Author:

gkronber

Message:

#1314 Merged MATLAB formatter from branch into trunk and implemented additional symbols related to dynamic modeling.

File:

• trunk/sources/HeuristicLab.Problems.DataAnalysis/3.3/Symbolic/Formatters/SymbolicExpressionTreeMATLABFormatter.cs (copied) (copied from branches/DataAnalysis.Extensions/HeuristicLab.Problems.DataAnalysis/3.3/Symbolic/Formatters/SymbolicExpressionTreeMATLABFormatter.cs) (6 diffs)

trunk/sources/HeuristicLab.Problems.DataAnalysis/3.3/Symbolic/Formatters/SymbolicExpressionTreeMATLABFormatter.cs

@@ -31 +31 @@
 using System.Collections.Generic;
 using System;
+using System.Globalization;
 
 namespace HeuristicLab.Encodings.SymbolicExpressionTreeEncoding.Formatters {
 
   [Item("SymbolicExpressionTreeMATLABFormatter", "String formatter for string representations of symbolic expression trees in MATLAB syntax.")]
-  public class SymbolicExpressionTreeMATLABFormatter : NamedItem, ISymbolicExpressionTreeStringFormatter {
-
-    protected SymbolicExpressionTreeMATLABFormatter(SymbolicExpressionTreeMATLABFormatter original, Cloner cloner) : base(original, cloner) { }
+  [StorableClass]
+  public sealed class SymbolicExpressionTreeMATLABFormatter : NamedItem, ISymbolicExpressionTreeStringFormatter {
+    private int currentLag;
+
+    [StorableConstructor]
+    private SymbolicExpressionTreeMATLABFormatter(bool deserializing) : base(deserializing) { }
+    private SymbolicExpressionTreeMATLABFormatter(SymbolicExpressionTreeMATLABFormatter original, Cloner cloner) : base(original, cloner) { }
     public SymbolicExpressionTreeMATLABFormatter()
       : base() {
       Name = "MATLAB String Formatter";
     }
+    public override IDeepCloneable Clone(Cloner cloner) {
+      return new SymbolicExpressionTreeMATLABFormatter(this, cloner);
+    }
+    private int currentIndexNumber;
+    public string CurrentIndexVariable {
+      get {
+        return "i" + currentIndexNumber;
+      }
+    }
+    private void ReleaseIndexVariable() {
+      currentIndexNumber--;
+    }
+
+    private string AllocateIndexVariable() {
+      currentIndexNumber++;
+      return CurrentIndexVariable;
+    }
 
     public string Format(SymbolicExpressionTree symbolicExpressionTree) {
+      currentLag = 0;
+      currentIndexNumber = 0;
       return FormatRecursively(symbolicExpressionTree.Root);
     }
 
     private string FormatRecursively(SymbolicExpressionTreeNode node) {
-
       Symbol symbol = node.Symbol;
-
       StringBuilder stringBuilder = new StringBuilder();
 
@@ -63 +85 @@
           stringBuilder.AppendLine("  " + variableName + " = Data(:, ???);");
         stringBuilder.AppendLine();
-        stringBuilder.AppendLine("  for i = size(Data,1):-1:1");
-        stringBuilder.AppendLine("    Target_estimated(i) = " + FormatRecursively(node.SubTrees[0]) + ";");
+        stringBuilder.AppendLine("  for "+CurrentIndexVariable+" = size(Data,1):-1:1");
+        stringBuilder.AppendLine("    Target_estimated("+CurrentIndexVariable+") = " + FormatRecursively(node.SubTrees[0]) + ";");
         stringBuilder.AppendLine();
         stringBuilder.AppendLine("function y = log_(x)");
         stringBuilder.AppendLine("  if(x<=0) y = NaN;");
         stringBuilder.AppendLine("  else     y = log(x);");
+        stringBuilder.AppendLine(
+@"
+function y = fivePoint(f0, f1, f3, f4)
+  y = (f0 + 2*f1 - 2*f3 - f4) / 8");
         return stringBuilder.ToString();
       }
@@ -104 +130 @@
       } else if (symbol is Constant) {
         ConstantTreeNode constantTreeNode = node as ConstantTreeNode;
-        stringBuilder.Append(constantTreeNode.Value.ToString().Replace(",", "."));
+        stringBuilder.Append(constantTreeNode.Value.ToString(CultureInfo.InvariantCulture));
       } else if (symbol is Cosine) {
         stringBuilder.Append("cos(");
@@ -143 +169 @@
         stringBuilder.Append(FormatRecursively(node.SubTrees[2]));
       } else if (symbol is LaggedVariable) {
+        // this if must be checked before if(symbol is LaggedVariable)
         LaggedVariableTreeNode laggedVariableTreeNode = node as LaggedVariableTreeNode;
-        stringBuilder.Append(laggedVariableTreeNode.Weight.ToString().Replace(",", "."));
+        stringBuilder.Append(laggedVariableTreeNode.Weight.ToString(CultureInfo.InvariantCulture));
         stringBuilder.Append("*");
-        stringBuilder.Append(laggedVariableTreeNode.VariableName + "(i-" + laggedVariableTreeNode.Lag + ")");
+        stringBuilder.Append(laggedVariableTreeNode.VariableName + LagToString(currentLag + laggedVariableTreeNode.Lag));
       } else if (symbol is LessThan) {
         stringBuilder.Append("((");
@@ -195 +222 @@
       } else if (symbol is HeuristicLab.Problems.DataAnalysis.Symbolic.Symbols.Variable) {
         VariableTreeNode variableTreeNode = node as VariableTreeNode;
-        stringBuilder.Append(variableTreeNode.Weight.ToString().Replace(",", "."));
+        stringBuilder.Append(variableTreeNode.Weight.ToString(CultureInfo.InvariantCulture));
         stringBuilder.Append("*");
-        stringBuilder.Append(variableTreeNode.VariableName + "(i)");
+        stringBuilder.Append(variableTreeNode.VariableName + LagToString(currentLag));
+      } else if (symbol is Power) {
+        stringBuilder.Append("(");
+        stringBuilder.Append(FormatRecursively(node.SubTrees[0]));
+        stringBuilder.Append(")^round(");
+        stringBuilder.Append(FormatRecursively(node.SubTrees[1]));
+        stringBuilder.Append(")");
+      } else if (symbol is Root) {
+        stringBuilder.Append("(");
+        stringBuilder.Append(FormatRecursively(node.SubTrees[0]));
+        stringBuilder.Append(")^(1 / round(");
+        stringBuilder.Append(FormatRecursively(node.SubTrees[1]));
+        stringBuilder.Append("))");
+      } else if (symbol is Derivative) {
+        stringBuilder.Append("fivePoint(");
+        // f0
+        stringBuilder.Append(FormatRecursively(node.SubTrees[0]));
+        stringBuilder.Append(", ");
+        // f1
+        currentLag--;
+        stringBuilder.Append(FormatRecursively(node.SubTrees[0]));
+        stringBuilder.Append(", ");
+        // f3
+        currentLag -= 2;
+        stringBuilder.Append(FormatRecursively(node.SubTrees[0]));
+        stringBuilder.Append(", ");
+        currentLag--;
+        // f4
+        stringBuilder.Append(FormatRecursively(node.SubTrees[0]));
+        stringBuilder.Append(")");
+        currentLag += 4;
+      } else if (symbol is Integral) {
+        var laggedNode = node as LaggedTreeNode;
+        string prevCounterVariable = CurrentIndexVariable;
+        string counterVariable = AllocateIndexVariable();
+        stringBuilder.AppendLine(" sum (map(@(" + counterVariable + ") " + FormatRecursively(node.SubTrees[0]) + ", (" + prevCounterVariable + "+" + laggedNode.Lag + "):" + prevCounterVariable + "))");
+        ReleaseIndexVariable();
+      } else if (symbol is TimeLag) {
+        var laggedNode = node as LaggedTreeNode;
+        currentLag += laggedNode.Lag;
+        stringBuilder.Append(FormatRecursively(node.SubTrees[0]));
+        currentLag -= laggedNode.Lag;
       } else {
         stringBuilder.Append("ERROR");
@@ -203 +271 @@
 
       stringBuilder.Append(")");
-
       return stringBuilder.ToString();
     }
 
-    public override IDeepCloneable Clone(Cloner cloner) {
-      return new SymbolicExpressionTreeMATLABFormatter(this, cloner);
-    }
+
+    private string LagToString(int lag) {
+      if (lag < 0) {
+        return "(" + CurrentIndexVariable + "" + lag + ")";
+      } else if (lag > 0) {
+        return "(" + CurrentIndexVariable + "+" + lag + ")";
+      } else return "(" + CurrentIndexVariable + ")";
+    }
+
   }
 }