From d0c45ed7c919dc3c37d2d59c7ef14dcf71fc18b2 Mon Sep 17 00:00:00 2001
From: s233039 <s233039@student.dtu.dk>
Date: Fri, 12 Apr 2024 14:50:47 +0200
Subject: [PATCH] QIM themed colormap

---
 docs/assets/screenshots/viz-colormaps-qim.png | Bin 0 -> 769 bytes
 docs/viz.md                                   |   1 +
 qim3d/viz/colormaps.py                        |  27 ++++++++++++++++--
 3 files changed, 25 insertions(+), 3 deletions(-)
 create mode 100644 docs/assets/screenshots/viz-colormaps-qim.png

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
GIT binary patch
literal 769
zcmeAS@N?(olHy`uVBq!ia0y~yV0y&Bz!1s7#=yW(UFf)gfq{Xg*vT`5gM)*kh9jke
zfq_A?#5JNMI6tkVJh3R1Aw4fYH&wSdxhOR?uQ(&W<UR9#I|c>@J&+>jqSVBa%=|os
zkj&gv2DhS21w$hR$ATgSBLgE71p^~1V?!%b1BC!z1!vdbkoh7_^B5SIB0XIkLn`LH
zy%y+q$U(#*@b#@bOsi8S{z^Y!ka_w_%&O-#Q#t!*wurqe+8JYKzwY|>_G#_Y@9R(7
zee`W{blUv6w{5KpuU$QV+xYvUYPqs&Q{R{7=3RbMw0Tc@UefvH=dEVQ-fLcb{&vy-
zbLVnr+WuN&fBk&b{qyJczL;14ZcX#o`j0hpzij{X?$e9^lWyOfxtZ@?{kQb2wDXIu
zhRt93W6k^Kw~P!8@hV3c859&oi6ImQA6Nt5uzFqyIJ1m_fq}u()z4*}Q$iB}phfAo

literal 0
HcmV?d00001

diff --git a/docs/viz.md b/docs/viz.md
index 604b3a59..3ddd34a1 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 fb146071..fc0e54b8 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)
+    ```
+    ![colormap objects](assets/screenshots/viz-colormaps-qim.png)
+"""
+colormaps.register(qim)
\ No newline at end of file
-- 
GitLab