legendgroup and other legend attrs

Author: dmt0

src/components/fields/AxesCreator.js

@@ -99,7 +99,6 @@ UnconnectedAxisCreator.propTypes = {
   attr: PropTypes.string,
   label: PropTypes.string,
   options: PropTypes.array,
-  canAddAxis: PropTypes.bool,
   container: PropTypes.object,
   fullContainer: PropTypes.object,
   updateContainer: PropTypes.func,

src/components/fields/GroupCreator.js

@@ -0,0 +1,75 @@
+import React, {Component} from 'react';
+import {connectToContainer} from 'lib';
+import Field from './Field';
+import Dropdown from './Dropdown';
+import PropTypes from 'prop-types';
+import Button from '../widgets/Button';
+import {PlusIcon} from 'plotly-icons';
+
+class UnconnectedGroupCreator extends Component {
+  canAddGroup() {
+    const {fullContainer, attr} = this.props;
+    const currentGroup = fullContainer[attr];
+    const currentTraceIndex = fullContainer.index;
+
+    return (
+      !currentGroup ||
+      this.context.fullData.some(d => d.index !== currentTraceIndex && d[attr] === currentGroup)
+    );
+  }
+
+  addAndUpdateGroup() {
+    const allGroups = this.context.fullData
+      .map(t => parseInt(t[this.props.attr], 10))
+      .filter(n => Number.isInteger(n));
+    // don't want to pass empty array to max
+    allGroups.push(0);
+
+    const lastGroupNumber = Math.max.apply(Math, allGroups);
+
+    this.props.updatePlot(lastGroupNumber + 1);
+  }
+
+  render() {
+    const {localize: _, data} = this.context;
+    const {attr, label, prefix, updatePlot} = this.props;
+
+    const options = [{label: _('None'), value: ''}];
+    const allGroups = [...new Set(data.map(t => t[attr]))].filter(g => Boolean(g));
+    allGroups.forEach(g => options.push({label: `${prefix} ${g}`, value: g}));
+    options.sort((a, b) => a.value - b.value);
+
+    const icon = <PlusIcon />;
+    const addButton = this.canAddGroup() ? (
+      <Button variant="no-text" icon={icon} onClick={() => this.addAndUpdateGroup()} />
+    ) : (
+      <Button variant="no-text--disabled" icon={icon} onClick={() => {}} />
+    );
+
+    return (
+      <Dropdown
+        label={label}
+        attr={attr}
+        clearable={false}
+        options={options}
+        updatePlot={updatePlot}
+        extraComponent={addButton}
+      />
+    );
+  }
+}
+
+UnconnectedGroupCreator.propTypes = {
+  attr: PropTypes.string,
+  fullContainer: PropTypes.object,
+  prefix: PropTypes.string,
+  ...Field.propTypes,
+};
+
+UnconnectedGroupCreator.contextTypes = {
+  localize: PropTypes.func,
+  data: PropTypes.array,
+  fullData: PropTypes.array,
+};
+
+export default connectToContainer(UnconnectedGroupCreator);

src/components/fields/SubplotCreator.js

@@ -1,15 +1,15 @@
+import React, {Component} from 'react';
 import Dropdown from './Dropdown';
 import Info from './Info';
 import PropTypes from 'prop-types';
-import React, {Component} from 'react';
 import {EDITOR_ACTIONS, SUBPLOT_TO_ATTR} from 'lib/constants';
 import Button from '../widgets/Button';
 import {PlusIcon} from 'plotly-icons';
 import {connectToContainer, traceTypeToAxisType, getSubplotTitle} from 'lib';
 import {PlotlySection} from 'components';
 
 class UnconnectedSingleSubplotCreator extends Component {
-  canAddAxis() {
+  canAddSubplot() {
     const currentAxisId = this.props.fullContainer[this.props.attr];
     const currentTraceIndex = this.props.fullContainer.index;
     return this.context.fullData.some(
@@ -62,7 +62,7 @@ class UnconnectedSingleSubplotCreator extends Component {
 
   render() {
     const icon = <PlusIcon />;
-    const extraComponent = this.canAddAxis() ? (
+    const extraComponent = this.canAddSubplot() ? (
       <Button variant="no-text" icon={icon} onClick={() => this.addAndUpdateSubplot()} />
     ) : (
       <Button variant="no-text--disabled" icon={icon} onClick={() => {}} />
@@ -86,7 +86,6 @@ UnconnectedSingleSubplotCreator.propTypes = {
   layoutAttr: PropTypes.string,
   label: PropTypes.string,
   options: PropTypes.array,
-  canAddAxis: PropTypes.bool,
   container: PropTypes.object,
   fullContainer: PropTypes.object,
   updateContainer: PropTypes.func,

src/components/fields/index.js

@@ -21,6 +21,7 @@ import TraceSelector from './TraceSelector';
 import ErrorBars from './ErrorBars';
 import AxesCreator from './AxesCreator';
 import SubplotCreator from './SubplotCreator';
+import GroupCreator from './GroupCreator';
 import UpdateMenuButtons from './UpdateMenuButtons';
 import {FilterOperation, FilterValue} from './FilterOperation';
 import MarkerSize from './MarkerSize';
@@ -100,6 +101,7 @@ export {
   TraceSelector,
   AxesCreator,
   SubplotCreator,
+  GroupCreator,
   AxisOverlayDropdown,
   AxisSide,
   UpdateMenuButtons,

src/components/index.js

@@ -39,6 +39,7 @@ import {
   RangesliderVisible,
   AxesCreator,
   SubplotCreator,
+  GroupCreator,
   SymbolSelector,
   TextEditor,
   TraceOrientation,
@@ -146,6 +147,7 @@ export {
   SingleSidebarItem,
   AxesCreator,
   SubplotCreator,
+  GroupCreator,
   SymbolSelector,
   TextEditor,
   SubplotAccordion,

src/default_panels/StyleLegendPanel.js

@@ -63,13 +63,16 @@ const StyleLegendPanel = (props, {localize: _}) => (
         />
       </PlotlySection>
       <PlotlySection name={_('Trace Order')}>
-        <Radio
+        <Dropdown
           attr="legend.traceorder"
           options={[
             {label: _('Normal'), value: 'normal'},
             {label: _('Reversed'), value: 'reversed'},
+            {label: _('Grouped'), value: 'grouped'},
+            {label: _('Reversed and Grouped'), value: 'reversed+grouped'},
           ]}
         />
+        <Numeric label={_('Gap Between Groups')} attr="legend.tracegroupgap" units="px" />
       </PlotlySection>
     </PlotlyFold>
   </TraceRequiredPanel>

src/default_panels/StyleTracesPanel.js

@@ -30,6 +30,7 @@ import {
   ErrorBars,
   DataSelector,
   VisibilitySelect,
+  GroupCreator,
 } from '../components';
 import {
   BinningNumeric,
@@ -43,12 +44,15 @@ import {
 const StyleTracesPanel = (props, {localize: _}) => (
   <TraceAccordion canGroup>
     <TextEditor label={_('Name')} attr="name" richTextOnly />
-    <ShowInLegend
-      label={_('Show in Legend')}
-      attr="showlegend"
-      options={[{label: _('Show'), value: true}, {label: _('Hide'), value: false}]}
-    />
     <NumericFraction label={_('Trace Opacity')} attr="opacity" />
+    <PlotlySection name={_('Legend')}>
+      <ShowInLegend
+        label={_('Show in Legend')}
+        attr="showlegend"
+        options={[{label: _('Show'), value: true}, {label: _('Hide'), value: false}]}
+      />
+      <GroupCreator label={_('Legend Group')} prefix={_('Group')} attr="legendgroup" />
+    </PlotlySection>
     <PlotlySection name={_('Cones & Streamtubes')}>
       <Numeric label={_('Size')} attr="sizeref" />
       <Dropdown