def decode_event()

in transformer-xl/utils/performance_event_repo.py [0:0]

• 19 lines of code
• 5 McCabe index (conditional complexity)

    def decode_event(self, index):
        try:
            event_name = self.ids_to_events[index]
            event_splits = event_name.split('_')
            event_type, event_value = '_'.join(event_splits[:-1]), int(event_splits[-1])
            if event_type == 'NOTE_ON':
                return note_seq.performance_lib.PerformanceEvent(
                    event_type=note_seq.performance_lib.PerformanceEvent.NOTE_ON, event_value=event_value)
            elif event_type == 'NOTE_OFF':
                return note_seq.performance_lib.PerformanceEvent(
                    event_type=note_seq.performance_lib.PerformanceEvent.NOTE_OFF, event_value=event_value)
            elif event_type == 'TIME_SHIFT':
                return note_seq.performance_lib.PerformanceEvent(
                    event_type=note_seq.performance_lib.PerformanceEvent.TIME_SHIFT, event_value=event_value)
            elif event_type == 'VELOCITY':
                return note_seq.performance_lib.PerformanceEvent(
                    event_type=note_seq.performance_lib.PerformanceEvent.VELOCITY, event_value=event_value)
        except:
            raise ValueError('Unknown event index: %s' % index)