# plots.rb require 'Tioga/FigureMaker' require './plot_styles.rb' class MyPlots include Math include Tioga include FigureConstants include MyPlotStyles def t @figure_maker end def initialize @figure_maker = FigureMaker.default t.save_dir = 'plots_out' t.def_eval_function { |str| eval(str) } @data_filename = "data/datalog.data" @opacity_data = nil @image_right_margin = 0.07 @margin = 0.1 @header = Dvector.new @positions = Dvector.new @blues = Dvector.new @reds = Dvector.new @greens = Dvector.new @data_array = [@positions, @blues, @reds, @greens] @have_data = false @big_blues = Dvector.new @big_blues_scale = 11.0 t.def_figure("Blues") { blues } t.def_figure("Reds") { reds } t.def_figure("Greens") { greens } t.def_figure("Blues_axes_1") { blues_axes_1 } t.def_figure("Blues_axes_2") { blues_axes_2 } t.def_figure("Blues_without_padding") { blues_without_padding } t.def_figure("Log_Reds") { log_reds } t.def_figure("Side_by_Side") { side_by_side } t.def_figure("Two_Ys") { two_yaxes } t.def_figure("Three_Ys") { three_yaxes } t.def_figure("Legends") { legends } t.def_figure("Two_Ys_with_Legends") { two_yaxes_with_legends } t.def_figure("Reds_and_Blues") { reds_blues } t.def_figure("Legend_Inside") { legend_inside } t.def_figure("Legend_Outside") { legend_outside } t.def_figure("Legend_Inside_with_Background") { legend_inside_with_background } t.def_figure("Legend_Inside_with_Background_using_legend_background") { legend_inside_with_background_using_legend_background } t.def_figure("Inset") { inset } t.def_figure("French_Decimal_Separator") { french_decimal_separator } t.def_figure("Column_Triplets") { column_triplets } t.def_figure("Row_Triplets") { row_triplets } t.def_figure("Two_Rows") { two_rows } t.def_figure("Three_Rows") { rows } t.def_figure("Columns") { columns } t.def_figure("Array") { array } t.def_figure("Trio") { trio } t.def_figure("Collage") { collage } t.def_figure("Blues_with_grid") { blues_with_grid } t.def_figure("Blues_reversed_axes") { blues_reversed_axes } t.def_figure("Labels") { labels } t.def_figure("Error_Bars") { error_bars } t.def_figure("Error_Bars_X") { error_bars_x } t.def_figure("Error_Bars_Y") { error_bars_y } t.def_figure("Error_Bars2") { error_bars2 } t.def_figure("Arrows") { arrows } t.def_figure("Special_Y") { special_y } t.def_figure("Sampled_Splines") { sampled_splines } t.def_figure("Sampled_pm_cubics") { sampled_pm_cubics } t.def_figure("Steps") { steps } t.def_figure("Bezier") { bezier } t.def_figure("Sampled_Data") { sampled_data } t.def_figure("Sampled_with_grid") { sampled_with_grid } t.def_figure("Contours") { samples_with_contours } t.def_figure("Greens_clipped") { greens_clipped } t.def_figure("Blues_far_off") { blues_far_off } t.def_figure("Blues_far_off_vincent") { t.vincent_or_bill = true blues_far_off } # Starting to play with axes specifications t.def_figure("Axes_fun") { axes_fun } t.model_number = -1 t.def_enter_page_function { enter_page } # Uncomment the followiing line if you want tioga to leave # its temporary files behind. t.autocleanup = false end def enter_page set_default_plot_style t.default_page_width = 72*3 # in big-points (1/72 inch) t.default_page_height = t.default_page_width t.default_enter_page_function end def french_decimal_separator t.tex_preamble += <<'EOD' \def\frenchsep#1{\frenchsepma#1\endoffrenchsep} \def\fseat#1{\frenchsepma} \def\frenchsepma{\futurelet\next\frenchsepmw} \def\frenchsepmw{\ifx\next\endoffrenchsep\let\endoffrenchsep=\relax% \else\if\next.\ifmmode\mathord,\else,\fi% \else\next\fi\expandafter \fseat\fi} EOD t.yaxis_numeric_label_tex = '$\frenchsep{#1}$' blues end def read_data if t.in_subplot || (@have_data && !t.need_to_reload_data) return end Dvector.read_row(@data_filename, 1, @header) model = @header[0].round if (model != t.model_number) || t.need_to_reload_data Dvector.read(@data_filename, @data_array, 2) t.model_number = model @big_blues = @blues.mul(@big_blues_scale) end @have_data = true t.need_to_reload_data = false end def show_model_number(pos = 1, shift = 4) if !(t.in_subplot) && t.model_number > 0 t.show_text('text' => t.model_number.to_s, 'side' => TOP, 'pos' => pos, 'shift' => shift, 'scale' => 0.8, 'justification' => RIGHT_JUSTIFIED) end end def plot_boundaries(xs,ys,margin,ymin=nil,ymax=nil,reverse_xaxis=false,reverse_yaxis=false) xmin = xs.min xmax = xs.max ymin = ys.min if ymin == nil ymax = ys.max if ymax == nil width = (xmax == xmin)? 1 : xmax - xmin height = (ymax == ymin)? 1 : ymax - ymin left_boundary = xmin - margin * width right_boundary = xmax + margin * width top_boundary = ymax + margin * height bottom_boundary = ymin - margin * height if reverse_xaxis tmp = left_boundary; left_boundary = right_boundary; right_boundary = tmp end if reverse_yaxis tmp = top_boundary; top_boundary = bottom_boundary; bottom_boundary = tmp end return [ left_boundary, right_boundary, top_boundary, bottom_boundary ] end def blues(opts={}) opts = { 'major_grid' => false, 'minor_grid' => false, }.merge(opts) read_data t.title_color = Crimson t.do_box_labels('Blues Plot', 'Position', '\textcolor[rgb]{0,0,1}{Blues}') show_model_number xs = @positions ys = @blues t.show_plot(plot_boundaries(xs,ys,@margin,-1,1)) { t.show_grid(opts) t.show_polyline(xs,ys,Blue) } end def blues_without_padding p t.default_frame_right t.default_frame_right = 1.0 p t.default_frame_right blues end def blues_far_off(opts={}) opts = { 'major_grid' => false, 'minor_grid' => false, }.merge(opts) read_data t.title_color = Crimson t.do_box_labels('(shifted) Blues Plot ' + if t.vincent_or_bill '(Vincent)' else '(Bill)' end, 'Position', '\textcolor[rgb]{0,0,1}{Blues}') show_model_number xs = @positions ys = @blues xs *= 0.1 xs += 100 ys *= 6 t.show_plot(plot_boundaries(xs,ys,@margin,-6,6)) { t.show_grid(opts) t.show_polyline(xs,ys,Blue) } end def blues_axes_1 read_data t.title_color = Crimson t.do_box_labels('Blues Plot --- without top and right', 'Position', '\textcolor[rgb]{0,0,1}{Blues}') show_model_number xs = @positions ys = @blues t.top_edge_visible = false t.right_edge_visible = false t.show_plot(plot_boundaries(xs,ys,@margin,-1,1)) { t.show_polyline(xs,ys,Blue) } end def blues_axes_2 read_data t.title_color = Crimson t.do_box_labels('Blues Plot', 'Position', '\textcolor[rgb]{0,0,1}{Blues}') show_model_number xs = @positions ys = @blues t.top_edge_visible = false t.right_edge_visible = false t.bottom_edge_visible = false t.xaxis_loc = AT_Y_ORIGIN t.show_plot(plot_boundaries(xs,ys,@margin,-1,1)) { t.show_polyline(xs,ys,Blue) } end def blues_reversed_axes # reverse both axes read_data t.do_box_labels('Blues Plot', 'Position', '\textcolor[rgb]{0,0,1}{Blues}') show_model_number xs = @positions ys = @blues t.show_plot(plot_boundaries(xs,ys,@margin,-1,1,true,true)) { t.show_polyline(xs,ys,Blue) } end def blues_with_grid blues('major_grid' => true) end def reds sans_serif_style unless t.in_subplot read_data t.do_box_labels('Reds Plot', 'Position', '\textcolor{Crimson}{Reds}') show_model_number xs = @positions ys = @reds t.show_plot(plot_boundaries(xs,ys,@margin,-1,1)) { t.show_polyline(xs,ys,Red) } end def log_reds read_data t.do_box_labels('loglog Reds Plot', 'Position', '\textcolor[rgb]{1,0,0}{Reds}') show_model_number xs = @positions ys = @reds t.xaxis_log_values = true t.yaxis_log_values = true t.xaxis_numeric_label_frequency = 3 t.show_plot(plot_boundaries(xs,ys,@margin,-1,1)) { t.show_polyline(xs,ys,Red) } end def greens read_data show_model_number t.do_box_labels('Greens Plot', 'Position', '\textcolor[rgb]{0,0.5,0}{Greens}') xs = @positions ys = @greens greens_min = -4 greens_max = -0.01 boundaries = plot_boundaries(xs,ys,@margin,greens_min,greens_max) t.show_plot(boundaries) { t.show_polyline(xs,ys,Green) } end # Same thing as green, but with X/Y clipped: def greens_clipped read_data show_model_number t.do_box_labels('Greens Plot Clipped', 'Position', '\textcolor[rgb]{0,0.5,0}{Greens, clipped}') xs = @positions ys = @greens greens_min = -4 greens_max = -0.01 boundaries = [ 1, 4, 0, -4 ] t.show_plot(boundaries) { t.show_polyline(xs,ys,Green) } end def reds_blues read_data t.set_aspect_ratio(1) t.do_box_labels("Reds and Blues", "Position", "Values") boundaries = setup_lines(@positions, [@blues, @reds], -1, 1) xs = @positions t.show_plot(boundaries) do t.show_polyline(xs,@blues,Blue,'really Blues') t.show_polyline(xs,@reds,Red,'Reds') end show_model_number end def side_by_side read_data t.landscape t.do_box_labels('Side by Side', 'Position', nil) t.subplot('right_margin' => 0.5) { t.yaxis_loc = t.ylabel_side = LEFT; t.right_edge_type = AXIS_LINE_ONLY; reds } t.subplot('left_margin' => 0.5) { t.yaxis_loc = t.ylabel_side = RIGHT; t.left_edge_type = AXIS_LINE_ONLY; greens } show_model_number end def two_yaxes read_data t.do_box_labels('Same X, 2 Different Y\'s', 'Position', nil) t.subplot { t.yaxis_loc = t.ylabel_side = LEFT; t.right_edge_type = AXIS_HIDDEN; reds } t.subplot { t.yaxis_loc = t.ylabel_side = RIGHT; t.left_edge_type = AXIS_HIDDEN; greens } show_model_number end def three_yaxes read_data t.rescale(0.75) t.ylabel_shift = 1.4 t.set_subframe('right_margin' => 0.3, 'top_margin' => 0.1, 'bottom_margin' => 0.1) t.do_box_labels('Same X, 3 Different Y\'s', 'Position', nil) t.subplot { t.yaxis_loc = t.ylabel_side = LEFT; t.right_edge_type = AXIS_HIDDEN reds } t.subplot { t.yaxis_loc = t.ylabel_side = RIGHT; t.left_edge_type = AXIS_HIDDEN greens } t.context do xs = @positions ys = @blues bounds = plot_boundaries(xs,ys.mul(2),@margin,-1,1) left_boundary = bounds[0] right_boundary = bounds[1] top_boundary = bounds[2] bottom_boundary = bounds[3] dx = right_boundary - left_boundary t.set_bounds(bounds) t.show_polyline(xs,ys,Blue) spec = { 'ticks_outside' => false, 'ticks_inside' => true, 'from' => [right_boundary + 0.3*dx, top_boundary], 'to' => [right_boundary + 0.3*dx, bottom_boundary], } t.show_axis(spec) t.yaxis_loc = t.ylabel_side = RIGHT t.ylabel_shift = 5 t.show_ylabel('\textcolor[rgb]{0,0,1}{Blues}') end show_model_number end def two_yaxes_with_legends read_data t.do_box_labels('Same X, Different Y\'s', 'Position', nil) t.legend_text_dy = 1.2 t.show_plot_with_legend('legend_left_margin' => 0.65, 'plot_scale' => 1, 'legend_scale' => 1.3, 'plot_right_margin' => 0) { # NOTE: use subfigure rather than subplot to get combined legend info # subplot would be correct if each had its own legend t.subfigure { t.yaxis_loc = t.ylabel_side = LEFT; t.right_edge_type = AXIS_HIDDEN; t.show_plot(plot_boundaries(@positions,@blues,@margin,-1,1)) { t.show_polyline(@positions,@blues,Blue,'Blues') } } t.subfigure { t.yaxis_loc = t.ylabel_side = RIGHT; t.left_edge_type = AXIS_HIDDEN; t.show_plot(plot_boundaries(@positions,@reds,@margin,-1,1)) { t.show_polyline(@positions,@reds,Red,'Reds') } } } show_model_number end def legend_inside read_data show_model_number t.show_plot_with_legend( 'legend_top_margin' => 0.60, 'legend_left_margin' => 0.05, 'plot_scale' => 1, 'legend_scale' => 1.3, 'legend_background_function' => lambda { |bnds| do_legend_background(bnds) }, 'plot_right_margin' => 0) { reds_blues } end def legend_inside_with_background read_data show_model_number t.show_plot_with_legend( 'legend_left_margin' => 0.5, 'legend_top_margin' => 0.3, 'plot_scale' => 1, 'legend_scale' => 1.3, 'plot_right_margin' => 0) { # First, the plot: reds_blues # Then, a background rectangle for the legend t.context do t.fill_color = White t.fill_opacity = 0.7 t.fill_rect(3.2,-0.2, 2.3, 0.5) end } end def legend_inside_with_background_using_legend_background read_data show_model_number t.measure_legends = true t.show_plot_with_legend( 'legend_left_margin' => 0.5, 'legend_top_margin' => 0.3, 'plot_scale' => 1, 'legend_scale' => 1.3, 'legend_background_function' => lambda { |bnds| do_real_legend_background(bnds) }, 'plot_right_margin' => 0) { reds_blues } end def do_legend_background(bnds) t.fill_color = Teal t.fill_opacity = 0.4 left = bnds[0] right = 0.35 #bnds[1] top = 0.85 # bnds[2] bot = 0.25 #bnds[3] t.fill_rounded_rect(left, bot, right-left, top-bot, 0.03, 0.03) t.stroke_color = Black t.stroke_width = 0.7 t.stroke_rounded_rect(left, bot, right-left, top-bot, 0.03, 0.03) end def do_real_legend_background(bnds) t.fill_color = Teal t.fill_opacity = 0.4 left = bnds[0] right = bnds[1] top = bnds[2] bot = bnds[3] t.fill_rounded_rect(left, bot, right-left, top-bot, 0.03, 0.03) t.stroke_color = Black t.stroke_width = 0.7 t.stroke_rounded_rect(left, bot, right-left, top-bot, 0.03, 0.03) end def legend_outside read_data show_model_number t.show_plot_with_legend('legend_scale' => 1.3) { reds_blues t.show_marker( 'marker' => Circle, 'scale' => 0.5, 'fill_color' => Teal, 'Xs' => [0,2,4,6], 'Ys' => [0.2,-0.1,0.3,-0.5], 'legend' => 'Spots' ) } end def inset read_data show_model_number reds t.subplot([0.25,0.45,0.5,0.1]) do t.rescale(0.5) blues end end def legends read_data t.subfigure('top_margin' => 0.6) do t.rescale(0.5) t.subplot('right_margin' => 0.6) { legend_inside } t.subplot('left_margin' => 0.56) { legend_outside } end end def column_triplets read_data show_model_number column_margin = 0.15 t.rescale(0.5) t.subplot(t.column_margins( 'num_columns' => 3, 'column' => 1, 'column_margin' => column_margin)) { blues } t.subplot(t.column_margins( 'num_columns' => 3, 'column' => 2, 'column_margin' => column_margin)) { reds } t.subplot(t.column_margins( 'num_columns' => 3, 'column' => 3, 'column_margin' => column_margin)) { greens } end def row_triplets read_data show_model_number row_margin = 0.15 t.rescale(0.5) t.subplot(t.row_margins( 'num_rows' => 3, 'row' => 1, 'row_margin' => row_margin)) { blues } t.subplot(t.row_margins( 'num_rows' => 3, 'row' => 2, 'row_margin' => row_margin)) { reds } t.subplot(t.row_margins( 'num_rows' => 3, 'row' => 3, 'row_margin' => row_margin)) { greens } end def rows read_data t.landscape show_model_number t.do_box_labels('Blues, Reds, Greens', 'Position', nil) t.rescale(0.8) num_plots = 3 t.subplot(t.row_margins('num_rows' => num_plots, 'row' => 1)) do t.xaxis_type = AXIS_WITH_TICKS_ONLY blues end t.subplot(t.row_margins('num_rows' => num_plots, 'row' => 2)) do t.xaxis_type = AXIS_WITH_TICKS_ONLY t.top_edge_type = AXIS_HIDDEN reds end t.subplot(t.row_margins('num_rows' => num_plots, 'row' => 3)) do t.top_edge_type = AXIS_HIDDEN greens end end def two_rows read_data t.landscape show_model_number t.do_box_labels('Blues, Reds, Greens', 'Position', nil) t.rescale(0.8) num_plots = 2 t.subplot(t.row_margins('num_rows' => num_plots, 'row' => 1)) do t.xaxis_type = AXIS_WITH_TICKS_ONLY blues end t.subplot(t.row_margins('num_rows' => num_plots, 'row' => 2)) do t.top_edge_type = AXIS_HIDDEN greens end end def columns read_data t.set_aspect_ratio(2) show_model_number t.rescale(0.8) t.do_box_labels('Blues, Reds, Greens', 'Position', 'Values for Colors') num_plots = 3 t.subplot(t.column_margins('num_columns' => 3, 'column' => 1)) do t.right_edge_type = AXIS_HIDDEN blues end t.subplot(t.column_margins('num_columns' => 3, 'column' => 2)) do t.yaxis_type = AXIS_WITH_TICKS_ONLY t.right_edge_type = AXIS_HIDDEN reds end t.subplot(t.column_margins('num_columns' => 3, 'column' => 3)) do t.yaxis_type = AXIS_WITH_TICKS_ONLY t.right_edge_type = AXIS_WITH_TICKS_ONLY greens end end def array read_data show_model_number t.do_box_labels("Array of Plots", "Position", "Values for Colors") row_margin = 0.0; num_rows = 2 column_margin = 0.0; num_columns = 3 t.subplot(t.row_margins('num_rows' => num_rows, 'row' => 1, 'row_margin' => row_margin)) do t.xaxis_type = AXIS_WITH_TICKS_ONLY margins = t.column_margins('num_columns' => num_columns, 'column' => 1, 'column_margin' => column_margin) t.subplot(margins) do t.right_edge_type = AXIS_HIDDEN blues end margins = t.column_margins('num_columns' => num_columns, 'column' => 2, 'column_margin' => column_margin) t.subplot(margins) do t.yaxis_type = AXIS_WITH_TICKS_ONLY t.right_edge_type = AXIS_HIDDEN reds end margins = t.column_margins('num_columns' => num_columns, 'column' => 3, 'column_margin' => column_margin) t.subplot(margins) do t.yaxis_type = AXIS_WITH_TICKS_ONLY t.right_edge_type = AXIS_WITH_TICKS_ONLY blues end end t.subplot(t.row_margins('num_rows' => num_rows, 'row' => 2, 'row_margin' => row_margin)) do t.top_edge_type = AXIS_HIDDEN margins = t.column_margins('num_columns' => num_columns, 'column' => 1, 'column_margin' => column_margin) t.subplot(margins) do t.right_edge_type = AXIS_HIDDEN reds end margins = t.column_margins('num_columns' => num_columns, 'column' => 2, 'column_margin' => column_margin) t.subplot(margins) do t.yaxis_type = AXIS_WITH_TICKS_ONLY t.right_edge_type = AXIS_HIDDEN blues end margins = t.column_margins('num_columns' => num_columns, 'column' => 3, 'column_margin' => column_margin) t.subplot(margins) do t.yaxis_type = AXIS_WITH_TICKS_ONLY t.right_edge_type = AXIS_WITH_TICKS_ONLY reds end end end def setup_lines(xs, yarry, ymin, ymax) margin = 0.1 num_lines = yarry.length return nil unless num_lines > 0 xmin = xs.min xmax = xs.max width = (xmax == xmin)? 1 : xmax - xmin height = (ymax == ymin)? 1 : ymax - ymin return [ xmin - margin * width, xmax + margin * width, ymax + margin * height, ymin - margin * height ] end def trio read_data show_model_number t.rescale(0.6) t.subplot('bottom_margin' => 0.6, 'left_margin' => 0.15, 'right_margin' => 0.15) { rows } t.subplot('top_margin' => 0.60) { side_by_side } end def collage read_data show_model_number t.rescale(0.5) t.no_xlabel t.line_width = 0.5 t.title_shift = 0.5 t.subplot( 'left_margin' => 0.1, 'right_margin' => 0.5, 'bottom_margin' => 0.75) { side_by_side } t.yaxis_loc = RIGHT t.subplot( 'left_margin' => 0.75, 'right_margin' => 0.07, 'top_margin' => 0.2, 'bottom_margin' => 0.55) { t.ylabel_shift = 0.3; reds } t.yaxis_loc = LEFT t.subplot( 'right_margin' => 0.7, 'top_margin' => 0.4) { rows } t.subplot( 'left_margin' => 0.5, 'top_margin' => 0.73) { legend_outside } t.subplot( 'left_margin' => 0.4, 'right_margin' => 0.35, 'top_margin' => 0.36, 'bottom_margin' => 0.4) { t.no_ylabel; blues } end def special_y read_data show_model_number t.ylabel_shift += 1.2 t.do_box_labels('Special Y Axis', 'Position', "Y Values") t.yaxis_numeric_label_angle = -90 t.yaxis_locations_for_major_ticks = [ -10.0, -6.0, -PI, 0.0, PI, 6.0, 10.0 ] yzero = '\begin{minipage}[t][1cm][c]{1cm}\begin{flushright}' + '\footnotesize Name this Point' + ' \end{flushright} \end{minipage}' t.yaxis_tick_labels = [ '-IX', '-VI', '$-\pi$', yzero, '$\pi$', 'VI', 'IX' ] t.yaxis_type = AXIS_WITH_MAJOR_TICKS_AND_NUMERIC_LABELS t.stroke_color = Blue t.yaxis_numeric_label_justification = RIGHT_JUSTIFIED t.right_edge_type = AXIS_WITH_MAJOR_TICKS_ONLY t.yaxis_numeric_label_shift = 0 xs = @positions ys = @big_blues t.show_plot(plot_boundaries(xs,ys,@margin,-11, 11)) do t.append_points_to_path(xs,ys) t.stroke end end def do_labels_plot(title,&cmd) t.do_box_labels(title, "Position", "Blues") t.stroke_color = Blue read_data xs = @positions ys = @blues t.stroke_color = Blue num_pts = @positions.length @pt1 = (num_pts * 0.25 - t.model_number).round.mod(num_pts) @pt2 = (num_pts * 0.75 - t.model_number).round.mod(num_pts) t.show_plot(plot_boundaries(xs,ys,@margin,-1,1)) do t.append_points_to_path(xs,ys) t.stroke cmd.call end show_model_number end def do_label(xs, ys, pt, fmt, color, marker_scale, text_scale) x = xs[pt]; y = ys[pt] shift = text_scale * t.default_text_height_dy * 0.8 shift = -shift if t.yaxis_reversed t.show_label('text' => sprintf(fmt, x, y), 'x' => x, 'y' => y + shift, 'justification' => CENTERED, 'color' => color, 'scale' => text_scale) t.show_marker('marker' => Bullet, 'point' => [x,y], 'scale' => marker_scale, 'color' => color) end def labels xs = @positions ys = @blues do_labels_plot("Labels on Points") do do_label(xs, ys, @pt1, "(%0.2f, %0.2f)", Red, 0.5, 0.6) do_label(xs, ys, @pt2, "(%0.2f, %0.2f)", Red, 0.5, 0.6) end end def error_bars xs = @positions ys = @blues do_labels_plot("Error Bars on Points") do t.show_error_bars('x' => xs[@pt2], 'y' => ys[@pt2], 'dx' => 0.2, 'dy' => 0.12) t.show_error_bars('x' => xs[@pt1], 'y' => ys[@pt1], 'dx' => 0.23, 'dy' => 0.15, 'color' => Crimson) end end def error_bars_y xs = @positions ys = @blues do_labels_plot("Error Bars on Points") do t.show_error_bars('x' => xs[@pt2], 'y' => ys[@pt2], 'dy' => 0.12) t.show_error_bars('x' => xs[@pt1], 'y' => ys[@pt1], 'dy' => 0.15, 'color' => Crimson) end end def error_bars_x xs = @positions ys = @blues do_labels_plot("Error Bars on Points") do t.show_error_bars('x' => xs[@pt2], 'y' => ys[@pt2], 'dx' => 0.2) t.show_error_bars('x' => xs[@pt1], 'y' => ys[@pt1], 'dx' => 0.23, 'color' => Crimson) end end def error_bars2 t.do_box_labels("Another Example of Error Bars", "Number of Particles", "Crossing Times") xs = Dvector[3, 4, 5, 6, 7, 8, 9, 12] ys = Dvector[72, 10, 17, 20, 15.5, 11, 11.5, 11.5] deltaT = Dvector[13, 2.8, 3, 3.5, 2.7, 1.9, 2, 2] t.show_plot([0, 15, 100, 0]) do t.stroke_color = Blue t.show_polyline(xs, ys) xs.size.times { |i| t.show_error_bars('x' => xs[i], 'y' => ys[i], 'dx' => 0, 'dy' => deltaT[i]) } t.show_marker('Xs' => xs, 'Ys' => ys, 'marker' => Bullet, 'scale' => 0.3, 'color' => Red); end end def arrows xs = @positions ys = @blues do_labels_plot("Arrows") do pt3 = (3*@pt1+@pt2)/4 pt4 = (@pt1+3*@pt2)/4 pt5 = (@pt1+6*@pt2)/7 mid_pt = (@pt1+@pt2)/2 t.show_arrow('head' => [xs[@pt1], ys[@pt1]], 'tail' => [xs[pt3], ys[pt3]], 'color' => Crimson) t.show_arrow('head' => [xs[mid_pt], ys[mid_pt]], 'tail' => [xs[pt4], ys[pt4]], 'color' => Lime) t.show_arrow('head' => [xs[@pt2], ys[@pt2]], 'tail' => [xs[pt5], ys[pt5]], 'color' => Magenta) end end def sampled_splines t.do_box_labels("Sampled Splines", "Position", "Average Count") xs = Dvector[ 1.0, 2.0, 5.0, 6.0, 7.0, 8.0, 10.0, 13.0, 17.0 ] ys = Dvector[ 3.0, 3.7, 3.9, 4.2, 5.7, 6.6, 7.1, 6.7, 4.5 ] data_pts = xs.size x_first = 0.0; x_last = 18.0; y_first = y_last = 2.5 x_results = Dvector[] y_results = Dvector[] stps = t.make_steps( 'dest_xs' => x_results, 'dest_ys' => y_results, 'xs' => xs, 'ys' => ys, 'x_first' => x_first, 'y_first' => y_first, 'x_last' => x_last, 'y_last' => y_last) t.show_plot([-1, 19, 8, 2]) do t.fill_color = FloralWhite t.fill_frame smooth_pts = 4*(data_pts-1) + 1 dx = (xs[data_pts-1] - xs[0])/(smooth_pts-1) sample_xs = Dvector.new(smooth_pts) { |i| i*dx + xs[0] } result_ys = t.make_spline_interpolated_points( 'sample_xs' => sample_xs, 'xs' => xs, 'ys' => ys, 'start_slope' => 2.5*(ys[1]-ys[0])/(xs[1]-xs[0])) t.stroke_color = Blue t.append_points_to_path(sample_xs, result_ys) t.stroke t.show_marker('Xs' => sample_xs, 'Ys' => result_ys, 'marker' => Bullet, 'scale' => 0.4, 'color' => Green); t.show_marker('Xs' => xs, 'Ys' => ys, 'marker' => Bullet, 'scale' => 0.6, 'color' => Red); end end def sampled_pm_cubics t.do_box_labels("Sampled pm cubics", "Position", "Average Count") xs = Dvector[ 1.0, 2.0, 5.0, 6.0, 7.0, 8.0, 10.0, 13.0, 17.0 ] ys = Dvector[ 3.0, 3.7, 3.9, 4.2, 5.7, 6.6, 7.1, 6.7, 4.5 ] data_pts = xs.size csdata_interpolant = Dvector.create_pm_cubic_interpolant(xs, ys) smooth_pts = 4*(data_pts-1) + 1 dx = (xs[data_pts-1] - xs[0])/(smooth_pts-1) pm_xs = Dvector.new(smooth_pts) { |i| i*dx + xs[0] } pm_ys = pm_xs.dup pm_ys.length.times {|i| pm_ys[i] = Dvector.pm_cubic_interpolate(pm_xs[i], csdata_interpolant) } t.show_plot([-1, 19, 8, 2]) do t.fill_color = FloralWhite t.fill_frame t.stroke_color = Blue t.append_points_to_path(pm_xs, pm_ys) t.stroke t.show_marker('Xs' => pm_xs, 'Ys' => pm_ys, 'marker' => Bullet, 'scale' => 0.4, 'color' => Green); t.show_marker('Xs' => xs, 'Ys' => ys, 'marker' => Bullet, 'scale' => 0.6, 'color' => Red); end end def steps t.do_box_labels("Steps", "Position", "Average Count") xs = Dvector[ 1.0, 2.0, 5.0, 6.0, 7.0, 8.0, 10.0, 13.0, 17.0 ] ys = Dvector[ 3.0, 3.7, 3.9, 4.2, 5.7, 6.6, 7.1, 6.7, 4.5 ] data_pts = xs.size x_first = 0.0; x_last = 18.0; y_first = y_last = 2.5 x_results = Dvector[0] y_results = Dvector[0] stps = t.make_steps( 'xs' => xs, 'ys' => ys, 'x_first' => x_first, 'y_first' => y_first, 'x_last' => x_last, 'y_last' => y_last) t.show_plot([-1, 19, 8, 2]) do t.fill_color = FloralWhite t.fill_frame t.stroke_color = Blue t.append_points_to_path(stps[0], stps[1]) t.stroke t.show_marker('Xs' => xs, 'Ys' => ys, 'marker' => Bullet, 'scale' => 0.6, 'color' => Red); end end def bezier # append bezier curves t.do_box_labels("bezier curves", "Position", "Average Count") xs = Dvector[ 1.0, 2.0, 5.0, 6.0, 7.0, 8.0, 10.0, 13.0, 17.0 ] ys = Dvector[ 3.0, 3.7, 3.9, 4.2, 5.7, 6.6, 7.1, 6.7, 4.5 ] scale = Dvector[ 0.6, 0.9 ] t.show_plot([-1, 19, 8, 2]) do t.fill_color = FloralWhite t.fill_frame start_slope = 2.5*(ys[1]-ys[0])/(xs[1]-xs[0]) interp = t.make_interpolant( 'xs' => xs, 'ys' => ys, 'start_slope' => start_slope) t.append_interpolant_to_path(interp) t.stroke_color = Black t.stroke t.show_marker('Xs' => xs, 'Ys' => ys, 'marker' => [ Bullet, Star ], 'scale' => scale, 'color' => [ Red, Blue, Teal ]); end end def sampled_image(title, colormap = nil) t.do_box_labels(title, 'Log Density', 'Log Temperature') data = get_press_image xs = @eos_logRHOs ys = @eos_logTs colormap = t.mellow_colormap if colormap == nil t.show_plot([@eos_xmin, @eos_xmax, @eos_ymax, @eos_ymin]) do t.fill_color = Wheat t.fill_frame clip_press_image t.show_image( 'll' => [xs.min, ys.min], 'lr' => [xs.max, ys.min], 'ul' => [xs.min, ys.max], 'color_space' => colormap, 'data' => data, 'value_mask' => 255, 'w' => @eos_data_xlen, 'h' => @eos_data_ylen) end end def sampled_image_with_grid(title, colormap = nil) t.do_box_labels(title, 'Log Density', 'Log Temperature') data = get_press_image xs = @eos_logRHOs ys = @eos_logTs colormap = t.mellow_colormap if colormap == nil t.show_plot([@eos_xmin, @eos_xmax, @eos_ymax, @eos_ymin]) do t.fill_color = Wheat t.fill_frame t.context do clip_press_image t.show_image( 'll' => [xs.min, ys.min], 'lr' => [xs.max, ys.min], 'ul' => [xs.min, ys.max], 'color_space' => colormap, 'data' => data, 'value_mask' => 255, 'w' => @eos_data_xlen, 'h' => @eos_data_ylen) end t.show_grid('stroke_color' => Teal, 'rescale_lines' => 0.5, 'major_grid' => true) end end def color_bar(ylabel, levels = nil) xmin = 0; xmax = 1; xmid = 0.5 t.rescale(0.8) t.xaxis_type = AXIS_LINE_ONLY t.xaxis_loc = BOTTOM t.top_edge_type = AXIS_LINE_ONLY t.yaxis_loc = t.ylabel_side = RIGHT t.yaxis_type = AXIS_WITH_TICKS_AND_NUMERIC_LABELS t.left_edge_type = AXIS_WITH_TICKS_ONLY t.ylabel_shift += 0.5 t.yaxis_major_tick_length *= 0.6 t.yaxis_minor_tick_length *= 0.5 t.show_ylabel(ylabel); t.no_ylabel t.show_plot('boundaries' => [xmin, xmax, @image_zmax, @image_zmin]) do t.axial_shading( 'start_point' => [xmid, @image_zmin], 'end_point' => [xmid, @image_zmax], 'colormap' => t.mellow_colormap ) if levels != nil t.stroke_color = Gray t.line_width = 1.5 levels.each { |level| t.stroke_line(xmin, level, xmax, level) } end end end def sampled_data t.rescale(0.8) title = 'Log Opacity' t.subplot('right_margin' => @image_right_margin) { sampled_image(title) } t.subplot('left_margin' => 0.95, 'top_margin' => 0.05, 'bottom_margin' => 0.05) { color_bar(title) } end def sampled_with_grid t.rescale(0.8) title = 'Log Opacity' t.subplot('right_margin' => @image_right_margin) { sampled_image_with_grid(title) } t.subplot('left_margin' => 0.95, 'top_margin' => 0.05, 'bottom_margin' => 0.05) { color_bar(title) } end def read_press_image_data return unless @opacity_data == nil @eos_xmin = -8.5; @eos_xmax = 2.5 @eos_ymin = 5.7; @eos_ymax = 7.0 @image_zmin = -3 @image_zmax = 6 data = Dvector.read("data/logRHOs_for_EoS.data") @eos_logRHOs = data[0] @eos_data_xlen = @eos_logRHOs.size @eos_xmin = @eos_logRHOs.min; @eos_xmax = @eos_logRHOs.max data = Dvector.read("data/logTs_for_EoS.data") @eos_logTs = data[0] @eos_data_ylen = @eos_logTs.size @eos_ymin = @eos_logTs.min; @eos_ymax = @eos_logTs.max @opacity_data = Dtable.new(@eos_data_xlen, @eos_data_ylen) @opacity_data.read("data/Opacity_EoS.data") end def get_press_image read_press_image_data return t.create_image_data( @opacity_data, 'min_value' => @image_zmin, 'max_value' => @image_zmax, 'masking' => true) end def clip_press_image t.move_to_point(t.bounds_left, t.bounds_bottom) t.append_point_to_path(t.bounds_left, 4.2) t.append_point_to_path(-3, t.bounds_top) t.append_point_to_path(t.bounds_right, t.bounds_top) t.append_point_to_path(t.bounds_right, t.bounds_bottom) t.close_path t.clip end def samples_with_contours t.rescale(0.8) title = 'Log Opacity' levels = Array.new (0..5).each { |i| levels << i + 0.4 } t.subplot('right_margin' => @image_right_margin) { sampled_image(title) } t.subplot('left_margin' => 0.95, 'top_margin' => 0.05, 'bottom_margin' => 0.05) { color_bar(title, levels) } t.subplot('right_margin' => @image_right_margin) do t.xaxis_type = t.yaxis_type = AXIS_WITH_TICKS_ONLY t.no_title; t.no_xlabel; t.no_ylabel bounds = [@eos_xmin, @eos_xmax, @eos_ymax, @eos_ymin] t.show_plot(bounds) do clip_press_image t.stroke_color = SlateGray t.line_width = 1 gaps = Array.new dict = { 'gaps' => gaps, 'xs' => @eos_logRHOs, 'ys' => @eos_logTs, 'data' => @opacity_data } levels.each do |level| dict['level'] = level pts_array = t.make_contour(dict) t.append_points_with_gaps_to_path(pts_array[0], pts_array[1], gaps, true) t.stroke end end end end # This plot is to demonstrate the new power of #show_axis. def axes_fun t.do_box_labels("Funny axes", "", "") t.xaxis_labels_color = [0.3,0.5,0.3] t.yaxis_labels_color = [0.7,0.2,0.4] t.show_plot([-1, 19, 8, 2]) do spec = { 'from' => [3,3], 'to' => [3,7], 'ticks_outside' => true, 'ticks_inside' => false, 'log' => true } t.show_axis(spec) p t.axis_information(RIGHT) spec2 = { 'from' => [12,3], 'to' => [12,7], 'ticks_outside' => true, 'ticks_inside' => false, 'major_ticks' => [4,6], 'labels' => ["$a$", "$b$"], 'color' => [0,0.8,0.3] } t.show_axis(spec2) spec3 = { 'from' => [17,3], 'to' => [17,7], 'ticks_outside' => true, 'ticks_inside' => false, 'shift' => 1.3, 'labels' => ["$a$", "$b$", 'c', 'd', 'e'], 'stroke_color' => [0.2,0.4,0.7], 'labels_color' => [0.3,0.2,0.4] } t.show_axis(spec3) end t.context do t.set_bounds([-1, 19, 8, 2]) spec = { 'from' => [4,1.2], 'to' => [12,1.2], 'major_ticks' => [ 4.5, 8.8] } t.show_axis(spec) end end end MyPlots.new