kumi 0.0.18 → 0.0.20

data/golden/array_index/expected/schema_ruby.rb

@@ -0,0 +1,75 @@
+# Autogenerated by Kumi Codegen
+module Kumi::Compiled::KUMI_6638f46fadb918fd0d9801db1cc74a8644071d3e15e8db0cebb4c1b7699ed811
+  def self.from(input_data = nil)
+    instance = Object.new
+    instance.extend(self)
+    instance.instance_variable_set(:@input, input_data)
+    instance
+  end
+
+  def self.__kumi_executable__
+    instance = Object.new
+    instance.extend(self)
+    instance
+  end
+
+  def update(input_data)
+    @input = @input.merge(input_data)
+    self
+  end
+
+  def [](name)
+    case name
+    when :box then _box
+    when :col_major then _col_major
+    when :sum_ij then _sum_ij
+    else raise KeyError, "Unknown declaration"
+    end
+  end
+
+  def _box(input = @input)
+    out = []
+    t5 = input["x"] || input[:x]
+    t5.each_with_index do |x_el_6, x_i_7|
+      out_1 = []
+      t29 = x_el_6.length
+      t11 = x_i_7 * t29
+      x_el_6.each_with_index do |y_el_8, y_i_9|
+        t12 = t11 + y_i_9
+        out_1 << t12
+      end
+      out << out_1
+    end
+    out
+  end
+
+  def _col_major(input = @input)
+    out = []
+    t13 = input["x"] || input[:x]
+    t19 = t13.length
+    t13.each_with_index do |x_el_14, x_i_15|
+      out_1 = []
+      x_el_14.each_with_index do |y_el_16, y_i_17|
+        t20 = y_i_17 * t19
+        t21 = t20 + x_i_15
+        out_1 << t21
+      end
+      out << out_1
+    end
+    out
+  end
+
+  def _sum_ij(input = @input)
+    out = []
+    t22 = input["x"] || input[:x]
+    t22.each_with_index do |x_el_23, x_i_24|
+      out_1 = []
+      x_el_23.each_with_index do |y_el_25, y_i_26|
+        t27 = x_i_24 + y_i_26
+        out_1 << t27
+      end
+      out << out_1
+    end
+    out
+  end
+end

data/golden/array_index/expected/snast.txt

@@ -0,0 +1,33 @@
+(SNAST
+  (VALUE W
+    (Call :core.array_size
+      (InputRef x.y key_chain=[]) :: [x] -> array
+    ) :: [x] -> integer
+  ) :: [x] -> integer
+  (VALUE box
+    (Call :core.add
+      (Call :core.mul
+        (IndexRef i input=x.__index(i)) :: [x] -> integer
+        (Ref W) :: [x] -> integer
+      ) :: [x] -> integer
+      (IndexRef j input=x.y.__index(j)) :: [x, y] -> integer
+    ) :: [x, y] -> integer
+  ) :: [x, y] -> integer
+  (VALUE col_major
+    (Call :core.add
+      (Call :core.mul
+        (IndexRef j input=x.y.__index(j)) :: [x, y] -> integer
+        (Call :core.array_size
+          (InputRef x key_chain=[]) :: [] -> array
+        ) :: [] -> integer
+      ) :: [x, y] -> integer
+      (IndexRef i input=x.__index(i)) :: [x] -> integer
+    ) :: [x, y] -> integer
+  ) :: [x, y] -> integer
+  (VALUE sum_ij
+    (Call :core.add
+      (IndexRef i input=x.__index(i)) :: [x] -> integer
+      (IndexRef j input=x.y.__index(j)) :: [x, y] -> integer
+    ) :: [x, y] -> integer
+  ) :: [x, y] -> integer
+)

data/golden/array_index/expected.json

@@ -0,0 +1,5 @@
+{
+  "box":       [[0,1,2,3,4],[5,6,7,8,9]],
+  "col_major": [[0,2,4,6,8],[1,3,5,7,9]],
+  "sum_ij":    [[0,1,2,3,4],[1,2,3,4,5]]
+}

data/golden/array_index/input.json

@@ -0,0 +1 @@
+{ "x": [ [0,0,0,0,0], [0,0,0,0,0] ] }

data/golden/array_index/schema.kumi

@@ -0,0 +1,19 @@
+schema do
+  input do
+    array :x, index: :i do
+      array :y, index: :j do
+        integer :_ # placeholder; values unused
+      end
+    end
+  end
+
+  # width per row
+  let   :W, fn(:array_size, input.x.y)
+
+  # row-major numbering: i*W + j
+  value :box, (index(:i) * W) + index(:j)
+
+  # variants:
+  value :col_major, (index(:j) * fn(:array_size, input.x)) + index(:i)
+  value :sum_ij,    index(:i) + index(:j)
+end

data/golden/array_operations/expected/ast.txt

@@ -0,0 +1,50 @@
+(Root
+  inputs: [
+    (InputDeclaration :items :array
+      [
+        (InputDeclaration :item :hash
+          [
+            (InputDeclaration :price :float)
+            (InputDeclaration :quantity :integer)
+            (InputDeclaration :category :string)
+          ]
+        )
+      ]
+    )
+    (InputDeclaration :tax_rate :float)
+  ]
+  values: [
+    (ValueDeclaration :subtotals
+      (CallExpression :multiply
+        (InputElementReference items.item.price)
+        (InputElementReference items.item.quantity)
+      )
+    )
+    (ValueDeclaration :discounted_price
+      (CallExpression :multiply
+        (InputElementReference items.item.price)
+        (Literal 0.9)
+      )
+    )
+    (ValueDeclaration :is_valid_quantity
+      (CallExpression :>
+        (InputElementReference items.item.quantity)
+        (Literal 0)
+      )
+    )
+  ]
+  traits: [
+    (TraitDeclaration :expensive_items
+      (CallExpression :>
+        (InputElementReference items.item.price)
+        (Literal 100.0)
+      )
+    )
+    (TraitDeclaration :electronics
+      (CallExpression :==
+        (InputElementReference items.item.category)
+        (Literal "electronics")
+      )
+    )
+  ]
+)

data/golden/array_operations/expected/input_plan.txt

@@ -0,0 +1,10 @@
+items: array (array) access_mode=element
+  └─> item: element_access
+  item: hash (hash)
+    └─> price: property_access
+    └─> quantity: property_access
+    └─> category: property_access
+    price: float
+    quantity: integer
+    category: string
+tax_rate: float

data/golden/array_operations/expected/lir_00_unoptimized.txt

@@ -0,0 +1,47 @@
+(LIR
+  (Declaration subtotals
+    %t1 = load_input "items" :: array
+    loop items id=L1 in %t1 as el=%items_el_2, idx=%items_i_3
+      %t4 = load_field items_el_2["price"] :: float
+      %t5 = load_field items_el_2["quantity"] :: integer
+      %t6 = call core.mul(%t4, %t5) :: float
+      yield %t6
+    end_loop
+  )
+  (Declaration discounted_price
+    %t7 = load_input "items" :: array
+    loop items id=L2 in %t7 as el=%items_el_8, idx=%items_i_9
+      %t10 = load_field items_el_8["price"] :: float
+      %t11 = const 0.9 :: float
+      %t12 = call core.mul(%t10, %t11) :: float
+      yield %t12
+    end_loop
+  )
+  (Declaration is_valid_quantity
+    %t13 = load_input "items" :: array
+    loop items id=L3 in %t13 as el=%items_el_14, idx=%items_i_15
+      %t16 = load_field items_el_14["quantity"] :: integer
+      %t17 = const 0 :: integer
+      %t18 = call core.gt(%t16, %t17) :: boolean
+      yield %t18
+    end_loop
+  )
+  (Declaration expensive_items
+    %t19 = load_input "items" :: array
+    loop items id=L4 in %t19 as el=%items_el_20, idx=%items_i_21
+      %t22 = load_field items_el_20["price"] :: float
+      %t23 = const 100.0 :: float
+      %t24 = call core.gt(%t22, %t23) :: boolean
+      yield %t24
+    end_loop
+  )
+  (Declaration electronics
+    %t25 = load_input "items" :: array
+    loop items id=L5 in %t25 as el=%items_el_26, idx=%items_i_27
+      %t28 = load_field items_el_26["category"] :: string
+      %t29 = const "electronics" :: string
+      %t30 = call core.eq(%t28, %t29) :: boolean
+      yield %t30
+    end_loop
+  )
+)

data/golden/array_operations/expected/lir_01_hoist_scalar_references.txt

@@ -0,0 +1,47 @@
+(LIR
+  (Declaration subtotals
+    %t1 = load_input "items" :: array
+    loop items id=L1 in %t1 as el=%items_el_2, idx=%items_i_3
+      %t4 = load_field items_el_2["price"] :: float
+      %t5 = load_field items_el_2["quantity"] :: integer
+      %t6 = call core.mul(%t4, %t5) :: float
+      yield %t6
+    end_loop
+  )
+  (Declaration discounted_price
+    %t7 = load_input "items" :: array
+    loop items id=L2 in %t7 as el=%items_el_8, idx=%items_i_9
+      %t10 = load_field items_el_8["price"] :: float
+      %t11 = const 0.9 :: float
+      %t12 = call core.mul(%t10, %t11) :: float
+      yield %t12
+    end_loop
+  )
+  (Declaration is_valid_quantity
+    %t13 = load_input "items" :: array
+    loop items id=L3 in %t13 as el=%items_el_14, idx=%items_i_15
+      %t16 = load_field items_el_14["quantity"] :: integer
+      %t17 = const 0 :: integer
+      %t18 = call core.gt(%t16, %t17) :: boolean
+      yield %t18
+    end_loop
+  )
+  (Declaration expensive_items
+    %t19 = load_input "items" :: array
+    loop items id=L4 in %t19 as el=%items_el_20, idx=%items_i_21
+      %t22 = load_field items_el_20["price"] :: float
+      %t23 = const 100.0 :: float
+      %t24 = call core.gt(%t22, %t23) :: boolean
+      yield %t24
+    end_loop
+  )
+  (Declaration electronics
+    %t25 = load_input "items" :: array
+    loop items id=L5 in %t25 as el=%items_el_26, idx=%items_i_27
+      %t28 = load_field items_el_26["category"] :: string
+      %t29 = const "electronics" :: string
+      %t30 = call core.eq(%t28, %t29) :: boolean
+      yield %t30
+    end_loop
+  )
+)

data/golden/array_operations/expected/lir_02_inlined.txt

@@ -0,0 +1,47 @@
+(LIR
+  (Declaration subtotals
+    %t1 = load_input "items" :: array
+    loop items id=L1 in %t1 as el=%items_el_2, idx=%items_i_3
+      %t4 = load_field items_el_2["price"] :: float
+      %t5 = load_field items_el_2["quantity"] :: integer
+      %t6 = call core.mul(%t4, %t5) :: float
+      yield %t6
+    end_loop
+  )
+  (Declaration discounted_price
+    %t7 = load_input "items" :: array
+    loop items id=L2 in %t7 as el=%items_el_8, idx=%items_i_9
+      %t10 = load_field items_el_8["price"] :: float
+      %t11 = const 0.9 :: float
+      %t12 = call core.mul(%t10, %t11) :: float
+      yield %t12
+    end_loop
+  )
+  (Declaration is_valid_quantity
+    %t13 = load_input "items" :: array
+    loop items id=L3 in %t13 as el=%items_el_14, idx=%items_i_15
+      %t16 = load_field items_el_14["quantity"] :: integer
+      %t17 = const 0 :: integer
+      %t18 = call core.gt(%t16, %t17) :: boolean
+      yield %t18
+    end_loop
+  )
+  (Declaration expensive_items
+    %t19 = load_input "items" :: array
+    loop items id=L4 in %t19 as el=%items_el_20, idx=%items_i_21
+      %t22 = load_field items_el_20["price"] :: float
+      %t23 = const 100.0 :: float
+      %t24 = call core.gt(%t22, %t23) :: boolean
+      yield %t24
+    end_loop
+  )
+  (Declaration electronics
+    %t25 = load_input "items" :: array
+    loop items id=L5 in %t25 as el=%items_el_26, idx=%items_i_27
+      %t28 = load_field items_el_26["category"] :: string
+      %t29 = const "electronics" :: string
+      %t30 = call core.eq(%t28, %t29) :: boolean
+      yield %t30
+    end_loop
+  )
+)

data/golden/array_operations/expected/lir_03_cse.txt

@@ -0,0 +1,47 @@
+(LIR
+  (Declaration subtotals
+    %t1 = load_input "items" :: array
+    loop items id=L1 in %t1 as el=%items_el_2, idx=%items_i_3
+      %t4 = load_field items_el_2["price"] :: float
+      %t5 = load_field items_el_2["quantity"] :: integer
+      %t6 = call core.mul(%t4, %t5) :: float
+      yield %t6
+    end_loop
+  )
+  (Declaration discounted_price
+    %t7 = load_input "items" :: array
+    loop items id=L2 in %t7 as el=%items_el_8, idx=%items_i_9
+      %t10 = load_field items_el_8["price"] :: float
+      %t11 = const 0.9 :: float
+      %t12 = call core.mul(%t10, %t11) :: float
+      yield %t12
+    end_loop
+  )
+  (Declaration is_valid_quantity
+    %t13 = load_input "items" :: array
+    loop items id=L3 in %t13 as el=%items_el_14, idx=%items_i_15
+      %t16 = load_field items_el_14["quantity"] :: integer
+      %t17 = const 0 :: integer
+      %t18 = call core.gt(%t16, %t17) :: boolean
+      yield %t18
+    end_loop
+  )
+  (Declaration expensive_items
+    %t19 = load_input "items" :: array
+    loop items id=L4 in %t19 as el=%items_el_20, idx=%items_i_21
+      %t22 = load_field items_el_20["price"] :: float
+      %t23 = const 100.0 :: float
+      %t24 = call core.gt(%t22, %t23) :: boolean
+      yield %t24
+    end_loop
+  )
+  (Declaration electronics
+    %t25 = load_input "items" :: array
+    loop items id=L5 in %t25 as el=%items_el_26, idx=%items_i_27
+      %t28 = load_field items_el_26["category"] :: string
+      %t29 = const "electronics" :: string
+      %t30 = call core.eq(%t28, %t29) :: boolean
+      yield %t30
+    end_loop
+  )
+)

data/golden/array_operations/expected/lir_04_1_loop_fusion.txt

@@ -0,0 +1,47 @@
+(LIR
+  (Declaration subtotals
+    %t1 = load_input "items" :: array
+    loop items id=L1 in %t1 as el=%items_el_2, idx=%items_i_3
+      %t4 = load_field items_el_2["price"] :: float
+      %t5 = load_field items_el_2["quantity"] :: integer
+      %t6 = call core.mul(%t4, %t5) :: float
+      yield %t6
+    end_loop
+  )
+  (Declaration discounted_price
+    %t7 = load_input "items" :: array
+    loop items id=L2 in %t7 as el=%items_el_8, idx=%items_i_9
+      %t10 = load_field items_el_8["price"] :: float
+      %t11 = const 0.9 :: float
+      %t12 = call core.mul(%t10, %t11) :: float
+      yield %t12
+    end_loop
+  )
+  (Declaration is_valid_quantity
+    %t13 = load_input "items" :: array
+    loop items id=L3 in %t13 as el=%items_el_14, idx=%items_i_15
+      %t16 = load_field items_el_14["quantity"] :: integer
+      %t17 = const 0 :: integer
+      %t18 = call core.gt(%t16, %t17) :: boolean
+      yield %t18
+    end_loop
+  )
+  (Declaration expensive_items
+    %t19 = load_input "items" :: array
+    loop items id=L4 in %t19 as el=%items_el_20, idx=%items_i_21
+      %t22 = load_field items_el_20["price"] :: float
+      %t23 = const 100.0 :: float
+      %t24 = call core.gt(%t22, %t23) :: boolean
+      yield %t24
+    end_loop
+  )
+  (Declaration electronics
+    %t25 = load_input "items" :: array
+    loop items id=L5 in %t25 as el=%items_el_26, idx=%items_i_27
+      %t28 = load_field items_el_26["category"] :: string
+      %t29 = const "electronics" :: string
+      %t30 = call core.eq(%t28, %t29) :: boolean
+      yield %t30
+    end_loop
+  )
+)

data/golden/array_operations/expected/lir_04_loop_invcm.txt

@@ -0,0 +1,47 @@
+(LIR
+  (Declaration subtotals
+    %t1 = load_input "items" :: array
+    loop items id=L1 in %t1 as el=%items_el_2, idx=%items_i_3
+      %t4 = load_field items_el_2["price"] :: float
+      %t5 = load_field items_el_2["quantity"] :: integer
+      %t6 = call core.mul(%t4, %t5) :: float
+      yield %t6
+    end_loop
+  )
+  (Declaration discounted_price
+    %t7 = load_input "items" :: array
+    %t11 = const 0.9 :: float
+    loop items id=L2 in %t7 as el=%items_el_8, idx=%items_i_9
+      %t10 = load_field items_el_8["price"] :: float
+      %t12 = call core.mul(%t10, %t11) :: float
+      yield %t12
+    end_loop
+  )
+  (Declaration is_valid_quantity
+    %t13 = load_input "items" :: array
+    %t17 = const 0 :: integer
+    loop items id=L3 in %t13 as el=%items_el_14, idx=%items_i_15
+      %t16 = load_field items_el_14["quantity"] :: integer
+      %t18 = call core.gt(%t16, %t17) :: boolean
+      yield %t18
+    end_loop
+  )
+  (Declaration expensive_items
+    %t19 = load_input "items" :: array
+    %t23 = const 100.0 :: float
+    loop items id=L4 in %t19 as el=%items_el_20, idx=%items_i_21
+      %t22 = load_field items_el_20["price"] :: float
+      %t24 = call core.gt(%t22, %t23) :: boolean
+      yield %t24
+    end_loop
+  )
+  (Declaration electronics
+    %t25 = load_input "items" :: array
+    %t29 = const "electronics" :: string
+    loop items id=L5 in %t25 as el=%items_el_26, idx=%items_i_27
+      %t28 = load_field items_el_26["category"] :: string
+      %t30 = call core.eq(%t28, %t29) :: boolean
+      yield %t30
+    end_loop
+  )
+)

data/golden/array_operations/expected/lir_06_const_prop.txt

@@ -0,0 +1,47 @@
+(LIR
+  (Declaration subtotals
+    %t1 = load_input "items" :: array
+    loop items id=L1 in %t1 as el=%items_el_2, idx=%items_i_3
+      %t4 = load_field items_el_2["price"] :: float
+      %t5 = load_field items_el_2["quantity"] :: integer
+      %t6 = call core.mul(%t4, %t5) :: float
+      yield %t6
+    end_loop
+  )
+  (Declaration discounted_price
+    %t7 = load_input "items" :: array
+    %t11 = const 0.9 :: float
+    loop items id=L2 in %t7 as el=%items_el_8, idx=%items_i_9
+      %t10 = load_field items_el_8["price"] :: float
+      %t12 = call core.mul(%t10, %t11) :: float
+      yield %t12
+    end_loop
+  )
+  (Declaration is_valid_quantity
+    %t13 = load_input "items" :: array
+    %t17 = const 0 :: integer
+    loop items id=L3 in %t13 as el=%items_el_14, idx=%items_i_15
+      %t16 = load_field items_el_14["quantity"] :: integer
+      %t18 = call core.gt(%t16, %t17) :: boolean
+      yield %t18
+    end_loop
+  )
+  (Declaration expensive_items
+    %t19 = load_input "items" :: array
+    %t23 = const 100.0 :: float
+    loop items id=L4 in %t19 as el=%items_el_20, idx=%items_i_21
+      %t22 = load_field items_el_20["price"] :: float
+      %t24 = call core.gt(%t22, %t23) :: boolean
+      yield %t24
+    end_loop
+  )
+  (Declaration electronics
+    %t25 = load_input "items" :: array
+    %t29 = const "electronics" :: string
+    loop items id=L5 in %t25 as el=%items_el_26, idx=%items_i_27
+      %t28 = load_field items_el_26["category"] :: string
+      %t30 = call core.eq(%t28, %t29) :: boolean
+      yield %t30
+    end_loop
+  )
+)

data/golden/array_operations/expected/nast.txt

@@ -0,0 +1,32 @@
+(NAST
+  (VALUE subtotals
+    (Call :"core.mul"
+    (InputRef [:items, :item, :price])
+    (InputRef [:items, :item, :quantity])
+  )
+  )
+  (VALUE discounted_price
+    (Call :"core.mul"
+    (InputRef [:items, :item, :price])
+    (Const 0.9)
+  )
+  )
+  (VALUE is_valid_quantity
+    (Call :"core.gt"
+    (InputRef [:items, :item, :quantity])
+    (Const 0)
+  )
+  )
+  (TRAIT expensive_items
+    (Call :"core.gt"
+    (InputRef [:items, :item, :price])
+    (Const 100.0)
+  )
+  )
+  (TRAIT electronics
+    (Call :"core.eq"
+    (InputRef [:items, :item, :category])
+    (Const "electronics")
+  )
+  )
+)

data/golden/array_operations/expected/schema_javascript.mjs

@@ -0,0 +1,60 @@
+export function _subtotals(input) {
+  let out = [];
+  let t1 = input["items"];
+  t1.forEach((items_el_2, items_i_3) => {
+    let t4 = items_el_2["price"];
+    let t5 = items_el_2["quantity"];
+    let t6 = t4 * t5;
+    out.push(t6);
+  });
+  return out;
+}
+
+export function _discounted_price(input) {
+  let out = [];
+  let t7 = input["items"];
+  const t11 = 0.9;
+  t7.forEach((items_el_8, items_i_9) => {
+    let t10 = items_el_8["price"];
+    let t12 = t10 * t11;
+    out.push(t12);
+  });
+  return out;
+}
+
+export function _is_valid_quantity(input) {
+  let out = [];
+  let t13 = input["items"];
+  const t17 = 0;
+  t13.forEach((items_el_14, items_i_15) => {
+    let t16 = items_el_14["quantity"];
+    let t18 = t16 > t17;
+    out.push(t18);
+  });
+  return out;
+}
+
+export function _expensive_items(input) {
+  let out = [];
+  let t19 = input["items"];
+  const t23 = 100.0;
+  t19.forEach((items_el_20, items_i_21) => {
+    let t22 = items_el_20["price"];
+    let t24 = t22 > t23;
+    out.push(t24);
+  });
+  return out;
+}
+
+export function _electronics(input) {
+  let out = [];
+  let t25 = input["items"];
+  const t29 = "electronics";
+  t25.forEach((items_el_26, items_i_27) => {
+    let t28 = items_el_26["category"];
+    let t30 = t28 == t29;
+    out.push(t30);
+  });
+  return out;
+}
+