#!/usr/bin/python

# Copyright (c) 2001-2006 MUSIC TECHNOLOGY GROUP (MTG)
#                         UNIVERSITAT POMPEU FABRA
#
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA


import sys
import re

def Lab2Plot(labfile, output, segmentationAttribute, filter, childScope, labelAttribute):
	print >> output, '<?xml version="1.0" encoding="UTF-8" standalone="no" ?>'
	print >> output, '<DescriptorsPool>'
	print >> output, '\t<ScopePool name="Song" size="1">'
	tokens = list()
	labelFilter = re.compile(filter)
	for line in labfile:
		segment = line.split(" ",3)
		if len(segment)!=3:
			print >> sys.stderr, "Found a line with %s tokens, 3 expected"%len(segment)
			continue
		segment[2]=segment[2].rstrip().lstrip()
		if not labelFilter.match(segment[2]): continue
		tokens.append(segment)

	sampleRate=44100

	print >> output, '\t\t<AttributePool name="%s" size="%s">'%(segmentationAttribute, len(tokens)*2) ,
	for segment in tokens :
		#In samples
		#print >> output, int(round(float(segment[0])*sampleRate)), int(round(float(segment[1])*sampleRate)),
		#In seconds
		print >> output, segment[0], segment[1],
	print >> output, '</AttributePool>'
	print >> output, '\t</ScopePool>'

	if childScope!=None:
		print >> output, '\t<ScopePool name="%s" size="%s">' % (childScope, len(tokens))
		print >> output, '\t\t<AttributePool name="%s">' % (labelAttribute)

		for segment in tokens :
			print >> output, '\t\t\t<Enumerated>%s</Enumerated>' % ( segment[2] ) 
		print >> output, '\t\t</AttributePool>'
		print >> output, '\t</ScopePool>'
	print >> output, '</DescriptorsPool>'
	return dict.fromkeys([ segment[2] for segment in tokens ]).keys() # unique labels

def generateSchema(mainScope, segmentationAttribute, childScope, labelAttribute, values):
	result = ''
	result += '<?xml version="1.0" encoding="UTF-8" standalone="no" ?>\n'
	result += '<DescriptionScheme>\n'
	result += '\t<Attributes>\n'
	result += '\t\t<Attribute scope="%s" name="%s" type="Segmentation">\n' % (mainScope, segmentationAttribute)
	result += '\t\t\t<ChildScope>%s</ChildScope>\n' %(childScope)
	result += '\t\t\t<SegmentationPolicy>%s</SegmentationPolicy>\n' %("Discontinuous")
	result += '\t\t</Attribute>\n'
	result += '\t\t<Attribute scope="%s" name="%s" type="Enumerated">\n' % (childScope, labelAttribute)
	result += '\t\t\t<EnumerationValues>'
	result += " ".join(values)
	result += '</EnumerationValues>\n'
	result += '\t\t</Attribute>\n'
	result += '\t</Attributes>\n'
	result += '</DescriptionScheme>\n'
	return result
	

if __name__ == "__main__":
	# Default parameters
	filter = '.*'
	childScope = None
	labelAttribute = None

	if len(sys.argv) == 3:
		(labFileList,segmentationAttribute) = sys.argv[1:3]
	elif len(sys.argv) == 4:
		(labFileList,segmentationAttribute,filter) = sys.argv[1:4]
	elif len(sys.argv) == 5:
		(labFileList,segmentationAttribute,childScope,labelAttribute) = sys.argv[1:5]
	elif len(sys.argv) == 6:
		(labFileList,segmentationAttribute,filter,childScope,labelAttribute) = sys.argv[1:6]
	else:
		print >> sys.stderr, "Lab2Pools v1.0: Wavesurfer Lab file to CLAM Annotator pools converter"
		print >> sys.stderr
		print >> sys.stderr, "Usage: " + sys.argv[0] + " <labFile> <segmentationAttr> [<labelFilter>] [<childScope> <labelAttribute>]"
		print >> sys.stderr
		print >> sys.stderr, " <labFileList>       A list of annotation files from wave surfer ('-' means stdin)"
		print >> sys.stderr, " <segmentationAttr>  Attribute name for the segmentation"
		print >> sys.stderr, " <labelFilter>       A regular expresion to filter segments by label (default: all segments)"
		print >> sys.stderr, " <childScope>        If defined the child scope that should contain the label attribute (by default it is discarded)"
		print >> sys.stderr, " <labelAttribute>    The attribute that holds the label"
		sys.exit()

	if labFileList == '-' : 
		labFileList = sys.stdin
	else :
		labFileList = file(labFileList)

	labelValues = [];
	for labfilename in labFileList :
		labfilename = labfilename.lstrip().rstrip()
		values = Lab2Plot( 
				file(labfilename), 
				file(labfilename+".pool","w"), 
				segmentationAttribute, 
				filter, 
				childScope, labelAttribute)
		labelValues += values
	labelValues = dict.fromkeys(labelValues).keys() # unique
	labelValues.sort()
	print generateSchema("Song", segmentationAttribute, childScope, labelAttribute, labelValues)