import type { Values } from '../../types/correlation.js';

/**
 * Sets the origin (default) label for each correlation, e.g. C1 or H2, depending on its order.
 *
 * @param {Values} correlations
 */
export function setLabels(correlations: Values): Values {
  const atomTypeCounts: Record<string, number> = {};
  for (const correlation of correlations) {
    if (!(atomTypeCounts[correlation.atomType] ?? false)) {
      atomTypeCounts[correlation.atomType] = 0;
    }
    atomTypeCounts[correlation.atomType]++;
    correlation.label.origin = `${correlation.atomType}${
      atomTypeCounts[correlation.atomType]
    }`;
  }

  return correlations;
}