features.ts

// eslint-disable-next-line @typescript-eslint/no-explicit-any
export default function extractFeatures(audioData: any) {
    const features: number[] = [];

    for (const [algorithm, identifier] of Object.entries(algorithms)) {
        const algorithmFunc = essentia[algorithm];

        const value = algorithmFunc.call(essentia, audioData);

        if (algorithm === "KeyExtractor") {
            // Parse key returns array of 2 values
            features.push(...parseKey(value));
        } else {
            value[identifier] = Number(value[identifier].toFixed(2));
            features.push(value[identifier]);
        }
    }

    return features;
}