90 lines
2.7 KiB
Python
90 lines
2.7 KiB
Python
|
from __future__ import absolute_import
|
||
|
|
from future.builtins import object
|
||
|
|
from .external.pybass import *
|
||
|
|
from functools import update_wrapper
|
||
|
|
|
||
|
|
EAX_ENVIRONMENTS = {
|
||
|
|
'generic': EAX_ENVIRONMENT_GENERIC,
|
||
|
|
'padded_cell': EAX_ENVIRONMENT_PADDEDCELL,
|
||
|
|
'room': EAX_ENVIRONMENT_ROOM,
|
||
|
|
'bathroom': EAX_ENVIRONMENT_BATHROOM,
|
||
|
|
'living_room': EAX_ENVIRONMENT_LIVINGROOM,
|
||
|
|
'stone_room': EAX_ENVIRONMENT_STONEROOM,
|
||
|
|
'auditorium': EAX_ENVIRONMENT_AUDITORIUM,
|
||
|
|
'concert_hall': EAX_ENVIRONMENT_CONCERTHALL,
|
||
|
|
'cave': EAX_ENVIRONMENT_CAVE,
|
||
|
|
'arena': EAX_ENVIRONMENT_ARENA,
|
||
|
|
'hanger': EAX_ENVIRONMENT_HANGAR,
|
||
|
|
'carpeted_hallway': EAX_ENVIRONMENT_CARPETEDHALLWAY,
|
||
|
|
'hallway': EAX_ENVIRONMENT_HALLWAY,
|
||
|
|
'stone_corridor': EAX_ENVIRONMENT_STONECORRIDOR,
|
||
|
|
'alley': EAX_ENVIRONMENT_ALLEY,
|
||
|
|
'forest': EAX_ENVIRONMENT_FOREST,
|
||
|
|
'city': EAX_ENVIRONMENT_CITY,
|
||
|
|
'mountains': EAX_ENVIRONMENT_MOUNTAINS,
|
||
|
|
'quarry': EAX_ENVIRONMENT_QUARRY,
|
||
|
|
'plain': EAX_ENVIRONMENT_PLAIN,
|
||
|
|
'parking_lot': EAX_ENVIRONMENT_PARKINGLOT,
|
||
|
|
'sewer_pipe': EAX_ENVIRONMENT_SEWERPIPE,
|
||
|
|
'underwater': EAX_ENVIRONMENT_UNDERWATER,
|
||
|
|
'drugged': EAX_ENVIRONMENT_DRUGGED,
|
||
|
|
'dizzy': EAX_ENVIRONMENT_DIZZY,
|
||
|
|
'psychotic': EAX_ENVIRONMENT_PSYCHOTIC
|
||
|
|
}
|
||
|
|
|
||
|
|
class BassError (Exception):
|
||
|
|
"""Error that is raised when there is a problem with a Bass call."""
|
||
|
|
|
||
|
|
def __init__(self, code, description):
|
||
|
|
self.code = code
|
||
|
|
self.description = description
|
||
|
|
|
||
|
|
def __str__(self):
|
||
|
|
return '%d, %s' % (self.code, self.description)
|
||
|
|
|
||
|
|
def bass_call(function, *args):
|
||
|
|
"""Makes a call to bass and raises an exception if it fails."""
|
||
|
|
res = function(*args)
|
||
|
|
if res == 0 or res == -1:
|
||
|
|
code = BASS_ErrorGetCode()
|
||
|
|
raise BassError(code, get_error_description(code))
|
||
|
|
return res
|
||
|
|
|
||
|
|
def bass_call_0(function, *args):
|
||
|
|
"""Makes a call to bass and raises an exception if it fails. Does not consider 0 an error."""
|
||
|
|
res = function(*args)
|
||
|
|
if res == -1:
|
||
|
|
code = BASS_ErrorGetCode()
|
||
|
|
raise BassError(code, get_error_description(code))
|
||
|
|
return res
|
||
|
|
|
||
|
|
def update_3d_system(func):
|
||
|
|
"""Decorator to automatically update the 3d system after a function call."""
|
||
|
|
def update_3d_system_wrapper(*args, **kwargs):
|
||
|
|
val = func(*args, **kwargs)
|
||
|
|
bass_call( BASS_Apply3D)
|
||
|
|
return val
|
||
|
|
update_wrapper(update_3d_system_wrapper, func)
|
||
|
|
return update_3d_system_wrapper
|
||
|
|
|
||
|
|
class FlagObject(object):
|
||
|
|
"""An object which translates bass flags into human-readable/usable items"""
|
||
|
|
|
||
|
|
def flags_for(self, **flags):
|
||
|
|
res = 0
|
||
|
|
for k, v in flags.items():
|
||
|
|
if v:
|
||
|
|
res |= self.flag_mapping[k]
|
||
|
|
return res
|
||
|
|
|
||
|
|
def setup_flag_mapping(self):
|
||
|
|
self.flag_mapping = {
|
||
|
|
'loop': BASS_SAMPLE_LOOP,
|
||
|
|
'autofree': BASS_STREAM_AUTOFREE,
|
||
|
|
'mono': BASS_SAMPLE_MONO,
|
||
|
|
'software': BASS_SAMPLE_SOFTWARE,
|
||
|
|
'three_d': BASS_SAMPLE_3D,
|
||
|
|
'fx': BASS_SAMPLE_FX,
|
||
|
|
'decode': BASS_STREAM_DECODE,
|
||
|
|
}
|