Initial commit

applications/compaan/designs/compaan_io_test_fn/tb/python/stress_test.py

+#! /usr/bin/env python
+###############################################################################
+#
+# Copyright (dC) 2015
+# ASTRON (Netherlands Institute for Radio Astronomy) <http://www.astron.nl/>
+# P.O.Box 2, 7990 AA Dwingeloo, The Netherlands
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+#
+###############################################################################
+
+# Purpose: 
+# . Test Compaan's capability to send data to a terminal
+
+
+from common import *
+from random import randint
+import sys
+import threading
+import test_case
+import node_io
+import pi_diag_block_gen
+import pi_diag_data_buffer
+import pi_dp_offload_tx_hdr_dat_compaan_unb1_10g_terminal
+import pi_eth
+import pi_compaan
+from eth import *
+
+################################################################################
+## Constants and variables
+################################################################################
+# System definitions
+c_10g_data_w          = 32 # <= 32 for terminal
+c_blocksize           = 8 #  = frame_len
+c_nof_blocks_per_sync = 200000
+
+# Block gen definitions
+c_bg_ram_size         = 512
+c_bg_nof_streams      = 1
+c_bg_gap_size         = 3000
+c_write_block_gen     = True
+
+# UDP header definitions
+eth_src_mac           = 0x2286080008    # 10G MAC base address for UniBoard 
+eth_dst_mac           = 0x074306C700    #0x00074306C700  # 10G MAC address DOP36
+ip_src_addr_fn0       = 0xc0a80164      # 0xc0a80164 = 192.168.1.100
+ip_src_addr_fn1       = 0xc0a80165      # 0xc0a80165 = 192.168.1.101
+ip_src_addr_fn2       = 0xc0a80166      # 0xc0a80164 = 192.168.1.102
+ip_src_addr_fn3       = 0xc0a80167      # 0xc0a80165 = 192.168.1.103
+ip_dst_addr           = 0xc0a80102      # 0xc0a80102 = 192.168.1.2 = IP-address 10G in DOP36
+
+genLock   = threading.Lock()
+checkLock = threading.Lock()
+
+# Instantiate testcase and IO
+tc   = test_case.Testcase('TB -', '')
+io   = node_io.NodeIO(tc.nodeImages, tc.base_ip)
+
+# Instantiate compaan
+ca = pi_compaan.PiCompaan(tc, io, nof_inst=1)
+
+# Create block generator/data buffer instance
+bg = pi_diag_block_gen.PiDiagBlockGen(tc, io, c_bg_nof_streams, c_bg_ram_size, tc.nodeFn1Nrs )
+db = pi_diag_data_buffer.PiDiagDataBuffer(tc, io, instanceName = '', nofStreams=c_bg_nof_streams, ramSizePerStream=2048, nodeNr = tc.nodeBn0Nrs ) #nodeFn1Nrs
+
+# Create dp_offload_tx instance
+dpotx_hdr_dat = pi_dp_offload_tx_hdr_dat_compaan_unb1_10g_terminal.PiDpOffloadTxHdrDatCompaanUnb110GTerminal(tc, io, nof_inst=1)
+
+###############################################################################
+# Calculate and print the IP header checksum for FN0
+###############################################################################
+# Fixed header constants 
+IP_HEADER_LENGTH   = 20
+UDP_HEADER_LENGTH  = 8
+USR_HEADER_LENGTH  = 20
+USR_HDR_WORD_ALIGN = 2
+NOF_PAYLOAD_BYTES  = c_blocksize * 8
+
+ip_version         = 4
+ip_header_length   = 5 # 5 32b words
+ip_services        = 0
+ip_total_length    = IP_HEADER_LENGTH+UDP_HEADER_LENGTH+USR_HEADER_LENGTH+USR_HDR_WORD_ALIGN+NOF_PAYLOAD_BYTES - 7 # 6196B
+ip_identification  = 0
+ip_flags           = 2
+ip_fragment_offset = 0
+ip_time_to_live    = 127
+ip_protocol        = 17
+ip_header_checksum = 0            # to be calculated
+hdr_bits_common    = CommonBits(ip_version         ,4)  & \
+                     CommonBits(ip_header_length   ,4)  & \
+                     CommonBits(ip_services        ,8)  & \
+                     CommonBits(ip_total_length    ,16) & \
+                     CommonBits(ip_identification  ,16) & \
+                     CommonBits(ip_flags           ,3)  & \
+                     CommonBits(ip_fragment_offset ,13) & \
+                     CommonBits(ip_time_to_live    ,8)  & \
+                     CommonBits(ip_protocol        ,8)  & \
+                     CommonBits(ip_header_checksum ,16)
+
+hdr_bits_fn0       = hdr_bits_common & \
+                     CommonBits(ip_src_addr_fn0    ,32) & \
+                     CommonBits(ip_dst_addr        ,32)
+
+hdr_bits_fn1       = hdr_bits_common & \
+                     CommonBits(ip_src_addr_fn1    ,32) & \
+                     CommonBits(ip_dst_addr        ,32)
+
+hdr_bits_fn2       = hdr_bits_common & \
+                     CommonBits(ip_src_addr_fn2    ,32) & \
+                     CommonBits(ip_dst_addr        ,32)
+
+hdr_bits_fn3       = hdr_bits_common & \
+                     CommonBits(ip_src_addr_fn3    ,32) & \
+                     CommonBits(ip_dst_addr        ,32)
+
+hdr_bytes_fn0   = CommonBytes(hdr_bits_fn0.data, IP_HEADER_LENGTH)
+hdr_bytes_fn1   = CommonBytes(hdr_bits_fn1.data, IP_HEADER_LENGTH)
+hdr_bytes_fn2   = CommonBytes(hdr_bits_fn2.data, IP_HEADER_LENGTH)
+hdr_bytes_fn3   = CommonBytes(hdr_bits_fn3.data, IP_HEADER_LENGTH)
+
+tc.append_log(3, 'IP header checksum FN0: %d' % ip_hdr_checksum(hdr_bytes_fn0))
+tc.append_log(3, 'IP header checksum FN1: %d' % ip_hdr_checksum(hdr_bytes_fn1))
+tc.append_log(3, 'IP header checksum FN2: %d' % ip_hdr_checksum(hdr_bytes_fn2))
+tc.append_log(3, 'IP header checksum FN3: %d' % ip_hdr_checksum(hdr_bytes_fn3))
+
+################################################################################
+## Write settings
+################################################################################
+tc.append_log(3, 'Setting DPOTX for node FN0')
+
+dpotx_hdr_dat.write(node_nrs=tc.nodeFn0Nrs, inst_nrs=tc.gpNumbers, registers=[('eth_src_mac', eth_src_mac + 0)], regmap=dpotx_hdr_dat.regmap)
+dpotx_hdr_dat.write(node_nrs=tc.nodeFn0Nrs, inst_nrs=tc.gpNumbers, registers=[('eth_dst_mac', eth_dst_mac)], regmap=dpotx_hdr_dat.regmap)
+dpotx_hdr_dat.write(node_nrs=tc.nodeFn0Nrs, inst_nrs=tc.gpNumbers, registers=[('ip_src_addr', ip_src_addr_fn0)], regmap=dpotx_hdr_dat.regmap)
+dpotx_hdr_dat.write(node_nrs=tc.nodeFn0Nrs, inst_nrs=tc.gpNumbers, registers=[('ip_dst_addr', ip_dst_addr)], regmap=dpotx_hdr_dat.regmap)
+dpotx_hdr_dat.write(node_nrs=tc.nodeFn0Nrs, inst_nrs=tc.gpNumbers, registers=[('ip_header_checksum', ip_hdr_checksum(hdr_bytes_fn0))], regmap=dpotx_hdr_dat.regmap)
+
+dpotx_hdr_dat.read(node_nrs=tc.nodeFn0Nrs, inst_nrs=tc.gpNumbers, regmap=dpotx_hdr_dat.regmap)
+
+tc.append_log(3, 'Setting BG for node BN0')
+bg.write_block_gen_settings(samplesPerPacket=c_blocksize, blocksPerSync=c_nof_blocks_per_sync, gapSize=c_bg_gap_size, memLowAddr=0, memHighAddr=c_bg_ram_size-1, BSNInit=10)
+bg.read_block_gen_settings()
+###############################################################################
+## Data generator thread class
+###############################################################################
+class Generator(threading.Thread):
+  def __init__(self, lock, amount = 0, low = 1, high = 1000):
+    threading.Thread.__init__(self)
+    self.amount  = amount
+    self.low     = low
+    self.high    = high
+    self.lock    = lock
+    self.dataArr = []
+  def generate(self):
+    with self.lock:
+      #print('Acquired lock')
+      self.dataArr = []
+      for i in range(self.amount):
+        self.dataArr.append(randint(self.low, self.high))
+      
+class Checker(threading.Thread):
+  def __init__(self, lock):
+    threading.Thread.__init__(self)
+    self.lock     = lock
+    self.dataArr  = []
+    self.checkArr = []
+  # Set array as reference for checking
+  def setCheck(self, checkArr):
+    self.checkArr = checkArr
+  # Set array to check on and strip 0's
+  def setDataAndStrip(self, dataArr):
+    self.dataArr = filter(lambda a: a != 0, dataArr)
+  # Check with earlier given arrays  
+  def check(self):
+    for i in range(len(self.checkArr)):
+      if(self.dataArr[i] != self.checkArr[i]*2):
+        self.doNothing = True
+        #tc.set_result("FAILED")
+        #sys.exit("{} != {} *2".format(self.dataArr[i], self.checkArr[i]))
+    
+################################################################################
+## Write data to the block generator
+################################################################################
+# Define threads
+gen_t = Generator(genLock, c_bg_ram_size/2)
+che_t = Checker(checkLock)
+# Start threads
+gen_t.start() 
+che_t.start()
+# Generate initial data
+gen_t.generate()
+
+# Start by reading the current data in the databuffer to enable overwriting it
+#read_e = []
+#for i in range(1):
+#  read_e.append(flatten(db.read_data_buffer(streamNr=i, n=2048, radix='uns', width=c_10g_data_w, nofColumns=16)))
+
+while True:
+  # Wait for the data to be generated
+  gen_t.join()
+  # Copy the data so a new set can be generated in the meantime
+  dataArr = gen_t.dataArr
+  # Set as check array for Checker()
+  che_t.setCheck(gen_t.dataArr)
+  # Generate a new set of data in the background for next iteration
+  gen_t.generate()
+  
+  # Send data to the block_gen
+  print('Sending data:')
+  print(dataArr)
+  for i in range(c_bg_nof_streams):
+    bg.write_waveform_ram(data=dataArr, channelNr=i)
+  bg.write_enable()  
+  
+  do_until_ge(db.read_nof_words, ms_retry=1000, val=1024, s_timeout=3600)
+  
+  # Clear and refill db_out
+  db_out = []
+  for i in range(1):
+    db_out.append(flatten(db.read_data_buffer(streamNr=i, n=2048, radix='uns', width=c_10g_data_w, nofColumns=16)))
+  
+  che_t.setDataAndStrip(db_out[0])
+  che_t.check()
+  
+  #print db_out
+  
+  #repeat
\ No newline at end of file