Upload TMIDIX.py

TMIDIX.py

@@ -1471,6 +1471,8 @@ import csv
 
 import tqdm
 
+import multiprocessing
+
 from itertools import zip_longest
 from itertools import groupby
 from collections import Counter
@@ -5589,7 +5591,7 @@ def split_melody(enhanced_melody_score_notes,
   for e in enhanced_melody_score_notes:
     dtime = max(0, min(max_score_time, e[1]-pe[1]))
 
-    if dtime > max(durs):
+    if dtime > stime:
       if chu:
         mel_chunks.append(chu)
       chu = []
@@ -9603,7 +9605,7 @@ def escore_notes_to_text_description(escore_notes,
 
     all_patches = [e[6] for e in escore_notes]
 
-    patches = ordered_set(all_patches)
+    patches = ordered_set(all_patches)[:16]
     
     instruments = [alpha_str(Number2patch[p]) for p in patches if p < 128]
     
@@ -9762,7 +9764,6 @@ def escore_notes_to_text_description(escore_notes,
             
             description += 'This ' + mood + ' melody has '
             
-                
         else:
             description += 'TThis drum track has '
             
@@ -9776,19 +9777,17 @@ def escore_notes_to_text_description(escore_notes,
         if instruments:
             
             if comp_type not in ['monophonic melody', 'drum track']:
+                
                 description += 'The song '
                 
-            else:
-                description += 'The melody '
-
-            if len(instruments) == 1:
-                description += 'is played on a solo ' + instruments[0] + '.'
+                if len(instruments) > 1:
                 
-                description += '\n'
-
-            else:
-                description += 'features ' + NUMERALS[max(0, min(15, len(instruments)-1))] + ' instruments: '
-                description += ', '.join(instruments[:-1]) + ' and ' + instruments[-1] + '.'
+                    description += 'features ' + NUMERALS[max(0, min(15, len(instruments)-1))] + ' instruments: '
+                    description += ', '.join(instruments[:-1]) + ' and ' + instruments[-1] + '.'
+                    
+                else:
+                    description += 'features one instrument: ' + instruments[0] + '.'
+                    
                 
                 description += '\n'
                 
@@ -9829,7 +9828,7 @@ def escore_notes_to_text_description(escore_notes,
                 description += 'The song has ' + NUMERALS[len(lead_melodies)-1] + ' distinct lead ' + mword + ' played on '
                 
                 if len(lead_melodies) > 1:               
-                    description += ', '.join(lead_melodies[:-1][0]) + ' and ' + lead_melodies[-1][0] + '.'
+                    description += ', '.join([l[0] for l in lead_melodies[:-1]]) + ' and ' + lead_melodies[-1][0] + '.'
                     
                 else:
                     description += lead_melodies[0][0] + '.'
@@ -9847,7 +9846,7 @@ def escore_notes_to_text_description(escore_notes,
                 description += 'The song has ' + NUMERALS[len(base_melodies)-1] + ' distinct base ' + mword + ' played on '
                 
                 if len(base_melodies) > 1:               
-                    description += ', '.join(base_melodies[:-1][0]) + ' and ' + base_melodies[-1][0] + '.'
+                    description += ', '.join([b[0] for b in base_melodies[:-1]]) + ' and ' + base_melodies[-1][0] + '.'
                     
                 else:
                     description += base_melodies[0][0] + '.'
@@ -9865,7 +9864,7 @@ def escore_notes_to_text_description(escore_notes,
                 description += 'The song has ' + NUMERALS[len(lead_melodies)-1] + ' distinct lead ' + lmword + ' played on '
                 
                 if len(lead_melodies) > 1:               
-                    description += ', '.join(lead_melodies[:-1][0]) + ' and ' + lead_melodies[-1][0] + '.'
+                    description += ', '.join([l[0] for l in lead_melodies[:-1]]) + ' and ' + lead_melodies[-1][0] + '.'
                     
                 else:
                     description += lead_melodies[0][0] + '.'
@@ -9879,7 +9878,7 @@ def escore_notes_to_text_description(escore_notes,
                 description += ' And ' + NUMERALS[len(base_melodies)-1] + ' distinct base ' + bmword + ' played on '
                 
                 if len(base_melodies) > 1:               
-                    description += ', '.join(base_melodies[:-1][0]) + ' and ' + base_melodies[-1][0] + '.'
+                    description += ', '.join([b[0] for b in base_melodies[:-1]]) + ' and ' + base_melodies[-1][0] + '.'
                     
                 else:
                     description += base_melodies[0][0] + '.'
@@ -11048,7 +11047,7 @@ LEAD_INSTRUMENTS = [0, 1, 2, 3, 4, 5, 6, 7, # Piano
                     8, 9, 10, 11, 12, 13, 14, 15, # Chromatic Percussion
                     16, 17, 18, 19, 20, 21, 22, 23, # Organ
                     24, 25, 26, 27, 28, 29, 30, 31, # Guitar
-                    40, 41, # Strings
+                    40, 41, 46, # Strings
                     52, 53, 54, # Ensemble
                     56, 57, 59, 60, # Brass
                     64, 65, 66, 67, 68, 69, 70, 71, # Reed
@@ -11091,6 +11090,97 @@ def escore_notes_pitches_range(escore_notes,
     else:
         return [ -1] * 6
 
+###################################################################################
+
+def escore_notes_core(escore_notes, core_len=128):
+
+    cscore = chordify_score([1000, escore_notes])
+
+    chords = []
+    chords_idxs = []
+    
+    for i, c in enumerate(cscore):
+        
+        pitches = [e[4] for e in c if e[3] != 9]
+
+        if pitches:
+            tones_chord = sorted(set([p % 12 for p in pitches]))
+
+            if tones_chord not in ALL_CHORDS_SORTED:
+                tones_chord = check_and_fix_tones_chord(tones_chord)
+
+            chords.append(ALL_CHORDS_SORTED.index(tones_chord))
+            chords_idxs.append(i)
+
+    mid = len(chords_idxs) // 2
+    clen = core_len // 2
+    
+    sidx = chords_idxs[mid-clen]
+    eidx = chords_idxs[mid+clen]
+
+    core_chords = chords[mid-clen:mid+clen]
+    core_score = flatten(cscore[sidx:eidx])
+    
+    return core_score, core_chords
+
+###################################################################################
+
+def multiprocessing_wrapper(function, data_list):
+    
+    with multiprocessing.Pool() as pool:
+        
+        results = []
+        
+        for result in tqdm.tqdm(pool.imap_unordered(function, data_list), total=len(data_list)):
+            results.append(result)
+            
+    return results
+
+###################################################################################
+
+def rle_encode_ones(matrix, div_mod=-1):
+    
+    flat_list = [val for row in matrix for val in row]
+    
+    encoding = []
+    i = 0
+    
+    while i < len(flat_list):
+        
+        if flat_list[i] == 1:
+            
+            start_index = i
+            count = 1
+            i += 1
+            
+            while i < len(flat_list) and flat_list[i] == 1:
+                count += 1
+                i += 1
+            
+            if div_mod > 0:
+                encoding.append((start_index // div_mod, start_index % div_mod))
+                
+            else:
+                encoding.append(start_index)
+            
+        else:
+            i += 1
+            
+    return encoding
+
+###################################################################################
+
+def rle_decode_ones(encoding, size=(128, 128)):
+    
+    flat_list = [0] * (size[0] * size[1])
+    
+    for start_index in encoding:
+        flat_list[start_index] = 1
+        
+    matrix = [flat_list[i * size[1]:(i + 1) * size[1]] for i in range(size[0])]
+    
+    return matrix
+
 ###################################################################################
 #  
 # This is the end of the TMIDI X Python module