/** @format */ /** * External dependencies */ import { find, findIndex, get } from 'lodash'; import { max as d3Max } from 'd3-array'; import { axisBottom as d3AxisBottom, axisLeft as d3AxisLeft } from 'd3-axis'; import { format as d3Format } from 'd3-format'; import { scaleBand as d3ScaleBand, scaleLinear as d3ScaleLinear, scaleTime as d3ScaleTime, } from 'd3-scale'; import { event as d3Event, select as d3Select } from 'd3-selection'; import { line as d3Line } from 'd3-shape'; import { format as formatDate } from '@wordpress/date'; /** * Internal dependencies */ import { formatCurrency } from 'lib/currency'; import { dayTicksThreshold } from 'lib/date'; /** * Describes `smallestFactor` * @param {number} inputNum - any double or integer * @returns {integer} smallest factor of num */ export const getFactors = inputNum => { const numFactors = []; for ( let i = 1; i <= Math.floor( Math.sqrt( inputNum ) ); i += 1 ) { if ( inputNum % i === 0 ) { numFactors.push( i ); inputNum / i !== i && numFactors.push( inputNum / i ); } } numFactors.sort( ( x, y ) => x - y ); // numeric sort return numFactors; }; /** * Describes `getUniqueKeys` * @param {array} data - The chart component's `data` prop. * @returns {array} of unique category keys */ export const getUniqueKeys = data => { return [ ...new Set( data.reduce( ( accum, curr ) => { Object.keys( curr ).forEach( key => key !== 'date' && accum.push( key ) ); return accum; }, [] ) ), ]; }; /** * Describes `getOrderedKeys` * @param {array} data - The chart component's `data` prop. * @param {array} uniqueKeys - from `getUniqueKeys`. * @returns {array} of unique category keys ordered by cumulative total value */ export const getOrderedKeys = ( data, uniqueKeys ) => uniqueKeys .map( key => ( { key, focus: true, total: data.reduce( ( a, c ) => a + c[ key ].value, 0 ), visible: true, } ) ) .sort( ( a, b ) => b.total - a.total ); /** * Describes `getLineData` * @param {array} data - The chart component's `data` prop. * @param {array} orderedKeys - from `getOrderedKeys`. * @returns {array} an array objects with a category `key` and an array of `values` with `date` and `value` properties */ export const getLineData = ( data, orderedKeys ) => orderedKeys.map( row => ( { key: row.key, focus: row.focus, visible: row.visible, values: data.map( d => ( { date: d.date, focus: row.focus, label: get( d, [ row.key, 'label' ], '' ), value: get( d, [ row.key, 'value' ], 0 ), visible: row.visible, } ) ), } ) ); /** * Describes `getUniqueDates` * @param {array} lineData - from `GetLineData` * @param {function} parseDate - D3 time format parser * @returns {array} an array of unique date values sorted from earliest to latest */ export const getUniqueDates = ( lineData, parseDate ) => { return [ ...new Set( lineData.reduce( ( accum, { values } ) => { values.forEach( ( { date } ) => accum.push( date ) ); return accum; }, [] ) ), ].sort( ( a, b ) => parseDate( a ) - parseDate( b ) ); }; export const getColor = ( key, params ) => { const smallColorScales = [ [], [ 0.5 ], [ 0.333, 0.667 ], [ 0.2, 0.5, 0.8 ], [ 0.12, 0.375, 0.625, 0.88 ], ]; let keyValue = 0; const len = params.orderedKeys.length; const idx = findIndex( params.orderedKeys, d => d.key === key ); if ( len < 5 ) { keyValue = smallColorScales[ len ][ idx ]; } else { keyValue = idx / ( params.orderedKeys.length - 1 ); } return params.colorScheme( keyValue ); }; /** * Describes getXScale * @param {array} uniqueDates - from `getUniqueDates` * @param {number} width - calculated width of the charting space * @returns {function} a D3 scale of the dates */ export const getXScale = ( uniqueDates, width ) => d3ScaleBand() .domain( uniqueDates ) .rangeRound( [ 0, width ] ) .paddingInner( 0.1 ); /** * Describes getXGroupScale * @param {array} orderedKeys - from `getOrderedKeys` * @param {function} xScale - from `getXScale` * @returns {function} a D3 scale for each category within the xScale range */ export const getXGroupScale = ( orderedKeys, xScale ) => d3ScaleBand() .domain( orderedKeys.filter( d => d.visible ).map( d => d.key ) ) .rangeRound( [ 0, xScale.bandwidth() ] ) .padding( 0.07 ); /** * Describes getXLineScale * @param {array} uniqueDates - from `getUniqueDates` * @param {number} width - calculated width of the charting space * @returns {function} a D3 scaletime for each date */ export const getXLineScale = ( uniqueDates, width ) => d3ScaleTime() .domain( [ new Date( uniqueDates[ 0 ] ), new Date( uniqueDates[ uniqueDates.length - 1 ] ) ] ) .rangeRound( [ 0, width ] ); /** * Describes and rounds the maximum y value to the nearest thousadn, ten-thousand, million etc. * @param {array} lineData - from `getLineData` * @returns {number} the maximum value in the timeseries multiplied by 4/3 */ export const getYMax = lineData => { const yMax = 4 / 3 * d3Max( lineData, d => d3Max( d.values.map( date => date.value ) ) ); const pow3Y = Math.pow( 10, ( ( Math.log( yMax ) * Math.LOG10E + 1 ) | 0 ) - 2 ) * 3; return Math.ceil( yMax / pow3Y ) * pow3Y; }; /** * Describes getYScale * @param {number} height - calculated height of the charting space * @param {number} yMax - from `getYMax` * @returns {function} the D3 linear scale from 0 to the value from `getYMax` */ export const getYScale = ( height, yMax ) => d3ScaleLinear() .domain( [ 0, yMax ] ) .rangeRound( [ height, 0 ] ); /** * Describes getyTickOffset * @param {number} height - calculated height of the charting space * @param {number} scale - ratio of the expected width to calculated width (given the viewbox) * @param {number} yMax - from `getYMax` * @returns {function} the D3 linear scale from 0 to the value from `getYMax`, offset by 12 pixels down */ export const getYTickOffset = ( height, scale, yMax ) => d3ScaleLinear() .domain( [ 0, yMax ] ) .rangeRound( [ height + scale * 12, scale * 12 ] ); /** * Describes getyTickOffset * @param {function} xLineScale - from `getXLineScale`. * @param {function} yScale - from `getYScale`. * @returns {function} the D3 line function for plotting all category values */ export const getLine = ( xLineScale, yScale ) => d3Line() .x( d => xLineScale( new Date( d.date ) ) ) .y( d => yScale( d.value ) ); /** * Calculate the maximum number of ticks allowed in the x-axis based on the width and layout of the chart * @param {integer} width - calculated page width * @param {string} layout - standard, comparison or compact chart types * @returns {integer} number of x-axis ticks based on width and chart layout */ const calculateMaxXTicks = ( width, layout ) => { if ( width < 783 ) { return 7; } else if ( width >= 783 && width < 1129 ) { return 12; } else if ( width >= 1130 && width < 1365 ) { if ( layout === 'standard' ) { return 16; } else if ( layout === 'comparison' ) { return 12; } else if ( layout === 'compact' ) { return 7; } } else if ( width >= 1365 ) { if ( layout === 'standard' ) { return 31; } else if ( layout === 'comparison' ) { return 16; } else if ( layout === 'compact' ) { return 12; } } return 16; }; /** * Filter out irrelevant dates so only the first date of each month is kept. * @param {array} dates - string dates. * @returns {array} Filtered dates. */ const getFirstDatePerMonth = dates => { return dates.filter( ( date, i ) => i === 0 || new Date( date ).getMonth() !== new Date( dates[ i - 1 ] ).getMonth() ); }; /** * Get x-axis ticks given the unique dates and the increment factor. * @param {array} uniqueDates - all the unique dates from the input data for the chart * @param {integer} incrementFactor - increment factor for the visible ticks. * @returns {array} Ticks for the x-axis. */ const getXTicksFromIncrementFactor = ( uniqueDates, incrementFactor ) => { const ticks = []; for ( let idx = 0; idx < uniqueDates.length; idx = idx + incrementFactor ) { ticks.push( uniqueDates[ idx ] ); } // If the first or last date is missing from the ticks array, add it back in. if ( ticks[ 0 ] !== uniqueDates[ 0 ] ) { ticks.unshift( uniqueDates[ 0 ] ); } if ( ticks[ ticks.length - 1 ] !== uniqueDates[ uniqueDates.length - 1 ] ) { ticks.push( uniqueDates[ uniqueDates.length - 1 ] ); } return ticks; }; /** * Calculates the increment factor between ticks so there aren't more than maxTicks. * @param {array} uniqueDates - all the unique dates from the input data for the chart * @param {integer} maxTicks - maximum number of ticks that can be displayed in the x-axis * @returns {integer} x-axis ticks increment factor */ const calculateXTicksIncrementFactor = ( uniqueDates, maxTicks ) => { let factors = []; let i = 1; // First we get all the factors of the length of the uniqueDates array // if the number is a prime number or near prime (with 3 factors) then we // step down by 1 integer and try again. while ( factors.length <= 3 ) { factors = getFactors( uniqueDates.length - i ); i += 1; } return factors.find( f => uniqueDates.length / f < maxTicks ); }; /** * Returns ticks for the x-axis. * @param {array} uniqueDates - all the unique dates from the input data for the chart * @param {integer} width - calculated page width * @param {string} layout - standard, comparison or compact chart types * @param {string} interval - string of the interval used in the graph (hour, day, week...) * @returns {integer} number of x-axis ticks based on width and chart layout */ export const getXTicks = ( uniqueDates, width, layout, interval ) => { const maxTicks = calculateMaxXTicks( width, layout ); if ( uniqueDates.length >= dayTicksThreshold && interval === 'day' ) { uniqueDates = getFirstDatePerMonth( uniqueDates ); } if ( uniqueDates.length <= maxTicks ) { return uniqueDates; } const incrementFactor = calculateXTicksIncrementFactor( uniqueDates, maxTicks ); return getXTicksFromIncrementFactor( uniqueDates, incrementFactor ); }; /** * Describes getDateSpaces * @param {array} data - The chart component's `data` prop. * @param {array} uniqueDates - from `getUniqueDates` * @param {number} width - calculated width of the charting space * @param {function} xLineScale - from `getXLineScale` * @returns {array} that icnludes the date, start (x position) and width to layout the mouseover rectangles */ export const getDateSpaces = ( data, uniqueDates, width, xLineScale ) => uniqueDates.map( ( d, i ) => { const datapoints = find( data, { date: d } ); const xNow = xLineScale( new Date( d ) ); const xPrev = i >= 1 ? xLineScale( new Date( uniqueDates[ i - 1 ] ) ) : xLineScale( new Date( uniqueDates[ 0 ] ) ); const xNext = i < uniqueDates.length - 1 ? xLineScale( new Date( uniqueDates[ i + 1 ] ) ) : xLineScale( new Date( uniqueDates[ uniqueDates.length - 1 ] ) ); let xWidth = i === 0 ? xNext - xNow : xNow - xPrev; const xStart = i === 0 ? 0 : xNow - xWidth / 2; xWidth = i === 0 || i === uniqueDates.length - 1 ? xWidth / 2 : xWidth; return { date: d, start: uniqueDates.length > 1 ? xStart : 0, width: uniqueDates.length > 1 ? xWidth : width, values: Object.keys( datapoints ) .filter( key => key !== 'date' ) .map( key => { return { key, value: datapoints[ key ].value, date: d, }; } ), }; } ); export const drawAxis = ( node, params ) => { const xScale = params.type === 'line' ? params.xLineScale : params.xScale; const yGrids = []; for ( let i = 0; i < 4; i++ ) { yGrids.push( i / 3 * params.yMax ); } const ticks = params.xTicks.map( d => ( params.type === 'line' ? new Date( d ) : d ) ); node .append( 'g' ) .attr( 'class', 'axis' ) .attr( 'aria-hidden', 'true' ) .attr( 'transform', `translate(0, ${ params.height })` ) .call( d3AxisBottom( xScale ) .tickValues( ticks ) .tickFormat( d => params.xFormat( d instanceof Date ? d : new Date( d ) ) ) ); node .append( 'g' ) .attr( 'class', 'axis axis-month' ) .attr( 'aria-hidden', 'true' ) .attr( 'transform', `translate(0, ${ params.height + 20 })` ) .call( d3AxisBottom( xScale ) .tickValues( ticks ) .tickFormat( ( d, i ) => { const monthDate = d instanceof Date ? d : new Date( d ); let prevMonth = i !== 0 ? ticks[ i - 1 ] : ticks[ i ]; prevMonth = prevMonth instanceof Date ? prevMonth : new Date( prevMonth ); return i === 0 || params.x2Format( monthDate ) !== params.x2Format( prevMonth ) ? params.x2Format( monthDate ) : ''; } ) ) .call( g => g.select( '.domain' ).remove() ); node .selectAll( '.axis-month .tick text' ) .style( 'font-size', `${ Math.round( params.scale * 10 ) }px` ); node .append( 'g' ) .attr( 'class', 'pipes' ) .attr( 'transform', `translate(0, ${ params.height })` ) .call( d3AxisBottom( xScale ) .tickValues( ticks ) .tickSize( 5 ) .tickFormat( '' ) ); node .append( 'g' ) .attr( 'class', 'grid' ) .attr( 'transform', `translate(-${ params.margin.left },0)` ) .call( d3AxisLeft( params.yScale ) .tickValues( yGrids ) .tickSize( -params.width - params.margin.left - params.margin.right ) .tickFormat( '' ) ) .call( g => g.select( '.domain' ).remove() ); node .append( 'g' ) .attr( 'class', 'axis y-axis' ) .attr( 'aria-hidden', 'true' ) .attr( 'transform', 'translate(-50, 0)' ) .attr( 'text-anchor', 'start' ) .call( d3AxisLeft( params.yTickOffset ) .tickValues( yGrids ) .tickFormat( d => d3Format( params.yFormat )( d !== 0 ? d : 0 ) ) ); node .selectAll( '.y-axis .tick text' ) .style( 'font-size', `${ Math.round( params.scale * 10 ) }px` ); node.selectAll( '.domain' ).remove(); node .selectAll( '.axis' ) .selectAll( '.tick' ) .select( 'line' ) .remove(); }; const getTooltipRowLabel = ( d, row, params ) => d[ row.key ].labelDate ? formatDate( params.pointLabelFormat, d[ row.key ].labelDate ) : row.key; const getFormatedTotal = ( total, valueType ) => { let rowTotal = total; switch ( valueType ) { case 'average': rowTotal = Math.round( total ); break; case 'currency': rowTotal = formatCurrency( total ); break; case 'number': break; } return rowTotal; }; const showTooltip = ( params, d, position ) => { const keys = params.orderedKeys.filter( row => row.visible ).map( row => `