aboutsummaryrefslogtreecommitdiffstats
path: root/mpm/python/usrp_mpm/periphs.py
blob: 3a78f224ba53844f60d83f2ae45c1ab7141eb8d5 (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
#
# Copyright 2017 Ettus Research (National Instruments)
#
# 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/>.
#
"""
 Module
"""

import libpyusrp_periphs as lib
import re
import logging
log = logging.Logger("usrp_mpm.periphs")


def init_periph_manager(mb_type=None, db_types={}, fpga=None):
    # Detect motherboard type if not already specified
    if mb_type is None:
        mb_files = lib.helper.find_mb_file()
        with open(mb_files, "r") as f:
            info = "".join(f.readlines())
            device_type = re.match("^.*USRP;([-\w])+;.*", info)
        if device_type is None:
            log.fatal("Could not determine device type from {}".format(info))
            exit(1)
        mb_type = device_type.group(1)
    # Check if we have an implementation for this motherboard type
    try:
        device_class = getattr(lib, mb_type)
    except AttributeError:
        log.fatal("Motherboard class implementation for {} device not found!".format(mb_type))
        exit(1)

    # Detect daughterboard type if not already specified
    if not db_types:
        db_files = lib.helper.find_db_files()
        db_types = {}
        for db in db_files:
            with open(db, "r") as f:
                info = "".join(f.readlines())
                device_type = re.match("^.*SLOT;([\w]);DB;([-\w])+;.*", info)
                if device_type is None:
                    log.fatal("Could not determine device type from: {}".format(info))
                    exit(2)
                slot = device_type.group(1)
                db_type = device_type.group(2)
                db_types.update({slot: db_type})
    # Check if we have an implementation for the daughterboard types
    for db in db_types.values():
        try:
            getattr(lib.db, db)
        except AttributeError:
            log.fatal("Daughterboard class implementation for {} device not found!".format(db))
            exit(1)

    # Next steps
    #
    # 1. Load FPGA image
    # 2. Use motherboard and daughterboard types to load the FPGA image
    # 3. Create periph_manager object wth given mb_type + db_types information