From 5d183710d9ad709ee28f8b10208d4b37e819b279 Mon Sep 17 00:00:00 2001 From: Tom Smeding Date: Sun, 16 Oct 2022 16:55:50 +0200 Subject: Colorschemes --- mandelhs.cabal | 1 + render/Inferno.hs | 272 ++++++++++++++++++++++++++++++++++++++++++++++++++++++ render/Render.hs | 4 +- 3 files changed, 276 insertions(+), 1 deletion(-) create mode 100644 render/Inferno.hs 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) = -- cgit v1.2.3-70-g09d2