infect-frontend/www/src/js/models/matrix/calculateXPositions.js

infect fronend

import getArrayDiff from '../../helpers/getArrayDiff';
import debug from 'debug';
const log = debug('infect:calculateXPositions');

/**
* Gets coordinates for all elements (antibiotics, substance classes) on x axes. Depends on view
* information (diameter etc. that depends on matrix size).
*
* @param {Array} sortedAntibiotics			Array as returned by AntibioticStore.sortedAntibiotics
* @param {Number} diameter					diameter of a single resistance
* @param {Number} space						Space between resistances (of an antibiotic)
* @returns {Map} x positions				Map with
*											- keys for every antibiotic and substance class
*											- value x positions for every entry (object with properties
*											  left and right) *or* undefined if substanceClass is not visible
*											  because parent substance class is contracted.
*/
export default function calculateXPositions(sortedAntibiotics, diameter, space) {

	// Get all visible antibiotics
	const visibleAB = sortedAntibiotics.filter((ab) => ab.visible);
	log('Visible antibiotics are %o', visibleAB);

	// Return value:
	// Key: antibiotic or substanceClass
	// Value: {left: x, right: x}; right is needed for substanceClasses only.
	const xMap = new Map();

	// Loop all visible antibiotics
	visibleAB.reduce((prev, current, index) => {


		// Get substance class of current antibiotic
		const substanceClasses = current.antibiotic.getSubstanceClasses();


		// CONTRACTED
		// Contracted substance classes, parent first
		const contractedClasses = substanceClasses.filter((sClass) => !sClass.expanded).reverse();
		// Contracted class is same as in previous loop: Just return
		const unchangedContractedClass = contractedClasses.length && contractedClasses[0] === prev.contractedSubstanceClass;
		prev.contractedSubstanceClass = contractedClasses.length ? contractedClasses[0] : undefined;
		if (unchangedContractedClass) {
			return prev;
		}


		// Get changes in substance classes (substance class chain of of current ab compared to 
		// substance class chain of previous ab)
		const diff = getArrayDiff(prev.previousSubstanceClasses, current.antibiotic.getSubstanceClasses());
		prev.previousSubstanceClasses = current.antibiotic.getSubstanceClasses();


		// Substance class removed (from previous to current antibiotic): Add right property
		diff.removed.forEach((sClass) => {
			xMap.get(sClass).right = prev.x;
		});


		// Substance class added (from previous to current antibiotic):
		// Add space before new substanceClass divider. Only one divider is drawn, even if multiple
		// substanceClasses were added.
		if (diff.added.length) prev.x += space;
		diff.added.forEach((sClass) => {
			xMap.set(sClass, {
				left: prev.x
			});
		});


		// Antibiotic
		prev.x += space;
		xMap.set(current, {
			left: prev.x
			, right: prev.x + diameter
		});


		// Last round: Add right property to all currently open substanceClasses
		if (index === visibleAB.length - 1) substanceClasses.forEach((sClass) => xMap.get(sClass).right = prev.x + diameter);


		prev.x += diameter;
		return prev;

	}, {
		// Used to detect changes in classes
		previousSubstanceClasses: []
		// Used to only add 1 diameter if classes are contracted. If contracted
		// class changes, we need to add another one. Always take highest ranking
		// contracted substance class (grand parent over parent)
		, contractedSubstanceClass: undefined
		, x: 0
	});

	log('xPositions are %o', xMap);
	return xMap;

}