source: src/mlx/soundsched.py@ 1092:f3c535a548c2

from .sound import startSound
from . import const
from . import fs

import threading

#------------------------------------------------------------------------------

## @package mlx.soundsched
#
# Sound playback scheduling.
#
# The logger may play certain sound files when certain conditions hold. For
# example, if the landing or the strobe lights are switched on during taxi, and
# we go to the takeoff state, a voice says "Cabin crew, please take your seats
# for takeoff". Some sounds are played only when the pilot presses a hotkey, if
# the program is set up that way.
#
# Each sound is represented by an instance of a subclass of \ref Sound. If a
# new \ref mlx.fs.AircraftState "aircraft state" comes in from the simulator,
# the \ref Sound.shouldPlay "shouldPlay" functions of the registered sound
# objects are called with that state, and if the function returns \c True, the
# corresponding sound file is played.
#
# The \ref ChecklistScheduler class handles the playback of the checklists,
# which are basically sequences of sound files played back one-by-one when a
# certain hotkey is pressed.

#------------------------------------------------------------------------------

class Sound(object):
    """A sound (file) that should be played in certain circumstances."""

    # Common lock for sounds
    _lock = threading.Lock()

    def __init__(self, name):
        """Construct the sound object for the sound file with the given
        name."""
        self._name = name
        
        self._playing = 0

    @property
    def playing(self):
        """Determine if sound is playing or not."""
        return self._playing>0

    def check(self, flight, state, pilotHotkeyPressed):
        """Check if the file should be played, and if it should be, play it."""
        if self.shouldPlay(flight, state, pilotHotkeyPressed):
            with Sound._lock:
                self._playing += 1
            startSound(self._name, finishCallback = self._playbackDone)

    def shouldPlay(self, flight, state, pilotHotkeyPressed):
        """Determine if the sound should be played.

        This default implementation returns False."""
        return False

    def _playbackDone(self, success, extra):
        """Called when the playback of thee sound has finished (or failed)."""
        if success is None:
            print("Failed to start sound", self._name)
        elif not success:
            print("Failed to finish sound", self._name)
        with Sound._lock:            
            self._playing -= 1

#------------------------------------------------------------------------------

class ScreamSound(Sound):
    """A screaming sound that is played under certain circumstance."""
    def __init__(self):
        """Construct the screaming sound."""
        super(ScreamSound, self).__init__(const.SOUND_SCREAM)

    def shouldPlay(self, flight, state, pilotHotkeyPressed):
        """Determine if the sound should be played.

        It should be played if it is not being played and the absolute value of
        the vertical speed is greater than 6000 fpm."""
        return not self.playing and abs(state.vs)>6000        

#------------------------------------------------------------------------------

class SimpleSound(Sound):
    """A simple sound that should be played only once, in a certain flight
    stage when the hotkey has been pressed or it is not the pilot who controls
    the sounds.

    If it is not the pilot who controls the sounds, it may be delayed relative
    to the first detection of a certain flight stage."""
    def __init__(self, name, stage, delay = 0.0, extraCondition = None,
                 considerHotkey = True, previousSound = None):
        """Construct the simple sound."""
        super(SimpleSound, self).__init__(name)
        
        self._stage = stage
        self._delay = delay
        self._extraCondition = extraCondition
        self._considerHotkey = considerHotkey
        self._previousSound = previousSound

        self._played = False
        self._stageDetectionTime = None

    def shouldPlay(self, flight, state, pilotHotkeyPressed):
        """Determine if the sound should be played."""
        if flight.stage!=self._stage or self._played or \
           (self._previousSound is not None and
            self._previousSound.playing):
            return False

        toPlay = False
        if flight.config.pilotControlsSounds and self._considerHotkey:
            toPlay = pilotHotkeyPressed
        else:
            if self._stageDetectionTime is None:
                self._stageDetectionTime = state.timestamp
            toPlay = state.timestamp>=(self._stageDetectionTime + self._delay)
            if toPlay and self._extraCondition is not None:
                toPlay = self._extraCondition(flight, state)

        if toPlay:
            self._played = True

        return toPlay

#------------------------------------------------------------------------------

class TaxiSound(SimpleSound):
    """The taxi sound.

    It first plays the Malev theme song, then an aircraft-specific taxi
    sound. The playback is started only, if the boarding sound is not being
    played."""

    _sounds = { const.AIRCRAFT_B736  : const.SOUND_TAXI_BOEING737NG,
                const.AIRCRAFT_B737  : const.SOUND_TAXI_BOEING737NG,
                const.AIRCRAFT_B738  : const.SOUND_TAXI_BOEING737NG,
                const.AIRCRAFT_B738C : const.SOUND_TAXI_BOEING737NG,
                const.AIRCRAFT_B762  : const.SOUND_TAXI_BOEING767,
                const.AIRCRAFT_B763  : const.SOUND_TAXI_BOEING767,
                const.AIRCRAFT_F70   : const.SOUND_TAXI_F70 }

    def __init__(self, flight, boardingSound = None):
        """Construct the taxi sound."""
        super(TaxiSound, self).__init__(const.SOUND_MALEV,
                                        const.STAGE_PUSHANDTAXI,
                                        previousSound = boardingSound,
                                        extraCondition = lambda _flight, state:
                                        state.groundSpeed>5)

        self._flight = flight

    def _playbackDone(self, success, extra):
        """Called when the playback is done.

        It starts playing the aircraft type-specific taxi sound, if any."""
        super(TaxiSound, self)._playbackDone(success, extra)
        aircraftType = self._flight.aircraftType
        sounds = TaxiSound._sounds
        if aircraftType in sounds:
            startSound(sounds[aircraftType])

#------------------------------------------------------------------------------

class TouchdownApplause(Sound):
    """An applause sound that is played after a gentle touchdown."""
    def __init__(self):
        super(TouchdownApplause, self).__init__(const.SOUND_APPLAUSE)

        self._touchdownTime = None
        self._played = False
        
    def shouldPlay(self, flight, state, pilotHotkeyPressed):
        """Determine if the sound should be played.

        It should be played if we are 2 seconds after a gentle touchdown."""
        if self._played or flight.tdRate is None:
            return False

        if self._touchdownTime is None:
            self._touchdownTime = state.timestamp

        if state.timestamp>=(self._touchdownTime + 2) and \
           flight.tdRate>=-150 and flight.tdRate<0:
            self._played = True
            return True
        else:
            return False
        
#------------------------------------------------------------------------------

class SoundScheduler(object):
    """A scheduler for the sounds."""
    def __init__(self, flight):
        """Construct the sound scheduler for the given flight."""
        self._flight = flight
        self._sounds = []

        self._sounds.append(ScreamSound())

        boardingSound = SimpleSound(const.SOUND_BOARDING,
                                    const.STAGE_BOARDING,
                                    delay = 10.0)
        self._sounds.append(boardingSound)
        self._sounds.append(TaxiSound(flight, boardingSound))
        self._sounds.append(SimpleSound(const.SOUND_CAPTAIN_TAKEOFF,
                                        const.STAGE_TAKEOFF,
                                        extraCondition = lambda flight, state:
                                        state.onTheGround,
                                        considerHotkey = False))
        self._sounds.append(SimpleSound(const.SOUND_CRUISE, const.STAGE_CRUISE))
        self._sounds.append(SimpleSound(const.SOUND_DESCENT, const.STAGE_DESCENT,
                                        extraCondition = lambda flight, state:
                                        state.altitude<15000))
        self._sounds.append(TouchdownApplause())
        self._sounds.append(SimpleSound(const.SOUND_TAXIAFTERLAND,
                                        const.STAGE_TAXIAFTERLAND,
                                        delay = 10.0))

    def schedule(self, state, pilotHotkeyPressed):
        """Schedule any sound, if needed."""
        flight = self._flight
        if flight.config.enableSounds:
            for sound in self._sounds:
                sound.check(flight, state, pilotHotkeyPressed)

#------------------------------------------------------------------------------

class ChecklistScheduler(object):
    """A scheduler for the checklist sounds"""
    def __init__(self, flight):
        """Construct the checklist scheduler for the given flight."""
        self._flight = flight
        self._checklist = None
        self._itemIndex = 0

    def hotkeyPressed(self):
        """Called when the checklist hotkey is pressed."""
        flight = self._flight
        config = flight.config
        if config.enableChecklists and flight.aircraftType is not None:
            if self._checklist is None:
                self._checklist = config.getChecklist(flight.aircraftType)

            index = self._itemIndex
            if index>=len(self._checklist):
                fs.sendMessage(const.MESSAGETYPE_INFORMATION,
                               "End of checklist")
            else:
                startSound(self._checklist[index])
                self._itemIndex += 1

#------------------------------------------------------------------------------