victor@2: '''
victor@2: Created on 10/11/2014
victor@2: 
victor@2: @organization: Lancaster University & University of Leeds
victor@2: @version: 1.0
victor@2: Created on 11/12/2014
victor@2: 
victor@2: @author: Victor Padilla
victor@2: @contact: v.padilla@lancaster.ac.uk
victor@2: 
victor@2: Functions related to the full score alignment.
victor@2: It takes the n parts and align with the best omr full score
victor@2: for the missing rest measuress
victor@2: '''
victor@2: from music21 import converter
victor@2: from music21 import stream
victor@2: from Functions import HashFunctions
victor@2: from Functions import FilesFunctions
victor@2: 
victor@2: 
victor@2: class FullScoreAlignment:
victor@2:     '''
victor@2:     Class for the full score alignment
victor@2:     
victor@2:     usage:
victor@2:     
victor@2:     fsa=FullScoreAlignment()   
victor@2:     idCompleteScoreBetter=fsa.getIdBetterOMRFullScore(fsOMRs,partsNumber)
victor@2:     finalScore=fsa.runSynchronisingMeasuresNJ(subdirnameFullScore,subdirnameParts,fsOMRs[idCompleteScoreBetter])  
victor@2:     finalScore.write("musicxml", dirGeneral+'/finalScore.xml')   
victor@2:         
victor@2:     '''
victor@2:     def runSynchronisingMeasuresNJ(self,subdirnameParts,completeScore):
victor@2:         '''
victor@2:         This function takes the different result.S2.xml files to construct
victor@2:         the final output.
victor@2:         
victor@2:         Example:
victor@2:         fsa=FullScoreAlignment()   
victor@2:         idCompleteScoreBetter=fsa.getIdBetterOMRFullScore(fsOMRs,partsNumber)
victor@2:         finalScore=fsa.runSynchronisingMeasuresNJ(subdirnameFullScore,subdirnameParts,fsOMRs[idCompleteScoreBetter])  
victor@2:         finalScore.write("musicxml", dirGeneral+'/finalScore.xml')   
victor@2:         '''
victor@2: 
victor@2:         gapOMRs=[]
victor@2:         parts=[]
victor@2:         completeScoreParts=[]
victor@2:         
victor@2:         # Convert to Hash the better full score
victor@2:         hf=HashFunctions()
victor@2:         for p in completeScore.getElementsByClass(stream.Part):
victor@2:             sc=stream.Score()
victor@2:             sc.append(p)
victor@2:             completeScoreParts.append(sc)
victor@2:         completeScoreHash=hf.getHash(completeScoreParts)
victor@2: 
victor@2:         # Convert to hash the parts
victor@2:         for dirname in sorted(subdirnameParts):
victor@2:             print "--- ",dirname
victor@2:             path = dirname+"/XML/"
victor@2:             partId=int(dirname.split('\\')[-1])
victor@2:             print partId
victor@2:             resultS2File=path+ "result.S2.xml"
victor@2:             resultS2=converter.parse(resultS2File,forceSource=False)
victor@2:             parts.append(resultS2)
victor@2:         partsHash=hf.getHash(parts)   
victor@2: 
victor@2: 
victor@2:         for partHash in partsHash:
victor@2:             index=partsHash.index(partHash)
victor@2:             h=[]
victor@2:             h.append(partHash)
victor@2:             h.append(completeScoreHash[index])
victor@2:             hf.alignHash(h)
victor@2:             partsHash[index]=h[0]
victor@2:             completeScoreHash[index]=h[1]
victor@2: 
victor@2:             gaps=hf.getGapsFromHashArray(h)
victor@2:             gapOMRs.append(gaps)
victor@2:             if(index<len(partsHash)-1):
victor@2:                 nextPart=completeScoreHash[index+1]
victor@2:                 for gap in gaps[1]:
victor@2:                     nextPart.insert(gap,"*")
victor@2: 
victor@2:         print "--------------trace back-----------"
victor@2:         print gapOMRs
victor@2:         sc=stream.Score() 
victor@2:         gaps=gapOMRs[len(partsHash)-1][1]
victor@2:         for i in range(len(partsHash)-1):
victor@2:             completeScoreHash[i]=hf.removeHashGaps(completeScoreHash[i])
victor@2:             completeScoreHash[i]=hf.addHashGaps(completeScoreHash[i],gaps)
victor@2:             partsHash[i]=hf.removeHashGaps(partsHash[i])
victor@2:             h=[]
victor@2:             h.append(partsHash[i])
victor@2:             h.append(completeScoreHash[i])
victor@2:             
victor@2:             h=hf.alignHash(h)
victor@2:             partsHash[i]=h[0]
victor@2:             completeScoreHash[i]=h[1]
victor@2:              
victor@2:         print "--------------reconstruct scores-----------" 
victor@2:          
victor@2:         streams,gaps=hf.reconstructHash(parts,partsHash)
victor@2:         for s in streams:
victor@2:             part=s.getElementsByClass(stream.Part)[0]
victor@2:             sc.append(part)
victor@2:         return sc
victor@2:         
victor@2:                               
victor@2:     
victor@2:     def getIdBetterOMRFullScore(self,fsOMRs_files,partsNumber):
victor@2:         '''
victor@2:         Tries to obtain the best full omr score
victor@2:         based on the parts and the length.
victor@2:         
victor@2:         Omrs increase the part numbers and add new rest measures
victor@2:         to synchronize
victor@2:         
victor@2:         usage:
victor@2:         # fsOMRs is an array with the  full score omrs
victor@2:         # for string quartet, 4 parts
victor@2:         idCompleteScoreBetter=fsa.getIdBetterOMRFullScore(fsOMRs,4)
victor@2:         
victor@2:         
victor@2:         '''
victor@2:         print "---------Calculating better OMR for alignment------------"
victor@2:         ff=FilesFunctions()
victor@2:         measuresLengthBetter=1000000 #maximum
victor@2:         partsLengthBetter=1000 #maximum
victor@2:         idCompleteScoreBetter=1000 #maximum
victor@2:         for cs_file in fsOMRs_files:
victor@2:             cs=ff.getOMR(cs_file)
victor@2:             if cs!=[]:
victor@2:                 try: 
victor@2:                     isEqualParts=False
victor@2:                     isLessMeasures=False
victor@2:                     isBetterOMR=False
victor@2:                     idCompleteScore=fsOMRs_files.index(cs_file)
victor@2:                     measuresLength=len(cs.getElementsByClass(stream.Part)[0].getElementsByClass(stream.Measure))
victor@2:                     partsLength=len(cs.getElementsByClass(stream.Part))
victor@2:                     print idCompleteScore,measuresLength,partsLength
victor@2:         
victor@2:                     if(partsLength==partsNumber):
victor@2:                         isEqualParts=True
victor@2:                     if(measuresLength<measuresLengthBetter):
victor@2:                         isLessMeasures=True
victor@2:         
victor@2:                     if isLessMeasures and isEqualParts:
victor@2:                         isBetterOMR=True
victor@2:                     if isBetterOMR:
victor@2:                         idCompleteScoreBetter=idCompleteScore
victor@2:                         measuresLengthBetter=measuresLength
victor@2:                         partsLengthBetter=partsLength
victor@2:                 except:
victor@2:                     print "error OMR"
victor@2:         print idCompleteScoreBetter,measuresLengthBetter,partsLengthBetter
victor@2:         return idCompleteScoreBetter
victor@2:     
victor@2: 
victor@2:                  
victor@2:                   
victor@2: