Changeset 15881

Timestamp:

04/03/18 08:14:29 (6 years ago)

Author:

fbaching

Message:

#2911: adding IfThenElse, NOT operator, AVG function
change visibility of util methods

File:

• branches/2911_TSQLSymbolicExpressionFormatter/HeuristicLab.Problems.DataAnalysis.Symbolic/3.4/Formatters/TSQLExpressionFormatter.cs (modified) (10 diffs)

branches/2911_TSQLSymbolicExpressionFormatter/HeuristicLab.Problems.DataAnalysis.Symbolic/3.4/Formatters/TSQLExpressionFormatter.cs

@@ -31 +31 @@
 
 namespace HeuristicLab.Problems.DataAnalysis.Symbolic {
-  public static class StringBuilderExtensions {
-    public static void AppendIndented(this StringBuilder strBuilder, int level, string text) {
-      strBuilder.Append(new String(' ', level * 2));
+  internal static class StringBuilderExtensions {
+    internal static void AppendIndented(this StringBuilder strBuilder, int level, string text) {
+      strBuilder.Append(new string(' ', level * 2));
       strBuilder.Append(text);
     }
-    public static void AppendLineIndented(this StringBuilder strBuilder, int level, string text) {
-      strBuilder.Append(new String(' ', level * 2));
+    internal static void AppendLineIndented(this StringBuilder strBuilder, int level, string text) {
+      strBuilder.Append(new string(' ', level * 2));
       strBuilder.AppendLine(text);
     }
@@ -104 +104 @@
 
       //add variable declaration for convenience
-      strBuilder.AppendLineIndented(1,$"-- added variable declaration for convenience");
+      strBuilder.AppendLineIndented(1, $"-- added variable declaration for convenience");
       foreach (var name in sortedVarcharIdentifiers)
         strBuilder.AppendLineIndented(1, $"-- DECLARE {name} NVARCHAR(max) = ''");
@@ -114 +114 @@
 
     private void FormatRecursively(int level, ISymbolicExpressionTreeNode node, StringBuilder strBuilder) {
-      // TODO: adapt to interpreter semantics. The HL interpreter also allows Boolean operations on reals
       if (node.Subtrees.Any()) {
         if (node.Symbol is Addition) {
@@ -121 +120 @@
           FormatOperator(level, node, "AND", strBuilder);
         } else if (node.Symbol is Average) {
-          throw new NotSupportedException($"Symbol {node.Symbol.GetType().Name} not yet supported.");
+          FormatAverage(level, node, strBuilder);
         } else if (node.Symbol is Cosine) {
           FormatFunction(level, node, "COS", strBuilder);
@@ -131 +130 @@
           FormatOperator(level, node, ">", strBuilder);
         } else if (node.Symbol is IfThenElse) {
-          throw new NotSupportedException($"Symbol {node.Symbol.GetType().Name} not yet supported.");
+          FormatIfThenElse(level, node, strBuilder);
         } else if (node.Symbol is LessThan) {
           FormatOperator(level, node, "<", strBuilder);
@@ -139 +138 @@
           FormatOperator(level, node, "*", strBuilder);
         } else if (node.Symbol is Not) {
-          throw new NotSupportedException($"Symbol {node.Symbol.GetType().Name} not yet supported.");
+          FormatOperator(level, node, "NOT LIKE", strBuilder);
         } else if (node.Symbol is Or) {
           FormatOperator(level, node, "OR", strBuilder);
@@ -159 +158 @@
           FormatRoot(level, node, strBuilder);
         } else {
-          throw new NotSupportedException("Formatting of symbol: " + node.Symbol + " not supported for C# symbolic expression tree formatter.");
+          throw new NotSupportedException("Formatting of symbol: " + node.Symbol + " not supported for TSQL symbolic expression tree formatter.");
         }
       } else {
@@ -175 +174 @@
           throw new NotSupportedException($"Symbol {node.Symbol.GetType().Name} not yet supported.");
         } else {
-          throw new NotSupportedException("Formatting of symbol: " + node.Symbol + " not supported for C# symbolic expression tree formatter.");
+          throw new NotSupportedException("Formatting of symbol: " + node.Symbol + " not supported for TSQL symbolic expression tree formatter.");
         }
       }
+    }
+
+    private void FormatIfThenElse(int level, ISymbolicExpressionTreeNode node, StringBuilder strBuilder) {
+      strBuilder.Append("CASE ISNULL((SELECT 1 WHERE");
+      FormatRecursively(level,node.GetSubtree(0), strBuilder);
+      strBuilder.AppendLine("),0)");
+      strBuilder.AppendIndented(level,"WHEN 1 THEN ");
+      FormatRecursively(level, node.GetSubtree(1), strBuilder);
+      strBuilder.AppendLine();
+      strBuilder.AppendIndented(level, "WHEN 0 THEN ");
+      FormatRecursively(level, node.GetSubtree(2), strBuilder);
+      strBuilder.AppendLine();
+      strBuilder.AppendIndented(level, "END");
+    }
+
+    private void FormatAverage(int level, ISymbolicExpressionTreeNode node, StringBuilder strBuilder) {
+      strBuilder.Append("(");
+      foreach (var child in node.Subtrees) {
+        FormatRecursively(level, child, strBuilder);
+        if (child != node.Subtrees.Last())
+          strBuilder.Append(" + ");
+      }
+      strBuilder.Append($") / {node.Subtrees.Count()}");
     }
 
@@ -204 +226 @@
         FormatRecursively(level++, child, strBuilder);
         if (child != node.Subtrees.Last())
-          strBuilder.Append( ", ");
+          strBuilder.Append(", ");
       }
       strBuilder.Append(")");
@@ -244 +266 @@
     private void FormatFactor(int level, FactorVariableTreeNode node, StringBuilder strBuilder) {
       strBuilder.AppendLine($"( ");
-      strBuilder.AppendLineIndented(level+1,$"CASE {VariableName2Identifier(node.VariableName)}");
+      strBuilder.AppendLineIndented(level + 1, $"CASE {VariableName2Identifier(node.VariableName)}");
       foreach (var name in node.Symbol.GetVariableValues(node.VariableName)) {
         strBuilder.AppendLineIndented(level + 2, $"WHEN '{name}' THEN {node.GetValue(name).ToString(CultureInfo.InvariantCulture)}");
       }
-      strBuilder.AppendLineIndented(level+1, "ELSE NULL END");
+      strBuilder.AppendLineIndented(level + 1, "ELSE NULL END");
       strBuilder.AppendIndented(level, ")");
     }