sampling: remove extra function call

romanz · Aug 8, 2014 · f79c4c1 · f79c4c1
1 parent 449d74c
commit f79c4c1
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/amodem/sampling.py b/amodem/sampling.py
@@ -47,11 +47,6 @@ def __iter__(self):
         return self
 
     def next(self):
-        return self._sample() * self.gain
-
-    __next__ = next
-
-    def _sample(self):
         offset = self.offset
         # offset = k + (j / self.resolution)
         k = int(offset)  # integer part
@@ -64,7 +59,10 @@ def _sample(self):
             self.index += 1
 
         self.offset += self.freq
-        return np.dot(coeffs, self.buff)
+        return np.dot(coeffs, self.buff) * self.gain
+
+    __next__ = next
+
 
 if __name__ == '__main__':
     import common