mirror of
https://github.com/onyx-and-iris/vmr-http.git
synced 2026-04-06 09:59:11 +00:00
onyx-and-iris
0d6d91c20a
it uses a factory function to create an eq router for each kind. add models.eq.EQParams. Use it to define on and ab properties. minor bump
97 lines
4.1 KiB
Python
97 lines
4.1 KiB
Python
"""Module for equalizer related endpoints, supporting both strip and bus parents via a factory function."""
|
|
|
|
from fastapi import APIRouter, Depends
|
|
|
|
from vmr_http.dependencies import get_voicemeeter_client
|
|
from vmr_http.models.eq import EQChannelCellParams, EQParams
|
|
|
|
|
|
def create_router(parent_router_kind: str) -> APIRouter:
|
|
"""Create an APIRouter for equalizer endpoints, with the specified parent kind ('strip' or 'bus')."""
|
|
if parent_router_kind not in ('strip', 'bus'):
|
|
raise ValueError(f'Invalid router kind: {parent_router_kind}')
|
|
parent_attr = parent_router_kind
|
|
|
|
def get_parent(voicemeeter, index):
|
|
return getattr(voicemeeter, parent_attr)[index]
|
|
|
|
cell_router = APIRouter()
|
|
|
|
@cell_router.patch('')
|
|
@cell_router.put('')
|
|
async def update_eq_channel_cell_params(
|
|
index: int,
|
|
channel_index: int,
|
|
cell_index: int,
|
|
params: EQChannelCellParams,
|
|
voicemeeter=Depends(get_voicemeeter_client),
|
|
):
|
|
"""Update one or more parameters for the specified eq channel cell."""
|
|
cell = get_parent(voicemeeter, index).eq.channel[channel_index].cell[cell_index]
|
|
updated = {}
|
|
for key, value in params.model_dump(exclude_unset=True).items():
|
|
setattr(cell, key, value)
|
|
updated[key] = getattr(cell, key)
|
|
return updated
|
|
|
|
@cell_router.get('/on')
|
|
async def get_eq_channel_cell_on(
|
|
index: int, channel_index: int, cell_index: int, voicemeeter=Depends(get_voicemeeter_client)
|
|
):
|
|
"""Get the current on status for the specified eq channel cell."""
|
|
return {'on': get_parent(voicemeeter, index).eq.channel[channel_index].cell[cell_index].on}
|
|
|
|
@cell_router.get('/type')
|
|
async def get_eq_channel_cell_type(
|
|
index: int, channel_index: int, cell_index: int, voicemeeter=Depends(get_voicemeeter_client)
|
|
):
|
|
"""Get the current type for the specified eq channel cell."""
|
|
return {'type': get_parent(voicemeeter, index).eq.channel[channel_index].cell[cell_index].type}
|
|
|
|
@cell_router.get('/f')
|
|
async def get_eq_channel_cell_f(
|
|
index: int, channel_index: int, cell_index: int, voicemeeter=Depends(get_voicemeeter_client)
|
|
):
|
|
"""Get the current f value for the specified eq channel cell."""
|
|
return {'f': get_parent(voicemeeter, index).eq.channel[channel_index].cell[cell_index].f}
|
|
|
|
@cell_router.get('/gain')
|
|
async def get_eq_channel_cell_gain(
|
|
index: int, channel_index: int, cell_index: int, voicemeeter=Depends(get_voicemeeter_client)
|
|
):
|
|
"""Get the current gain value for the specified eq channel cell."""
|
|
return {'gain': get_parent(voicemeeter, index).eq.channel[channel_index].cell[cell_index].gain}
|
|
|
|
@cell_router.get('/q')
|
|
async def get_eq_channel_cell_q(
|
|
index: int, channel_index: int, cell_index: int, voicemeeter=Depends(get_voicemeeter_client)
|
|
):
|
|
"""Get the current q value for the specified eq channel cell."""
|
|
return {'q': get_parent(voicemeeter, index).eq.channel[channel_index].cell[cell_index].q}
|
|
|
|
router = APIRouter()
|
|
router.include_router(cell_router, prefix='/channel/{channel_index}/cell/{cell_index}')
|
|
|
|
@router.patch('')
|
|
@router.put('')
|
|
async def update_eq_params(index: int, params: EQParams, voicemeeter=Depends(get_voicemeeter_client)):
|
|
"""Update one or more equalizer parameters for the specified index."""
|
|
eq = get_parent(voicemeeter, index).eq
|
|
updated = {}
|
|
for key, value in params.model_dump(exclude_unset=True).items():
|
|
setattr(eq, key, value)
|
|
updated[key] = getattr(eq, key)
|
|
return updated
|
|
|
|
@router.get('/on')
|
|
async def get_eq_on(index: int, voicemeeter=Depends(get_voicemeeter_client)):
|
|
"""Get the current equalizer on status for the specified index."""
|
|
return {'on': get_parent(voicemeeter, index).eq.on}
|
|
|
|
@router.get('/ab')
|
|
async def get_eq_ab(index: int, voicemeeter=Depends(get_voicemeeter_client)):
|
|
"""Get the current equalizer A/B status for the specified index."""
|
|
return {'ab': get_parent(voicemeeter, index).eq.ab}
|
|
|
|
return router
|