Bug colormap

Author: vojtatom

studio/src/components/Shared/StyleScalars.tsx

@@ -2,6 +2,8 @@ import { Listbox } from '@headlessui/react';
 
 import { Histogram, Scalars } from '@data/types';
 
+import { useApplyStyle } from '@hooks/useApplyStyle';
+import { useLastStyle } from '@hooks/useLastStyle';
 import { useUpdateStyles } from '@hooks/useStyleUpdate';
 
 import { StyleHistogram } from './StyleHistogram';
@@ -11,10 +13,12 @@ const colormaps = ['plasma', 'magma', 'inferno', 'viridis'];
 export function StyleScalars(props: { scalars: Scalars; histogram: Histogram }) {
     const { scalars, histogram } = props;
     const updateStyles = useUpdateStyles();
+    const [, applyStyle] = useLastStyle();
 
     const handleChange = (value: string) => {
         scalars.colormap = value;
         updateStyles();
+        applyStyle();
     };
 
     return (

studio/src/hooks/useStyleUpdate.ts

@@ -8,8 +8,9 @@ export function useUpdateStyles() {
     const ctx = React.useContext(context);
 
     const setStyle = (style?: Style, metadata?: Metadata) => {
-        autoUpdateStyle(metadata ?? ctx.metadata, style ?? ctx.styles);
-        ctx.setStyles({ ...style });
+        const s = style ?? ctx.styles;
+        autoUpdateStyle(metadata ?? ctx.metadata, s);
+        ctx.setStyles({ ...s });
     };
 
     return setStyle;