marker size for multiple traces

Author: dmt0

src/components/fields/Field.js

@@ -24,6 +24,7 @@ class Field extends Component {
       children,
       label,
       multiValued,
+      suppressMultiValuedMessage,
       units,
       extraComponent,
     } = this.props;
@@ -68,7 +69,7 @@ class Field extends Component {
         ) : null}
         <div className={fieldClass}>
           {children}
-          {multiValued ? (
+          {multiValued && !suppressMultiValuedMessage ? (
             <MenuPanel label={getMultiValueText('title', _)} ownline question>
               <div className="info__title">{getMultiValueText('title', _)}</div>
               <div className="info__text">{getMultiValueText('text', _)}</div>
@@ -94,6 +95,7 @@ Field.propTypes = {
   label: PropTypes.any,
   units: PropTypes.string,
   multiValued: PropTypes.bool,
+  suppressMultiValuedMessage: PropTypes.bool,
   children: PropTypes.node,
   extraComponent: PropTypes.any,
 };

src/components/fields/MarkerSize.js

@@ -5,14 +5,27 @@ import {connectToContainer} from 'lib';
 import RadioBlocks from '../widgets/RadioBlocks';
 import Numeric from './Numeric';
 import DataSelector from './DataSelector';
-
-const getType = value => (Array.isArray(value) ? 'variable' : 'constant');
+import {MULTI_VALUED} from 'lib/constants';
 
 class UnconnectedMarkerSize extends Component {
   constructor(props, context) {
     super(props, context);
 
-    const type = getType(props.fullValue);
+    let type = null;
+    if (
+      !props.container.marker ||
+      (props.container.marker && !props.container.marker.sizesrc)
+    ) {
+      type = 'constant';
+    } else if (
+      props.container.marker &&
+      Array.isArray(props.container.marker.size) &&
+      props.fullContainer.marker &&
+      Array.isArray(props.fullContainer.marker.size)
+    ) {
+      type = 'variable';
+    }
+
     this.state = {
       type,
       value: {
@@ -30,6 +43,11 @@ class UnconnectedMarkerSize extends Component {
     this.props.updatePlot(this.state.value[type]);
     if (type === 'constant') {
       this.context.updateContainer({['marker.sizesrc']: null});
+    } else {
+      this.context.updateContainer({
+        ['marker.size']: null,
+        ['marker.sizesrc']: null,
+      });
     }
   }
 
@@ -45,32 +63,37 @@ class UnconnectedMarkerSize extends Component {
   }
 
   render() {
-    const {attr} = this.props;
+    const {attr, fullValue} = this.props;
     const {localize: _} = this.context;
+    const {type, value} = this.state;
     const options = [
       {label: _('Constant'), value: 'constant'},
       {label: _('Variable'), value: 'variable'},
     ];
+    const multiValued =
+      this.props.multiValued ||
+      (Array.isArray(fullValue) && fullValue.includes(MULTI_VALUED));
 
     return (
       <div>
-        <Field {...this.props} attr={attr}>
+        <Field {...this.props} multiValued={multiValued} attr={attr}>
           <RadioBlocks
             options={options}
-            activeOption={this.state.type}
+            activeOption={type}
             onOptionChange={this.setType}
           />
-          {this.state.type === 'constant' ? (
+          {type === 'constant' ? (
             <Numeric
+              suppressMultiValuedMessage
               attr="marker.size"
               updatePlot={this.setValue}
-              fullValue={this.state.value.constant}
+              fullValue={value.constant}
             />
-          ) : (
+          ) : multiValued ? null : (
             <DataSelector
+              suppressMultiValuedMessage
               attr="marker.size"
               updatePlot={this.setValue}
-              fullValue={this.state.value.variable}
             />
           )}
         </Field>

src/lib/connectTraceToPlot.js

@@ -75,17 +75,25 @@ export default function connectTraceToPlot(WrappedComponent) {
       };
 
       if (traceIndexes.length > 1) {
-        const multiValuedContainer = deepCopyPublic(fullTrace);
+        const multiValuedFullContainer = deepCopyPublic(fullTrace);
         fullData.forEach(t =>
+          Object.keys(t).forEach(key =>
+            setMultiValuedContainer(multiValuedFullContainer, t, key, {
+              searchArrays: true,
+            })
+          )
+        );
+        const multiValuedContainer = deepCopyPublic(trace);
+        data.forEach(t =>
           Object.keys(t).forEach(key =>
             setMultiValuedContainer(multiValuedContainer, t, key, {
               searchArrays: true,
             })
           )
         );
-        this.childContext.fullContainer = multiValuedContainer;
+        this.childContext.fullContainer = multiValuedFullContainer;
         this.childContext.defaultContainer = fullTrace;
-        this.childContext.container = {};
+        this.childContext.container = multiValuedContainer;
       }
 
       if (trace && fullTrace) {

src/lib/multiValues.js

@@ -17,10 +17,8 @@ function deepCopyPublic(value) {
 }
 
 function setMultiValuedContainer(intoObj, fromObj, key, config = {}) {
-  var intoVal = intoObj[key],
-    fromVal = fromObj[key];
-
-  var searchArrays = config.searchArrays;
+  const intoVal = intoObj[key];
+  const fromVal = fromObj[key];
 
   // don't merge private attrs
   if (
@@ -51,20 +49,24 @@ function setMultiValuedContainer(intoObj, fromObj, key, config = {}) {
   } else if (Array.isArray(intoVal)) {
     // in data, other arrays are data, which we don't care about
     // for styling purposes
-    if (!searchArrays) {
+    if (!config.searchArrays) {
       return;
     }
-    // in layout though, we need to recurse into arrays
-    for (var i = 0; i < fromVal.length; i++) {
-      setMultiValuedContainer(intoVal, fromVal, i, searchArrays);
+    if (!Array.isArray(fromVal)) {
+      intoObj[key] = MULTI_VALUED;
+    } else {
+      // in layout though, we need to recurse into arrays
+      for (let i = 0; i < fromVal.length; i++) {
+        setMultiValuedContainer(intoVal, fromVal, i, config);
+      }
     }
   } else if (isPlainObject(fromVal)) {
     // recurse into objects
     if (!isPlainObject(intoVal)) {
       throw new Error('tried to merge object into non-object: ' + key);
     }
     Object.keys(fromVal).forEach(function(key2) {
-      setMultiValuedContainer(intoVal, fromVal, key2, searchArrays);
+      setMultiValuedContainer(intoVal, fromVal, key2, config);
     });
   } else if (isPlainObject(intoVal)) {
     throw new Error('tried to merge non-object into object: ' + key);