charma 0.1.0

data/lib/charma/violin_chart.rb

@@ -0,0 +1,121 @@
+# frozen_string_literal: true
+
+module Charma
+  class ViolinChart < Chart
+    def initialize(opts)
+      super(opts)
+    end
+
+    # TODO: LineChart に同じようなメソッドがあるのでなんとかする
+    def calc_range(sym)
+      r0 = @opts[:series].map{ |e| e[sym] }.flatten.minmax
+      dist = r0[1] - r0[0]
+      delta = dist==0 ? 1 : dist*0.1
+      raw_range =
+        if 0<=r0[0]
+          [[r0[0]-delta, 0].max, r0[1]+delta]
+        else
+          [r0[0]-delta, r0[1]+delta]
+        end
+      raw_range.map{ |e| unscale_value( sym, e ) }
+    end
+
+    # TODO: BarChart に同じメソッドがあるのでなんとかする
+    def render_xticks(pdf, area)
+      rects = area.hsplit(*Array.new(@opts[:x_ticks].size){ 1 }).map{ |rc0|
+        rc0.hsplit(1,8,1)[1]
+      }
+      draw_samesize_texts( pdf, rects, @opts[:x_ticks], valign: :top )
+    end
+
+    def draw_violin(pdf, rect, yrange, hists, cols)
+      ratio = 0.75
+      _, violins, = rect.hsplit( (1-ratio)/2, ratio, (1-ratio)/2 )
+      v_rects = violins.hsplit(*Array.new(hists.size,1))
+      v_rects.zip(hists, cols) do |rc, hist, col|
+        cx = rc.x + rc.w/2
+        h = rc.h / hist.size.to_f
+        hist.each.with_index do |f, ix|
+          w = rc.w * f
+          top = rc.bottom + h*(ix+1)
+          edge = 1e-1 # バーの隙間を埋める
+          fill_rect( pdf, Rect.new( cx-w/2, top + edge, w, h + edge*2 ), col )
+        end
+      end
+    end
+
+    def meansize( vals )
+      sum=0.0
+      count=0
+      vals.each do |vvv|
+        vvv.each do |vv|
+          sum += vv.size
+          count+=1
+        end
+      end
+      sum / count
+    end
+
+    def make_histograms( vals, range )
+      hist_size = [10,Math.sqrt( meansize(vals) ).round].max
+      min = range.min
+      step = (range.max - min).to_f / hist_size
+      bottoms = Array.new(hist_size){ |ix| min + (ix+1)*step }
+      raw_h = vals.map{ |vvv|
+        vvv.map{ |vv|
+          vv.each.with_object([0]*hist_size){ |v,o|
+            ix = bottoms.index{ |b| v<b }
+            o[ix]+=1
+          }
+        }
+      }
+      ratio = 1.0 / raw_h.flatten.max
+      raw_h.map{ |vvv|
+        vvv.map{ |vv|
+          vv.map{ |e| e*ratio }
+        }
+      }
+    end
+
+    def render_chart(pdf, rect, yrange)
+      y_values = @opts[:series].map{ |s| s[:y] }.transpose
+      bar_areas = rect.hsplit(*Array.new(y_values.size,1))
+      cols = if y_values.first.size==1
+        colors(y_values.size).map{ |e| [e] }
+      else
+        [colors(y_values.first.size)] * y_values.size
+      end
+      hists = make_histograms(y_values, yrange)
+      hists.zip(bar_areas, cols).each do |h, rc, c|
+        draw_violin(pdf, rc, yrange, h, c)
+      end
+    end
+
+    # BarChart の render とほぼ同じなのでなんとかする
+    def render( pdf, rect )
+      stroke_rect(pdf, rect)
+      title_text = @opts[:title]
+      title, main, ticks, bottom = rect.vsplit(
+        (title_text ? 1 : 0),
+        7,
+        (@opts[:x_ticks] ? 0.5 : 0),
+        (bottom_legend? ? 0.5 : 0))
+      draw_text( pdf, title, title_text ) if title_text
+      hratio = [(@opts[:y_label] ? 1 : 0), 1, 10]
+      ylabel, yticks, chart = main.hsplit(*hratio)
+      yrange = @opts[:y_range] || calc_range(:y)
+      render_chart(pdf, chart, yrange)
+      if @opts[:y_label]
+        render_rottext(pdf, ylabel, @opts[:y_label] )
+      end
+      if @opts[:x_ticks]
+        _, _, xticks = ticks.hsplit(*hratio)
+        render_xticks(pdf, xticks)
+      end
+      yvalues = tick_values(:y, yrange)
+      render_yticks(pdf, yticks, yrange, yvalues)
+      render_y_grid(pdf, chart, yrange, yvalues)
+      render_legend(pdf, bottom) if bottom_legend?
+    end
+  end
+end

data/lib/charma.rb

@@ -0,0 +1,14 @@
+# frozen_string_literal: true
+
+require "prawn"
+require_relative "charma/version"
+require_relative "charma/error"
+require_relative "charma/rect"
+require_relative "charma/chart"
+
+require_relative "charma/bar_chart"
+require_relative "charma/line_chart"
+require_relative "charma/violin_chart"
+
+require_relative "charma/page"
+require_relative "charma/document"

metadata

@@ -0,0 +1,143 @@
+--- !ruby/object:Gem::Specification
+name: charma
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Nabetani
+autorequire: 
+bindir: exe
+cert_chain: []
+date: 2019-03-24 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: prawn
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 2.2.2
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 2.2.2
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+- !ruby/object:Gem::Dependency
+  name: pry
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+description: Create chart in PDF
+email:
+- takenori@nabetani.sakura.ne.jp
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".rspec"
+- ".rubocop.yml"
+- ".travis.yml"
+- CODE_OF_CONDUCT.md
+- Gemfile
+- Gemfile.lock
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/console
+- bin/setup
+- charma.gemspec
+- examples/bar_charts.pdf
+- examples/bar_charts.rb
+- examples/line_charts.pdf
+- examples/line_charts.rb
+- examples/runall.rb
+- examples/violin_charts.pdf
+- examples/violin_charts.rb
+- lib/charma.rb
+- lib/charma/bar_chart.rb
+- lib/charma/chart.rb
+- lib/charma/document.rb
+- lib/charma/error.rb
+- lib/charma/line_chart.rb
+- lib/charma/page.rb
+- lib/charma/rect.rb
+- lib/charma/version.rb
+- lib/charma/violin_chart.rb
+homepage: https://github.com/nabetani/charma
+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.0.3
+signing_key: 
+specification_version: 4
+summary: Create chart in PDF
+test_files: []