diff --git a/docs/assets/screenshots/viz-colormaps-qim.png b/docs/assets/screenshots/viz-colormaps-qim.png new file mode 100644 index 0000000000000000000000000000000000000000..b5b214a41bc0ce361fc51939494d55a83b458779 Binary files /dev/null and b/docs/assets/screenshots/viz-colormaps-qim.png differ diff --git a/docs/viz.md b/docs/viz.md index 604b3a59a2fc0c86f1039975d349c913bc8db9f5..3ddd34a1c58617de014cf57e6479ad32fb826e72 100644 --- a/docs/viz.md +++ b/docs/viz.md @@ -17,3 +17,4 @@ The `qim3d` library aims to provide easy ways to explore and get insights from v options: members: - objects + - qim diff --git a/qim3d/viz/colormaps.py b/qim3d/viz/colormaps.py index fb14607182ec70e9e2f9179234977a969d150e37..fc0e54b8e1724984ff3f973b0eb19f0eb5a06985 100644 --- a/qim3d/viz/colormaps.py +++ b/qim3d/viz/colormaps.py @@ -6,6 +6,7 @@ import colorsys from typing import Union, Tuple import numpy as np from matplotlib.colors import LinearSegmentedColormap +from matplotlib import colormaps from qim3d.io.logger import log @@ -104,8 +105,28 @@ def objects( randRGBcolors[-1] = background_color # Create colormap - objects_cmap = LinearSegmentedColormap.from_list( - "objects_cmap", randRGBcolors, N=nlabels + objects = LinearSegmentedColormap.from_list( + "objects", randRGBcolors, N=nlabels ) - return objects_cmap + return objects + + + +qim = LinearSegmentedColormap.from_list('qim', + [(0.6, 0.0, 0.0), #990000 + (1.0, 0.6, 0.0), #ff9900 + ]) +""" +Defines colormap in QIM logo colors. Can be accessed as module attribute or easily by ```cmap = 'qim'``` + +Example: + ```python + + import qim3d + + display(qim3d.viz.colormaps.qim) + ``` +  +""" +colormaps.register(qim) \ No newline at end of file