Sha256: a263c4c4c82b87047b9fc7e68d01acd9b9071cb9a1de7766d73f03e5b8f466c8
Contents?: true
Size: 1.78 KB
Versions: 30
Compression:
Stored size: 1.78 KB
Contents
module Steep
module AST
module Types
class Union
attr_reader :types
attr_reader :location
def initialize(types:, location: nil)
@types = types
@location = location
end
def self.build(types:, location: nil)
types.flat_map do |type|
if type.is_a?(Union)
type.types
else
[type]
end
end.map do |type|
case type
when AST::Types::Any
return AST::Types::Any.new(location: location)
when AST::Types::Top
return AST::Types::Top.new(location: location)
when AST::Types::Bot
nil
else
type
end
end.compact.uniq.yield_self do |tys|
if tys.length == 1
tys.first
else
new(types: tys.sort_by(&:hash), location: location)
end
end
end
def ==(other)
other.is_a?(Union) &&
other.types == types
end
def hash
self.class.hash ^ types.hash
end
alias eql? ==
def subst(s)
self.class.build(location: location,
types: types.map {|ty| ty.subst(s) })
end
def to_s
"(#{types.map(&:to_s).sort.join(" | ")})"
end
def free_variables
types.each.with_object(Set.new) do |type, set|
set.merge(type.free_variables)
end
end
include Helper::ChildrenLevel
def level
[0] + level_of_children(types)
end
def with_location(new_location)
self.class.new(types: types, location: new_location)
end
end
end
end
end
Version data entries
30 entries across 30 versions & 1 rubygems