pokemon-color-search/web/score.js

const currentScores = {};
const currentBestClusterIndices = {};

const rescoreAll = target => pokemonData.forEach(({ name, jab, rgb }) => {
  currentScores[name] = {
    jab: {
      total: applyMetrics(jab.total, target.jab),
      clusters: jab.clusters.map(c => applyMetrics(c, target.jab)),
    },
    rgb: {
      total: applyMetrics(rgb.total, target.rgb),
      clusters: rgb.clusters.map(c => applyMetrics(c, target.rgb)),
    },
  };
});

const getBestClusterIndex = (pkmn, space, { sortMetric, scaleOption, sortOrder }) => {
  // get the scales
  const scales = scaleOption(pkmn[space]);
  // and multiply with the intended metric, and find the index of the best value
  return currentScores[pkmn.name][space].clusters
    .map((c, i) => [c[sortMetric] * scales[i], i])
    .reduce((a, b) => sortOrder(a[0], b[0]) > 0 ? b : a)[1];
}

const getBest = (number, space, clusterSettings, { sortMetric, scaleOption, sortOrder }) => {
  let valueExtractor;
  if (clusterSettings) {
    valueExtractor = pkmn => {
      const index = getBestClusterIndex(pkmn, space, clusterSettings);
      // save the index for rendering
      currentBestClusterIndices[pkmn.name] = { ...(currentBestClusterIndices[pkmn.name] || {}), [space]: index };
      // and then get the *actual* score according to the sort metric
      return scaleOption(pkmn[space])[index] * currentScores[pkmn.name][space].clusters[index][sortMetric];      
    };
  } else {
    // ignore scaleOption if not using clusters
    valueExtractor = pkmn => currentScores[pkmn.name][space].total[sortMetric];
  }
  
  return pokemonData
    .slice()
    .sort((a, b) => sortOrder(valueExtractor(a), valueExtractor(b)))
    .slice(0, number);
};