Improving colormap support

solidsgroup · Dec 18, 2023 · 35c08e8 · 35c08e8
1 parent f98aa38
commit 35c08e8
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/bleMD/bleMDColormaps.py b/bleMD/bleMDColormaps.py
@@ -51,13 +51,16 @@ def get(self,pos):
             g.append(igreen.get(k))
             b.append(iblue.get(k))
         return p, r, g, b
+
     elif "colors" in cmap.keys():
         p = []
         r = []
         g = []
         b = []
-        for i,c in enumerate(cmap['colors']):
-            p.append(i/len(cmap['colors']))
+        mod = int(len(cmap["colors"])/32)
+        if not mod: mod = 1
+        for i,c in enumerate(cmap['colors'][::mod]):
+            p.append(mod*i/len(cmap['colors']))
             r.append(c[0])
             g.append(c[1])
             b.append(c[2])

diff --git a/bleMD/bleMDNodes.py b/bleMD/bleMDNodes.py
@@ -104,7 +104,7 @@ def defaultSettings():
     for scene in bpy.data.scenes:
         scene.render.engine = 'CYCLES'
 
-    bpy.context.window.workspace = bpy.data.workspaces['Shading']
+    #bpy.context.window.workspace = bpy.data.workspaces['Shading']
     for n in range(3,11): # This whole bit of code is hacky
         for area in bpy.data.screens[n].areas: 
            if area.type == 'VIEW_3D':