diff pytouhou/formats/fmt.py @ 325:cddfd3cb4797

Add music support for >PCB.
author Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
date Thu, 21 Jun 2012 15:01:01 +0200
parents
children d8aab27a2ab2
line wrap: on
line diff
new file mode 100644
--- /dev/null
+++ b/pytouhou/formats/fmt.py
@@ -0,0 +1,69 @@
+# -*- encoding: utf-8 -*-
+##
+## Copyright (C) 2012 Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
+##
+## This program is free software; you can redistribute it and/or modify
+## it under the terms of the GNU General Public License as published
+## by the Free Software Foundation; version 3 only.
+##
+## This program is distributed in the hope that it will be useful,
+## but WITHOUT ANY WARRANTY; without even the implied warranty of
+## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+## GNU General Public License for more details.
+##
+
+
+from struct import unpack
+
+
+class Track(object):
+    def __init__(self):
+        self.name = ''
+
+        # loop info
+        self.intro = 0
+        #self.unknown
+        self.start = 0
+        self.duration = 0
+
+        # WAVE header
+        self.wFormatTag = 1
+        self.wChannels = 2
+        self.dwSamplesPerSec = 44100
+        self.dwAvgBytesPerSec = 176400
+        self.wBlockAlign = 4
+        self.wBitsPerSample = 16
+
+
+class FMT(list):
+    @classmethod
+    def read(cls, file):
+        self = cls()
+
+        file.seek(0)
+        while True:
+            track = Track()
+            track.name = unpack('<16s', file.read(16))[0]
+            if not ord(track.name[0]):
+                break
+
+            # loop info
+            track.intro, unknown, track.start, track.duration = unpack('<IIII', file.read(16))
+
+            # WAVE header
+            (track.wFormatTag,
+             track.wChannels,
+             track.dwSamplesPerSec,
+             track.dwAvgBytesPerSec,
+             track.wBlockAlign,
+             track.wBitsPerSample) = unpack('<HHLLHH', file.read(16))
+
+            assert track.wFormatTag == 1 # We don’t support non-PCM formats
+            assert track.dwAvgBytesPerSec == track.dwSamplesPerSec * track.wBlockAlign
+            assert track.wBlockAlign == track.wChannels * track.wBitsPerSample // 8
+            assert b'\00\00\00\00' == file.read(4)
+
+            self.append(track)
+
+        return self
+