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
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
|
# p3.rb -- for Pythagorian 3 body plots
require 'Tioga/FigureMaker'
class Pythagorian_3_body
include Math
include Tioga
include FigureConstants
def t
@figure_maker
end
def initialize(filename)
@data_filename = filename
@figure_maker = FigureMaker.default
t.def_eval_function { |str| eval(str) }
t.save_dir = 'p3_out'
t.def_figure("All") { all }
t.def_figure("Reds") { reds }
t.def_figure("Blues") { blues }
t.def_figure("Greens") { greens }
t.def_figure("Section") { section }
Dvector.read(filename, [ # don't bother with the z coords since are all 0
@time = Dvector.new,
@xred = Dvector.new,
@yred = Dvector.new,
nil, # z1
@xgreen = Dvector.new,
@ygreen = Dvector.new,
nil, # z2
@xblue = Dvector.new,
@yblue = Dvector.new,
nil] # z3
)
@plot_bounds = [ -5, 5, 5, -5 ]
@section_start = 0 # time
@section_length = 1 # in time units
@weight = 0.5 # default line weight for non-movie plots
@scale = 0.8
t.def_enter_page_function { enter_page }
end
def enter_page
t.page_setup(11*72/2,8.5*72/2)
t.set_frame_sides(0.15,0.85,0.85,0.15) # left, right, top, bottom in page coords
end
def set_style
t.rescale(@scale)
t.line_width = @weight
end
def blues
set_style
t.do_box_labels('Blue Star: Mass 5 $\mathrm{M_{\odot}}$', 'x', 'y')
t.show_plot(@plot_bounds) { t.show_polyline(@xblue, @yblue, Blue) }
end
def reds
set_style
t.do_box_labels('Red Star: Mass 3 $\mathrm{M_{\odot}}$', 'x', 'y')
t.show_plot(@plot_bounds) { t.show_polyline(@xred, @yred, Red) }
end
def greens
set_style
t.do_box_labels('Green Star: Mass 4 $\mathrm{M_{\odot}}$', 'x', 'y')
t.show_plot(@plot_bounds) { t.show_polyline(@xgreen, @ygreen, Green) }
end
def piece_of_line(xs, ys, loc, color)
t.stroke_color = color
t.stroke_line(xs[loc], ys[loc], xs[loc+1], ys[loc+1])
end
def show_section(start_time, duration)
set_style
t.do_box_labels("Time from #{start_time} to #{start_time + duration}", "x", "y")
end_time = start_time + duration
start_loc = @time.where_closest(start_time)
loc = start_loc
t.show_plot(@plot_bounds) do
while (loc+1 < @time.size and @time[loc] <= end_time) do
piece_of_line(@xblue, @yblue, loc, Blue)
piece_of_line(@xred, @yred, loc, Red)
piece_of_line(@xgreen, @ygreen, loc, Green)
loc = loc + 1
end
end
end
def section
show_section(@section_start, @section_length)
@section_start = @section_start + 1
@section_start = 0 if @section_start+@section_length >= @time[-1]
# each time section is called it moves to the next section of the data
end
def all
show_section(0, @time[-1])
end
def movie(num_frames = 200, line_width = 2, bounds = [ -7, 7, 7, -7 ])
puts "make movie"
t.save_dir = 'movie_out'
system('rm movie_out/*')
t.add_model_number = true
section_plot_number = t.figure_names.index('Section')
lw = @weight
@weight = line_width
bnds = @plot_bounds
@plot_bounds = bounds
@section_start = @time[0]
frame_size = (@time[-1] - @time[0]) / num_frames
@section_length = frame_size * 1.5
num_frames.times do |i|
t.model_number = i
@section_start = frame_size * i
t.make_preview_pdf(section_plot_number)
end
@weight = lw
@plot_bounds = bnds
system('cd movie_out; rm *.txt *.aux *.log *.tex *_figure.pdf')
t.save_dir = 'p3_out'
t.add_model_number = false
return 'finished'
end
end
$p3 = Pythagorian_3_body.new('p3.data')
|
