[ruby-list:23437] Re: dump a single object

咳といいます。

> > まさかやってないだろうと勝手に思い込んでました。
> > Rubyで環状なオブジェクトを dump するスクリプトを書いて、
> > ruby-list にポストする寸前にmashal 試したらびっくり。
> > ちゃんと共有されているオブジェクトは共有されて dump/load され
> > るんですよね。無駄になってしまった。# あ、xmarshal に混ぜれば良いのか。
> 
> それいいですね。ぼくも興味あります。

じゃああげます。恥ずかしいけど。

# Marshalのdump/load を試してびっくりした瞬間のソースですから
# サンプルが半端な状態です。勘で読んでください。

使い方)
  s = RefMarshal.dump(root), 
  root = RefMarshal.load(s)

私のはダサくて root から全てのオブジェクトを表にしまうのですが、
marshal.cはそのオブジェクトで必要な分しか表を作らないんです。


> > marshal.c読んでいて思ったんですけど、どうでしょう。
> > オブジェクトの構造をなめたりする作業はmarshalに任せて
> > marshalフォーマットの echoder/decoder みたいなの。

> そうなんですよ。ぼくが行き違いのメールに書いた、
>   > で、ぼくのやりたいのは、チェック*する*ほうです。ざっくり言えば、
>   > Marshal の出力形式が Hash になるというやつです。
> てのは、まさにそれなんですね。Iterator でアクセスできれば吉だと。
> 
> ただ、いちいち marshal を解析するより、marshal がそれを提供する
> ほうが美しいかなと。ドキュメントがないと、フォーマットの解析は
> ソースを読むしかなく、変更の波及が怖いです。

それはそうですね。
もし marshal.c に手を入れるなら、Marshal済みオブジェクトみたいのが
返ってくる方がいいかもしれないです。
下の RefMarshal::Map のような…

    

# Reference Marshal
#
# $Date: 2000/05/19 23:33:04 $
# Copyright (c) 2000 Masatoshi SEKI
#
# refmarshal.rb is copyrighted free software by Masatoshi SEKI.
# You can redistribute it and/or modify it under the same term as Ruby.

class Object
# __iv_get, __iv_set : import from xmarshal.rb
  def __iv_get(name)
    raise "invalid ivar name #{name}" if name !~ /^@\w+$/
    eval name
  end

  def __iv_set(name, val)
    raise "invalid ivar name #{name}" if name !~ /^@\w+$/
    eval "#{name} = val"
  end
end

module RefMarshal
  class Ref
    def initialize(obj)
      @ref = obj.id
    end
    attr_reader(:ref)
  end

  class Map
    def initialize(root)
      @root = root.id
      @map = {}
      ref_walk(root)
    end

    private
    def ref_walk(src)
      return if @map[src.id]

      @map[src.id] = true	# reserved
      begin
	obj = src.dup
      rescue TypeError		# nil, true, false ...
	@map[src.id] = src
	return
      end

      case src
      when Array
	src.each_with_index do |v,i|
	  ref_walk(v)
	  obj[i] = Ref.new(v)
	end
      when Hash
	src.each do |k,v|
	  ref_walk(k)
	  ref_walk(v)
	  obj.delete(k)
	  obj[Ref.new(k)] = Ref.new(v)
	end
      end

      ivs = obj.instance_variables
      ivs.each do |name|
	v = obj.__iv_get(name)
	ref = Ref.new(v)
	obj.__iv_set(name, ref)
	ref_walk(v)
      end
      @map[src.id] = obj
    end

    def resolve_ref(obj)
      case obj
      when Array
	obj.each_with_index do |v,i|
	  obj[i] = @map[v.ref] if v.kind_of? Ref
	end
      when Hash
	keys = obj.keys
	keys.each do |k|
	  v = obj[k]
	  if k.kind_of? Ref
	    obj.delete(k) 
	    k = @map[k.ref]
	  end
	  v = @map[v.ref] if v.kind_of? Ref
	  obj[k] = v
	end
      end
      
      ivs = obj.instance_variables
      ivs.each do |name|
	v = obj.__iv_get(name)
	if v.kind_of? Ref
	  v = @map[v.ref]
	  obj.__iv_set(name, v)
	end
      end
    end
    
    def resolve
      @map.each do |k, v|
	resolve_ref(v)
      end
      @map[@root]
    end

    public
    def root
      return @root unless @map
      @root = resolve
      @map = nil	# resolved
      @root
    end
  end

  def RefMarshal.dump(root, port=nil)
    m = Map.new(root)
    port ? Marshal.dump(m, port) : Marshal.dump(m)
  end

  def RefMarshal.load(data)
    m = Marshal.load(data)
    m.root
  end
end

# 以下はsample 
class Linked
  def initialize(v)
    @value = v
    @fwd = @bwd = self
  end
  attr_accessor(:fwd, :bwd, :value)
  
  def add(v)
    cell = Linked.new(v)
    cell.fwd = self.fwd
    cell.bwd = self
    self.fwd = cell
  end

  def each
    yield(@value)

    curr = self.fwd
    while curr != self
      yield(curr.value)
      curr = curr.fwd
    end
  end
end

list = Linked.new('I')
list.add('like')
list.add('Ruby')

list.each do |v|
  p v
end

p Marshal.dump(list)

toolong = Marshal.load(Marshal.dump(list))

toolong.each do |v|
  p v
end

p toolong