def __init__(self, surround_config):
surround = Surround(
[WranglingData(),
ModellingAndPrediction(),
DisplayOutput()], __name__)
surround.set_config(surround_config)
super().__init__(surround)
def test_surround_config(self):
path = os.path.dirname(__file__)
config = Config()
config.read_config_files([os.path.join(path, "config.yaml")])
surround = Surround([HelloStage()])
surround.set_config(config)
data = BasicData()
surround.process(data)
self.assertEqual(data.config_value, "Scott")
def test_surround_override(self):
path = os.path.dirname(__file__)
surround = Surround([FirstStage()])
config = Config()
config.read_config_files([os.path.join(path, "stages.yaml")])
surround.set_config(config)
data = BasicData()
surround.process(data)
self.assertEqual(data.stage1, "first stage")
self.assertEqual(data.stage2, "second stage")
def main():
sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
sock.bind(('0.0.0.0', 1024))
surround = Surround([ValidateData(), VadDetection()])
config = Config()
surround.set_config(config)
surround.init_stages()
audio_input = []
last_packet_time = time.time()
packet_id = -1
while True:
source_addr = None
while len(audio_input) < 48000:
# Retrieve data from client (9600 samples in bytes = 9600 * 2 bytes (2 bytes per sample))
data_bytes, source_addr = sock.recvfrom(2400 * 2 + 4)
# If last packet received was 5 seconds ago, clear cache
if last_packet_time + 5 < time.time():
audio_input.clear()
last_packet_time = time.time()
# Get the packet id from the end of the byte array
packet_id = int.from_bytes(data_bytes[2400 * 2:], sys.byteorder, signed=True)
# Convert the byte array into an array of float samples (-1 to 1)
for i in range(0, len(data_bytes) - 4, 2):
sample = int.from_bytes(data_bytes[i : i + 2], sys.byteorder, signed=True)
sample /= 32767.0
audio_input.append(sample)
# Process the audio data for voice activity
data = VadData(audio_input)
surround.process(data)
# Rolling window of 2400 samples (50ms)
audio_input = audio_input[2400:]
if data.error is None and data.output_data is not None:
print("Noise: " + str(data.output_data[0] * 100.0) + " Voice: " + str(data.output_data[1] * 100.0))
# Sending the results back to who made the request
results = { "id": packet_id, "noise": float(data.output_data[0]), "voice": float(data.output_data[1]) }
sock.sendto(json.dumps(results).encode(), (source_addr[0], 25565))
else:
print(data.error)
break
def main():
logging.basicConfig(level=logging.INFO)
surround = Surround(
[WranglingData(),
ModellingAndPrediction(),
DisplayOutput()])
surround_config = Config()
surround_config.read_config_files(["config.yaml"])
surround.set_config(surround_config)
surround.init_stages()
# Fetch the data from the data folder.
raw_data = fetch_data(surround_config)
# epl_data is created as the SurroundData type.
epl_data = EplData()
# raw_data is fed into epl_data.
epl_data.feed_data(raw_data)
# Surround process started.
surround.process(epl_data)
import logging
from surround import Surround, Config
from svr.SVR_stages import FeedData, SVRData, ComputeForecast, PlotResult
logging.basicConfig(level=logging.INFO)
# def main():
# wrapper = PipelineWrapper()
# config = wrapper.get_config()
# output = wrapper.run(json.dumps({"data": "hello"}))
# with open(os.path.join(config["output_path"], "output.txt"), 'w') as f:
# f.write(output["output"])
# logging.info(output)
if __name__ == "__main__":
logging.basicConfig(level=logging.INFO)
surround = Surround([FeedData(), ComputeForecast(), PlotResult()])
surround_config = Config()
surround_config.read_config_files(["config.yaml"])
surround.set_config(surround_config)
surround.init_stages()
svr_data = SVRData()
svr_data.get_data()
surround.process(svr_data)