def encode_event()

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

• 14 lines of code
• 6 McCabe index (conditional complexity)

    def encode_event(self, event):

        event_name = None
        if event.event_type == note_seq.performance_lib.PerformanceEvent.NOTE_ON:
            event_name = f"NOTE_ON_{event.event_value}" 
        elif event.event_type == note_seq.performance_lib.PerformanceEvent.NOTE_OFF:
            event_name = f"NOTE_OFF_{event.event_value}" 
        elif event.event_type == note_seq.performance_lib.PerformanceEvent.TIME_SHIFT:
            event_name = f"TIME_SHIFT_{event.event_value}" 
        elif event.event_type == note_seq.performance_lib.PerformanceEvent.VELOCITY:
            event_name = f"VELOCITY_{event.event_value}"  
            
        if event_name:
            return self.events_to_ids[event_name]
        else:
            raise ValueError(f"Unknown event type: {event.event_type}")