From 12f6bf916fbe4357305522cc361b522e2482e9f5 Mon Sep 17 00:00:00 2001
From: KIU Shueng Chuan <nixchuan@gmail.com>
Date: Tue, 17 Jan 2017 22:04:05 +0800
Subject: [PATCH] fix fft premature slicing away of 0 freq bin

also fixes:
- use rfft for better efficiency
- use rfftfreq to compute coords correctly
  - works for both odd/even lengths
- python3: integer division needed for numpy indexing
---
 pyqtgraph/graphicsItems/PlotDataItem.py | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/pyqtgraph/graphicsItems/PlotDataItem.py b/pyqtgraph/graphicsItems/PlotDataItem.py
index 37245bec..11184ae6 100644
--- a/pyqtgraph/graphicsItems/PlotDataItem.py
+++ b/pyqtgraph/graphicsItems/PlotDataItem.py
@@ -679,10 +679,9 @@ class PlotDataItem(GraphicsObject):
             x2 = np.linspace(x[0], x[-1], len(x))
             y = np.interp(x2, x, y)
             x = x2
-        f = np.fft.fft(y) / len(y)
-        y = abs(f[1:len(f)/2])
-        dt = x[-1] - x[0]
-        x = np.linspace(0, 0.5*len(x)/dt, len(y))
+        f = np.fft.rfft(y) / y.size
+        x = np.fft.rfftfreq(y.size)
+        y = np.abs(f)
         return x, y
     
 def dataType(obj):