| #!/usr/bin/env python3 | |
| """ | |
| Localize from four WAV files. Fill in the paths and receiver | |
| positions, then run. | |
| Run: python examples/from_wav.py | |
| """ | |
| import numpy as np | |
| from ir_source_localizer import localize, load_wav | |
| # Edit these: | |
| WAV_PATHS = ["r0.wav", "r1.wav", "r2.wav", "r3.wav"] | |
| RECEIVERS = [ | |
| (4.5, 3.5, 1.0), | |
| (0.5, 3.5, 1.0), | |
| (4.5, 0.5, 1.0), | |
| (0.5, 0.5, 1.0), | |
| ] | |
| SOURCE_HEIGHT = 1.0 | |
| SIGMA_GUESS = 4e-4 # adjust to your pulse | |
| X_RANGE = (0.2, 4.8) | |
| Y_RANGE = (0.2, 3.8) | |
| # Load | |
| irs = [] | |
| sample_rate = None | |
| for path in WAV_PATHS: | |
| data, sr = load_wav(path) | |
| if sample_rate is None: | |
| sample_rate = sr | |
| if sr != sample_rate: | |
| raise SystemExit(f"sample rate mismatch in {path}") | |
| irs.append(data) | |
| # Localize | |
| result = localize( | |
| irs=irs, sample_rate=sample_rate, receivers=RECEIVERS, | |
| sz=SOURCE_HEIGHT, | |
| x_range=X_RANGE, y_range=Y_RANGE, | |
| sigma_guess=SIGMA_GUESS, | |
| ) | |
| print(result) |