dnet/pySSTV
1
0
Fork 0
mirror of https://github.com/dnet/pySSTV.git synced 2026-01-07 17:20:17 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

replaced struct with array

Browse source
This commit is contained in:
András Veres-Szentkirályi 2013-06-26 11:22:30 +02:00
parent 808a47fa6d
commit 7560c8f37f
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
pysstv/sstv.py
View file

@ -5,7 +5,7 @@ from math import sin, pi, floor
from random import random from random import random
from contextlib import closing from contextlib import closing
from itertools import imap from itertools import imap
import struct from array import array
import wave import wave
FREQ_VIS_BIT1 = 1100 FREQ_VIS_BIT1 = 1100
@ -38,13 +38,13 @@ class SSTV(object):
def write_wav(self, filename): def write_wav(self, filename):
"""writes the whole image to a Microsoft WAV file""" """writes the whole image to a Microsoft WAV file"""
fmt = '<' + self.BITS_TO_STRUCT[self.bits] fmt = self.BITS_TO_STRUCT[self.bits]
data = ''.join(struct.pack(fmt, b) for b in self.gen_samples()) data = array(fmt, self.gen_samples())
with closing(wave.open(filename, 'wb')) as wav: with closing(wave.open(filename, 'wb')) as wav:
wav.setnchannels(self.nchannels) wav.setnchannels(self.nchannels)
wav.setsampwidth(self.bits // 8) wav.setsampwidth(self.bits // 8)
wav.setframerate(self.samples_per_sec) wav.setframerate(self.samples_per_sec)
wav.writeframes(data) wav.writeframes(data.tostring())
def gen_samples(self): def gen_samples(self):
"""generates discrete samples from gen_values() """generates discrete samples from gen_values()