# compute the two scaled butterfly equations
# A o--->----o---->---
#    \      /
#     \    /
#      \  /
#       \/
#       /\
#      /  \
#     /    \
#    /      \
# B o--->----o---->---
#      (-1)      (W)

from efficient_complex_multiplier import EfficientComplexMultiplier

class ButterflyProcessor(object):
	"""docstring for ClassName"""
	def __init__(self, scale):
		super(ButterflyProcessor, self).__init__()
		self.scale = scale
		self.efficient_complex_multiplier = EfficientComplexMultiplier(0);

	def butterfly(self, Are, Aim, Bre, Bim, C, C_plus_S, C_minus_S):
		if self.scale == 1:
			res = self.butterfly_scaled(Are, Aim, Bre, Bim, C, C_plus_S, C_minus_S)
		else:
			res = self.butterfly_not_scaled(Are, Aim, Bre, Bim, C, C_plus_S, C_minus_S)
		return res

	def butterfly_not_scaled(self, Are, Aim, Bre, Bim, C, C_plus_S, C_minus_S):
		out1_re = ( Are + Bre )
		out1_im = ( Aim + Bim )
		tmp_re = ( Are - Bre )
		tmp_im = ( Aim - Bim )
		res = self.efficient_complex_multiplier.multiply(tmp_re, tmp_im, C, C_plus_S, C_minus_S)
		out2_re = res[0]
		out2_im = res[1]
		out1 = out1_re + 1j * out1_im
		out2 = out2_re + 1j * out2_im
		return [out1, out2]

	def butterfly_scaled(self, Are, Aim, Bre, Bim, C, C_plus_S, C_minus_S):
		out1_re = ( Are + Bre ) / 2
		out1_im = ( Aim + Bim ) / 2
		tmp_re = ( Are - Bre ) / 2
		tmp_im = ( Aim - Bim ) / 2
		res = self.efficient_complex_multiplier.multiply(tmp_re, tmp_im, C, C_plus_S, C_minus_S)
		out2_re = res[0]
		out2_im = res[1]
		out1 = out1_re + 1j * out1_im
		out2 = out2_re + 1j * out2_im
		return [out1, out2]