Skip to content

Commit

Permalink
fix: line path with ordered xValues (#824)
Browse files Browse the repository at this point in the history
  • Loading branch information
nickofthyme authored Sep 17, 2020
1 parent 7aba794 commit 5a73a3a
Show file tree
Hide file tree
Showing 8 changed files with 147 additions and 16 deletions.
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
14 changes: 14 additions & 0 deletions integration/tests/line_stories.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -42,4 +42,18 @@ describe('Line series stories', () => {
);
});
});

describe('Line paths for ordered values', () => {
it('should render correct line path - non-stacked', async () => {
await common.expectChartAtUrlToMatchScreenshot(
'http://localhost:9001/?path=/story/line-chart--test-path-ordering&knob-enable orderOrdinalBinsBy=true&knob-Stacked=false',
);
});

it('should render correct line path - stacked', async () => {
await common.expectChartAtUrlToMatchScreenshot(
'http://localhost:9001/?path=/story/line-chart--test-path-ordering&knob-enable orderOrdinalBinsBy=true&knob-Stacked=true',
);
});
});
});
2 changes: 1 addition & 1 deletion src/chart_types/xy_chart/utils/fill_series.ts
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@ import { SpecId } from '../../../utils/ids';
import { DataSeries } from './series';

/**
*
* Fill missing x values in all data series
* @param series
* @param xValues
* @internal
Expand Down
58 changes: 43 additions & 15 deletions src/chart_types/xy_chart/utils/series.ts
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,7 @@ import { splitSpecsByGroupId, YBasicSeriesSpec } from '../domains/y_domain';
import { LastValues } from '../state/utils/types';
import { applyFitFunctionToDataSeries } from './fit_function_utils';
import { BasicSeriesSpec, SeriesTypes, SeriesSpecs, SeriesNameConfigOptions, StackMode } from './specs';
import { formatStackedDataSeriesValues } from './stacked_series_utils';
import { formatStackedDataSeriesValues, datumXSortPredicate } from './stacked_series_utils';

/** @internal */
export const SERIES_DELIMITER = ' - ';
Expand Down Expand Up @@ -329,6 +329,22 @@ function castToNumber(value: any, nonNumericValues: any[]): number | null {
return num;
}

/**
* Sorts data based on order of xValues
* @param dataSeries
* @param xValues
* @param xScaleType
*/
const getSortedDataSeries = (
dataSeries: DataSeries[],
xValues: Set<string | number>,
xScaleType: ScaleType,
): DataSeries[] =>
dataSeries.map(({ data, ...rest }) => ({
...rest,
data: data.sort(datumXSortPredicate(xScaleType, [...xValues.values()])),
}));

/** @internal */
export function getFormattedDataseries(
specs: YBasicSeriesSpec[],
Expand Down Expand Up @@ -359,8 +375,12 @@ export function getFormattedDataseries(
const { stackMode } = groupSpecs;
// format stacked data series
const stackedDataSeries = getDataSeriesBySpecGroup(groupSpecs.stacked, availableDataSeries);
const fittedDataSeries = applyFitFunctionToDataSeries(stackedDataSeries.dataSeries, seriesSpecs, xScaleType);
const fittedAndStackedDataSeries = formatStackedDataSeriesValues(fittedDataSeries, xValues, stackMode);
const fittedStackedDataSeries = applyFitFunctionToDataSeries(
getSortedDataSeries(stackedDataSeries.dataSeries, xValues, xScaleType),
seriesSpecs,
xScaleType,
);
const fittedAndStackedDataSeries = formatStackedDataSeriesValues(fittedStackedDataSeries, xValues, stackMode);

stackedFormattedDataSeries.push({
groupId,
Expand All @@ -371,10 +391,15 @@ export function getFormattedDataseries(

// format non stacked data series
const nonStackedDataSeries = getDataSeriesBySpecGroup(groupSpecs.nonStacked, availableDataSeries);
const fittedNonStackedDataSeries = applyFitFunctionToDataSeries(
getSortedDataSeries(nonStackedDataSeries.dataSeries, xValues, xScaleType),
seriesSpecs,
xScaleType,
);
nonStackedFormattedDataSeries.push({
groupId,
counts: nonStackedDataSeries.counts,
dataSeries: applyFitFunctionToDataSeries(nonStackedDataSeries.dataSeries, seriesSpecs, xScaleType),
dataSeries: fittedNonStackedDataSeries,
});
});
return {
Expand Down Expand Up @@ -485,20 +510,23 @@ export function getDataSeriesBySpecId(
globalXValues.add(xValue);
}
}

const xValues =
isOrdinalScale || !isNumberArray
? getSortedOrdinalXValues(globalXValues, mutatedXValueSums, orderOrdinalBinsBy)
: new Set(
[...globalXValues].sort((a, b) => {
if (typeof a === 'string' || typeof b === 'string') {
return 0;
}
return a - b;
}),
);

return {
dataSeriesBySpecId,
seriesCollection,
xValues:
isOrdinalScale || !isNumberArray
? getSortedOrdinalXValues(globalXValues, mutatedXValueSums, orderOrdinalBinsBy)
: new Set(
[...globalXValues].sort((a, b) => {
if (typeof a === 'string' || typeof b === 'string') {
return 0;
}
return a - b;
}),
),
xValues,
fallbackScale: !isOrdinalScale && !isNumberArray ? ScaleType.Ordinal : undefined,
};
}
Expand Down
88 changes: 88 additions & 0 deletions stories/line/10_test_path_ordering.tsx
Original file line number Diff line number Diff line change
@@ -0,0 +1,88 @@
/*
* Licensed to Elasticsearch B.V. under one or more contributor
* license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright
* ownership. Elasticsearch B.V. licenses this file to you under
* the Apache License, Version 2.0 (the "License"); you may
* not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/

import { boolean } from '@storybook/addon-knobs';
import { shuffle } from 'lodash';
import React from 'react';

import { Axis, LineSeries, Chart, Position, ScaleType, Settings, Direction } from '../../src';
import { SB_SOURCE_PANEL } from '../utils/storybook';

/**
* Data to defined consistent order of series when using shuffle
*/
const data1 = [
// DestAirportID: Descending Carrier: Descending Max AvgTicketPrice Average AvgTicketPrice Test
{ x: 'XIY', g: 'JetBeats', y1: 1195.87316894531, y2: 735.960746071555, z: 735.960746071555 },
{ x: 'XIY', g: 'Kibana Airlines', y1: 1079.14624023438, y2: 742.831329345703, z: 742.831329345703 },
{ x: 'XIY', g: 'ES-Air', y1: 929.561462402344, y2: 765.738806152344, z: 765.738806152344 },
{ x: 'XIY', g: 'Logstash Airways', y1: 836.307922363281, y2: 487.398278808594, z: 487.398278808594 },
];

const data2 = [
{ x: 'XHBU', g: 'JetBeats', y1: 1193.38342285156, y2: 702.543407440186, z: 702.543407440186 },
{ x: 'XHBU', g: 'Kibana Airlines', y1: 1159.03503417969, y2: 606.558886210124, z: 606.558886210124 },
{ x: 'XHBU', g: 'ES-Air', y1: 996.849731445313, y2: 752.394683837891, z: 752.394683837891 },
{ x: 'XHBU', g: 'Logstash Airways', y1: 909.167602539063, y2: 564.171913146973, z: 564.171913146973 },
{ x: 'NGO', g: 'ES-Air', y1: 1189.08776855469, y2: 1189.08776855469, z: 1189.08776855469 },
{ x: 'SCL', g: 'Logstash Airways', y1: 1176.63818359375, y2: 1031.0576171875, z: 1031.0576171875 },
{ x: 'VE05', g: 'Kibana Airlines', y1: 1189.53845214844, y2: 563.195382859972, z: 563.195382859972 },
{ x: 'VE05', g: 'Logstash Airways', y1: 998.839538574219, y2: 467.636221313477, z: 467.636221313477 },
{ x: 'VE05', g: 'JetBeats', y1: 900.798461914063, y2: 569.146169026693, z: 569.146169026693 },
{ x: 'VE05', g: 'ES-Air', y1: 820.462463378906, y2: 541.392608642578, z: 541.392608642578 },
];

export const Example = () => {
const orderOrdinalBinsBy = boolean('enable orderOrdinalBinsBy', true);

return (
<Chart className="story-chart">
<Settings
orderOrdinalBinsBy={
orderOrdinalBinsBy
? {
direction: Direction.Descending,
}
: undefined
}
/>
<Axis id="bottom" position={Position.Bottom} showOverlappingTicks />
<Axis id="left2" position={Position.Left} tickFormat={(d: any) => `$${Number(d).toFixed(2)}`} />

<LineSeries
id="bars1"
fit="linear"
xScaleType={ScaleType.Ordinal}
yScaleType={ScaleType.Linear}
xAccessor="x"
yAccessors={['y1', 'y2']}
splitSeriesAccessors={['g']}
stackAccessors={boolean('Stacked', true) ? ['g'] : undefined}
data={[...data1, ...(orderOrdinalBinsBy ? shuffle(data2) : data2)]}
/>
</Chart>
);
};

// storybook configuration
Example.story = {
parameters: {
options: { selectedPanel: SB_SOURCE_PANEL },
},
};
1 change: 1 addition & 0 deletions stories/line/line.stories.tsx
Original file line number Diff line number Diff line change
Expand Up @@ -35,3 +35,4 @@ export { Example as curvedWithAxisAndLegend } from './6_curved';
export { Example as multipleWithAxisAndLegend } from './7_multiple';
export { Example as stackedWithAxisAndLegend } from './8_stacked';
export { Example as multiSeriesWithLogValues } from './9_multi_series';
export { Example as testPathOrdering } from './10_test_path_ordering';

0 comments on commit 5a73a3a

Please sign in to comment.