Skip to content

Commit

Permalink
brightness reduction in colormap gen
Browse files Browse the repository at this point in the history
  • Loading branch information
nmwsharp committed Jan 1, 2025
1 parent a3a9a81 commit 9d5c3a4
Showing 1 changed file with 16 additions and 5 deletions.
21 changes: 16 additions & 5 deletions misc/generate_colormap_constant.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,27 +2,38 @@

import numpy as np
import matplotlib.cm
import matplotlib.pyplot as plt
import colorsys

nValues = 500;

# get a matplotlib colormap
# cmapName = 'Spectral'
# cmap = matplotlib.cm.get_cmap(cmapName)
cmapName = 'hsv'
cmap = plt.get_cmap(cmapName)

# get a cmocean colormap
import cmocean
cmapName = 'phase'
cmap = cmocean.cm.phase
# import cmocean
# cmapName = 'phase'
# cmap = cmocean.cm.phase


print("static const Colormap CM_" + cmapName.upper() + " = {")
print(" \"" + cmapName + "\",")

def reduce_brightness_rgb(color, factor):
r, g, b = color
h, l, s = colorsys.rgb_to_hls(r, g, b)
l = max(0, l * factor) # Ensure lightness remains non-negative
r, g, b = colorsys.hls_to_rgb(h, l, s)
return tuple(x for x in (r, g, b))

dataStr = "{"
for i in range(nValues):

floatInd = float(i) / (nValues-1)
color = cmap(floatInd)
color = (color[0], color[1], color[2])
# color = reduce_brightness_rgb(color, 0.85) # optional: dim bright colormaps a little

dataStr += "{" + str(color[0]) + "," + str(color[1]) + "," + str(color[2]) + "},"

Expand Down

0 comments on commit 9d5c3a4

Please sign in to comment.