diff --git a/sstv.py b/sstv.py index a41f1e6..6388e7d 100644 --- a/sstv.py +++ b/sstv.py @@ -3,7 +3,8 @@ from __future__ import division, with_statement from math import sin, pi, floor from random import random -import struct +from contextlib import closing +import struct, wave FREQ_VIS_BIT1 = 1100 FREQ_SYNC = 1200 @@ -25,23 +26,13 @@ class SSTV(object): BITS_TO_STRUCT = {8: 'b', 16: 'h'} def write_wav(self, filename): - bytes_per_sec = self.bits // 8 fmt = '<' + self.BITS_TO_STRUCT[self.bits] data = ''.join(struct.pack(fmt, b) for b in self.gen_samples()) - payload = ''.join(( - 'WAVE', - 'fmt ', - struct.pack('