-
Notifications
You must be signed in to change notification settings - Fork 0
/
CAudioFileIO.py
55 lines (41 loc) · 1.19 KB
/
CAudioFileIO.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
"""Audio file IO class
"""
import numpy as np
from scipy.io.wavfile import read
import aifc
from os.path import splitext
class CAudioFileIO(object):
def __init__(self):
self.data = np.zeros(())
self.sampleRate = None
self.fileExtension = None
def setData(self, data):
self.data = data
def getData(self):
return self.data
def setSampleRate(self, sampleRate):
self.sampleRate = float(sampleRate)
def getSampleRate(self):
return self.sampleRate
def getFileExtension(self):
return self.fileExtension
def setAudioObject(self, newObject):
self = newObject
def getAudioObject(self):
return object
def readAudio(self, filePath):
fileName, fileExtension = splitext(filePath)
self.fileExtension = fileExtension
if self.fileExtension == '.wav':
self.sampleRate, self.data = read(filePath)
elif self.fileExtension == '.aif':
file = aifc.open(filePath)
self.sampleRate = float(file.getframerate())
numFrames = file.getnframes()
self.data = file.readframes(numFrames)
file.close()
elif self.fileExtension == '.pcm':
rawData = np.memmap(filePath, dtype='h', mode='r')
self.data = np.asarray(rawData)
else:
raise AttributeError("Invalid file type")