import type { Experiments } from '../../types/experiment.js';
import type { Spectra } from '../../types/spectrum.js';

/**
 * Get all different experiments from spectra data.
 *
 * @param {Spectra} spectraData
 */

export function getExperiments(spectraData: Spectra): Experiments {
  if (!spectraData) return {};
  const experiments: Experiments = {};
  for (const spectrum of spectraData) {
    if (spectrum.info.isFid) continue;
    if (!experiments[`${spectrum.info.dimension}D`]) {
      experiments[`${spectrum.info.dimension}D`] = {};
    }
    const experiment = spectrum.info.experiment;
    if (!experiments[`${spectrum.info.dimension}D`][experiment]) {
      experiments[`${spectrum.info.dimension}D`][experiment] = [];
    }
    experiments[`${spectrum.info.dimension}D`][experiment].push(spectrum);
  }

  return experiments;
}