red-palette 0.4.0 → 0.5.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 3122a06d16fd374d4d82abe074c2dbff00a2b75f546f785c9cc72ded5d0a2148
-  data.tar.gz: d4003bbc74cdfdaddf32775b07d2870f1b3f4f3c23da36ec1735a23f007802be
+  metadata.gz: 1adfe150392dfb94f321411798f406a2dd9e9dcb7183c049f28363a84bf1ed9a
+  data.tar.gz: f123f69082f526c4aa4be8f5e8055956c6a06a4d7aeff8ff15d9af951ba23a66
 SHA512:
-  metadata.gz: 73d4618b458ddfec249662b3b3082ce61a3482671f8f6d26365fe510dc7fd9f3b6e3c78ea7eb9ddacff856184e376aa0957d3d0f4b5f7faa465e1ee584e29fcb
-  data.tar.gz: e0c74affc25e2f964c6f076caa739c7431d7b0c6957912198cfe2a930fafabdfc30bce259eda1e4a2333d070f2032f3b912a1ff0da83a6dc817364bd9b57254c
+  metadata.gz: 69e32980c20d3a9a6d7334145cfb9b168d637bf15ec7ed2e4c2b5725e48e06de1962d7dd6ea08734bd868e31109c03f05bf9257371927e1390fa3c6ddc067790
+  data.tar.gz: 0ef1c3883ce5ea83312489458793ca1a370cb4fa9b2da784d5cdf12f04ddb7f8374a2a0c47a4b092978249a54f45c5339f1b4172bab0bae0b560e768d3a923b8

data/lib/palette.rb

@@ -185,5 +185,5 @@ class Palette
     end
   end
 
-  self.default = Palette.new("deep").freeze
+  self.default = Palette.new("tab10").freeze
 end

data/lib/palette/colors.rb

@@ -79,8 +79,43 @@ class Palette
     pal
   end
 
-  def self.matplotlib_colors(name, n_colors=6)
-    raise NotImplementedError,
-          "Matplotlib's colormap emulation has not been implemented"
+  def self.matplotlib_colors(name, n_colors=6, as_cmap: false)
+    if name.end_with?("_d")
+      sub_name = name[0..-2]
+      if sub_name.end_with?("_r")
+        reverse = true
+        sub_name = sub_name[0..-2]
+      else
+        reverse = false
+      end
+      colors = Palette.new(sub_name, 2).colors
+      colors << Colors::RGB.parse("#333333")
+      colors.reverse! if reverse
+      cmap = blend_cmap(colors, n_colors, as_cmap: true)
+    else
+      cmap = Colors::ColormapRegistry[name]
+    end
+
+    return cmap if as_cmap
+
+    bins = if MPL_QUAL_PALS.include?(name)
+             Helper.linspace(0r, 1r, MPL_QUAL_PALS[name])[0, n_colors]
+           else
+             Helper.linspace(0r, 1r, Integer(n_colors) + 2)[1...-1]
+           end
+    colors = cmap[bins]
+    return colors
+  end
+
+  def self.blend_colors(colors, n_colors=6, as_cmap: false)
+    name = "blend"
+    cmap = Colors::LinearSegmentedColormap.new_from_list(name, colors)
+    if as_cmap
+      return cmap
+    else
+      bins = Helper.linspace(0r, 1r, Integer(n_colors))
+      colors = cmap[bins]
+      return colors
+    end
   end
 end

data/lib/palette/version.rb

@@ -1,3 +1,3 @@
 class Palette
-  VERSION = "0.4.0"
+  VERSION = "0.5.0"
 end

data/red-palette.gemspec

@@ -32,7 +32,7 @@ Gem::Specification.new do |spec|
   spec.files += Dir.glob("doc/text/*")
   spec.test_files += Dir.glob("test/**/*")
 
-  spec.add_dependency("red-colors", ">= 0.2.0")
+  spec.add_dependency("red-colors", ">= 0.3.0")
 
   spec.add_development_dependency("bundler")
   spec.add_development_dependency("rake")

data/test/palette_test.rb

@@ -3,7 +3,7 @@ class PaletteTest < Test::Unit::TestCase
 
   sub_test_case(".default") do
     test("the initial value") do
-      assert_equal(Palette.new("deep"), Palette.default)
+      assert_equal(Palette.new("tab10"), Palette.default)
     end
   end
 
@@ -146,18 +146,40 @@ class PaletteTest < Test::Unit::TestCase
 
   sub_test_case(".matplotlib_colors") do
     test("Set3") do
-      omit("Not implemented yet")
       palette = Palette.new("Set3")
       assert_equal("Set3", palette.name)
       assert_equal(Palette::QUAL_PALETTE_SIZES["Set3"],
                    palette.n_colors)
-      assert_equal(Palette::SEABORN_PALETTES["Set3"].map {|c|
-                     Colors::RGB.parse(c)
-                   },
-                   palette.colors)
+      assert_equal([
+        Colors::RGB.new(0.55294117647058827, 0.82745098039215681, 0.7803921568627451),
+        Colors::RGB.new(1.0,                 0.92941176470588238, 0.43529411764705883)
+      ],
+      [
+        palette.colors[0],
+        palette.colors[-1],
+      ])
+    end
+  end
+
+  sub_test_case("seaborn's palette") do
+    test("flare") do
+      palette = Palette.new("flare", 256)
+      assert_equal("flare", palette.name)
+      assert_equal(256, palette.n_colors)
+      assert_equal([
+        Colors::RGB.new(0.92907237, 0.68878959, 0.50411509),
+        Colors::RGB.new(0.29408557, 0.13721193, 0.38442775),
+      ],
+      [
+        palette.colors[0],
+        palette.colors[-1],
+      ])
     end
   end
 
+  sub_test_case(".blend_colors") do
+  end
+
   test("desaturation feature") do
     desaturated_colors = Palette.new(["#ff0000", "#00ff0099"], desaturate_factor: 0.8).colors
     assert_near(Colors::HSL.new(0, 0.8r, 0.5r).to_rgb,

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: red-palette
 version: !ruby/object:Gem::Version
-  version: 0.4.0
+  version: 0.5.0
 platform: ruby
 authors:
 - Kenta Murata
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2021-06-13 00:00:00.000000000 Z
+date: 2021-06-15 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: red-colors
@@ -16,14 +16,14 @@ dependencies:
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: 0.2.0
+        version: 0.3.0
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: 0.2.0
+        version: 0.3.0
 - !ruby/object:Gem::Dependency
   name: bundler
   requirement: !ruby/object:Gem::Requirement