konpeito 0.1.0

data/lib/konpeito/ui/chart/scatter_chart.rb

@@ -0,0 +1,303 @@
+# ScatterChart - XY scatter plot with point markers
+# Supports multiple series, hover detection, value labels
+
+class ScatterChart < BaseChart
+  def initialize(x_data, y_data, series_names)
+    super()
+    @x_data = x_data          # Array of Array[Float] (one per series)
+    @y_data = y_data          # Array of Array[Float] (one per series)
+    @series_names = series_names
+    @point_radius = 5.0
+    @show_grid = true
+    @data_x_min = 0.0
+    @data_x_max = 1.0
+    @data_y_min = 0.0
+    @data_y_max = 1.0
+    @x_scale = nil
+    @y_scale = nil
+    @x_ticks = nil
+    @y_ticks = nil
+    @num_series = 0
+    @chart_px = 0.0
+    @chart_py = 0.0
+    @chart_pw = 0.0
+    @chart_ph = 0.0
+    compute_scatter_range
+  end
+
+  def point_radius(r)
+    @point_radius = r
+    self
+  end
+
+  def show_grid(v)
+    @show_grid = v
+    self
+  end
+
+  def set_data(x_data, y_data, series_names)
+    @x_data = x_data
+    @y_data = y_data
+    @series_names = series_names
+    compute_scatter_range
+    mark_dirty
+    update
+  end
+
+  def render_chart(painter, px, py, pw, ph)
+    return if @x_data.length == 0
+    @chart_px = px
+    @chart_py = py
+    @chart_pw = pw
+    @chart_ph = ph
+    @num_series = @x_data.length
+    setup_scatter_scales
+    draw_scatter_grid(painter)
+    draw_y_axis(painter, px, py, pw, ph, @y_ticks, @y_scale)
+    draw_x_axis_line(painter, px, py, pw, ph)
+    draw_scatter_x_ticks(painter)
+    draw_all_scatter_points(painter)
+    draw_scatter_legend(painter, px, py)
+  end
+
+  def setup_scatter_scales
+    @x_ticks = compute_ticks(@data_x_min, @data_x_max, 5)
+    @y_ticks = compute_ticks(@data_y_min, @data_y_max, 5)
+    x_min = @data_x_min
+    x_max = @data_x_max
+    y_min = @data_y_min
+    y_max = @data_y_max
+    if @x_ticks.length > 0
+      x_min = @x_ticks[0]
+      x_max = @x_ticks[@x_ticks.length - 1]
+    end
+    if @y_ticks.length > 0
+      y_min = @y_ticks[0]
+      y_max = @y_ticks[@y_ticks.length - 1]
+    end
+    bottom = @chart_py + @chart_ph
+    right = @chart_px + @chart_pw
+    @x_scale = LinearScale.new(x_min, x_max, @chart_px, right)
+    @y_scale = LinearScale.new(y_min, y_max, bottom, @chart_py)
+  end
+
+  def draw_scatter_grid(painter)
+    return if !@show_grid
+    bc = $theme.border
+    gc = painter.with_alpha(bc, 40)
+    # Vertical grid lines at x ticks
+    i = 0
+    while i < @x_ticks.length
+      xx = @x_scale.map(@x_ticks[i])
+      bottom = @chart_py + @chart_ph
+      painter.draw_line(xx, @chart_py, xx, bottom, gc, 1.0)
+      i = i + 1
+    end
+  end
+
+  def draw_scatter_x_ticks(painter)
+    lc = $theme.text_secondary
+    i = 0
+    while i < @x_ticks.length
+      xx = @x_scale.map(@x_ticks[i])
+      label = painter.number_to_string(@x_ticks[i])
+      lw = painter.measure_text_width(label, $theme.font_family, 11.0)
+      ascent = painter.get_text_ascent($theme.font_family, 11.0)
+      label_y = @chart_py + @chart_ph + 14.0 + ascent
+      painter.draw_text(label, xx - lw / 2.0, label_y, $theme.font_family, 11.0, lc)
+      i = i + 1
+    end
+  end
+
+  def draw_all_scatter_points(painter)
+    si = 0
+    while si < @num_series
+      c = series_color(si)
+      draw_scatter_series(painter, si, c)
+      si = si + 1
+    end
+  end
+
+  def draw_scatter_series(painter, si, c)
+    xs = @x_data[si]
+    ys = @y_data[si]
+    n = xs.length
+    if ys.length < n
+      n = ys.length
+    end
+    j = 0
+    while j < n
+      xx = @x_scale.map(xs[j])
+      yy = @y_scale.map(ys[j])
+      r = @point_radius
+      if @hover_series == si
+        if @hover_index == j
+          r = @point_radius + 3.0
+        end
+      end
+      painter.fill_circle(xx, yy, r, c)
+      if @hover_series == si
+        if @hover_index == j
+          draw_scatter_label(painter, xs[j], ys[j], xx, yy, r)
+        end
+      end
+      j = j + 1
+    end
+  end
+
+  def draw_scatter_label(painter, xv, yv, xx, yy, r)
+    xl = painter.number_to_string(xv)
+    yl = painter.number_to_string(yv)
+    label = "(" + xl + ", " + yl + ")"
+    lw = painter.measure_text_width(label, $theme.font_family, 10.0)
+    tc = $theme.text_primary
+    painter.draw_text(label, xx - lw / 2.0, yy - r - 4.0, $theme.font_family, 10.0, tc)
+  end
+
+  def draw_scatter_legend(painter, px, py)
+    if @show_legend
+      if @series_names.length > 1
+        colors = []
+        si = 0
+        while si < @series_names.length
+          colors << series_color(si)
+          si = si + 1
+        end
+        draw_legend(painter, @series_names, colors, px + 8.0, py - 20.0)
+      end
+    end
+  end
+
+  def update_hover
+    return if @x_scale == nil
+    mx = @mouse_x
+    my = @mouse_y
+    px = plot_x
+    py = plot_y
+    pw = plot_w
+    ph = plot_h
+    if mx < px
+      @hover_index = -1
+      @hover_series = -1
+      return
+    end
+    if mx > px + pw
+      @hover_index = -1
+      @hover_series = -1
+      return
+    end
+    if my < py
+      @hover_index = -1
+      @hover_series = -1
+      return
+    end
+    if my > py + ph
+      @hover_index = -1
+      @hover_series = -1
+      return
+    end
+    find_nearest_scatter_point(mx, my)
+  end
+
+  def find_nearest_scatter_point(mx, my)
+    best_si = -1
+    best_j = -1
+    best_dist = 999999.0
+    si = 0
+    while si < @num_series
+      xs = @x_data[si]
+      ys = @y_data[si]
+      n = xs.length
+      if ys.length < n
+        n = ys.length
+      end
+      j = 0
+      while j < n
+        xx = @x_scale.map(xs[j])
+        yy = @y_scale.map(ys[j])
+        dx = mx - xx
+        dy = my - yy
+        d = dx * dx + dy * dy
+        if d < best_dist
+          best_dist = d
+          best_si = si
+          best_j = j
+        end
+        j = j + 1
+      end
+      si = si + 1
+    end
+    threshold = (@point_radius + 8.0) * (@point_radius + 8.0)
+    if best_dist < threshold
+      @hover_series = best_si
+      @hover_index = best_j
+    else
+      @hover_series = -1
+      @hover_index = -1
+    end
+  end
+
+  private
+
+  def compute_scatter_range
+    @data_x_min = 0.0
+    @data_x_max = 1.0
+    @data_y_min = 0.0
+    @data_y_max = 1.0
+    first = true
+    si = 0
+    while si < @x_data.length
+      xs = @x_data[si]
+      ys = @y_data[si]
+      n = xs.length
+      if ys.length < n
+        n = ys.length
+      end
+      j = 0
+      while j < n
+        xv = xs[j]
+        yv = ys[j]
+        if first
+          @data_x_min = xv
+          @data_x_max = xv
+          @data_y_min = yv
+          @data_y_max = yv
+          first = false
+        else
+          if xv < @data_x_min
+            @data_x_min = xv
+          end
+          if xv > @data_x_max
+            @data_x_max = xv
+          end
+          if yv < @data_y_min
+            @data_y_min = yv
+          end
+          if yv > @data_y_max
+            @data_y_max = yv
+          end
+        end
+        j = j + 1
+      end
+      si = si + 1
+    end
+    xr = @data_x_max - @data_x_min
+    yr = @data_y_max - @data_y_min
+    if xr > 0.0
+      @data_x_max = @data_x_max + xr * 0.05
+      @data_x_min = @data_x_min - xr * 0.05
+    else
+      @data_x_max = @data_x_max + 1.0
+    end
+    if yr > 0.0
+      @data_y_max = @data_y_max + yr * 0.05
+      @data_y_min = @data_y_min - yr * 0.05
+    else
+      @data_y_max = @data_y_max + 1.0
+    end
+  end
+end
+
+def ScatterChart(x_data, y_data, series_names)
+  ScatterChart.new(x_data, y_data, series_names)
+end

data/lib/konpeito/ui/chart/stacked_bar_chart.rb

@@ -0,0 +1,276 @@
+# StackedBarChart - stacked bar chart widget
+# Supports multiple series stacked on top of each other, hover highlight
+
+class StackedBarChart < BaseChart
+  def initialize(categories, series_data, series_names)
+    super()
+    @categories = categories
+    @series_data = series_data
+    @series_names = series_names
+    @show_values = false
+    @bar_radius = 0.0
+    @data_min = 0.0
+    @data_max = 0.0
+    @y_scale = nil
+    @y_ticks = nil
+    @band = nil
+    @bar_w = 2.0
+    @baseline_y = 0.0
+    @num_series = 0
+    @num_cats = 0
+    @y_range_min = 0.0
+    @y_range_max = 1.0
+    @chart_px = 0.0
+    @chart_py = 0.0
+    @chart_pw = 0.0
+    @chart_ph = 0.0
+    compute_stacked_range
+  end
+
+  def show_values(v)
+    @show_values = v
+    self
+  end
+
+  def bar_radius(r)
+    @bar_radius = r
+    self
+  end
+
+  def set_data(categories, series_data, series_names)
+    @categories = categories
+    @series_data = series_data
+    @series_names = series_names
+    compute_stacked_range
+    mark_dirty
+    update
+  end
+
+  def render_chart(painter, px, py, pw, ph)
+    return if @categories.length == 0
+    return if @series_data.length == 0
+    @chart_px = px
+    @chart_py = py
+    @chart_pw = pw
+    @chart_ph = ph
+    @num_series = @series_data.length
+    @num_cats = @categories.length
+    setup_stacked_ticks
+    setup_stacked_y_scale
+    setup_stacked_band
+    @baseline_y = @y_scale.map(0.0)
+    bmax = @chart_py + @chart_ph
+    if @baseline_y > bmax
+      @baseline_y = bmax
+    end
+    draw_y_axis(painter, px, py, pw, ph, @y_ticks, @y_scale)
+    draw_x_axis_line(painter, px, py, pw, ph)
+    draw_stacked_cat_labels(painter)
+    draw_all_stacked_bars(painter)
+    draw_stacked_legend(painter, px, py)
+  end
+
+  def setup_stacked_ticks
+    @y_ticks = compute_ticks(@data_min, @data_max, 5)
+    @y_range_min = @data_min
+    @y_range_max = @data_max
+    if @y_ticks.length > 0
+      @y_range_min = @y_ticks[0]
+      @y_range_max = @y_ticks[@y_ticks.length - 1]
+    end
+  end
+
+  def setup_stacked_y_scale
+    bottom = @chart_py + @chart_ph
+    @y_scale = LinearScale.new(@y_range_min, @y_range_max, bottom, @chart_py)
+  end
+
+  def setup_stacked_band
+    right = @chart_px + @chart_pw
+    nc = @num_cats * 1.0
+    @band = BandScale.new(nc, @chart_px, right, 12.0)
+    @bar_w = @band.band_width
+  end
+
+  def draw_stacked_cat_labels(painter)
+    lc = $theme.text_secondary
+    i = 0
+    while i < @num_cats
+      i_f = i * 1.0
+      bx = @band.map(i_f)
+      label = @categories[i]
+      lw = painter.measure_text_width(label, $theme.font_family, 11.0)
+      ascent = painter.get_text_ascent($theme.font_family, 11.0)
+      label_x = bx + @bar_w / 2.0 - lw / 2.0
+      label_y = @chart_py + @chart_ph + 14.0 + ascent
+      painter.draw_text(label, label_x, label_y, $theme.font_family, 11.0, lc)
+      i = i + 1
+    end
+  end
+
+  def draw_all_stacked_bars(painter)
+    ci = 0
+    while ci < @num_cats
+      draw_stacked_column(painter, ci)
+      ci = ci + 1
+    end
+  end
+
+  def draw_stacked_column(painter, ci)
+    ci_f = ci * 1.0
+    bx = @band.map(ci_f)
+    cumulative = 0.0
+    si = 0
+    while si < @num_series
+      val = @series_data[si][ci]
+      top_val = cumulative + val
+      bar_bottom = @y_scale.map(cumulative)
+      bar_top = @y_scale.map(top_val)
+      bar_h = bar_bottom - bar_top
+      if bar_h < 0.0
+        bar_h = 0.0 - bar_h
+        bar_top = bar_bottom
+      end
+      c = series_color(si)
+      if @hover_index == ci
+        if @hover_series == si
+          c = painter.lighten_color(c, 0.3)
+        end
+      end
+      painter.fill_round_rect(bx, bar_top, @bar_w, bar_h, @bar_radius, c)
+      if @show_values
+        if val > 0.0
+          vl = painter.number_to_string(val)
+          vw = painter.measure_text_width(vl, $theme.font_family, 9.0)
+          ascent = painter.get_text_ascent($theme.font_family, 9.0)
+          vx = bx + @bar_w / 2.0 - vw / 2.0
+          vy = bar_top + bar_h / 2.0 + ascent / 2.0
+          painter.draw_text(vl, vx, vy, $theme.font_family, 9.0, 4294967295)
+        end
+      end
+      cumulative = top_val
+      si = si + 1
+    end
+  end
+
+  def draw_stacked_legend(painter, px, py)
+    if @show_legend
+      if @series_names.length > 1
+        colors = []
+        si = 0
+        while si < @series_names.length
+          colors << series_color(si)
+          si = si + 1
+        end
+        draw_legend(painter, @series_names, colors, px + 8.0, py - 20.0)
+      end
+    end
+  end
+
+  def update_hover
+    return if @band == nil
+    mx = @mouse_x
+    my = @mouse_y
+    px = plot_x
+    py = plot_y
+    pw = plot_w
+    ph = plot_h
+    if mx < px
+      @hover_index = -1
+      @hover_series = -1
+      return
+    end
+    if mx > px + pw
+      @hover_index = -1
+      @hover_series = -1
+      return
+    end
+    if my < py
+      @hover_index = -1
+      @hover_series = -1
+      return
+    end
+    if my > py + ph
+      @hover_index = -1
+      @hover_series = -1
+      return
+    end
+    find_stacked_hover(mx, my)
+  end
+
+  def find_stacked_hover(mx, my)
+    @hover_index = -1
+    @hover_series = -1
+    ci = 0
+    while ci < @num_cats
+      ci_f = ci * 1.0
+      bx = @band.map(ci_f)
+      if mx >= bx
+        if mx <= bx + @bar_w
+          # Found the category, now find which series segment
+          cumulative = 0.0
+          si = 0
+          while si < @num_series
+            val = @series_data[si][ci]
+            top_val = cumulative + val
+            bar_bottom = @y_scale.map(cumulative)
+            bar_top = @y_scale.map(top_val)
+            if bar_top > bar_bottom
+              tmp = bar_top
+              bar_top = bar_bottom
+              bar_bottom = tmp
+            end
+            if my >= bar_top
+              if my <= bar_bottom
+                @hover_index = ci
+                @hover_series = si
+                return
+              end
+            end
+            cumulative = top_val
+            si = si + 1
+          end
+          @hover_index = ci
+          return
+        end
+      end
+      ci = ci + 1
+    end
+  end
+
+  private
+
+  def compute_stacked_range
+    @data_min = 0.0
+    @data_max = 1.0
+    return if @series_data.length == 0
+    return if @categories.length == 0
+    max_stack = 0.0
+    ci = 0
+    while ci < @categories.length
+      stack_total = 0.0
+      si = 0
+      while si < @series_data.length
+        if ci < @series_data[si].length
+          stack_total = stack_total + @series_data[si][ci]
+        end
+        si = si + 1
+      end
+      if stack_total > max_stack
+        max_stack = stack_total
+      end
+      ci = ci + 1
+    end
+    @data_max = max_stack
+    range = @data_max - @data_min
+    if range > 0.0
+      @data_max = @data_max + range * 0.1
+    else
+      @data_max = @data_min + 1.0
+    end
+  end
+end
+
+def StackedBarChart(categories, series_data, series_names)
+  StackedBarChart.new(categories, series_data, series_names)
+end