Best of Ruby Quiz - GEDCOM Parser

[0] 首先，分析书后源代码，整题建立在下面的假设上：Level 标号从0开始，依次递增，且递增跨度为1，输入不存在错误

也正是这一点假设，LEVEL值可以不作为结点属性处理，即LEVEL值实际上是节点在堆栈中位置的代表，用堆栈结构可以轻松化解LEVEL带来的麻烦

 

[1] 第一种想法是用REXML，将字符串解析后存入XML结构，最后统一输出，源代码和书上一样

require 'rexml/document'
doc = REXML::Document.new '<gedcom/>'
stack = [doc.root]
IO.read(ARGV[0]).each do |line|
  next if line =~ /^\s*$/
  line =~ /^\s*(\d+)\s+(@\S+@|\S+)\s*(.*?)$/ or raise "Invalid GEDCOM"
  level , tag , data = $1.to_i , $2 , $3
  stack.pop while (level + 1 < stack.size)
  parent = stack.last
  if tag =~ /^@(\S+)@$/
    ele = parent.add_element data
    ele.attributes['id'] = tag
  else
    ele = parent.add_element tag
    ele.text = data
  end
  stack.push ele
end
File.open("output_std.txt","w") do |file|
  doc.write(file,0)
end

 

 [2] 当数据规模很大时，不可能将所有数据存入XML结构内存，再统一输出，So fight and run

 

class Node
  def initialize(tag_or_id,data = "")
    if tag_or_id =~ /@.*@/
      @name , @myid , @value = data , tag_or_id , ""
    else
      @name , @value ,@myid = tag_or_id , data , ""
    end
  end
  
  def to_s_first
    s = @myid.empty? ? "<#{@name}>\n" : "<#{@name} id=\"#{@myid}\">\n" 
    s += (@value+"\n") unless @value.empty?
    s
  end
  
  def to_s_last
    "</#{@name}>\n"
  end
  
end

stack = [Node.new "gedcom"]
File.open("output.txt","w") do |file|
  file.print stack.first.to_s_first
  IO.read($*[0]).each do |line|
    next if line =~ /^\s*$/
    line =~ /^\s*(\d+)\s+(@\S+@|\S+)\s*(.*?)$/ or raise "error"
    level , tag_or_id , data = $1.to_i , $2 , $3
    file.print stack.pop.to_s_last while (level + 1 < stack.size)
    node = Node.new(tag_or_id,data)
    file.print node.to_s_first
    stack.push node  
  end
  file.print stack.first.to_s_last
end

将节点打包成类，第二步解析交给类构造去做，输出也交给类做

在入栈时输出第一部分，在出栈时输出第二部分

自然想到将栈也打包

 

 1require 'rexml/text'
 2
 3class Node
 4  def initialize(tag_or_id,data = "")
 5    if tag_or_id =~ /@.*@/
 6      @name , @myid , @value = data , tag_or_id , ""
 7    else
 8      @name , @value ,@myid = tag_or_id , data , ""
 9    end
  end
  
  def to_s_first
    s = @myid.empty? ? "<#{@name}>\n" : "<#{@name} id=\'#{@myid}\'>\n" 
    s += (@value+"\n") unless @value.empty?
    s
  end
  
  def to_s_last
    "</#{@name}>\n"
  end
  
end

class Stack < Array
    def push(obj)
        raise "type error" unless obj.is_a? Node
        print obj.to_s_first        
        super(obj)
    end
    
    def pop
        print self.last.to_s_last
        super
    end
end

def file_write_env(file)
    $stdout =     file
    yield
    $stdout = STDOUT
end

stack = Stack.new
File.open("output.txt","w") do |file|
    file_write_env(file) do
        stack.push(Node.new "gedcom")
          IO.read($*[0]).each do |line|
            next if line =~ /^\s*$/
            line =~ /^\s*(\d+)\s+(@\S+@|\S+)\s*(.*?)$/ or raise "error"
            level , tag_or_id , data = $1.to_i , $2 , REXML::Text::normalize($3)
            stack.pop while (level + 1 < stack.size)
            stack.push Node.new(tag_or_id,data)  
          end
        stack.pop
    end
end

第50行调用  REXML::Text::normalize 将字符串escaping ，达到和XML输出差不多的效果，完成字符转义

 

[PS] 有一点不很明白，用REXML构造法得到的输出，和最后的方法得到的输出在空格处理上有一定差异，REXML对于多空格只输出一个，而最后两种方法忠实于输入