kumi 0.0.5 → 0.0.7

data/lib/kumi/vectorization_metadata.rb

@@ -0,0 +1,108 @@
+# frozen_string_literal: true
+
+module Kumi
+  # Metadata system for vectorization detection and handling
+  module VectorizationMetadata
+    # Tracks which declarations are arrays with children (vectorizable)
+    class ArrayDeclarationTracker
+      def initialize
+        @array_declarations = {}
+      end
+
+      def register_array(name, children)
+        @array_declarations[name] = children.map(&:name)
+      end
+
+      def array_declaration?(name)
+        @array_declarations.key?(name)
+      end
+
+      def array_children(name)
+        @array_declarations[name] || []
+      end
+
+      def all_arrays
+        @array_declarations.keys
+      end
+    end
+
+    # Detects vectorized operations in expressions
+    class VectorizationDetector
+      def initialize(array_tracker)
+        @array_tracker = array_tracker
+      end
+
+      # Check if an expression should be vectorized
+      def vectorized_expression?(expression)
+        case expression
+        when Kumi::Syntax::CallExpression
+          vectorized_call?(expression)
+        when Kumi::Syntax::InputElementReference
+          vectorized_element_reference?(expression)
+        else
+          false
+        end
+      end
+
+      # Check if a function call should be treated as a reducer
+      def reducer_function?(fn_name, args)
+        REDUCER_FUNCTIONS.include?(fn_name) &&
+          args.any? { |arg| vectorized_expression?(arg) }
+      end
+
+      private
+
+      REDUCER_FUNCTIONS = %i[sum min max size length first last].freeze
+
+      def vectorized_call?(call_expr)
+        # Arithmetic operations between array elements are vectorized
+        ARITHMETIC_OPERATIONS.include?(call_expr.fn_name) &&
+          call_expr.args.any? { |arg| vectorized_expression?(arg) }
+      end
+
+      def vectorized_element_reference?(elem_ref)
+        return false unless elem_ref.path.size >= 2
+
+        array_name, _field_name = elem_ref.path
+        @array_tracker.array_declaration?(array_name)
+      end
+
+      ARITHMETIC_OPERATIONS = %i[add subtract multiply divide modulo power].freeze
+    end
+
+    # Metadata about how values should be computed
+    class ComputationMetadata
+      attr_reader :vectorized_values, :reducer_values, :scalar_values
+
+      def initialize
+        @vectorized_values = Set.new
+        @reducer_values = Set.new
+        @scalar_values = Set.new
+      end
+
+      def mark_vectorized(name)
+        @vectorized_values.add(name)
+      end
+
+      def mark_reducer(name)
+        @reducer_values.add(name)
+      end
+
+      def mark_scalar(name)
+        @scalar_values.add(name)
+      end
+
+      def vectorized?(name)
+        @vectorized_values.include?(name)
+      end
+
+      def reducer?(name)
+        @reducer_values.include?(name)
+      end
+
+      def scalar?(name)
+        @scalar_values.include?(name)
+      end
+    end
+  end
+end

data/lib/kumi/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Kumi
-  VERSION = "0.0.5"
+  VERSION = "0.0.7"
 end

data/lib/kumi.rb

@@ -9,4 +9,18 @@ loader.setup
 
 module Kumi
   extend Schema
+
+  def self.inspector_from_schema
+    Inspector.new(@__syntax_tree__, @__analyzer_result__, @__compiled_schema__)
+  end
+
+  def self.reset!
+    @__syntax_tree__ = nil
+    @__analyzer_result__ = nil
+    @__compiled_schema__ = nil
+    @__schema_metadata__ = nil
+  end
+
+  # Reset on require to avoid state leakage in tests
+  reset!
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: kumi
 version: !ruby/object:Gem::Version
-  version: 0.0.5
+  version: 0.0.7
 platform: ruby
 authors:
 - André Muta
@@ -35,24 +35,40 @@ files:
 - LICENSE.txt
 - README.md
 - Rakefile
+- docs/AST.md
+- docs/DSL.md
+- docs/FUNCTIONS.md
+- docs/SYNTAX.md
 - docs/development/README.md
 - docs/development/error-reporting.md
-- documents/AST.md
-- documents/DSL.md
-- documents/FUNCTIONS.md
-- documents/SYNTAX.md
+- docs/features/README.md
+- docs/features/analysis-cascade-mutual-exclusion.md
+- docs/features/analysis-type-inference.md
+- docs/features/analysis-unsat-detection.md
+- docs/features/array-broadcasting.md
+- docs/features/input-declaration-system.md
+- docs/features/performance.md
+- docs/schema_metadata.md
+- docs/schema_metadata/broadcasts.md
+- docs/schema_metadata/cascades.md
+- docs/schema_metadata/declarations.md
+- docs/schema_metadata/dependencies.md
+- docs/schema_metadata/evaluation_order.md
+- docs/schema_metadata/examples.md
+- docs/schema_metadata/inferred_types.md
+- docs/schema_metadata/inputs.md
 - examples/deep_schema_compilation_and_evaluation_benchmark.rb
 - examples/federal_tax_calculator_2024.rb
 - examples/game_of_life.rb
 - examples/simple_rpg_game.rb
 - examples/static_analysis_errors.rb
 - examples/wide_schema_compilation_and_evaluation_benchmark.rb
-- lib/generators/trait_engine/templates/schema_spec.rb.erb
 - lib/kumi.rb
 - lib/kumi/analyzer.rb
 - lib/kumi/analyzer/analysis_state.rb
 - lib/kumi/analyzer/constant_evaluator.rb
-- lib/kumi/analyzer/passes/definition_validator.rb
+- lib/kumi/analyzer/passes/broadcast_detector.rb
+- lib/kumi/analyzer/passes/declaration_validator.rb
 - lib/kumi/analyzer/passes/dependency_resolver.rb
 - lib/kumi/analyzer/passes/input_collector.rb
 - lib/kumi/analyzer/passes/name_indexer.rb
@@ -69,7 +85,6 @@ files:
 - lib/kumi/compiled_schema.rb
 - lib/kumi/compiler.rb
 - lib/kumi/constraint_relationship_solver.rb
-- lib/kumi/domain.rb
 - lib/kumi/domain/enum_analyzer.rb
 - lib/kumi/domain/range_analyzer.rb
 - lib/kumi/domain/validator.rb
@@ -95,28 +110,43 @@ files:
 - lib/kumi/function_registry/math_functions.rb
 - lib/kumi/function_registry/string_functions.rb
 - lib/kumi/function_registry/type_functions.rb
-- lib/kumi/input.rb
 - lib/kumi/input/type_matcher.rb
 - lib/kumi/input/validator.rb
 - lib/kumi/input/violation_creator.rb
-- lib/kumi/parser/build_context.rb
-- lib/kumi/parser/dsl.rb
-- lib/kumi/parser/dsl_cascade_builder.rb
-- lib/kumi/parser/expression_converter.rb
-- lib/kumi/parser/guard_rails.rb
-- lib/kumi/parser/input_builder.rb
-- lib/kumi/parser/input_proxy.rb
-- lib/kumi/parser/parser.rb
-- lib/kumi/parser/schema_builder.rb
-- lib/kumi/parser/sugar.rb
+- lib/kumi/json_schema.rb
+- lib/kumi/json_schema/generator.rb
+- lib/kumi/json_schema/validator.rb
+- lib/kumi/ruby_parser.rb
+- lib/kumi/ruby_parser/build_context.rb
+- lib/kumi/ruby_parser/declaration_reference_proxy.rb
+- lib/kumi/ruby_parser/dsl.rb
+- lib/kumi/ruby_parser/dsl_cascade_builder.rb
+- lib/kumi/ruby_parser/expression_converter.rb
+- lib/kumi/ruby_parser/guard_rails.rb
+- lib/kumi/ruby_parser/input_builder.rb
+- lib/kumi/ruby_parser/input_field_proxy.rb
+- lib/kumi/ruby_parser/input_proxy.rb
+- lib/kumi/ruby_parser/nested_input.rb
+- lib/kumi/ruby_parser/parser.rb
+- lib/kumi/ruby_parser/schema_builder.rb
+- lib/kumi/ruby_parser/sugar.rb
 - lib/kumi/schema.rb
 - lib/kumi/schema_instance.rb
-- lib/kumi/syntax.rb
-- lib/kumi/syntax/declarations.rb
-- lib/kumi/syntax/expressions.rb
+- lib/kumi/schema_metadata.rb
+- lib/kumi/syntax/array_expression.rb
+- lib/kumi/syntax/call_expression.rb
+- lib/kumi/syntax/cascade_expression.rb
+- lib/kumi/syntax/case_expression.rb
+- lib/kumi/syntax/declaration_reference.rb
+- lib/kumi/syntax/hash_expression.rb
+- lib/kumi/syntax/input_declaration.rb
+- lib/kumi/syntax/input_element_reference.rb
+- lib/kumi/syntax/input_reference.rb
+- lib/kumi/syntax/literal.rb
 - lib/kumi/syntax/node.rb
 - lib/kumi/syntax/root.rb
-- lib/kumi/syntax/terminal_expressions.rb
+- lib/kumi/syntax/trait_declaration.rb
+- lib/kumi/syntax/value_declaration.rb
 - lib/kumi/types.rb
 - lib/kumi/types/builder.rb
 - lib/kumi/types/compatibility.rb
@@ -124,6 +154,7 @@ files:
 - lib/kumi/types/inference.rb
 - lib/kumi/types/normalizer.rb
 - lib/kumi/types/validator.rb
+- lib/kumi/vectorization_metadata.rb
 - lib/kumi/version.rb
 - scripts/generate_function_docs.rb
 homepage: https://github.com/amuta/kumi
@@ -151,6 +182,5 @@ required_rubygems_version: !ruby/object:Gem::Requirement
 requirements: []
 rubygems_version: 3.7.1
 specification_version: 4
-summary: A declarative DSL that transforms business logic into a statically-checked
-  dependency graph
+summary: A Declarative logic framework with static analysis for Ruby.
 test_files: []

data/lib/generators/trait_engine/templates/schema_spec.rb.erb

@@ -1,27 +0,0 @@
-# frozen_string_literal: true
-require "rails_helper"
-
-RSpec.describe <%= schema_constant %> do
-  # Single shared instance for every example
-  let(:schema) { described_class }
-
-  # Minimal dummy context so every binding can run
-  let(:ctx) do
-    {
-<% leaf_keys.each do |k| -%>
-      <%= "#{k}:" %> nil,
-<% end -%>
-    }
-  end
-
-<% expose_names.each do |name| -%>
-  describe "<%= name %>" do
-    it "evaluates without raising" do
-      expect {
-        schema.evaluate_binding(:<%= name %>, ctx)
-      }.not_to raise_error
-    end
-  end
-
-<% end -%>
-end

data/lib/kumi/domain.rb

@@ -1,8 +0,0 @@
-# frozen_string_literal: true
-
-module Kumi
-  module Domain
-  end
-end
-
-require_relative "domain/validator"

data/lib/kumi/input.rb

@@ -1,8 +0,0 @@
-# frozen_string_literal: true
-
-module Kumi
-  module Input
-  end
-end
-
-require_relative "input/validator"

data/lib/kumi/syntax/declarations.rb

@@ -1,26 +0,0 @@
-# frozen_string_literal: true
-
-module Kumi
-  module Syntax
-    module Declarations
-      Attribute = Struct.new(:name, :expression) do
-        include Node
-
-        def children = [expression]
-      end
-
-      Trait = Struct.new(:name, :expression) do
-        include Node
-
-        def children = [expression]
-      end
-
-      # For field metadata declarations inside input blocks
-      FieldDecl = Struct.new(:name, :domain, :type) do
-        include Node
-
-        def children = []
-      end
-    end
-  end
-end

data/lib/kumi/syntax/expressions.rb

@@ -1,34 +0,0 @@
-# frozen_string_literal: true
-
-module Kumi
-  module Syntax
-    module Expressions
-      CallExpression = Struct.new(:fn_name, :args) do
-        include Node
-
-        def children = args
-      end
-      CascadeExpression = Struct.new(:cases) do
-        include Node
-
-        def children = cases
-      end
-
-      WhenCaseExpression = Struct.new(:condition, :result) do
-        include Node
-
-        def children = [condition, result]
-      end
-
-      ListExpression = Struct.new(:elements) do
-        include Node
-
-        def children = elements
-
-        def size
-          elements.size
-        end
-      end
-    end
-  end
-end

data/lib/kumi/syntax/terminal_expressions.rb

@@ -1,30 +0,0 @@
-# frozen_string_literal: true
-
-require_relative "node"
-
-module Kumi
-  module Syntax
-    module TerminalExpressions
-      # Leaf expressions that represent a value or reference and terminate a branch.
-
-      Literal = Struct.new(:value) do
-        include Node
-
-        def children = []
-      end
-
-      # For field usage/reference in expressions (input.field_name)
-      FieldRef = Struct.new(:name) do
-        include Node
-
-        def children = []
-      end
-
-      Binding = Struct.new(:name) do
-        include Node
-
-        def children = []
-      end
-    end
-  end
-end

data/lib/kumi/syntax.rb

@@ -1,9 +0,0 @@
-# frozen_string_literal: true
-
-module Kumi
-  module Syntax
-    include Declarations
-    include Expressions
-    include TerminalExpressions
-  end
-end