1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
|
# -*- coding: utf-8 -*-
# Libs should not require rubygems, commenting out
# require 'rubygems'
require 'graphviz'
require 'graphviz/family_tree/generation'
require 'graphviz/family_tree/person'
require 'graphviz/family_tree/couple'
require 'graphviz/family_tree/sibling'
class GraphViz
class FamilyTree
# Create a new family tree
#
# require 'graphviz/family_tree'
# t = GraphViz::FamilyTree.new do
# ...
# end
def initialize( &block )
@persons = {}
@graph = GraphViz.new( "FamilyTree", :use => :neato )
@generation_number = 0
@generations = []
@couples = {}
instance_eval(&block) if block
end
# Add a new generation in the tree
#
# require 'graphviz/family_tree'
# t = GraphViz::FamilyTree.new do
# generation do
# ...
# end
# generation do
# ...
# end
# end
def generation( &b )
gen = GraphViz::FamilyTree::Generation.new( @graph, @persons, self, @generation_number )
gen.make( &b )
@generations << gen
@generation_number += 1
end
def persons #:nodoc:
@persons ||= {}
end
def add_couple( x, y, node ) #:nodoc:
@couples[x] = {} if @couples[x].nil?
@couples[x][y] = GraphViz::FamilyTree::Couple.new( @graph, node, [x, y] )
@couples[y] = {} if @couples[y].nil?
@couples[y][x] = @couples[x][y]
end
# Get a couple (GraphViz::FamilyTree::Couple)
def couple( x, y )
@couples[x][y]
end
def method_missing(sym, *args, &block) #:nodoc:
persons[sym.to_s]
end
# Family size
def size
@persons.size
end
# Get the graph
def graph
maxY = @generations.size
biggestGen, maxX = biggestGenerationNumberAndSize
puts "#{maxY} generations"
puts "Plus grosse generation : ##{biggestGen} avec #{maxX} personnes"
puts "traitement des générations..."
puts " #{biggestGen}:"
@generations[biggestGen].persons.each do |id, person|
puts " - #{id} : #{person.class}"
end
puts " Up..."
(0...biggestGen).reverse_each do |genNumber|
puts " #{genNumber}:"
@generations[genNumber].persons.each do |id, person|
puts " - #{id} : #{person.class}"
end
end
puts " Down..."
((biggestGen+1)...maxY).each do |genNumber|
puts " #{genNumber}:"
@generations[genNumber].persons.each do |id, person|
puts " - #{id} : #{person.class}"
end
end
@graph
end
private
def biggestGenerationNumberAndSize
size = 0
number = 0
@generations.each do |gen|
if gen.size > size
size = gen.size
number = gen.number
end
end
return number, size
end
end
end
|
