mirror of
https://github.com/AsahiLinux/m1n1
synced 2024-11-22 22:53:04 +00:00
225 lines
6.5 KiB
Python
Executable file
225 lines
6.5 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
# SPDX-License-Identifier: MIT
|
|
import sys, pathlib
|
|
import time
|
|
sys.path.append(str(pathlib.Path(__file__).resolve().parents[1]))
|
|
|
|
# speaker_amp.py -- play audio through the embedded speaker on Mac mini
|
|
#
|
|
# sample usage with sox:
|
|
#
|
|
# sox INPUT_FILE -t raw -r 48000 -c 1 -e signed-int -b 32 -L - gain -63 | python3 ./speaker_amp.py
|
|
#
|
|
# (expects mono, 24-bit signed samples padded to 32 bits on the msb side)
|
|
|
|
import argparse
|
|
import os.path
|
|
import code
|
|
import sys
|
|
|
|
from m1n1.setup import *
|
|
from m1n1.hw.dart import DART, DARTRegs
|
|
from m1n1.hw.admac import *
|
|
from m1n1.hw.i2c import I2C
|
|
|
|
# this here is an embedded console so that one can poke while
|
|
# the descriptors keep being filled in
|
|
class PollingConsole(code.InteractiveConsole):
|
|
def __init__(self, locals=None, filename="<console>"):
|
|
global patch_stdout, PromptSession, FileHistory
|
|
global Thread, Queue, Empty
|
|
from prompt_toolkit import PromptSession
|
|
from prompt_toolkit.history import FileHistory
|
|
from prompt_toolkit.patch_stdout import patch_stdout
|
|
from threading import Thread
|
|
from queue import Queue, Empty
|
|
|
|
super().__init__(locals, filename)
|
|
|
|
self._qu_input = Queue()
|
|
self._qu_result = Queue()
|
|
self._should_exit = False
|
|
|
|
self.session = PromptSession(history=FileHistory(os.path.expanduser("~/.m1n1-history")))
|
|
self._other_thread = Thread(target=self._other_thread_main, daemon=False)
|
|
self._other_thread.start()
|
|
|
|
def __enter__(self):
|
|
self._patch = patch_stdout()
|
|
self._patch.__enter__()
|
|
return self
|
|
|
|
def __exit__(self, exc_type, exc_val, exc_tb):
|
|
self._patch.__exit__(exc_type, exc_val, exc_tb)
|
|
|
|
def _other_thread_main(self):
|
|
first = True
|
|
|
|
while True:
|
|
if first:
|
|
more_input = False
|
|
first = False
|
|
else:
|
|
more_input = self._qu_result.get()
|
|
|
|
try:
|
|
self._qu_input.put(self.session.prompt("(♫♫) " if not more_input else "... "))
|
|
except EOFError:
|
|
self._qu_input.put(None)
|
|
return
|
|
|
|
def poll(self):
|
|
if self._should_exit:
|
|
return False
|
|
|
|
try:
|
|
line = self._qu_input.get(timeout=0.01)
|
|
except Empty:
|
|
return True
|
|
if line is None:
|
|
self._should_exit = True
|
|
return False
|
|
self._qu_result.put(self.push(line))
|
|
return True
|
|
|
|
|
|
class NoConsole:
|
|
def poll(self):
|
|
time.sleep(0.01)
|
|
return True
|
|
|
|
def __enter__(self):
|
|
return self
|
|
|
|
def __exit__(self, exc_type, exc_val, exc_tb):
|
|
pass
|
|
|
|
|
|
argparser = argparse.ArgumentParser()
|
|
argparser.add_argument("--console", action='store_true')
|
|
argparser.add_argument("-f", "--file", "--input", "--samples",
|
|
type=str, default=None,
|
|
help='input filename to take samples from ' \
|
|
'(default: standard input)')
|
|
argparser.add_argument("-b", "--bufsize", type=int, default=1024*32,
|
|
help='size of buffers to keep submitting to DMA')
|
|
args = argparser.parse_args()
|
|
|
|
if args.console and args.file is None:
|
|
print("Specify file with samples (option -f) if using console")
|
|
sys.exit(1)
|
|
inputf = open(args.file, "rb") if args.file is not None else sys.stdin.buffer
|
|
|
|
|
|
p.pmgr_adt_clocks_enable("/arm-io/i2c1")
|
|
p.pmgr_adt_clocks_enable("/arm-io/admac-sio")
|
|
p.pmgr_adt_clocks_enable("/arm-io/dart-sio")
|
|
p.pmgr_adt_clocks_enable("/arm-io/mca-switch")
|
|
|
|
|
|
i2c1 = I2C(u, "/arm-io/i2c1")
|
|
|
|
dart_base, _ = u.adt["/arm-io/dart-sio"].get_reg(0) # stream index 2
|
|
dart = DART(iface, DARTRegs(u, dart_base), util=u)
|
|
dart.initialize()
|
|
|
|
admac = ADMAC(u, "/arm-io/admac-sio", dart, debug=True)
|
|
tx_chan = admac.chans[4]
|
|
|
|
tx_chan.disable()
|
|
tx_chan.reset()
|
|
tx_chan.poll() # read stale reports
|
|
tx_chan.buswidth = E_BUSWIDTH.W_32BIT
|
|
tx_chan.framesize = E_FRAME.F_1_WORD
|
|
|
|
mca_switch0_base = 0x2_3840_0000 # size: 0x1_8000
|
|
mca_switch1_base = 0x2_3830_0000 # size: 0x3_0000
|
|
|
|
for off in [0x0, 0x100, 0x4000, 0x4100]:
|
|
p.write32(mca_switch0_base + off, 0x0)
|
|
p.write32(mca_switch0_base + off, 0x2)
|
|
|
|
|
|
p.write32(mca_switch0_base + 0x4104, 0x0)
|
|
p.write32(mca_switch0_base + 0x4108, 0x0)
|
|
p.write32(mca_switch0_base + 0x410c, 0xfe)
|
|
|
|
p.write32(mca_switch1_base + 0x8000, 0x102048)
|
|
# bits 0x0000e0 influence clock
|
|
# 0x00000f influence sample serialization
|
|
|
|
# clock
|
|
p.write32(0x23b0400d8, 0x06000000) # 48 ksps, zero out for ~96 ksps
|
|
|
|
p.write32(mca_switch0_base + 0x0600, 0xe) # 0x8 or have zeroed samples, 0x6 or have no clock
|
|
p.write32(mca_switch0_base + 0x0604, 0x200) # sensitive in mask 0xf00, any other value disables clock
|
|
p.write32(mca_switch0_base + 0x0608, 0x4) # 0x4 or zeroed samples
|
|
|
|
# toggle the GPIO line driving the speaker-amp IC reset
|
|
p.write32(0x23c1002d4, 0x76a02) # invoke reset
|
|
p.write32(0x23c1002d4, 0x76a03) # take out of reset
|
|
|
|
|
|
tx_chan.submit(inputf.read(args.bufsize))
|
|
tx_chan.enable()
|
|
|
|
|
|
# accesses to 0x100-sized blocks in the +0x4000 region require
|
|
# the associated enable bit cleared, or they cause SErrors
|
|
def mca_switch_unk_disable():
|
|
for off in [0x4000, 0x4100, 0x4300]:
|
|
p.write32(mca_switch0_base + off, 0x0)
|
|
|
|
def mca_switch_unk_enable():
|
|
for off in [0x4000, 0x4100, 0x4300]:
|
|
p.write32(mca_switch0_base + off, 0x1)
|
|
|
|
p.write32(mca_switch0_base + 0x4104, 0x2)
|
|
mca_switch_unk_enable()
|
|
|
|
|
|
# by ADT and leaked schematic, i2c1 contains TAS5770L,
|
|
# which is not a public part. but there's e.g. TAS2770
|
|
# with similar registers
|
|
#
|
|
# https://www.ti.com/product/TAS2770
|
|
#
|
|
# if the speaker-amp IC loses clock on the serial sample input,
|
|
# it automatically switches to software shutdown.
|
|
#
|
|
|
|
i2c1.write_reg(0x31, 0x08, [0x40])
|
|
i2c1.write_reg(0x31, 0x0a, [0x06, 0x00, 0x1a])
|
|
i2c1.write_reg(0x31, 0x1b, [0x01, 0x82, 0x06])
|
|
i2c1.write_reg(0x31, 0x16, [0x50, 0x04])
|
|
i2c1.write_reg(0x31, 0x0d, [0x00])
|
|
#i2c1.write_reg(0x31, 0x03, [0x14])
|
|
|
|
# amplifier gain, presumably this is the lowest setting
|
|
i2c1.write_reg(0x31, 0x03, [0x0])
|
|
|
|
# take the IC out of software shutdown
|
|
i2c1.write_reg(0x31, 0x02, [0x0c])
|
|
|
|
|
|
with (PollingConsole(locals()) if args.console else NoConsole()) as cons:
|
|
try:
|
|
while cons.poll():
|
|
while (not tx_chan.can_submit()) and cons.poll():
|
|
tx_chan.poll()
|
|
|
|
if not cons.poll():
|
|
break
|
|
|
|
tx_chan.submit(inputf.read(args.bufsize))
|
|
except KeyboardInterrupt:
|
|
pass
|
|
|
|
|
|
# mute
|
|
i2c1.write_reg(0x31, 0x02, [0x0d])
|
|
|
|
# software shutdown
|
|
i2c1.write_reg(0x31, 0x02, [0x0e])
|
|
|
|
tx_chan.disable()
|