dpdklibs_udp_receiver.py

Go to the documentation of this file.

#!/usr/bin/env python
import socket
import sys
import binascii
import fddetdataformats
import time
import click
 
N_STREAM = 128
FRAME_SIZE = 7200
FRAME_TS_GAP = 2048
FRAME_TS_GAP = 2000
 
def print_header(wib_frame,prefix="\t"):
    header = wib_frame.get_daqheader()
    print(f'{prefix}Version: 0x{header.version:x}')
    print(f'{prefix}Detector ID: 0x{header.det_id:x}')
    print(f'{prefix}(Crate,Slot,Stream): (0x{header.crate_id:x},0x{header.slot_id:x},0x{header.stream_id:x})')
    print(f'{prefix}Timestamp: 0x{header.timestamp:x}')
    print(f'{prefix}Seq ID: {header.seq_id}') 
    print(f'{prefix}Block length: 0x{header.block_length:x}')
 
def dump_data(data):
    print(f'Size of the message received: {len(data)}')
    data2=data
 
 
    # if len(data)%2 ==1:
        # data2=data[0:-1]
    # print("\n".join(str(binascii.hexlify(data2,' ', bytes_per_sep=8)).split(' ')))
    n_word = (len(data) // 8) +len(data) % 8
    for i in range(n_word):
        w = int.from_bytes(data[i*8:(i+1)*8], byteorder='little', signed=False)
        print(f"0x{w:016x}")
 
@click.command()
@click.option('-d', '--dump', is_flag=True, default=False)
@click.option('-c', '--count', type=int, default=None)
@click.option('-p', '--port', type=int, default=0x4444)
@click.option('-g', '--gap', type=int, default=None)
@click.option('-f', '--frame-type', type=click.Choice(['wib', 'tde']), default='wib')
def main(dump, count, port, gap, frame_type):
    s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM, 0)
 
    s.bind(('', port))
 
    prev_stream = {}
    i=0
    dtstart = time.time()
    dtlast = dtstart
    sampling = 100000
    match frame_type:
        case 'wib':
            frame_class = fddetdataformats.WIBEthFrame
            port = port if not port is None else 0x4444
            gap = gap if not gap is None else 2048
        case 'tde':
            frame_class = fddetdataformats.TDEEthFrame
            port = port if not port is None else 54323
            gap = gap if not gap is None else 2000
            
    print('Starting receiver')
    while (count==None or i<count):
    # while i<10:
        data, address = s.recvfrom(20000)
        wf = frame_class(data)
        header = wf.get_daqheader()
 
 
        # hdr_id = header.stream_id
        hdr_id = (header.det_id, header.crate_id, header.slot_id, header.stream_id)
        
        # if hdr_id < N_STREAM:
        stream_ts = header.timestamp
        # print(hdr_id, header.seq_id, hex(stream_ts))
        if dump:
            dump_data(data[0:32])
 
        if hdr_id not in prev_stream:
            pass
        else:
            prev_strm_ts = prev_stream[hdr_id]
            if (stream_ts - prev_strm_ts) != FRAME_TS_GAP:
                print(f'delta_ts {stream_ts-prev_strm_ts} for {hdr_id} ')
 
        
        # if prev_stream[hdr_id] is None:
            # pass
        # elif (stream_ts-prev_stream[hdr_id]) != 2048:
            # print(f'delta_ts {stream_ts-prev_stream[hdr_id]} for det {header.det_id} strm {hdr_id} ')
        prev_stream[hdr_id] = stream_ts
 
 
        i+=1;
        if i%100000 ==0:
            dtnow = time.time()
            avg_throughput = FRAME_SIZE*i/(1000000*(dtnow-dtstart))
            throughput = FRAME_SIZE*sampling/(1000000*(dtnow-dtlast))
            print(f'Received {i} packets; throughput = {throughput:.3f} MB/s [avg = {avg_throughput:.3f} MB/s]')
            dtlast = dtnow
            print_header(wf)
            for k,ts in prev_stream.items():
                if ts is None:
                    continue
                print(f"stream {k}: last_ts {ts}")
 
            # b = data
            # for i in rnage(len(b)//8):
            #     w = [f"{s:02x}" for s in b[8*i:8*(i+1)]]
            #     w.reverse()
            #     print("0x"+''.join(w))
 
if __name__ == '__main__':
    main()