comparison fft/fft.js/src/real.rb @ 25:66f9fd5ac611

Bring in some more of the third-party code

+def load(offset = 0, i = nil, stride = nil)
+  "#{real(x, offset, i, stride)}"
+end
+
+def creal(x, offset = 0, i = nil, stride = nil)
+  if stride
+    "#{x}[2 * ((#{offset}) + (#{stride}) * (#{i}))]"
+  elsif i
+    "#{x}[2 * ((#{offset}) + (#{i}))]"
+  elsif offset
+    "#{x}[2 * (#{offset})]"
+  else
+    "#{x}[0]"
+  end
+end
+
+def cimag(x, offset = 0, i = nil, stride = nil)
+  if stride
+    "#{x}[2 * ((#{offset}) + (#{stride}) * (#{i})) + 1]"
+  elsif i
+    "#{x}[2 * ((#{offset}) + (#{i})) + 1]"
+  elsif offset
+    "#{x}[2 * (#{offset}) + 1]"
+  else
+    "#{x}[1]"
+  end
+end
+
+def cload(value, x, offset = 0, i = nil, stride = nil)
+  "var #{value}_r = #{real(x, offset, i, stride)}, #{value}_i = #{imag(x, offset, i, stride)}"
+end
+
+def cstore(value, x, offset = 0, i = nil, stride = nil)
+  "#{real(x, offset, i, stride)} = #{value}_r, #{imag(x, offset, i, stride)} = #{value}_i"
+end
+
+def cadd(result, a, b)
+  "var #{result}_r = #{a}_r + #{b}_r, #{result}_i = #{a}_i + #{b}_i"
+end
+
+def csub(result, a, b)
+  "var #{result}_r = #{a}_r - #{b}_r, #{result}_i = #{a}_i - #{b}_i"
+end
+
+def cmul(result, a, b)
+  "var #{result}_r = #{a}_r * #{b}_r - #{a}_i * #{b}_i, #{result}_i = #{a}_r * #{b}_i + #{a}_i * #{b}_r"
+end