@vitessce/statistical-plots 2.0.3 → 3.0.0

package/src/CellSetSizesPlotSubscriber.js

@@ -1,16 +1,18 @@
-import React, { useMemo } from 'react';
+import React, { useMemo, useState } from 'react';
 import {
   TitleInfo,
   useCoordination, useLoaders,
   useUrls, useReady, useGridItemSize,
   useObsSetsData,
-  registerPluginViewType,
 } from '@vitessce/vit-s';
+import { isEqual } from 'lodash-es';
 import { ViewType, COMPONENT_COORDINATION_TYPES } from '@vitessce/constants-internal';
-import { mergeObsSets, treeToSetSizesBySetNames } from '@vitessce/sets-utils';
+import {
+  mergeObsSets, treeToSetSizesBySetNames, filterPathsByExpansionAndSelection, findChangedHierarchy,
+} from '@vitessce/sets-utils';
 import { capitalize } from '@vitessce/utils';
-import CellSetSizesPlot from './CellSetSizesPlot';
-import { useStyles } from './styles';
+import CellSetSizesPlot from './CellSetSizesPlot.js';
+import { useStyles } from './styles.js';
 
 /**
  * A subscriber component for `CellSetSizePlot`,
@@ -42,6 +44,7 @@ export function CellSetSizesPlotSubscriber(props) {
     obsSetSelection: cellSetSelection,
     obsSetColor: cellSetColor,
     additionalObsSets: additionalCellSets,
+    obsSetExpansion: cellSetExpansion,
   }, {
     setObsSetSelection: setCellSetSelection,
     setObsSetColor: setCellSetColor,
@@ -52,6 +55,11 @@ export function CellSetSizesPlotSubscriber(props) {
   const [width, height, containerRef] = useGridItemSize();
   const [urls, addUrl] = useUrls(loaders, dataset);
 
+  // the name of the hierarchy that was clicked on last
+  const [currentHierarchy, setCurrentHierarchy] = useState([]);
+  // the previous cell set that was selected
+  const [prevCellSetSelection, setPrevCellSetSelection] = useState([]);
+
   // Get data from loaders using the data hooks.
   const [{ obsSets: cellSets }, obsSetsStatus] = useObsSetsData(
     loaders, dataset, addUrl, true,
@@ -68,12 +76,57 @@ export function CellSetSizesPlotSubscriber(props) {
     [cellSets, additionalCellSets],
   );
 
-  // From the cell sets hierarchy and the list of selected cell sets,
-  // generate the array of set sizes data points for the bar plot.
-  const data = useMemo(() => (mergedCellSets && cellSetSelection && cellSetColor
-    ? treeToSetSizesBySetNames(mergedCellSets, cellSetSelection, cellSetColor, theme)
-    : []
-  ), [mergedCellSets, cellSetSelection, cellSetColor, theme]);
+  const data = useMemo(() => {
+    if (cellSetSelection && cellSetColor && mergedCellSets && cellSets) {
+      let newHierarchy = currentHierarchy;
+
+      if (cellSetSelection) {
+        const changedHierarchy = findChangedHierarchy(prevCellSetSelection, cellSetSelection);
+        setPrevCellSetSelection(cellSetSelection);
+
+        if (changedHierarchy) {
+          setCurrentHierarchy(changedHierarchy);
+          newHierarchy = changedHierarchy;
+        }
+      }
+
+      const cellSetPaths = filterPathsByExpansionAndSelection(
+        mergedCellSets,
+        newHierarchy,
+        cellSetExpansion,
+        cellSetSelection,
+      );
+
+      if (mergedCellSets && cellSets && cellSetSelection && cellSetColor) {
+        return treeToSetSizesBySetNames(
+          mergedCellSets,
+          cellSetPaths,
+          cellSetSelection,
+          cellSetColor,
+          theme,
+        );
+      }
+    }
+    return [];
+  }, [
+    mergedCellSets,
+    cellSetSelection,
+    cellSetExpansion,
+    cellSetColor,
+    theme,
+  ]);
+
+  const onBarSelect = (setNamePath, wasGrayedOut, selectOnlyEnabled = false) => {
+    if (selectOnlyEnabled) {
+      setCellSetSelection([setNamePath]);
+      return;
+    }
+    if (!wasGrayedOut) {
+      setCellSetSelection(cellSetSelection.filter(d => !isEqual(d, setNamePath)));
+    } else if (wasGrayedOut) {
+      setCellSetSelection([...cellSetSelection, setNamePath]);
+    }
+  };
 
   return (
     <TitleInfo
@@ -86,6 +139,7 @@ export function CellSetSizesPlotSubscriber(props) {
       <div ref={containerRef} className={classes.vegaContainer}>
         <CellSetSizesPlot
           data={data}
+          onBarSelect={onBarSelect}
           theme={theme}
           width={width}
           height={height}
@@ -95,11 +149,3 @@ export function CellSetSizesPlotSubscriber(props) {
     </TitleInfo>
   );
 }
-
-export function register() {
-  registerPluginViewType(
-    ViewType.OBS_SET_SIZES,
-    CellSetSizesPlotSubscriber,
-    COMPONENT_COORDINATION_TYPES[ViewType.OBS_SET_SIZES],
-  );
-}

package/src/ExpressionHistogram.js

@@ -1,7 +1,14 @@
-import React from 'react';
-import clamp from 'lodash/clamp';
+import React, { useState, useEffect, useCallback } from 'react';
+import { clamp, debounce } from 'lodash-es';
 import { VegaPlot, VEGA_THEMES } from '@vitessce/vega';
 
+/**
+ * We use debounce, so that onSelect is called only after the user has finished the selection.
+ * Due to vega-lite limitations, we cannot use the vega-lite signals to implement this.
+ * See this issue: https://github.com/vega/vega-lite/issues/5728
+ * See this for reference on what is supported: https://vega.github.io/vega-lite/docs/selection.html
+ */
+
 /**
  * Gene expression histogram displayed as a bar chart,
  * implemented with the VegaPlot component.
@@ -29,14 +36,18 @@ export default function ExpressionHistogram(props) {
     height,
     marginRight = 90,
     marginBottom = 50,
+    onSelect,
   } = props;
 
+  const [selectedRanges, setSelectedRanges] = useState([]);
+
   const xTitle = geneSelection && geneSelection.length >= 1
     ? 'Normalized Expression Value'
     : 'Total Normalized Transcript Count';
 
   const spec = {
-    mark: { type: 'bar' },
+    data: { values: data },
+    mark: 'bar',
     encoding: {
       x: {
         field: 'value',
@@ -50,15 +61,59 @@ export default function ExpressionHistogram(props) {
         title: 'Number of Cells',
       },
       color: { value: 'gray' },
+      opacity: {
+        condition: { selection: 'brush', value: 1 },
+        value: 0.7,
+      },
     },
+    params: [
+      {
+        name: 'brush',
+        select: { type: 'interval', encodings: ['x'] },
+      },
+    ],
     width: clamp(width - marginRight, 10, Infinity),
     height: clamp(height - marginBottom, 10, Infinity),
     config: VEGA_THEMES[theme],
   };
 
+
+  const handleSignal = (name, value) => {
+    if (name === 'brush') {
+      setSelectedRanges(value.value);
+    }
+  };
+
+
+  // eslint-disable-next-line react-hooks/exhaustive-deps
+  const debouncedOnSelect = useCallback(debounce((ranges, latestOnSelect) => {
+    latestOnSelect(ranges);
+    // We set a debounce timer of 1000ms: the assumption here is that the user has
+    // finished the selection when there's been no mouse movement on the histogram for a second.
+    // We do not pass any dependencies for the useCallback
+    // since we only want to define the debounced function once (on the initial render).
+  }, 1000), []);
+
+  useEffect(() => {
+    if (!selectedRanges || selectedRanges.length === 0) return () => {};
+
+    // Call the debounced function instead of directly calling onSelect
+    debouncedOnSelect(selectedRanges, onSelect);
+
+    // Clean up the debounce timer when the component unmounts or the dependency changes
+    return () => {
+      debouncedOnSelect.cancel();
+    };
+  // We only want to call the debounced function when the selectedRanges changes.
+  // eslint-disable-next-line react-hooks/exhaustive-deps
+  }, [selectedRanges]);
+
+  const signalListeners = { brush: handleSignal };
+
   return (
     <VegaPlot
       data={data}
+      signalListeners={signalListeners}
       spec={spec}
     />
   );

package/src/ExpressionHistogramSubscriber.js

@@ -1,16 +1,17 @@
-import React, { useMemo } from 'react';
+import React, {
+  useMemo, useCallback,
+} from 'react';
 import { sum } from 'd3-array';
 import {
   TitleInfo,
   useCoordination, useLoaders,
   useUrls, useReady, useGridItemSize,
   useObsFeatureMatrixData, useFeatureSelection,
-  registerPluginViewType,
 } from '@vitessce/vit-s';
 import { ViewType, COMPONENT_COORDINATION_TYPES } from '@vitessce/constants-internal';
-import ExpressionHistogram from './ExpressionHistogram';
-import { useStyles } from './styles';
-
+import { setObsSelection, getObsInfoFromDataWithinRange } from '@vitessce/sets-utils';
+import ExpressionHistogram from './ExpressionHistogram.js';
+import { useStyles } from './styles.js';
 /**
  * A subscriber component for `ExpressionHistogram`,
  * which listens for gene selection updates and
@@ -38,6 +39,13 @@ export function ExpressionHistogramSubscriber(props) {
     featureType,
     featureValueType,
     featureSelection: geneSelection,
+    additionalObsSets: additionalCellSets,
+    obsSetColor: cellSetColor,
+  }, {
+    setAdditionalObsSets: setAdditionalCellSets,
+    setObsSetColor: setCellSetColor,
+    setObsColorEncoding: setCellColorEncoding,
+    setObsSetSelection: setCellSetSelection,
   }] = useCoordination(
     COMPONENT_COORDINATION_TYPES[ViewType.FEATURE_VALUE_HISTOGRAM],
     coordinationScopes,
@@ -69,11 +77,12 @@ export function ExpressionHistogramSubscriber(props) {
   // generate the array of data points for the histogram.
   const data = useMemo(() => {
     if (firstGeneSelected && obsFeatureMatrix && expressionData) {
-      // Create new cellColors map based on the selected gene.
-      return Array.from(expressionData[0]).map((_, index) => {
-        const value = expressionData[0][index];
+      return obsIndex.map((cellId, cellIndex) => {
+        const value = expressionData[0][cellIndex];
+        // Create new cellColors map based on the selected gene.
         const normValue = value * 100 / 255;
-        return { value: normValue, gene: firstGeneSelected };
+        const newItem = { value: normValue, gene: firstGeneSelected, cellId };
+        return newItem;
       });
     }
     if (obsFeatureMatrix) {
@@ -82,12 +91,28 @@ export function ExpressionHistogramSubscriber(props) {
         const values = obsFeatureMatrix.data
           .subarray(cellIndex * numGenes, (cellIndex + 1) * numGenes);
         const sumValue = sum(values) * 100 / 255;
-        return { value: sumValue, gene: null };
+        const newItem = { value: sumValue, gene: null, cellId };
+        return newItem;
       });
     }
     return null;
   }, [obsIndex, featureIndex, obsFeatureMatrix, firstGeneSelected, expressionData]);
 
+  const onSelect = useCallback((value) => {
+    const geneName = firstGeneSelected ? [firstGeneSelected, 'values'].join(' ') : 'transcript count';
+
+    const selectedCellIds = getObsInfoFromDataWithinRange(value, data);
+    setObsSelection(
+      selectedCellIds, additionalCellSets, cellSetColor,
+      setCellSetSelection, setAdditionalCellSets, setCellSetColor,
+      setCellColorEncoding,
+      'Selection ',
+      `: based on ${geneName} in range [${value[0].toFixed(1)}, ${value[1].toFixed(1)}] `,
+    );
+  }, [additionalCellSets, cellSetColor, data, setAdditionalCellSets,
+    setCellColorEncoding, setCellSetColor, setCellSetSelection, firstGeneSelected,
+  ]);
+
   return (
     <TitleInfo
       title={`Expression Histogram${(firstGeneSelected ? ` (${firstGeneSelected})` : '')}`}
@@ -99,6 +124,7 @@ export function ExpressionHistogramSubscriber(props) {
       <div ref={containerRef} className={classes.vegaContainer}>
         <ExpressionHistogram
           geneSelection={geneSelection}
+          onSelect={onSelect}
           data={data}
           theme={theme}
           width={width}
@@ -108,11 +134,3 @@ export function ExpressionHistogramSubscriber(props) {
     </TitleInfo>
   );
 }
-
-export function register() {
-  registerPluginViewType(
-    ViewType.FEATURE_VALUE_HISTOGRAM,
-    ExpressionHistogramSubscriber,
-    COMPONENT_COORDINATION_TYPES[ViewType.FEATURE_VALUE_HISTOGRAM],
-  );
-}

package/src/index.js

@@ -1,6 +1,6 @@
-export { CellSetExpressionPlotSubscriber, register as registerCellSetExpression } from './CellSetExpressionPlotSubscriber';
-export { CellSetSizesPlotSubscriber, register as registerCellSetSizes } from './CellSetSizesPlotSubscriber';
-export { ExpressionHistogramSubscriber, register as registerExpressionHistogram } from './ExpressionHistogramSubscriber';
-export { default as CellSetSizesPlot } from './CellSetSizesPlot';
-export { default as CellSetExpressionPlot } from './CellSetExpressionPlot';
-export { default as ExpressionHistogram } from './ExpressionHistogram';
+export { CellSetExpressionPlotSubscriber } from './CellSetExpressionPlotSubscriber.js';
+export { CellSetSizesPlotSubscriber } from './CellSetSizesPlotSubscriber.js';
+export { ExpressionHistogramSubscriber } from './ExpressionHistogramSubscriber.js';
+export { default as CellSetSizesPlot } from './CellSetSizesPlot.js';
+export { default as CellSetExpressionPlot } from './CellSetExpressionPlot.js';
+export { default as ExpressionHistogram } from './ExpressionHistogram.js';

package/src/styles.js

@@ -1,4 +1,4 @@
-import { makeStyles } from '@material-ui/core/styles';
+import { makeStyles } from '@material-ui/core';
 
 export const useStyles = makeStyles(() => ({
   vegaContainer: {