sampling: refactor iteration

amodem/sampling.py

@@ -53,21 +53,19 @@ class Sampler(object):
         self.offset += offset
 
     def next(self):
-        res = self._sample() * self.gain
+        return self._sample() * self.gain
-        self.offset += self.freq
-        return res
 
     def _sample(self):
         coeffs, begin = self.interp.get(self.offset)
         end = begin + self.interp.coeff_len
-        while True:
+        while self.index < end:
-            if self.index == end:
-                return np.dot(coeffs, self.buff)
-
             self.buff[:-1] = self.buff[1:]
             self.buff[-1] = self.src.next()  # throws StopIteration
             self.index += 1
 
+        self.offset += self.freq
+        return np.dot(coeffs, self.buff)
+
 if __name__ == '__main__':
     import common
     import sys