61 lines
1.4 KiB
Ruby
61 lines
1.4 KiB
Ruby
module Sof
|
|
class HashNode < Node
|
|
def initialize ref
|
|
super(ref)
|
|
@children = []
|
|
end
|
|
attr_reader :children
|
|
def add key , val
|
|
@children << [key,val]
|
|
end
|
|
def out io , level = 0
|
|
super
|
|
short = true
|
|
children.each do |k,v|
|
|
short = false unless k.is_a?(SimpleNode)
|
|
short = false unless v.is_a?(SimpleNode)
|
|
end
|
|
if(short and children.length < 7 )
|
|
short_out(io,level)
|
|
else
|
|
long_out(io , level)
|
|
end
|
|
end
|
|
def short_out(io,level)
|
|
io.write("{")
|
|
children.each_with_index do |child , i|
|
|
key , val = child
|
|
key.out(io , level + 1)
|
|
io.write " => "
|
|
val.out(io , level + 1)
|
|
io.write ", " unless (i+1) == children.length
|
|
end
|
|
io.write("}")
|
|
end
|
|
def long_out io , level
|
|
indent = " " * level
|
|
children.each_with_index do |child , i|
|
|
key , val = child
|
|
io.write "\n#{indent}" unless i == 0
|
|
io.write "-"
|
|
key.out(io , level + 1)
|
|
io.write " => "
|
|
val.out(io , level + 1)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
Hash.class_eval do
|
|
def to_sof_node(writer , level , ref)
|
|
node = Sof::HashNode.new(ref)
|
|
each do |key , object|
|
|
k = writer.to_sof_node( key ,level + 1)
|
|
v = writer.to_sof_node( object ,level +1)
|
|
node.add(k , v)
|
|
end
|
|
node
|
|
end
|
|
end
|
|
|