Skip to content

mljs/spectra-fitting

Repository files navigation

ml-spectra-fitting

NPM version build status npm download

This is a spectra fitting package to optimize the position (x), max intensity (y), full width at half maximum (FWHM = width) and the ratio of gaussian contribution (mu) if it's required. It supports three kind of shapes:

Name Equation
Gaussian
Lorentzian
Pseudo Voigt

where

It is a wrapper of ml-levenberg-marquardt

Installation

$ npm install ml-spectra-fitting

Example

import { optimize } from 'ml-spectra-fitting';
import { SpectrumGenerator } from 'spectrum-generator';

const generator = new SpectrumGenerator({
  nbPoints: 101,
  from: -1,
  to: 1,
});

// by default the kind of shape is gaussian;
generator.addPeak({ x: 0.5, y: 0.2 }, { fwhm: 0.2 });
generator.addPeak(
  { x: -0.5, y: 0.2 },
  {
    shape: {
      kind: 'lorentzian',
      fwhm: 0.1,
    },
  },
);

//points to fit {x, y};
let data = generator.getSpectrum();
console.log(JSON.stringify({ x: Array.from(data.x), y: Array.from(data.y) }));
//the approximate values to be optimized, It could coming from a peak picking with ml-gsd
let peaks = [
  {
    x: -0.5,
    y: 0.22,
    shape: {
      kind: 'gaussian',
      fwhm: 0.25,
    },
  },
  {
    x: 0.52,
    y: 0.18,
    shape: {
      kind: 'gaussian',
      fwhm: 0.18,
    },
  },
];

// the function receive an array of peak with {x, y, fwhm} as a guess
// and return a list of objects
let fittedParams = optimize(data, peaks, { shape: { kind: 'pseudoVoigt' } });

console.log(fittedParams);
const result = {
  error: 0.12361588652854476,
  iterations: 100,
  peaks: [
    {
      x: -0.5000014532421942,
      y: 0.19995307937326137,
      shape: {
        kind: 'pseudoVoigt',
        fwhm: 0.10007670374735196,
        mu: 0.004731136777288483,
      },
    },
    {
      x: 0.5001051783652894,
      y: 0.19960010175400406,
      shape: {
        kind: 'pseudoVoigt',
        fwhm: 0.19935932346969124,
        mu: 1,
      },
    },
  ],
};

For data with and combination of signals with shapes between gaussian and lorentzians, we could use the kind pseudovoigt to fit the data.

import { optimize } from 'ml-spectra-fitting';
import { SpectrumGenerator } from 'spectrum-generator';

const generator = new SpectrumGenerator({
  nbPoints: 101,
  from: -1,
  to: 1,
});

// by default the kind of shape is gaussian;
generator.addPeak({ x: 0.5, y: 0.2 }, { fwhm: 0.2 });
generator.addPeak(
  { x: -0.5, y: 0.2 },
  {
    shape: {
      kind: 'lorentzian',
      fwhm: 0.1,
    },
  },
);

//points to fit {x, y};
let data = generator.getSpectrum();
console.log(JSON.stringify({ x: Array.from(data.x), y: Array.from(data.y) }));
//the approximate values to be optimized, It could coming from a peak picking with ml-gsd
let peaks = [
  {
    x: -0.5,
    y: 0.22,
    shape: {
      kind: 'gaussian',
      fwhm: 0.25,
    },
  },
  {
    x: 0.52,
    y: 0.18,
    shape: {
      kind: 'gaussian',
      fwhm: 0.18,
    },
  },
];

// the function receive an array of peak with {x, y, fwhm} as a guess
// and return a list of objects
let fittedParams = optimize(data, peaks, { shape: { kind: 'pseudoVoigt' } });

console.log(fittedParams);
const result = {
  error: 0.12361588652854476,
  iterations: 100,
  peaks: [
    {
      x: -0.5000014532421942,
      y: 0.19995307937326137,
      shape: {
        kind: 'pseudoVoigt',
        fwhm: 0.10007670374735196,
        mu: 0.004731136777288483,
      },
    },
    {
      x: 0.5001051783652894,
      y: 0.19960010175400406,
      shape: {
        kind: 'pseudoVoigt',
        fwhm: 0.19935932346969124,
        mu: 1,
      },
    },
  ],
};

License

MIT