from gtoolkit_bridge.PythonBridge.telemetry import gtTrace
from gtoolkit_bridge.PythonBridge.telemetry import TelemetrySignal
from gtoolkit_bridge import gtView

class PivotComparisonSignal(TelemetrySignal):
	def __init__(self, message, element, pivot):
		super().__init__(message)
		self.element = element
		self.pivot = pivot

	@gtView
	def gtViewPivotComparison(self, aBuilder):
		clist = aBuilder.columnedList()
		clist.title("PivotComparison")
		clist.priority(5)
		clist.items(lambda: [ ("message",self.message), ("element",self.element), ("pivot",self.pivot) ])
		clist.column("property", lambda each: each[0])
		clist.column("value", lambda each: each[1])
		return clist
		
class LampSort:
	def __init__(self, data):
		self.data = data

	def __copy__(self):
		return LampSort(self.data.copy())

	def __repr__(self):
		return 'LampSort'+(self.data.__repr__())

	@gtTrace
	def sort(self):
		intervals = set([range(0, len(self.data))])
		while intervals:
			selectedInterval = self.selectInterval(intervals)
			if len(selectedInterval) > 1:
				self.joinIntervals(intervals, self.partition(selectedInterval))
		return self.data
		
	@gtTrace
	def partition(self, interval):
		pivot = self.selectPivot(interval)
		TelemetrySignal("stash_pivot")
		self.swap(interval.start, interval.stop-1)
		pivotIndex = interval.start
		for i in range(interval.start, interval.stop-1):
			if self.data[i] < pivot:
				PivotComparisonSignal("smaller_than_pivot", self.data[i], pivot)
				self.swap(i, pivotIndex)
				pivotIndex += 1
			else:
				PivotComparisonSignal("larger_than_pivot", self.data[i], pivot)
		TelemetrySignal("restore_pivot")
		self.swap(interval.stop-1, pivotIndex)
		return self.splitInterval(interval, pivotIndex)
	
	@gtTrace
	def selectInterval(self, intervals):
		return intervals.pop()
		
	@gtTrace
	def joinIntervals(self, intervals, newIntervals):
		intervals.update(newIntervals)
	
	@gtTrace
	def selectPivot(self, interval):
		return self.data[interval.start]
		
	@gtTrace
	def splitInterval(self, interval, pivotIndex):
		left = range(interval.start, pivotIndex)
		right = range(pivotIndex+1, interval.stop)
		return [left, right]
		
	@gtTrace
	def swap(self, index1, index2):
		tmp = self.data[index1]
		self.data[index1] = self.data[index2]
		self.data[index2] = tmp