File size: 2,180 Bytes
3c428bc |
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 |
input_dir = "<path_to_your_xml_files>" # Replace with the path to your folder containing XML (.xml, .mxl, .musicxml) files
import os
import math
import random
import subprocess
from tqdm import tqdm
from multiprocessing import Pool
def convert_xml2abc(file_list):
cmd = 'cmd /u /c python utils/xml2abc.py -d 8 -x '
for file in tqdm(file_list):
filename = file.split('/')[-1]
output_dir = file.split('/')[:-1]
output_dir[0] = output_dir[0] + '_abc'
output_dir = '/'.join(output_dir)
os.makedirs(output_dir, exist_ok=True)
try:
p = subprocess.Popen(cmd + '"' + file + '"', stdout=subprocess.PIPE, shell=True)
result = p.communicate()
output = result[0].decode('utf-8')
if output == '':
with open("logs/xml2abc_error_log.txt", "a", encoding="utf-8") as f:
f.write(file + '\n')
continue
else:
with open(output_dir + '/' + ".".join(filename.split(".")[:-1]) + '.abc', 'w', encoding='utf-8') as f:
f.write(output)
except Exception as e:
with open("logs/xml2abc_error_log.txt", "a", encoding="utf-8") as f:
f.write(file + ' ' + str(e) + '\n')
pass
if __name__ == '__main__':
file_list = []
os.makedirs("logs", exist_ok=True)
# Traverse the specified folder for XML/MXL files
for root, dirs, files in os.walk(input_dir):
for file in files:
if not file.endswith((".mxl", ".xml", ".musicxml")):
continue
filename = os.path.join(root, file).replace("\\", "/")
file_list.append(filename)
# Prepare for multiprocessing
file_lists = []
random.shuffle(file_list)
for i in range(os.cpu_count()):
start_idx = int(math.floor(i * len(file_list) / os.cpu_count()))
end_idx = int(math.floor((i + 1) * len(file_list) / os.cpu_count()))
file_lists.append(file_list[start_idx:end_idx])
pool = Pool(processes=os.cpu_count())
pool.map(convert_xml2abc, file_lists)
|