feat: add sheet setting to disable symbolic expression simplification

Disabling symbolic expression simplification can speed up calculations for sheets with complex expressions. Does not impact numeric results.

public/dimensional_analysis.py

@@ -440,6 +440,7 @@ class StatementsAndSystems(TypedDict):
     statements: list[InputStatement]
     systemDefinitions: list[SystemDefinition]
     customBaseUnits: NotRequired[CustomBaseUnits]
+    simplifySymbolicExpressions: bool
 
 def is_code_function_query_statement(statement: InputAndSystemStatement) -> TypeGuard[CodeFunctionQueryStatement]:
     return statement.get("isCodeFunctionQuery", False)
@@ -1680,19 +1681,27 @@ def subs_wrapper(expression: Expr, subs: dict[str, str] | dict[str, Expr | float
         return cast(Expr, expression.xreplace(subs))
 
 
-def get_evaluated_expression(expression: Expr, parameter_subs: dict[Symbol, Expr]) -> tuple[ExprWithAssumptions, str | list[list[str]]]:
+def get_evaluated_expression(expression: Expr,
+                             parameter_subs: dict[Symbol, Expr],
+                             simplify_symbolic_expressions: bool) -> tuple[ExprWithAssumptions, str | list[list[str]]]:
     expression = cast(Expr, expression.xreplace(parameter_subs))
     expression = replace_placeholder_funcs(expression, "sympy_func")
     expression = cast(Expr, expression.doit())
     if not is_matrix(expression):
-        symbolic_expression = custom_latex(cancel(expression))
+        if simplify_symbolic_expressions:
+            symbolic_expression = custom_latex(cancel(expression))
+        else:
+            symbolic_expression = custom_latex(expression)
     else:
         symbolic_expression = []
         for i in range(expression.rows):
             row = []
             symbolic_expression.append(row)
             for j in range(expression.cols):
-                row.append(custom_latex(cancel(expression[i,j])))
+                if simplify_symbolic_expressions:
+                    row.append(custom_latex(cancel(expression[i,j])))
+                else:
+                    row.append(custom_latex(cast(Expr, expression[i,j])))
 
     evaluated_expression = cast(ExprWithAssumptions, expression.evalf(PRECISION))
     return evaluated_expression, symbolic_expression
@@ -1762,7 +1771,8 @@ def get_hashable_matrix_units(matrix_result: MatrixResult) -> tuple[tuple[str, .
     return tuple(rows)
 
 def evaluate_statements(statements: list[InputAndSystemStatement],
-                        custom_base_units: CustomBaseUnits | None) -> tuple[list[Result | FiniteImagResult | list[PlotResult] | MatrixResult], dict[int,bool]]:
+                        custom_base_units: CustomBaseUnits | None,
+                        simplify_symbolic_expressions: bool) -> tuple[list[Result | FiniteImagResult | list[PlotResult] | MatrixResult], dict[int,bool]]:
     num_statements = len(statements)
 
     if num_statements == 0:
@@ -1986,7 +1996,7 @@ def evaluate_statements(statements: list[InputAndSystemStatement],
         else:
             expression = cast(Expr, item["expression"].doit())
 
-            evaluated_expression, symbolic_expression = get_evaluated_expression(expression, parameter_subs)
+            evaluated_expression, symbolic_expression = get_evaluated_expression(expression, parameter_subs, simplify_symbolic_expressions)
             dimensional_analysis_expression, dim_sub_error = get_dimensional_analysis_expression(dimensional_analysis_subs, expression)
 
             if not is_matrix(evaluated_expression):
@@ -2092,13 +2102,15 @@ def evaluate_statements(statements: list[InputAndSystemStatement],
     return combine_plot_results(results_with_ranges[:num_statements], statement_plot_info), numerical_system_cell_unit_errors
 
 
-def get_query_values(statements: list[InputAndSystemStatement], custom_base_units: CustomBaseUnits | None):
+def get_query_values(statements: list[InputAndSystemStatement],
+                     custom_base_units: CustomBaseUnits | None,
+                     simplify_symbolic_expressions: bool):
     error: None | str = None
 
     results: list[Result | FiniteImagResult | list[PlotResult] | MatrixResult] = []
     numerical_system_cell_errors: dict[int, bool] = {}
     try:
-        results, numerical_system_cell_errors = evaluate_statements(statements, custom_base_units)
+        results, numerical_system_cell_errors = evaluate_statements(statements, custom_base_units, simplify_symbolic_expressions)
     except DuplicateAssignment as e:
         error = f"Duplicate assignment of variable {e}"
     except ReferenceCycle as e:
@@ -2190,6 +2202,7 @@ def solve_sheet(statements_and_systems):
     statements: list[InputAndSystemStatement] = cast(list[InputAndSystemStatement], statements_and_systems["statements"])
     system_definitions = statements_and_systems["systemDefinitions"]
     custom_base_units = statements_and_systems.get("customBaseUnits", None)
+    simplify_symbolic_expressions = statements_and_systems["simplifySymbolicExpressions"]
 
     system_results: list[SystemResult] = []
     equation_to_system_cell_map: dict[int,int] = {}
@@ -2234,7 +2247,7 @@ def solve_sheet(statements_and_systems):
     error: str | None
     results: list[Result | FiniteImagResult | list[PlotResult] | MatrixResult]
     numerical_system_cell_errors: dict[int, bool]
-    error, results, numerical_system_cell_errors = get_query_values(statements, custom_base_units)
+    error, results, numerical_system_cell_errors = get_query_values(statements, custom_base_units, simplify_symbolic_expressions)
 
     # If there was a numerical solve, check to make sure there were not unit mismatches
     # between the lhs and rhs of each equality in the system

src/App.svelte

@@ -50,6 +50,7 @@
     SkipToContent,
     HeaderUtilities,
     HeaderGlobalAction,
+    Checkbox,
     Content,
     SideNav,
     SideNavMenuItem,
@@ -827,7 +828,12 @@
 
     statements.push(...endStatements);
 
-    return {statements: statements, systemDefinitions: systemDefinitions, customBaseUnits: $config.customBaseUnits};
+    return {
+      statements: statements,
+      systemDefinitions: systemDefinitions, 
+      customBaseUnits: $config.customBaseUnits,
+      simplifySymbolicExpressions: $config.simplifySymbolicExpressions
+    };
   }
 
   function checkParsingErrors() {
@@ -1087,9 +1093,8 @@ Please include a link to this sheet in the email to assist in debugging the prob
       // old documents in database will not have the insertedSheets property or a config property
       $insertedSheets = sheet.insertedSheets ?? [];
       $config = sheet.config ?? getDefaultConfig();
-      if (!$config.customBaseUnits) {
-        $config.customBaseUnits = getDefaultBaseUnits();
-      }
+      $config.customBaseUnits = $config.customBaseUnits ?? getDefaultBaseUnits(); // customBaseUnits may not exist
+      $config.simplifySymbolicExpressions = $config.simplifySymbolicExpressions ?? true; // simplifySymboicExpressions may not exist
     
       if (!$history.map(item => item.hash !== "file" ? getSheetHash(new URL(item.url)) : "").includes(getSheetHash(window.location))) {
         $history = requestHistory;
@@ -2630,6 +2635,11 @@ Please include a link to this sheet in the email to assist in debugging the prob
             <Tab label="Default Units" />
             <svelte:fragment slot="content">
               <TabContent>
+                <Checkbox 
+                  labelText="Automatically Simplify Symbolic Expressions (unchecking will speed up sheet updates)"
+                  bind:checked={$config.simplifySymbolicExpressions}
+                  on:change={() => $mathCellChanged = true}
+                />
                 <MathCellConfigDialog
                   bind:this={mathCellConfigDialog}
                   bind:mathCellConfig={$config.mathCellConfig}

src/sheet/Sheet.ts

@@ -10,7 +10,7 @@ export type Sheet = {
   system_results: (SystemResult)[];
   nextId: number;
   sheetId: string;
-  insertedSheets?: InsertedSheet[]; // early sheets did not have an insertedSheets property  
+  insertedSheets?: InsertedSheet[]; // early sheets did not have this property  
 };
 
 export type InsertedSheet = {
@@ -21,7 +21,8 @@ export type InsertedSheet = {
 
 export type Config = {
   mathCellConfig: MathCellConfig;
-  customBaseUnits?: CustomBaseUnits; // same early sheets won't have this property
+  customBaseUnits?: CustomBaseUnits; // some early sheets won't have this property
+  simplifySymbolicExpressions?: boolean; // some early sheets won't have this property
 };
 
 type Notation = "auto" | "fixed" | "exponential" | "engineering";
@@ -41,7 +42,8 @@ type FormatOptions = {
 export function getDefaultConfig(): Config {
   return {
     mathCellConfig: getDefaultMathCellConfig(),
-    customBaseUnits: getDefaultBaseUnits()
+    customBaseUnits: getDefaultBaseUnits(),
+    simplifySymbolicExpressions: true
   };
 }
 

src/types.ts

@@ -41,6 +41,7 @@ export type StatementsAndSystems = {
   statements: Statement[];
   systemDefinitions: SystemDefinition[];
   customBaseUnits?: CustomBaseUnits;
+  simplifySymbolicExpressions: boolean;
 }