From 3a81a8ef83b55329b82ac2f6a54b868cad50ad2e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Andr=C3=A1s=20Veres-Szentkir=C3=A1lyi?= <vsza@vsza.hu>
Date: Mon, 17 Jun 2013 13:44:24 +0200
Subject: [PATCH] replaced WAV writer with stdlib wave module

---
 sstv.py | 23 +++++++----------------
 1 file changed, 7 insertions(+), 16 deletions(-)

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('<IHHIIHH', 16, 1, 1, self.samples_per_sec,
-					self.samples_per_sec * bytes_per_sec, bytes_per_sec,
-					self.bits),
-				'data',
-				struct.pack('<I', len(data))
-				))
-		header = 'RIFF' + struct.pack('<I', len(payload) + len(data))
-		with file(filename, 'wb') as wav:
-			wav.write(header)
-			wav.write(payload)
-			wav.write(data)
+		with closing(wave.open(filename, 'wb')) as wav:
+			wav.setnchannels(1)
+			wav.setsampwidth(self.bits // 8)
+			wav.setframerate(self.samples_per_sec)
+			wav.writeframes(data)
 
 	def gen_samples(self):
 		"""generates bits from gen_values"""