用于信号处理的奇数函数?

时间:2018-09-18 06:51:06

标签: javascript signal-processing fft

嗨!我希望这是一个可以接受的问题。

通过一些用于信号处理的代码,我遇到了一个奇怪的函数:

let kInd = (k1, pow) => {

  let k2 = 0;
  let k3 = 0;

  for (let i = 0; i < pow; i++) {
    k3 = k1 >> 1;
    k2 = 2 * (k2 - k3) + k1;
    k1 = k3;
  }

  return k2;

};

在傅立叶变换计算结束时调用此函数,以交换实数+虚数数组对中的索引:

let fft = samples => {

  let pow = Math.log2(samples.length); // `samples.length` is expected to be 2^int

  // ... a bunch of code to generate `rBuff` and `iBuff` arrays representing 
  // real and imaginary components of fourier values

  // Now make use of `kInd`; conditionally swap some indexes in `rBuff` and `iBuff`:
  for (let i = 0; i < rBuff.length; i++) {
    let k = kInd(i, pow);
    if (k >= i) continue;
    [ rBuff[i], rBuff[k] ] = [ rBuff[k], rBuff[i] ];
    [ iBuff[i], iBuff[k] ] = [ iBuff[k], iBuff[i] ];
  }

  // ... A bit of code to convert to power spectrum and return result

};

我的问题是: kInd到底在做什么?我已经运行它来输出一些示例值;似乎随着k1参数的增加,它以几乎随机的顺序输出2的幂。对kInd的微小更改会导致fft的结果完全错误。

谢谢!

(注意:让我知道是否有更多代码会有所帮助。为了读者的缘故,请尽量使内容简短!)

3 个答案:

答案 0 :(得分:4)

这实现了FFT算法的butterfly操作。

例如,正在运行...

console.log([0,1,2,3,4,5,6,7].map(i => kInd(i, 3)))

...打印...

[ 0, 4, 2, 6, 1, 5, 3, 7 ]

...这是图中的映射:

http://www.alwayslearn.com/DFT%20and%20FFT%20Tutorial/DFTandFFT_FFT_Butterfly_8_Input.html

答案 1 :(得分:1)

似乎有点重新排序。

let kInd = (k1, pow) =>{
  let k2 = 0;
  let k3 = 0;

  for (let i = 0; i < pow; i++) {
    k3 = k1 >> 1;
    k2 = 2 * (k2 - k3) + k1;
    k1 = k3;
  }
  return k2;
};

const format = (s, p = 5) => s.toString().padStart(p);

var i,
    l = 16,
    pow = Math.log2(l);
    
for (i = 0; i < l; i++) {
    document.getElementById('out').innerHTML += `${format(i)} ${format(kInd(i, pow))}<br>`;
}
<pre id="out"></pre>

答案 2 :(得分:1)

对于任何给定的pow,馈送递增的i会产生k s的周期。循环的长度为2^pow,下面的代码将针对每个i在递增的pow上显示两个循环。有趣的是i的模式导致k大于或等于i(在下面的输出中以粗体显示)。

对于每个战俘,k(i) >= ii=0开头的模式是什么:

战俘0:TFFF...

战俘1:TTFFF...

战俘2:TTFTFTFFF...

战俘3:TTTTFTFTFFF...

战俘4:TTTTFTTTFTFTFFFTFFF...

战俘5:TTTTTTTTFTTTFTTTFTFTFTFTFFFTFFFTFFF...

let kInd = (k1, pow) => {
  let k2 = 0;
  let k3 = 0;

  for (let i = 0; i < pow; i++) {
    k3 = k1 >> 1;
    k2 = 2 * (k2 - k3) + k1;
    k1 = k3;
  }
  return k2;
};

for (let p = 0; p < 7; p++) {
  for (let i = 0; i < Math.pow(2, p) * 2; i++) {
    let k = kInd(i, p)
    if (k >= i) {
      $('#output').append("<p><b>" + i + "&nbsp;&nbsp;&nbsp;" + p + "->" + k + "</b></p>");
    } else {
      $('#output').append("<p>" + i + "&nbsp;&nbsp;&nbsp;" + p + "->" + k + "</p>");
    }
  }
  $('#output').append("<p>----------------</p>");
}
<script src="https://ajax.googleapis.com/ajax/libs/jquery/2.1.1/jquery.min.js"></script>
<div id="output"></div>

我知道这不是答案,但我想提供帮助,并且注释不适合我要放在此处的格式。