25 lines
495 B
Ruby
25 lines
495 B
Ruby
require 'set'
|
|
|
|
class ComparableSet < Set
|
|
def add new
|
|
# smileys only
|
|
return super new unless new.respond_to? :>
|
|
|
|
grep(new.class) do |o|
|
|
return self if o > new
|
|
delete o
|
|
end
|
|
super new
|
|
end
|
|
|
|
alias_method :<<, :add
|
|
|
|
# Set#merge bypasses enumerating the set's contents,
|
|
# so the subclassed #add would never be called
|
|
def merge enum
|
|
enum.is_a?(Enumerable) or raise ArgumentError, "value must be enumerable"
|
|
enum.each { |o| add(o) }
|
|
self
|
|
end
|
|
end
|