/
source.py
55 lines (43 loc) · 1.75 KB
/
source.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
# from system
import struct
import sys
import time
import random
# from current dir
import meta_data
from my_top_block import my_top_block
from mac_layer import mac_layer
class source:
def __init__(self, options, crn_manager):
self.pktno = 0
self.options = options
self.crn_manager = crn_manager
self.tb = my_top_block(self.rx_callback, self.options)
self.tb.rxpath.set_carrier_threshold(options.carrier_threshold)
self.mac_layer_ = mac_layer(self.crn_manager)
self.routing_request_cnt = 0
def run(self):
while 1:
self.crn_manager.process_con.acquire()
if self.crn_manager.process_flag == 0:
self.crn_manager.process_con.wait()
self.crn_manager.process_con.release()
if len(self.mac_layer_.buffer) == 0 and self.crn_manager.status == 0:
for i in range(meta_data.batch_size):
self.generate_pakcage()
if self.crn_manager.status != 2:
self.mac_layer_.tx_run()
time.sleep(meta_data.min_time)
def generate_pakcage(self):
# prepare data and save to buffer
# structure:
# | packet_no | sender | receiverer | data |
# | 2 bytes | 2 bytes | 2 bytes | ...... |
pkt_size = int(meta_data.packet_size)
self.pktno += 1
pkt_sender_id = self.crn_manager.id
pkt_receiver_id = self.crn_manager.next_hop
data = (pkt_size - 6) * chr(self.pktno & 0xff)
self.mac_layer_.buffer.append([self.pktno, pkt_sender_id, pkt_receiver_id, data])
def rx_callback(self, ok, payload):
pass