File size: 3,588 Bytes
9965bf6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 |
import os
import pretty_midi
import argparse
def chunk_midi(input_path, output_dir, chunk_length=10.24):
# Ensure the output directory exists
if not os.path.exists(output_dir):
os.makedirs(output_dir)
for midi_file_name in os.listdir(input_path):
if not (midi_file_name.endswith('.midi') or midi_file_name.endswith('.mid')):
continue # Skip non-midi files
full_path = os.path.join(input_path, midi_file_name)
try:
midi_data = pretty_midi.PrettyMIDI(full_path)
except Exception as e:
print(f"Error processing {midi_file_name}: {e}")
continue # Skip to the next file if an error occurs
end_time = midi_data.get_end_time() # Get end time directly with pretty_midi
num_chunks = int(end_time // chunk_length) + (1 if end_time % chunk_length > 0 else 0)
base_name, file_extension = os.path.splitext(midi_file_name)
for i in range(num_chunks):
start_time = i * chunk_length
segment_end_time = min((i + 1) * chunk_length, midi_data.get_end_time())
# Create a new MIDI object for each chunk
chunk_midi_data = pretty_midi.PrettyMIDI()
# Merge non-drum instruments into a single instrument
merged_instrument = pretty_midi.Instrument(program=0, is_drum=False)
for instrument in midi_data.instruments:
if not instrument.is_drum:
for note in instrument.notes:
if start_time <= note.start < segment_end_time:
# Shift the note start and end times to start at 0
new_note = pretty_midi.Note(
velocity=note.velocity,
pitch=note.pitch,
start=note.start - start_time,
end=note.end - start_time
)
merged_instrument.notes.append(new_note)
else:
# If it's a drum instrument, just adjust the note times and append it
new_drum_instrument = pretty_midi.Instrument(program=instrument.program, is_drum=True, name=instrument.name)
new_drum_instrument.notes = [note for note in instrument.notes if start_time <= note.start < segment_end_time]
for note in new_drum_instrument.notes:
note.start -= start_time
note.end -= start_time
chunk_midi_data.instruments.append(new_drum_instrument)
# Add the merged instrument to the MIDI object
chunk_midi_data.instruments.append(merged_instrument)
# Save the chunk with the same extension as the original file
new_midi_name = "{}_{}{}".format(base_name, i, file_extension)
chunk_midi_data.write(os.path.join(output_dir, new_midi_name))
if __name__ == "__main__":
parser = argparse.ArgumentParser(description="Chunk MIDI files into specified lengths.")
parser.add_argument("--input_path", type=str, help="Path to the directory containing the MIDI files to chunk.")
parser.add_argument("--output_dir", type=str, help="Path to the directory where the chunked MIDI files will be saved.")
parser.add_argument("--chunk_length", type=float, default=10.24, help="length to chunk the midi file to (s).")
args = parser.parse_args()
chunk_midi(args.input_path, args.output_dir, chunk_length=args.chunk_length)
|