# Ruby入門＞イテレータ

イテレータ定義で複数の引数をブロックに渡す

イテレータ定義では、ブロックに渡す引数を複数にすることもできる。
yield文に、複数の引数を渡すようにします。

require "pp"
 
# 1人のプロフィールのクラス。名前、年齢、出身国
class Prof
    attr_accessor :name, :age, :country
    def initialize(name, age, country=nil)
        @name = name
        @age = age
        @country = country
    end
end
 
# プロフィールのリストのクラス
class ProfList
    def initialize()
        @proflist = Array.new
    end
    # 新しい人を加える
    def add(prof)
        @proflist.push(prof)
    end
    # 人数を返す
    def length
        @proflist.length
    end
    # n番目の人を別の人に入れ替える
    def []=(n, prof)
        @proflist[n] = prof
    end
    # n番目の人を返す
    def [](n)
        @proflist[n]
    end
    # リストからプロフィールを削除
    def delete(prof)
        @proflist.delete(prof)
    end
    # イテレータ定義
    def each
        @proflist.each{|prof|
            yield(prof)    
        }
    end
    # イテレータ定義2。名前のみイテレート。
    def each_name
        @proflist.each{|prof|
            yield(prof.name)    
        }
    end
    # イテレータ定義3。名前、年齢、出身国のイテレート。
    def each_all
        @proflist.each{|prof|
            yield(prof.name, prof.age, prof.country)    
        }
    end
end
 
# テストルーチン
proflist = ProfList.new
 
# プロフィールをいくつか作る
p1 = Prof.new("Tom William", 45, "America")
p2 = Prof.new("De Pedoro", 23, "Spain")
p3 = Prof.new("Naoko Yamada", 37, "Japan")
 
# プロフィールリストに人（プロフィール）を追加
proflist.add(p1)
proflist.add(p2)
proflist.add(p3)
 
# プロフィールを1人ずつ表示。
# ProfListクラスの独自定義イテレータ。
puts "--- original iterator"
proflist.each{|prof|
    puts "#{prof.name} - age:#{prof.age} country:#{prof.country}"
}
 
# 名前のみイテレート。
puts "--- name only iterator"
proflist.each_name{|name|
    puts name
}
 
# 名前、年齢、出身国のイテレート。
puts "--- name,age,country iterator"
proflist.each_all{|name, age, country|
    puts "#{name} - age:#{age} country:#{country}"
}

実行結果。

--- original iterator
Tom William - age:45 country:America
De Pedoro - age:23 country:Spain
Naoko Yamada - age:37 country:Japan
--- name only iterator
Tom William
De Pedoro
Naoko Yamada
--- name,age,country iterator
Tom William - age:45 country:America
De Pedoro - age:23 country:Spain
Naoko Yamada - age:37 country:Japan

>> 次の記事 : 引数のあるイテレータ

<< 前の記事 : クラス内にイテレータ定義