def run(gates_list, clock: Signal = None, n_pulse=None, display=None):
        if n_pulse is None:
            n_pulse = -1
        if clock is None:
            depend = []
            for gate in gates_list:
                gate.logic(depend)

            if display:
                for lst in display:
                    if type(lst[0]) in CircuitRunner.NOT_GATE:
                        print("".join([str(r.output.output) for r in lst]))
                    else:
                        print("".join([str(r.output) for r in lst]))
        else:
            while not n_pulse == 0:
                n_pulse -= 1
                clock.pulse()

                depend = []
                for gate in gates_list:
                    gate.logic(depend)

                if display:
                    for lst in display:
                        if type(lst[0]) in CircuitRunner.NOT_GATE:
                            print("".join([str(r.output.output) for r in lst]))
                        else:
                            print("".join([str(r.output) for r in lst]))
Esempio n. 2
0
def show_statistics(read_signal: Signal):
    print("Wartość średnia:                                        " +
          str(read_signal.average_signal_value()))
    print("Wartość średnia bezwzględna:                            " +
          str(read_signal.absolute_average_signal_value()))
    print("Moc średnia:                                            " +
          str(read_signal.average_power_of_signal()))
    print("Wariancja sygnału w przedziale wokół wartości średniej: " +
          str(read_signal.signal_variance()))
    print("Wartość skuteczna:                                      " +
          str(read_signal.effective_value()))
def test2():
    clock = Signal()
    d1 = D_FlipFlop(clock, None, "d1")
    not1 = Not(d1, "not")
    d1.set_input(not1)
    d1.set()

    for _ in range(20):
        clock.pulse()
        d1.logic()
        print(d1)
Esempio n. 4
0
def extractSignalsFromWav(path):
    file = scipy.io.wavfile.read(path)
    sampleRate = file[0]
    numberOfSamples = len(file[1])
    signals = []
    if numberOfSamples > 0:
        if isinstance(file[1][0], np.ndarray):
            channels = zip(*file[1])
            for channel in channels:
                signals.append(Signal(channel, sampleRate))
        else:
            signals.append(Signal(file[1], sampleRate))
    return signals
def johnson_counter(n=100):
    clock = Signal()
    bits = [D_FlipFlop(clock, None, f"d{i}") for i in range(n)]
    for i in range(1, n):
        bits[i].set_input(bits[i - 1])
        bits[i].reset()

    bits[0].set_input(Not(bits[-1], "not"))
    bits[0].reset()

    for _ in range(4 * n):
        clock.pulse()
        bits[0].logic()
        print("".join([str(b.q()) for b in bits]))
def n_bit_adder():
    clock = Signal()
    n = 200
    a, b = "01001" * 40, "01110" * 40

    d1 = [D_FlipFlop(clock, None, f"a{i}") for i in range(n)]
    d2 = [D_FlipFlop(clock, None, f"b{i}") for i in range(n)]

    adder = [FullAdder(None, None, f"adder{i}") for i in range(n)]

    res = [D_FlipFlop(clock, None, f"r{i}") for i in range(n)]

    for i in range(n):
        d1[i].set_input(d1[i])
        d2[i].set_input(d2[i])
        adder[i].set_input((d1[i], d2[i]))
        adder[i].set_cin(Zero() if i == 0 else adder[i - 1].cout)

        res[i].set_input(adder[i].sum)
        res[i].reset()

        if a[n - i - 1] == '0':
            d1[i].reset()
        else:
            d1[i].set()

        if b[n - 1 - i] == '0':
            d2[i].reset()
        else:
            d2[i].set()

    CircuitRunner.run(res, clock, 3, [res])
def test1():
    clock = Signal()
    l1 = D_Latch(clock, None, "l1")

    l1.set_input(l1)
    l1.set()

    CircuitRunner.run([l1], clock, 4, [[l1]])
Esempio n. 8
0
    def generate(self, amplitude: float, start_time: float, duration: float,
                 period: float, sampling_frequency: float) -> Signal:
        x = self.get_arguments(start_time, duration, sampling_frequency)

        y = amplitude * np.sin(((2 * math.pi) / period) * (x - start_time))

        return Signal(start_time, 1 / period, sampling_frequency, y,
                      SignalType.REAL, SignalPeriodic.YES)
    def generate(self, amplitude: float, start_time: float, duration: float, period: float,
                 sampling_frequency: float) -> Signal:
        x = self.get_arguments(start_time, duration, sampling_frequency)

        y = np.random.rand(len(x)) * (2 * amplitude) - amplitude
        z = list(y)

        return Signal(start_time, 1 / period, sampling_frequency, z, SignalType.REAL, SignalPeriodic.NO)
Esempio n. 10
0
    def generate(self, amplitude: float, start_time: float, duration: float, period: float, fill_factor: float,
                 sampling_frequency: float) -> Signal:
        x = self.get_arguments(start_time, duration, sampling_frequency)
        frequency = 1 / period

        y = amplitude * signal.square(x * frequency, duty=fill_factor)

        return Signal(start_time, frequency, sampling_frequency, y, SignalType.REAL, SignalPeriodic.YES)
Esempio n. 11
0
    def perform_signal_action(self, signal1: Signal, signal2: Signal,
                              operation: operator) -> Signal:
        check_compatibility(signal1, signal2)
        singal1_values = signal1.samples
        signal2_values = signal2.samples

        new_values = operation(singal1_values, signal2_values)

        return Signal(signal1.start_time, signal1.signal_frequency,
                      signal1.sampling_frequency, new_values, SignalType.REAL,
                      get_periodical(signal1, signal2))
Esempio n. 12
0
    def generate(self, amplitude: float, start_time: float, duration: float,
                 sampling_frequency: float) -> Signal:
        x = self.get_arguments(start_time, duration, sampling_frequency)
        if start_time <= 0 and abs(round(start_time)) <= duration:
            index_of_jump = round(abs(start_time) / duration * (len(x) - 1))
            y = amplitude * signal.unit_impulse(len(x), index_of_jump)
        else:
            y = x * 0.0

        return Signal(start_time, 0.0, sampling_frequency, list(y),
                      SignalType.REAL, SignalPeriodic.NO)
Esempio n. 13
0
 def __init__(self):
     self.explodeOnRemove = False
     self.exploded = False
     self.unloaded = False
     self.model = None
     self.additionalModels = []
     self.animationSequencer = None
     self.animationStateObject = None
     self.released = False
     self.wreckID = None
     self._audioEntities = []
     self._audioEntity = None
     self.logger = logging.getLogger('spaceObject.' +
                                     self.__class__.__name__)
     self.logger = SpaceObjectLogAdapter(self.logger, so_id=self.id)
     self.modelLoadedEvent = locks.Event()
     self.modelLoadSignal = Signal()
     self.explosionModel = None
     self.typeID = None
     self.typeData = {}
     self.explosionManager = ExplosionManager()
Esempio n. 14
0
    def generate(self, amplitude: float, start_time: float, duration: float,
                 jump_time: float, sampling_frequency: float) -> Signal:
        number_of_low_values = (jump_time - start_time) * sampling_frequency
        number_of_high_values = (duration -
                                 (jump_time - start_time)) * sampling_frequency

        y_low = [0.0] * round(number_of_low_values)
        y_high = [amplitude] * round(number_of_high_values)

        y = y_low + [(amplitude / 2)] + y_high

        return Signal(start_time, 0.0, sampling_frequency, list(y),
                      SignalType.REAL, SignalPeriodic.NO)
Esempio n. 15
0
 def generate(self, amplitude: float, start_time: float, duration: float,
              period: float, fill_factor: float,
              sampling_frequency: float) -> Signal:
     x = self.get_arguments(start_time, 2 * math.pi,
                            sampling_frequency / (2 * math.pi * 10))
     frequency = 1 / period
     y = ((amplitude * signal.sawtooth(x, fill_factor) + amplitude) *
          0.5) - 0.5 * amplitude
     y2 = list(y)
     for i in range(round(duration / period) - 1):
         y2 = y2 + list(y)
     return Signal(start_time, frequency, sampling_frequency, y2,
                   SignalType.REAL, SignalPeriodic.YES)
Esempio n. 16
0
def draw_histogram(signal: Signal, number_of_compartment: int):
    if signal.signal_periodic == SignalPeriodic.YES:
        number_of_samples_in_one_period = signal.get_number_of_samples_in_one_period(
        )
        number_of_samples_in_period = len(
            signal.samples
        ) // number_of_samples_in_one_period * number_of_samples_in_one_period
    else:
        number_of_samples_in_period = len(signal.samples)

    bins = count_intervals2(signal, number_of_compartment)
    plt.hist(signal.samples[:number_of_samples_in_period],
             bins=bins,
             edgecolor="k")
    plt.xticks(bins)
Esempio n. 17
0
    def division(self, signal1: Signal, signal2: Signal):
        check_compatibility(signal1, signal2)
        signal1_values = signal1.samples
        signal2_values = signal2.samples

        new_values = list(range(len(signal1_values)))
        for i in range(len(signal1_values)):
            if signal2_values[i] != 0.0:
                new_values[i] = signal1_values[i] / signal2_values[i]
            else:
                new_values[i] = signal1_values[i]

        return Signal(signal1.start_time, signal1.signal_frequency,
                      signal1.sampling_frequency, new_values, SignalType.REAL,
                      get_periodical(signal1, signal2))
Esempio n. 18
0
    def generate(self, amplitude: float, start_time: float, duration: float,
                 period: float, probability: float,
                 sampling_frequency: float) -> Signal:

        y = list(range(round(duration * sampling_frequency)))
        for i in range(round(duration * sampling_frequency)):
            random_number = np.random.rand()
            if random_number < probability:
                y[i] = amplitude
            else:
                y[i] = 0

        frequency = 1 / period
        return Signal(start_time, frequency, sampling_frequency, list(y),
                      SignalType.REAL, SignalPeriodic.NO)
Esempio n. 19
0
    def filter(self, signal: Signal) -> Signal:
        x = signal.samples
        n = len(x)

        splot = list(range(n))
        for i in range(n):
            point = 0
            for j in range(self.m - 1):
                point += (self.filter_impulsive_answer(j) *
                          get_value(x, i - j))
                # point += high_pass(j) * self.hanning(j) * self.filter_impulsive_answer(j) * get_value(x, i - j)
            splot[i] = point

        signal2 = Signal(signal.start_time, signal.signal_frequency,
                         signal.sampling_frequency, splot, signal.signal_type,
                         signal.signal_periodic)
        return signal2
Esempio n. 20
0
    def __init__(self,
                 signal,
                 filterDetail,
                 lowestFrequency=None,
                 highestFrequency=None):
        self.signal = signal
        self.highestFrequency = highestFrequency
        self.lowestFrequency = lowestFrequency
        self.filterDetail = filterDetail

        frequencies = getDFTFrequencies(filterDetail,
                                        self.signal.getSampleRate())
        filterFreq = list(
            map(
                lambda f: 1.0 if f <
                (highestFrequency or self.signal.getSampleRate() / 2.0) and f >
                (lowestFrequency or 0.0) else 0.0, frequencies))
        filterSequence = list(np.fft.irfft(filterFreq))
        self.filterSignal = Normalize(
            Signal(filterSequence, self.signal.getSampleRate()))
        super(BandPassFilter, self).__init__(self.signal, self.filterSignal)
Esempio n. 21
0
 def generate(self, amplitude: float, start_time: float, duration: float, period: float, fill_factor: float,
              sampling_frequency: float) -> Signal:
     x = self.get_arguments(start_time, duration, sampling_frequency)
     frequency = 1/period
     y = (amplitude * signal.sawtooth(x, fill_factor) + amplitude) * 0.5
     return Signal(start_time, frequency, sampling_frequency, list(y), SignalType.REAL, SignalPeriodic.YES)
Esempio n. 22
0
class SpaceObject(decometaclass.WrapBlueClass('destiny.ClientBall')):
    __persistdeco__ = 0
    __update_on_reload__ = 1

    def __init__(self):
        self.explodeOnRemove = False
        self.exploded = False
        self.unloaded = False
        self.model = None
        self.additionalModels = []
        self.animationSequencer = None
        self.animationStateObject = None
        self.released = False
        self.wreckID = None
        self._audioEntities = []
        self._audioEntity = None
        self.logger = logging.getLogger('spaceObject.' +
                                        self.__class__.__name__)
        self.logger = SpaceObjectLogAdapter(self.logger, so_id=self.id)
        self.modelLoadedEvent = locks.Event()
        self.modelLoadSignal = Signal()
        self.explosionModel = None
        self.typeID = None
        self.typeData = {}
        self.explosionManager = ExplosionManager()

    def GetPositionCurve(self):
        return self

    def GetTypeID(self):
        if self.typeID is None:
            self.typeID = self.typeData.get('typeID', None)
        return self.typeID

    def SetServices(self, spaceMgr, serviceMgr):
        self.spaceMgr = spaceMgr
        self.sm = serviceMgr
        self.spaceObjectFactory = serviceMgr.GetService(
            'sofService').spaceObjectFactory

    def Prepare(self):
        self.typeID = self.typeData.get('typeID', None)
        self.LoadModel()
        self.Assemble()

    def HasBlueInterface(self, obj, interfaceName):
        if hasattr(obj, 'TypeInfo'):
            return interfaceName in obj.TypeInfo()[1]
        return False

    def _GetComponentRegistry(self):
        return self.ballpark.componentRegistry

    def TriggerAnimation(self, state, **kwargs):
        if self.animationSequencer is None:
            return
        self.logger.debug(
            'SpaceObject: Trigger animation %s with parameters %s', state,
            kwargs)
        for parameterName, parameterValue in kwargs.iteritems():
            self.animationSequencer.SetStateParameter(state, parameterName,
                                                      parameterValue)

        self.animationSequencer.GoToState(state)
        if self.animationStateObject:
            self.RemoveAndClearModel(self.animationStateObject)
            self.animationStateObject = None
        if state in self.typeData['animationStateObjects']:
            dnaToLoad = self.typeData['animationStateObjects'][state]
            self.animationStateObject = self.spaceObjectFactory.BuildFromDNA(
                dnaToLoad)
            self._SetupModelAttributes(self.animationStateObject,
                                       '%d_%s' % (self.id, state))
            self.animationStateObject.rotationCurve = self.model.rotationCurve
            self._AddModelToScene(self.animationStateObject)

    def GetCurrentAnimationState(self, stateMachineName):
        if self.animationSequencer is None:
            return
        for stateMachine in self.animationSequencer.stateMachines:
            if stateMachine.name == stateMachineName:
                if stateMachine.currentState is None:
                    return
                else:
                    return stateMachine.currentState.name

    def GetModel(self):
        if not self.model:
            if blue.os.isOnMainTasklet:
                return None
            self.modelLoadedEvent.wait()
        return self.model

    def GetDNA(self):
        materialSetID = self.typeData.get('slimItem').skinMaterialSetID
        return gfxutils.BuildSOFDNAFromTypeID(self.typeData['typeID'],
                                              materialSetID=materialSetID)

    def _LoadModelResource(self, fileName=None):
        self.logger.debug('LoadModel: %s', fileName)
        model = None
        sofDNA = self.GetDNA()
        self.logger.debug("LoadModel fileName='%s' sofDNA='%s'", fileName,
                          sofDNA)
        if sofDNA is not None and fileName is None:
            model = self.spaceObjectFactory.BuildFromDNA(sofDNA)
        else:
            if fileName is None:
                fileName = self.typeData.get('graphicFile')
            if fileName is not None and len(fileName):
                model = blue.resMan.LoadObject(fileName)
        if model is None:
            self.logger.error(
                'Error: Object type %s has invalid graphicFile, using graphicID: %s',
                self.typeData['typeID'], self.typeData['graphicID'])
        return model

    def _SetupModelAndAddToScene(self, fileName=None, loadedModel=None):
        if loadedModel:
            model = loadedModel
        else:
            model = self._LoadModelResource(fileName)
        if self.released:
            return None
        if not model:
            self.logger.error(
                'Could not load model for spaceobject. FileName:%s typeID:%s',
                fileName, getattr(self, 'typeID', '?'))
            return None
        self._SetupModelAttributes(model, '%d' % self.id)
        self._AddModelToScene(model)
        return model

    def _SetupModelAttributes(self, model, objectName):
        model.translationCurve = self
        model.rotationCurve = self
        model.name = objectName
        if hasattr(model, 'useCurves'):
            model.useCurves = 1
        if model and hasattr(model, 'albedoColor'):
            model.albedoColor = eveSpaceObject.GetAlbedoColor(model)

    def _AddModelToScene(self, model):
        if model is not None:
            scene = self.spaceMgr.GetScene()
            if scene is not None:
                scene.objects.append(model)
            else:
                raise RuntimeError('Invalid object loaded by spaceObject: %s' %
                                   str(model))

    def LoadAdditionalModel(self, fileName=None):
        model = self._SetupModelAndAddToScene(fileName)
        if fileName is not None:
            self.additionalModels.append(model)
        return model

    def NotifyModelLoaded(self):
        if self.model is not None:
            self.logger.debug('SpaceObject - NotifyModelLoaded')
            self.modelLoadedEvent.set()
            self.modelLoadSignal()
            self.sm.GetService('FxSequencer').NotifyModelLoaded(self.id)
        else:
            self.logger.warning(
                'SpaceObject - NotifyModelLoaded called without a model present, no notification was done'
            )

    def RegisterForModelLoad(self, func):
        self.modelLoadSignal.connect(func)

    def LoadModel(self, fileName=None, loadedModel=None):
        self.model = self._SetupModelAndAddToScene(fileName, loadedModel)
        if self.model is None:
            return
        self.SetupAnimationInformation(self.model)
        self.NotifyModelLoaded()

    def SetupAnimationInformation(self, model):
        self._SetupAnimationStateMachines(model)
        self._SetupAnimationUpdater(model)

    def SetAnimationSequencer(self, model):
        if model is not None and hasattr(model, 'animationSequencer'):
            self.animationSequencer = model.animationSequencer
        else:
            self.animationSequencer = None

    def _SetupAnimationStateMachines(self, model):
        animationStates = self.typeData['animationStates']
        if len(animationStates) == 0:
            return
        spaceobjanimation.LoadAnimationStates(animationStates,
                                              cfg.graphicStates, model,
                                              trinity)
        self.SetAnimationSequencer(model)

    def _SetupAnimationUpdater(self, model):
        if not hasattr(
                model,
                'animationUpdater') or not self.typeData['animationStates']:
            return
        if self._audioEntity is None:
            self._audioEntity = self._GetGeneralAudioEntity(model=model)
        if model is not None and model.animationUpdater is not None:
            model.animationUpdater.eventListener = self._audioEntity

    def Assemble(self):
        pass

    def GetStaticRotation(self):
        rot = self.typeData.get('dunRotation', None)
        if rot:
            yaw, pitch, roll = map(math.radians, rot)
            return geo2.QuaternionRotationSetYawPitchRoll(yaw, pitch, roll)
        else:
            return (0.0, 0.0, 0.0, 1.0)

    def SetStaticRotation(self):
        if self.model is None:
            return
        self.model.rotationCurve = None
        rot = self.typeData.get('dunRotation', None)
        if rot:
            yaw, pitch, roll = map(math.radians, rot)
            quat = geo2.QuaternionRotationSetYawPitchRoll(yaw, pitch, roll)
            if hasattr(self.model, 'rotation'):
                if type(self.model.rotation) == types.TupleType:
                    self.model.rotation = quat
                else:
                    self.model.rotation.SetYawPitchRoll(yaw, pitch, roll)
            else:
                self.model.rotationCurve = trinity.TriRotationCurve()
                self.model.rotationCurve.value = quat
                if self.animationStateObject is not None:
                    self.animationStateObject.rotationCurve = self.model.rotationCurve

    def _FindClosestBallDir(self, constgrp):
        bp = self.sm.StartService('michelle').GetBallpark()
        dist = 1e+100
        closestID = None
        for ballID, slimItem in bp.slimItems.iteritems():
            if slimItem.groupID == constgrp:
                test = bp.DistanceBetween(self.id, ballID)
                if test < dist:
                    dist = test
                    closestID = ballID

        if closestID is None:
            return (1.0, 0.0, 0.0)
        ball = bp.GetBall(closestID)
        direction = geo2.Vec3SubtractD((self.x, self.y, self.z),
                                       (ball.x, ball.y, ball.z))
        return direction

    def FindClosestMoonDir(self):
        return self._FindClosestBallDir(const.groupMoon)

    def FindClosestPlanetDir(self):
        return self._FindClosestBallDir(const.groupPlanet)

    def GetStaticDirection(self):
        return self.typeData.get('dunDirection', None)

    def SetStaticDirection(self):
        if self.model is None:
            return
        self.model.rotationCurve = None
        direction = self.GetStaticDirection()
        if direction is None:
            self.logger.error(
                'No static direction defined - no rotation will be applied')
            return
        self.AlignToDirection(direction)

    def AlignToDirection(self, direction):
        if not self.model:
            return
        zaxis = direction
        if geo2.Vec3LengthSqD(zaxis) > 0.0:
            zaxis = geo2.Vec3NormalizeD(zaxis)
            xaxis = geo2.Vec3CrossD(zaxis, (0, 1, 0))
            if geo2.Vec3LengthSqD(xaxis) == 0.0:
                zaxis = geo2.Vec3AddD(zaxis, mathCommon.RandomVector(0.0001))
                zaxis = geo2.Vec3NormalizeD(zaxis)
                xaxis = geo2.Vec3CrossD(zaxis, (0, 1, 0))
            xaxis = geo2.Vec3NormalizeD(xaxis)
            yaxis = geo2.Vec3CrossD(xaxis, zaxis)
        else:
            self.logger.error('Invalid direction (%s). Unable to rotate it.',
                              direction)
            return
        mat = ((xaxis[0], xaxis[1], xaxis[2], 0.0), (yaxis[0], yaxis[1],
                                                     yaxis[2], 0.0),
               (-zaxis[0], -zaxis[1], -zaxis[2], 0.0), (0.0, 0.0, 0.0, 1.0))
        quat = geo2.QuaternionRotationMatrix(mat)
        if hasattr(self.model, 'modelRotationCurve'):
            if not self.model.modelRotationCurve:
                self.model.modelRotationCurve = trinity.TriRotationCurve(
                    0.0, 0.0, 0.0, 1.0)
            self.model.modelRotationCurve.value = quat
        else:
            self.model.rotationCurve = None

    def UnSync(self):
        if self.model is None:
            return
        startTime = long(random.random() * 123456.0 * 1234.0)
        scaling = 0.95 + random.random() * 0.1
        curves = timecurves.ReadCurves(self.model)
        timecurves.ResetTimeCurves(curves, startTime, scaling)

    def Display(self, display=1, canYield=True):
        if self.model is None:
            if display:
                self.logger.warning('Display - No model')
            return
        if canYield:
            blue.synchro.Yield()
        if eve.session.shipid == self.id and display and self.IsCloaked():
            self.sm.StartService('FxSequencer').OnSpecialFX(
                self.id, None, None, None, None, 'effects.CloakNoAmim', 0, 1,
                0, 5, 0)
            return
        if self.model:
            self.model.display = display

    def IsCloaked(self):
        return self.isCloaked

    def OnDamageState(self, damageState):
        pass

    def GetDamageState(self):
        bp = sm.GetService('michelle').GetBallpark()
        if bp is not None:
            return bp.GetDamageState(self.id)

    def _UpdateImpacts(self):
        states = self.GetDamageState()
        if states is not None and self.model is not None:
            damageState = [(d if d is not None else 0.0) for d in states]
            self.model.SetImpactDamageState(damageState[0], damageState[1],
                                            damageState[2], True)

    def DoFinalCleanup(self):
        if not self.sm.IsServiceRunning('FxSequencer'):
            return
        self.sm.GetService('FxSequencer').RemoveAllBallActivations(self.id)
        self.ClearExplosion()
        if not self.released:
            self.explodeOnRemove = False
            self.Release()
        elif self.HasModels():
            scene = self.spaceMgr.GetScene()
            self.ClearAndRemoveAllModels(scene)

    def ClearExplosion(self, model=None):
        if hasattr(self, 'gfx') and self.gfx is not None:
            self.RemoveAndClearModel(self.gfx)
            self.gfx = None
        if self.explosionModel is not None:
            if getattr(self, 'explosionDisplayBinding', False):
                self.explosionDisplayBinding.destinationObject = None
                self.explosionDisplayBinding = None
            self.RemoveAndClearModel(self.explosionModel)
            self.explosionModel = None

    def Release(self, origin=None):
        uthread2.StartTasklet(self._Release, origin)

    def _Release(self, origin=None):
        if self.released:
            return
        self.released = True
        if self.explodeOnRemove:
            delay = self.Explode()
            if delay:
                blue.synchro.SleepSim(delay)
        self.Display(display=0, canYield=False)
        for model in self.additionalModels:
            if model is not None:
                model.display = False

        if hasattr(self.model, 'animationSequencer'):
            self.model.animationSequencer = None
        self.animationSequencer = None
        if hasattr(self.model, 'animationUpdater'
                   ) and self.model.animationUpdater is not None:
            self.model.animationUpdater.eventListener = None
        self._audioEntities = []
        self._audioEntity = None
        scene = self.spaceMgr.GetScene()
        camera = sm.GetService('sceneManager').GetActiveSpaceCamera()
        lookingAt = camera.GetLookAtItemID()
        interestID = camera.GetTrackItemID()
        if self.explodeOnRemove and (self.id == lookingAt
                                     or interestID == self.id):
            self.RemoveAllModelsFromScene(scene)
        else:
            self.ClearAndRemoveAllModels(scene)

    def HasModels(self):
        return self.model is not None

    def ClearAndRemoveAllModels(self, scene):
        self.RemoveAndClearModel(self.model, scene)
        self.model = None
        for m in self.additionalModels:
            self.RemoveAndClearModel(m, scene)

        self.additionalModels = []
        if self.animationStateObject is not None:
            self.RemoveAndClearModel(self.animationStateObject, scene)
        self.animationStateObject = None

    def RemoveAllModelsFromScene(self, scene):
        if scene is None:
            return
        scene.objects.fremove(self.model)
        for m in self.additionalModels:
            scene.objects.fremove(m)

        if self.animationStateObject is not None:
            scene.objects.fremove(self.animationStateObject)

    def RemoveAndClearModel(self, model, scene=None):
        if model:
            self._Clearcurves(model)
        else:
            self.released = True
            return
        self.RemoveFromScene(model, scene)

    def _Clearcurves(self, model):
        if hasattr(model, 'translationCurve'):
            model.translationCurve = None
            model.rotationCurve = None
        if hasattr(model, 'observers'):
            for ob in model.observers:
                ob.observer = None

    def RemoveFromScene(self, model, scene):
        if scene is None:
            scene = self.spaceMgr.GetScene()
        if scene:
            scene.objects.fremove(model)

    def GetExplosionInfo(self):
        raceName = self.typeData.get('sofRaceName', None)
        return eveSpaceObject.GetDeathExplosionInfo(self.model, self.radius,
                                                    raceName)

    def GetExplosionLookAtDelay(self):
        return eveSpaceObject.GetDeathExplosionLookDelay(
            self.model, self.radius)

    def Explode(self,
                explosionURL=None,
                scaling=1.0,
                managed=False,
                delay=0.0):
        if self.exploded:
            return False
        self.sm.ScatterEvent('OnObjectExplode', self.GetModel())
        self.exploded = True
        delayedRemove = delay
        self.explodedTime = blue.os.GetTime()
        if gfxsettings.Get(gfxsettings.UI_EXPLOSION_EFFECTS_ENABLED):
            if SpaceObjectExplosionManager.USE_EXPLOSION_BUCKETS:
                explosionBucket = fsdExplosionBuckets.GetExplosionBucketByTypeID(
                    self.typeData['typeID'])
                if explosionBucket:
                    self.logger.debug('Exploding with explosion bucket')
                    scene = sm.GetService('space').GetScene()
                    wreckSwitchTime, _, __ = SpaceObjectExplosionManager.ExplodeBucketForBall(
                        self, scene)
                    return wreckSwitchTime
            if managed:
                gfx = self.explosionManager.GetExplosion(
                    explosionURL, callback=self.ClearExplosion)
            else:
                if explosionURL is None:
                    self.logger.error(
                        'explosionURL not set when calling Explode. Possibly wrongly authored content. typeID: %s',
                        self.typeID)
                    explosionURL, (delay, scaling) = self.GetExplosionInfo()
                explosionURL = explosionURL.replace('.blue', '.red').replace(
                    '/Effect/', '/Effect3/')
                gfx = trinity.Load(explosionURL)
                if not gfx:
                    self.logger.error(
                        'Failed to load explosion: %s - using default',
                        explosionURL)
                    gfx = trinity.Load(
                        'res:/Model/Effect3/Explosion/entityExplode_large.red')
                if isinstance(gfx, trinity.EveEffectRoot2):
                    msg = 'ExplosionManager circumvented, explosion not managed for %s. (Class:%s, Type:%s)'
                    self.logger.warning(msg, explosionURL,
                                        self.__class__.__name__, self.typeID)
                    gfx.Start()
                elif not isinstance(
                        gfx,
                    (trinity.EveRootTransform, trinity.EveEffectRoot2)):
                    root = trinity.EveRootTransform()
                    root.children.append(gfx)
                    root.name = explosionURL
                    gfx = root
            gfx.translationCurve = self
            self.explosionModel = gfx
            scale = scaling
            gfx.scaling = (gfx.scaling[0] * scale, gfx.scaling[1] * scale,
                           gfx.scaling[2] * scale)
            scene = self.spaceMgr.GetScene()
            if scene is not None:
                scene.objects.append(gfx)
        if self.wreckID is not None:
            wreckBall = self.sm.StartService('michelle').GetBall(self.wreckID)
            if wreckBall is not None:
                uthread2.StartTasklet(wreckBall.DisplayWreck, delayedRemove)
        return delayedRemove

    def PrepareForFiring(self):
        pass

    def GetEventNameFromSlimItem(self, defaultSoundUrl):
        slimItem = self.typeData.get('slimItem')
        eventName = spaceobjaudio.GetSoundUrl(slimItem, defaultSoundUrl)
        return eventName

    def SetupAmbientAudio(self, defaultSoundUrl=None):
        audioUrl = self.GetEventNameFromSlimItem(defaultSoundUrl)
        if audioUrl is None:
            return
        audentity = self._GetGeneralAudioEntity()
        if audentity is not None:
            spaceobjaudio.PlayAmbientAudio(audentity, audioUrl)

    def SetupSharedAmbientAudio(self, defaultSoundUrl=None):
        eventName = self.GetEventNameFromSlimItem(defaultSoundUrl)
        if eventName is None or self.model is None:
            return
        spaceobjaudio.SetupSharedEmitterForAudioEvent(self.model, eventName)

    def LookAtMe(self):
        pass

    def _GetGeneralAudioEntity(self, model=None, recreate=False):
        if model is None:
            model = self.model
        if model is None:
            self._audioEntity = None
            self.logger.warning('model is None, cannot play audio.')
        elif recreate or self._audioEntity is None:
            self._audioEntity = spaceobjaudio.SetupAudioEntity(model)
            self._audioEntities.append(self._audioEntity)
        return self._audioEntity

    def PlayGeneralAudioEvent(self, eventName):
        audentity = self._GetGeneralAudioEntity()
        if audentity is not None:
            spaceobjaudio.SendEvent(audentity, eventName)

    def GetNamedAudioEmitterFromObservers(self, emitterName):
        if getattr(self, 'model', None) is None:
            return
        for triObserver in self.model.observers:
            if triObserver.observer.name.lower() == emitterName:
                return triObserver.observer

    def PlaySound(self, event):
        if self.model is None:
            return
        if hasattr(self.model, 'observers'):
            for obs in self.model.observers:
                obs.observer.SendEvent(unicode(event))
                return

        self.logger.error(
            "Space Object: %s can't play sound. Sound observer not found.",
            self.typeData.get('typeName', None))
Esempio n. 23
0
 def transformFilterSignal(self, filterSignal):
     signal = filterSignal.getRange()
     halfLen = int(len(signal) / 2)
     signal = list(signal[halfLen:]) + list(signal[:halfLen])
     return Signal(signal, filterSignal.getSampleRate())
Esempio n. 24
0
 def get_new_x_values(self, signal: Signal, step: int):
     end_time = self._get_end_time(signal)
     return linspace(
         signal.start_time, end_time,
         int(step * signal.get_number_of_samples_in_one_period()))