red-colors 0.2.0 → 0.3.0

data/lib/colors/colormap_data/seaborn_builtin.rb

@@ -0,0 +1,10 @@
+module Colors
+  module ColormapRegistry
+    register_listed_colormap("rocket")
+    register_listed_colormap("mako")
+    register_listed_colormap("icefire")
+    register_listed_colormap("vlag")
+    register_listed_colormap("flare")
+    register_listed_colormap("crest")
+  end
+end

data/lib/colors/colormap_registry.rb

@@ -0,0 +1,62 @@
+module Colors
+  module ColormapRegistry
+    @registry = {}
+
+    def self.[](name)
+      return name if name.is_a?(Colormap)
+
+      name = String.try_convert(name)
+      if @registry.key?(name)
+        return @registry[name]
+      else
+        raise ArgumentError, "Unknown colormap name: %p" % name
+      end
+    end
+
+    def self.register(cmap, name: nil, override_builtin: false)
+      case name
+      when String, Symbol
+        name = name.to_s
+      when nil
+        name = cmap.name
+        if name.nil?
+          raise ArgumentError, "`name` cannot be omitted for unnamed colormaps"
+        end
+      else
+        name = String.try_convert(name)
+        if name.nil?
+          raise ArgumentError, "`name` must be convertible to a String by to_str"
+        end
+      end
+
+      if @registry.key?(name)
+        existing = @registry[name]
+        if BUILTIN_COLORMAPS.key?(name)
+          unless override_builtin
+            raise ArgumentError,
+                  "Trying to re-register a builtin colormap: %p" % name
+          end
+        end
+        warn "Trying to re-register the colormap %p which already exists" % name
+      end
+
+      unless cmap.is_a?(Colormap)
+        raise ArgumentError,
+              "Invalid value for registering a colormap (%p for a Colormap)" % cmap
+      end
+
+      @registry[name] = cmap
+    end
+
+    def self.unregister(name)
+      if @registry.key?(name)
+        if BUILTIN_COLORMAPS.key?(name)
+          raise ArgumentError,
+                "Unable to unregister the colormap %p which is a builtin colormap" % name
+        end
+      else
+        @registry.delete(name)
+      end
+    end
+  end
+end

data/lib/colors/helper.rb

@@ -7,7 +7,8 @@ module Colors
 
     private def check_range(value, range, name)
       return value if range.cover?(value)
-      check_fail ArgumentError, "#{name} must be in #{range}, but #{value} is given"
+      check_fail ArgumentError,
+                 "#{name} must be in #{range}, but %p is given" % value
     end
 
     private def check_fail(exc_class, message)

data/lib/colors/linear_segmented_colormap.rb

@@ -17,7 +17,7 @@ module Colors
       else
         ary = Array.try_convert(colors)
         if ary.nil?
-          raise ArgumentError, "colors must be convertible to an array"
+          raise ArgumentError, "colors must be convertible to an array: %p for %s" % [colors, name]
         else
           colors = ary
         end
@@ -115,5 +115,23 @@ module Colors
 
       return lut.map {|v| v.clamp(0, 1).to_f }
     end
+
+    private def make_reverse_colormap(name)
+      segmented_data_r = self.segmented_data.map { |key, data|
+        data_r = if data.respond_to?(:call)
+                   make_inverse_func(data)
+                 else
+                   data.reverse_each.map do |x, y0, y1|
+                     [1r - x, y1, y0]
+                   end
+                 end
+        [key, data_r]
+      }.to_h
+      LinearSegmentedColormap.new(name, segmented_data_r, n_colors: self.n_colors, gamma: self.gamma)
+    end
+
+    private def make_inverse_func(f)
+      ->(x) { f(1 - x) }
+    end
   end
 end

data/lib/colors/listed_colormap.rb

@@ -37,5 +37,9 @@ module Colors
       @initialized = true
       update_extreme_colors
     end
+
+    private def make_reverse_colormap(name)
+      ListedColormap.new(self.colors.reverse, name: name, n_colors: self.n_colors)
+    end
   end
 end

data/lib/colors/version.rb

@@ -1,3 +1,3 @@
 module Colors
-  VERSION = "0.2.0"
+  VERSION = "0.3.0"
 end

data/red-colors.gemspec

@@ -27,6 +27,7 @@ Gem::Specification.new do |spec|
     "#{spec.name}.gemspec",
   ]
   spec.files += [".yardopts"]
+  spec.files += Dir.glob("data/colormaps/*.json")
   spec.files += Dir.glob("lib/**/*.rb")
   spec.files += Dir.glob("image/*.*")
   spec.files += Dir.glob("doc/text/*")

data/test/test-linear-segmented-colormap.rb

@@ -115,5 +115,24 @@ class ColorsLinearSegmentedColormapTest < Test::Unit::TestCase
                    after
                  ])
   end
+
+  def test_reversed
+    cm = Colors::LinearSegmentedColormap.new_from_list("four", [:red, :magenta, :green, :yellow, :blue], n_colors: 5)
+    cm.under_color = :black
+    cm.over_color = :white
+    rev = cm.reversed
+    assert_equal([
+                   "four_r",
+                   cm[[0, 1, 2, 3, 4]],
+                   Colors[:black],
+                   Colors[:white]
+                 ],
+                 [
+                   rev.name,
+                   rev[[4, 3, 2, 1, 0]],
+                   rev.under_color,
+                   rev.over_color
+                 ])
+  end
 end
 

data/test/test-listed-colormap.rb

@@ -112,4 +112,23 @@ class ColorsListedColormapTest < Test::Unit::TestCase
                    after
                  ])
   end
+
+  def test_reversed
+    cm = Colors::ListedColormap.new([:red, :green, :blue], name: "three")
+    cm.under_color = :orange
+    cm.over_color = :yellow
+    rev = cm.reversed
+    assert_equal([
+                   "three_r",
+                   cm[[0, 1, 2]],
+                   Colors[:orange],
+                   Colors[:yellow]
+                 ],
+                 [
+                   rev.name,
+                   rev[[2, 1, 0]],
+                   rev.under_color,
+                   rev.over_color
+                 ])
+  end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: red-colors
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.3.0
 platform: ruby
 authors:
 - Kenta Murata
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2021-06-12 00:00:00.000000000 Z
+date: 2021-06-14 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: matrix
@@ -106,11 +106,33 @@ files:
 - LICENSE.txt
 - README.md
 - Rakefile
+- data/colormaps/cividis.json
+- data/colormaps/coolwarm.json
+- data/colormaps/crest.json
+- data/colormaps/flare.json
+- data/colormaps/gist_earth.json
+- data/colormaps/gist_ncar.json
+- data/colormaps/icefire.json
+- data/colormaps/inferno.json
+- data/colormaps/magma.json
+- data/colormaps/mako.json
+- data/colormaps/nipy_spectral.json
+- data/colormaps/pink.json
+- data/colormaps/plasma.json
+- data/colormaps/rocket.json
+- data/colormaps/turbo.json
+- data/colormaps/twilight.json
+- data/colormaps/viridis.json
+- data/colormaps/vlag.json
 - lib/colors.rb
 - lib/colors/abstract_color.rb
 - lib/colors/alpha_component.rb
 - lib/colors/color_data.rb
 - lib/colors/colormap.rb
+- lib/colors/colormap_data.rb
+- lib/colors/colormap_data/matplotlib_builtin.rb
+- lib/colors/colormap_data/seaborn_builtin.rb
+- lib/colors/colormap_registry.rb
 - lib/colors/convert.rb
 - lib/colors/helper.rb
 - lib/colors/hsl.rb