742 lines
No EOL
19 KiB
Python
742 lines
No EOL
19 KiB
Python
import re
|
|
from argparse import ArgumentParser
|
|
from raspyrfm import *
|
|
import threading
|
|
import time
|
|
|
|
PARAM_ID = ('i', 'id')
|
|
PARAM_HOUSE = ('o', 'house')
|
|
PARAM_GROUP = ('g', 'group')
|
|
PARAM_UNIT = ('u', 'unit')
|
|
PARAM_COMMAND = ('a', 'command')
|
|
PARAM_CODE = ('c', 'code')
|
|
PARAM_DIPS = ('d', 'dips')
|
|
|
|
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 _encode_command(self, command):
|
|
if command in self._commands:
|
|
return self._commands[command]
|
|
else:
|
|
raise Exception("Invalid command")
|
|
|
|
def _decode_command(self, symbols):
|
|
for k in self._commands:
|
|
if self._commands[k] == symbols:
|
|
return k
|
|
raise Exception("Unknown command")
|
|
|
|
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 = [PARAM_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 {"code": symbols}, tb, rep
|
|
|
|
class ITTristate(TristateBase): #old intertechno
|
|
def __init__(self):
|
|
self._name = "ittristate"
|
|
TristateBase.__init__(self)
|
|
self.params = [PARAM_HOUSE, PARAM_GROUP, PARAM_UNIT, PARAM_COMMAND]
|
|
self._commands = {"on": "FF", "off": "F0"}
|
|
|
|
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 += self._encode_command(params["command"])
|
|
return self._build_frame(symbols, timebase, repetitions)
|
|
|
|
def decode(self, pulsetrain):
|
|
symbols, tb, rep = self._decode_symbols(pulsetrain[0:-2])
|
|
if symbols:
|
|
return {
|
|
"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": self._decode_command(symbols[10:12].upper()),
|
|
}, tb, rep
|
|
|
|
class Brennenstuhl(TristateBase): #old intertechno
|
|
def __init__(self):
|
|
self._name = "brennenstuhl"
|
|
TristateBase.__init__(self)
|
|
self.params = [PARAM_DIPS, PARAM_UNIT, PARAM_COMMAND]
|
|
self._commands = {"on": "0F", "off": "F0"}
|
|
|
|
def encode(self, params, timebase=None, repetitions=None):
|
|
symbols = ""
|
|
for c in params["dips"]:
|
|
symbols += '0' if c == '1' else 'F'
|
|
for i in range(4):
|
|
symbols += '0' if (int(params["unit"]) - 1) == i else 'F'
|
|
symbols += "F"
|
|
symbols += self._encode_command(params["command"])
|
|
return self._build_frame(symbols, timebase, repetitions)
|
|
|
|
def decode(self, pulsetrain):
|
|
symbols, tb, rep = self._decode_symbols(pulsetrain[0:-2])
|
|
if symbols:
|
|
dips = ""
|
|
for s in symbols[0:5]:
|
|
dips += "1" if s == '0' else "0"
|
|
unit = 0
|
|
for u in symbols[5:9]:
|
|
unit += 1
|
|
if u == '0':
|
|
break
|
|
|
|
return {
|
|
"dips": dips,
|
|
"unit": unit,
|
|
"command": self._decode_command(symbols[10:12].upper()),
|
|
}, 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, PARAM_UNIT, PARAM_COMMAND]
|
|
self._commands = {"on": "1", "off": "0"}
|
|
|
|
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 += self._encode_command(params["command"])
|
|
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:
|
|
return {
|
|
"id": int(symbols[:26], 2),
|
|
"unit": self._decode_unit(symbols[28:32]),
|
|
"command": self._decode_command(symbols[27])
|
|
}, 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, PARAM_UNIT, PARAM_COMMAND]
|
|
self._commands = {"on": "1", "learn": "1", "off": "0"}
|
|
|
|
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 += self._encode_command(params["command"])
|
|
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:
|
|
return {
|
|
"id": int(symbols[:20], 2),
|
|
"unit": self._decode_unit(symbols[21:24]),
|
|
"command": self._decode_command(symbols[20])
|
|
}, tb, rep
|
|
|
|
class Emylo(PWM1):
|
|
def __init__(self):
|
|
PWM1.__init__(self)
|
|
self._name = "emylo"
|
|
self.params = [PARAM_ID, PARAM_COMMAND]
|
|
self._commands = {'A': '0001', 'B': '0010', 'C': '0100', 'D': '1000'}
|
|
|
|
def encode(self, params, timebase=None, repetitions=None):
|
|
symbols = ""
|
|
symbols += "{:020b}".format(int(params["id"]))
|
|
symbols += self._encode_command(params["command"])
|
|
return self._build_frame(symbols, timebase, repetitions)
|
|
|
|
def decode(self, pulsetrain):
|
|
symbols, tb, rep = self._decode_symbols(pulsetrain[:-2])
|
|
if symbols:
|
|
return {
|
|
"id": int(symbols[:20], 2),
|
|
"command": self._decode_command(symbols[-4:])
|
|
}, 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,PARAM_UNIT, PARAM_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:
|
|
return {
|
|
"id": int(symbols[13:21] + symbols[22:30], 2),
|
|
"unit": int(symbols[31:39], 2) + 1,
|
|
"command": int(symbols[40:48], 2),
|
|
}, tb, rep
|
|
|
|
class Voltcraft(RcProtocol):
|
|
'''
|
|
PPM: Pulse Position Modulation
|
|
Pulse in middle of a symbol: 0, end of symbol: 1
|
|
Used by Voltcraft RC30
|
|
'''
|
|
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, PARAM_UNIT, PARAM_COMMAND]
|
|
self._commands = {"off": "000", "alloff": "100", "on": "010", "allon": "110", "dimup": "101", "dimdown": "111"}
|
|
RcProtocol.__init__(self)
|
|
|
|
def encode(self, params, timebase=None, repetitions=None):
|
|
if params["command"] in ["on", "off"]:
|
|
unit = int(params["unit"])-1
|
|
else:
|
|
unit = 3
|
|
|
|
symbols = "{:012b}".format(int(params["id"]))[::-1]
|
|
symbols += "{:02b}".format(unit)[::-1]
|
|
symbols += self._encode_command(params["command"])
|
|
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:
|
|
return {
|
|
"id": int(symbols[0:12][::-1], 2),
|
|
"unit": int(symbols[12:14][::-1], 2) + 1,
|
|
"command": self._decode_command(symbols[14:17])
|
|
}, 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 = [PARAM_CODE]
|
|
RcProtocol.__init__(self)
|
|
|
|
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'),
|
|
'101100': (-1, -1 , 'allon'), '011100': (-1, -1, 'alloff')
|
|
}
|
|
|
|
def __init__(self):
|
|
PWM2.__init__(self)
|
|
self._name = "pilota"
|
|
self.params = [PARAM_ID, PARAM_GROUP, PARAM_UNIT, PARAM_COMMAND]
|
|
|
|
def encode(self, params, timebase=None, repetitions=None):
|
|
symbols = '01'
|
|
u = None
|
|
if params["command"] in ["allon", "alloff"]:
|
|
params["unit"] = -1
|
|
params["group"] = -1
|
|
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"
|
|
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]]
|
|
return {
|
|
"id": int(symbols[8:24][::-1], 2),
|
|
"group": c[0],
|
|
"unit": c[1],
|
|
"command": c[2],
|
|
}, 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(),
|
|
Brennenstuhl(),
|
|
Intertechno(),
|
|
Hama(),
|
|
Logilight(),
|
|
Emylo(),
|
|
#PWM2(),
|
|
Voltcraft(),
|
|
#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):
|
|
res = []
|
|
succ = False
|
|
for p in protocols:
|
|
try:
|
|
params, tb, rep = p.decode(train)
|
|
if params:
|
|
succ = True
|
|
if not rep:
|
|
res.append({"protocol": p._name, "params": params})
|
|
except Exception as e:
|
|
pass
|
|
|
|
self.__rxcb(res if succ else None, train)
|
|
|
|
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 Exception as e:
|
|
print("Encode error: " + e.message)
|
|
|
|
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) |