optimize unary minus for constants

2026-03-23 10:14:06 +03:00
parent daa2e4e747
commit 5cb52f7bb7
1 changed files with 5 additions and 1 deletions
--- a/lib/ruby_algebra/parser.rb
+++ b/lib/ruby_algebra/parser.rb
@@ -183,7 +183,11 @@ module RubyAlgebra
      when :positive
        out_stack.push a
      when :negative
-        out_stack.push Multiplication.new(Constant.new(-1), a)
+        if a.type == :constant
+          out_stack.push Constant.new(-a.value)
+        else
+          out_stack.push Multiplication.new(Constant.new(-1), a)
+        end
      end
    end