nodejs-audio-visualizer 2.1.4 → 4.0.0

package/src/tests/dsp.test.ts

@@ -16,10 +16,9 @@ describe('fft', function () {
   });
 
   it('getSpectrum', function () {
-    var signal = [1, 0, 0, 0, 1, 0, 0, 0];
-    const fft = [8.317766166719343, 0, 11.090354888959125, 0];
+    const signal = Array(Math.pow(2, 12)).fill(1);
 
-    const result = getSpectrum(signal);
-    expect(result).deep.equal(fft);
+    const result = getSpectrum(signal, 44100);
+    expect(result).to.have.length(24);
   });
 });

package/src/tests/image.test.ts

@@ -1,10 +1,10 @@
 import { expect } from 'chai';
 import path from 'path';
-import fs from 'fs';
 import {
   drawRect,
   Color,
-  createVisualizerFrame,
+  createSpectrumVisualizerFrameGenerator,
+  createPolarVisualizerFrameGenerator,
   parseImage,
   convertToBmp,
   mixValues,
@@ -12,10 +12,6 @@ import {
 } from '../image';
 import { createBpmEncoder } from '../bpmEncoder';
 
-import frameSnapshotSpectrumDown from './snapshots/frameSnapshotSpectrumDown.json';
-import frameSnapshotSpectrumUp from './snapshots/frameSnapshotSpectrumUp.json';
-import frameSnapshotSpectrumUpOpacity50 from './snapshots/frameSnapshotSpectrumUpOpacity50.json';
-
 describe('image', function () {
   it('mixValues', function () {
     expect(mixValues(420, 0.69, 96)).equal(320);
@@ -81,12 +77,14 @@ describe('image', function () {
     expect(imageDstBuffer.data).deep.equal(expectedImageData);
   });
 
-  it('createVisualizerFrame', async function () {
-    const backgroundImagePath = path.resolve('example/media/background.png');
+  it('createSpectrumVisualizerFrameGenerator', async function () {
+    const expectedImageLength = 6220854;
+    const backgroundImagePath = path.resolve('example/media/horses.png');
     const backgroundImageBmpBuffer = await convertToBmp(backgroundImagePath);
     const backgroundImage = parseImage(backgroundImageBmpBuffer);
     const bpmEncoder = createBpmEncoder({ width: backgroundImage.width, height: backgroundImage.height });
     const backgroundImageBuffer = bpmEncoder(backgroundImage.data);
+    const createVisualizerFrame = createSpectrumVisualizerFrameGenerator();
     const frameSpectrumDown = createVisualizerFrame({
       backgroundImageBuffer,
       spectrum: [0.5, 0, 1],
@@ -98,7 +96,7 @@ describe('image', function () {
       opacity: 1,
     });
     const resultSpectrumDown = frameSpectrumDown.data.toJSON().data;
-    expect(resultSpectrumDown).deep.equal(frameSnapshotSpectrumDown);
+    expect(resultSpectrumDown).to.have.length(expectedImageLength);
 
     const frameSpectrumUp = createVisualizerFrame({
       backgroundImageBuffer,
@@ -111,7 +109,20 @@ describe('image', function () {
       opacity: 1,
     });
     const resultSpectrumUp = frameSpectrumUp.data.toJSON().data;
-    expect(resultSpectrumUp).deep.equal(frameSnapshotSpectrumUp);
+    expect(resultSpectrumUp).to.have.length(expectedImageLength);
+
+    const frameSpectrumMirror = createVisualizerFrame({
+      backgroundImageBuffer,
+      spectrum: [0.1, 1, 0],
+      size: { width: 25, height: 23 },
+      position: { x: 10, y: 5 },
+      rotation: 'mirror',
+      margin: 4,
+      color: { red: 0, green: 123, blue: 69 },
+      opacity: 1,
+    });
+    const resultSpectrumMirror = frameSpectrumMirror.data.toJSON().data;
+    expect(resultSpectrumMirror).to.have.length(expectedImageLength);
 
     const frameSpectrumUpOpacity50 = createVisualizerFrame({
       backgroundImageBuffer,
@@ -124,6 +135,88 @@ describe('image', function () {
       opacity: 0.5,
     });
     const resultSpectrumUpOpacity50 = frameSpectrumUpOpacity50.data.toJSON().data;
-    expect(resultSpectrumUpOpacity50).deep.equal(frameSnapshotSpectrumUpOpacity50);
+    expect(resultSpectrumUpOpacity50).to.have.length(expectedImageLength);
+  });
+
+  it('createPolarVisualizerFrameGenerator', async function () {
+    const expectedImageLength = 6220854;
+    const backgroundImagePath = path.resolve('example/media/horses.png');
+    const backgroundImageBmpBuffer = await convertToBmp(backgroundImagePath);
+    const backgroundImage = parseImage(backgroundImageBmpBuffer);
+    const bpmEncoder = createBpmEncoder({ width: backgroundImage.width, height: backgroundImage.height });
+    const backgroundImageBuffer = bpmEncoder(backgroundImage.data);
+    const createPolarVisualizerFrame = createPolarVisualizerFrameGenerator();
+
+    const framePolarBasic = createPolarVisualizerFrame({
+      backgroundImageBuffer,
+      spectrum: [0.5, 0.8, 0.3, 0.9, 0.6, 0.4, 0.7, 0.2],
+      centerX: 500,
+      centerY: 400,
+      innerRadius: 100,
+      maxBarLength: 150,
+      barWidth: 5,
+      color: { red: 255, green: 100, blue: 50 },
+      opacity: 1,
+    });
+    const resultPolarBasic = framePolarBasic.data.toJSON().data;
+    expect(resultPolarBasic).to.have.length(expectedImageLength);
+
+    const framePolarVolume = createPolarVisualizerFrame({
+      backgroundImageBuffer,
+      spectrum: [0.2, 0.5, 0.8, 0.3, 0.6, 0.9, 0.4, 0.7],
+      centerX: 500,
+      centerY: 400,
+      innerRadius: 80,
+      maxBarLength: 120,
+      barWidth: 4,
+      color: { red: 100, green: 200, blue: 255 },
+      opacity: 1,
+      spectrumEffect: 'volume',
+    });
+    const resultPolarVolume = framePolarVolume.data.toJSON().data;
+    expect(resultPolarVolume).to.have.length(expectedImageLength);
+
+    const framePolarSmooth = createPolarVisualizerFrame({
+      backgroundImageBuffer,
+      spectrum: [0.7, 0.4, 0.9, 0.2, 0.5, 0.8, 0.3, 0.6],
+      centerX: 500,
+      centerY: 400,
+      innerRadius: 90,
+      maxBarLength: 130,
+      barWidth: 6,
+      color: { red: 255, green: 255, blue: 0 },
+      opacity: 1,
+      spectrumEffect: 'smooth',
+    });
+    const resultPolarSmooth = framePolarSmooth.data.toJSON().data;
+    expect(resultPolarSmooth).to.have.length(expectedImageLength);
+
+    const framePolarOpacity50 = createPolarVisualizerFrame({
+      backgroundImageBuffer,
+      spectrum: [0.6, 0.3, 0.8, 0.5, 0.4, 0.7, 0.2, 0.9],
+      centerX: 500,
+      centerY: 400,
+      innerRadius: 100,
+      maxBarLength: 140,
+      barWidth: 5,
+      color: { red: 0, green: 255, blue: 128 },
+      opacity: 0.5,
+    });
+    const resultPolarOpacity50 = framePolarOpacity50.data.toJSON().data;
+    expect(resultPolarOpacity50).to.have.length(expectedImageLength);
+
+    const framePolarHexColor = createPolarVisualizerFrame({
+      backgroundImageBuffer,
+      spectrum: [0.5, 0.6, 0.7, 0.8, 0.9, 0.4, 0.3, 0.2],
+      centerX: 500,
+      centerY: 400,
+      innerRadius: 110,
+      maxBarLength: 160,
+      barWidth: 7,
+      color: '#FF5733',
+      opacity: 1,
+    });
+    const resultPolarHexColor = framePolarHexColor.data.toJSON().data;
+    expect(resultPolarHexColor).to.have.length(expectedImageLength);
   });
 });

package/src/tests/index.test.ts

@@ -42,7 +42,7 @@ describe('index', function () {
         path: 'audioPath'
       },
       image: {
-        path: 'example/media/background.png'
+        path: 'example/media/horses.png'
       },
       outVideo: {
         path: 'outVideoPath',

package/src/typings.d.ts

@@ -0,0 +1,7 @@
+declare module 'fft-js' {
+  export function fft(signal: number[]): [number, number][];
+  export const util: {
+    fftFreq(phasors: [number, number][], sampleRate: number): number[];
+    fftMag(phasors: [number, number][]): number[];
+  };
+}

package/dist/spectrum.d.ts

@@ -1 +0,0 @@
-export declare const createSpectrumAnalyzer: (bufferSize: number) => (buffer: number[]) => number[];

package/dist/spectrum.js

@@ -1,72 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.createSpectrumAnalyzer = function (bufferSize) {
-    var real = new Float64Array(bufferSize);
-    var imag = new Float64Array(bufferSize);
-    var reverseTable = new Uint32Array(bufferSize);
-    var limit = 1;
-    var bit = bufferSize >> 1;
-    while (limit < bufferSize) {
-        for (var i = 0; i < limit; i++) {
-            reverseTable[i + limit] = reverseTable[i] + bit;
-        }
-        limit = limit << 1;
-        bit = bit >> 1;
-    }
-    var sinTable = new Float64Array(bufferSize);
-    var cosTable = new Float64Array(bufferSize);
-    for (var i = 0; i < bufferSize; i++) {
-        sinTable[i] = Math.sin(-Math.PI / i);
-        cosTable[i] = Math.cos(-Math.PI / i);
-    }
-    var calculateSpectrum = function () {
-        var spectrum = new Array(bufferSize / 2);
-        var bSi = 2 / bufferSize;
-        for (var i = 0, N = bufferSize / 2; i < N; i++) {
-            var rval = real[i];
-            var ival = imag[i];
-            var mag = bSi * Math.sqrt(rval * rval + ival * ival);
-            spectrum[i] = mag;
-        }
-        return spectrum;
-    };
-    return function (buffer) {
-        var k = Math.floor(Math.log(bufferSize) / Math.LN2);
-        if (Math.pow(2, k) !== bufferSize) {
-            throw new Error('Invalid buffer size, must be a power of 2');
-        }
-        if (bufferSize !== buffer.length) {
-            throw new Error("Supplied buffer is not the same size as defined FFT. FFT Size: " + bufferSize + " Buffer Size: " + buffer.length);
-        }
-        var halfSize = 1;
-        for (var i = 0; i < bufferSize; i++) {
-            real[i] = buffer[reverseTable[i]];
-            imag[i] = 0;
-        }
-        while (halfSize < bufferSize) {
-            var phaseShiftStepReal = cosTable[halfSize];
-            var phaseShiftStepImag = sinTable[halfSize];
-            var currentPhaseShiftReal = 1;
-            var currentPhaseShiftImag = 0;
-            for (var fftStep = 0; fftStep < halfSize; fftStep++) {
-                var i = fftStep;
-                while (i < bufferSize) {
-                    var off = i + halfSize;
-                    var tr = (currentPhaseShiftReal * real[off]) - (currentPhaseShiftImag * imag[off]);
-                    var ti = (currentPhaseShiftReal * imag[off]) + (currentPhaseShiftImag * real[off]);
-                    real[off] = real[i] - tr;
-                    imag[off] = imag[i] - ti;
-                    real[i] += tr;
-                    imag[i] += ti;
-                    i += halfSize << 1;
-                }
-                var tmpReal = currentPhaseShiftReal;
-                currentPhaseShiftReal = (tmpReal * phaseShiftStepReal) - (currentPhaseShiftImag * phaseShiftStepImag);
-                currentPhaseShiftImag = (tmpReal * phaseShiftStepImag) + (currentPhaseShiftImag * phaseShiftStepReal);
-            }
-            halfSize = halfSize << 1;
-        }
-        return calculateSpectrum();
-    };
-};
-//# sourceMappingURL=spectrum.js.map

package/dist/spectrum.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"spectrum.js","sourceRoot":"","sources":["../src/spectrum.ts"],"names":[],"mappings":";;AAAa,QAAA,sBAAsB,GAAG,UAAC,UAAkB;IACvD,IAAM,IAAI,GAAG,IAAI,YAAY,CAAC,UAAU,CAAC,CAAC;IAC1C,IAAM,IAAI,GAAG,IAAI,YAAY,CAAC,UAAU,CAAC,CAAC;IAC1C,IAAM,YAAY,GAAG,IAAI,WAAW,CAAC,UAAU,CAAC,CAAC;IAEjD,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,IAAI,GAAG,GAAG,UAAU,IAAI,CAAC,CAAC;IAE1B,OAAO,KAAK,GAAG,UAAU,EAAE;QACzB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,EAAE,CAAC,EAAE,EAAE;YAC9B,YAAY,CAAC,CAAC,GAAG,KAAK,CAAC,GAAG,YAAY,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC;SACjD;QAED,KAAK,GAAG,KAAK,IAAI,CAAC,CAAC;QACnB,GAAG,GAAG,GAAG,IAAI,CAAC,CAAC;KAChB;IAED,IAAM,QAAQ,GAAG,IAAI,YAAY,CAAC,UAAU,CAAC,CAAC;IAC9C,IAAM,QAAQ,GAAG,IAAI,YAAY,CAAC,UAAU,CAAC,CAAC;IAE9C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,EAAE,CAAC,EAAE,EAAE;QACnC,QAAQ,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC;QACrC,QAAQ,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC;KACtC;IAED,IAAM,iBAAiB,GAAG;QACxB,IAAM,QAAQ,GAAa,IAAI,KAAK,CAAC,UAAU,GAAG,CAAC,CAAC,CAAC;QACrD,IAAM,GAAG,GAAG,CAAC,GAAG,UAAU,CAAC;QAE3B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE;YAC9C,IAAM,IAAI,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;YACrB,IAAM,IAAI,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;YACrB,IAAM,GAAG,GAAG,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,GAAG,IAAI,GAAG,IAAI,GAAG,IAAI,CAAC,CAAC;YAEvD,QAAQ,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC;SACnB;QAED,OAAO,QAAQ,CAAC;IAClB,CAAC,CAAA;IAED,OAAO,UAAC,MAAgB;QACtB,IAAM,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC;QAEtD,IAAI,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC,KAAK,UAAU,EAAE;YACjC,MAAM,IAAI,KAAK,CAAC,2CAA2C,CAAC,CAAC;SAC9D;QACD,IAAI,UAAU,KAAK,MAAM,CAAC,MAAM,EAAE;YAChC,MAAM,IAAI,KAAK,CAAC,oEAAkE,UAAU,sBAAiB,MAAM,CAAC,MAAQ,CAAC,CAAC;SAC/H;QAED,IAAI,QAAQ,GAAG,CAAC,CAAC;QAEjB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,EAAE,CAAC,EAAE,EAAE;YACnC,IAAI,CAAC,CAAC,CAAC,GAAG,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC;YAClC,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;SACb;QAED,OAAO,QAAQ,GAAG,UAAU,EAAE;YAC5B,IAAM,kBAAkB,GAAG,QAAQ,CAAC,QAAQ,CAAC,CAAC;YAC9C,IAAM,kBAAkB,GAAG,QAAQ,CAAC,QAAQ,CAAC,CAAC;YAE9C,IAAI,qBAAqB,GAAG,CAAC,CAAC;YAC9B,IAAI,qBAAqB,GAAG,CAAC,CAAC;YAE9B,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,GAAG,QAAQ,EAAE,OAAO,EAAE,EAAE;gBACnD,IAAI,CAAC,GAAG,OAAO,CAAC;gBAEhB,OAAO,CAAC,GAAG,UAAU,EAAE;oBACrB,IAAM,GAAG,GAAG,CAAC,GAAG,QAAQ,CAAC;oBACzB,IAAM,EAAE,GAAG,CAAC,qBAAqB,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,qBAAqB,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;oBACrF,IAAM,EAAE,GAAG,CAAC,qBAAqB,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,qBAAqB,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;oBAErF,IAAI,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC;oBACzB,IAAI,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC;oBACzB,IAAI,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;oBACd,IAAI,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;oBAEd,CAAC,IAAI,QAAQ,IAAI,CAAC,CAAC;iBACpB;gBAED,IAAM,OAAO,GAAG,qBAAqB,CAAC;gBACtC,qBAAqB,GAAG,CAAC,OAAO,GAAG,kBAAkB,CAAC,GAAG,CAAC,qBAAqB,GAAG,kBAAkB,CAAC,CAAC;gBACtG,qBAAqB,GAAG,CAAC,OAAO,GAAG,kBAAkB,CAAC,GAAG,CAAC,qBAAqB,GAAG,kBAAkB,CAAC,CAAC;aACvG;YAED,QAAQ,GAAG,QAAQ,IAAI,CAAC,CAAC;SAC1B;QAED,OAAO,iBAAiB,EAAE,CAAC;IAC7B,CAAC,CAAC;AACJ,CAAC,CAAC"}

package/src/spectrum.ts

@@ -1,91 +0,0 @@
-export const createSpectrumAnalyzer = (bufferSize: number) => {
-  const real = new Float64Array(bufferSize);
-  const imag = new Float64Array(bufferSize);
-  const reverseTable = new Uint32Array(bufferSize);
-
-  let limit = 1;
-  let bit = bufferSize >> 1;
-
-  while (limit < bufferSize) {
-    for (let i = 0; i < limit; i++) {
-      reverseTable[i + limit] = reverseTable[i] + bit;
-    }
-
-    limit = limit << 1;
-    bit = bit >> 1;
-  }
-
-  const sinTable = new Float64Array(bufferSize);
-  const cosTable = new Float64Array(bufferSize);
-
-  for (let i = 0; i < bufferSize; i++) {
-    sinTable[i] = Math.sin(-Math.PI / i);
-    cosTable[i] = Math.cos(-Math.PI / i);
-  }
-
-  const calculateSpectrum = () => {
-    const spectrum: number[] = new Array(bufferSize / 2);
-    const bSi = 2 / bufferSize;
-
-    for (var i = 0, N = bufferSize / 2; i < N; i++) {
-      const rval = real[i];
-      const ival = imag[i];
-      const mag = bSi * Math.sqrt(rval * rval + ival * ival);
-
-      spectrum[i] = mag;
-    }
-
-    return spectrum;
-  }
-
-  return (buffer: number[]) => {
-    const k = Math.floor(Math.log(bufferSize) / Math.LN2);
-
-    if (Math.pow(2, k) !== bufferSize) {
-      throw new Error('Invalid buffer size, must be a power of 2');
-    }
-    if (bufferSize !== buffer.length) {
-      throw new Error(`Supplied buffer is not the same size as defined FFT. FFT Size: ${bufferSize} Buffer Size: ${buffer.length}`);
-    }
-
-    let halfSize = 1;
-
-    for (let i = 0; i < bufferSize; i++) {
-      real[i] = buffer[reverseTable[i]];
-      imag[i] = 0;
-    }
-
-    while (halfSize < bufferSize) {
-      const phaseShiftStepReal = cosTable[halfSize];
-      const phaseShiftStepImag = sinTable[halfSize];
-
-      let currentPhaseShiftReal = 1;
-      let currentPhaseShiftImag = 0;
-
-      for (var fftStep = 0; fftStep < halfSize; fftStep++) {
-        let i = fftStep;
-
-        while (i < bufferSize) {
-          const off = i + halfSize;
-          const tr = (currentPhaseShiftReal * real[off]) - (currentPhaseShiftImag * imag[off]);
-          const ti = (currentPhaseShiftReal * imag[off]) + (currentPhaseShiftImag * real[off]);
-
-          real[off] = real[i] - tr;
-          imag[off] = imag[i] - ti;
-          real[i] += tr;
-          imag[i] += ti;
-
-          i += halfSize << 1;
-        }
-
-        const tmpReal = currentPhaseShiftReal;
-        currentPhaseShiftReal = (tmpReal * phaseShiftStepReal) - (currentPhaseShiftImag * phaseShiftStepImag);
-        currentPhaseShiftImag = (tmpReal * phaseShiftStepImag) + (currentPhaseShiftImag * phaseShiftStepReal);
-      }
-
-      halfSize = halfSize << 1;
-    }
-
-    return calculateSpectrum();
-  };
-};