summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorTom Smeding <tom@tomsmeding.com>2022-10-16 16:55:50 +0200
committerTom Smeding <tom@tomsmeding.com>2022-10-16 16:55:50 +0200
commit5d183710d9ad709ee28f8b10208d4b37e819b279 (patch)
treeb07eedff1a532af02e26fc3a406bb656c72b4f47
parentd616a03191028113287a28c0b134071e8ba2cf1b (diff)
ColorschemesHEADmaster
-rw-r--r--mandelhs.cabal1
-rw-r--r--render/Inferno.hs272
-rw-r--r--render/Render.hs4
3 files changed, 276 insertions, 1 deletions
diff --git a/mandelhs.cabal b/mandelhs.cabal
index e149c59..fb33b46 100644
--- a/mandelhs.cabal
+++ b/mandelhs.cabal
@@ -46,6 +46,7 @@ executable mandelhs-render
main-is:
Render.hs
other-modules:
+ Inferno
build-depends:
mandelhs-lib,
base >= 4.13 && < 4.15,
diff --git a/render/Inferno.hs b/render/Inferno.hs
new file mode 100644
index 0000000..4aeb8bb
--- /dev/null
+++ b/render/Inferno.hs
@@ -0,0 +1,272 @@
+module Inferno where
+
+import Data.Word
+
+
+-- | matplotlib's "inferno" color scheme, with input inverted (i.e. 1 - input)
+colorscheme_infernoinv :: Double -> (Word8, Word8, Word8)
+colorscheme_infernoinv fraction =
+ let idx = max 0 $ min 255 $ round ((1 - fraction) * 255)
+ (r, g, b) = infernoData !! idx
+ tow8 x = round (max 0 (min 255 (x * 255)))
+ in (tow8 r, tow8 g, tow8 b)
+
+-- Source: https://github.com/matplotlib/matplotlib/blob/deb7b1145b71c9f60b87d4eaf32e848b1ea5ebe6/lib/matplotlib/_cm_listed.py#L260-L515
+infernoData :: [(Double, Double, Double)]
+infernoData =
+ [(0.001462, 0.000466, 0.013866),
+ (0.002267, 0.001270, 0.018570),
+ (0.003299, 0.002249, 0.024239),
+ (0.004547, 0.003392, 0.030909),
+ (0.006006, 0.004692, 0.038558),
+ (0.007676, 0.006136, 0.046836),
+ (0.009561, 0.007713, 0.055143),
+ (0.011663, 0.009417, 0.063460),
+ (0.013995, 0.011225, 0.071862),
+ (0.016561, 0.013136, 0.080282),
+ (0.019373, 0.015133, 0.088767),
+ (0.022447, 0.017199, 0.097327),
+ (0.025793, 0.019331, 0.105930),
+ (0.029432, 0.021503, 0.114621),
+ (0.033385, 0.023702, 0.123397),
+ (0.037668, 0.025921, 0.132232),
+ (0.042253, 0.028139, 0.141141),
+ (0.046915, 0.030324, 0.150164),
+ (0.051644, 0.032474, 0.159254),
+ (0.056449, 0.034569, 0.168414),
+ (0.061340, 0.036590, 0.177642),
+ (0.066331, 0.038504, 0.186962),
+ (0.071429, 0.040294, 0.196354),
+ (0.076637, 0.041905, 0.205799),
+ (0.081962, 0.043328, 0.215289),
+ (0.087411, 0.044556, 0.224813),
+ (0.092990, 0.045583, 0.234358),
+ (0.098702, 0.046402, 0.243904),
+ (0.104551, 0.047008, 0.253430),
+ (0.110536, 0.047399, 0.262912),
+ (0.116656, 0.047574, 0.272321),
+ (0.122908, 0.047536, 0.281624),
+ (0.129285, 0.047293, 0.290788),
+ (0.135778, 0.046856, 0.299776),
+ (0.142378, 0.046242, 0.308553),
+ (0.149073, 0.045468, 0.317085),
+ (0.155850, 0.044559, 0.325338),
+ (0.162689, 0.043554, 0.333277),
+ (0.169575, 0.042489, 0.340874),
+ (0.176493, 0.041402, 0.348111),
+ (0.183429, 0.040329, 0.354971),
+ (0.190367, 0.039309, 0.361447),
+ (0.197297, 0.038400, 0.367535),
+ (0.204209, 0.037632, 0.373238),
+ (0.211095, 0.037030, 0.378563),
+ (0.217949, 0.036615, 0.383522),
+ (0.224763, 0.036405, 0.388129),
+ (0.231538, 0.036405, 0.392400),
+ (0.238273, 0.036621, 0.396353),
+ (0.244967, 0.037055, 0.400007),
+ (0.251620, 0.037705, 0.403378),
+ (0.258234, 0.038571, 0.406485),
+ (0.264810, 0.039647, 0.409345),
+ (0.271347, 0.040922, 0.411976),
+ (0.277850, 0.042353, 0.414392),
+ (0.284321, 0.043933, 0.416608),
+ (0.290763, 0.045644, 0.418637),
+ (0.297178, 0.047470, 0.420491),
+ (0.303568, 0.049396, 0.422182),
+ (0.309935, 0.051407, 0.423721),
+ (0.316282, 0.053490, 0.425116),
+ (0.322610, 0.055634, 0.426377),
+ (0.328921, 0.057827, 0.427511),
+ (0.335217, 0.060060, 0.428524),
+ (0.341500, 0.062325, 0.429425),
+ (0.347771, 0.064616, 0.430217),
+ (0.354032, 0.066925, 0.430906),
+ (0.360284, 0.069247, 0.431497),
+ (0.366529, 0.071579, 0.431994),
+ (0.372768, 0.073915, 0.432400),
+ (0.379001, 0.076253, 0.432719),
+ (0.385228, 0.078591, 0.432955),
+ (0.391453, 0.080927, 0.433109),
+ (0.397674, 0.083257, 0.433183),
+ (0.403894, 0.085580, 0.433179),
+ (0.410113, 0.087896, 0.433098),
+ (0.416331, 0.090203, 0.432943),
+ (0.422549, 0.092501, 0.432714),
+ (0.428768, 0.094790, 0.432412),
+ (0.434987, 0.097069, 0.432039),
+ (0.441207, 0.099338, 0.431594),
+ (0.447428, 0.101597, 0.431080),
+ (0.453651, 0.103848, 0.430498),
+ (0.459875, 0.106089, 0.429846),
+ (0.466100, 0.108322, 0.429125),
+ (0.472328, 0.110547, 0.428334),
+ (0.478558, 0.112764, 0.427475),
+ (0.484789, 0.114974, 0.426548),
+ (0.491022, 0.117179, 0.425552),
+ (0.497257, 0.119379, 0.424488),
+ (0.503493, 0.121575, 0.423356),
+ (0.509730, 0.123769, 0.422156),
+ (0.515967, 0.125960, 0.420887),
+ (0.522206, 0.128150, 0.419549),
+ (0.528444, 0.130341, 0.418142),
+ (0.534683, 0.132534, 0.416667),
+ (0.540920, 0.134729, 0.415123),
+ (0.547157, 0.136929, 0.413511),
+ (0.553392, 0.139134, 0.411829),
+ (0.559624, 0.141346, 0.410078),
+ (0.565854, 0.143567, 0.408258),
+ (0.572081, 0.145797, 0.406369),
+ (0.578304, 0.148039, 0.404411),
+ (0.584521, 0.150294, 0.402385),
+ (0.590734, 0.152563, 0.400290),
+ (0.596940, 0.154848, 0.398125),
+ (0.603139, 0.157151, 0.395891),
+ (0.609330, 0.159474, 0.393589),
+ (0.615513, 0.161817, 0.391219),
+ (0.621685, 0.164184, 0.388781),
+ (0.627847, 0.166575, 0.386276),
+ (0.633998, 0.168992, 0.383704),
+ (0.640135, 0.171438, 0.381065),
+ (0.646260, 0.173914, 0.378359),
+ (0.652369, 0.176421, 0.375586),
+ (0.658463, 0.178962, 0.372748),
+ (0.664540, 0.181539, 0.369846),
+ (0.670599, 0.184153, 0.366879),
+ (0.676638, 0.186807, 0.363849),
+ (0.682656, 0.189501, 0.360757),
+ (0.688653, 0.192239, 0.357603),
+ (0.694627, 0.195021, 0.354388),
+ (0.700576, 0.197851, 0.351113),
+ (0.706500, 0.200728, 0.347777),
+ (0.712396, 0.203656, 0.344383),
+ (0.718264, 0.206636, 0.340931),
+ (0.724103, 0.209670, 0.337424),
+ (0.729909, 0.212759, 0.333861),
+ (0.735683, 0.215906, 0.330245),
+ (0.741423, 0.219112, 0.326576),
+ (0.747127, 0.222378, 0.322856),
+ (0.752794, 0.225706, 0.319085),
+ (0.758422, 0.229097, 0.315266),
+ (0.764010, 0.232554, 0.311399),
+ (0.769556, 0.236077, 0.307485),
+ (0.775059, 0.239667, 0.303526),
+ (0.780517, 0.243327, 0.299523),
+ (0.785929, 0.247056, 0.295477),
+ (0.791293, 0.250856, 0.291390),
+ (0.796607, 0.254728, 0.287264),
+ (0.801871, 0.258674, 0.283099),
+ (0.807082, 0.262692, 0.278898),
+ (0.812239, 0.266786, 0.274661),
+ (0.817341, 0.270954, 0.270390),
+ (0.822386, 0.275197, 0.266085),
+ (0.827372, 0.279517, 0.261750),
+ (0.832299, 0.283913, 0.257383),
+ (0.837165, 0.288385, 0.252988),
+ (0.841969, 0.292933, 0.248564),
+ (0.846709, 0.297559, 0.244113),
+ (0.851384, 0.302260, 0.239636),
+ (0.855992, 0.307038, 0.235133),
+ (0.860533, 0.311892, 0.230606),
+ (0.865006, 0.316822, 0.226055),
+ (0.869409, 0.321827, 0.221482),
+ (0.873741, 0.326906, 0.216886),
+ (0.878001, 0.332060, 0.212268),
+ (0.882188, 0.337287, 0.207628),
+ (0.886302, 0.342586, 0.202968),
+ (0.890341, 0.347957, 0.198286),
+ (0.894305, 0.353399, 0.193584),
+ (0.898192, 0.358911, 0.188860),
+ (0.902003, 0.364492, 0.184116),
+ (0.905735, 0.370140, 0.179350),
+ (0.909390, 0.375856, 0.174563),
+ (0.912966, 0.381636, 0.169755),
+ (0.916462, 0.387481, 0.164924),
+ (0.919879, 0.393389, 0.160070),
+ (0.923215, 0.399359, 0.155193),
+ (0.926470, 0.405389, 0.150292),
+ (0.929644, 0.411479, 0.145367),
+ (0.932737, 0.417627, 0.140417),
+ (0.935747, 0.423831, 0.135440),
+ (0.938675, 0.430091, 0.130438),
+ (0.941521, 0.436405, 0.125409),
+ (0.944285, 0.442772, 0.120354),
+ (0.946965, 0.449191, 0.115272),
+ (0.949562, 0.455660, 0.110164),
+ (0.952075, 0.462178, 0.105031),
+ (0.954506, 0.468744, 0.099874),
+ (0.956852, 0.475356, 0.094695),
+ (0.959114, 0.482014, 0.089499),
+ (0.961293, 0.488716, 0.084289),
+ (0.963387, 0.495462, 0.079073),
+ (0.965397, 0.502249, 0.073859),
+ (0.967322, 0.509078, 0.068659),
+ (0.969163, 0.515946, 0.063488),
+ (0.970919, 0.522853, 0.058367),
+ (0.972590, 0.529798, 0.053324),
+ (0.974176, 0.536780, 0.048392),
+ (0.975677, 0.543798, 0.043618),
+ (0.977092, 0.550850, 0.039050),
+ (0.978422, 0.557937, 0.034931),
+ (0.979666, 0.565057, 0.031409),
+ (0.980824, 0.572209, 0.028508),
+ (0.981895, 0.579392, 0.026250),
+ (0.982881, 0.586606, 0.024661),
+ (0.983779, 0.593849, 0.023770),
+ (0.984591, 0.601122, 0.023606),
+ (0.985315, 0.608422, 0.024202),
+ (0.985952, 0.615750, 0.025592),
+ (0.986502, 0.623105, 0.027814),
+ (0.986964, 0.630485, 0.030908),
+ (0.987337, 0.637890, 0.034916),
+ (0.987622, 0.645320, 0.039886),
+ (0.987819, 0.652773, 0.045581),
+ (0.987926, 0.660250, 0.051750),
+ (0.987945, 0.667748, 0.058329),
+ (0.987874, 0.675267, 0.065257),
+ (0.987714, 0.682807, 0.072489),
+ (0.987464, 0.690366, 0.079990),
+ (0.987124, 0.697944, 0.087731),
+ (0.986694, 0.705540, 0.095694),
+ (0.986175, 0.713153, 0.103863),
+ (0.985566, 0.720782, 0.112229),
+ (0.984865, 0.728427, 0.120785),
+ (0.984075, 0.736087, 0.129527),
+ (0.983196, 0.743758, 0.138453),
+ (0.982228, 0.751442, 0.147565),
+ (0.981173, 0.759135, 0.156863),
+ (0.980032, 0.766837, 0.166353),
+ (0.978806, 0.774545, 0.176037),
+ (0.977497, 0.782258, 0.185923),
+ (0.976108, 0.789974, 0.196018),
+ (0.974638, 0.797692, 0.206332),
+ (0.973088, 0.805409, 0.216877),
+ (0.971468, 0.813122, 0.227658),
+ (0.969783, 0.820825, 0.238686),
+ (0.968041, 0.828515, 0.249972),
+ (0.966243, 0.836191, 0.261534),
+ (0.964394, 0.843848, 0.273391),
+ (0.962517, 0.851476, 0.285546),
+ (0.960626, 0.859069, 0.298010),
+ (0.958720, 0.866624, 0.310820),
+ (0.956834, 0.874129, 0.323974),
+ (0.954997, 0.881569, 0.337475),
+ (0.953215, 0.888942, 0.351369),
+ (0.951546, 0.896226, 0.365627),
+ (0.950018, 0.903409, 0.380271),
+ (0.948683, 0.910473, 0.395289),
+ (0.947594, 0.917399, 0.410665),
+ (0.946809, 0.924168, 0.426373),
+ (0.946392, 0.930761, 0.442367),
+ (0.946403, 0.937159, 0.458592),
+ (0.946903, 0.943348, 0.474970),
+ (0.947937, 0.949318, 0.491426),
+ (0.949545, 0.955063, 0.507860),
+ (0.951740, 0.960587, 0.524203),
+ (0.954529, 0.965896, 0.540361),
+ (0.957896, 0.971003, 0.556275),
+ (0.961812, 0.975924, 0.571925),
+ (0.966249, 0.980678, 0.587206),
+ (0.971162, 0.985282, 0.602154),
+ (0.976511, 0.989753, 0.616760),
+ (0.982257, 0.994109, 0.631017),
+ (0.988362, 0.998364, 0.644924)]
diff --git a/render/Render.hs b/render/Render.hs
index 095df8d..05a1ee3 100644
--- a/render/Render.hs
+++ b/render/Render.hs
@@ -19,6 +19,7 @@ import System.Environment (getArgs)
import System.Exit (die, exitFailure)
import MandelHSlib
+import Inferno
type Colorscheme = Double -> (Word8, Word8, Word8)
@@ -50,7 +51,8 @@ curve x start end
colorschemes :: Map String Colorscheme
colorschemes = Map.fromList
[("yellow", colorscheme_yellow)
- ,("purple", colorscheme_purple)]
+ ,("purple", colorscheme_purple)
+ ,("infernoinv", colorscheme_infernoinv)]
renderFractal :: Colorscheme -> Fractal -> Pix.Image Pix.PixelRGB8
renderFractal colors (Fractal (w, h) _ arr) =