Sha256: 016e9f0d6348ddb2f27b21a1b49677b15a4105a8af25c10b289d0549ff30c395
Contents?: true
Size: 1.26 KB
Versions: 11
Compression:
Stored size: 1.26 KB
Contents
# frozen_string_literal: true
module RBS
class Substitution
attr_reader :mapping
attr_accessor :instance_type
def empty?
mapping.empty? && instance_type.nil?
end
def initialize()
@mapping = {}
end
def add(from:, to:)
mapping[from] = to
end
def self.build(variables, types, instance_type: nil, &block)
unless variables.size == types.size
raise "Broken substitution: variables=#{variables}, types=#{types}"
end
mapping = variables.zip(types).to_h
self.new.tap do |subst|
mapping.each do |v, t|
type = block_given? ? yield(t) : t
subst.add(from: v, to: type)
end
subst.instance_type = instance_type
end
end
def apply(ty)
case ty
when Types::Variable
# @type var ty: Types::Variable
mapping[ty.name] || ty
when Types::Bases::Instance
if t = instance_type
t
else
ty
end
else
ty
end
end
def without(*vars)
Substitution.new.tap do |subst|
subst.mapping.merge!(mapping)
vars.each do |var|
subst.mapping.delete(var)
end
subst.instance_type = self.instance_type
end
end
end
end
Version data entries
11 entries across 11 versions & 2 rubygems