Add bag operators.

sorear · Feb 19, 2012 · 49030d6 · 49030d6
1 parent 8488bba
commit 49030d6
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/lib/CORE.setting b/lib/CORE.setting
@@ -2193,6 +2193,16 @@ sub bag(*@a) {
     Bag.new(|@a);
 }
 
+proto sub infix:<⊍>($, $ --> Bag) is equiv(&infix:<X>) {*}
+multi sub infix:<⊍>(Any $a, Any $b --> Bag) { bag($a) ⊍ bag($b) }
+multi sub infix:<⊍>(Bag $a, Bag $b --> Bag) { bag(($a ∪ $b).map({ ; $_ => $a{$_} min $b{$_} })) }
+only  sub infix:<(.)>($a, $b --> Bag) is equiv(&infix:<X>) { $a ⊍ $b }
+
+proto sub infix:<⊎>($, $ --> Bag) is equiv(&infix:<X>) {*}
+multi sub infix:<⊎>(Any $a, Any $b --> Bag) { bag($a) ⊎ bag($b) }
+multi sub infix:<⊎>(Bag $a, Bag $b --> Bag) { bag(($a ∪ $b).map({ ; $_ => $a{$_} + $b{$_} })) }
+only  sub infix:<(+)>($a, $b --> Bag) is equiv(&infix:<X>) { $a ⊎ $b }
+
 class KeyBag does Associative {
     has Int %!elems; # should be UInt