import re
from argparse import ArgumentParser
from raspyrfm import *
import threading
import time

PARAM_ID = ('i', 'id')

class RcProtocol:
	def __init__(self):
		self.__numbits = 0
		self._ookdata = bytearray()
		self.__bbuf = 0
		self.__bval = 0
		self._parser = ArgumentParser()
		self._lastdecode = None
		self._lastdecodetime = 0

		sympulses = []
		for i in self._symbols:
			sympulses += self._symbols[i]
		sympulses.sort(reverse=True)
		i = 0
		while i<len(sympulses) - 1:
			if (sympulses[i] == sympulses[i+1]):
				del sympulses[i]
			else:
				i += 1
		p1 = sympulses.pop(0)
		p2 = sympulses.pop(0)
		f = (1.0 * p1 / p2 - 1) / (1.0 * p1/p2 + 2)
		self._minwidth = self._timebase - self._timebase * f
		self._maxwidth = self._timebase + self._timebase * f
		
	def _reset(self):
		self.__numbits = 0
		self._ookdata = bytearray()
		self.__bbuf = 0
		self.__bval = 0

	def _add_pulses(self, numbits):
		for num in numbits:
			self.__bval ^= 1
			for i in range(num):
				self.__bbuf <<= 1
				self.__bbuf |= self.__bval
				self.__numbits += 1
				if self.__numbits == 8:
					self._ookdata.append(self.__bbuf)
					self.__bbuf = 0
					self.__numbits = 0
					
	def _add_finish(self):
		if (self.__numbits > 0):
			self.__bval ^= 1
			self._add_pulses([8 - self.__numbits])

	def _add_symbols(self, symbols):
		for s in symbols:
			sym = self._symbols[s]
			for pulse in sym:
				self._add_pulses([pulse])

	def _match_symbol(self, pulsetrain, symbol):
		if len(pulsetrain) != len(symbol):
			return False

		sumpulse = 0
		sumstep = 0
		for i, v in enumerate(symbol):
			if not (self._minwidth <= 1.0 * pulsetrain[i] / v <= self._maxwidth):
				return
			sumpulse += pulsetrain[i]
			sumstep += v
		return (sumpulse, sumstep)

	def _decode_symbols(self, pulsetrain):
		#match symbols
		dec = ""
		pos = 0
		sumpulse = 0
		sumstep = 0
		while pos < len(pulsetrain):	
			match = None
			for s in self._symbols:
				slen = len(self._symbols[s])
				match = self._match_symbol(pulsetrain[pos:pos+slen], self._symbols[s])
				if match:
					dec += s
					pos += slen
					sumpulse += match[0]
					sumstep += match[1]
					break

			if not match:
				return None, None, None

		if re.match("^" + self._pattern + "$", dec):
			rep = True
			if (self._lastdecode != dec) or (time.time() - self._lastdecodetime > 0.5):
				self._lastdecode = dec
				rep = False
			self._lastdecodetime = time.time()
			return dec, int(1.0 * sumpulse / sumstep), rep
			
		return None, None, None
		
	def _build_frame(self, symbols, timebase=None, repetitions=None):
		self._reset()

		if hasattr(self, '_header'):
			self._add_pulses(self._header)
		self._add_symbols(symbols)
		if hasattr(self, '_footer'):
			self._add_pulses(self._footer)
		self._add_finish()
		if repetitions is None:
			repetitions = self._repetitions
		return self._ookdata * repetitions, timebase if timebase else self._timebase

	def decode(self, pulsetrain):
		pass

	def encode(self, params):
		pass

class TristateBase(RcProtocol): #Baseclass for old intertechno, Brennenstuhl, ...
	def __init__(self):
		self._timebase = 300
		self._repetitions = 4
		self._pattern = "[01fF]{12}"
		self._symbols = { 
			'0': [1, 4, 1, 4],
			'1': [4, 1, 4, 1],
			'f': [1, 4, 4, 1],
			'F': [1, 4, 4, 1],
		}
		self._footer = [1, 31]
		RcProtocol.__init__(self)

	def _encode_int(self, ival, digits):
		code = ""
		for i in range(digits):
			code += "f" if (ival & 0x01) > 0 else "0"
			ival >>= 1
		return code

	def _decode_int(self, tristateval):
		i = 0
		while tristateval != "":
			i <<= 1
			if tristateval[-1] != '0':
				i |= 1
			tristateval = tristateval[:-1]
		return i


class Tristate(TristateBase): #old intertechno
	def __init__(self):
		self._name = "tristate"
		TristateBase.__init__(self)
		self.params = [
			('c', 'code'),
		]

	def encode(self, params, timebase=None, repetitions=None):
		return self._build_frame(params["code"], timebase, repetitions)

	def decode(self, pulsetrain):
		symbols, tb, rep = self._decode_symbols(pulsetrain[0:-2])
		if symbols:
			return self._name, {"code": symbols}, tb, rep

class ITTristate(TristateBase): #old intertechno
	def __init__(self):
		self._name = "ittristate"
		TristateBase.__init__(self)
		self.params = [
			('o', 'house'), #A-P
			('g', 'group'), #1-4
			('u', 'unit'),  #1-4
			('a', 'command'), #on|off
		]

	def encode(self, params, timebase=None, repetitions=None):
		symbols = ""
		symbols += self._encode_int(ord(params["house"][0]) - ord('A'), 4)
		symbols += self._encode_int(int(params["unit"]) - 1, 2)
		symbols += self._encode_int(int(params["group"]) - 1, 2)
		symbols += "0f"
		symbols += "ff" if params["command"] == "on" else "f0"
		return self._build_frame(symbols, timebase, repetitions)

	def decode(self, pulsetrain):
		symbols, tb, rep = self._decode_symbols(pulsetrain[0:-2])
		if symbols:
			params = {
				"house": chr(self._decode_int(symbols[:4]) + ord('A')),
				"unit": self._decode_int(symbols[4:6]) + 1,
				"group": self._decode_int(symbols[6:8]) + 1,
				"command": "on" if symbols[10:12].upper() == "FF" else "off",
			}
			return self._name, params, tb, rep

class PPM1(RcProtocol): #Intertechno, Hama, ...
	'''
	PDM1: Pulse Position Modulation
	Every bit consists of 2 shortpulses. Long distance between these pulses 2 pulses -> 1, else -> 0
	Frame: header, payload, footer
	Used by Intertechno self learning, Hama, ...
	'''
	def __init__(self):
		self._timebase = 250
		self._repetitions = 4
		self._pattern = "[01]{32}"
		self._header = [1, 11]
		self._symbols = {
			'0': [1, 1, 1, 5],
			'1': [1, 5, 1, 1],
		}
		self._footer = [1, 39]
		RcProtocol.__init__(self)

class Intertechno(PPM1):
	def __init__(self):
		PPM1.__init__(self)
		self._name = "intertechno"
		self._timebase = 275
		self.params = [
			PARAM_ID,
			('u', 'unit'),
			('a', 'command'),
		]

	def _encode_unit(self, unit):
		return "{:04b}".format(int(unit) - 1)

	def _decode_unit(self, unit):
		return int(unit, 2) + 1

	def encode(self, params, timebase=None, repetitions=None):
		symbols = ""
		symbols += "{:026b}".format(int(params["id"]))
		symbols += "0" #group
		symbols += "1" if params["command"] == "on" else "0"
		symbols += self._encode_unit(params["unit"])
		return self._build_frame(symbols, timebase, repetitions)

	def decode(self, pulsetrain):
		symbols, tb, rep = self._decode_symbols(pulsetrain[2:-2])
		if symbols:
			params = {
				"id": int(symbols[:26], 2),
				"unit": self._decode_unit(symbols[28:32]),
				"command": "on" if symbols[27] == "1" else "off"
			}
			return self._name, params, tb, rep

class Hama(Intertechno):
	def __init__(self):
		Intertechno.__init__(self)
		self._name = "hama"
		self._timebase = 250
	
	def _encode_unit(self, unit):
		return "{:04b}".format(16 - int(unit))

	def _decode_unit(self, unit):
		return 16 - int(unit, 2)

		
class PWM1(RcProtocol):
	'''
	PWM1: Pulse Width Modulation
	Wide pulse -> 1, small pulse -> 0
	Frame: header, payload, footer
	Used by Emylo, Logilight, ...
	'''
	def __init__(self):
		self._timebase = 300
		self._repetitions = 6
		self._pattern = "[01]{24}"
		self._symbols = { 
			'1': [3, 1],
			'0': [1, 3],
		}
		self._footer = [1, 31]
		RcProtocol.__init__(self)

class Logilight(PWM1):
	def __init__(self):
		PWM1.__init__(self)
		self._name = "logilight"
		self.params = [
			PARAM_ID,
			('u', 'unit'),
			('a', 'command'),
		]

	def _encode_unit(self, unit):
		res = ""
		mask = 0x01
		while mask < 0x08:
			res += '1' if ((int(unit) - 1) & mask) == 0 else '0'
			mask <<= 1
		return res

	def _decode_unit(self, unit):
		i = 0
		while unit:
			i <<= 1
			if unit[-1] == '0':
				i |= 1
			unit = unit[:-1]
		return i + 1

	def encode(self, params, timebase=None, repetitions=None):
		symbols = ""
		symbols += "{:020b}".format(int(params["id"]))
		symbols += "1" if params["command"] in ["on", "learn"] else "0"
		symbols += self._encode_unit(params["unit"])
		if (params["command"] == "learn"):
			repetitions = 10
		return self._build_frame(symbols, timebase, repetitions)

	def decode(self, pulsetrain):
		symbols, tb, rep = self._decode_symbols(pulsetrain[:-2])
		if symbols:
			params = {
				"id": int(symbols[:20], 2),
				"unit": self._decode_unit(symbols[21:24]),
				"command": "on" if symbols[20] == "1" else "off"
			}
			return self._name, params, tb, rep

class Emylo(PWM1):
	def __init__(self):
		PWM1.__init__(self)
		self._name = "emylo"
		self.params = [
			PARAM_ID,
			('k', 'key'),
		]
		self.__keys = {'A': '0001', 'B': '0010', 'C': '0100', 'D': '1000'}

	def encode(self, params, timebase=None, repetitions=None):
		symbols = ""
		symbols += "{:020b}".format(int(params["id"]))
		if not params["key"] in self.__keys:
			return
		symbols += self.__keys[params["key"]]
		return self._build_frame(symbols, timebase, repetitions)

	def decode(self, pulsetrain):
		symbols, tb, rep = self._decode_symbols(pulsetrain[:-2])
		if symbols:
			key = None
			for k in self.__keys:
				if self.__keys[k] == symbols[-4:]:
					key = k
					break
			if key is None:
				return

			params = {
				"id": int(symbols[:20], 2),
				"key": key,
			}
			return self._name, params, tb, rep

class FS20(RcProtocol):
	def __init__(self):
		self._name = "fs20"
		self._timebase = 200
		self._repetitions = 6
		self._pattern = "0000000000001[01]{45}"
		self._symbols = { 
			'0': [2, 2],
			'1': [3, 3],
		}
		self._header = [2, 2] * 12 + [3, 3]
		self._footer = [1, 100]
		self.params = [
			PARAM_ID,
			('u', 'unit'),
			('a', 'command')
		]
		RcProtocol.__init__(self)
		
	def __encode_byte(self, b):
		b &= 0xFF
		result = '{:08b}'.format(b)
		par = 0
		while b:
			par ^= 1
			b &= b-1
		result += '1' if par != 0 else '0'
		return result

	def encode(self, params, timebase=None, repetitions=None):
		symbols = ""
		id = int(params["id"])
		unit = int(params["unit"]) - 1
		command = int(params["command"])
		symbols += self.__encode_byte((id >> 8))
		symbols += self.__encode_byte(id)
		symbols += self.__encode_byte(unit)
		symbols += self.__encode_byte(command)
		q = 0x06 + (id >> 8) + (id & 0xFF) + unit + command
		symbols += self.__encode_byte(q)
		return self._build_frame(symbols, timebase, repetitions)
		
	def decode(self, pulsetrain):
		symbols, tb, rep = self._decode_symbols(pulsetrain[0:-2])
		if symbols:
			params = {
				"id": int(symbols[13:21] + symbols[22:30], 2),
				"unit": int(symbols[31:39], 2) + 1,
				"command": int(symbols[40:48], 2),
			}
			return self._name, params, tb, rep

class Voltcraft(RcProtocol):
	'''
	PPM: Pulse Position Modulation
	Pulse in middle of a symbol: 0, end of symbol: 1
	Used by Voltcraft RC30
	'''
	__commands = {0: "off", 1: "alloff", 2: "on", 3: "allon", 5: "dimup", 7: "dimdown"}
	def __init__(self):
		self._name = "voltcraft"
		self._timebase = 600
		self._repetitions = 4
		self._pattern = "[01]{20}"
		self._symbols = {
			'0': [1, 2],
			'1': [2, 1],
		}
		self._header = [1]
		self._footer = [132]
		self.params = [
			PARAM_ID,
			('u', 'unit'),
			('a', 'command')
		]
		RcProtocol.__init__(self)

	def encode(self, params, timebase=None, repetitions=None):
		cmd = None
		for k in self.__commands:
			if self.__commands[k] == params["command"]:
				cmd = k
				break
			
		if not (cmd in [0, 2]):
			unit = 3
		else:
			unit = int(params["unit"])-1

		symbols = "{:012b}".format(int(params["id"]))[::-1]
		symbols += "{:02b}".format(unit)[::-1]
		symbols += "{:03b}".format(cmd)[::-1]
		symbols += "0"
		symbols += "1" if (symbols[12] == "1") ^ (symbols[14] == "1") ^ (symbols[16] == "1") else "0"
		symbols += "1" if (symbols[13] == "1") ^ (symbols[15] == "1") ^ (symbols[17] == "1") else "0"
		return self._build_frame(symbols, timebase, repetitions)

	def decode(self, pulsetrain):
		symbols, tb, rep = self._decode_symbols(pulsetrain[1:-1])
		if symbols:
			params = {
				"id": int(symbols[0:12][::-1], 2),
				"unit": int(symbols[12:14][::-1], 2) + 1,
				"command": self.__commands[int(symbols[14:17][::-1], 2)]
			}
			return self._name, params, tb, rep

class PWM2(RcProtocol): 
	'''
	PWM2: Pulse Width Modulation
	Wide pulse -> 0, small pulse -> 1
	Frame: header, payload, footer
	Used by Pilota casa
	'''
	def __init__(self):
		self._name = "pwm2"
		self._timebase = 600
		self._repetitions = 10
		self._pattern = "[01]{32}"
		self._symbols = { 
			'1': [1, 2],
			'0': [2, 1],
		}
		self._footer = [1, 11]
		self.params = [
			('c', 'code')
		]
		RcProtocol.__init__(self)

	def encode(self, params, timebase=None, repetitions=None):
		symbols = params["code"]
		return self._build_frame(symbols, timebase, repetitions)

	def decode(self, pulsetrain):
		symbols, tb, rep = self._decode_symbols(pulsetrain[:-2])
		
		if symbols:
			x1 = symbols[2:8]
			x2 = x1[5] + x1[4] + x1[3] + x1[2] + x1[1] + x1[0] 
			print(symbols, symbols[2:4], symbols[4:8], int(x1, 2), int(x2, 2))
		return
		
		if symbols:
			return {
				"protocol": self._name,
				"code": code,
				"timebase": tb,
			}, rep

class PilotaCasa(PWM2):
	__codes = {
		'110001': (1, 1, 'on'), '111110': (1, 1, 'off'),
		'011001': (1, 2, 'on'), '010001': (1, 2, 'off'),
		'101001': (1, 3, 'on'), '100001': (1, 3, 'off'),
		'111010': (2, 1, 'on'), '110010': (2, 1, 'off'),
		'010110': (2, 2, 'on'), '011010': (2, 2, 'off'),
		'100110': (2, 3, 'on'), '101010': (2, 3, 'off'),
		'110111': (3, 1, 'on'), '111011': (3, 1, 'off'),
		'011111': (3, 2, 'on'), '010111': (3, 2, 'off'),
		'101111': (3, 3, 'on'), '100111': (3, 3, 'off'),
		'111101': (4, 1, 'on'), '110101': (4, 1, 'off'),
		'010011': (4, 2, 'on'), '011101': (4, 2, 'off'),
		'100011': (4, 3, 'on'), '101101': (4, 3, 'off'),
	}

	def __init__(self):
		PWM2.__init__(self)
		self._name = "pilota"
		self.params = [
			PARAM_ID,
			('g', 'group'),
			('u', 'unit'),
			('a', 'command')
		]

	def encode(self, params, timebase=None, repetitions=None):
		symbols = '01'
		u = None
		for k, v in self.__codes.items():
			if v[0] == int(params["group"]) and v[1] == int(params["unit"]) and v[2] == params["command"]:
				u = k
				break
		symbols += u
		symbols += "{:016b}".format(int(params["id"]))[::-1]
		symbols += "11111111"
		print(symbols)
		return self._build_frame(symbols, timebase, repetitions)
		
	def decode(self, pulsetrain):
		symbols, tb, rep = self._decode_symbols(pulsetrain[:-2])

		if symbols and (symbols[2:8] in self.__codes):
			c = self.__codes[symbols[2:8]]
			params = {
				"id": int(symbols[8:24][::-1], 2),
				"group": c[0],
				"unit": c[1],
				"command": c[2],
			}
			return self._name, params, tb, rep

class PCPIR(RcProtocol): #pilota casa PIR sensor
	def __init__(self):
		self._name = "pcpir"
		self._timebase = 400
		self._repetitions = 5
		self._pattern = "[01]{12}"
		self._symbols = { 
			'1': [1, 3, 1, 3],
			'0': [1, 3, 3, 1],
		}
		RcProtocol.__init__(self)
		self._parser.add_argument("-c", "--code", required=True)

	def decode(self, pulsetrain):
		code, tb, rep = self._decode_symbols(pulsetrain[0:-2])
		if code:
			return {
				"protocol": self._name,
				"code": code,
				"timebase": tb,
			}, rep

	def encode(self, args):
		self._reset()
		self._add_symbols(args.code)
		self._add_pulses([1, 12])
		self._add_finish()
		return self._ookdata, self._timebase, self._repetitions

protocols = [
	Tristate(),
	ITTristate(),
	Intertechno(),
	Hama(),
	Logilight(),
	Emylo(),
	#PWM2(),
	Voltcraft(),
	#PDM32(),
	#PCPIR(),
	#PDM1(),
	FS20(),
	PilotaCasa()
]

def get_protocol(name):
	for p in protocols:
		if p._name == name:
			return p
	return None


RXDATARATE = 20.0 #kbit/s	
class RfmPulseTRX(threading.Thread):
	def __init__(self, module, rxcb, frequency):
		self.__rfm = RaspyRFM(module, RFM69)
		self.__rfm.set_params(
			Freq = frequency, #MHz
			Bandwidth = 500, #kHz
			SyncPattern = [],
			RssiThresh = -105, #dBm
			ModulationType = rfm69.OOK,
			OokThreshType = 1, #peak thresh
			OokPeakThreshDec = 3,
			Preamble = 0,
			TxPower = 13
		)
		self.__rxtraincb = rxcb
		self.__event = threading.Event()
		self.__event.set()
		threading.Thread.__init__(self)
		self.daemon = True
		self.start()
		
	def run(self):
		while True:
			self.__event.wait()
			self.__rfm.set_params(
				Datarate = RXDATARATE #kbit/s
			)
			self.__rfm.start_receive(self.__rxcb)
	
	def __rxcb(self, rfm):
		bit = False
		cnt = 1
		train = []
		bitfifo = 0
		while self.__event.isSet():
			fifo = rfm.read_fifo_wait(64)

			for b in fifo:
				mask = 0x80
				while mask != 0:
					newbit = (b & mask) != 0

					''' filter
					bitfifo <<= 1
					if newbit:
						bitfifo |= 1
					v = bitfifo & 0x3
					c = 0
					while v > 0:
						v &= v - 1
						c += 1
					'''

					if newbit == bit:
						cnt += 1
					else:
						if cnt < 150*RXDATARATE/1000: #<150 us
							train *= 0 #clear
						elif cnt > 4000*RXDATARATE/1000:
							if not bit:
								train.append(cnt)
								if len(train) > 20:
									self.__rxtraincb(train)
							train *= 0 #clear
						elif len(train) > 0 or bit:
							train.append(cnt)
						cnt = 1
						bit = not bit
					mask >>= 1

	def send(self, train, timebase):
		self.__event.clear()
		self.__rfm.set_params(
			Datarate = 1000.0 / timebase
		)
		self.__event.set()
		self.__rfm.send(train)

class RcTransceiver(threading.Thread):
	def __init__(self, module, frequency, rxcallback):
		threading.Thread.__init__(self)
		self.__lock = threading.Lock()
		self.__event = threading.Event()
		self.__trainbuf = []
		self.daemon = True
		self.start()
		self.__rxcb = rxcallback
		self.__rfmtrx = RfmPulseTRX(module, self.__pushPulseTrain, frequency)

	def __del__(self):
		del self.__rfmtrx
	
	def __pushPulseTrain(self, train):
		self.__lock.acquire()
		self.__trainbuf.append(list(train))
		self.__event.set()
		self.__lock.release()

	def __decode(self, train):
		dec = None
		res = []
		succ = False
		for p in protocols:
			dec = p.decode(train)
			if dec:
				print(dec)
				succ = True
		return res, succ

	def send(self, protocol, params, timebase=None, repeats=None):
		proto = get_protocol(protocol)
		if proto:
			try:
				txdata, tb = proto.encode(params, timebase, repeats)
				self.__rfmtrx.send(txdata, tb)
			except:
				print("Encode error")

	def run(self):
		while True:
			self.__event.wait()
			self.__lock.acquire()
			train = None
			if len(self.__trainbuf) > 0:
				train = self.__trainbuf.pop()
			else:
				self.__event.clear()
			self.__lock.release()
			if (train != None):
				for i, v in enumerate(train):
					train[i] = int(v * 1000 / RXDATARATE) #convert to microseconds
				self.__decode(train)