# 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 re
import cStringIO
import sys

class Aggregator :
	class Exception :
		def __init__(self, message) :
			self.what = message
		def __str__(self) :
			return self.what

	class Copier :
		regexp = re.compile("^copy\s*(\d+)\s*(\w*)::(\w*)\s*(\w*)::(\w*)")
		def __init__(self, aggregator, source, scope, attribute, targetScope, targetAttribute) :
			self.aggregator = aggregator
			self.source = int(source)
			self.scope = scope
			self.attribute = attribute
			self.targetScope = targetScope
			self.targetAttribute = targetAttribute

		def apply(self) :
			target = self.aggregator.target
			source = self.aggregator.sources[self.source-1]
			target.InsertAttribute(source,
				self.scope, self.attribute,
				self.targetScope, self.targetAttribute)
			target.TranslateChildScope(self.targetScope, self.targetAttribute, self.aggregator.scopeMappings[self.source])

		def dump(self, file) :
			print >> file, "copy", self.source, \
				(self.scope + "::" + self.attribute), \
				(self.targetScope + "::" + self.targetAttribute)

	def __init__(self, script) :
		errors = ""
		self.copiers = []
		self.scopeMappings = {}
		self.requiredSources = 0
		lineNumber = 0
		for line in script :
			lineNumber=lineNumber+1
			usefulContent = line.split("#")[0].strip()
			if usefulContent=="": continue
			match = Aggregator.Copier.regexp.match(usefulContent)
			if not match :
				errors += "Parser error at line %d\n>%s"%(lineNumber,line)
				continue
			(source,scope,attribute,targetScope,targetAttribute) = match.groups()
			source = int(source)
			self.copiers.append(
				Aggregator.Copier(self, source,
					scope, attribute,
					targetScope, targetAttribute))
			if self.requiredSources < source :
				self.requiredSources = source
			try :
				sourceScopeMappings = self.scopeMappings[source]
			except :
				self.scopeMappings[source] = {}
				sourceScopeMappings = self.scopeMappings[source]
			sourceScopeMappings[scope] = targetScope

		if errors != "" :
			raise Aggregator.Exception(errors)

	def run(self, target, sources) :
		if len(sources) < self.requiredSources :
			raise "The aggregation script requires %d sources but only %d provided"% \
				(self.requiredSources,len(sources))
		self.target = target
		self.sources = sources
		for copier in self.copiers :
			copier.apply()

	def dump(self, file) :
		for copier in self.copiers :
			copier.dump(file)