class NodeB(object):
"""
Node B radio.
Node B is the base station or `headquarters` node with which the
AV communicates.
"""
def __init__(self):
self.data = NodeBData()
self.packet = Packet('B')
self.radio = RadioAPI()
self.tx_packet_number = 1
self.rx_packet_list = []
# self.reconfigure = False
# self.receive_stream = False
# self.send_data = False
def run(self):
"""
Run Node B.
This function starts the operation of the Node B radio,
excecuting all the one-time start-up funcitons, before handing
off to the finite state machine.
"""
parser = argparse.ArgumentParser()
parser.add_argument("-f", type=float, default=434e6, metavar='frequency', dest='frequency',
nargs=1, help="Transmit frequency (default: %(default)s)")
parser.add_argument("-m", type=str, default='gfsk', metavar='modulation', dest='modulation',
choices=['gfsk', 'fsk', 'ask'],
help="Select modulation from [%(choices)s] (default: %(default)s)")
parser.add_argument("-p" "--power", type=int, default=17, metavar='power', dest='power',
choices=[8, 11, 14, 17],
help="Select transmit power from [%(choices)s] (default: %(default)s)")
parser.add_argument("-r" "--bitrate", type=float, default=4.8e3, metavar='bitrate',
dest='bitrate', help="Set bitrate (default: %(default)s)")
args = parser.parse_args()
self.frequency = args.frequency
modulation = args.modulation
power = args.power
data_rate = args.bitrate
self.radio.startup()
self.radio.configure_radio(power, self.frequency, data_rate, modulation)
self.fsm()
def _receive_packet(self):
"""
Receive packet.
This function blocks until it receives a packet.
"""
(self.rssi_level, rx_packet) = self.radio.receive(rx_fifo_threshold=63, timeout=None)
# rx_packet = self.radio.receive(rx_fifo_threshold=63, timeout=None)
pkt_num, t, loc, flags, data = self.packet.parse_packet(rx_packet)
# print "packet received"
# print pkt_num, loc, flags, data
return pkt_num, loc, flags, data
def _send_packet(self, mode, received_packets = None):
"""
Transmit data.
Parameters
----------
mode : str
Mode of operation, used to create appropriate packet header.
One of {`ack_command` | `send_data`}.
received_packets : int, opt
Number of packets received.
"""
location = 44
if mode == 'ack_command':
self.packet.set_flags_node_b(ack_command = True)
payload = self.data.pack_data(mode)
elif mode == 'send_data':
if received_packets == None:
print "mode is `send_data`, but `received_packets` is `None`"
raise ValueError
self.packet.set_flags_node_b(send_data = True)
payload = self.data.pack_data(mode, received_packets)
else:
print 'error in _send_packet, no mode specified'
raise ValueError
tx_packet = self.packet.make_packet(self.tx_packet_number, location, payload)
self.tx_packet_number += 1
self.radio.transmit(tx_packet)
def fsm(self):
"""
Node B finite state machine.
"""
while True:
pkt_num, loc, flags, data = self._receive_packet()
if (flags & 0x80) == 0x80: # receive stream of packets
print "receiving stream of packets, pkt_num = ", pkt_num
self.rx_packet_list.append(pkt_num)
# print "received data stream packet"
continue
elif (flags & 0x40) == 0x40: # receive data update request
print "received request for data update"
received_packets = len(self.rx_packet_list)
time.sleep(0.5)
self._send_packet('send_data', received_packets)
self.rx_packet_list = []
continue
elif (flags & 0x20) == 0x20: # receive reconfiguration request
print "received request for reconfiguration"
# self.rx_packet_list = []
mod, eirp, bitrate = self.data.unpack_data('reconfig', data)
print mod, eirp, bitrate
time.sleep(1.0)
self._send_packet('ack_command')
self.radio.configure_radio(eirp, self.frequency, bitrate, mod)
print "reconfigured radio with new waveform, waiting for more packets..."
continue
else:
print bin(flags)
print "error in Node B FSM, will reset and continue"
continue
class RxOnePacket(object):
"""
Receiver.
"""
def __init__(self):
self.data = NodeBData()
self.packet = Packet('B')
self.radio = RadioAPI()
self.tx_packet_number = 1
self.rx_packet_list = []
self.rssi_level = 0
def _configure_radio(self, power, frequency, data_rate, modulation):
"""
Configure radio for operation.
"""
self.radio.configure_radio(power, frequency, data_rate, modulation)
def _receive_packet(self):
"""
Receive packet.
This function blocks until it receives a packet.
"""
(self.rssi_level, rx_packet) = self.radio.receive(rx_fifo_threshold=64,
timeout=None)
pkt_num, t, loc, flags, data = self.packet.parse_packet(rx_packet)
return pkt_num, loc, flags, data
def _send_packet(self, mode, received_packets=None):
"""
Transmit data.
Parameters
----------
mode : str
Mode of operation, used to create appropriate packet header.
One of {`ack_command` | `send_data`}.
received_packets : int, opt
Number of packets received.
"""
location = 44
if mode == 'ack_command':
self.packet.set_flags_node_b(ack_command=True)
payload = self.data.pack_data(mode)
elif mode == 'send_data':
if received_packets == None:
print "mode is `send_data`, but `received_packets` is `None`"
raise ValueError
self.packet.set_flags_node_b(send_data=True)
payload = self.data.pack_data(mode, received_packets)
else:
print 'error in _send_packet, no mode specified'
raise ValueError
tx_packet = self.packet.make_packet(self.tx_packet_number, location,
payload)
self.tx_packet_number += 1
self.radio.transmit(tx_packet)
def run(self):
"""
Run receiver.
This function starts the operation of the Node B radio,
excecuting all the one-time start-up funcitons, before handing
off to the finite state machine.
"""
parser = argparse.ArgumentParser()
parser.add_argument("-f",
type=float,
default=434e6,
metavar='frequency',
dest='frequency',
nargs=1,
help="Transmit frequency (default: %(default)s)")
parser.add_argument(
"-m",
type=str,
default='gfsk',
metavar='modulation',
dest='modulation',
choices=['gfsk', 'fsk', 'ask'],
help="Select modulation from [%(choices)s] (default: %(default)s)")
parser.add_argument(
"-p"
"--power",
type=int,
default=17,
metavar='power',
dest='power',
choices=[8, 11, 14, 17],
help=
"Select transmit power from [%(choices)s] (default: %(default)s)")
parser.add_argument("-r"
"--bitrate",
type=float,
default=4.8e3,
metavar='bitrate',
dest='bitrate',
help="Set bitrate (default: %(default)s)")
args = parser.parse_args()
self.frequency = args.frequency
modulation = args.modulation
power = args.power
data_rate = args.bitrate
self.radio.startup()
self.radio.configure_radio(power, self.frequency, data_rate,
modulation)
pkt_num, loc, flags, data = self._receive_packet()
print "packet number %d received" % (pkt_num)
def shutdown(self):
self.radio.shutdown()
class RxOnePacket(object):
"""
Receiver.
"""
def __init__(self):
self.data = NodeBData()
self.packet = Packet('B')
self.radio = RadioAPI()
self.tx_packet_number = 1
self.rx_packet_list = []
self.rssi_level = 0
def _configure_radio(self, power, frequency, data_rate, modulation):
"""
Configure radio for operation.
"""
self.radio.configure_radio(power, frequency, data_rate, modulation)
def _receive_packet(self):
"""
Receive packet.
This function blocks until it receives a packet.
"""
(self.rssi_level, rx_packet) = self.radio.receive(rx_fifo_threshold=64, timeout=None)
pkt_num, t, loc, flags, data = self.packet.parse_packet(rx_packet)
return pkt_num, loc, flags, data
def _send_packet(self, mode, received_packets = None):
"""
Transmit data.
Parameters
----------
mode : str
Mode of operation, used to create appropriate packet header.
One of {`ack_command` | `send_data`}.
received_packets : int, opt
Number of packets received.
"""
location = 44
if mode == 'ack_command':
self.packet.set_flags_node_b(ack_command = True)
payload = self.data.pack_data(mode)
elif mode == 'send_data':
if received_packets == None:
print "mode is `send_data`, but `received_packets` is `None`"
raise ValueError
self.packet.set_flags_node_b(send_data = True)
payload = self.data.pack_data(mode, received_packets)
else:
print 'error in _send_packet, no mode specified'
raise ValueError
tx_packet = self.packet.make_packet(self.tx_packet_number, location, payload)
self.tx_packet_number += 1
self.radio.transmit(tx_packet)
def run(self):
"""
Run receiver.
This function starts the operation of the Node B radio,
excecuting all the one-time start-up funcitons, before handing
off to the finite state machine.
"""
parser = argparse.ArgumentParser()
parser.add_argument("-f", type=float, default=434e6, metavar='frequency', dest='frequency',
nargs=1, help="Transmit frequency (default: %(default)s)")
parser.add_argument("-m", type=str, default='gfsk', metavar='modulation', dest='modulation',
choices=['gfsk', 'fsk', 'ask'],
help="Select modulation from [%(choices)s] (default: %(default)s)")
parser.add_argument("-p" "--power", type=int, default=17, metavar='power', dest='power',
choices=[8, 11, 14, 17],
help="Select transmit power from [%(choices)s] (default: %(default)s)")
parser.add_argument("-r" "--bitrate", type=float, default=4.8e3, metavar='bitrate',
dest='bitrate', help="Set bitrate (default: %(default)s)")
args = parser.parse_args()
self.frequency = args.frequency
modulation = args.modulation
power = args.power
data_rate = args.bitrate
self.radio.startup()
self.radio.configure_radio(power, self.frequency, data_rate, modulation)
pkt_num, loc, flags, data = self._receive_packet()
print "packet number %d received" %(pkt_num)
def shutdown(self):
self.radio.shutdown()
class NodeB(object):
"""
Node B radio.
Node B is the base station or `headquarters` node with which the
AV communicates.
"""
def __init__(self):
self.data = NodeBData()
self.packet = Packet('B')
self.radio = RadioAPI()
self.tx_packet_number = 1
self.rx_packet_list = []
# self.reconfigure = False
# self.receive_stream = False
# self.send_data = False
def run(self):
"""
Run Node B.
This function starts the operation of the Node B radio,
excecuting all the one-time start-up funcitons, before handing
off to the finite state machine.
"""
parser = argparse.ArgumentParser()
parser.add_argument("-f",
type=float,
default=434e6,
metavar='frequency',
dest='frequency',
nargs=1,
help="Transmit frequency (default: %(default)s)")
parser.add_argument(
"-m",
type=str,
default='gfsk',
metavar='modulation',
dest='modulation',
choices=['gfsk', 'fsk', 'ask'],
help="Select modulation from [%(choices)s] (default: %(default)s)")
parser.add_argument(
"-p"
"--power",
type=int,
default=17,
metavar='power',
dest='power',
choices=[8, 11, 14, 17],
help=
"Select transmit power from [%(choices)s] (default: %(default)s)")
parser.add_argument("-r"
"--bitrate",
type=float,
default=4.8e3,
metavar='bitrate',
dest='bitrate',
help="Set bitrate (default: %(default)s)")
args = parser.parse_args()
self.frequency = args.frequency
modulation = args.modulation
power = args.power
data_rate = args.bitrate
self.radio.startup()
self.radio.configure_radio(power, self.frequency, data_rate,
modulation)
self.fsm()
def _receive_packet(self):
"""
Receive packet.
This function blocks until it receives a packet.
"""
rx_packet = self.radio.receive(rx_fifo_threshold=64, timeout=None)
pkt_num, t, loc, flags, data = self.packet.parse_packet(rx_packet)
# print "packet received"
# print pkt_num, loc, flags, data
return pkt_num, loc, flags, data
def _send_packet(self, mode, received_packets=None):
"""
Transmit data.
Parameters
----------
mode : str
Mode of operation, used to create appropriate packet header.
One of {`ack_command` | `send_data`}.
received_packets : int, opt
Number of packets received.
"""
location = 44
if mode == 'ack_command':
self.packet.set_flags_node_b(ack_command=True)
payload = self.data.pack_data(mode)
elif mode == 'send_data':
if received_packets == None:
print "mode is `send_data`, but `received_packets` is `None`"
raise ValueError
self.packet.set_flags_node_b(send_data=True)
payload = self.data.pack_data(mode, received_packets)
else:
print 'error in _send_packet, no mode specified'
raise ValueError
tx_packet = self.packet.make_packet(self.tx_packet_number, location,
payload)
self.tx_packet_number += 1
self.radio.transmit(tx_packet)
def fsm(self):
"""
Node B finite state machine.
"""
while True:
pkt_num, loc, flags, data = self._receive_packet()
if (flags & 0x80) == 0x80: # receive stream of packets
print "receiving stream of packets, pkt_num = ", pkt_num
self.rx_packet_list.append(pkt_num)
# print "received data stream packet"
continue
elif (flags & 0x40) == 0x40: # receive data update request
print "received request for data update"
received_packets = len(self.rx_packet_list)
time.sleep(0.5)
self._send_packet('send_data', received_packets)
self.rx_packet_list = []
continue
elif (flags & 0x20) == 0x20: # receive reconfiguration request
print "received request for reconfiguration"
# self.rx_packet_list = []
mod, eirp, bitrate = self.data.unpack_data('reconfig', data)
print mod, eirp, bitrate
time.sleep(1.0)
self._send_packet('ack_command')
self.radio.configure_radio(eirp, self.frequency, bitrate, mod)
print "reconfigured radio with new waveform, waiting for more packets..."
continue
else:
print bin(flags)
print "error in Node B FSM, will reset and continue"
continue
class StandAloneRadioB(object):
def __init__(self):
self.kb = KnowledgeBase()
self.data = NodeBData()
self.packet = Packet('B')
self.radio = RadioAPI()
self.tx_packet_number = 1
self.rx_packet_list = []
def _configure_radio(self, power, frequency, data_rate, modulation):
"""
Configure radio for operation.
"""
self.radio.configure_radio(power, frequency, data_rate, modulation)
def _send_packet(self):
"""
Transmit packet.
"""
self.packet.set_flags_node_b(ack_packet=True)
location = self.kb.get_state()['current_location']
payload = self.data.pack_data(self.rx_packet_number, self.goodput)
tx_packet = self.packet.make_packet(self.tx_packet_number, location,
payload)
self.tx_packet_number += 1
# print "processing latency = %f" %(time.time()-self.tic,)
self.radio.transmit(tx_packet)
# print "reply transmitted"
def _receive_packet(self):
"""
Receive packet.
"""
rx_packet = self.radio.receive(rx_fifo_threshold=64, timeout=None)
# self.tic = time.time()
if rx_packet == []: # this occurs when timeout has been exceeded
return
else:
self.rx_packet_number, time_stamp, location, flags, data = self.packet.parse_packet(
rx_packet)
# print "packet_number=%d time_stamp=%f location=%d flags=0x%x" %(packet_number, time_stamp,
# location, flags)
del_time = time.time() - time_stamp
self.goodput = 50 * 8 / del_time
print "goodput for packet #%d = %f bits/second" % (
self.rx_packet_number, self.goodput)
self.rx_packet_list.append(self.rx_packet_number)
def _listen(self):
"""
Listen before talk.
"""
status = self.radio.listen(rssi_threshold=100, timeout=1.0)
if status == 'clear':
print "channel clear"
def _fsm(self):
self._listen()
time.sleep(0.01)
self._receive_packet()
time.sleep(0.3) # to account for tx->rx processing latency in node_a
self._send_packet()
time.sleep(0.01)
def run(self):
"""
Run the radio subsystem.
"""
self.radio.startup()
default_radio_profile = location = self.kb.get_state(
)['default_radio_profile']
power = default_radio_profile['power']
frequency = default_radio_profile['frequency']
data_rate = default_radio_profile['data_rate']
modulation = default_radio_profile['modulation']
self._configure_radio(power, frequency, data_rate, modulation)
state = "listen"
while True:
self._fsm()
# if state == "listen":
# self._listen()
# state = "receive"
# time.sleep(0.1)
# elif state == "receive":
# self._receive_packet()
# state = "send"
# time.sleep(0.1)
# elif state == "send":
# self._send_packet()
# state = "listen"
# time.sleep(0.1)
# else:
# print "+++ Melon melon melon +++"
# state = "listen"
def shutdown(self):
print "\n\n\n"
print "number of received packets = %d" % (len(self.rx_packet_list), )
print "\n\n\n"
self.radio.shutdown()