chart_fun 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 51d64455dae1b9a4e9f07726a206a085c165bdfbabbccee76edd7d6b10753f6d
+  data.tar.gz: 16b5e4f0feb4d5a36e6595d2e77a04c71984a8349bac378d5f2c33dd7d6fab86
+SHA512:
+  metadata.gz: 49095ffe16fcdb7d464fe1f4990dab6c3bed4f48d85d95a149e6e16e0cab2396ebb01767902827275df0ec8afdceeb5313ecd0bd4ee634e0d83965c5a44c77e5
+  data.tar.gz: c8734a4850a64172a20e3262fffcf5f6a8487343ac6b4638ed33ecc35d46eb3693637f1cb16342bd37d64e95782dabc596a58ad55fbc063a2004ac1ff18479cc

data/chart_fun.rb

@@ -0,0 +1,5 @@
+require "erubi"
+require 'date'
+require_relative 'src/plot'
+require_relative 'src/data_value'
+require_relative 'src/axis'

data/src/axis.rb

@@ -0,0 +1,55 @@
+# frozen_string_literal: true
+
+class Axis
+  attr_reader :title, :start_point, :end_point, :grid_lines
+
+  def initialize(min, max, start_point, end_point, count, title, grid_lines)
+    @min = min
+    @max = max
+    @start_point = start_point
+    @end_point = end_point
+    @type = min.type
+    @count = count
+    @title = title
+    @grid_lines = grid_lines
+  end
+
+  def labels
+    (0...@count).map do |i|
+      value = interpolate(i, 0, @count - 1, @min.value, @max.value)
+      {
+        text: label_string(value),
+        x: perpendicular[0] * 25 + interpolate(i, 0, @count - 1, @start_point[:x], @end_point[:x]),
+        y: perpendicular[1] * 25 + interpolate(i, 0, @count - 1, @start_point[:y], @end_point[:y])
+      }
+    end
+  end
+
+  def label_string(x)
+    case @type
+    when :date
+      Time.at(x).strftime("%Y-%m-%d")
+    else
+      x.to_s
+    end
+  end
+
+  def title_pos
+    {
+      x: perpendicular[0] * 75 + (@start_point[:x] + (@end_point[:x] - @start_point[:x]) / 2),
+      y: perpendicular[1] * 75 + (@start_point[:y] + (@end_point[:y] - @start_point[:y]) / 2)
+    }
+  end
+
+  private
+
+  def perpendicular
+    direction = [@start_point[:x] - @end_point[:x], @start_point[:y] - @end_point[:y]]
+    perpendicular = [direction[1], -direction[0]]
+    perpendicular.map { |i| i / Math.sqrt(perpendicular[0] ** 2 + perpendicular[1] ** 2) }
+  end
+
+  def interpolate(value, from_low, from_high, to_low, to_high)
+    (value - from_low) * (to_high - to_low) / (from_high - from_low) + to_low
+  end
+end

data/src/chart.svg.erb

@@ -0,0 +1,60 @@
+<svg height="<%= @height %>" width="<%= @width %>" xmlns="http://www.w3.org/2000/svg" font-family="Arial" style="background-color: #333333">
+  <text x="<%= @width / 2 %>" y="50" fill="white" dominant-baseline="middle" text-anchor="middle" font-size="40">
+    <%= @title %>
+  </text>
+
+  <% [@x_axis, @y_axis].compact.each do |axis| %>
+    <line x1="<%= axis.start_point[:x] %>" y1="<%= axis.start_point[:y] %>" x2="<%= axis.end_point[:x] %>" y2="<%= axis.end_point[:y] %>" style="stroke:white;stroke-width:2"/>
+
+    <% axis.labels.each do |label| %>
+      <text x="<%= label[:x] %>" y="<%= label[:y] %>" fill="white" dominant-baseline="middle" text-anchor="middle">
+        <%= label[:text] %>
+      </text>
+    <% end %>
+  <% end %>
+
+  <% if @x_axis&.title %>
+    <text x="<%= @x_axis.title_pos[:x] %>" y="<%= @x_axis.title_pos[:y] %>" fill="white" dominant-baseline="middle" text-anchor="middle" font-size="30">
+      <%= @x_axis.title %>
+    </text>
+  <% end %>
+
+  <% if @y_axis&.title %>
+    <text x="<%= @y_axis.title_pos[:x] %>" y="<%= @y_axis.title_pos[:y] %>" fill="white" dominant-baseline="middle" text-anchor="middle" font-size="30" transform="rotate(-90 <%= @y_axis.title_pos[:x] %> <%= @y_axis.title_pos[:y] %>)">
+      <%= @y_axis.title %>
+    </text>
+  <% end %>
+
+  <% if @x_axis&.grid_lines %>
+    <% @x_axis.labels.each do |label| %>
+      <line x1="<%= label[:x] %>" y1="<%= plot_area[:bottom] %>" x2="<%= label[:x] %>" y2="<%= plot_area[:bottom] + 10 %>" style="stroke:white;stroke-width:2"/>
+      <line x1="<%= label[:x] %>" y1="<%= plot_area[:top] %>" x2="<%= label[:x] %>" y2="<%= plot_area[:bottom] %>" style="stroke:white;stroke-width:1"/>
+    <% end %>
+  <% end %>
+
+  <% if @y_axis&.grid_lines %>
+    <% @y_axis.labels.each do |label| %>
+      <line x1="<%= plot_area[:left] %>" y1="<%= label[:y] %>" x2="<%= plot_area[:left] - 10 %>" y2="<%= label[:y] %>" style="stroke:white;stroke-width:2"/>
+      <line x1="<%= plot_area[:left] %>" y1="<%= label[:y] %>" x2="<%= plot_area[:right] %>" y2="<%= label[:y] %>" style="stroke:white;stroke-width:1"/>
+    <% end %>
+  <% end %>
+
+  <% if @scatter %>
+    <% point_positions.each do |x, y| %>
+      <circle cx="<%= x %>" cy="<%= y %>" r="3" fill="white"/>
+    <% end %>
+  <% end %>
+
+  <% if @line %>
+    <% point_positions.each_cons(2) do |(x1, y1), (x2, y2)| %>
+      <line x1="<%= x1 %>" y1="<%= y1 %>" x2="<%= x2 %>" y2="<%= y2 %>" style="stroke:white;stroke-width:2"/>
+    <% end %>
+  <% end %>
+
+  <% if @trend_line %>
+    <mask id="plot_area">
+      <rect x="<%= plot_area[:left] %>" y="<%= plot_area[:top] %>" width="<%= plot_area[:width] %>" height="<%= plot_area[:height] %>" fill="white"/>
+    </mask>
+    <line x1="<%= trend_line_points[0][0] %>" y1="<%= trend_line_points[0][1] %>" x2="<%= trend_line_points[1][0] %>" y2="<%= trend_line_points[1][1] %>" style="stroke:white;stroke-width:2" mask="url(#plot_area)"/>
+  <% end %>
+</svg>

data/src/data_value.rb

@@ -0,0 +1,38 @@
+# frozen_string_literal: true
+
+class DataValue
+  attr_reader :type
+
+  def initialize(value)
+    @value = value
+    @type = if date_time?(value)
+              :date
+            elsif Integer(value)
+              :integer
+            else
+              :string
+            end
+  end
+
+  def value
+    case @type
+    when :date
+      DateTime.parse(@value).to_time.to_i
+    else
+      @value.to_i
+    end
+  end
+
+  def to_s
+    @value.to_s
+  end
+
+  private
+
+  def date_time?(val)
+    DateTime.parse(val)
+    true
+  rescue Date::Error
+    false
+  end
+end

data/src/plot.rb

@@ -0,0 +1,114 @@
+# frozen_string_literal: true
+
+class Plot
+  def initialize(data, height: 1000, width: 1000)
+    @height = height
+    @width = width
+    @data = data.map { |line| [DataValue.new(line[0]), DataValue.new(line[1])] }
+  end
+
+  def scatter
+    @scatter = true
+    self
+  end
+
+  def line
+    @line = true
+    self
+  end
+
+  def title(title)
+    @title = title
+    self
+  end
+
+  def x_axis(title: nil, count: 10, grid_lines: true)
+    @x_axis_options = { title: title, count: count, grid_lines: grid_lines }
+    self
+  end
+
+  def y_axis(title: nil, count: 10, grid_lines: true)
+    @y_axis_options = { title: title, count: count, grid_lines: grid_lines }
+    self
+  end
+
+  def trend_line
+    @trend_line = true
+    self
+  end
+
+  def to_svg
+    @x_axis = Axis.new(
+      min_x,
+      max_x,
+      { x: plot_area[:left], y: plot_area[:bottom] },
+      { x: plot_area[:right], y: plot_area[:bottom] },
+      @x_axis_options[:count],
+      @x_axis_options[:title],
+      @x_axis_options[:grid_lines]
+    )
+
+    @y_axis = Axis.new(
+      max_y,
+      min_y,
+      { x: plot_area[:left], y: plot_area[:top] },
+      { x: plot_area[:left], y: plot_area[:bottom] },
+      @y_axis_options[:count],
+      @y_axis_options[:title],
+      @y_axis_options[:grid_lines]
+    )
+
+    template = File.read(File.join(__dir__, 'chart.svg.erb'))
+    eval(Erubi::Engine.new(template).src)
+  end
+
+  def trend_line_points
+    x_values = point_positions.map(&:first)
+    y_values = point_positions.map(&:last)
+    x_mean = x_values.sum / x_values.size.to_f
+    y_mean = y_values.sum / y_values.size.to_f
+    slope = x_values.zip(y_values).map { |x, y| (x - x_mean) * (y - y_mean) }.sum / x_values.map { |x| (x - x_mean) ** 2 }.sum
+    intercept = y_mean - slope * x_mean
+    [
+      [plot_area[:left], slope * plot_area[:left] + intercept],
+      [plot_area[:right], slope * plot_area[:right] + intercept]
+    ]
+  end
+
+  private
+
+  def plot_area
+    top = @title ? 100 : 50
+    left = @y_axis_options&.key?(:title) ? 100 : 50
+    right = @width - 50
+    bottom = @x_axis_options&.key?(:title) ? @height - 100 : @height - 50
+    width = right - left
+    height = bottom - top
+    { top:, left:, right:, bottom:, width:, height: }
+  end
+
+  def point_positions
+    @data.map do |x, y|
+      [
+        interpolate(x.value, min_x.value, max_x.value, plot_area[:left], plot_area[:right]),
+        interpolate(y.value, min_y.value, max_y.value, plot_area[:bottom], plot_area[:top])
+      ]
+    end.sort_by(&:first)
+  end
+
+  def interpolate(value, fromLow, fromHigh, toLow, toHigh)
+    (value - fromLow) * (toHigh - toLow) / (fromHigh - fromLow) + toLow
+  end
+
+  def x_values = @data.map(&:first)
+
+  def min_x = x_values.min_by(&:value)
+
+  def max_x = x_values.max_by(&:value)
+
+  def y_values = @data.map(&:last)
+
+  def min_y = y_values.min_by(&:value)
+
+  def max_y = y_values.max_by(&:value)
+end

metadata

@@ -0,0 +1,62 @@
+--- !ruby/object:Gem::Specification
+name: chart_fun
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Max Hatfull
+autorequire:
+bindir: bin
+cert_chain: []
+date: 2025-01-08 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: erubi
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: A simple plotting library for Ruby
+email:
+- max.hatfull@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- chart_fun.rb
+- src/axis.rb
+- src/chart.svg.erb
+- src/data_value.rb
+- src/plot.rb
+homepage: https://github.com/MaxHatfull/chart_fun
+licenses:
+- MIT
+metadata: {}
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.5.16
+signing_key:
+specification_version: 4
+summary: A simple plotting library for Ruby
+test_files: []