RubyGems - kumi - Versions diffs - 0.0.25 → 0.0.26 - Mend

kumi 0.0.25 → 0.0.26

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (177) hide show

checksums.yaml +4 -4
data/CHANGELOG.md +9 -0
data/README.md +70 -71
data/data/functions/agg/boolean.yaml +6 -2
data/data/functions/agg/numeric.yaml +32 -16
data/data/functions/agg/string.yaml +4 -3
data/data/functions/core/arithmetic.yaml +62 -14
data/data/functions/core/boolean.yaml +12 -6
data/data/functions/core/comparison.yaml +25 -13
data/data/functions/core/constructor.yaml +16 -8
data/data/functions/core/select.yaml +3 -1
data/data/functions/core/stencil.yaml +14 -5
data/data/functions/core/string.yaml +9 -4
data/data/kernels/ruby/agg/numeric.yaml +1 -1
data/docs/UNSAT_DETECTION.md +83 -0
data/golden/array_element/expected/nast.txt +1 -1
data/golden/array_element/expected/schema_ruby.rb +1 -1
data/golden/array_index/expected/nast.txt +7 -7
data/golden/array_index/expected/schema_ruby.rb +1 -1
data/golden/array_operations/expected/nast.txt +2 -2
data/golden/array_operations/expected/schema_ruby.rb +1 -1
data/golden/array_operations/expected/snast.txt +3 -3
data/golden/cascade_logic/expected/schema_ruby.rb +1 -1
data/golden/cascade_logic/expected/snast.txt +2 -2
data/golden/chained_fusion/expected/nast.txt +2 -2
data/golden/chained_fusion/expected/schema_ruby.rb +1 -1
data/golden/element_arrays/expected/nast.txt +2 -2
data/golden/element_arrays/expected/schema_ruby.rb +1 -1
data/golden/element_arrays/expected/snast.txt +1 -1
data/golden/empty_and_null_inputs/expected/nast.txt +3 -3
data/golden/empty_and_null_inputs/expected/schema_ruby.rb +1 -1
data/golden/function_overload/expected/ast.txt +29 -0
data/golden/function_overload/expected/input_plan.txt +4 -0
data/golden/function_overload/expected/lir_00_unoptimized.txt +18 -0
data/golden/function_overload/expected/lir_01_hoist_scalar_references.txt +18 -0
data/golden/function_overload/expected/lir_02_inlined.txt +20 -0
data/golden/function_overload/expected/lir_03_cse.txt +20 -0
data/golden/function_overload/expected/lir_04_1_loop_fusion.txt +20 -0
data/golden/function_overload/expected/lir_04_loop_invcm.txt +20 -0
data/golden/function_overload/expected/lir_06_const_prop.txt +20 -0
data/golden/function_overload/expected/nast.txt +22 -0
data/golden/function_overload/expected/schema_javascript.mjs +12 -0
data/golden/function_overload/expected/schema_ruby.rb +39 -0
data/golden/function_overload/expected/snast.txt +22 -0
data/golden/function_overload/input.json +8 -0
data/golden/function_overload/schema.kumi +19 -0
data/golden/game_of_life/expected/lir_00_unoptimized.txt +4 -4
data/golden/game_of_life/expected/lir_01_hoist_scalar_references.txt +4 -4
data/golden/game_of_life/expected/lir_02_inlined.txt +16 -16
data/golden/game_of_life/expected/lir_03_cse.txt +20 -16
data/golden/game_of_life/expected/lir_04_1_loop_fusion.txt +20 -16
data/golden/game_of_life/expected/lir_04_loop_invcm.txt +20 -16
data/golden/game_of_life/expected/lir_06_const_prop.txt +20 -16
data/golden/game_of_life/expected/nast.txt +4 -4
data/golden/game_of_life/expected/schema_javascript.mjs +4 -2
data/golden/game_of_life/expected/schema_ruby.rb +5 -3
data/golden/game_of_life/expected/snast.txt +10 -10
data/golden/hash_keys/expected/schema_ruby.rb +1 -1
data/golden/hash_value/expected/nast.txt +1 -1
data/golden/hash_value/expected/schema_ruby.rb +1 -1
data/golden/hash_value/expected/snast.txt +1 -1
data/golden/hierarchical_complex/expected/nast.txt +3 -3
data/golden/hierarchical_complex/expected/schema_ruby.rb +1 -1
data/golden/hierarchical_complex/expected/snast.txt +3 -3
data/golden/inline_rename_scope_leak/expected/nast.txt +3 -3
data/golden/inline_rename_scope_leak/expected/schema_ruby.rb +1 -1
data/golden/input_reference/expected/nast.txt +2 -2
data/golden/input_reference/expected/schema_ruby.rb +1 -1
data/golden/interleaved_fusion/expected/nast.txt +2 -2
data/golden/interleaved_fusion/expected/schema_ruby.rb +1 -1
data/golden/let_inline/expected/nast.txt +4 -4
data/golden/let_inline/expected/schema_ruby.rb +1 -1
data/golden/loop_fusion/expected/nast.txt +1 -1
data/golden/loop_fusion/expected/schema_ruby.rb +1 -1
data/golden/min_reduce_scope/expected/nast.txt +3 -3
data/golden/min_reduce_scope/expected/schema_ruby.rb +1 -1
data/golden/min_reduce_scope/expected/snast.txt +1 -1
data/golden/mixed_dimensions/expected/nast.txt +2 -2
data/golden/mixed_dimensions/expected/schema_ruby.rb +1 -1
data/golden/multirank_hoisting/expected/nast.txt +7 -7
data/golden/multirank_hoisting/expected/schema_ruby.rb +1 -1
data/golden/nested_hash/expected/nast.txt +1 -1
data/golden/nested_hash/expected/schema_ruby.rb +1 -1
data/golden/reduction_broadcast/expected/nast.txt +3 -3
data/golden/reduction_broadcast/expected/schema_ruby.rb +1 -1
data/golden/reduction_broadcast/expected/snast.txt +1 -1
data/golden/roll/expected/schema_ruby.rb +1 -1
data/golden/shift/expected/schema_ruby.rb +1 -1
data/golden/shift_2d/expected/schema_ruby.rb +1 -1
data/golden/simple_math/expected/lir_00_unoptimized.txt +1 -1
data/golden/simple_math/expected/lir_01_hoist_scalar_references.txt +1 -1
data/golden/simple_math/expected/lir_02_inlined.txt +1 -1
data/golden/simple_math/expected/lir_03_cse.txt +1 -1
data/golden/simple_math/expected/lir_04_1_loop_fusion.txt +1 -1
data/golden/simple_math/expected/lir_04_loop_invcm.txt +1 -1
data/golden/simple_math/expected/lir_06_const_prop.txt +1 -1
data/golden/simple_math/expected/nast.txt +5 -5
data/golden/simple_math/expected/schema_ruby.rb +1 -1
data/golden/simple_math/expected/snast.txt +2 -2
data/golden/streaming_basics/expected/nast.txt +8 -8
data/golden/streaming_basics/expected/schema_ruby.rb +1 -1
data/golden/streaming_basics/expected/snast.txt +1 -1
data/golden/tuples/expected/lir_00_unoptimized.txt +5 -5
data/golden/tuples/expected/lir_01_hoist_scalar_references.txt +5 -5
data/golden/tuples/expected/lir_02_inlined.txt +5 -5
data/golden/tuples/expected/lir_03_cse.txt +5 -5
data/golden/tuples/expected/lir_04_1_loop_fusion.txt +5 -5
data/golden/tuples/expected/lir_04_loop_invcm.txt +5 -5
data/golden/tuples/expected/lir_06_const_prop.txt +5 -5
data/golden/tuples/expected/nast.txt +4 -4
data/golden/tuples/expected/schema_ruby.rb +1 -1
data/golden/tuples/expected/snast.txt +6 -6
data/golden/tuples_and_arrays/expected/lir_00_unoptimized.txt +1 -1
data/golden/tuples_and_arrays/expected/lir_01_hoist_scalar_references.txt +1 -1
data/golden/tuples_and_arrays/expected/lir_02_inlined.txt +2 -2
data/golden/tuples_and_arrays/expected/lir_03_cse.txt +2 -2
data/golden/tuples_and_arrays/expected/lir_04_1_loop_fusion.txt +2 -2
data/golden/tuples_and_arrays/expected/lir_04_loop_invcm.txt +2 -2
data/golden/tuples_and_arrays/expected/lir_06_const_prop.txt +2 -2
data/golden/tuples_and_arrays/expected/nast.txt +3 -3
data/golden/tuples_and_arrays/expected/schema_ruby.rb +1 -1
data/golden/tuples_and_arrays/expected/snast.txt +2 -2
data/golden/us_tax_2024/expected/ast.txt +63 -670
data/golden/us_tax_2024/expected/input_plan.txt +8 -45
data/golden/us_tax_2024/expected/lir_00_unoptimized.txt +253 -863
data/golden/us_tax_2024/expected/lir_01_hoist_scalar_references.txt +253 -863
data/golden/us_tax_2024/expected/lir_02_inlined.txt +1215 -5139
data/golden/us_tax_2024/expected/lir_03_cse.txt +587 -2460
data/golden/us_tax_2024/expected/lir_04_1_loop_fusion.txt +632 -2480
data/golden/us_tax_2024/expected/lir_04_loop_invcm.txt +587 -2460
data/golden/us_tax_2024/expected/lir_06_const_prop.txt +587 -2460
data/golden/us_tax_2024/expected/nast.txt +123 -826
data/golden/us_tax_2024/expected/schema_javascript.mjs +127 -581
data/golden/us_tax_2024/expected/schema_ruby.rb +135 -610
data/golden/us_tax_2024/expected/snast.txt +155 -858
data/golden/us_tax_2024/expected.json +120 -1
data/golden/us_tax_2024/input.json +18 -9
data/golden/us_tax_2024/schema.kumi +48 -178
data/golden/with_constants/expected/lir_00_unoptimized.txt +1 -1
data/golden/with_constants/expected/lir_01_hoist_scalar_references.txt +1 -1
data/golden/with_constants/expected/lir_02_inlined.txt +1 -1
data/golden/with_constants/expected/lir_03_cse.txt +1 -1
data/golden/with_constants/expected/lir_04_1_loop_fusion.txt +1 -1
data/golden/with_constants/expected/lir_04_loop_invcm.txt +1 -1
data/golden/with_constants/expected/lir_06_const_prop.txt +1 -1
data/golden/with_constants/expected/nast.txt +2 -2
data/golden/with_constants/expected/schema_ruby.rb +1 -1
data/golden/with_constants/expected/snast.txt +2 -2
data/lib/kumi/analyzer.rb +12 -12
data/lib/kumi/core/analyzer/passes/formal_constraint_propagator.rb +236 -0
data/lib/kumi/core/analyzer/passes/input_collector.rb +22 -4
data/lib/kumi/core/analyzer/passes/nast_dimensional_analyzer_pass.rb +64 -18
data/lib/kumi/core/analyzer/passes/normalize_to_nast_pass.rb +9 -4
data/lib/kumi/core/analyzer/passes/snast_pass.rb +3 -1
data/lib/kumi/core/analyzer/passes/unsat_detector.rb +172 -198
data/lib/kumi/core/error_reporter.rb +36 -1
data/lib/kumi/core/errors.rb +33 -1
data/lib/kumi/core/functions/function_spec.rb +5 -4
data/lib/kumi/core/functions/loader.rb +17 -1
data/lib/kumi/core/functions/overload_resolver.rb +164 -0
data/lib/kumi/core/functions/type_error_reporter.rb +118 -0
data/lib/kumi/core/functions/type_rules.rb +155 -35
data/lib/kumi/core/types/inference.rb +29 -22
data/lib/kumi/core/types/normalizer.rb +29 -45
data/lib/kumi/core/types/validator.rb +16 -27
data/lib/kumi/core/types/value_objects.rb +116 -0
data/lib/kumi/core/types.rb +45 -37
data/lib/kumi/registry_v2/loader.rb +90 -0
data/lib/kumi/registry_v2.rb +18 -1
data/lib/kumi/version.rb +1 -1
metadata +21 -7
data/lib/kumi/core/analyzer/unsat_constant_evaluator.rb +0 -59
data/lib/kumi/core/atom_unsat_solver.rb +0 -396
data/lib/kumi/core/constraint_relationship_solver.rb +0 -641
data/lib/kumi/core/types/builder.rb +0 -23
data/lib/kumi/core/types/compatibility.rb +0 -96
data/lib/kumi/core/types/formatter.rb +0 -26

data/golden/us_tax_2024/expected/nast.txt CHANGED Viewed

@@ -1,95 +1,95 @@
 (NAST
   (VALUE state_tax
-    (Call :"core.mul"
+    (Call :multiply
     (InputRef [:income])
     (InputRef [:state_rate])
   )
   )
   (VALUE local_tax
-    (Call :"core.mul"
+    (Call :multiply
     (InputRef [:income])
     (InputRef [:local_rate])
   )
   )
-  (VALUE taxable_single
-    (Call :"agg.max"
+  (VALUE taxable
+    (Call :max
     (Tuple
-      (Call :"core.sub"
+      (Call :subtract
         (InputRef [:income])
-        (InputRef [:fed, :single, :std])
+        (InputRef [:statuses, :status, :std])
       )
       (Const 0)
     )
   )
   )
-  (VALUE lo_s
-    (InputRef [:fed, :single, :rates, :bracket, :lo])
+  (VALUE lo
+    (InputRef [:statuses, :status, :rates, :bracket, :lo])
   )
-  (VALUE hi_s
-    (InputRef [:fed, :single, :rates, :bracket, :hi])
+  (VALUE hi
+    (InputRef [:statuses, :status, :rates, :bracket, :hi])
   )
-  (VALUE rate_s
-    (InputRef [:fed, :single, :rates, :bracket, :rate])
+  (VALUE rate
+    (InputRef [:statuses, :status, :rates, :bracket, :rate])
   )
   (VALUE big_hi
     (Const 100000000000.0)
   )
-  (VALUE hi_s_eff
-    (Call :__select__
+  (VALUE hi_eff
+    (Call :select
     (Call :"core.eq"
-      (Ref hi_s)
+      (Ref hi)
       (Const -1)
     )
     (Ref big_hi)
-    (Ref hi_s)
+    (Ref hi)
   )
   )
-  (VALUE amt_s
-    (Call :"core.clamp"
-    (Call :"core.sub"
-      (Ref taxable_single)
-      (Ref lo_s)
+  (VALUE amt
+    (Call :clamp
+    (Call :subtract
+      (Ref taxable)
+      (Ref lo)
     )
     (Const 0)
-    (Call :"core.sub"
-      (Ref hi_s_eff)
-      (Ref lo_s)
+    (Call :subtract
+      (Ref hi_eff)
+      (Ref lo)
     )
   )
   )
-  (VALUE fed_tax_single
-    (Call :"agg.sum"
-    (Call :"core.mul"
-      (Ref amt_s)
-      (Ref rate_s)
+  (VALUE fed_tax
+    (Call :sum
+    (Call :multiply
+      (Ref amt)
+      (Ref rate)
     )
   )
   )
-  (VALUE in_br_s
-    (Call :"core.and"
+  (VALUE in_br
+    (Call :and
     (Call :"core.gte"
-      (Ref taxable_single)
-      (Ref lo_s)
+      (Ref taxable)
+      (Ref lo)
     )
     (Call :"core.lt"
-      (Ref taxable_single)
-      (Ref hi_s_eff)
+      (Ref taxable)
+      (Ref hi_eff)
     )
   )
   )
-  (VALUE fed_marginal_single
+  (VALUE fed_marg
     (Call :sum
     (Call :__select__
-      (Ref in_br_s)
-      (Ref rate_s)
+      (Ref in_br)
+      (Ref rate)
       (Const 0)
     )
   )
   )
-  (VALUE fed_eff_single
-    (Call :"core.div"
-    (Ref fed_tax_single)
-    (Call :"agg.max"
+  (VALUE fed_eff
+    (Call :divide
+    (Ref fed_tax)
+    (Call :max
       (Tuple
         (InputRef [:income])
         (Const 1.0)
@@ -103,9 +103,9 @@
   (VALUE ss_rate
     (Const 0.062)
   )
-  (VALUE ss_tax_s
-    (Call :"core.mul"
-    (Call :"agg.min"
+  (VALUE ss_tax
+    (Call :multiply
+    (Call :min
       (Tuple
         (InputRef [:income])
         (Ref ss_wage_base)
@@ -117,25 +117,22 @@
   (VALUE med_base_rate
     (Const 0.0145)
   )
-  (VALUE med_tax_s
-    (Call :"core.mul"
+  (VALUE med_tax
+    (Call :multiply
     (InputRef [:income])
     (Ref med_base_rate)
   )
   )
-  (VALUE addl_threshold_single
-    (Const 200000.0)
-  )
   (VALUE addl_med_rate
     (Const 0.009)
   )
-  (VALUE addl_med_tax_s
-    (Call :"core.mul"
-    (Call :"agg.max"
+  (VALUE addl_med_tax
+    (Call :multiply
+    (Call :max
       (Tuple
-        (Call :"core.sub"
+        (Call :subtract
           (InputRef [:income])
-          (Ref addl_threshold_single)
+          (InputRef [:statuses, :status, :addl_threshold])
         )
         (Const 0)
       )
@@ -143,19 +140,19 @@
     (Ref addl_med_rate)
   )
   )
-  (VALUE fica_tax_single
-    (Call :"core.add"
-    (Call :"core.add"
-      (Ref ss_tax_s)
-      (Ref med_tax_s)
+  (VALUE fica_tax
+    (Call :add
+    (Call :add
+      (Ref ss_tax)
+      (Ref med_tax)
     )
-    (Ref addl_med_tax_s)
+    (Ref addl_med_tax)
   )
   )
-  (VALUE fica_eff_single
-    (Call :"core.div"
-    (Ref fica_tax_single)
-    (Call :"agg.max"
+  (VALUE fica_eff
+    (Call :divide
+    (Ref fica_tax)
+    (Call :max
       (Tuple
         (InputRef [:income])
         (Const 1.0)
@@ -163,22 +160,22 @@
     )
   )
   )
-  (VALUE total_tax_single
-    (Call :"core.add"
-    (Call :"core.add"
-      (Call :"core.add"
-        (Ref fed_tax_single)
-        (Ref fica_tax_single)
+  (VALUE total_tax
+    (Call :add
+    (Call :add
+      (Call :add
+        (Ref fed_tax)
+        (Ref fica_tax)
       )
       (Ref state_tax)
     )
     (Ref local_tax)
   )
   )
-  (VALUE total_eff_single
-    (Call :"core.div"
-    (Ref total_tax_single)
-    (Call :"agg.max"
+  (VALUE total_eff
+    (Call :divide
+    (Ref total_tax)
+    (Call :max
       (Tuple
         (InputRef [:income])
         (Const 1.0)
@@ -186,791 +183,91 @@
     )
   )
   )
-  (VALUE after_tax_single
-    (Call :"core.sub"
+  (VALUE after_tax
+    (Call :subtract
     (InputRef [:income])
-    (Ref total_tax_single)
+    (Ref total_tax)
   )
   )
-  (VALUE take_home_single
-    (Call :"core.sub"
-    (Ref after_tax_single)
+  (VALUE take_home
+    (Call :subtract
+    (Ref after_tax)
     (InputRef [:retirement_contrib])
   )
   )
-  (VALUE taxable_mj
-    (Call :"agg.max"
-    (Tuple
-      (Call :"core.sub"
-        (InputRef [:income])
-        (InputRef [:fed, :married_joint, :std])
-      )
-      (Const 0)
-    )
-  )
-  )
-  (VALUE lo_mj
-    (InputRef [:fed, :married_joint, :rates, :bracket, :lo])
-  )
-  (VALUE hi_mj
-    (InputRef [:fed, :married_joint, :rates, :bracket, :hi])
-  )
-  (VALUE rate_mj
-    (InputRef [:fed, :married_joint, :rates, :bracket, :rate])
-  )
-  (VALUE hi_mj_eff
-    (Call :__select__
-    (Call :"core.eq"
-      (Ref hi_mj)
-      (Const -1)
-    )
-    (Ref big_hi)
-    (Ref hi_mj)
-  )
-  )
-  (VALUE amt_mj
-    (Call :"core.clamp"
-    (Call :"core.sub"
-      (Ref taxable_mj)
-      (Ref lo_mj)
-    )
-    (Const 0)
-    (Call :"core.sub"
-      (Ref hi_mj_eff)
-      (Ref lo_mj)
-    )
-  )
-  )
-  (VALUE fed_tax_mj
-    (Call :"agg.sum"
-    (Call :"core.mul"
-      (Ref amt_mj)
-      (Ref rate_mj)
-    )
-  )
-  )
-  (VALUE in_br_mj
-    (Call :"core.and"
-    (Call :"core.gte"
-      (Ref taxable_mj)
-      (Ref lo_mj)
-    )
-    (Call :"core.lt"
-      (Ref taxable_mj)
-      (Ref hi_mj_eff)
-    )
-  )
-  )
-  (VALUE fed_marginal_mj
-    (Call :sum
-    (Call :__select__
-      (Ref in_br_mj)
-      (Ref rate_mj)
-      (Const 0)
-    )
-  )
-  )
-  (VALUE fed_eff_mj
-    (Call :"core.div"
-    (Ref fed_tax_mj)
-    (Call :"agg.max"
-      (Tuple
-        (InputRef [:income])
-        (Const 1.0)
-      )
+  (VALUE summary
+    (Hash
+    (Pair filing_status
+      (InputRef [:statuses, :status, :name])
     )
-  )
-  )
-  (VALUE ss_tax_mj
-    (Ref ss_tax_s)
-  )
-  (VALUE med_tax_mj
-    (Ref med_tax_s)
-  )
-  (VALUE addl_threshold_mj
-    (Const 250000.0)
-  )
-  (VALUE addl_med_tax_mj
-    (Call :"core.mul"
-    (Call :"agg.max"
-      (Tuple
-        (Call :"core.sub"
-          (InputRef [:income])
-          (Ref addl_threshold_mj)
+    (Pair federal
+      (Hash
+        (Pair marginal
+          (Ref fed_marg)
         )
-        (Const 0)
-      )
-    )
-    (Ref addl_med_rate)
-  )
-  )
-  (VALUE fica_tax_mj
-    (Call :"core.add"
-    (Call :"core.add"
-      (Ref ss_tax_mj)
-      (Ref med_tax_mj)
-    )
-    (Ref addl_med_tax_mj)
-  )
-  )
-  (VALUE fica_eff_mj
-    (Call :"core.div"
-    (Ref fica_tax_mj)
-    (Call :"agg.max"
-      (Tuple
-        (InputRef [:income])
-        (Const 1.0)
-      )
-    )
-  )
-  )
-  (VALUE total_tax_mj
-    (Call :"core.add"
-    (Call :"core.add"
-      (Call :"core.add"
-        (Ref fed_tax_mj)
-        (Ref fica_tax_mj)
-      )
-      (Ref state_tax)
-    )
-    (Ref local_tax)
-  )
-  )
-  (VALUE total_eff_mj
-    (Call :"core.div"
-    (Ref total_tax_mj)
-    (Call :"agg.max"
-      (Tuple
-        (InputRef [:income])
-        (Const 1.0)
-      )
-    )
-  )
-  )
-  (VALUE after_tax_mj
-    (Call :"core.sub"
-    (InputRef [:income])
-    (Ref total_tax_mj)
-  )
-  )
-  (VALUE take_home_mj
-    (Call :"core.sub"
-    (Ref after_tax_mj)
-    (InputRef [:retirement_contrib])
-  )
-  )
-  (VALUE taxable_ms
-    (Call :"agg.max"
-    (Tuple
-      (Call :"core.sub"
-        (InputRef [:income])
-        (InputRef [:fed, :married_separate, :std])
-      )
-      (Const 0)
-    )
-  )
-  )
-  (VALUE lo_ms
-    (InputRef [:fed, :married_separate, :rates, :bracket, :lo])
-  )
-  (VALUE hi_ms
-    (InputRef [:fed, :married_separate, :rates, :bracket, :hi])
-  )
-  (VALUE rate_ms
-    (InputRef [:fed, :married_separate, :rates, :bracket, :rate])
-  )
-  (VALUE hi_ms_eff
-    (Call :__select__
-    (Call :"core.eq"
-      (Ref hi_ms)
-      (Const -1)
-    )
-    (Ref big_hi)
-    (Ref hi_ms)
-  )
-  )
-  (VALUE amt_ms
-    (Call :"core.clamp"
-    (Call :"core.sub"
-      (Ref taxable_ms)
-      (Ref lo_ms)
-    )
-    (Const 0)
-    (Call :"core.sub"
-      (Ref hi_ms_eff)
-      (Ref lo_ms)
-    )
-  )
-  )
-  (VALUE fed_tax_ms
-    (Call :"agg.sum"
-    (Call :"core.mul"
-      (Ref amt_ms)
-      (Ref rate_ms)
-    )
-  )
-  )
-  (VALUE in_br_ms
-    (Call :"core.and"
-    (Call :"core.gte"
-      (Ref taxable_ms)
-      (Ref lo_ms)
-    )
-    (Call :"core.lt"
-      (Ref taxable_ms)
-      (Ref hi_ms_eff)
-    )
-  )
-  )
-  (VALUE fed_marginal_ms
-    (Call :sum
-    (Call :__select__
-      (Ref in_br_ms)
-      (Ref rate_ms)
-      (Const 0)
-    )
-  )
-  )
-  (VALUE fed_eff_ms
-    (Call :"core.div"
-    (Ref fed_tax_ms)
-    (Call :"agg.max"
-      (Tuple
-        (InputRef [:income])
-        (Const 1.0)
-      )
-    )
-  )
-  )
-  (VALUE ss_tax_ms
-    (Ref ss_tax_s)
-  )
-  (VALUE med_tax_ms
-    (Ref med_tax_s)
-  )
-  (VALUE addl_threshold_ms
-    (Const 125000.0)
-  )
-  (VALUE addl_med_tax_ms
-    (Call :"core.mul"
-    (Call :"agg.max"
-      (Tuple
-        (Call :"core.sub"
-          (InputRef [:income])
-          (Ref addl_threshold_ms)
+        (Pair effective
+          (Ref fed_eff)
         )
-        (Const 0)
-      )
-    )
-    (Ref addl_med_rate)
-  )
-  )
-  (VALUE fica_tax_ms
-    (Call :"core.add"
-    (Call :"core.add"
-      (Ref ss_tax_ms)
-      (Ref med_tax_ms)
-    )
-    (Ref addl_med_tax_ms)
-  )
-  )
-  (VALUE fica_eff_ms
-    (Call :"core.div"
-    (Ref fica_tax_ms)
-    (Call :"agg.max"
-      (Tuple
-        (InputRef [:income])
-        (Const 1.0)
-      )
-    )
-  )
-  )
-  (VALUE total_tax_ms
-    (Call :"core.add"
-    (Call :"core.add"
-      (Call :"core.add"
-        (Ref fed_tax_ms)
-        (Ref fica_tax_ms)
-      )
-      (Ref state_tax)
-    )
-    (Ref local_tax)
-  )
-  )
-  (VALUE total_eff_ms
-    (Call :"core.div"
-    (Ref total_tax_ms)
-    (Call :"agg.max"
-      (Tuple
-        (InputRef [:income])
-        (Const 1.0)
-      )
-    )
-  )
-  )
-  (VALUE after_tax_ms
-    (Call :"core.sub"
-    (InputRef [:income])
-    (Ref total_tax_ms)
-  )
-  )
-  (VALUE take_home_ms
-    (Call :"core.sub"
-    (Ref after_tax_ms)
-    (InputRef [:retirement_contrib])
-  )
-  )
-  (VALUE taxable_hoh
-    (Call :"agg.max"
-    (Tuple
-      (Call :"core.sub"
-        (InputRef [:income])
-        (InputRef [:fed, :head_of_household, :std])
-      )
-      (Const 0)
-    )
-  )
-  )
-  (VALUE lo_h
-    (InputRef [:fed, :head_of_household, :rates, :bracket, :lo])
-  )
-  (VALUE hi_h
-    (InputRef [:fed, :head_of_household, :rates, :bracket, :hi])
-  )
-  (VALUE rate_h
-    (InputRef [:fed, :head_of_household, :rates, :bracket, :rate])
-  )
-  (VALUE hi_h_eff
-    (Call :__select__
-    (Call :"core.eq"
-      (Ref hi_h)
-      (Const -1)
-    )
-    (Ref big_hi)
-    (Ref hi_h)
-  )
-  )
-  (VALUE amt_h
-    (Call :"core.clamp"
-    (Call :"core.sub"
-      (Ref taxable_hoh)
-      (Ref lo_h)
-    )
-    (Const 0)
-    (Call :"core.sub"
-      (Ref hi_h_eff)
-      (Ref lo_h)
-    )
-  )
-  )
-  (VALUE fed_tax_hoh
-    (Call :"agg.sum"
-    (Call :"core.mul"
-      (Ref amt_h)
-      (Ref rate_h)
-    )
-  )
-  )
-  (VALUE in_br_h
-    (Call :"core.and"
-    (Call :"core.gte"
-      (Ref taxable_hoh)
-      (Ref lo_h)
-    )
-    (Call :"core.lt"
-      (Ref taxable_hoh)
-      (Ref hi_h_eff)
-    )
-  )
-  )
-  (VALUE fed_marginal_hoh
-    (Call :sum
-    (Call :__select__
-      (Ref in_br_h)
-      (Ref rate_h)
-      (Const 0)
-    )
-  )
-  )
-  (VALUE fed_eff_hoh
-    (Call :"core.div"
-    (Ref fed_tax_hoh)
-    (Call :"agg.max"
-      (Tuple
-        (InputRef [:income])
-        (Const 1.0)
-      )
-    )
-  )
-  )
-  (VALUE ss_tax_h
-    (Ref ss_tax_s)
-  )
-  (VALUE med_tax_h
-    (Ref med_tax_s)
-  )
-  (VALUE addl_threshold_hoh
-    (Const 200000.0)
-  )
-  (VALUE addl_med_tax_h
-    (Call :"core.mul"
-    (Call :"agg.max"
-      (Tuple
-        (Call :"core.sub"
-          (InputRef [:income])
-          (Ref addl_threshold_hoh)
+        (Pair tax
+          (Ref fed_tax)
         )
-        (Const 0)
-      )
-    )
-    (Ref addl_med_rate)
-  )
-  )
-  (VALUE fica_tax_hoh
-    (Call :"core.add"
-    (Call :"core.add"
-      (Ref ss_tax_h)
-      (Ref med_tax_h)
-    )
-    (Ref addl_med_tax_h)
-  )
-  )
-  (VALUE fica_eff_hoh
-    (Call :"core.div"
-    (Ref fica_tax_hoh)
-    (Call :"agg.max"
-      (Tuple
-        (InputRef [:income])
-        (Const 1.0)
-      )
-    )
-  )
-  )
-  (VALUE total_tax_hoh
-    (Call :"core.add"
-    (Call :"core.add"
-      (Call :"core.add"
-        (Ref fed_tax_hoh)
-        (Ref fica_tax_hoh)
-      )
-      (Ref state_tax)
-    )
-    (Ref local_tax)
-  )
-  )
-  (VALUE total_eff_hoh
-    (Call :"core.div"
-    (Ref total_tax_hoh)
-    (Call :"agg.max"
-      (Tuple
-        (InputRef [:income])
-        (Const 1.0)
       )
     )
-  )
-  )
-  (VALUE after_tax_hoh
-    (Call :"core.sub"
-    (InputRef [:income])
-    (Ref total_tax_hoh)
-  )
-  )
-  (VALUE take_home_hoh
-    (Call :"core.sub"
-    (Ref after_tax_hoh)
-    (InputRef [:retirement_contrib])
-  )
-  )
-  (VALUE summary
-    (Hash
-    (Pair single
+    (Pair fica
       (Hash
-        (Pair federal
-          (Hash
-            (Pair marginal
-              (Ref fed_marginal_single)
-            )
-            (Pair effective
-              (Ref fed_eff_single)
-            )
-            (Pair tax
-              (Ref fed_tax_single)
-            )
-          )
-        )
-        (Pair fica
-          (Hash
-            (Pair effective
-              (Ref fica_eff_single)
-            )
-            (Pair tax
-              (Ref fica_tax_single)
-            )
-          )
-        )
-        (Pair state
-          (Hash
-            (Pair marginal
-              (InputRef [:state_rate])
-            )
-            (Pair effective
-              (InputRef [:state_rate])
-            )
-            (Pair tax
-              (Ref state_tax)
-            )
-          )
-        )
-        (Pair local
-          (Hash
-            (Pair marginal
-              (InputRef [:local_rate])
-            )
-            (Pair effective
-              (InputRef [:local_rate])
-            )
-            (Pair tax
-              (Ref local_tax)
-            )
-          )
-        )
-        (Pair total
-          (Hash
-            (Pair effective
-              (Ref total_eff_single)
-            )
-            (Pair tax
-              (Ref total_tax_single)
-            )
-          )
+        (Pair effective
+          (Ref fica_eff)
         )
-        (Pair after_tax
-          (Ref after_tax_single)
-        )
-        (Pair retirement_contrib
-          (InputRef [:retirement_contrib])
-        )
-        (Pair take_home
-          (Ref take_home_single)
+        (Pair tax
+          (Ref fica_tax)
         )
       )
     )
-    (Pair married_joint
+    (Pair state
       (Hash
-        (Pair federal
-          (Hash
-            (Pair marginal
-              (Ref fed_marginal_mj)
-            )
-            (Pair effective
-              (Ref fed_eff_mj)
-            )
-            (Pair tax
-              (Ref fed_tax_mj)
-            )
-          )
-        )
-        (Pair fica
-          (Hash
-            (Pair effective
-              (Ref fica_eff_mj)
-            )
-            (Pair tax
-              (Ref fica_tax_mj)
-            )
-          )
-        )
-        (Pair state
-          (Hash
-            (Pair marginal
-              (InputRef [:state_rate])
-            )
-            (Pair effective
-              (InputRef [:state_rate])
-            )
-            (Pair tax
-              (Ref state_tax)
-            )
-          )
+        (Pair marginal
+          (InputRef [:state_rate])
         )
-        (Pair local
-          (Hash
-            (Pair marginal
-              (InputRef [:local_rate])
-            )
-            (Pair effective
-              (InputRef [:local_rate])
-            )
-            (Pair tax
-              (Ref local_tax)
-            )
-          )
+        (Pair effective
+          (InputRef [:state_rate])
         )
-        (Pair total
-          (Hash
-            (Pair effective
-              (Ref total_eff_mj)
-            )
-            (Pair tax
-              (Ref total_tax_mj)
-            )
-          )
-        )
-        (Pair after_tax
-          (Ref after_tax_mj)
-        )
-        (Pair retirement_contrib
-          (InputRef [:retirement_contrib])
-        )
-        (Pair take_home
-          (Ref take_home_mj)
+        (Pair tax
+          (Ref state_tax)
         )
       )
     )
-    (Pair married_separate
+    (Pair local
       (Hash
-        (Pair federal
-          (Hash
-            (Pair marginal
-              (Ref fed_marginal_ms)
-            )
-            (Pair effective
-              (Ref fed_eff_ms)
-            )
-            (Pair tax
-              (Ref fed_tax_ms)
-            )
-          )
-        )
-        (Pair fica
-          (Hash
-            (Pair effective
-              (Ref fica_eff_ms)
-            )
-            (Pair tax
-              (Ref fica_tax_ms)
-            )
-          )
-        )
-        (Pair state
-          (Hash
-            (Pair marginal
-              (InputRef [:state_rate])
-            )
-            (Pair effective
-              (InputRef [:state_rate])
-            )
-            (Pair tax
-              (Ref state_tax)
-            )
-          )
+        (Pair marginal
+          (InputRef [:local_rate])
         )
-        (Pair local
-          (Hash
-            (Pair marginal
-              (InputRef [:local_rate])
-            )
-            (Pair effective
-              (InputRef [:local_rate])
-            )
-            (Pair tax
-              (Ref local_tax)
-            )
-          )
+        (Pair effective
+          (InputRef [:local_rate])
         )
-        (Pair total
-          (Hash
-            (Pair effective
-              (Ref total_eff_ms)
-            )
-            (Pair tax
-              (Ref total_tax_ms)
-            )
-          )
-        )
-        (Pair after_tax
-          (Ref after_tax_ms)
-        )
-        (Pair retirement_contrib
-          (InputRef [:retirement_contrib])
-        )
-        (Pair take_home
-          (Ref take_home_ms)
+        (Pair tax
+          (Ref local_tax)
         )
       )
     )
-    (Pair head_of_household
+    (Pair total
       (Hash
-        (Pair federal
-          (Hash
-            (Pair marginal
-              (Ref fed_marginal_hoh)
-            )
-            (Pair effective
-              (Ref fed_eff_hoh)
-            )
-            (Pair tax
-              (Ref fed_tax_hoh)
-            )
-          )
-        )
-        (Pair fica
-          (Hash
-            (Pair effective
-              (Ref fica_eff_hoh)
-            )
-            (Pair tax
-              (Ref fica_tax_hoh)
-            )
-          )
+        (Pair effective
+          (Ref total_eff)
         )
-        (Pair state
-          (Hash
-            (Pair marginal
-              (InputRef [:state_rate])
-            )
-            (Pair effective
-              (InputRef [:state_rate])
-            )
-            (Pair tax
-              (Ref state_tax)
-            )
-          )
-        )
-        (Pair local
-          (Hash
-            (Pair marginal
-              (InputRef [:local_rate])
-            )
-            (Pair effective
-              (InputRef [:local_rate])
-            )
-            (Pair tax
-              (Ref local_tax)
-            )
-          )
-        )
-        (Pair total
-          (Hash
-            (Pair effective
-              (Ref total_eff_hoh)
-            )
-            (Pair tax
-              (Ref total_tax_hoh)
-            )
-          )
-        )
-        (Pair after_tax
-          (Ref after_tax_hoh)
-        )
-        (Pair retirement_contrib
-          (InputRef [:retirement_contrib])
-        )
-        (Pair take_home
-          (Ref take_home_hoh)
+        (Pair tax
+          (Ref total_tax)
         )
       )
     )
+    (Pair after_tax
+      (Ref after_tax)
+    )
+    (Pair retirement_contrib
+      (InputRef [:retirement_contrib])
+    )
+    (Pair take_home
+      (Ref take_home)
+    )
   )
   )
 )