Gecko [ v3.cesa.co.za ]

Name	Size	Permission	Date
annotate	[ DIR ]	drwxr-xr-x	2025-06-05 11:22
ast	[ DIR ]	drwxr-xr-x	2025-06-05 11:22
collection	[ DIR ]	drwxr-xr-x	2025-06-05 11:22
definition_builder	[ DIR ]	drwxr-xr-x	2025-06-05 11:22
parser_compat	[ DIR ]	drwxr-xr-x	2025-06-05 11:22
prototype	[ DIR ]	drwxr-xr-x	2025-06-05 11:22
test	[ DIR ]	drwxr-xr-x	2025-06-05 11:22
ancestor_graph.rb	2.47 KB	-rw-r--r--	2022-04-12 11:11
annotate.rb	157 B	-rw-r--r--	2022-04-12 11:11
buffer.rb	1 KB	-rw-r--r--	2022-04-12 11:11
builtin_names.rb	1.59 KB	-rw-r--r--	2022-04-12 11:11
char_scanner.rb	292 B	-rw-r--r--	2022-04-12 11:11
cli.rb	27.19 KB	-rw-r--r--	2022-04-12 11:11
collection.rb	295 B	-rw-r--r--	2022-04-12 11:11
constant.rb	461 B	-rw-r--r--	2022-04-12 11:11
constant_table.rb	5.04 KB	-rw-r--r--	2022-04-12 11:11
definition.rb	9.47 KB	-rw-r--r--	2022-04-12 11:11
definition_builder.rb	32.63 KB	-rw-r--r--	2022-04-12 11:11
environment.rb	15.26 KB	-rw-r--r--	2022-04-12 11:11
environment_loader.rb	3.32 KB	-rw-r--r--	2022-04-12 11:11
environment_walker.rb	4.2 KB	-rw-r--r--	2022-04-12 11:11
errors.rb	11.68 KB	-rw-r--r--	2022-04-12 11:11
factory.rb	301 B	-rw-r--r--	2022-04-12 11:11
location_aux.rb	2.73 KB	-rw-r--r--	2022-04-12 11:11
locator.rb	5.39 KB	-rw-r--r--	2022-04-12 11:11
method_type.rb	2.65 KB	-rw-r--r--	2022-04-12 11:11
namespace.rb	2.12 KB	-rw-r--r--	2022-04-12 11:11
parser_aux.rb	1.3 KB	-rw-r--r--	2022-04-12 11:11
repository.rb	2.83 KB	-rw-r--r--	2022-04-12 11:11
substitution.rb	1.23 KB	-rw-r--r--	2022-04-12 11:11
test.rb	2.57 KB	-rw-r--r--	2022-04-12 11:11
type_alias_dependency.rb	2.82 KB	-rw-r--r--	2022-04-12 11:11
type_alias_regularity.rb	3.14 KB	-rw-r--r--	2022-04-12 11:11
type_name.rb	1.63 KB	-rw-r--r--	2022-04-12 11:11
type_name_resolver.rb	1.56 KB	-rw-r--r--	2022-04-12 11:11
types.rb	27.19 KB	-rw-r--r--	2022-04-12 11:11
validator.rb	4.6 KB	-rw-r--r--	2022-04-12 11:11
variance_calculator.rb	4.45 KB	-rw-r--r--	2022-04-12 11:11
vendorer.rb	2.03 KB	-rw-r--r--	2022-04-12 11:11
version.rb	35 B	-rw-r--r--	2022-04-12 11:11
writer.rb	7.91 KB	-rw-r--r--	2022-04-12 11:11

Rename

module RBS
  class AncestorGraph
    InstanceNode = _ = Struct.new(:type_name, keyword_init: true)
    SingletonNode = _ = Struct.new(:type_name, keyword_init: true)

attr_reader :env
    attr_reader :ancestor_builder
    attr_reader :parents
    attr_reader :children

def initialize(env:, ancestor_builder: DefinitionBuilder::AncestorBuilder.new(env: env))
      @env = env
      @ancestor_builder = ancestor_builder
      build()
    end

def build()
      @parents = {}
      @children = {}

env.class_decls.each_key do |type_name|
        build_ancestors(InstanceNode.new(type_name: type_name), ancestor_builder.one_instance_ancestors(type_name))
        build_ancestors(SingletonNode.new(type_name: type_name), ancestor_builder.one_singleton_ancestors(type_name))
      end
      env.interface_decls.each_key do |type_name|
        build_ancestors(InstanceNode.new(type_name: type_name), ancestor_builder.one_interface_ancestors(type_name))
      end
    end

def build_ancestors(node, ancestors)
      ancestors.each_ancestor do |ancestor|
        case ancestor
        when Definition::Ancestor::Instance
          register(child: node, parent: InstanceNode.new(type_name: ancestor.name))
        when Definition::Ancestor::Singleton
          register(child: node, parent: SingletonNode.new(type_name: ancestor.name))
        end
      end
    end

def register(parent:, child:)
      (parents[child] ||= Set[]) << parent
      (children[parent] ||= Set[]) << child
    end

def each_parent(node, &block)
      if block
        parents[node]&.each(&block)
      else
        enum_for :each_parent, node
      end
    end

def each_child(node, &block)
      if block
        children[node]&.each(&block)
      else
        enum_for :each_child, node
      end
    end

def each_ancestor(node, yielded: Set[], &block)
      if block
        each_parent(node) do |parent|
          unless yielded.member?(parent)
            yielded << parent
            yield parent
            each_ancestor(parent, yielded: yielded, &block)
          end
        end
      else
        enum_for :each_ancestor, node
      end
    end

def each_descendant(node, yielded: Set[], &block)
      if block
        each_child(node) do |child|
          unless yielded.member?(child)
            yielded << child
            yield child
            each_descendant(child, yielded: yielded, &block)
          end
        end
      else
        enum_for :each_descendant, node
      end
    end
  end
end