import { allowedSignalKinds } from '../../constants/allowedSignalKinds.js';
import type {
  Experiment1DSignal,
  Experiment1DSignals,
  ExperimentsType,
} from '../../types/experiment.js';
import type { Spectrum1D } from '../../types/spectrum.js';
import { checkMatch } from '../general/checkMatch.js';

/**
 * Get all different 1D signals from experiments with allowed signal kinds in "signalKindsToInclude".
 *
 * @param {ExperimentsType} experiments1D
 */
export function getSignals1D(
  experiments1D: ExperimentsType,
): Experiment1DSignals {
  // store valid signals from 1D experiments
  const signals1DByAtomType: Experiment1DSignals = {};
  for (const atomType in experiments1D) {
    const signals: Experiment1DSignal[] = [];
    const experiment = experiments1D[atomType] || [];

    if (experiment.length === 0) continue;

    const index = 0;
    const spectrum1D = experiment[index] as Spectrum1D;

    for (const range of spectrum1D.ranges.values) {
      for (const signal of range.signals) {
        if (!signal.kind || !allowedSignalKinds.includes(signal.kind)) continue;
        if (!checkExistence(signal, signals)) {
          signals.push({
            experimentType: '1d',
            experimentID: spectrum1D.id,
            integration: signal.integration || range.integration,
            atomType,
            signal: { ...signal },
          });
        }
      }
    }

    signals1DByAtomType[atomType] = signals;
  }
  return signals1DByAtomType;
}

function checkExistence(current: any, group: Experiment1DSignal[]) {
  for (const element of group) {
    if (checkMatch(element.signal.delta, current.delta, 0)) {
      return true;
    }
  }
  return false;
}