aboutsummaryrefslogtreecommitdiffstats
path: root/mpm/python/aurora_bist_test.py
blob: 9fe6f10b41831a6634d57d065d774dcf7f59733a (plain)
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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
#!/usr/bin/env python3
#
# Copyright 2017 Ettus Research, a National Instruments Company
#
# SPDX-License-Identifier: GPL-3.0-or-later
#
"""
Aurora BIST command line utility
"""

from __future__ import print_function
import argparse
import usrp_mpm as mpm
from usrp_mpm.uio import UIO
from usrp_mpm.aurora_control import AuroraControl

########################################################################
# command line options
########################################################################
def parse_args():
    " Create argparser, return args "
    parser = argparse.ArgumentParser(
        description='Controller for Ettus Aurora BIST Engine'
    )
    parser.add_argument(
        '--uio-dev', default='misc-enet-regs0',
        help='UIO device for master device peeks and pokes'
    )
    parser.add_argument(
        '--base-addr', type=int, default=0,
        help='Base address for register read/writes'
    )
    parser.add_argument(
        '--slave-uio-dev', default=None,
        help='UIO device for slave device peeks and pokes'
    )
    parser.add_argument(
        '--slave-base-addr', type=int, default=0,
        help='Base address for register read/writes on slave'
    )
    parser.add_argument(
        '--test', default='ber', choices=['ber', 'latency'],
        help='Type of test to run'
    )
    parser.add_argument(
        '--duration', type=int, default=10, help='Duration of test in seconds'
    )
    parser.add_argument(
        '--rate', type=int, default=1245, help='BIST throughput in MB/s'
    )
    parser.add_argument(
        '--loopback', action="store_true",
        help="Don't run a test, but set this Aurora into loopback mode"
    )
    return parser.parse_args()

########################################################################
# main
########################################################################
def main():
    " Dawaj, dawaj! "
    args = parse_args()
    # Initialize logger for downstream components
    mpm.get_main_logger().getChild('main')
    master_core = AuroraControl(
        UIO(label=args.uio_dev, read_only=False),
        args.base_addr,
    )
    slave_core = None if args.slave_uio_dev is None else AuroraControl(
        UIO(label=args.slave_uio_dev, read_only=False),
        args.slave_base_addr,
    )
    if args.loopback:
        master_core.reset_core()
        master_core.set_loopback(enable=True)
        return True
    # Run BIST
    if args.test == 'ber':
        print("Performing BER BIST test.")
        master_core.run_ber_loopback_bist(
            args.duration,
            args.rate * 8e6,
            slave_core,
        )
    else:
        print("Performing Latency BIST test.")
        master_core.run_latency_loopback_bist(
            args.duration,
            args.rate * 8e6,
            slave_core,
        )

if __name__ == '__main__':
    exit(not main())