heatgrid 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 8915d9ec489aa94d63d1f0eb343c687b565275aa
-  data.tar.gz: 8dd74d30266cbf9ccbb97cb93a6075bd2c2e9060
+  metadata.gz: ed2cbbef931c7150e0b4b89b132143304bfadeeb
+  data.tar.gz: f8f3385099930b4d43347c29a003d684a2f6d778
 SHA512:
-  metadata.gz: 2f1a3246db9b67065470a7697bc393c15e37bbe3881f30de9968f2b4367b8a9fc03818e3993b16327ef0a3f4fa848198ccf07d69955ad9a605d60b4958ab892e
-  data.tar.gz: 7ce9aaa298fc609d15a33fb8d2ea971d87da283cdd2b0718fc9ffd3b1acb14e5b4314419d3df1da0232d12b9f3e9177100a9d49d3cce3b01fc1cf7ca07f3894e
+  metadata.gz: be561d8760943944f1b752517df6b91b951e988856b6086e8f1b3b43ca99eaa275b04843e1d8b213bd92c90205663837b8c01b0fd8df4f317dfd87442f8309d4
+  data.tar.gz: f238283ca36c5a6b2c7f07c001dcc442b5b5aee1f79a21df906974460312e474d1d3dc4088b1d740c35b9fe00bd92822ca55fb2ee73fcd93ce2747b32ff5ce24

data/Rakefile

@@ -1 +1,7 @@
 require "bundler/gem_tasks"
+
+task :console do
+  require 'pry'
+  require './lib/heatgrid'
+  Pry.start
+end

data/heatgrid.gemspec

@@ -9,7 +9,7 @@ Gem::Specification.new do |spec|
   spec.authors       = ["Eugene Millerman"]
   spec.email         = ["e.v.millerman@gmail.com"]
   spec.summary       = %q{Generate a heatmap from a 2d array. }
-  spec.description   = %q{Generate a heatmap from a 2d array with rmagick.}
+  spec.description   = %q{Generate a grid based heatmap from a 2d array using rmagick.}
   spec.homepage      = ""
   spec.license       = "MIT"
 
@@ -20,5 +20,7 @@ Gem::Specification.new do |spec|
 
   spec.add_development_dependency "bundler", "~> 1.5"
   spec.add_development_dependency "rake"
-  # spec.add_development_dependency "rspec", "~>2.5.0"
+  spec.add_development_dependency "rspec", "~>2.5.0"
+  spec.add_development_dependency "pry"
+
 end

data/lib/heatgrid.rb

@@ -1,10 +1,6 @@
 require "heatgrid/version"
+require "heatgrid/grid"
 
 module Heatgrid
-  # Your code goes here...
-  class Heatgrid
-    def self.hi
-      puts "Hello World!"
-    end
-  end
+  
 end

data/lib/heatgrid/grid.rb

@@ -0,0 +1,38 @@
+require 'pry'
+module Heatgrid
+  class Grid
+    attr_accessor :original_grid, :normalized_grid
+    @max
+    
+    def initialize(in_array)
+      self.original_grid = in_array
+      self.normalized_grid = normalize(in_array)
+    end
+
+    def rgb_grid
+      normalized_grid.map do |row|
+        row.map do |val|
+          normalized_to_rgb(val)
+        end
+      end
+    end
+
+    def normalized_to_rgb(num)
+      if (0..0.5).include?(num)
+        [0,(255 * (num*2)).to_i,(255*((0.5-num)*2)).to_i]
+      else
+        [(255 * (num-0.5)*2).to_i,(255*(1-2*(num-0.5))).to_i,0]
+      end
+    end
+
+    def normalize(in_array)
+      max = max_value
+      in_array.map{|row| row.map{|v| v/max.to_f}}
+    end
+
+    def max_value
+      @max ||= original_grid.map(&:max).max
+    end
+
+  end
+end

data/lib/heatgrid/version.rb

@@ -1,3 +1,3 @@
 module Heatgrid
-  VERSION = "0.0.1"
+  VERSION = "0.0.2"
 end

data/spec/grid_spec.rb

@@ -0,0 +1,59 @@
+require 'spec_helper'
+
+TEST = [
+  [1,2,3,4],
+  [2,3,4,5],
+  [3,4,5,6],
+  [4,5,6,7]
+]
+
+RESULT = 
+[
+  [0.14285714285714285,0.2857142857142857,0.42857142857142855,0.5714285714285714],
+  [0.2857142857142857,0.42857142857142855,0.5714285714285714,0.7142857142857143],
+  [0.42857142857142855,0.5714285714285714,0.7142857142857143,0.8571428571428571],
+  [0.5714285714285714, 0.7142857142857143, 0.8571428571428571, 1.0]
+]
+
+RGB = [
+  [[0, 72, 182], [0, 145, 109], [0, 218, 36], [36, 218, 0]],
+  [[0, 145, 109], [0, 218, 36], [36, 218, 0], [109, 145, 0]],
+  [[0, 218, 36], [36, 218, 0], [109, 145, 0], [182, 72, 0]],
+  [[36, 218, 0], [109, 145, 0], [182, 72, 0], [255, 0, 0]]
+]
+
+describe Heatgrid::Grid do
+  let(:grid) {Heatgrid::Grid.new(TEST)}
+
+  describe "#new" do
+    
+    it "saves the original input" do
+      expect(grid.original_grid).to eq(TEST)
+    end
+
+    it "finds the maximum value" do
+      expect(grid.max_value).to eq(7)
+    end
+
+    it "calculates the normalized grid" do
+      expect(grid.normalized_grid).to eq(RESULT)
+    end
+
+    it "converts normalized values to rgb" do
+      expect(grid.normalized_to_rgb(0.0)).to eq([0,0,255])
+      expect(grid.normalized_to_rgb(0.1)).to eq([0,51,204])
+      expect(grid.normalized_to_rgb(0.25)).to eq([0,127,127])
+      expect(grid.normalized_to_rgb(0.5)).to eq([0,255,0])
+      expect(grid.normalized_to_rgb(0.75)).to eq([127,127,0])
+      expect(grid.normalized_to_rgb(1.0)).to eq([255,0,0])
+      expect(grid.normalized_to_rgb(0.9)).to eq([204,50,0])
+    end
+
+    it "gets the rgb values" do
+      expect(grid.rgb_grid).to eq(RGB)
+    end
+
+  end
+
+end
+

data/spec/spec_helper.rb

@@ -4,6 +4,9 @@
 # loaded once.
 #
 # See http://rubydoc.info/gems/rspec-core/RSpec/Core/Configuration
+require_relative '../lib/heatgrid'
+
+
 RSpec.configure do |config|
   config.treat_symbols_as_metadata_keys_with_true_values = true
   config.run_all_when_everything_filtered = true
@@ -13,5 +16,5 @@ RSpec.configure do |config|
   # order dependency and want to debug it, you can fix the order by providing
   # the seed, which is printed after each run.
   #     --seed 1234
-  config.order = 'random'
+  config.order = 'default'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: heatgrid
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors:
 - Eugene Millerman
@@ -38,7 +38,35 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
-description: Generate a heatmap from a 2d array with rmagick.
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 2.5.0
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 2.5.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'
+description: Generate a grid based heatmap from a 2d array using rmagick.
 email:
 - e.v.millerman@gmail.com
 executables: []
@@ -53,7 +81,9 @@ files:
 - Rakefile
 - heatgrid.gemspec
 - lib/heatgrid.rb
+- lib/heatgrid/grid.rb
 - lib/heatgrid/version.rb
+- spec/grid_spec.rb
 - spec/spec_helper.rb
 homepage: ''
 licenses:
@@ -80,4 +110,6 @@ signing_key:
 specification_version: 4
 summary: Generate a heatmap from a 2d array.
 test_files:
+- spec/grid_spec.rb
 - spec/spec_helper.rb
+has_rdoc: