135 lines
8.2 KiB
JavaScript
135 lines
8.2 KiB
JavaScript
export function specModel(options) {
|
||
let result = {}; //результат
|
||
let model = options.src.base; //массив точек базовой линии для построения упрощенной модели спектра
|
||
let imin = 0; //левая граница коридора
|
||
let imax = 0; //правая граница коридора
|
||
for (let i = 0; i < options.src.peaks.length; i++) {
|
||
let freq = options.src.peaks[i].freq; //получаем значение частоты гармоники из массива обнаруженных гармоник
|
||
let level = options.src.peaks[i].level; //получаем значение уровня гармоники из массива обнаруженных гармоник
|
||
imin = Math.round((freq - freq * 0.5 * options.tol / 100) / options.src.resolution);
|
||
imax = Math.round((freq + freq * 0.5 * options.tol / 100) / options.src.resolution);
|
||
if (imax > options.src.base.length) { imax = options.src.base.length - 1 }; //проверяем выход границы за размер массива
|
||
for (let j = imin; j <= imax; j++) { model[j] = options.src.base[j] + level };
|
||
};
|
||
|
||
//формируем результат
|
||
result["name"] = options.src.name + "_model";
|
||
result["resolution"] = options.src.resolution;
|
||
result["data"] = model;
|
||
|
||
//отрисовка графика на plot
|
||
if (options.canvas != undefined) {
|
||
options.canvas.add(
|
||
{
|
||
color: options.color,
|
||
name: result.name,
|
||
x: result.resolution,
|
||
y: model
|
||
}
|
||
);
|
||
};
|
||
|
||
return result;
|
||
}; //построение упрощенной модели спектра
|
||
|
||
export function getMask(options) {
|
||
let result = {}; //результат
|
||
let data = options.src1.data; //массив точек спектра для сравнения с портретом
|
||
let mask = options.src2.base; //массив точек базовой линии для построения портрета дефекта
|
||
let harms = options.harms; //кол-во гармоник портрета
|
||
if (options.freq * harms > options.src2.frequency) { harms = Math.trunc(options.src2.frequency / options.freq) }; //проверяем максимальное кол-во гармоник в портрете
|
||
let df = 0; //отношение ширины фильтра частотному разрешению спектра
|
||
let dl = options.lvl; //разность уровней гармонической и случайной составляющей вибрации
|
||
if (options.filter != 0) {
|
||
df = options.src2.frequency / options.filter.width;
|
||
dl = 10 * Math.log10((options.lvl ** 2) / df + 1);
|
||
};
|
||
let k = options.coef; //коэффициент затухания
|
||
let x = 0; //индекс элемента массива для гармоники
|
||
let dx = 0; //коридор индексов
|
||
let xmin = 0; //левая граница коридора
|
||
let xmax = 0; //правая граница коридора
|
||
let correlation = 0; //коэффициент корреляции
|
||
|
||
function corr(a, b) {
|
||
let X = 0; //аргумент 1
|
||
let Y = 0; //аргумент 2
|
||
let Z = 0; //аргумент 3
|
||
let a_avg = a.reduce((acc, item) => (acc + item)) / a.length; //среднее значение массива 1
|
||
let b_avg = b.reduce((acc, item) => (acc + item)) / b.length; //среднее значение массива 2
|
||
|
||
for (let i = 0; i < a.length; i++) {
|
||
X += (a[i] - a_avg) * (b[i] - b_avg);
|
||
Y += (a[i] - a_avg) ** 2;
|
||
Z += (b[i] - b_avg) ** 2;
|
||
};
|
||
|
||
if (X <= 0) { return 0 } else { return X / (Math.sqrt(Y) * Math.sqrt(Z)) };
|
||
};
|
||
|
||
let arr1 = [];
|
||
let arr2 = [];
|
||
switch (options.type) {
|
||
case 0: //обычный ряд
|
||
for (let i = 1; i <= harms; i++) {
|
||
x = Math.trunc(i * options.freq / options.src2.resolution); //определяем инднекс элемента массива для гармоники
|
||
dx = Math.trunc(i * (options.freq * options.tol / 100) / options.src2.resolution); //коридор индексов массива для гармоники
|
||
xmin = x - Math.round(dx / 2); //определяем индекс левой границы коридора
|
||
xmax = x + Math.round(dx / 2); //определяем индекс правой границы коридора
|
||
|
||
if (xmax > options.src2.base.length) { xmax = options.src2.base.length - 2 }; //проверяем выход границы за размер массива
|
||
for (let j = xmin; j <= xmax; j++) { mask[j] = options.src2.base[j] + dl }; //записываем значение глубины модуляции для коридора
|
||
dl = dl - (k * dl); //снижаем глубину модуляции с коэффициентом затухания
|
||
|
||
for (let j = xmin - 2; j <= xmax + 2; j++) {
|
||
arr1.push(mask[j]);
|
||
arr2.push(data[j]);
|
||
}; //формируем массивы портретной гармоники и спектра (модели) под портретом
|
||
correlation += corr(arr1, arr2); //рассчитываем корреляцию данных из массивов
|
||
};
|
||
break;
|
||
case 1: //четные составляющие
|
||
for (let i = 1; i <= harms; i++) {
|
||
x = Math.trunc(i * options.freq / options.src2.resolution); //инднекс элемента массива для гармоники
|
||
dx = Math.trunc(i * (options.freq * options.tol / 100) / options.src2.resolution); //коридор индексов массива для гармоники
|
||
xmin = x - Math.round(dx / 2); //определяем индекс левой границы коридора
|
||
xmax = x + Math.round(dx / 2); //определяем индекс правой границы коридора
|
||
|
||
if (xmax > options.src2.base.length) { xmax = options.src2.base.length - 2 }; //проверяем выход границы за размер массива
|
||
if (i % 2 > 0) {
|
||
for (let j = xmin; j <= xmax; j++) { mask[j] = options.src2.base[j] + k * dl }; //записываем значение глубины модуляции для коридора (нечетная гармоника)
|
||
} else {
|
||
for (let j = xmin; j <= xmax; j++) { mask[j] = options.src2.base[j] + dl }; //записываем значение глубины модуляции для коридора (четная гармоника)
|
||
dl = dl - (k * dl); //снижаем глубину модуляции с коэффициентом затухания
|
||
};
|
||
|
||
for (let j = xmin - 2; j <= xmax + 2; j++) {
|
||
arr1.push(mask[j]);
|
||
arr2.push(data[j]);
|
||
}; //формируем массивы портретной гармоники и спектра (модели) под портретом
|
||
correlation += corr(arr1, arr2); //рассчитываем корреляцию данных из массивов
|
||
};
|
||
break;
|
||
default:
|
||
break;
|
||
};
|
||
|
||
//формируем результат
|
||
result["name"] = options.name; //имя маски дефекта
|
||
result["corr"] = correlation / harms; //средняя корреляция
|
||
result["data"] = mask; //массив данных портрета
|
||
|
||
//отрисовка графика на plot
|
||
if (options.canvas != undefined) {
|
||
options.canvas.add(
|
||
{
|
||
color: options.color,
|
||
name: options.name + " (" + result.corr.toFixed(2) + ")",
|
||
x: options.src2.resolution,
|
||
y: mask
|
||
}
|
||
);
|
||
};
|
||
|
||
return result;
|
||
}; //построение портрета дефекта
|