Rewrite :incr internal operator into += method call

vidarh · Sep 23, 2014 · b3edea5 · b3edea5
1 parent c288585
commit b3edea5
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/treeoutput.rb b/treeoutput.rb
@@ -67,6 +67,12 @@ def oper(o)
         else
           @vstack << E[:callm, leftv, :[], [rightv]]
         end
+      elsif o.sym == :incr
+        if ra and rightv[0] == :array
+          @vstack << E[:assign, leftv, [:callm, leftv, :"+", flatten(rightv[1..-1])]]
+        else
+          @vstack << E[:assign, leftv, [:callm, leftv, :"+", [rightv]]]
+        end
       elsif ra and rightv[0] == :comma and o.sym == :array || o.sym == :hash
         @vstack << E[o.sym, leftv].compact + flatten(rightv)
       elsif ra and rightv[0] == :comma and o.sym != :comma